openai-llm-rag

Build error

bstraehle commited on Dec 20, 2023

Commit

170a11d

1 Parent(s): ccc81c9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -138,7 +138,10 @@ def wandb_trace(rag_option, prompt, completion, chain, status_msg, start_time_ms
             "model": config["model"],
             "temperature": config["temperature"],
         },
-        inputs = {"rag_option": rag_option if (str(status_msg) == "") else "","prompt": str(prompt if (str(status_msg) == "") else ""),"prompt_template": str((llm_template if (rag_option == RAG_OFF) else rag_template) if (str(status_msg) == "") else ""),"docs_meta": "" if (rag_option == RAG_OFF or str(status_msg) != "") else docs_meta},
         outputs = {"result": result},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms

             "model": config["model"],
             "temperature": config["temperature"],
         },
+        inputs = {"rag_option": rag_option if (str(status_msg) == "") else "",
+                  "prompt": str(prompt if (str(status_msg) == "") else ""),
+                  "prompt_template": str((llm_template if (rag_option == RAG_OFF) else rag_template) if (str(status_msg) == "") else ""),
+                  "docs_meta": "" if (rag_option == RAG_OFF or str(status_msg) != "") else docs_meta},
         outputs = {"result": result},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms