langchain_community.callbacks.tracers.wandb.RunProcessor

class langchain_community.callbacks.tracers.wandb.RunProcessor(wandb_module: Any, trace_module: Any)

Handles the conversion of a LangChain Runs into a WBTraceTree.

Methods

__init__(wandb_module, trace_module)
build_tree(runs)	Builds a nested dictionary from a list of runs. :param runs: The list of runs to build the tree from. :return: The nested dictionary representing the langchain Run in a tree structure compatible with WBTraceTree.
flatten_run(run)	Utility to flatten a nest run object into a list of runs.
modify_serialized_iterative(runs[, ...])	Utility to modify the serialized field of a list of runs dictionaries.
process_model(run)	Utility to process a run for wandb model_dict serialization.
process_span(run)	Converts a LangChain Run into a W&B Trace Span.
truncate_run_iterative(runs[, keep_keys])	Utility to truncate a list of runs dictionaries to only keep the specified

Parameters

• wandb_module (Any) –

• trace_module (Any) –

__init__(wandb_module: Any, trace_module: Any)

Parameters

• wandb_module (Any) –

• trace_module (Any) –

build_tree(runs: List[Dict[str, Any]]) → Dict[str, Any]

Builds a nested dictionary from a list of runs. :param runs: The list of runs to build the tree from. :return: The nested dictionary representing the langchain Run in a tree

structure compatible with WBTraceTree.

Parameters

runs (List[Dict[str, Any]]) –

Return type

Dict[str, Any]

flatten_run(run: Dict[str, Any]) → List[Dict[str, Any]]

Utility to flatten a nest run object into a list of runs. :param run: The base run to flatten. :return: The flattened list of runs.

Parameters

run (Dict[str, Any]) –

Return type

List[Dict[str, Any]]

modify_serialized_iterative(runs: List[Dict[str, Any]], exact_keys: Tuple[str, ...] = (), partial_keys: Tuple[str, ...] = ()) → List[Dict[str, Any]]

Utility to modify the serialized field of a list of runs dictionaries. removes any keys that match the exact_keys and any keys that contain any of the partial_keys. recursively moves the dictionaries under the kwargs key to the top level. changes the “id” field to a string “_kind” field that tells WBTraceTree how to visualize the run. promotes the “serialized” field to the top level.

Parameters

• runs (List[Dict[str, Any]]) – The list of runs to modify.

• exact_keys (Tuple[str, ...]) – A tuple of keys to remove from the serialized field.

• partial_keys (Tuple[str, ...]) – A tuple of partial keys to remove from the serialized field.

Returns

The modified list of runs.

Return type

List[Dict[str, Any]]

process_model(run: Run) → Optional[Dict[str, Any]]

Utility to process a run for wandb model_dict serialization. :param run: The run to process. :return: The convert model_dict to pass to WBTraceTree.

Parameters

run (Run) –

Return type

Optional[Dict[str, Any]]

process_span(run: Run) → Optional['Span']

Converts a LangChain Run into a W&B Trace Span. :param run: The LangChain Run to convert. :return: The converted W&B Trace Span.

Parameters

run (Run) –

Return type

Optional[‘Span’]

truncate_run_iterative(runs: List[Dict[str, Any]], keep_keys: Tuple[str, ...] = ()) → List[Dict[str, Any]]

Utility to truncate a list of runs dictionaries to only keep the specified

keys in each run.

Parameters

• runs (List[Dict[str, Any]]) – The list of runs to truncate.

• keep_keys (Tuple[str, ...]) – The keys to keep in each run.

Returns

The truncated list of runs.

Return type

List[Dict[str, Any]]