Spaces:

bstraehle
/

rag

Running

bstraehle commited on Dec 21, 2023

Commit

9bf30b7

1 Parent(s): 51b4724

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -107,7 +107,7 @@ def document_retrieval_mongodb(llm, prompt):
 def llm_chain(llm, prompt):
     llm_chain = LLMChain(llm = llm, prompt = LLM_CHAIN_PROMPT, verbose = False)
-    completion = llm_chain.generate({"question": prompt})
     return completion, llm_chain
 def rag_chain(llm, prompt, db):

 def llm_chain(llm, prompt):
     llm_chain = LLMChain(llm = llm, prompt = LLM_CHAIN_PROMPT, verbose = False)
+    completion = llm_chain.run({"question": prompt})
     return completion, llm_chain
 def rag_chain(llm, prompt, db):