Update app.py

app.py

@@ -128,8 +128,7 @@ def wandb_trace(rag_option, prompt, completion, chain, err_msg, start_time_ms, e
     wandb.init(project = "openai-llm-rag")
     trace = Trace(
         kind = "chain",
-        #name = type(chain).__name__ if (chain != None) else "",
-        name = str(chain) if (chain != None) else "",
+        name = type(chain).__name__ if (chain != None) else "",
         status_code = "success" if (str(err_msg) == "") else "error",
         status_message = str(err_msg),
         metadata = {
@@ -187,6 +186,7 @@ def invoke(openai_api_key, rag_option, prompt):
         raise gr.Error(e)
     finally:
         end_time_ms = round(time.time() * 1000)
+        log(chain)
         wandb_trace(rag_option, prompt, completion, chain, err_msg, start_time_ms, end_time_ms)
     return result