Skip to content

trading_agent.tools.tool_search_tool

The tool_search tool: describes the MCP tools that match some keywords.

MCP tools are listed in the prompt by name only. This tool returns the full description and argument schema of the ones the desk asks about, so it can call them with call_mcp_tool.

Typical usage example:

tool = ToolSearchTool(manager)

ToolSearchArguments

Bases: BaseModel

The arguments of tool_search.

Attributes:

Name Type Description
query str

Keywords, a tool name, or a server name to see all of its tools.
limit int

The most tools to describe.

ToolSearchTool 

ToolSearchTool(manager)

Bases: Tool

Finds MCP tools and returns how to call them.

Creates the tool.

Parameters:

Name Type Description Default
manager McpClientManager

The gateway's MCP connections.

 required
Source code in src/trading_agent/tools/tool_search_tool.py 
39
40
41
42
43
44
45
def __init__(self, manager: mcp_client_manager.McpClientManager):
    """Creates the tool.

    Args:
        manager (mcp_client_manager.McpClientManager): The gateway's MCP connections.
    """
    self._manager = manager

run async 

run(arguments)

Ranks the tools by how many keywords they contain.

Parameters:

Name Type Description Default
arguments ToolSearchArguments

The query and the limit.

 required

Returns:

Type Description
ToolResult

The matching tools with their schemas, or the names of all tools when nothing matches.
Source code in src/trading_agent/tools/tool_search_tool.py 
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
async def run(self, arguments: ToolSearchArguments) -> tool.ToolResult:
    """Ranks the tools by how many keywords they contain.

    Args:
        arguments (ToolSearchArguments): The query and the limit.

    Returns:
        tool.ToolResult: The matching tools with their schemas, or the names of all tools when nothing matches.
    """
    tools = self._manager.tools()
    if not tools:
        return tool.ToolResult.error(
            'No MCP server is ready, so there are no MCP tools right now.'
        )
    words = arguments.query.lower().split()
    scored = []
    for candidate in tools:
        if candidate.server.lower() == arguments.query.strip().lower():
            scored.append((len(words) + 1, candidate))
            continue
        haystack = f'{candidate.server} {candidate.name} {candidate.description}'.lower()
        score = 0
        for word in words:
            if word in haystack:
                score += 1
        if score:
            scored.append((score, candidate))
    if not scored:
        names = []
        for candidate in tools:
            names.append(f'{candidate.server}/{candidate.name}')
        return tool.ToolResult.success(
            f'No MCP tool matches {arguments.query!r}. The tools are: {", ".join(names)}.'
        )
    scored.sort(key=lambda pair: pair[0], reverse=True)
    lines = []
    for _, candidate in scored[: arguments.limit]:
        lines.append(f'server: {candidate.server}')
        lines.append(f'tool: {candidate.name}')
        lines.append(f'description: {candidate.description}')
        lines.append(
            f'arguments schema: {json.dumps(candidate.input_schema, ensure_ascii=False)}'
        )
        lines.append('')
    return tool.ToolResult.success('\n'.join(lines).strip())