Update app.py

app.py

@@ -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.run({"question": prompt})
+    completion = llm_chain.generate({"question": prompt})
     return completion, llm_chain
 
 def rag_chain(llm, prompt, db):
@@ -179,6 +179,7 @@ def invoke(openai_api_key, rag_option, prompt):
             result = completion["result"]
         else:
             result, chain = llm_chain(llm, prompt)
+            print(result)
             completion = result
     except Exception as e:
         status_msg = e