chunk-your-tools 2.0.3

MCP tool schema decomposition and recomposition
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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
72
73

"""Smoke tests for chunk-your-tools installed from PyPI."""

from __future__ import annotations

import pytest

from chunk_your_tools import build_catalog_index

from example_snapshot import (
    catalog_dict_from_snapshot,
    extract_snapshot_parts,
    load_snapshot,
    resolve_snapshot_path,
    write_output,
)


def test_build_catalog_index_from_registry() -> None:
    tool = {
        "id": "mcp__test__foo",
        "server": "test",
        "tool": "mcp__test__foo",
        "summary": "A test tool",
        "full_schema": {
            "id": "mcp__test__foo",
            "name": "mcp__test__foo",
            "description": "A test tool",
            "inputSchema": {
                "type": "object",
                "properties": {
                    "required_field": {"type": "string"},
                    "optional_field": {"type": "string", "description": "opt"},
                },
                "required": ["required_field"],
            },
        },
    }
    index = build_catalog_index([tool], [])
    assert "schemas/decomposed/mcp__test__foo.json" in index.files

    meta = index.tool_schema_metadata()
    types = {
        entry["file_path"]: entry["type"]
        for entry in meta.get("decomposed") or []
        if isinstance(entry, dict)
    }
    assert types["schemas/decomposed/mcp__test__foo.json"] == "tool"
    assert types["schemas/decomposed/mcp__test__foo/optional_field.json"] == "property"


def test_retrieve_from_example_file(
    example_file: str | None,
    output_file: str | None,
) -> None:
    if example_file is None:
        pytest.skip("pass --file to run against a local debug snapshot")
    snapshot_path = resolve_snapshot_path(example_file)
    data = load_snapshot(snapshot_path)
    _build_tools, _survivor, _expected = extract_snapshot_parts(data)

    catalog = catalog_dict_from_snapshot(data)
    json_chunks = catalog.get("json") or []
    md_chunks = catalog.get("md") or []
    assert json_chunks, "build_catalog_index produced no json chunks"
    assert md_chunks, "build_catalog_index produced no md enum chunks"
    assert any(
        "/schemas/decomposed/" in entry.get("file_path", "")
        and entry.get("file_path", "").endswith(".json")
        for entry in json_chunks
        if isinstance(entry, dict)
    ), "expected per-property decomposed json chunks"

    write_output(catalog, output_file)