Skip to content

datarobot_genai.drtools.panels.truncate

truncate

Structure-preserving truncation for LLM-facing panel payloads (from wren-mcp).

truncate_for_llm 

truncate_for_llm(data: Any, *, max_string_length: int = 200, max_array_items: int = 5, max_object_keys: int = 30, max_depth: int = 6, _current_depth: int = 0) -> Any

Truncate nested data for LLM consumption while preserving structure.

Arrays keep the first max_array_items entries (plus a remaining-count marker), strings are capped at max_string_length chars, objects keep the first max_object_keys keys, and nesting beyond max_depth collapses to a one-line summary. Primitives pass through unchanged.

Source code in datarobot_genai/drtools/panels/truncate.py 
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
def truncate_for_llm(
    data: Any,
    *,
    max_string_length: int = 200,
    max_array_items: int = 5,
    max_object_keys: int = 30,
    max_depth: int = 6,
    _current_depth: int = 0,
) -> Any:
    """Truncate nested data for LLM consumption while preserving structure.

    Arrays keep the first ``max_array_items`` entries (plus a remaining-count
    marker), strings are capped at ``max_string_length`` chars, objects keep the
    first ``max_object_keys`` keys, and nesting beyond ``max_depth`` collapses to
    a one-line summary. Primitives pass through unchanged.
    """
    if _current_depth >= max_depth:
        return _summarize_truncated(data)

    if isinstance(data, str):
        if len(data) > max_string_length:
            return data[:max_string_length] + f"... ({len(data) - max_string_length} more chars)"
        return data

    recurse_kwargs = {
        "max_string_length": max_string_length,
        "max_array_items": max_array_items,
        "max_object_keys": max_object_keys,
        "max_depth": max_depth,
        "_current_depth": _current_depth + 1,
    }

    if isinstance(data, list):
        truncated_list: list[Any] = [
            truncate_for_llm(item, **recurse_kwargs) for item in data[:max_array_items]
        ]
        if len(data) > max_array_items:
            truncated_list.append(f"... ({len(data) - max_array_items} more items)")
        return truncated_list

    if isinstance(data, dict):
        keys = list(data.keys())
        truncated_dict: dict[str, Any] = {
            key: truncate_for_llm(data[key], **recurse_kwargs) for key in keys[:max_object_keys]
        }
        if len(keys) > max_object_keys:
            truncated_dict["..."] = f"({len(keys) - max_object_keys} more keys)"
        return truncated_dict

    return data

truncate_source_code 

truncate_source_code(src: str, max_lines: int = 50) -> str

Truncate source code to its first max_lines lines.

Source code in datarobot_genai/drtools/panels/truncate.py 
87
88
89
90
91
92
def truncate_source_code(src: str, max_lines: int = 50) -> str:
    """Truncate source code to its first ``max_lines`` lines."""
    lines = src.splitlines()
    if len(lines) <= max_lines:
        return src
    return "\n".join(lines[:max_lines]) + f"\n... ({len(lines) - max_lines} more lines)"