Python+AI实战:用ChatGPT API 5分钟写一个问答机器人

小飞兽 AI技术 24 次阅读 2026-07-14

一、准备工作

    • 注册 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 要保密,用环境变量更安全
  • 生产环境要加错误处理和重试机制