Spaces:
Build error
Build error
Update trace.py
Browse files
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):
|