Function export_traces

utils/trace_export_utils.py:356–385

A high-level wrapper to export traces to a specified format. Args: traces: A list of trace objects. output_path: The path to save the file. format: The desired output format ('jsonl', 'json', 'hdf5'). compress: If True, compress the output (currently for jso

(
    traces: List[Any],
    output_path: Union[str, Path],
    format: str = "jsonl",
    compress: bool = False,
    metadata: Optional[Dict[str, Any]] = None,
)

def export_traces(
    traces: List[Any],
    output_path: Union[str, Path],
    format: str = "jsonl",
    compress: bool = False,
    metadata: Optional[Dict[str, Any]] = None,
) -> None:
    """
    A high-level wrapper to export traces to a specified format.

    Args:
        traces: A list of trace objects.
        output_path: The path to save the file.
        format: The desired output format ('jsonl', 'json', 'hdf5').
        compress: If True, compress the output (currently for jsonl only).
        metadata: Optional metadata (for json and hdf5 formats only).
    """
    # Normalize format string to lowercase
    format = format.lower()

    # Dispatch to the appropriate export function based on the format
    if format == "jsonl":
        export_traces_jsonl(traces, output_path, compress=compress)
    elif format == "json":
        export_traces_json(traces, output_path, metadata=metadata)
    elif format == "hdf5":
        export_traces_hdf5(traces, output_path, metadata=metadata)
    else:
        # Raise an error for unsupported formats
        raise ValueError(f"Unsupported format: {format}. Use 'jsonl', 'json', or 'hdf5'")


def load_traces(