agentops-cockpit 0.3.0__py3-none-any.whl → 0.4.1__py3-none-any.whl

agent_ops_cockpit/ops/mcp_hub.py

@@ -1,35 +1,80 @@
-from typing import List, Dict, Any
+from typing import List, Dict, Any, Optional
 import asyncio
+import json
+import os
+from mcp import ClientSession, StdioServerParameters
+from mcp.client.stdio import stdio_client
 
 class MCPHub:
     """
     Model Context Protocol (MCP) Hub.
-    Optimizes tool discovery, execution, and cost across multiple providers.
+    Provides a unified interface for tool discovery and execution across
+    multiple MCP servers (Google Search, SQL, internal tools).
     """
 
     def __init__(self):
+        self.servers: Dict[str, StdioServerParameters] = {}
         self.registry = {
-            "search": {"type": "mcp", "provider": "google-search", "status": "optimized"},
-            "db": {"type": "mcp", "provider": "alloydb-vector", "status": "optimized"},
-            "legacy_crm": {"type": "rest_api", "provider": "internal", "status": "deprecated"}
+            "search": {"type": "mcp", "provider": "google-search", "server": "google-search-mcp"},
+            "db": {"type": "mcp", "provider": "alloydb", "server": "postgres-mcp"},
+            "legacy_crm": {"type": "rest", "provider": "internal", "status": "deprecated"}
         }
 
-    async def execute_tool(self, tool_name: str, args: Dict[str, Any]):
+    def register_server(self, name: str, command: str, args: List[str] = None):
+        """Registers a local MCP server."""
+        self.servers[name] = StdioServerParameters(
+            command=command,
+            args=args or [],
+            env=os.environ.copy()
+        )
+
+    async def execute_tool(self, tool_name: str, arguments: Dict[str, Any]):
         """
-        Executes a tool via MCP if available, else falls back to legacy.
-        Logs metrics for the Flight Recorder.
+        Executes a tool call using the Model Context Protocol.
         """
         if tool_name not in self.registry:
-            raise ValueError(f"Tool {tool_name} not found in MCP Registry.")
-        
+            raise ValueError(f"Tool {tool_name} not found in MCP registry.")
+
         config = self.registry[tool_name]
         
-        if config["status"] == "deprecated":
-            print(f"⚠️  WARNING: Using legacy Tool API for '{tool_name}'. Migrate to MCP for 30% lower latency.")
-        
-        print(f"🛠️  Executing tool '{tool_name}' via {config['type']} protocol...")
-        await asyncio.sleep(0.1) # Simulating execution
-        
-        return {"result": f"Data from {tool_name}", "protocol": config["type"]}
+        # If it's a legacy tool, handle it separately
+        if config["type"] == "rest":
+            print(f"⚠️  Executing legacy REST tool: {tool_name}")
+            return await self._mock_legacy_exec(tool_name, arguments)
+
+        server_name = config.get("server")
+        if not server_name or server_name not in self.servers:
+            # Fallback to mock for demo/unconfigured environments
+            print(f"ℹ️  MCP Server '{server_name}' not configured. Running in simulated mode.")
+            return await self._mock_mcp_exec(tool_name, arguments)
+
+        # Real MCP Protocol Execution
+        async with stdio_client(self.servers[server_name]) as (read, write):
+            async with ClientSession(read, write) as session:
+                await session.initialize()
+                result = await session.call_tool(tool_name, arguments)
+                return {
+                    "result": result.content,
+                    "protocol": "mcp-v1",
+                    "server": server_name
+                }
+
+    async def _mock_mcp_exec(self, tool_name: str, args: Dict[str, Any]):
+        await asyncio.sleep(0.2)
+        return {
+            "result": f"Simulated MCP response for {tool_name}",
+            "protocol": "mcp-virtual",
+            "assurance": 0.95
+        }
+
+    async def _mock_legacy_exec(self, tool_name: str, args: Dict[str, Any]):
+        await asyncio.sleep(0.5)
+        return {
+            "result": f"Legacy response for {tool_name}",
+            "protocol": "rest-legacy",
+            "warning": "MIGRATE_TO_MCP"
+        }
 
 global_mcp_hub = MCPHub()
