Update app.py

app.py

@@ -118,8 +118,13 @@ def rag_chain(llm, prompt, db):
 def wandb_trace(rag_option, prompt, completion, chain, status_msg, start_time_ms, end_time_ms):
     if (chain != None):
         if (type(chain).__name__ == "LLMChain"):
-            print("1=" + str(chain.llm))
-            print("2=" + str(chain.prompt))
+            print("1=" + str(chain.llm.client))
+            print("1=" + str(chain.llm.async_client))
+            print("1=" + str(chain.llm.model_name))
+            print("1=" + str(chain.llm.temperature))
+            print("1=" + str(chain.llm.openai_proxy))
+            print("2=" + str(chain.prompt.input_variables))
+            print("2=" + str(chain.prompt.template))
         if (type(chain).__name__ == "RetrievalQA"):
             print(chain)
             #print("3=" + str(chain.llm))