Samarth991 commited on
Commit
c338f45
·
verified ·
1 Parent(s): 578f8e2

Update QnA.py

Browse files
Files changed (1) hide show
  1. QnA.py +4 -3
QnA.py CHANGED
@@ -86,12 +86,13 @@ def Q_A(vectorstore,question,API_KEY,compressor=False):
86
  compression_retriever = ContextualCompressionRetriever(base_compressor=compressor,base_retriever=retriever)
87
  retriever = compression_retriever
88
 
89
-
90
  if 'reliable' in question.lower() or 'relaibility' in question.lower():
91
- question_answer_chain = create_stuff_documents_chain(chat_llm, prompt_template_for_relaibility())
92
 
93
  else:
94
- question_answer_chain = create_stuff_documents_chain(chat_llm, prompt_template_to_analyze_resume())
 
 
95
 
96
  chain = create_retrieval_chain(retriever, question_answer_chain)
97
  result = chain.invoke({'input':question})
 
86
  compression_retriever = ContextualCompressionRetriever(base_compressor=compressor,base_retriever=retriever)
87
  retriever = compression_retriever
88
 
 
89
  if 'reliable' in question.lower() or 'relaibility' in question.lower():
90
+ prompt = prompt_template_for_relaibility()
91
 
92
  else:
93
+ prompt = prompt_template_to_analyze_resume()
94
+
95
+ question_answer_chain = create_stuff_documents_chain(chat_llm, prompt)
96
 
97
  chain = create_retrieval_chain(retriever, question_answer_chain)
98
  result = chain.invoke({'input':question})