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 |
)
|