Update app.py

app.py

@@ -36,7 +36,7 @@ config = {
     "chunk_overlap": 150,
     "chunk_size": 1500,
     "k": 3,
-    "model": "gpt-4",
+    "model_name": "gpt-4",
     "temperature": 0,
 }
 
@@ -128,14 +128,15 @@ def wandb_trace(rag_option, prompt, completion, chain, err_msg, start_time_ms, e
     wandb.init(project = "openai-llm-rag")
     trace = Trace(
         kind = "chain",
-        name = type(chain).__name__ if (chain != None) else "",
+        #name = type(chain).__name__ if (chain != None) else "",
+        name = str(chain) if (chain != None) else "",
         status_code = "success" if (str(err_msg) == "") else "error",
         status_message = str(err_msg),
         metadata = {
             "chunk_overlap": "" if (rag_option == RAG_OFF) else config["chunk_overlap"],
             "chunk_size": "" if (rag_option == RAG_OFF) else config["chunk_size"],
             "k": "" if (rag_option == RAG_OFF) else config["k"],
-            "model": config["model"],
+            "model_name": config["model_name"],
             "temperature": config["temperature"],
         },
         inputs = {"rag_option": rag_option if (str(err_msg) == "") else "",
@@ -162,7 +163,7 @@ def invoke(openai_api_key, rag_option, prompt):
     err_msg = ""
     try:
         start_time_ms = round(time.time() * 1000)
-        llm = ChatOpenAI(model_name = config["model"], 
+        llm = ChatOpenAI(model_name = config["model_name"], 
                          openai_api_key = openai_api_key, 
                          temperature = config["temperature"])
         if (rag_option == RAG_CHROMA):