+# Example registration (commented out as it requires local binaries)
+# global_mcp_hub.register_server("google-search-mcp", "npx", ["-y", "@modelcontextprotocol/server-google-search"])

agent_ops_cockpit/ops/swarm.py

@@ -0,0 +1,71 @@
+import asyncio
+from typing import List, Dict, Any, Optional
+from dataclasses import dataclass
+from rich.console import Console
+from rich.panel import Panel
+
+console = Console()
+
+@dataclass
+class SwarmMessage:
+    sender: str
+    recipient: str
+    content: str
+    evidence_packet: Optional[Dict[str, Any]] = None
+
+class MultiAgentOrchestrator:
+    """
+    Standardizes Swarm/Coordinator patterns using the A2A spec.
+    """
+    
+    def __init__(self):
+        self.agents: Dict[str, Any] = {}
+        self.history: List[SwarmMessage] = []
+
+    def register_agent(self, name: str, agent_func):
+        self.agents[name] = agent_func
+        console.print(f"🤖 Agent [bold cyan]{name}[/bold cyan] registered in swarm.")
+
+    async def dispatch(self, sender: str, recipient: str, message: str):
+        """Dispatches a message with an A2A Reasoning Evidence Packet."""
+        console.print(f"\n📡 [dim]A2A Transmission:[/dim] [bold]{sender}[/bold] -> [bold]{recipient}[/bold]")
+        
+        # Simulated Evidence Packet for Governance
+        evidence = {
+            "assurance_score": 0.99,
+            "origin_vpc": "secure-engine-zone",
+            "pii_scrubbed": True
+        }
+        
+        swarm_msg = SwarmMessage(sender, recipient, message, evidence)
+        self.history.append(swarm_msg)
+        
+        if recipient in self.agents:
+            response = await self.agents[recipient](message, evidence)
+            return response
+        else:
+            return {"error": f"Agent {recipient} not found."}
+
+    def get_swarm_report(self):
+        console.print(Panel.fit("🐝 [bold]Swarm Orchestration Trace[/bold]", border_style="yellow"))
+        for msg in self.history:
+            console.print(f"[blue]{msg.sender}[/blue] -> [green]{msg.recipient}[/green]: {msg.content}")
+
+def run_swarm_demo():
+    orchestrator = MultiAgentOrchestrator()
+    
+    async def researcher(query, evidence):
+        return f"Research results for {query} (Evidence verified: {evidence['assurance_score']})"
+        
+    async def writer(query, evidence):
+        return f"Professional summary of {query}"
+
+    orchestrator.register_agent("Researcher", researcher)
+    orchestrator.register_agent("Writer", writer)
+    
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(orchestrator.dispatch("Orchestrator", "Researcher", "Analyze market trends"))
+    orchestrator.get_swarm_report()
+
+if __name__ == "__main__":
+    run_swarm_demo()

