怎么将chatgpt接入flask
时间: 2024-03-09 22:43:22 浏览: 88
将ChatGPT接入Flask可以实现一个基本的聊天机器人应用。下面是一种简单的实现方式:
1. 首先,确保你已经安装了Flask和OpenAI的Python库。
2. 创建一个Flask应用,并导入必要的库:
```python
from flask import Flask, request, jsonify
import openai
```
3. 设置OpenAI的API密钥:
```python
openai.api_key = 'YOUR_API_KEY'
```
4. 创建一个路由来处理聊天请求:
```python
app = Flask(__name__)
@app.route('/chat', methods=['POST'])
def chat():
data = request.get_json()
message = data['message']
# 调用ChatGPT进行回答
response = openai.Completion.create(
engine='text-davinci-002',
prompt=message,
max_tokens=50,
temperature=0.7,
n=1,
stop=None,
temperature=0.7
)
# 提取回答并返回
answer = response.choices[0].text.strip()
return jsonify({'answer': answer})
```
5. 运行Flask应用:
```python
if __name__ == '__main__':
app.run()
```
现在,你可以使用POST请求向`/chat`发送聊天消息,并获得ChatGPT的回答。例如,你可以使用curl命令:
```bash
curl -X POST -H "Content-Type: application/json" -d '{"message":"你好"}' http://localhost:5000/chat
```
这将返回一个JSON对象,其中包含ChatGPT的回答。
阅读全文