Update app.py

app.py

@@ -14,8 +14,7 @@ AGENT_LANGCHAIN  = "LangChain"
 AGENT_LLAMAINDEX = "LlamaIndex"
 
 config = {
-    "model": "gpt-4o",
-    "model_llamaindex": "gpt-4",
+    "model": "gpt-4-0613",
     "temperature": 0
 }
 
@@ -70,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