Update app.py
Browse files
app.py
CHANGED
@@ -140,9 +140,9 @@ def wandb_trace(rag_option, prompt, completion, status_msg, start_time_ms, end_t
|
|
140 |
inputs = {"rag_option": rag_option if (str(status_msg) == "") else "",
|
141 |
"prompt": str(prompt if (str(status_msg) == "") else ""),
|
142 |
"prompt_template": str((llm_template if (rag_option == "Off") else rag_template) if (str(status_msg) == "") else ""),
|
143 |
-
|
144 |
-
|
145 |
-
|
146 |
outputs = {"result": result},
|
147 |
start_time_ms = start_time_ms,
|
148 |
end_time_ms = end_time_ms
|
|
|
140 |
inputs = {"rag_option": rag_option if (str(status_msg) == "") else "",
|
141 |
"prompt": str(prompt if (str(status_msg) == "") else ""),
|
142 |
"prompt_template": str((llm_template if (rag_option == "Off") else rag_template) if (str(status_msg) == "") else ""),
|
143 |
+
"document_0": "" if (rag_option == "Off" or str(status_msg) != "") else str(document_0),
|
144 |
+
"document_1": "" if (rag_option == "Off" or str(status_msg) != "") else str(document_1),
|
145 |
+
"document_2": "" if (rag_option == "Off" or str(status_msg) != "") else str(document_2)},
|
146 |
outputs = {"result": result},
|
147 |
start_time_ms = start_time_ms,
|
148 |
end_time_ms = end_time_ms
|