首先,确保你已经安装了 OpenAI 的 Python 库。你可以使用以下命令来安装:
pip install openai
你需要从 OpenAI 的官网获取一个 API 密钥。然后在代码中设置该密钥。
下面是一个简单的代码示例,展示如何调用 ChatGPT API:
解释import openai# 设置你的 OpenAI API 密钥openai.api_key = 'YOUR_API_KEY'def chat_with_gpt(prompt): try: # 发送请求到 ChatGPT
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{"role": "user", "content": prompt}
]
) # 获取并返回 AI 的回复
reply = response['choices'][0]['message']['content'] return reply except Exception as e: return f"发生错误: {str(e)}"if __name__ == "__main__":
user_input = input("请输入你的问题: ")
response = chat_with_gpt(user_input) print("AI 的回复:", response)
将上述代码保存为 Python 文件(例如 chatgpt.py),然后在终端中运行:
python chatgpt.py
输入你的问题,程序将返回 ChatGPT 的回复。
API 密钥:在 openai.api_key 中替换为你的真实 API 密钥。
模型:这里使用的是 gpt-3.5-turbo,你也可以使用其他支持的模型。
异常处理:代码中包含了简单的异常处理,以帮助捕捉并打印可能发生的错误。
你可以根据需求扩展代码,例如:
持续对话:可以将 messages 中的历史对话记录下来,从而实现上下文更丰富的对话。
更多参数:openai.ChatCompletion.create 方法支持许多其他参数,如 temperature、max_tokens 等,可以根据需要调整。