Update app.py

app.py

@@ -43,7 +43,7 @@ def invoke(openai_api_key, prompt, rag_option):
         start_time_ms = round(time.time() * 1000)
 
         if (rag_option == RAG_LANGCHAIN):
-            completion, chain, cb = rag_chain(config, rag_option, prompt)
+            completion, chain, cb = rag_chain(config, prompt)
 
             result = completion["result"]
         elif (rag_option == RAG_LLAMAINDEX):
@@ -60,16 +60,16 @@ def invoke(openai_api_key, prompt, rag_option):
     finally:
         end_time_ms = round(time.time() * 1000)
         
-        trace_wandb(config,
-                    rag_option == RAG_OFF, 
-                    prompt, 
-                    completion, 
-                    result, 
-                    chain, 
-                    cb, 
-                    err_msg, 
-                    start_time_ms, 
-                    end_time_ms)
+        #trace_wandb(config,
+        #            rag_option == RAG_OFF, 
+        #            prompt, 
+        #            completion, 
+        #            result, 
+        #            chain, 
+        #            cb, 
+        #            err_msg, 
+        #            start_time_ms, 
+        #            end_time_ms)
 
     return result