trulens.core.utils.trace_compression

trulens.core.utils.trace_compression

Experimental Trace compression utilities to reduce token usage in feedback functions. This module provides functionality to compress trace data while preserving essential information needed for evaluation. Use with caution.

Classes

TraceCompressor

Compresses trace data to reduce token usage while preserving essential information.

Functions

__init__

__init__()

Initialize the trace compressor.

compress_trace

compress_trace(trace_data: Any) -> Dict[str, Any]

Compress trace data to reduce token usage.

PARAMETER	DESCRIPTION
trace_data	The raw trace data to compress TYPE: Any

RETURNS	DESCRIPTION
Dict[str, Any]	Compressed trace data with essential information preserved

Functions

compress_trace_for_feedback

compress_trace_for_feedback(
    trace_data: Any,
) -> Dict[str, Any]

Convenience function to compress trace data for feedback functions.

PARAMETER	DESCRIPTION
trace_data	The trace data to compress TYPE: Any

RETURNS	DESCRIPTION
Dict[str, Any]	Compressed trace data

compress_multiple_traces

compress_multiple_traces(
    traces: List[Any],
) -> List[Dict[str, Any]]

Compress multiple traces efficiently.

PARAMETER	DESCRIPTION
traces	List of trace data to compress TYPE: List[Any]

RETURNS	DESCRIPTION
List[Dict[str, Any]]	List of compressed traces