Update app.py

app.py

@@ -37,7 +37,6 @@ def invoke(openai_api_key, prompt, rag_option):
         elif (rag_option == RAG_LLAMAINDEX):
             rag_ingestion_llamaindex(config)
     
-    chain = None
     completion = ""
     result = ""
     callback = ""
@@ -64,15 +63,15 @@ def invoke(openai_api_key, prompt, rag_option):
     finally:
         end_time_ms = round(time.time() * 1000)
         
-        #trace_wandb(config,
-        #            rag_option, 
-        #            prompt, 
-        #            completion, 
-        #            result, 
-        #            callback, 
-        #            err_msg, 
-        #            start_time_ms, 
-        #            end_time_ms)
+        trace_wandb(config,
+                    rag_option, 
+                    prompt, 
+                    completion, 
+                    result, 
+                    callback, 
+                    err_msg, 
+                    start_time_ms, 
+                    end_time_ms)
 
     return result