Update app.py
Browse files
app.py
CHANGED
@@ -69,18 +69,18 @@ class AdvancedPdfChatbot:
|
|
69 |
self.overall_chain = self.CustomChain(refinement_chain=refinement_chain, qa_chain=qa_chain)
|
70 |
|
71 |
def setup_conversation_chain(self):
|
72 |
-
|
73 |
-
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
-
|
78 |
-
|
79 |
-
|
80 |
-
|
81 |
-
|
82 |
-
|
83 |
-
|
84 |
|
85 |
class CustomChain(Chain):
|
86 |
def __init__(self, refinement_chain, qa_chain):
|
|
|
69 |
self.overall_chain = self.CustomChain(refinement_chain=refinement_chain, qa_chain=qa_chain)
|
70 |
|
71 |
def setup_conversation_chain(self):
|
72 |
+
refinement_chain = LLMChain(
|
73 |
+
llm=self.refinement_llm,
|
74 |
+
prompt=self.refinement_prompt,
|
75 |
+
output_key='refined_query'
|
76 |
+
)
|
77 |
+
qa_chain = ConversationalRetrievalChain.from_llm(
|
78 |
+
self.llm,
|
79 |
+
retriever=self.db.as_retriever(),
|
80 |
+
memory=self.memory,
|
81 |
+
combine_docs_chain_kwargs={"prompt": self.prompt}
|
82 |
+
)
|
83 |
+
self.overall_chain = self.CustomChain(refinement_chain=refinement_chain, qa_chain=qa_chain)
|
84 |
|
85 |
class CustomChain(Chain):
|
86 |
def __init__(self, refinement_chain, qa_chain):
|