stravinsky 0.2.67__py3-none-any.whl → 0.4.66__py3-none-any.whl

mcp_bridge/server_tools.py

@@ -1,8 +1,7 @@
-from mcp.types import Tool, Prompt
-from typing import List
+from mcp.types import Prompt, Tool
 
 
-def get_tool_definitions() -> List[Tool]:
+def get_tool_definitions() -> list[Tool]:
     """Return all Tool definitions for the Stravinsky MCP server."""
     return [
         Tool(
@@ -12,6 +11,7 @@ def get_tool_definitions() -> List[Tool]:
                 "type": "object",
                 "properties": {},
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="system_restart",
@@ -20,6 +20,113 @@ def get_tool_definitions() -> List[Tool]:
                 "type": "object",
                 "properties": {},
             },
+            meta={"defer_loading": True},
+        ),
+        Tool(
+            name="tool_search",
+            description="Search for tools by name, description, or category. Returns matching tools with their descriptions and parameters. Use this to discover available tools before using them.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "query": {
+                        "type": "string",
+                        "description": "Search query to match against tool names, descriptions, and categories",
+                    },
+                    "category": {
+                        "type": "string",
+                        "description": "Optional category filter (e.g., 'semantic', 'lsp', 'agent', 'git')",
+                    },
+                    "top_k": {
+                        "type": "integer",
+                        "description": "Maximum number of results to return",
+                        "default": 5,
+                    },
+                },
+                "required": ["query"],
+            },
+        ),
+        Tool(
+            name="list_directory",
+            description="List files and directories in a path. Uses caching for faster repeated access.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "path": {"type": "string", "description": "Directory path to list"}
+                },
+                "required": ["path"],
+            },
+            meta={"defer_loading": True},
+        ),
+        Tool(
+            name="read_file",
+            description=(
+                "Read the contents of a file. Supports smart truncation and log-awareness. "
+                "For log files (.log, .out, .err), defaults to reading the last 100 lines."
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "path": {"type": "string", "description": "Absolute or relative path to the file"},
+                    "offset": {
+                        "type": "integer", 
+                        "description": "Line number to start reading from (0-indexed)",
+                        "default": 0
+                    },
+                    "limit": {
+                        "type": "integer", 
+                        "description": "Maximum number of lines to read",
+                    },
+                },
+                "required": ["path"],
+            },
+            meta={"defer_loading": True},
+        ),
+        Tool(
+            name="write_file",
+            description="Write content to a file. Invalidates related cache entries.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "path": {"type": "string", "description": "Path to the file to write"},
+                    "content": {"type": "string", "description": "Content to write to the file"},
+                },
+                "required": ["path", "content"],
+            },
+            meta={"defer_loading": True},
+        ),
+        Tool(
+            name="replace",
+            description="Replace text in a file. Invalidates related cache entries.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "path": {"type": "string", "description": "Path to the file to modify"},
+                    "old_string": {"type": "string", "description": "The exact literal text to replace"},
+                    "new_string": {"type": "string", "description": "The exact literal text to replace with"},
+                    "instruction": {"type": "string", "description": "Detailed description of the change"},
+                    "expected_replacements": {
+                        "type": "integer", 
+                        "description": "Number of replacements expected",
+                        "default": 1
+                    },
+                },
+                "required": ["path", "old_string", "new_string", "instruction"],
+            },
+            meta={"defer_loading": True},
+        ),
+        Tool(
+            name="run_shell_command",
+            description="Execute a shell command. Invalidates cache if it looks like a write operation.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "command": {"type": "string", "description": "Exact bash command to execute"},
+                    "description": {"type": "string", "description": "Brief description of the command's purpose"},
+                    "dir_path": {"type": "string", "description": "Optional directory to run the command in"},
+                },
+                "required": ["command", "description"],
+            },
+            meta={"defer_loading": True},
         ),
         Tool(
             name="invoke_gemini",
@@ -76,6 +183,59 @@ def get_tool_definitions() -> List[Tool]:
                     },
                 },
                 "required": ["prompt"],
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="invoke_gemini_agentic",
+            description=(
+                "Invoke Gemini with function calling for agentic tasks. "
+                "Implements a multi-turn agentic loop: sends prompt with tool definitions, "
+                "executes tool calls, and iterates until final response or max_turns reached. "
+                "Supports both API key (GEMINI_API_KEY) and OAuth authentication."
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "prompt": {
+                        "type": "string",
+                        "description": "The task prompt for the agentic loop",
+                    },
+                    "model": {
+                        "type": "string",
+                        "description": "Gemini model to use (default: gemini-3-flash)",
+                        "default": "gemini-3-flash",
+                    },
+                    "max_turns": {
+                        "type": "integer",
+                        "description": "Maximum number of tool-use turns (default: 10)",
+                        "default": 10,
+                    },
+                    "timeout": {
+                        "type": "integer",
+                        "description": "Request timeout in seconds (default: 120)",
+                        "default": 120,
+                    },
+                    "agent_context": {
+                        "type": "object",
+                        "description": "Optional agent metadata for logging (agent_type, task_id, description)",
+                        "properties": {
+                            "agent_type": {
+                                "type": "string",
+                                "description": "Type of agent (explore, delphi, frontend, etc.)",
+                            },
+                            "task_id": {
+                                "type": "string",
+                                "description": "Background task ID if running as agent",
+                            },
+                            "description": {
+                                "type": "string",
+                                "description": "Short description of what the agent is doing",
+                            },
+                        },
+                    },
+                },
+                "required": ["prompt"],
             },
         ),
         Tool(
@@ -113,6 +273,12 @@ def get_tool_definitions() -> List[Tool]:
                         "description": "Tokens reserved for internal reasoning (e.g. o1 / o3)",
                         "default": 0,
                     },
