bstraehle commited on
Commit
930bd8b
·
1 Parent(s): 8d60a3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -132,8 +132,8 @@ def wandb_trace(prompt, completion, rag_option):
132
  },
133
  #start_time_ms=start_time_ms,
134
  #end_time_ms=end_time_ms,
135
- inputs={"prompt": prompt, "template": LLM_CHAIN_PROMPT},
136
- outputs={"completion": completion},
137
  )
138
  trace.log("test")
139
  wandb.finish()
 
132
  },
133
  #start_time_ms=start_time_ms,
134
  #end_time_ms=end_time_ms,
135
+ inputs={"prompt": prompt, "template": "todo"},
136
+ outputs={"completion": json.loads(completion)},
137
  )
138
  trace.log("test")
139
  wandb.finish()