Spaces:

bstraehle
/

rag

Running

bstraehle commited on Dec 20, 2023

Commit

303c50b

1 Parent(s): 979e095

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -141,9 +141,9 @@ def wandb_trace(rag_option, prompt, completion, status_msg, start_time_ms, end_t
                   "prompt": str(prompt if (str(status_msg) == "") else ""),
                   "prompt_template": str((llm_template if (rag_option == "Off") else rag_template) if (str(status_msg) == "") else "")},
         outputs = {"result": str(result),
-                   "document_0": "" if (rag_option == "Off") else str(document_0),
-                   "document_1": "" if (rag_option == "Off") else str(document_1),
-                   "document_2": "" if (rag_option == "Off") else str(document_2)},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms
     )

                   "prompt": str(prompt if (str(status_msg) == "") else ""),
                   "prompt_template": str((llm_template if (rag_option == "Off") else rag_template) if (str(status_msg) == "") else "")},
         outputs = {"result": str(result),
+                   "document_0": "" if (rag_option == "Off") else document_0,
+                   "document_1": "" if (rag_option == "Off") else document_1,
+                   "document_2": "" if (rag_option == "Off") else document_2},
         start_time_ms = start_time_ms,
         end_time_ms = end_time_ms
     )