diff --git a/src/ragas/callbacks.py b/src/ragas/callbacks.py index d587e43a4..23332c61c 100644 --- a/src/ragas/callbacks.py +++ b/src/ragas/callbacks.py @@ -164,7 +164,8 @@ def parse_run_traces( prompt_trace = traces[prompt_uuid] output = prompt_trace.outputs.get("output", {}) output = output[0] if isinstance(output, list) else output - prompt_traces[f"{prompt_trace.name}"] = { + prompt_traces[f"prompt_{i+1}"] = { + "name" : prompt_trace.name, "input": prompt_trace.inputs.get("data", {}), "output": output, }