Spaces:

bstraehle
/

rag

Running

bstraehle commited on Dec 23, 2023

Commit

9e0f123

•

1 Parent(s): 4fb4308

Update trace.py

Files changed (1) hide show

trace.py CHANGED Viewed

@@ -2,6 +2,8 @@ import wandb
 from wandb.sdk.data_types.trace_tree import Trace
 def wandb_trace(rag_option,
                 prompt,
                 completion,

 from wandb.sdk.data_types.trace_tree import Trace
+WANDB_API_KEY = os.environ["WANDB_API_KEY"]
 def wandb_trace(rag_option,
                 prompt,
                 completion,