Update app.py

app.py

@@ -116,7 +116,8 @@ def rag_chain(llm, prompt, db):
     return completion, rag_chain
 
 def wandb_trace(rag_option, prompt, completion, chain, status_msg, start_time_ms, end_time_ms):
-    #print(chain.retriever)
+    if (chain != None):
+        print(chain.type)
     wandb.init(project = "openai-llm-rag")
     if (rag_option == "Off" or str(status_msg) != ""):
         result = completion
@@ -159,7 +160,7 @@ def invoke(openai_api_key, rag_option, prompt):
         raise gr.Error("Prompt is required.")
     completion = ""
     result = ""
-    chain = ""
+    chain = None
     status_msg = ""
     try:
         start_time_ms = round(time.time() * 1000)