Python+AI实战:用ChatGPT API 5分钟写一个问答机器人
一、准备工作
- 注册 OpenAI 账号并获取 API Key
- 安装 Python 3.8+
- 安装 openai 库:
pip install openai
二、获取 API Key
- 点右上角头像 → API Keys → Create new secret key
- 复制 Key,注意不要泄露或提交到 GitHub
三、代码实现
import openai
openai.api_key = "你的API Key"
def ask_chatgpt(question):
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[{"role": "user", "content": question}]
)
return response.choices[0].message.content
测试
answer = ask_chatgpt("什么是Python?")
print(answer)
四、进阶:加对话历史
messages = [
{"role": "system", "content": "你是一个热情的Python老师。"}
]
while True:
user_input = input("你: ")
if user_input == "quit":
break
messages.append({"role": "user", "content": user_input})
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=messages
)
answer = response.choices[0].message.content
print(f"AI: {answer}")
messages.append({"role": "assistant", "content": answer})
五、注意事项
- API 按 token 计费,注意控制提问长度
- API Key 要保密,用环境变量更安全
- 生产环境要加错误处理和重试机制