Update app.py
Browse files
app.py
CHANGED
@@ -118,7 +118,7 @@ def rag_chain(llm, prompt, db):
|
|
118 |
|
119 |
def wandb_trace(rag_option, prompt, completion, status_msg, start_time_ms, end_time_ms):
|
120 |
wandb.init(project = "openai-llm-rag")
|
121 |
-
if (rag_option == "Off"
|
122 |
result = completion
|
123 |
else:
|
124 |
result = completion["result"]
|
@@ -141,9 +141,9 @@ def wandb_trace(rag_option, prompt, completion, status_msg, start_time_ms, end_t
|
|
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"
|
145 |
-
"document_1": "" if (rag_option == "Off"
|
146 |
-
"document_2": "" if (rag_option == "Off"
|
147 |
start_time_ms = start_time_ms,
|
148 |
end_time_ms = end_time_ms
|
149 |
)
|
|
|
118 |
|
119 |
def wandb_trace(rag_option, prompt, completion, status_msg, start_time_ms, end_time_ms):
|
120 |
wandb.init(project = "openai-llm-rag")
|
121 |
+
if (rag_option == "Off" or str(status_msg) != ""):
|
122 |
result = completion
|
123 |
else:
|
124 |
result = completion["result"]
|
|
|
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" or str(status_msg) != "") else str(document_0),
|
145 |
+
"document_1": "" if (rag_option == "Off" or str(status_msg) != "") else str(document_1),
|
146 |
+
"document_2": "" if (rag_option == "Off" or str(status_msg) != "") else str(document_2)},
|
147 |
start_time_ms = start_time_ms,
|
148 |
end_time_ms = end_time_ms
|
149 |
)
|