Hugging Face's logo

Spaces:
bstraehle
/
rag
Running

App Files Community
bstraehle commited on
Commit
fcd3a75
•
1 Parent(s): e02bd6d

Update trace.py

Browse files
Files changed (1) hide show
  1. trace.py +2 -0
trace.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  from wandb.sdk.data_types.trace_tree import Trace
2
 
3
  def wandb_trace(rag_option, prompt, completion, result, generation_info, llm_output, chain, err_msg, start_time_ms, end_time_ms):
 
1
+ import wandb
2
+
3
  from wandb.sdk.data_types.trace_tree import Trace
4
 
5
  def wandb_trace(rag_option, prompt, completion, result, generation_info, llm_output, chain, err_msg, start_time_ms, end_time_ms):