在上一节中我们学会了创建assistant,那么怎么开始使用助手构建一个聊天机器人呢?在这一节,我们就来看如何运行。
首先,先进行包的导入和相关环境变量设置
import os
from openai import OpenAI
from dotenv import load_dotenv
load_dotenv()
openai_api_key = os.getenv("OPENAI_API_KEY")
assistant_id = os.getenv("ASSISTANT_ID")
client = OpenAI(api_key=openai_api_key)
上一节中我们新创建了一个assistant并且把ID写入了.env文件,这个时候用os.getenv就可以直接从文件中获取到了
接下来我们需要创建一个vector store来放我们要上传的文件
#create vector store
vector_store = client.beta.vector_stores.create(name="Policies")
print(f"Vector Store Id - {vector_store.id}")
可以打印一下ID
接下来上传文件,在file_paths中可以放置多个文件,直接填入文件的地址即可
#upload work from home policy file
file_paths = ["work_from_home_policy.txt"]
file_streams = [open(file_path, "rb") for file_path in file_paths]
现在,把文件添加到vector store
#add files to vector store
file_batch = client.beta.vector_stores.file_batches.upload_and_poll(
vector_store_id