+                    "reasoning_effort": {
+                        "type": "string",
+                        "description": "Reasoning effort for reasoning models (o1, o3): low, medium, high",
+                        "enum": ["low", "medium", "high"],
+                        "default": "medium",
+                    },
                     "agent_context": {
                         "type": "object",
                         "description": "Optional agent metadata for logging (agent_type, task_id, description)",
@@ -134,6 +300,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["prompt"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="get_project_context",
@@ -144,6 +311,7 @@ def get_tool_definitions() -> List[Tool]:
                     "project_path": {"type": "string", "description": "Path to the project root"},
                 },
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="get_system_health",
@@ -152,6 +320,7 @@ def get_tool_definitions() -> List[Tool]:
                 "type": "object",
                 "properties": {},
             },
+            meta={"defer_loading": True},
         ),
         Tool(
             name="lsp_diagnostics",
@@ -168,6 +337,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["file_path"],
             },
+            meta={"defer_loading": True},
         ),
         Tool(
             name="ast_grep_search",
@@ -229,6 +399,7 @@ def get_tool_definitions() -> List[Tool]:
                     "limit": {"type": "integer", "description": "Max sessions", "default": 20},
                 },
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="session_read",
@@ -241,6 +412,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["session_id"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="session_search",
@@ -254,6 +426,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["query"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="skill_list",
@@ -264,6 +437,7 @@ def get_tool_definitions() -> List[Tool]:
                     "project_path": {"type": "string", "description": "Project directory"},
                 },
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="skill_get",
@@ -276,6 +450,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["name"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="task_spawn",
@@ -298,6 +473,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["prompt"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="task_status",
@@ -309,6 +485,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["task_id"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="task_list",
@@ -317,6 +494,7 @@ def get_tool_definitions() -> List[Tool]:
                 "type": "object",
                 "properties": {},
             },
+            meta={"defer_loading": True},
         ),
         Tool(
             name="agent_spawn",
@@ -384,6 +562,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["task_id"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="agent_output",
@@ -411,14 +590,42 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["task_id"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="agent_list",
-            description="List all background agent tasks with their status.",
+            description="List all background agent tasks with their status. By default shows all agents; use show_all=false to see only running/pending.",
             inputSchema={
                 "type": "object",
-                "properties": {},
+                "properties": {
+                    "show_all": {
+                        "type": "boolean",
+                        "description": "If false, only show running/pending agents. If true (default), show all.",
+                        "default": True,
+                    },
+                },
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="agent_cleanup",
+            description="Clean up old completed/failed/cancelled agents to reduce clutter in agent_list. Removes agents older than max_age_minutes.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "max_age_minutes": {
+                        "type": "integer",
+                        "description": "Remove agents older than this many minutes (default: 30)",
+                        "default": 30,
+                    },
+                    "statuses": {
+                        "type": "array",
+                        "items": {"type": "string"},
+                        "description": "List of statuses to remove (default: ['completed', 'failed', 'cancelled'])",
+                    },
+                },
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="agent_progress",
@@ -435,6 +642,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["task_id"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="lsp_hover",
@@ -451,6 +659,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["file_path", "line", "character"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="lsp_goto_definition",
@@ -467,6 +676,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["file_path", "line", "character"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="lsp_find_references",
@@ -488,6 +698,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["file_path", "line", "character"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="lsp_document_symbols",
@@ -499,6 +710,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["file_path"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="lsp_workspace_symbols",
@@ -518,6 +730,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["query"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="lsp_prepare_rename",
@@ -534,6 +747,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["file_path", "line", "character"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="lsp_rename",
@@ -556,6 +770,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["file_path", "line", "character", "new_name"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="lsp_code_actions",
@@ -572,6 +787,58 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["file_path", "line", "character"],
             },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="lsp_code_action_resolve",
+            description="Apply a specific code action/fix to a file (e.g., fix F401 unused import).",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "file_path": {"type": "string", "description": "Absolute path to the file"},
+                    "action_code": {
+                        "type": "string",
+                        "description": "Code action ID to apply (e.g., 'F401', 'E501' for Python)",
+                    },
+                    "line": {
+                        "type": "integer",
+                        "description": "Optional line number filter (1-indexed)",
+                    },
+                },
+                "required": ["file_path", "action_code"],
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="lsp_extract_refactor",
+            description="Extract code to a function or variable (Python via jedi).",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "file_path": {"type": "string", "description": "Absolute path to the file"},
+                    "start_line": {"type": "integer", "description": "Start line (1-indexed)"},
+                    "start_char": {"type": "integer", "description": "Start character (0-indexed)"},
+                    "end_line": {"type": "integer", "description": "End line (1-indexed)"},
+                    "end_char": {"type": "integer", "description": "End character (0-indexed)"},
+                    "new_name": {
+                        "type": "string",
+                        "description": "Name for extracted function/variable",
+                    },
+                    "kind": {
+                        "type": "string",
+                        "description": "'function' or 'variable' (default: function)",
+                    },
+                },
+                "required": [
+                    "file_path",
+                    "start_line",
+                    "start_char",
+                    "end_line",
+                    "end_char",
+                    "new_name",
+                ],
+            },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="lsp_servers",
@@ -580,6 +847,7 @@ def get_tool_definitions() -> List[Tool]:
                 "type": "object",
                 "properties": {},
             },
+            meta={"defer_loading": True},
         ),
         Tool(
             name="ast_grep_replace",
@@ -612,11 +880,475 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["pattern", "replacement"],
             },
+                    meta={"defer_loading": True},
+        ),
+        # --- SEMANTIC SEARCH ---
+        Tool(
+            name="semantic_search",
+            description=(
+                "Search codebase using natural language queries. Uses vector embeddings to find "
+                "semantically related code even without exact pattern matches. "
+                "Supports filtering by language and node type. "
+                "Example: 'find authentication logic' or 'error handling patterns'."
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "query": {
+                        "type": "string",
+                        "description": "Natural language search query (e.g., 'find authentication logic')",
+                    },
+                    "project_path": {
+                        "type": "string",
+                        "description": "Path to the project root",
+                        "default": ".",
+                    },
+                    "n_results": {
+                        "type": "integer",
+                        "description": "Maximum number of results to return",
+                        "default": 10,
+                    },
+                    "language": {
+                        "type": "string",
+                        "description": "Filter by language (e.g., 'py', 'ts', 'js')",
+                    },
+                    "node_type": {
+                        "type": "string",
+                        "description": "Filter by node type (e.g., 'function', 'class', 'method')",
+                    },
+                    "provider": {
+                        "type": "string",
+                        "description": "Embedding provider: ollama/mxbai (local/free), gemini (cloud/OAuth), openai (cloud/OAuth), huggingface (cloud)",
+                        "enum": ["ollama", "mxbai", "gemini", "openai", "huggingface"],
+                        "default": "ollama",
+                    },
+                },
+                "required": ["query"],
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="hybrid_search",
+            description=(
+                "Hybrid search combining semantic similarity with structural AST matching. "
+                "Use when you need both natural language understanding AND structural patterns. "
+                "Example: query='find authentication logic' + pattern='def $FUNC($$$):'"
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "query": {
+                        "type": "string",
+                        "description": "Natural language search query",
+                    },
+                    "pattern": {
+                        "type": "string",
+                        "description": "ast-grep pattern for structural matching (optional)",
+                    },
+                    "project_path": {
+                        "type": "string",
+                        "description": "Path to the project root",
+                        "default": ".",
+                    },
+                    "n_results": {
+                        "type": "integer",
+                        "description": "Maximum number of results to return",
+                        "default": 10,
+                    },
+                    "language": {
+                        "type": "string",
+                        "description": "Filter by language (e.g., 'py', 'ts', 'js')",
+                    },
+                    "node_type": {
+                        "type": "string",
+                        "description": "Filter by node type (e.g., 'function', 'class', 'method')",
+                    },
+                    "decorator": {
+                        "type": "string",
+                        "description": "Filter by decorator (e.g., '@property', '@staticmethod')",
+                    },
+                    "is_async": {
+                        "type": "boolean",
+                        "description": "Filter by async status (True = async only, False = sync only)",
+                    },
+                    "base_class": {
+                        "type": "string",
+                        "description": "Filter by base class (e.g., 'BaseClass')",
+                    },
+                    "provider": {
+                        "type": "string",
+                        "description": "Embedding provider: ollama/mxbai (local/free), gemini (cloud/OAuth), openai (cloud/OAuth), huggingface (cloud)",
+                        "enum": ["ollama", "mxbai", "gemini", "openai", "huggingface"],
+                        "default": "ollama",
+                    },
+                },
+                "required": ["query"],
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="find_code",
+            description=(
+                "Smart code search with automatic routing to optimal search strategy. "
+                "Automatically detects whether query is AST pattern (e.g., 'class $X'), "
+                "natural language (e.g., 'auth logic'), or complex query (e.g., 'JWT AND middleware'). "
+                "Routes to ast_grep, semantic_search, hybrid_search, or grep based on pattern detection. "
+                "**PREFERRED TOOL**: Use this instead of calling individual search tools directly."
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "query": {
+                        "type": "string",
+                        "description": "Search query (AST pattern, natural language, or text)",
+                    },
+                    "search_type": {
+                        "type": "string",
+                        "description": "Search strategy: 'auto' (default), 'ast', 'semantic', 'hybrid', 'grep', 'exact'",
+                        "enum": ["auto", "ast", "semantic", "hybrid", "grep", "exact"],
+                        "default": "auto",
+                    },
+                    "project_path": {
+                        "type": "string",
+                        "description": "Path to the project root",
+                        "default": ".",
+                    },
+                    "language": {
+                        "type": "string",
+                        "description": "Filter by language (e.g., 'py', 'ts', 'js')",
+                    },
+                    "n_results": {
+                        "type": "integer",
+                        "description": "Maximum number of results to return",
+                        "default": 10,
+                    },
+                    "provider": {
+                        "type": "string",
+                        "description": "Embedding provider for semantic search: ollama (default), gemini, openai",
+                        "default": "ollama",
+                    },
+                },
+                "required": ["query"],
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="semantic_index",
+            description=(
+                "Index a codebase for semantic search. Creates vector embeddings for all code files. "
+                "Run this before using semantic_search on a new project."
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "project_path": {
+                        "type": "string",
+                        "description": "Path to the project root",
+                        "default": ".",
+                    },
+                    "force": {
+                        "type": "boolean",
+                        "description": "If true, reindex everything. Otherwise, only new/changed files.",
+                        "default": False,
+                    },
+                    "provider": {
+                        "type": "string",
+                        "description": "Embedding provider: ollama/mxbai (local/free), gemini (cloud/OAuth), openai (cloud/OAuth), huggingface (cloud)",
+                        "enum": ["ollama", "mxbai", "gemini", "openai", "huggingface"],
+                        "default": "ollama",
+                    },
+                },
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="semantic_stats",
+            description="Get statistics about the semantic search index for a project.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "project_path": {
+                        "type": "string",
+                        "description": "Path to the project root",
+                        "default": ".",
+                    },
+                    "provider": {
+                        "type": "string",
+                        "description": "Embedding provider: ollama/mxbai (local/free), gemini (cloud/OAuth), openai (cloud/OAuth), huggingface (cloud)",
+                        "enum": ["ollama", "mxbai", "gemini", "openai", "huggingface"],
+                        "default": "ollama",
+                    },
+                },
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="start_file_watcher",
+            description=(
+                "Start automatic background reindexing when code files change. "
+                "Watches for .py file changes and triggers semantic_index automatically. "
+                "Run semantic_index() first before starting the watcher."
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "project_path": {
+                        "type": "string",
+                        "description": "Path to the project root",
+                        "default": ".",
+                    },
+                    "provider": {
+                        "type": "string",
+                        "description": "Embedding provider: ollama/mxbai (local/free), gemini (cloud/OAuth), openai (cloud/OAuth), huggingface (cloud)",
+                        "enum": ["ollama", "mxbai", "gemini", "openai", "huggingface"],
+                        "default": "ollama",
+                    },
+                    "debounce_seconds": {
+                        "type": "number",
+                        "description": "Wait time after file changes before reindexing (default: 2.0)",
+                        "default": 2.0,
+                    },
+                },
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="stop_file_watcher",
+            description="Stop the file watcher for a project.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "project_path": {
+                        "type": "string",
+                        "description": "Path to the project root",
+                        "default": ".",
+                    },
+                },
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="cancel_indexing",
+            description=(
+                "Cancel an ongoing semantic indexing operation. "
+                "Cancellation happens gracefully between batches - the current batch will complete before stopping."
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "project_path": {
+                        "type": "string",
+                        "description": "Path to the project root",
+                        "default": ".",
+                    },
+                    "provider": {
+                        "type": "string",
+                        "description": "Embedding provider (must match the one used for indexing)",
+                        "enum": ["ollama", "mxbai", "gemini", "openai", "huggingface"],
+                        "default": "ollama",
+                    },
+                },
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="delete_index",
+            description=(
+                "Delete semantic search index(es). Can delete for specific project+provider, "
+                "all providers for a project, or all indexes globally."
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "project_path": {
+                        "type": "string",
+                        "description": "Path to the project root (ignored if delete_all=true)",
+                        "default": ".",
+                    },
+                    "provider": {
+                        "type": "string",
+                        "description": "Specific provider to delete (if not specified, deletes all providers for project)",
+                        "enum": ["ollama", "mxbai", "gemini", "openai", "huggingface"],
+                    },
+                    "delete_all": {
+                        "type": "boolean",
+                        "description": "If true, delete ALL indexes for ALL projects (ignores project_path and provider)",
+                        "default": False,
+                    },
+                },
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="list_file_watchers",
+            description="List all active file watchers across all projects.",
+            inputSchema={
+                "type": "object",
+                "properties": {},
+            },
+            meta={"defer_loading": True},
+        ),
+        Tool(
+            name="multi_query_search",
+            description=(
+                "Search with LLM-expanded query variations for better recall. "
+                "Rephrases query into multiple semantic variations (e.g., 'database connection' -> "
+                "['SQLAlchemy engine setup', 'postgres connection', 'db session factory']) "
+                "and aggregates results using reciprocal rank fusion."
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "query": {
+                        "type": "string",
+                        "description": "Natural language search query",
+                    },
+                    "project_path": {
+                        "type": "string",
+                        "description": "Path to the project root",
+                        "default": ".",
+                    },
+                    "n_results": {
+                        "type": "integer",
+                        "description": "Maximum number of results to return",
+                        "default": 10,
+                    },
+                    "num_expansions": {
+                        "type": "integer",
+                        "description": "Number of query variations to generate",
+                        "default": 3,
+                    },
+                    "language": {
+                        "type": "string",
+                        "description": "Filter by language (e.g., 'py', 'ts')",
+                    },
+                    "node_type": {
+                        "type": "string",
+                        "description": "Filter by node type (e.g., 'function', 'class')",
+                    },
+                    "provider": {
+                        "type": "string",
+                        "description": "Embedding provider: ollama/mxbai (local/free), gemini (cloud/OAuth), openai (cloud/OAuth), huggingface (cloud)",
+                        "enum": ["ollama", "mxbai", "gemini", "openai", "huggingface"],
+                        "default": "ollama",
+                    },
+                },
+                "required": ["query"],
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="decomposed_search",
+            description=(
+                "Search by decomposing complex queries into focused sub-questions. "
+                "Breaks multi-part queries like 'Initialize the DB and create a user model' "
+                "into separate searches ('database initialization', 'user model definition') "
+                "and returns organized results for each part."
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "query": {
+                        "type": "string",
+                        "description": "Complex search query (may contain multiple concepts)",
+                    },
+                    "project_path": {
+                        "type": "string",
+                        "description": "Path to the project root",
+                        "default": ".",
+                    },
+                    "n_results": {
+                        "type": "integer",
+                        "description": "Maximum results per sub-query",
+                        "default": 10,
+                    },
+                    "language": {
+                        "type": "string",
+                        "description": "Filter by language",
+                    },
+                    "node_type": {
+                        "type": "string",
+                        "description": "Filter by node type",
+                    },
+                    "provider": {
+                        "type": "string",
+                        "description": "Embedding provider: ollama/mxbai (local/free), gemini (cloud/OAuth), openai (cloud/OAuth), huggingface (cloud)",
+                        "enum": ["ollama", "mxbai", "gemini", "openai", "huggingface"],
+                        "default": "ollama",
+                    },
+                },
+                "required": ["query"],
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="enhanced_search",
+            description=(
+                "Unified enhanced search combining query expansion and decomposition. "
+                "Automatically selects strategy based on query complexity: "
+                "simple queries use multi-query expansion, complex queries use decomposition. "
+                "Use mode='auto' (default), 'expand', 'decompose', or 'both'."
+            ),
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "query": {
+                        "type": "string",
+                        "description": "Search query (simple or complex)",
+                    },
+                    "project_path": {
+                        "type": "string",
+                        "description": "Path to the project root",
+                        "default": ".",
+                    },
+                    "n_results": {
+                        "type": "integer",
+                        "description": "Maximum number of results",
+                        "default": 10,
+                    },
+                    "mode": {
+                        "type": "string",
+                        "description": "Search mode: 'auto' (default), 'expand', 'decompose', or 'both'",
+                        "enum": ["auto", "expand", "decompose", "both"],
+                        "default": "auto",
+                    },
+                    "language": {
+                        "type": "string",
+                        "description": "Filter by language",
+                    },
+                    "node_type": {
+                        "type": "string",
+                        "description": "Filter by node type",
+                    },
+                    "provider": {
+                        "type": "string",
+                        "description": "Embedding provider: ollama/mxbai (local/free), gemini (cloud/OAuth), openai (cloud/OAuth), huggingface (cloud)",
+                        "enum": ["ollama", "mxbai", "gemini", "openai", "huggingface"],
+                        "default": "ollama",
+                    },
+                },
+                "required": ["query"],
+            },
+                    meta={"defer_loading": True},
+        ),
+        Tool(
+            name="get_cost_report",
+            description="Get a cost report for the current or specified session, breaking down token usage and cost by agent.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "session_id": {
+                        "type": "string",
+                        "description": "Optional session ID to filter by",
+                    },
+                },
+            },
+            meta={"defer_loading": True},
         ),
     ]
 
 
-def get_prompt_definitions() -> List[Prompt]:
+def get_prompt_definitions() -> list[Prompt]:
     """Return all Prompt definitions for the Stravinsky MCP server."""
     return [
         Prompt(