Stéphanie Kamgnia Wonkap commited on
Commit
0b0ce2d
1 Parent(s): 4055a36

fixing invoke

Browse files
Files changed (1) hide show
  1. src/generator.py +1 -1
src/generator.py CHANGED
@@ -41,6 +41,6 @@ def answer_with_rag(
41
  #print(final_prompt)
42
  # Redact an answer
43
  print("=> Generating answer...")
44
- response=retrieval_chain.invoke({'query':query})
45
 
46
  return response['answer'], response["context"]
 
41
  #print(final_prompt)
42
  # Redact an answer
43
  print("=> Generating answer...")
44
+ response=retrieval_chain.invoke(query)
45
 
46
  return response['answer'], response["context"]