以下是一个简单的 AI 聊天程序源码示例,使用 Python 语言和 openai 的 GPT 模型:
import openai
import requests
# 设置 openai 的 API 密钥
openai.api_key = "YOUR_API_KEY"
# 与 GPT 模型建立连接
model_engine = "text-davinci-002"
prompt = "Hello, I'm an AI chatbot. What would you like to talk about today?"
# 定义一个函数,用于与 GPT 模型交互并生成回答
def generate_answer(prompt, model_engine):
# 发送请求,获取 AI 回答
response = openai.Completion.create(
engine=model_engine,
prompt=prompt,
max_tokens=2048,
n=1,
stop=None,
temperature=0.7,
)
message = response.choices[0].text
# 返回 AI 回答
return message
# 循环接收用户输入,调用 GPT 模型生成回答,并输出到终端
while True:
user_input = input("> ")
prompt += user_input
answer = generate_answer(prompt, model_engine)
prompt += answer
print(answer)
这个程序使用 OpenAI 的 GPT 模型作为 AI 引擎,接收用户的输入,调用 GPT 模型生成回答,并输出到终端。你需要替换 YOUR_API_KEY 为你的 OpenAI API 密钥,以便程序可以成功连接到 GPT 模型。
评论