bstraehle commited on
Commit
4c63329
·
1 Parent(s): 0fa9763

Update trace.py

Browse files
Files changed (1) hide show
  1. trace.py +2 -0
trace.py CHANGED
@@ -6,6 +6,7 @@ WANDB_API_KEY = os.environ["WANDB_API_KEY"]
6
 
7
  def trace_wandb(config,
8
  rag_option,
 
9
  prompt,
10
  completion,
11
  result,
@@ -21,6 +22,7 @@ def trace_wandb(config,
21
  status_code = "success" if (str(err_msg) == "") else "error",
22
  status_message = str(err_msg),
23
  inputs = {"prompt": prompt,
 
24
  "rag_option": rag_option,
25
  "config": str(config)
26
  } if (str(err_msg) == "") else {},
 
6
 
7
  def trace_wandb(config,
8
  rag_option,
9
+ prompt_template,
10
  prompt,
11
  completion,
12
  result,
 
22
  status_code = "success" if (str(err_msg) == "") else "error",
23
  status_message = str(err_msg),
24
  inputs = {"prompt": prompt,
25
+ "prompt_template": prompt_template,
26
  "rag_option": rag_option,
27
  "config": str(config)
28
  } if (str(err_msg) == "") else {},