stravinsky 0.1.12__py3-none-any.whl

mcp_bridge/tools/skill_loader.py

@@ -0,0 +1,212 @@
+"""
+Skill Loader - Claude Code Slash Command Discovery
+
+Discovers and lists available skills (slash commands) from:
+1. Project-local .claude/commands/
+2. User-global ~/.claude/commands/
+
+Skills are markdown files with frontmatter defining the command behavior.
+"""
+
+import os
+import re
+from pathlib import Path
+from typing import Any
+
+
+def parse_frontmatter(content: str) -> tuple[dict[str, Any], str]:
+    """
+    Parse YAML frontmatter from markdown content.
+    
+    Returns:
+        Tuple of (metadata dict, body content)
+    """
+    if not content.startswith("---"):
+        return {}, content
+    
+    # Find the closing ---
+    end_match = content.find("---", 3)
+    if end_match == -1:
+        return {}, content
+    
+    frontmatter = content[3:end_match].strip()
+    body = content[end_match + 3:].strip()
+    
+    # Simple YAML parsing for key: value pairs
+    metadata = {}
+    for line in frontmatter.split("\n"):
+        if ":" in line:
+            key, _, value = line.partition(":")
+            key = key.strip()
+            value = value.strip().strip('"').strip("'")
+            metadata[key] = value
+    
+    return metadata, body
+
+
+def discover_skills(project_path: str | None = None) -> list[dict[str, Any]]:
+    """
+    Discover available skills/commands.
+    
+    Searches:
+    1. Project-local: {project}/.claude/commands/
+    2. User-global: ~/.claude/commands/
+    
+    Args:
+        project_path: Project directory to search (defaults to cwd)
+        
+    Returns:
+        List of skill definitions.
+    """
+    skills = []
+    search_paths = []
+    
+    # Project-local commands
+    project = Path(project_path) if project_path else Path.cwd()
+    project_commands = project / ".claude" / "commands"
+    if project_commands.exists():
+        search_paths.append(("project", project_commands))
+    
+    # User-global commands
+    user_commands = Path.home() / ".claude" / "commands"
+    if user_commands.exists():
+        search_paths.append(("user", user_commands))
+    
+    for scope, commands_dir in search_paths:
+        for md_file in commands_dir.glob("*.md"):
+            try:
+                content = md_file.read_text()
+                metadata, body = parse_frontmatter(content)
+                
+                skills.append({
+                    "name": md_file.stem,
+                    "scope": scope,
+                    "path": str(md_file),
+                    "description": metadata.get("description", ""),
+                    "allowed_tools": metadata.get("allowed-tools", "").split(",") if metadata.get("allowed-tools") else [],
+                    "body_preview": body[:200] + "..." if len(body) > 200 else body,
+                })
+            except Exception:
+                continue
+    
+    return skills
+
+
+def list_skills(project_path: str | None = None) -> str:
+    """
+    List all available skills for MCP tool.
+    
+    Args:
+        project_path: Project directory to search
+        
+    Returns:
+        Formatted skill listing.
+    """
+    skills = discover_skills(project_path)
+    
+    if not skills:
+        return "No skills found. Create .claude/commands/*.md files to add skills."
+    
+    lines = [f"Found {len(skills)} skill(s):\n"]
+    
+    for skill in skills:
+        scope_badge = "[project]" if skill["scope"] == "project" else "[user]"
+        lines.append(f"  /{skill['name']} {scope_badge}")
+        if skill["description"]:
+            lines.append(f"    {skill['description']}")
+    
+    return "\n".join(lines)
+
+
+def get_skill(name: str, project_path: str | None = None) -> str:
+    """
+    Get the content of a specific skill.
+    
+    Args:
+        name: Skill name (filename without .md)
+        project_path: Project directory to search
+        
+    Returns:
+        Skill content or error message.
+    """
+    skills = discover_skills(project_path)
+    
+    skill = next((s for s in skills if s["name"] == name), None)
+    if not skill:
+        available = ", ".join(s["name"] for s in skills)
+        return f"Skill '{name}' not found. Available: {available or 'none'}"
+    
+    try:
+        content = Path(skill["path"]).read_text()
+        metadata, body = parse_frontmatter(content)
+        
+        lines = [
+            f"## Skill: {name}",
+            f"**Scope**: {skill['scope']}",
+            f"**Path**: {skill['path']}",
+        ]
+        
+        if metadata.get("description"):
+            lines.append(f"**Description**: {metadata['description']}")
+        
+        if metadata.get("allowed-tools"):
+            lines.append(f"**Allowed Tools**: {metadata['allowed-tools']}")
+        
+        lines.extend(["", "---", "", body])
+        
+        return "\n".join(lines)
+        
+    except Exception as e:
+        return f"Error reading skill: {e}"
+
+
+def create_skill(
+    name: str,
+    description: str,
+    content: str,
+    scope: str = "project",
+    project_path: str | None = None,
+) -> str:
+    """
+    Create a new skill file.
+    
+    Args:
+        name: Skill name (will be used as filename)
+        description: Short description for frontmatter
+        content: Skill body content
+        scope: "project" or "user"
+        project_path: Project directory for project-scope skills
+        
+    Returns:
+        Success or error message.
+    """
+    # Sanitize name
+    name = re.sub(r"[^a-zA-Z0-9_-]", "-", name.lower())
+    
+    if scope == "project":
+        base_dir = Path(project_path) if project_path else Path.cwd()
+        commands_dir = base_dir / ".claude" / "commands"
+    else:
+        commands_dir = Path.home() / ".claude" / "commands"
+    
+    # Ensure directory exists
+    commands_dir.mkdir(parents=True, exist_ok=True)
+    
+    skill_path = commands_dir / f"{name}.md"
+    
+    if skill_path.exists():
+        return f"Skill '{name}' already exists at {skill_path}"
+    
+    # Create skill content
+    skill_content = f"""---
+description: {description}
+---
+
+{content}
+"""
+    
+    try:
+        skill_path.write_text(skill_content)
+        return f"Created skill '{name}' at {skill_path}"
+    except Exception as e:
+        return f"Error creating skill: {e}"

