agentops-cockpit 0.3.0__py3-none-any.whl → 0.4.0__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"])

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentops-cockpit
-Version: 0.3.0
+Version: 0.4.0
 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,20 @@ 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://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 +45,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 +102,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.
 

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

@@ -14,7 +14,7 @@ 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
@@ -23,8 +23,8 @@ agent_ops_cockpit/ops/secret_scanner.py,sha256=OKojiW8umarrp5ywS4InCTnzzky1hcdBm
 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.0.dist-info/METADATA,sha256=nNpVEjWuXzZupMXp2i0upE_RxTlj8FvInjCvDDBu2Hs,8319
+agentops_cockpit-0.4.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+agentops_cockpit-0.4.0.dist-info/entry_points.txt,sha256=SOGYPNtUGhMVgxLQ9dEYo7L3M_dvhWEU2eQz2zhaTkY,112
+agentops_cockpit-0.4.0.dist-info/licenses/LICENSE,sha256=XNJEk4bvf88tBnKqHdGBGi10l9yJWv2yLWPJvvVie1c,1071
+agentops_cockpit-0.4.0.dist-info/RECORD,,