stravinsky 0.4.18__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,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["file_path", "line", "character"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="lsp_code_action_resolve",
@@ -591,6 +807,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["file_path", "action_code"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="lsp_extract_refactor",
@@ -621,6 +838,7 @@ def get_tool_definitions() -> List[Tool]:
                     "new_name",
                 ],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="lsp_servers",
@@ -629,6 +847,7 @@ def get_tool_definitions() -> List[Tool]:
                 "type": "object",
                 "properties": {},
             },
+            meta={"defer_loading": True},
         ),
         Tool(
             name="ast_grep_replace",
@@ -661,6 +880,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["pattern", "replacement"],
             },
+                    meta={"defer_loading": True},
         ),
         # --- SEMANTIC SEARCH ---
         Tool(
@@ -705,6 +925,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["query"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="hybrid_search",
@@ -763,6 +984,53 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "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",
@@ -791,6 +1059,7 @@ def get_tool_definitions() -> List[Tool]:
                     },
                 },
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="semantic_stats",
@@ -811,6 +1080,7 @@ def get_tool_definitions() -> List[Tool]:
                     },
                 },
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="start_file_watcher",
@@ -840,6 +1110,7 @@ def get_tool_definitions() -> List[Tool]:
                     },
                 },
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="stop_file_watcher",
@@ -854,6 +1125,7 @@ def get_tool_definitions() -> List[Tool]:
                     },
                 },
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="cancel_indexing",
@@ -877,6 +1149,7 @@ def get_tool_definitions() -> List[Tool]:
                     },
                 },
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="delete_index",
@@ -904,6 +1177,7 @@ def get_tool_definitions() -> List[Tool]:
                     },
                 },
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="list_file_watchers",
@@ -912,6 +1186,7 @@ def get_tool_definitions() -> List[Tool]:
                 "type": "object",
                 "properties": {},
             },
+            meta={"defer_loading": True},
         ),
         Tool(
             name="multi_query_search",
@@ -960,6 +1235,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["query"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="decomposed_search",
@@ -1003,6 +1279,7 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "required": ["query"],
             },
+                    meta={"defer_loading": True},
         ),
         Tool(
             name="enhanced_search",
@@ -1052,11 +1329,26 @@ def get_tool_definitions() -> List[Tool]:
                 },
                 "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(

mcp_bridge/tools/__init__.py

@@ -1,12 +1,20 @@
 # Tools module
-from .model_invoke import invoke_gemini, invoke_openai, invoke_gemini_agentic
-from .code_search import lsp_diagnostics, ast_grep_search, ast_grep_replace, grep_search, glob_files
-from .session_manager import list_sessions, read_session, search_sessions, get_session_info
-from .skill_loader import list_skills, get_skill, create_skill
-from .agent_manager import agent_spawn, agent_output, agent_cancel, agent_list, agent_progress, agent_retry
-from .background_tasks import task_spawn, task_status, task_list
-from .continuous_loop import enable_ralph_loop, disable_ralph_loop
-from .query_classifier import classify_query, QueryCategory, QueryClassification
+from .agent_manager import (
+    agent_cancel,
+    agent_list,
+    agent_output,
+    agent_progress,
+    agent_retry,
+    agent_spawn,
+)
+from .background_tasks import task_list, task_spawn, task_status
+from .code_search import ast_grep_replace, ast_grep_search, glob_files, grep_search, lsp_diagnostics
+from .continuous_loop import disable_ralph_loop, enable_ralph_loop
+from .model_invoke import invoke_gemini, invoke_gemini_agentic, invoke_openai
+from .query_classifier import QueryCategory, QueryClassification, classify_query
+from .session_manager import get_session_info, list_sessions, read_session, search_sessions
+from .skill_loader import create_skill, get_skill, list_skills
+from .tool_search import format_search_results, search_tool_names, search_tools
 
 __all__ = [
     "QueryCategory",
@@ -23,6 +31,7 @@ __all__ = [
     "create_skill",
     "disable_ralph_loop",
     "enable_ralph_loop",
+    "format_search_results",
     "get_session_info",
     "get_skill",
     "glob_files",
@@ -35,6 +44,8 @@ __all__ = [
     "lsp_diagnostics",
     "read_session",
     "search_sessions",
+    "search_tool_names",
+    "search_tools",
     "task_list",
     "task_spawn",
     "task_status",