mcp_bridge/tools/task_runner.py

@@ -0,0 +1,97 @@
+"""
+Task Runner for Stravinsky background sub-agents.
+
+This script is executed as a background process to handle agent tasks,
+capture output, and update status in tasks.json.
+"""
+
+import argparse
+import json
+import logging
+import os
+import sys
+import asyncio
+from datetime import datetime
+from pathlib import Path
+
+# Setup logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s [%(levelname)s] %(message)s')
+logger = logging.getLogger("task_runner")
+
+async def run_task(task_id: str, base_dir: str):
+    base_path = Path(base_dir)
+    tasks_file = base_path / "tasks.json"
+    agents_dir = base_path / "agents"
+    
+    # Load task details
+    try:
+        with open(tasks_file, "r") as f:
+            tasks = json.load(f)
+        task = tasks.get(task_id)
+    except Exception as e:
+        logger.error(f"Failed to load tasks: {e}")
+        return
+
+    if not task:
+        logger.error(f"Task {task_id} not found")
+        return
+
+    prompt = task.get("prompt")
+    model = task.get("model", "gemini-2.0-flash")
+    
+    output_file = agents_dir / f"{task_id}.out"
+    agents_dir.mkdir(parents=True, exist_ok=True)
+
+    try:
+        # Import model invocation
+        from mcp_bridge.tools.model_invoke import invoke_gemini
+        
+        logger.info(f"Executing task {task_id} with model {model}...")
+        
+        # Execute the model call
+        result = await invoke_gemini(prompt=prompt, model=model)
+        
+        # Save result
+        with open(output_file, "w") as f:
+            f.write(result)
+            
+        # Update status
+        with open(tasks_file, "r") as f:
+            tasks = json.load(f)
+        
+        if task_id in tasks:
+            tasks[task_id].update({
+                "status": "completed",
+                "result": result,
+                "completed_at": datetime.now().isoformat()
+            })
+            with open(tasks_file, "w") as f:
+                json.dump(tasks, f, indent=2)
+                
+        logger.info(f"Task {task_id} completed successfully")
+        
+    except Exception as e:
+        logger.exception(f"Task {task_id} failed: {e}")
+        
+        # Update status with error
+        try:
+            with open(tasks_file, "r") as f:
+                tasks = json.load(f)
+            if task_id in tasks:
+                tasks[task_id].update({
+                    "status": "failed",
+                    "error": str(e),
+                    "completed_at": datetime.now().isoformat()
+                })
+                with open(tasks_file, "w") as f:
+                    json.dump(tasks, f, indent=2)
+        except:
+            pass
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--task-id", required=True)
+    parser.add_argument("--base-dir", required=True)
+    args = parser.parse_args()
+    
+    asyncio.run(run_task(args.task_id, args.base_dir))

mcp_bridge/utils/__init__.py

