openai-llm-rag

Build error

bstraehle commited on Dec 19, 2023

Commit

eddbb92

1 Parent(s): a33d0f3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -131,11 +131,11 @@ def wandb_trace(rag_option, prompt, prompt_template, result, completion, chain_n
         },
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms,
-        inputs = {{"rag_option": rag_option}, {"prompt": str(prompt)}, {"prompt_template": str(prompt_template)}},
-        outputs = {{"result": str(result)}, {"completion": str(completion)}},
         #model_dict = {"llm": str(llm)}
     )
-    trace.log("eval")
     wandb.finish()
 def invoke(openai_api_key, rag_option, prompt):

         },
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms,
+        inputs = {"rag_option": rag_option, "prompt": str(prompt), "prompt_template": str(prompt_template)},
+        outputs = {"result": str(result), "completion": str(completion)},
         #model_dict = {"llm": str(llm)}
     )
+    trace.log()
     wandb.finish()
 def invoke(openai_api_key, rag_option, prompt):