Spaces:
Sleeping
Sleeping
guangliang.yin commited on
Commit ·
2573026
1
Parent(s): f82d274
改为简单链
Browse files
app.py
CHANGED
|
@@ -17,6 +17,7 @@ from project.prompt.answer_by_private_prompt import (
|
|
| 17 |
)
|
| 18 |
from langchain.chains.combine_documents.refine import RefineDocumentsChain
|
| 19 |
from langchain.chains.llm import LLMChain
|
|
|
|
| 20 |
|
| 21 |
chain: Optional[Callable] = None
|
| 22 |
|
|
@@ -68,15 +69,17 @@ def web_loader(file, openai_key, puzhiai_key, zilliz_uri, user, password):
|
|
| 68 |
|
| 69 |
llm = ZhipuAILLM(model="glm-3-turbo", temperature=0.1, zhipuai_api_key=puzhiai_key)
|
| 70 |
|
| 71 |
-
initial_chain = LLMChain(llm=llm, prompt=DEFAULT_TEXT_QA_PROMPT)
|
| 72 |
-
refine_chain = LLMChain(llm=llm, prompt=DEFAULT_REFINE_PROMPT)
|
| 73 |
-
combine_documents_chain = RefineDocumentsChain(
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
|
| 77 |
-
|
| 78 |
-
|
| 79 |
-
)
|
|
|
|
|
|
|
| 80 |
|
| 81 |
chain = RetrievalQAWithSourcesChain(combine_documents_chain=combine_documents_chain,
|
| 82 |
retriever=docsearch.as_retriever())
|
|
@@ -100,7 +103,7 @@ if __name__ == "__main__":
|
|
| 100 |
"""
|
| 101 |
<h1><center>Langchain And Zilliz App</center></h1>
|
| 102 |
|
| 103 |
-
v.2.27.15.
|
| 104 |
|
| 105 |
"""
|
| 106 |
)
|
|
|
|
| 17 |
)
|
| 18 |
from langchain.chains.combine_documents.refine import RefineDocumentsChain
|
| 19 |
from langchain.chains.llm import LLMChain
|
| 20 |
+
from langchain.chains.combine_documents import create_stuff_documents_chain
|
| 21 |
|
| 22 |
chain: Optional[Callable] = None
|
| 23 |
|
|
|
|
| 69 |
|
| 70 |
llm = ZhipuAILLM(model="glm-3-turbo", temperature=0.1, zhipuai_api_key=puzhiai_key)
|
| 71 |
|
| 72 |
+
#initial_chain = LLMChain(llm=llm, prompt=DEFAULT_TEXT_QA_PROMPT)
|
| 73 |
+
#refine_chain = LLMChain(llm=llm, prompt=DEFAULT_REFINE_PROMPT)
|
| 74 |
+
#combine_documents_chain = RefineDocumentsChain(
|
| 75 |
+
# initial_llm_chain=initial_chain,
|
| 76 |
+
# refine_llm_chain=refine_chain,
|
| 77 |
+
# document_variable_name="context_str",
|
| 78 |
+
# initial_response_name="existing_answer",
|
| 79 |
+
# document_prompt=EXAMPLE_PROMPT,
|
| 80 |
+
#)
|
| 81 |
+
|
| 82 |
+
combine_documents_chain = create_stuff_documents_chain(llm, prompt=QUESTION_PROMPT)
|
| 83 |
|
| 84 |
chain = RetrievalQAWithSourcesChain(combine_documents_chain=combine_documents_chain,
|
| 85 |
retriever=docsearch.as_retriever())
|
|
|
|
| 103 |
"""
|
| 104 |
<h1><center>Langchain And Zilliz App</center></h1>
|
| 105 |
|
| 106 |
+
v.2.27.15.21
|
| 107 |
|
| 108 |
"""
|
| 109 |
)
|