@@ -0,0 +1 @@
+# Utility functions

stravinsky-0.1.12.dist-info/METADATA

@@ -0,0 +1,198 @@
+Metadata-Version: 2.4
+Name: stravinsky
+Version: 0.1.12
+Summary: MCP Bridge for Claude Code with Multi-Model Support. Install globally: claude mcp add --scope user stravinsky -- uvx stravinsky. Add to CLAUDE.md: See https://pypi.org/project/stravinsky/
+Project-URL: Repository, https://github.com/GratefulDave/stravinsky
+Project-URL: Issues, https://github.com/GratefulDave/stravinsky/issues
+Author: Stravinsky Team
+License: MIT
+Keywords: claude,gemini,mcp,oauth,openai
+Requires-Python: >=3.11
+Requires-Dist: aiofiles>=23.1.0
+Requires-Dist: cryptography>=41.0.0
+Requires-Dist: google-auth-oauthlib>=1.0.0
+Requires-Dist: google-auth>=2.20.0
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: jedi>=0.19.2
+Requires-Dist: keyring>=25.7.0
+Requires-Dist: mcp>=1.0.0
+Requires-Dist: openai>=1.0.0
+Requires-Dist: psutil>=5.9.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: rich>=13.0.0
+Requires-Dist: ruff>=0.14.10
+Provides-Extra: dev
+Requires-Dist: mypy>=1.10.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.23.0; extra == 'dev'
+Requires-Dist: pytest>=8.0.0; extra == 'dev'
+Requires-Dist: ruff>=0.4.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+<div align="center">
+  <img src="https://raw.githubusercontent.com/GratefulDave/stravinsky/main/assets/logo.png" width="300" alt="Stravinsky Logo">
+  <h1>Stravinsky</h1>
+  <p><strong>The Avant-Garde MCP Bridge for Claude Code</strong></p>
+  <p><em>Movement • Rhythm • Precision</em></p>
+</div>
+
+---
+
+## What is Stravinsky?
+
+**Multi-model AI orchestration** with OAuth authentication for Claude Code.
+
+## Features
+
+- 🔐 **OAuth Authentication** - Secure browser-based auth for Google (Gemini) and OpenAI (ChatGPT)
+- 🤖 **Multi-Model Support** - Seamlessly invoke Gemini and GPT models from Claude
+- 🛠️ **31 MCP Tools** - Model invocation, code search, LSP integrations, session management, and more
+- 🧠 **7 Specialized Agents** - Stravinsky (orchestrator), Delphi (advisor), Dewey (documentation), and more
+- 🔄 **Background Tasks** - Spawn parallel agents with full tool access via Claude Code CLI
+- 📝 **LSP Integration** - Full Language Server Protocol support for Python (jedi)
+- 🔍 **AST-Aware Search** - Structural code search and refactoring with ast-grep
+
+## Quick Start
+
+### Installation
+
+**From PyPI (Recommended):**
+
+```bash
+# One-shot with uvx - no installation needed!
+claude mcp add stravinsky -- uvx stravinsky
+
+# Or install globally first:
+uv tool install stravinsky
+claude mcp add stravinsky -- stravinsky
+```
+
+**From Source (for development):**
+
+```bash
+uv tool install --editable /path/to/stravinsky
+claude mcp add stravinsky -- stravinsky
+```
+
+### Authentication
+
+```bash
+# Login to Google (Gemini)
+stravinsky-auth login gemini
+
+# Login to OpenAI (ChatGPT Plus/Pro required)
+stravinsky-auth login openai
+
+# Check status
+stravinsky-auth status
+
+# Logout
+stravinsky-auth logout gemini
+```
+
+## Add to Your CLAUDE.md
+
+After installing globally, add this to your project's `CLAUDE.md`:
+
+```markdown
+## Stravinsky MCP (Parallel Agents)
+
+Use Stravinsky MCP tools. **DEFAULT: spawn parallel agents for multi-step tasks.**
+
+### Agent Tools
+
+- `agent_spawn(prompt, agent_type, description)` - Spawn background agent with full tool access
+- `agent_output(task_id, block)` - Get results (block=True to wait)
+- `agent_progress(task_id)` - Check real-time progress
+- `agent_list()` - Overview of all running agents
+- `agent_cancel(task_id)` - Stop a running agent
+
+### Agent Types
+
+- `explore` - Codebase search, "where is X?" questions
+- `dewey` - Documentation research, implementation examples
+- `frontend` - UI/UX work, component design
+- `delphi` - Strategic advice, architecture review
+
+### Parallel Execution (MANDATORY)
+
+For ANY task with 2+ independent steps:
+
+1. **Immediately use agent_spawn** for each independent component
+2. Fire all agents simultaneously, don't wait
+3. Monitor with agent_progress, collect with agent_output
+
+### ULTRATHINK / ULTRAWORK
+
+- **ULTRATHINK**: Engage exhaustive deep reasoning, multi-dimensional analysis
+- **ULTRAWORK**: Maximum parallel execution - spawn agents aggressively for every subtask
+```
+
+## Tools (31)
+
+| Category         | Tools                                                                              |
+| ---------------- | ---------------------------------------------------------------------------------- |
+| **Model Invoke** | `invoke_gemini`, `invoke_openai`, `get_system_health`                              |
+| **Environment**  | `get_project_context`, `task_spawn`, `task_status`, `task_list`                    |
+| **Agents**       | `agent_spawn`, `agent_output`, `agent_cancel`, `agent_list`, `agent_progress`      |
+| **Code Search**  | `ast_grep_search`, `ast_grep_replace`, `grep_search`, `glob_files`                 |
+| **LSP**          | `lsp_diagnostics`, `lsp_hover`, `lsp_goto_definition`, `lsp_find_references`, etc. |
+| **Sessions**     | `session_list`, `session_read`, `session_search`                                   |
+| **Skills**       | `skill_list`, `skill_get`                                                          |
+
+## Agent Prompts (7)
+
+| Prompt            | Purpose                                                       |
+| ----------------- | ------------------------------------------------------------- |
+| `stravinsky`      | Task orchestration, planning, and goal-oriented execution.    |
+| `delphi`          | Strategic technical advisor (GPT-based) for hard debugging.   |
+| `dewey`           | Documentation and multi-repository research specialist.       |
+| `explore`         | Specialized for codebase-wide search and structural analysis. |
+| `frontend`        | UI/UX Engineer (Gemini-optimized) for component prototyping.  |
+| `document_writer` | Technical documentation and specification writer.             |
+| `multimodal`      | Visual analysis expert for UI screenshots and diagrams.       |
+
+## Development
+
+```bash
+# Install in development mode
+uv pip install -e .
+
+# Run server
+stravinsky
+```
+
+## Project Structure
+
+```
+stravinsky/
+├── mcp_bridge/           # Python MCP server
+│   ├── server.py         # Entry point
+│   ├── auth/             # OAuth (Google & OpenAI)
+│   ├── tools/            # Model invoke, search, skills
+│   ├── prompts/          # Agent system prompts
+│   └── config/           # Bridge configuration
+├── pyproject.toml        # Build system
+└── README.md             # This file
+```
+
+## Troubleshooting
+
+### OpenAI "Port 1455 in use"
+
+The Codex CLI uses the same port. Stop it with: `killall codex`
+
+### OpenAI Authentication Failed
+
+- Ensure you have a ChatGPT Plus/Pro subscription
+- Tokens expire occasionally; run `stravinsky-auth login openai` to refresh
+
+## License
+
+MIT
+
+---
+
+<div align="center">
+  <small>Built with obsession by the Google Deepmind team.</small>
+</div>

