Update app.py

app.py

@@ -172,20 +172,26 @@ def invoke(openai_api_key, rag_option, prompt):
             db = document_retrieval_mongodb(llm, prompt)
             completion, chain = rag_chain(llm, prompt, db)
             result = completion["result"]
+            print("###")
+            print(completion)
+            print("###")
+            print(chain.llm_chain)
+            print("###")
         else:
             completion, chain = llm_chain(llm, prompt)
             result = completion.generations[0][0].text if (completion.generations[0] != None and 
                                                            completion.generations[0][0] != None) else ""
+            print("###")
+            print(completion)
+            print("###")
+            print(chain.llm.client)
+            print("###")
+
     except Exception as e:
         err_msg = e
         raise gr.Error(e)
     finally:
         end_time_ms = round(time.time() * 1000)
-        print("###")
-        print(completion)
-        print("###")
-        print(chain)
-        print("###")
         wandb_trace(rag_option, prompt, completion, result, chain, err_msg, start_time_ms, end_time_ms)
     return result