用python写一个
时间: 2023-07-01 09:06:54 浏览: 126
python 编写的一个代码
好的,以下是一个简单的 Python 代码,可以实现一个基于 ChitGPT 的 AI 对话程序:
```python
import openai
import time
# 设置 CSDN API 访问密钥
openai.api_key = "YOUR_API_KEY"
# 设置模型引擎 ID
model_engine_id = "YOUR_MODEL_ENGINE_ID"
# 创建一个新的对话 session
def create_session():
response = openai.Completion.create(
engine=model_engine_id,
prompt="Create a new session",
max_tokens=0,
n=1,
stop=None,
temperature=0.5
)
return response.choices[0].text.strip()
# 发送一条消息并获取 AI 的回复
def send_message(session_id, message):
response = openai.Completion.create(
engine=model_engine_id,
prompt=f"Session ID: {session_id}\nUser: {message}\nAI:",
max_tokens=50,
n=1,
stop=None,
temperature=0.5
)
return response.choices[0].text.strip()
# 主程序
def main():
# 创建一个新的对话 session
session_id = create_session()
print(f"Session ID: {session_id}")
while True:
# 从用户那里获取消息
message = input("You: ")
# 发送消息并获取回复
response = send_message(session_id, message)
# 打印 AI 的回复
print(f"AI: {response}")
# 等待一段时间,减慢对话速度
time.sleep(1)
if __name__ == "__main__":
main()
```
在运行代码之前,需要先从 CSDN 获取 API 访问密钥和模型引擎 ID,并替换代码中的 `YOUR_API_KEY` 和 `YOUR_MODEL_ENGINE_ID`。同时需要安装 CSDN 的 Python SDK,可以通过 `pip install openai` 命令来安装。运行代码后,程序会提示输入一条消息,然后会输出 AI 的回复,如下所示:
```
Session ID: xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx
You: Hello
AI: Hi there!
You: How are you doing?
AI: I'm doing well, thank you for asking. How about you?
You: I'm doing pretty good too
AI: That's great to hear! Is there anything you'd like to talk about?
...
```
阅读全文