stravinsky-0.1.12.dist-info/RECORD

@@ -0,0 +1,34 @@
+mcp_bridge/__init__.py,sha256=8hnj-sTcl7-N11RY05wAGqveB_WtfEciYsxjQkXAnmc,156
+mcp_bridge/server.py,sha256=3HbPqRCXN7N94pXh5MzElSN-sU9pHuuf0ntX4r3xOGA,34139
+mcp_bridge/auth/__init__.py,sha256=AGHNtKzqvZYMLQ35Qg6aOabpxBqmkR-pjXv8Iby9oMw,797
+mcp_bridge/auth/cli.py,sha256=z894Jb5buAtzFv8455JuzGeghT-cXphgyp82LryM_NU,6767
+mcp_bridge/auth/oauth.py,sha256=lFTVYN6kjXWAqqYpg9L6XbWYI-Pg-_LFvc4WZAlzM9k,12450
+mcp_bridge/auth/openai_oauth.py,sha256=0Ks2X-NXLCBzqs3xnbj9QLZpugICOX5qB5y5vtDENOo,11522
+mcp_bridge/auth/token_store.py,sha256=3A6TZJ7Wju6QfhALeX4IMhY5jzb9OWMrDzwRbfAukiU,5650
+mcp_bridge/config/__init__.py,sha256=uapHdrSWWrafVKD9CTB1J_7Dw0_RajRhoDGjy9zH21o,256
+mcp_bridge/config/hooks.py,sha256=WvWC6ZUc8y1IXPlGCjLYAAsGGigd5tWeGiw585OGNwA,4624
+mcp_bridge/prompts/__init__.py,sha256=pQmPxbdfS_S-ENB-VX33MXbgfGs8zHfLB_QOBtaPjis,322
+mcp_bridge/prompts/delphi.py,sha256=OgsG5l9lsomj4rpohCtdU28ZW15gugCPG9eYwN340d4,4910
+mcp_bridge/prompts/dewey.py,sha256=5-hRlOIJbfat9G_Lj-N-yhA7xk8c9lSOhK9yhSBqIs0,5261
+mcp_bridge/prompts/document_writer.py,sha256=hiCbxgTU8HKPJkS0eNpPPtzSqDXPreApU2OqiS6zh-0,5618
+mcp_bridge/prompts/explore.py,sha256=121R3bNFbb7AIte69eDtLfxdyHdOo3l16L6cUH1p-kk,3592
+mcp_bridge/prompts/frontend.py,sha256=j91I8k5vcVed13eeX-Ebiv49x9Qj4HO_SQN1xhB8TLQ,4943
+mcp_bridge/prompts/multimodal.py,sha256=Svw11N392LjshalasOd80X0Qw_qtOMqu_lD-_HmQDIo,1936
+mcp_bridge/prompts/stravinsky.py,sha256=dhZnB_gv4mK1s_9kEVmWGbR7q-Pn8Z98exJD_PiYvYQ,12635
+mcp_bridge/tools/__init__.py,sha256=YjNSlDzQ3xKezT-MlMfv3OXS9mlW86Xge0YBKpm1ACw,905
+mcp_bridge/tools/agent_manager.py,sha256=OaRdaJOhkblwuIXTwVfoyboQ8g-8_nI1h5iqnYM6u4A,22274
+mcp_bridge/tools/background_tasks.py,sha256=G1iQ00538q939-DJbzaEsA0uZI2TfwzawFCDx8QC5Hg,5178
+mcp_bridge/tools/code_search.py,sha256=sR-alLQuxaXUFB9hby_wQsQu3Io644wdnpdOM_vm0aw,9978
+mcp_bridge/tools/continuous_loop.py,sha256=MM3FnF3ULuR32h0tqJP8uF48iJg6R9dbyHy_36KLOls,2100
+mcp_bridge/tools/model_invoke.py,sha256=1oXd8rIQShd5ZLV4OdH8C-cwM42c4dCzBxLvD3rnrro,7059
+mcp_bridge/tools/project_context.py,sha256=bXKxuW1pGjtIbeNjMgpBoQL-d_CI94UPBVpRjUyhX20,4707
+mcp_bridge/tools/session_manager.py,sha256=tCVLLvO-Kttla7OxPImb_NSGL_9aW46ilq5ej_IcnlA,9252
+mcp_bridge/tools/skill_loader.py,sha256=6IhlEEPdNyLjqbtrPMnZdXz7u0KpGpR6EQkXykeC1VQ,6092
+mcp_bridge/tools/task_runner.py,sha256=OCYAXy7vZTHf_np1IITv7azLlwomBYqYI9qu6oQna44,2888
+mcp_bridge/tools/lsp/__init__.py,sha256=fLiII9qgeachI3MlkO6uGulfUH3T0YDeyEfO65bbxdw,549
+mcp_bridge/tools/lsp/tools.py,sha256=d0gR8WuOjxLKWIbIXw2n1k6jUOrkoZW7C7TuA08OQwg,16225
+mcp_bridge/utils/__init__.py,sha256=pbHV4nq5SLUYcAyTmLUZYrp293Ctud57X8hwsMGA_BM,20
+stravinsky-0.1.12.dist-info/METADATA,sha256=dtLX88AW28vQNUgzx3nZka--SVrkC2esotJ6bvutY24,6890
+stravinsky-0.1.12.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+stravinsky-0.1.12.dist-info/entry_points.txt,sha256=BISwF7i71Oen7jFVmBXz8fxiU11Cp415wPF0xXG2Q3s,97
+stravinsky-0.1.12.dist-info/RECORD,,

stravinsky-0.1.12.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

stravinsky-0.1.12.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+stravinsky = mcp_bridge.server:main
+stravinsky-auth = mcp_bridge.auth.cli:main