Update trace.py
Browse files
trace.py
CHANGED
@@ -21,9 +21,6 @@ def trace_wandb(config,
|
|
21 |
name = "" if (chain == None) else type(chain).__name__,
|
22 |
status_code = "success" if (str(err_msg) == "") else "error",
|
23 |
status_message = str(err_msg),
|
24 |
-
metadata = {"chunk_overlap": "" if (is_rag_off) else config["chunk_overlap"],
|
25 |
-
"chunk_size": "" if (is_rag_off) else config["chunk_size"],
|
26 |
-
} if (str(err_msg) == "") else {},
|
27 |
inputs = {"is_rag": not is_rag_off,
|
28 |
"prompt": prompt,
|
29 |
"chain_prompt": (str(chain.prompt) if (is_rag_off) else
|
|
|
21 |
name = "" if (chain == None) else type(chain).__name__,
|
22 |
status_code = "success" if (str(err_msg) == "") else "error",
|
23 |
status_message = str(err_msg),
|
|
|
|
|
|
|
24 |
inputs = {"is_rag": not is_rag_off,
|
25 |
"prompt": prompt,
|
26 |
"chain_prompt": (str(chain.prompt) if (is_rag_off) else
|