Update app.py
Browse files
app.py
CHANGED
|
@@ -137,13 +137,13 @@ def wandb_trace(rag_option, prompt, completion, status_msg, start_time_ms, end_t
|
|
| 137 |
"model": config["model"],
|
| 138 |
"temperature": config["temperature"],
|
| 139 |
},
|
| 140 |
-
inputs = {
|
| 141 |
-
|
| 142 |
-
|
| 143 |
-
outputs = {
|
| 144 |
-
|
| 145 |
-
|
| 146 |
-
|
| 147 |
start_time_ms = start_time_ms,
|
| 148 |
end_time_ms = end_time_ms
|
| 149 |
)
|
|
|
|
| 137 |
"model": config["model"],
|
| 138 |
"temperature": config["temperature"],
|
| 139 |
},
|
| 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 |
+
outputs = {"result": result,
|
| 144 |
+
"document_0": "" if (rag_option == "Off") else str(document_0),
|
| 145 |
+
"document_1": "" if (rag_option == "Off") else str(document_1),
|
| 146 |
+
"document_2": "" if (rag_option == "Off") else str(document_2)},
|
| 147 |
start_time_ms = start_time_ms,
|
| 148 |
end_time_ms = end_time_ms
|
| 149 |
)
|