ouroboros-ai 0.26.7.dev4__tar.gz → 0.26.7.dev5__tar.gz

{ouroboros_ai-0.26.7.dev4 → ouroboros_ai-0.26.7.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ouroboros-ai
-Version: 0.26.7.dev4
+Version: 0.26.7.dev5
 Summary: Specification-first workflow engine for AI coding agents. Works with Claude Code and Codex CLI.
 Author-email: Q00 <jqyu.lee@gmail.com>
 License-File: LICENSE

{ouroboros_ai-0.26.7.dev4 → ouroboros_ai-0.26.7.dev5}/src/ouroboros/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.26.7.dev4'
-__version_tuple__ = version_tuple = (0, 26, 7, 'dev4')
+__version__ = version = '0.26.7.dev5'
+__version_tuple__ = version_tuple = (0, 26, 7, 'dev5')
 
 __commit_id__ = commit_id = None

{ouroboros_ai-0.26.7.dev4 → ouroboros_ai-0.26.7.dev5}/src/ouroboros/mcp/server/adapter.py

@@ -569,6 +569,7 @@ def create_ouroboros_server(
     state_dir: Any | None = None,
     runtime_backend: str | None = None,
     llm_backend: str | None = None,
+    mcp_bridge: Any | None = None,
 ) -> MCPServerAdapter:
     """Create an Ouroboros MCP server with all tools and dependencies wired.
 
@@ -1239,6 +1240,15 @@ def create_ouroboros_server(
     # The server owns the shared event store lifecycle
     server.register_owned_resource(event_store)
 
+    # Inject bridge into all BridgeAwareMixin handlers (loop-based auto-discovery)
+    if mcp_bridge is not None:
+        from ouroboros.mcp.tools.bridge_mixin import inject_bridge
+
+        injected = [type(h).__name__ for h in tool_handlers if inject_bridge(h, mcp_bridge)]
+        if injected:
+            log.info("mcp.bridge.injected", handlers=injected)
+        server.register_owned_resource(mcp_bridge)
+
     # Register all tools with the server
     for handler in tool_handlers:
         server.register_tool(handler)

ouroboros_ai-0.26.7.dev5/src/ouroboros/mcp/tools/bridge_mixin.py

@@ -0,0 +1,52 @@
+"""BridgeAwareMixin for MCP tool handlers that need external MCP access.
+
+Handlers that inherit from this mixin will automatically receive
+an MCPClientManager reference when an MCPBridge is configured,
+via loop-based injection in the composition root.
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import Any
+
+
+@dataclass
+class BridgeAwareMixin:
+    """Mixin for handlers that need access to external MCP servers.
+
+    Provides ``mcp_manager`` and ``mcp_tool_prefix`` fields that are
+    injected by the composition root when an MCPBridge is configured.
+    Both fields default to None/"" so handlers work without a bridge.
+
+    Usage::
+
+        @dataclass
+        class MyHandler(BridgeAwareMixin):
+            other_field: str = ""
+
+            async def handle(self, arguments):
+                if self.mcp_manager:
+                    tools = await self.mcp_manager.list_all_tools()
+    """
+
+    mcp_manager: Any | None = field(default=None, repr=False)
+    mcp_tool_prefix: str = ""
+
+
+def inject_bridge(handler: object, bridge: object | None) -> bool:
+    """Inject bridge manager into a BridgeAwareMixin handler.
+
+    Args:
+        handler: A tool handler, possibly BridgeAwareMixin.
+        bridge: An MCPBridge instance (must have .manager and .tool_prefix).
+
+    Returns:
+        True if injection was performed, False otherwise.
+    """
+    if bridge is None or not isinstance(handler, BridgeAwareMixin):
+        return False
+
+    handler.mcp_manager = getattr(bridge, "manager", None)
+    handler.mcp_tool_prefix = getattr(bridge, "tool_prefix", "")
+    return True

{ouroboros_ai-0.26.7.dev4 → ouroboros_ai-0.26.7.dev5}/src/ouroboros/mcp/tools/execution_handlers.py

@@ -32,6 +32,7 @@ from ouroboros.core.worktree import (
 from ouroboros.evaluation.verification_artifacts import build_verification_artifacts
 from ouroboros.mcp.errors import MCPServerError, MCPToolError
 from ouroboros.mcp.job_manager import JobLinks, JobManager
+from ouroboros.mcp.tools.bridge_mixin import BridgeAwareMixin
 from ouroboros.mcp.types import (
     ContentType,
     MCPContentItem,
@@ -97,7 +98,7 @@ def _extract_inherited_effective_tools(arguments: dict[str, Any]) -> list[str] |
 
 
 @dataclass
-class ExecuteSeedHandler:
+class ExecuteSeedHandler(BridgeAwareMixin):
     """Handler for the execute_seed tool.
 
     Executes a seed (task specification) in the Ouroboros system.
@@ -394,6 +395,8 @@ class ExecuteSeedHandler:
                     adapter=agent_adapter,
                     event_store=event_store,
                     console=console,
+                    mcp_manager=self.mcp_manager,
+                    mcp_tool_prefix=self.mcp_tool_prefix,
                     debug=False,
                     enable_decomposition=True,
                     inherited_runtime_handle=inherited_runtime_handle,

ouroboros_ai-0.26.7.dev5/tests/unit/mcp/tools/test_mcp_manager_wiring.py

@@ -0,0 +1,145 @@
+"""Tests for mcp_manager wiring via BridgeAwareMixin and inject_bridge.
+
+Covers:
+- BridgeAwareMixin fields default to None/""
+- ExecuteSeedHandler inherits BridgeAwareMixin
+- inject_bridge auto-discovers and injects into mixin handlers
+- create_ouroboros_server loop-based bridge injection
+"""
+
+from __future__ import annotations
+
+from unittest.mock import MagicMock
+
+from ouroboros.mcp.tools.bridge_mixin import BridgeAwareMixin, inject_bridge
+from ouroboros.mcp.tools.definitions import get_ouroboros_tools
+from ouroboros.mcp.tools.execution_handlers import ExecuteSeedHandler
+
+
+class TestBridgeAwareMixin:
+    """Verify the mixin provides correct defaults."""
+
+    def test_handler_is_bridge_aware(self):
+        assert issubclass(ExecuteSeedHandler, BridgeAwareMixin)
+
+    def test_defaults_to_none(self):
+        handler = ExecuteSeedHandler()
+        assert handler.mcp_manager is None
+        assert handler.mcp_tool_prefix == ""
+
+    def test_accepts_mcp_manager_directly(self):
+        mock = MagicMock()
+        handler = ExecuteSeedHandler(mcp_manager=mock, mcp_tool_prefix="pfx_")
+        assert handler.mcp_manager is mock
+        assert handler.mcp_tool_prefix == "pfx_"
+
+
+class TestInjectBridge:
+    """Verify inject_bridge auto-discovery and injection."""
+
+    def test_injects_into_bridge_aware_handler(self):
+        handler = ExecuteSeedHandler()
+        bridge = MagicMock()
+        bridge.manager = MagicMock(name="FakeManager")
+        bridge.tool_prefix = "ext_"
+
+        result = inject_bridge(handler, bridge)
+
+        assert result is True
+        assert handler.mcp_manager is bridge.manager
+        assert handler.mcp_tool_prefix == "ext_"
+
+    def test_skips_non_bridge_aware_handler(self):
+        handler = MagicMock(spec=[])  # No BridgeAwareMixin
+        bridge = MagicMock()
+
+        result = inject_bridge(handler, bridge)
+        assert result is False
+
+    def test_skips_when_bridge_is_none(self):
+        handler = ExecuteSeedHandler()
+        result = inject_bridge(handler, None)
+        assert result is False
+        assert handler.mcp_manager is None
+
+    def test_handles_bridge_without_tool_prefix(self):
+        handler = ExecuteSeedHandler()
+        bridge = MagicMock(spec=["manager"])  # No tool_prefix attr
+        bridge.manager = MagicMock()
+
+        result = inject_bridge(handler, bridge)
+        assert result is True
+        assert handler.mcp_manager is bridge.manager
+        assert handler.mcp_tool_prefix == ""  # getattr fallback
+
+
+class TestGetOuroborosTools:
+    """Verify tools can be created and later injected."""
+
+    def test_default_tools_have_no_manager(self):
+        tools = get_ouroboros_tools()
+        exec_handler = tools[0]
+        assert isinstance(exec_handler, ExecuteSeedHandler)
+        assert exec_handler.mcp_manager is None
+
+    def test_tool_count_unchanged(self):
+        tools = get_ouroboros_tools()
+        assert len(tools) >= 20  # Sanity check
+
+
+class TestCompositionRoot:
+    """Verify create_ouroboros_server loop-based bridge injection."""
+
+    def test_bridge_injected_into_execute_handler(self):
+        from ouroboros.mcp.server.adapter import create_ouroboros_server
+
+        mock_bridge = MagicMock()
+        mock_bridge.manager = MagicMock(name="FakeManager")
+        mock_bridge.tool_prefix = "ext_"
+
+        server = create_ouroboros_server(mcp_bridge=mock_bridge)
+
+        exec_handler = None
+        for handler in server._tool_handlers.values():
+            if isinstance(handler, ExecuteSeedHandler):
+                exec_handler = handler
+                break
+
+        assert exec_handler is not None
+        assert exec_handler.mcp_manager is mock_bridge.manager
+        assert exec_handler.mcp_tool_prefix == "ext_"
+
+    def test_bridge_registered_as_owned_resource(self):
+        from ouroboros.mcp.server.adapter import create_ouroboros_server
+
+        mock_bridge = MagicMock()
+        mock_bridge.manager = MagicMock()
+        mock_bridge.tool_prefix = ""
+
+        server = create_ouroboros_server(mcp_bridge=mock_bridge)
+        assert mock_bridge in server._owned_resources
+
+    def test_no_bridge_leaves_handler_with_none(self):
+        from ouroboros.mcp.server.adapter import create_ouroboros_server
+
+        server = create_ouroboros_server()
+
+        exec_handler = None
+        for handler in server._tool_handlers.values():
+            if isinstance(handler, ExecuteSeedHandler):
+                exec_handler = handler
+                break
+
+        assert exec_handler is not None
+        assert exec_handler.mcp_manager is None
+
+    def test_no_bridge_not_in_owned_resources(self):
+        from ouroboros.mcp.server.adapter import create_ouroboros_server
+
+        server = create_ouroboros_server()
+        bridge_resources = [
+            r
+            for r in server._owned_resources
+            if hasattr(r, "tool_prefix") and hasattr(r, "manager")
+        ]
+        assert len(bridge_resources) == 0