Update app.py

app.py

@@ -128,27 +128,27 @@ def wandb_trace(rag_option, prompt, completion, result, generation_info, llm_out
         status_code = "success" if (str(err_msg) == "") else "error",
         status_message = str(err_msg),
         metadata = {
-            "chunk_overlap": config["chunk_overlap"] if (str(err_msg) == "" and rag_option != RAG_OFF) else "",
-            "chunk_size": config["chunk_size"] if (str(err_msg) == "" and rag_option != RAG_OFF) else "",
-        },
+            "chunk_overlap": config["chunk_overlap"] if (rag_option != RAG_OFF) else "",
+            "chunk_size": config["chunk_size"] if (rag_option != RAG_OFF) else "",
+        } if (str(err_msg) == "") else {},
         inputs = {"rag_option": rag_option,
                   "prompt": prompt,
-        } if (str(err_msg) == "") else "",
-        outputs = {"result": result if (str(err_msg) == "") else "",
-                   "generation_info": str(generation_info) if (str(err_msg) == "") else "",
-                   "llm_output": str(llm_output) if (str(err_msg) == "") else "",
-                   "completion": str(completion) if (str(err_msg) == "") else "",
-        },
-        model_dict = {"llm_client": (str(chain.llm.client) if (rag_option == RAG_OFF) else
-                                     str(chain.combine_documents_chain.llm_chain.llm.client)) if (str(err_msg) == "") else "",
-                      "llm_model_name": (str(chain.llm.model_name) if (rag_option == RAG_OFF) else
-                                         str(chain.combine_documents_chain.llm_chain.llm.model_name)) if (str(err_msg) == "") else "",
-                      "llm_temperature": (str(chain.llm.temperature) if (rag_option == RAG_OFF) else
-                                          str(chain.combine_documents_chain.llm_chain.llm.temperature)) if (str(err_msg) == "") else "",
-                      "chain_prompt": (str(chain.prompt) if (rag_option == RAG_OFF) else 
-                                       str(chain.combine_documents_chain.llm_chain.prompt)) if (str(err_msg) == "") else "",
-                      "chain_retriever": ("" if (rag_option == RAG_OFF) else str(chain.retriever)) if (str(err_msg) == "") else "",
-        },
+        } if (str(err_msg) == "") else {},
+        outputs = {"result": result,
+                   "generation_info": str(generation_info),
+                   "llm_output": str(llm_output),
+                   "completion": str(completion),
+        } if (str(err_msg) == "") else {},
+        model_dict = {"client": (str(chain.llm.client) if (rag_option == RAG_OFF) else
+                                 str(chain.combine_documents_chain.llm_chain.llm.client)),
+                      "model_name": (str(chain.llm.model_name) if (rag_option == RAG_OFF) else
+                                     str(chain.combine_documents_chain.llm_chain.llm.model_name)),
+                      "temperature": (str(chain.llm.temperature) if (rag_option == RAG_OFF) else
+                                      str(chain.combine_documents_chain.llm_chain.llm.temperature)),
+                      "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 str(chain.retriever)),
+        } if (str(err_msg) == "") else {},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms
     )