python写一个gpt聊天工具
时间: 2023-08-25 11:57:16 浏览: 175
推荐一个好用的ChatGPT工具
当然,下面是一个使用 Python 编写的简单的 GPT 聊天工具的示例代码:
```python
import requests
API_KEY = 'YOUR_API_KEY'
API_URL = 'https://api.gptapi.dev/chat/completions'
def generate_response(prompt):
headers = {
'Content-Type': 'application/json',
'Authorization': f'Bearer {API_KEY}'
}
data = {
'prompt': prompt,
'max_tokens': 50, # 设置生成的回复的最大长度
'temperature': 0.8 # 控制生成文本的创造性程度,值越大越随机,值越小越保守
}
response = requests.post(API_URL, headers=headers, json=data)
response.raise_for_status()
return response.json()['choices'][0]['text']
# 主循环
while True:
user_input = input('User: ')
if user_input.lower() in ['bye', 'goodbye']:
print('ChatGPT: Goodbye!')
break
response = generate_response(user_input)
print('ChatGPT:', response)
```
请注意,你需要将 `YOUR_API_KEY` 替换为你自己在 CSDN 开发者中心获取到的 API 密钥。此外,该示例仅展示了一个简单的循环,用户输入问题后,程序将调用 GPT API 生成回复并显示在控制台上。你可以根据自己的需求进行更加复杂和定制化的开发。记得在使用 GPT API 时遵守相关的使用规范和限制。
阅读全文