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 = {["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 |
)
|
|
|
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 |
)
|