Toggle Menu
Prev Up Next

# `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]]