Spaces:
Sleeping
Sleeping
guangliang.yin commited on
Commit ·
cd99b75
1
Parent(s): 6dc8814
优化向量查询数量
Browse files
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=
|
| 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':
|
| 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.
|
| 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 |
)
|