Update app.py

app.py

@@ -123,9 +123,9 @@ def wandb_trace(rag_option, prompt, prompt_template, result, completion, chain_n
         status_code = "SUCCESS" if (str(status_msg) == "") else "ERROR",
         status_message = str(status_msg),
         metadata={
-            "chunk_overlap": config["chunk_overlap"] if (rag_option == "Off") else "N/A",
-            "chunk_size": config["chunk_size"] if (rag_option == "Off") else "N/A",
-            "k": config["k"] if (rag_option == "Off") else "N/A",
+            "chunk_overlap": config["chunk_overlap"] if (rag_option != "Off") else "",
+            "chunk_size": config["chunk_size"] if (rag_option != "Off") else "",
+            "k": config["k"] if (rag_option != "Off") else "N/A",
             "model": config["model"],
             "temperature": config["temperature"],
         },
@@ -133,6 +133,7 @@ def wandb_trace(rag_option, prompt, prompt_template, result, completion, chain_n
         end_time_ms = end_time_ms,
         inputs = {"rag_option": rag_option, "prompt": prompt, "prompt_template": prompt_template},
         outputs = {"result": str(result), "completion": str(completion)},
+        model_dict = {"_kind": "openai"}
     )
     trace.log("test")
     wandb.finish()
@@ -149,8 +150,8 @@ def invoke(openai_api_key, rag_option, prompt):
     prompt_template = ""
     chain_name = ""
     status_msg = ""
-    start_time_ms = round(time.time() * 1000)
     try:
+        start_time_ms = round(time.time() * 1000)
         llm = ChatOpenAI(model_name = config["model"], 
                          openai_api_key = openai_api_key, 
                          temperature = config["temperature"])