moriire commited on
Commit
a49001a
·
verified ·
1 Parent(s): 1a4faaf

Update app/llm.py

Browse files
Files changed (1) hide show
  1. app/llm.py +1 -1
app/llm.py CHANGED
@@ -43,7 +43,7 @@ class RagChat:
43
  | llm
44
  | StrOutputParser()
45
  )
46
- chain.invoke({"context": self.search, "question": question})
47
 
48
  def search(self, question):
49
  docs = vectorstore.similarity_search(question)
 
43
  | llm
44
  | StrOutputParser()
45
  )
46
+ return chain.invoke({"context": self.search(question), "question": question})
47
 
48
  def search(self, question):
49
  docs = vectorstore.similarity_search(question)