Update app.py

app.py

@@ -123,17 +123,17 @@ 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 "",
-            "chunk_size": config["chunk_size"] if (rag_option != "Off") else "",
-            "k": config["k"] if (rag_option != "Off") else "N/A",
+            "chunk_overlap": "" if (rag_option == "Off") else config["chunk_overlap"],
+            "chunk_size": "" if (rag_option == "Off") else config["chunk_size"],
+            "k": "" if (rag_option == "Off") else config["k"],
             "model": config["model"],
             "temperature": config["temperature"],
         },
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms,
-        inputs = {"rag_option": rag_option, "prompt": prompt, "prompt_template": prompt_template},
+        inputs = {"rag_option": rag_option, "prompt": str(prompt), "prompt_template": str(prompt_template)},
         outputs = {"result": str(result), "completion": str(completion)},
-        model_dict = {"api_type": "openai", "model_alias": "gpt-4"}
+        model_dict = {"api_type": "openai"}
     )
     trace.log("test")
     wandb.finish()