guangliang.yin commited on
Commit
cd99b75
1 Parent(s): 6dc8814

优化向量查询数量

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -45,7 +45,7 @@ def web_loader(file, puzhiai_key, zilliz_uri, user, password):
45
  loader = TextLoader(file)
46
  docs = loader.load()
47
 
48
- text_splitter = CharacterTextSplitter(chunk_size=256, chunk_overlap=0)
49
  docs = text_splitter.split_documents(docs)
50
  #embeddings = OpenAIEmbeddings(model="text-embedding-ada-002", openai_api_key=openai_key)
51
  embeddings = ZhipuAIEmbeddings(zhipuai_api_key=puzhiai_key)
@@ -129,7 +129,7 @@ def web_loader(file, puzhiai_key, zilliz_uri, user, password):
129
  )
130
 
131
  chain = RetrievalQAWithSourcesChain(combine_documents_chain=combine_documents_chain,
132
- retriever=docsearch.as_retriever(search_kwargs={'k': 1}))
133
  return "success to load data"
134
 
135
 
@@ -150,7 +150,7 @@ if __name__ == "__main__":
150
  """
151
  <h1><center>Langchain And Zilliz App</center></h1>
152
 
153
- v.2.27.18.30
154
 
155
  """
156
  )
 
45
  loader = TextLoader(file)
46
  docs = loader.load()
47
 
48
+ text_splitter = CharacterTextSplitter(chunk_size=128, chunk_overlap=0)
49
  docs = text_splitter.split_documents(docs)
50
  #embeddings = OpenAIEmbeddings(model="text-embedding-ada-002", openai_api_key=openai_key)
51
  embeddings = ZhipuAIEmbeddings(zhipuai_api_key=puzhiai_key)
 
129
  )
130
 
131
  chain = RetrievalQAWithSourcesChain(combine_documents_chain=combine_documents_chain,
132
+ retriever=docsearch.as_retriever(search_kwargs={'k': 3}))
133
  return "success to load data"
134
 
135
 
 
150
  """
151
  <h1><center>Langchain And Zilliz App</center></h1>
152
 
153
+ v.2.28.10.46
154
 
155
  """
156
  )