Update app.py

app.py

@@ -69,17 +69,17 @@ def invoke(openai_api_key, prompt, agent_option):
     finally:
         end_time_ms = round(time.time() * 1000)
         
-        #trace_wandb(
-        #    config,
-        #    agent_option,
-        #    prompt, 
-        #    completion, 
-        #    result,
-        #    callback,
-        #    err_msg, 
-        #    start_time_ms, 
-        #    end_time_ms
-        #)
+        trace_wandb(
+            config,
+            agent_option,
+            prompt, 
+            completion, 
+            result,
+            callback,
+            err_msg, 
+            start_time_ms, 
+            end_time_ms
+        )
 
     return result