bstraehle commited on
Commit
1112ac1
1 Parent(s): 6db720d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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" || str(status_msg) != ""):
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" || str(status_msg) != "") else str(document_0),
145
- "document_1": "" if (rag_option == "Off" || str(status_msg) != "") else str(document_1),
146
- "document_2": "" if (rag_option == "Off" || str(status_msg) != "") else str(document_2)},
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
  )