Pavan178 commited on
Commit
fad38ab
·
verified ·
1 Parent(s): 04218dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -12
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
- 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):
 
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):