stravinsky 0.2.52__py3-none-any.whl → 0.4.18__py3-none-any.whl

mcp_bridge/server_tools.py

@@ -357,6 +357,11 @@ def get_tool_definitions() -> List[Tool]:
                         "description": "Maximum execution time in seconds",
                         "default": 300,
                     },
+                    "blocking": {
+                        "type": "boolean",
+                        "description": "If true, wait for agent completion and return result directly. Recommended for delphi consultations.",
+                        "default": False,
+                    },
                 },
                 "required": ["prompt"],
             },
@@ -568,6 +573,55 @@ def get_tool_definitions() -> List[Tool]:
                 "required": ["file_path", "line", "character"],
             },
         ),
+        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"],
+            },
+        ),
+        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",
+                ],
+            },
+        ),
         Tool(
             name="lsp_servers",
             description="List available LSP servers and their installation status.",
@@ -608,6 +662,397 @@ def get_tool_definitions() -> List[Tool]:
                 "required": ["pattern", "replacement"],
             },
         ),
+        # --- 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"],
+            },
+        ),
+        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"],
+            },
+        ),
+        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",
+                    },
+                },
+            },
+        ),
+        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",
+                    },
+                },
+            },
+        ),
+        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,
+                    },
+                },
+            },
+        ),
+        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": ".",
+                    },
+                },
+            },
+        ),
+        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",
+                    },
+                },
+            },
+        ),
+        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,
+                    },
+                },
+            },
+        ),
+        Tool(
+            name="list_file_watchers",
+            description="List all active file watchers across all projects.",
+            inputSchema={
+                "type": "object",
+                "properties": {},
+            },
+        ),
+        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"],
+            },
+        ),
+        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"],
+            },
+        ),
+        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"],
+            },
+        ),
     ]
 
 

mcp_bridge/tools/__init__.py

@@ -6,33 +6,37 @@ 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
 
 __all__ = [
+    "QueryCategory",
+    "QueryClassification",
+    "agent_cancel",
+    "agent_list",
+    "agent_output",
+    "agent_progress",
+    "agent_retry",
+    "agent_spawn",
+    "ast_grep_replace",
+    "ast_grep_search",
+    "classify_query",
+    "create_skill",
+    "disable_ralph_loop",
+    "enable_ralph_loop",
+    "get_session_info",
+    "get_skill",
+    "glob_files",
+    "grep_search",
     "invoke_gemini",
     "invoke_gemini_agentic",
     "invoke_openai",
-    "lsp_diagnostics",
-    "ast_grep_search",
-    "ast_grep_replace",
-    "grep_search",
-    "glob_files",
     "list_sessions",
+    "list_skills",
+    "lsp_diagnostics",
     "read_session",
     "search_sessions",
-    "get_session_info",
-    "list_skills",
-    "get_skill",
-    "create_skill",
-    "agent_spawn",
-    "agent_output",
-    "agent_retry",
-    "agent_cancel",
-    "agent_list",
-    "agent_progress",
+    "task_list",
     "task_spawn",
     "task_status",
-    "task_list",
-    "enable_ralph_loop",
-    "disable_ralph_loop",
 ]