ShynBui commited on
Commit
0ae7506
1 Parent(s): 73c3156

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -3
utils.py CHANGED
@@ -134,10 +134,9 @@ def get_question_from_summarize(summary, question, llm):
134
  new_qa_chain = new_qa_prompt | llm
135
  return new_qa_chain.invoke({'summary': summary, 'question': question}).content
136
 
137
- def get_final_answer(question, context, chat_history, prompt, llm):
138
  qa_prompt = ChatPromptTemplate.from_messages(
139
  [
140
- MessagesPlaceholder("chat_history"),
141
  ("system", prompt),
142
  ("human", '''
143
  Context: {context}
@@ -149,7 +148,7 @@ def get_final_answer(question, context, chat_history, prompt, llm):
149
 
150
  answer_chain = qa_prompt | llm
151
 
152
- answer = answer_chain.invoke({'question': question, 'context': context, 'chat_history': chat_history})
153
 
154
  return answer.content
155
 
 
134
  new_qa_chain = new_qa_prompt | llm
135
  return new_qa_chain.invoke({'summary': summary, 'question': question}).content
136
 
137
+ def get_final_answer(question, context, prompt, llm):
138
  qa_prompt = ChatPromptTemplate.from_messages(
139
  [
 
140
  ("system", prompt),
141
  ("human", '''
142
  Context: {context}
 
148
 
149
  answer_chain = qa_prompt | llm
150
 
151
+ answer = answer_chain.invoke({'question': question, 'context': context})
152
 
153
  return answer.content
154