Skip to content

Callers

callers

CallersTool — find call-sites of a symbol.

CallersTool

Bases: AXMTool

Find all call-sites of a function via tree-sitter.

Registered as ast_callers via axm.tools entry point. Workspace-aware: if path is a uv workspace root, searches across all member packages.

Source code in packages/axm-ast/src/axm_ast/tools/callers.py 
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
class CallersTool(AXMTool):
    """Find all call-sites of a function via tree-sitter.

    Registered as ``ast_callers`` via axm.tools entry point.
    Workspace-aware: if path is a uv workspace root, searches
    across all member packages.
    """

    agent_hint: str = (
        "Find all call-sites of a symbol"
        " — precise caller list with file, line, and enclosing function."
        " Not grep noise."
    )

    @property
    def name(self) -> str:
        """Return tool name for registry lookup."""
        return "ast_callers"

    def execute(
        self, *, path: str = ".", symbol: str | None = None, **kwargs: Any
    ) -> ToolResult:
        """Find all callers of a symbol.

        Args:
            path: Path to package or workspace directory.
            symbol: Symbol name to search for (required).

        Returns:
            ToolResult with caller list.
        """
        if not symbol:
            return ToolResult(success=False, error="symbol parameter is required")

        try:
            project_path = Path(path).resolve()
            if not project_path.is_dir():
                return ToolResult(
                    success=False, error=f"Not a directory: {project_path}"
                )

            from axm_ast.core.workspace import detect_workspace

            ws = detect_workspace(project_path)
            if ws is not None:
                from axm_ast.core.callers import find_callers_workspace
                from axm_ast.core.workspace import analyze_workspace

                ws = analyze_workspace(project_path)
                callers = find_callers_workspace(ws, symbol)
            else:
                from axm_ast.core.cache import get_package
                from axm_ast.core.callers import find_callers

                pkg = get_package(project_path)
                callers = find_callers(pkg, symbol)

            caller_data = [
                {
                    "module": c.module,
                    "line": c.line,
                    "context": c.context,
                    "call_expression": c.call_expression,
                }
                for c in callers
            ]

            return ToolResult(
                success=True,
                data={"callers": caller_data, "count": len(caller_data)},
                hint=(
                    "Tip: Use ast_impact(symbol) for full blast radius including tests."
                ),
            )
        except Exception as exc:
            return ToolResult(success=False, error=str(exc))
name property

Return tool name for registry lookup.

execute(*, path='.', symbol=None, **kwargs)

Find all callers of a symbol.

Parameters:

Name Type Description Default
path str

Path to package or workspace directory.

 '.'
symbol str | None

Symbol name to search for (required).

 None

Returns:

Type Description
ToolResult

ToolResult with caller list.
Source code in packages/axm-ast/src/axm_ast/tools/callers.py 
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
def execute(
    self, *, path: str = ".", symbol: str | None = None, **kwargs: Any
) -> ToolResult:
    """Find all callers of a symbol.

    Args:
        path: Path to package or workspace directory.
        symbol: Symbol name to search for (required).

    Returns:
        ToolResult with caller list.
    """
    if not symbol:
        return ToolResult(success=False, error="symbol parameter is required")

    try:
        project_path = Path(path).resolve()
        if not project_path.is_dir():
            return ToolResult(
                success=False, error=f"Not a directory: {project_path}"
            )

        from axm_ast.core.workspace import detect_workspace

        ws = detect_workspace(project_path)
        if ws is not None:
            from axm_ast.core.callers import find_callers_workspace
            from axm_ast.core.workspace import analyze_workspace

            ws = analyze_workspace(project_path)
            callers = find_callers_workspace(ws, symbol)
        else:
            from axm_ast.core.cache import get_package
            from axm_ast.core.callers import find_callers

            pkg = get_package(project_path)
            callers = find_callers(pkg, symbol)

        caller_data = [
            {
                "module": c.module,
                "line": c.line,
                "context": c.context,
                "call_expression": c.call_expression,
            }
            for c in callers
        ]

        return ToolResult(
            success=True,
            data={"callers": caller_data, "count": len(caller_data)},
            hint=(
                "Tip: Use ast_impact(symbol) for full blast radius including tests."
            ),
        )
    except Exception as exc:
        return ToolResult(success=False, error=str(exc))