Update trace.py
Browse files
trace.py
CHANGED
|
@@ -9,8 +9,6 @@ def wandb_trace(config,
|
|
| 9 |
prompt,
|
| 10 |
completion,
|
| 11 |
result,
|
| 12 |
-
generation_info,
|
| 13 |
-
llm_output,
|
| 14 |
chain,
|
| 15 |
cb,
|
| 16 |
err_msg,
|
|
@@ -34,8 +32,6 @@ def wandb_trace(config,
|
|
| 34 |
} if (str(err_msg) == "") else {},
|
| 35 |
outputs = {"result": result,
|
| 36 |
"cb": str(cb),
|
| 37 |
-
"generation_info": str(generation_info),
|
| 38 |
-
"llm_output": str(llm_output),
|
| 39 |
"completion": str(completion),
|
| 40 |
} if (str(err_msg) == "") else {},
|
| 41 |
model_dict = {"client": (str(chain.llm.client) if (is_rag_off) else
|
|
|
|
| 9 |
prompt,
|
| 10 |
completion,
|
| 11 |
result,
|
|
|
|
|
|
|
| 12 |
chain,
|
| 13 |
cb,
|
| 14 |
err_msg,
|
|
|
|
| 32 |
} if (str(err_msg) == "") else {},
|
| 33 |
outputs = {"result": result,
|
| 34 |
"cb": str(cb),
|
|
|
|
|
|
|
| 35 |
"completion": str(completion),
|
| 36 |
} if (str(err_msg) == "") else {},
|
| 37 |
model_dict = {"client": (str(chain.llm.client) if (is_rag_off) else
|