当前位置：首页 > article >正文

《python每天一小段》--（1）与GPT-3.5-turbo 模型进行对话

article 2024/11/14 20:08:19

对话如图：

在这里插入图片描述

配置环境变量

APIKey如何获得这边不做说明

在Windows操作系统中，你可以按照以下步骤设置环境变量：

打开“控制面板”。
在控制面板中，选择“系统和安全”。
选择“系统”。
在系统窗口中，选择“高级系统设置”。

在弹出的对话框中，选择“环境变量”按钮。
在这里插入图片描述

在环境变量对话框中，你可以设置系统级别的环境变量（对所有用户有效）或用户级别的环境变量（只对当前用户有效）。

在相应的区域中，你可以添加、编辑或删除环境变量。点击“新建”来添加一个新的环境变量。

输入变量名和变量值。对于API密钥，变量名可以是任意的，例如OPEN_API_KEY，而变量值则是你的API密钥。

确认所有更改并关闭对话框。
在这里插入图片描述

重启主机生效

在centos7操作系统中，可以按照以下步骤设置环境变量：

登录到你的Linux 系统。
打开终端应用程序或通过 SSH 连接到服务器
使用以下命令编辑/etc/profile 文件：
```
vi /etc/profile
```
这将使用vi 编辑器打开 /etc/profile文件。你也可以使用其他文本编辑器替代 vi
在文件的末尾添加以下行：
```
export OPEN_API_KEY="your_api_key"
```
将 your_api_key替换为你的 OpenAI API 密钥。
:wq保存文件并退出编辑器。
使用以下命令重新加载/etc/profile 文件：
```
source /etc/profile
```
这将使新的环境变量生效。
验证环境变量是否已成功添加，可以使用以下命令：
```
echo $OPEN_API_KEY
```
如果环境变量已正确设置，将看到你的 API 密钥的值。

注意：环境OPEN_API_KEY名称需要与代码中一致

如环境变量名称为OPEN_API_KEY，代码也必须相同：openai.api_key = os.getenv("OPEN_API_KEY")

安装openai库

打开cmd或者Powershell

pip install openai==0.28

使用 OpenAI 的 GPT-3.5-turbo 模型来进行对话问答。

代码

import openai

import os

# 从系统环境变量中获取API密钥
openai.api_key = os.getenv("OPEN_API_KEY")

#print(os.getenv("OPEN_API_KEY"))

def getAnswerFromOpenAi(question): 
    #创建一个 GPT-3 请求
    completion = openai.ChatCompletion.create(
        model="gpt-3.5-turbo-16k",
        messages=[
            {"role": "user", "content": question}
        ]
    )
    
    return str(completion.choices[0].message.content)

while True:
        question = input('我:')
        print('openAi正在思考，请稍等...')
        answer = getAnswerFromOpenAi(question)
        print('openAi:' + answer)