Update app.py

app.py

@@ -12,6 +12,14 @@ RAG_OFF     = "Off"
 RAG_CHROMA  = "Chroma"
 RAG_MONGODB = "MongoDB"
 
+config = {
+    "chunk_overlap": 150,
+    "chunk_size": 1500,
+    "k": 3,
+    "model_name": "gpt-4-0613",
+    "temperature": 0,
+}
+
 def invoke(openai_api_key, rag_option, prompt):
     if (openai_api_key == ""):
         raise gr.Error("OpenAI API Key is required.")
@@ -57,7 +65,8 @@ def invoke(openai_api_key, rag_option, prompt):
     finally:
         end_time_ms = round(time.time() * 1000)
         
-        wandb_trace(rag_option == RAG_OFF, 
+        wandb_trace(config,
+                    rag_option == RAG_OFF, 
                     prompt, 
                     completion, 
                     result,