openai库使用相关介绍
介绍
openai库是openai官方提供的api接口,用户可以通过该库与获取到的api-key得到相关的模型服务,并进行构建自己的ai_agent。
openai库与langchain是相同的性质,都属于ai开发的框架,只不过其中所涉及的api与服务不尽相同。具体细节不太清楚,langchain所能涉及到的公司不仅仅只有openai
该篇文章主要介绍openai库中assistant服务的介绍,其余api接口暂不涉及
其实具体详细内容可以前往查看咖哥的《大模型做应用开发 动手做AI Agent》一书
环境配置
- 首先需要安装相关的包
pip install openai
-
其次,我们需要去官网获取openai官方所提供的api-key或是TB上购买。(注意:该api-key需要是直连的,中转的api-ke部分服务不提供)
-
配置api-key到开发文件中,存在两种方法:一种是在开发文件中直接调用api-key,二是将其配置到环境变量文件(.env)中
- 在开发文件中直接调用api-key
from openai import OpenAI # 在这里输入你的 API 密钥 api_key = "your_api_key" client = OpenAI(api_key=api_key)
- 配置环境变量文件并调用该文件
#在你指定的文件夹下创建一个.txt文件,再修改名称为.env, #注意这是整个文件名为”.env“ #之后在.env文件中输入以下内容,注意保存修改后的文件 OPENAI_API_KEY="your api-key" OPENAI_BASE_URL="https://api.openai.com//v1" OPENAI_API_BASE="https://api.openai.com//v1"
之后的开发代码如下,load_dotenv会读取其中的OPENAI_API_KEY并作为参数传入OpenAI()函数中,就不需要显示调用
# 导入OpenAI库,并创建OpenAI客户端 import os from dotenv import load_dotenv load_dotenv(override=True) # 加载 .env 文件,并覆盖已有的环境变量 from openai import OpenAI client = OpenAI()
assistant服务使用介绍
基本逻辑:
1、创建assistant——2、创建线程thread——3、给线程thread上传信息message——4、运行run(创建run)——5、获取线程的消息序列进而获取assistant对消息message的处理结果
- 创建assistant
#此处也可以上传文件给模型进行处理
# 创建文件
file_path = './处理后的数据.csv'
file = client.files.create(
file=open(file_path, "rb"),
purpose='assistants',
)
# 创建一个包含这个文件的助手
#tools是指定这个助手可以使用哪些工具
assistant = client.beta.assistants.create(
instructions