Update app.py

app.py

@@ -121,9 +121,10 @@ def wandb_trace(rag_option, prompt, completion, chain, status_msg, start_time_ms
             print("1=" + str(chain.llm))
             print("2=" + str(chain.prompt))
         if (type(chain).__name__ == "RetrievalQA"):
-            print("3=" + str(chain.llm))
-            print("4=" + str(chain.chain_type_kwargs))
-            print("5=" + str(chain.retriever))
+            print(chain)
+            #print("3=" + str(chain.llm))
+            #print("4=" + str(chain.chain_type_kwargs))
+            #print("5=" + str(chain.retriever))
     wandb.init(project = "openai-llm-rag")
     if (rag_option == "Off" or str(status_msg) != ""):
         result = completion