{agentops_cockpit-0.3.0.dist-info → agentops_cockpit-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentops-cockpit
-Version: 0.3.0
+Version: 0.4.1
 Summary: Production-grade Agent Operations (AgentOps) Platform
 Project-URL: Homepage, https://github.com/enriquekalven/agent-ops-cockpit
 Project-URL: Bug Tracker, https://github.com/enriquekalven/agent-ops-cockpit/issues
@@ -11,6 +11,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.10
 Requires-Dist: gitpython>=3.1.0
+Requires-Dist: mcp>=0.1.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: typer>=0.9.0
 Description-Content-Type: text/markdown
@@ -18,10 +19,22 @@ Description-Content-Type: text/markdown
 # 🕹️ AgentOps Cockpit
 
 <div align="center">
+  <img src="https://raw.githubusercontent.com/enriquekalven/agent-cockpit/main/public/og-image.png" alt="AgentOps Cockpit Social Preview" width="100%" />
+</div>
+
+<div align="center">
+  <br />
+  <a href="https://agent-cockpit.web.app" target="_blank"><strong>🌐 Official Website & Live Demo</strong></a>
+  <br /><br />
+  <a href="https://deploy.cloud.google.com?repo=https://github.com/enriquekalven/agent-cockpit">
+    <img src="https://deploy.cloud.google.com/button.svg" alt="Deploy to Google Cloud" />
+  </a>
+  <br />
+  <br />
   <img src="https://img.shields.io/github/stars/enriquekalven/agent-cockpit?style=for-the-badge&color=ffd700" alt="GitHub Stars" />
   <img src="https://img.shields.io/github/license/enriquekalven/agent-cockpit?style=for-the-badge&color=007bff" alt="License" />
   <img src="https://img.shields.io/badge/Google-Well--Architected-4285F4?style=for-the-badge&logo=google-cloud" alt="Google Well-Architected" />
-  <img src="https://img.shields.io/badge/Status-Day%202%20Operations-10b981?style=for-the-badge" alt="Status" />
+  <img src="https://img.shields.io/badge/A2A_Standard-Enabled-10b981?style=for-the-badge" alt="A2A Standard" />
 </div>
 
 <br />
@@ -34,7 +47,12 @@ Description-Content-Type: text/markdown
 ---
 
 ## 📽️ The Mission
-Most AI agent templates stop at a single Python file and an API key. **The AgentOps Cockpit** is for developers moving into production. While optimized for **ADK**, it provides framework-agnostic governance, safety, and cost guardrails for the entire agentic ecosystem—from CrewAI to LangGraph. Based on the **[Google Well-Architected Framework for Agents](/docs/google-architecture)**.
+Most AI agent templates stop at a single Python file and an API key. **The AgentOps Cockpit** is for developers moving into production. It provides framework-agnostic governance, safety, and cost guardrails for the entire agentic ecosystem.
+
+### Key Pillars:
+- **Governance-as-Code**: Audit your agent against [Google Well-Architected](/docs/google-architecture) best practices.
+- **Agentic Trinity**: Dedicated layers for the Engine (Logic), Face (UX), and Cockpit (Ops).
+- **A2A Connectivity**: Implements the [Agent-to-Agent Transmission Standard](/A2A_GUIDE.md) for secure swarm orchestration.
 
 ---
 
@@ -86,6 +104,9 @@ Don't wait for your users to find prompt injections. Use the built-in Adversaria
 ### 🏛️ Arch Review & Framework Detection
 Every agent in the cockpit is graded against a framework-aware checklist. The Cockpit intelligently detects your stack—**Google ADK**, **OpenAI Agentkit**, **Anthropic Claude**, **Microsoft AutoGen/Semantic Kernel**, **AWS Bedrock Agents**, or **CopilotKit**—and runs a tailored audit against corresponding production standards. Use `make arch-review` to verify your **Governance-as-Code**.
 
+### 🕹️ MCP Connectivity Hub (Model Context Protocol)
+Stop building one-off tool integrations. The Cockpit provides a unified hub for **MCP Servers**. Connect to Google Search, Slack, or your internal databases via the standardized Model Context Protocol for secure, audited tool execution.
+
 ### 🧗 Quality Hill Climbing (ADK Evaluation)
 Following **Google ADK Evaluation** best practices, the Cockpit provides an iterative optimization loop. `make quality-baseline` runs your agent against a "Golden Dataset" using **LLM-as-a-Judge** scoring (Response Match & Tool Trajectory), climbing the quality curve until production-grade fidelity is reached.
 
@@ -133,9 +154,9 @@ make deploy-prod   # 🚀 1-click deploy to Google Cloud
 ---
 
 ## 🧭 Roadmap
-- [ ] **One-Click GitHub Action**: Automated audits on every PR.
-- [ ] **Multi-Agent Orchestrator**: Support for Swarm/Coordinator patterns.
-- [ ] **Visual Mission Control**: Real-time observability dashboard.
+- [x] **One-Click GitHub Action**: Automated governance audits on every PR.
+- [x] **Multi-Agent Orchestrator**: Standardized A2A Swarm/Coordinator patterns.
+- [ ] **Visual Mission Control**: Real-time cockpit observability dashboard.
 
 [View full roadmap →](/ROADMAP.md)
 

{agentops_cockpit-0.3.0.dist-info → agentops_cockpit-0.4.1.dist-info}/RECORD

@@ -14,17 +14,18 @@ agent_ops_cockpit/ops/arch_review.py,sha256=o8ZKYSrmtt-dw74QBROObKz-w8Z-ZwC4G_yk
 agent_ops_cockpit/ops/cost_optimizer.py,sha256=fisPPo1hykcDBqljs05OG8xn0MBA_HPg7X8SlNDsx0M,1454
 agent_ops_cockpit/ops/evidence.py,sha256=LRAW57c-2R4ICiMLtc-JA1Tu5dlfO9-VBSUMc3TCLuo,1051
 agent_ops_cockpit/ops/frameworks.py,sha256=gJdisK8JOs79BY5x0yKu75Lu8WesgDcGJgQrjL9AE7U,19054
-agent_ops_cockpit/ops/mcp_hub.py,sha256=IcQNvHvbUhl-PbGPEWvKlUljNVAp8f9QMJR9gypJyE8,1360
+agent_ops_cockpit/ops/mcp_hub.py,sha256=Km-gXPNU0ulrFOe7fg9kPvhS3enataLKN8zyOoyzF6k,3184
 agent_ops_cockpit/ops/memory_optimizer.py,sha256=whsKhAuJkEJRa2dxfVeJC_xxwDwKjhx5tnmOmkiKgIQ,1635
 agent_ops_cockpit/ops/orchestrator.py,sha256=WnJ7nv99Ir7lvkWq0EIOEHE2rRzgJv2E4iRi8oDQcPc,3904
 agent_ops_cockpit/ops/pii_scrubber.py,sha256=HBRzzYv97f8VqIx2Gse9o6UVf6QWXSuop-xF-wVhuKU,1524
 agent_ops_cockpit/ops/reliability.py,sha256=Vuh7ZShjZQkXI8CWhL67LeacwEE75JNM6HgRTGLmt7o,2003
 agent_ops_cockpit/ops/secret_scanner.py,sha256=OKojiW8umarrp5ywS4InCTnzzky1hcdBmOfGa-uVIuE,3124
+agent_ops_cockpit/ops/swarm.py,sha256=wptxkdz-ORr4hqmMeQ3tiqw93U4y4XDBtu4xdVToqeQ,2457
 agent_ops_cockpit/ops/ui_auditor.py,sha256=3Cmc8i3oMQ9Wa0hSkeR0t_J8_s1c-u1_kj2PwxDGD6o,5542
 agent_ops_cockpit/shadow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agent_ops_cockpit/shadow/router.py,sha256=HRsgrrd3sQeabi58Ub8pOaDL9c7j4WpayeT9D8zPvOo,2725
-agentops_cockpit-0.3.0.dist-info/METADATA,sha256=lL6BFg_T3mHYyu_n8FeH4qhPZmgRRZbXYWcWmSt_InY,7385
-agentops_cockpit-0.3.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-agentops_cockpit-0.3.0.dist-info/entry_points.txt,sha256=SOGYPNtUGhMVgxLQ9dEYo7L3M_dvhWEU2eQz2zhaTkY,112
-agentops_cockpit-0.3.0.dist-info/licenses/LICENSE,sha256=XNJEk4bvf88tBnKqHdGBGi10l9yJWv2yLWPJvvVie1c,1071
-agentops_cockpit-0.3.0.dist-info/RECORD,,
+agentops_cockpit-0.4.1.dist-info/METADATA,sha256=mgrusibi7Cx9GwdqPkenRC--OmNMFV9KqZWce4Vsfo4,8471
+agentops_cockpit-0.4.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+agentops_cockpit-0.4.1.dist-info/entry_points.txt,sha256=SOGYPNtUGhMVgxLQ9dEYo7L3M_dvhWEU2eQz2zhaTkY,112
+agentops_cockpit-0.4.1.dist-info/licenses/LICENSE,sha256=XNJEk4bvf88tBnKqHdGBGi10l9yJWv2yLWPJvvVie1c,1071
+agentops_cockpit-0.4.1.dist-info/RECORD,,