Yadukrishnan commited on
Commit
03f5bce
·
verified ·
1 Parent(s): 994d521

Update src/utils.py

Browse files
Files changed (1) hide show
  1. src/utils.py +2 -3
src/utils.py CHANGED
@@ -1,5 +1,4 @@
1
- def generate_response(question, retriever, llm_chain):
2
  """Generates a response using the RAG (retrieval-augmented generation) chain."""
3
- rag_chain = {"context": retriever, "question": question}
4
- response = llm_chain.invoke(rag_chain)
5
  return response.split('cutresponsefromhereplease')[-1]
 
1
+ def generate_response(question, rag_chain):
2
  """Generates a response using the RAG (retrieval-augmented generation) chain."""
3
+ response = rag_chain.invoke(question)
 
4
  return response.split('cutresponsefromhereplease')[-1]