Update app.py

app.py

@@ -115,7 +115,7 @@ def rag_chain(llm, prompt, db):
     completion = rag_chain({"query": prompt})
     return completion
 
-def wandb_trace(prompt, completion, rag_option):
+def wandb_trace(rag_option, prompt, prompt_template, completion):
     wandb.init(project = "openai-llm-rag")
     trace = Trace(
         name="test",
@@ -123,7 +123,6 @@ def wandb_trace(prompt, completion, rag_option):
         #status_code=status,
         #status_message=status_message,
         metadata={
-            "rag_option": rag_option,
             "chunk_overlap": config["chunk_overlap"],
             "chunk_size": config["chunk_size"],
             "k": config["k"],
@@ -132,7 +131,7 @@ def wandb_trace(prompt, completion, rag_option):
         },
         #start_time_ms=start_time_ms,
         #end_time_ms=end_time_ms,
-        inputs={"prompt": prompt, "template": "todo"},
+        inputs={"rag_option": rag_option, "prompt": prompt, "prompt_template": prompt_template},
         outputs={"completion": str(completion)},
     )
     trace.log("test")
@@ -171,7 +170,7 @@ def invoke(openai_api_key, rag_option, prompt):
         completion = e
         raise gr.Error(e)
     finally:
-        wandb_trace(prompt_template, prompt, completion, rag_option)
+        wandb_trace(rag_option, prompt, prompt_template, completion)
     return result
 
 description = """<strong>Overview:</strong> Context-aware multimodal reasoning application using a <strong>large language model (LLM)</strong> with