Update app.py

app.py

@@ -129,18 +129,20 @@ def wandb_trace(rag_option, prompt, completion, result, chain, err_msg, start_ti
         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_name": config["model_name"],
-            "temperature": config["temperature"],
+            #"k": "" if (rag_option == RAG_OFF) else config["k"],
+            #"model_name": config["model_name"],
+            #"temperature": config["temperature"],
         },
         inputs = {"rag_option": rag_option if (str(err_msg) == "") else "",
-                  "prompt": str(prompt if (str(err_msg) == "") else "")},
-                  #"prompt_template": str((llm_template if (rag_option == RAG_OFF) else rag_template) if (str(err_msg) == "") else "")},
+                  "prompt": str(prompt if (str(err_msg) == "") else ""),
+        },
         outputs = {"result": result if (str(err_msg) == "") else "",
-                   "completion": str(completion)},
-        model_dict = {"llm_client": chain.llm.client if (rag_option == RAG_OFF) else chain.combine_documents_chain.llm_chain.llm.client,
-                      "prompt_engineered": chain.prompt if (rag_option == RAG_OFF) else chain.combine_documents_chain.llm_chain.prompt,
-                      "retriever": "" if (rag_option == RAG_OFF) else chain.retriever},
+                   "completion": str(completion),
+        },
+        model_dict = {"llm_client": str(chain.llm.client) if (rag_option == RAG_OFF) else str(chain.combine_documents_chain.llm_chain.llm.client),
+                      "chain_prompt": str(chain.prompt) if (rag_option == RAG_OFF) else str(chain.combine_documents_chain.llm_chain.prompt),
+                      "retriever": "" if (rag_option == RAG_OFF) else chain.retriever,
+        },
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms
     )