+
95
-

GPTSimpleVectorIndex如何追加索引文档数据?

GPTSimpleVectorIndex如何追加索引文档数据?

from llama_index import GPTSimpleVectorIndex, ServiceContext 
from langchain import OpenAI

# 初始化GPT-3模型和ServiceContext
llm = OpenAI(temperature=0.7, model_name="gpt-3.5-turbo")
service_context = ServiceContext(llm_predictor=llm)

# 构建索引
docs = []  # 存放文档内容的列表
for file in os.listdir("docs"):
    with open(os.path.join("docs", file), "r") as f:
        docs.append(f.read())
index = GPTSimpleVectorIndex.from_documents(docs, service_context=service_context)
index.save("index.pkl")  

# 读取索引
index = GPTSimpleVectorIndex.load("index.pkl")  

# 查询索引
query = "我要找一篇关于机器学习的文章"
response = index.query(query, response_mode="compact")
print(response.response)


网友回复

+
15
-

from llama_index import GPTSimpleVectorIndex

index = GPTSimpleVecto...

点击查看剩余70%

我知道答案,我要回答