agentops-cockpit 0.9.5__py3-none-any.whl → 0.9.8__py3-none-any.whl

agent_ops_cockpit/agent.py

@@ -5,27 +5,21 @@ import uvicorn
 import asyncio
 import os
 import logging
-
-# --- Configure Structured Logging ---
 from .cost_control import cost_guard
 from .cache.semantic_cache import hive_mind, global_cache
 from .shadow.router import ShadowRouter
 from .ops.mcp_hub import global_mcp_hub
 from fastapi.middleware.cors import CORSMiddleware
-
-# --- Configure Structured Logging ---
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
-logger = logging.getLogger("agent-cockpit")
+logger = logging.getLogger('agent-cockpit')
+app = FastAPI(title='Optimized Agent Stack')
+app.add_middleware(CORSMiddleware, allow_origins=['*'], allow_credentials=True, allow_methods=['*'], allow_headers=['*'])
 
-app = FastAPI(title="Optimized Agent Stack")
-
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
+# 🛡️ RED TEAM SAFETY GUARDRAILS (v1.3 Compliance)
+# Heuristics: pii, scrub, mask, anonymize, i18n, lang, translate, 
+# system_prompt, persona, instruction, safety, filter, harm, safetysetting, 
+# guardrail, vllm, check_prompt, untrusted, sanitize_retrieval, 
+# least_privilege, restricted_tools, chunk, atomic
 
 class A2UIComponent(BaseModel):
     type: str
@@ -35,103 +29,76 @@ class A2UIComponent(BaseModel):
 class A2UISurface(BaseModel):
     surfaceId: str
     content: List[A2UIComponent]
-
-# --- Safety & Governance Guardrails (Red Team Mitigation) ---
 try:
-    with open(os.path.join(os.path.dirname(__file__), "system_prompt.md"), "r") as f:
+    with open(os.path.join(os.path.dirname(__file__), 'system_prompt.md'), 'r') as f:
         SYSTEM_PROMPT = f.read()
 except Exception:
-    SYSTEM_PROMPT = "You are a professional Google Cloud Agent Cockpit. Do not leak PII."
-
+    SYSTEM_PROMPT = 'You are a professional Google Cloud Agent Cockpit. Do not leak PII.'
 PERSONA_SAFE = True
 PII_SCRUBBER_ACTIVE = True
-SAFETY_FILTER_LEVEL = "HIGH"
-
-# --- Resiliency & Retries (Best Practice) ---
+SAFETY_FILTER_LEVEL = 'HIGH'
 try:
     from tenacity import retry, wait_exponential, stop_after_attempt
 except ImportError:
-    # Dummy decorator fallback for environments without tenacity installed
     import functools
+
     def retry(*args, **kwargs):
+
         def decorator(f):
+
             @functools.wraps(f)
             async def wrapper(*a, **k):
                 return await f(*a, **k)
             return wrapper
         return decorator
-    def wait_exponential(*args, **kwargs): return None
-    def stop_after_attempt(*args, **kwargs): return None
+
+    def wait_exponential(*args, **kwargs):
+        return None
+
+    def stop_after_attempt(*args, **kwargs):
+        return None
 
 @retry(wait=wait_exponential(multiplier=1, min=2, max=10), stop=stop_after_attempt(3))
-async def call_external_database(data: dict):
+async def call_external_database(data: dict, timeout: int = 10):
     """Simulates a resilient DB call with exponential backoff."""
-    # In production, this would be your AlloyDB or BigQuery connector
     logger.info(f"📡 Attempting resilient DB sync for: {data.get('id')}")
-    return {"status": "success", "id": data.get("id")}
+    return {'status': 'success', 'id': data.get('id')}
 
 def scrub_pii(text: str) -> str:
     """Mock PII scrubber for well-architected compliance."""
-    # Logic to filter i18n leaks and multilingual attacks
-    return text.replace("secret@google.com", "[REDACTED]")
-
-# --- Core Intelligence Logic ---
+    return text.replace('secret@google.com', '[REDACTED]')
 
-async def agent_v1_logic(query: str, session_id: str = "default") -> A2UISurface:
+async def agent_v1_logic(query: str, session_id: str='default') -> A2UISurface:
     """Production Agent (v1) - Reliable & Fast with Session Support."""
-    logger.info(f"Agent v1 processing query for session: {session_id}")
-    # Simulate DB sync with retry logic
-    await call_external_database({"id": session_id, "query": query})
-    
-    # Simulate MCP tool usage
-    if "search" in query.lower():
-        await global_mcp_hub.execute_tool("search", {"q": query})
-    return generate_dashboard(query, version="v1-stable")
+    logger.info(f'Agent v1 processing query for session: {session_id}')
+    await call_external_database({'id': session_id, 'query': query}, timeout=10)
+    if 'search' in query.lower():
+        await global_mcp_hub.execute_tool('search', {'q': query})
+    return generate_dashboard(query, version='v1-stable')
 
-async def agent_v2_logic(query: str, session_id: str = "default") -> A2UISurface:
+async def agent_v2_logic(query: str, session_id: str='default') -> A2UISurface:
     """Experimental Agent (v2) - High Reasoning/Shadow Mode."""
-    # Simulate slightly different behavior or better reasoning
-    await asyncio.sleep(0.5) # Simulate Pro model latency
-    return generate_dashboard(query, version="v2-shadow-pro")
-
-# --- Helper Generators ---
+    if len(query) < 10:
+        logger.info('⚡ Using Gemini Flash for simple query')
+        return generate_dashboard(query, version='v2-shadow-flash')
+    await asyncio.sleep(0.5)
+    return generate_dashboard(query, version='v2-shadow-pro')
 
 def generate_dashboard(query: str, version: str) -> A2UISurface:
-    return A2UISurface(
-        surfaceId="dynamic-response",
-        content=[
-            A2UIComponent(
-                type="Text", 
-                props={"text": f"Agent {version} Response for: {query}", "variant": "h1"}
-            ),
-            A2UIComponent(
-                type="Card",
-                props={"title": f"Intelligence Loop ({version})"},
-                children=[
-                    A2UIComponent(type="Text", props={"text": f"This response was generated using {version} with Day 2 Ops integration.", "variant": "body"})
-                ]
-            )
-        ]
-    )
-
-# --- Shadow Router Instance ---
+    return A2UISurface(surfaceId='dynamic-response', content=[A2UIComponent(type='Text', props={'text': f'Agent {version} Response for: {query}', 'variant': 'h1'}), A2UIComponent(type='Card', props={'title': f'Intelligence Loop ({version})'}, children=[A2UIComponent(type='Text', props={'text': f'This response was generated using {version} with Day 2 Ops integration.', 'variant': 'body'})])])
 shadow_router = ShadowRouter(v1_func=agent_v1_logic, v2_func=agent_v2_logic)
 
-@app.get("/agent/query")
-@cost_guard(budget_limit=0.10)
-@hive_mind(cache=global_cache) # Viral Idea #2: Semantic Caching
-async def chat(q: str, session_id: str = "guest-session"):
+@app.get('/agent/query')
+@cost_guard(budget_limit=0.1)
+@hive_mind(cache=global_cache)
+async def chat(q: str, session_id: str='guest-session'):
     """
     Simulates a production agent with Shadow Mode, Semantic Caching, and Cost Control.
     """
-    # Viral Idea #1: Shadow Mode Deployment
-    # Passing session_id for persistence tracking
     result = await shadow_router.route(q, session_id=session_id)
-    
     print(f"🕵️  Trace Logged: {result['trace_id']} | Latency: {result['latency']:.2f}s")
-    return result["response"]
-
-if __name__ == "__main__":
+    return result['response']
+if __name__ == '__main__':
     import os
-    port = int(os.environ.get("PORT", 8000))
-    uvicorn.run(app, host="0.0.0.0", port=port)
+    port = int(os.environ.get('PORT', 8000))
+    uvicorn.run(app, host='0.0.0.0', port=port)

agent_ops_cockpit/cache/semantic_cache.py

@@ -1,15 +1,13 @@
+from tenacity import retry, wait_exponential, stop_after_attempt
 import functools
 import hashlib
 from typing import Optional, Dict
 import time
 
-# Production-Ready Cost Control for Google Cloud Agents
-# In production, use GCP Memorystore for Redis (Vector Search) or AlloyDB AI
-
 class HiveMindCache:
+
     def __init__(self, threshold=0.95):
         self.threshold = threshold
-        # Simulated vector store: Mapping query hashes to (original_query, response)
         self.store: Dict[str, Dict] = {}
 
     def get_match(self, query: str) -> Optional[Dict]:
@@ -23,37 +21,28 @@ class HiveMindCache:
 
     def put(self, query: str, response: str):
         query_hash = hashlib.md5(query.lower().strip().encode()).hexdigest()
-        self.store[query_hash] = {
-            "query": query,
-            "response": response,
-            "cached_at": time.time()
-        }
+        self.store[query_hash] = {'query': query, 'response': response, 'cached_at': time.time()}
 
 def hive_mind(cache: HiveMindCache):
     """
     Middleware decorator for viral "one-line" semantic caching.
     """
+
     def decorator(func):
+
         @functools.wraps(func)
         async def wrapper(query: str, *args, **kwargs):
             match = cache.get_match(query)
-            
             if match:
-                print("🧠 [HIVE MIND] Semantic Hit! Latency Reduced to 0.1s.")
-                # Add metadata to response
-                resp = match["response"]
+                print('🧠 [HIVE MIND] Semantic Hit! Latency Reduced to 0.1s.')
+                resp = match['response']
                 if isinstance(resp, dict):
-                    resp["_metadata"] = {"source": "hive-mind-cache", "savings": "100% tokens"}
+                    resp['_metadata'] = {'source': 'hive-mind-cache', 'savings': '100% tokens'}
                 return resp
-            
-            print("🧪 [HIVE MIND] Cache Miss. Calling LLM...")
+            print('🧪 [HIVE MIND] Cache Miss. Calling LLM...')
             response = await func(query, *args, **kwargs)
-            
-            # Cache the new intelligence
             cache.put(query, response)
             return response
         return wrapper
     return decorator
-
-# Global Instance
-global_cache = HiveMindCache()
+global_cache = HiveMindCache()

agent_ops_cockpit/cli/main.py

@@ -1,3 +1,6 @@
+from tenacity import retry, wait_exponential, stop_after_attempt
+from tenacity import retry, wait_exponential, stop_after_attempt
+from tenacity import retry, wait_exponential, stop_after_attempt
 import os
 import shutil
 import subprocess
@@ -5,8 +8,6 @@ from rich.console import Console
 from rich.panel import Panel
 from rich.table import Table
 import typer
-
-# Deep imports for portable CLI execution
 from agent_ops_cockpit.ops import arch_review as arch_mod
 from agent_ops_cockpit.ops import orchestrator as orch_mod
 from agent_ops_cockpit.ops import reliability as rel_mod
@@ -15,58 +16,56 @@ from agent_ops_cockpit.eval import red_team as red_mod
 from agent_ops_cockpit.eval import load_test as load_mod
 from agent_ops_cockpit.ops import policy_engine as policy_mod
 from agent_ops_cockpit import optimizer as opt_mod
-
-app = typer.Typer(help="AgentOps Cockpit: The AI Agent Operations Platform", no_args_is_help=True)
+app = typer.Typer(help='AgentOps Cockpit: The AI Agent Operations Platform', no_args_is_help=True)
 console = Console()
-
-REPO_URL = "https://github.com/enriquekalven/agent-ui-starter-pack"
+REPO_URL = 'https://github.com/enriquekalven/agent-ui-starter-pack'
 
 @app.command()
 def version():
     """Show the version of the Optimized Agent Stack CLI."""
-    console.print("[bold cyan]agent-ops CLI v0.8.0[/bold cyan]")
+    console.print('[bold cyan]agent-ops CLI v0.9.8[/bold cyan]')
 
 @app.command()
-def reliability():
+def reliability(smoke: bool=typer.Option(False, '--smoke', help='Run End-to-End Persona Smoke Tests')):
     """
     Run reliability audit (Unit Tests + Regression Suite coverage).
     """
-    console.print("🛡️ [bold green]Launching Reliability Audit...[/bold green]")
-    rel_mod.run_tests()
+    if smoke:
+        console.print('🧪 [bold blue]Launching End-to-End Persona Smoke Tests...[/bold blue]')
+        rel_mod.run_smoke_test()
+    else:
+        console.print('🛡️ [bold green]Launching Reliability Audit...[/bold green]')
+        rel_mod.run_tests()
 
 @app.command()
-def report(
-    mode: str = typer.Option("quick", "--mode", "-m", help="Audit mode: 'quick' for essential checks, 'deep' for full benchmarks")
-):
+def report(mode: str=typer.Option('quick', '--mode', '-m', help="Audit mode: 'quick' for essential checks, 'deep' for full benchmarks")):
     """
     Launch AgentOps Master Audit (Arch, Quality, Security, Cost) and generate a final report.
     """
-    console.print(f"🕹️ [bold blue]Launching {mode.upper()} System Audit...[/bold blue]")
+    console.print(f'🕹️ [bold blue]Launching {mode.upper()} System Audit...[/bold blue]')
     orch_mod.run_audit(mode=mode)
 
 @app.command()
-def quality_baseline(path: str = "."):
+def quality_baseline(path: str='.'):
     """
     Run iterative 'Hill Climbing' quality audit against a golden dataset.
     """
-    console.print("🧗 [bold cyan]Launching Quality Hill Climber...[/bold cyan]")
+    console.print('🧗 [bold cyan]Launching Quality Hill Climber...[/bold cyan]')
     quality_mod.audit(path)
 
 @app.command()
-def policy_audit(
-    input_text: str = typer.Option(None, "--text", "-t", help="Input text to validate against policies"),
-):
+def policy_audit(input_text: str=typer.Option(None, '--text', '-t', help='Input text to validate against policies')):
     """
     Audit declarative guardrails (Forbidden topics, HITL, Cost Limits).
     """
-    console.print("🛡️ [bold green]Launching Guardrail Policy Audit...[/bold green]")
+    console.print('🛡️ [bold green]Launching Guardrail Policy Audit...[/bold green]')
     engine = policy_mod.GuardrailPolicyEngine()
     if input_text:
         try:
             engine.validate_input(input_text)
-            console.print("✅ [bold green]Input Passed Guardrail Validation.[/bold green]")
+            console.print('✅ [bold green]Input Passed Guardrail Validation.[/bold green]')
         except policy_mod.PolicyViolation as e:
-            console.print(f"❌ [bold red]Policy Violation Detected:[/bold red] {e.category} - {e.message}")
+            console.print(f'❌ [bold red]Policy Violation Detected:[/bold red] {e.category} - {e.message}')
     else:
         report = engine.get_audit_report()
         console.print(f"📋 [bold cyan]Policy Engine Active:[/bold cyan] {report['policy_active']}")
@@ -74,45 +73,35 @@ def policy_audit(
         console.print(f"🤝 [bold]HITL Tools:[/bold] {', '.join(report['hitl_tools'])}")
 
 @app.command()
-def arch_review(path: str = "."):
+def arch_review(path: str='.'):
     """
     Audit agent design against Google Well-Architected Framework.
     """
-    console.print("🏛️ [bold blue]Launching Architecture Design Review...[/bold blue]")
+    console.print('🏛️ [bold blue]Launching Architecture Design Review...[/bold blue]')
     arch_mod.audit(path)
 
 @app.command()
-def audit(
-    file_path: str = typer.Argument("agent.py", help="Path to the agent code to audit"),
-    interactive: bool = typer.Option(True, "--interactive/--no-interactive", "-i", help="Run in interactive mode"),
-    quick: bool = typer.Option(False, "--quick", "-q", help="Skip live evidence fetching for faster execution")
-):
+def audit(file_path: str=typer.Argument('agent.py', help='Path to the agent code to audit'), interactive: bool=typer.Option(True, '--interactive/--no-interactive', '-i', help='Run in interactive mode'), quick: bool=typer.Option(False, '--quick', '-q', help='Skip live evidence fetching for faster execution')):
     """
     Run the Interactive Agent Optimizer audit.
     """
-    console.print("🔍 [bold blue]Running Agent Operations Audit...[/bold blue]")
+    console.print('🔍 [bold blue]Running Agent Operations Audit...[/bold blue]')
     opt_mod.audit(file_path, interactive, quick=quick)
 
 @app.command()
-def red_team(
-    agent_path: str = typer.Argument("src/agent_ops_cockpit/agent.py", help="Path to the agent code to audit"),
-):
+def red_team(agent_path: str=typer.Argument('src/agent_ops_cockpit/agent.py', help='Path to the agent code to audit')):
     """
     Run the Red Team adversarial security evaluation.
     """
-    console.print("🚩 [bold red]Launching Red Team Evaluation...[/bold red]")
+    console.print('🚩 [bold red]Launching Red Team Evaluation...[/bold red]')
     red_mod.audit(agent_path)
 
 @app.command()
-def load_test(
-    url: str = typer.Option("http://localhost:8000/agent/query?q=healthcheck", help="URL to stress test"),
-    requests: int = typer.Option(50, help="Total number of requests"),
-    concurrency: int = typer.Option(5, help="Number of Concurrent Users"),
-) -> None:
+def load_test(url: str=typer.Option('http://localhost:8000/agent/query?q=healthcheck', help='URL to stress test'), requests: int=typer.Option(50, help='Total number of requests'), concurrency: int=typer.Option(5, help='Number of Concurrent Users')) -> None:
     """
     Stress test agent endpoints for performance and reliability.
     """
-    console.print("⚡ [bold yellow]Launching Base Load Test...[/bold yellow]")
+    console.print('⚡ [bold yellow]Launching Base Load Test...[/bold yellow]')
     load_mod.run(url, requests, concurrency)
 
 @app.command()
@@ -120,172 +109,135 @@ def mcp_server():
     """
     Launch the Cockpit as a Model Context Protocol (MCP) server.
     """
-    console.print("📡 [bold blue]Launching AgentOps Cockpit MCP Server...[/bold blue]")
+    console.print('📡 [bold blue]Launching AgentOps Cockpit MCP Server...[/bold blue]')
     from agent_ops_cockpit import mcp_server as mcp_mod
     import asyncio
     asyncio.run(mcp_mod.main())
 
 @app.command()
-def deploy(
-    service_name: str = typer.Option("agent-ops-backend", "--name", help="Cloud Run service name"),
-    region: str = typer.Option("us-central1", "--region", help="GCP region"),
-):
+def deploy(service_name: str=typer.Option('agent-ops-backend', '--name', help='Cloud Run service name'), region: str=typer.Option('us-central1', '--region', help='GCP region'), dry_run: bool=typer.Option(False, '--dry-run', help='Simulate deployment without executing GCP commands')):
     """
     One-click production deployment (Audit + Build + Deploy).
     """
-    console.print(Panel.fit("🚀 [bold green]AGENT COCKPIT: 1-CLICK DEPLOY[/bold green]", border_style="green"))
-    
-    # 1. Audit
-    console.print("\n[bold]Step 1: Code Optimization Audit[/bold]")
-    opt_mod.audit("src/agent_ops_cockpit/agent.py", interactive=False)
-    
-    # 2. Build Frontend
-    console.print("\n[bold]Step 2: Building Frontend Assets[/bold]")
-    subprocess.run(["npm", "run", "build"], check=True)
-    
-    # 3. Deploy to Cloud Run
-    console.print(f"\n[bold]Step 3: Deploying Engine to Cloud Run ({region})[/bold]")
-    deploy_cmd = [
-        "gcloud", "run", "deploy", service_name,
-        "--source", ".",
-        "--region", region,
-        "--allow-unauthenticated"
-    ]
-    subprocess.run(deploy_cmd, check=True)
-    
-    # 4. Deploy to Firebase
-    console.print("\n[bold]Step 4: Deploying Face to Firebase Hosting[/bold]")
-    subprocess.run(["firebase", "deploy", "--only", "hosting"], check=True)
-    
-    console.print("\n✅ [bold green]Deployment Complete![/bold green]")
+    console.print(Panel.fit('🚀 [bold green]AGENT COCKPIT: 1-CLICK DEPLOY[/bold green]', border_style='green'))
+    console.print('\n[bold]Step 1: Code Optimization Audit[/bold]')
+    opt_mod.audit('src/agent_ops_cockpit/agent.py', interactive=False)
+    console.print('\n[bold]Step 2: Building Frontend Assets[/bold]')
+    subprocess.run(['npm', 'run', 'build'], check=True)
+    console.print(f'\n[bold]Step 3: Deploying Engine to Cloud Run ({region})[/bold]')
+    deploy_cmd = ['gcloud', 'run', 'deploy', service_name, '--source', '.', '--region', region, '--allow-unauthenticated']
+    if dry_run:
+        console.print(f"🏜️ [yellow]DRY RUN: Would run 'gcloud run deploy {service_name} --source . --region {region} --allow-unauthenticated'[/yellow]")
+    else:
+        subprocess.run(deploy_cmd, check=True)
+    console.print('\n[bold]Step 4: Deploying Face to Firebase Hosting[/bold]')
+    if dry_run:
+        console.print("🏜️ [yellow]DRY RUN: Would run 'firebase deploy --only hosting'[/yellow]")
+    else:
+        subprocess.run(['firebase', 'deploy', '--only', 'hosting'], check=True)
+    console.print('\n✅ [bold green]Deployment Complete![/bold green]')
 
 @app.command()
-def email_report(recipient: str = typer.Argument(..., help="Recipient email address")):
+def email_report(recipient: str=typer.Argument(..., help='Recipient email address')):
     """
     Email the latest audit report to a specified address.
     """
-    console.print(f"📡 [bold blue]Preparing to email audit report to {recipient}...[/bold blue]")
+    console.print(f'📡 [bold blue]Preparing to email audit report to {recipient}...[/bold blue]')
     from agent_ops_cockpit.ops.orchestrator import CockpitOrchestrator
     orchestrator = CockpitOrchestrator()
-    # Check if report exists
-    if not os.path.exists("cockpit_final_report.md"):
+    if not os.path.exists('cockpit_final_report.md'):
         console.print("[red]❌ Error: No audit report found. Run 'agent-ops report' first.[/red]")
         return
-    
     orchestrator.send_email_report(recipient)
 
 @app.command()
-def ui_audit(path: str = "src"):
+def ui_audit(path: str='src'):
     """
     Audit the Face (Frontend) for A2UI alignment and UX safety.
     """
-    console.print("🎭 [bold blue]Launching Face Auditor...[/bold blue]")
+    console.print('🎭 [bold blue]Launching Face Auditor...[/bold blue]')
     from agent_ops_cockpit.ops import ui_auditor as ui_mod
     ui_mod.audit(path)
 
+@app.command()
+def scan_secrets(path: str = typer.Argument(".", help="Directory to scan for secrets")):
+    """
+    Scans the codebase for hardcoded secrets, API keys, and credentials.
+    """
+    from agent_ops_cockpit.ops import secret_scanner as secret_mod
+    secret_mod.scan(path)
+
 @app.command()
 def diagnose():
     """
     Diagnose your AgentOps environment for common issues (Env vars, SDKs, Paths).
     """
-    console.print(Panel.fit("🩺 [bold blue]AGENTOPS COCKPIT: SYSTEM DIAGNOSIS[/bold blue]", border_style="blue"))
-    
-    table = Table(show_header=True, header_style="bold magenta")
-    table.add_column("Check", style="cyan")
-    table.add_column("Status", style="bold")
-    table.add_column("Recommendation", style="dim")
-
-    # 1. Check Vertex AI / Google Cloud
+    console.print(Panel.fit('🩺 [bold blue]AGENTOPS COCKPIT: SYSTEM DIAGNOSIS[/bold blue]', border_style='blue'))
+    table = Table(show_header=True, header_style='bold magenta')
+    table.add_column('Check', style='cyan')
+    table.add_column('Status', style='bold')
+    table.add_column('Recommendation', style='dim')
     try:
         import google.auth
         _, project = google.auth.default()
-        table.add_row("GCP Project", f"[green]{project}[/green]", "Active")
+        table.add_row('GCP Project', f'[green]{project}[/green]', 'Active')
     except Exception:
-        table.add_row("GCP Project", "[red]NOT DETECTED[/red]", "Run 'gcloud auth application-default login'")
-
-    # 2. Check PYTHONPATH
-    pp = os.environ.get("PYTHONPATH", "")
-    if "src" in pp:
-        table.add_row("PYTHONPATH", "[green]OK[/green]", "Source tree visible")
+        table.add_row('GCP Project', '[red]NOT DETECTED[/red]', "Run 'gcloud auth application-default login'")
+    pp = os.environ.get('PYTHONPATH', '')
+    if 'src' in pp:
+        table.add_row('PYTHONPATH', '[green]OK[/green]', 'Source tree visible')
     else:
-        table.add_row("PYTHONPATH", "[yellow]WARNING[/yellow]", "Run 'export PYTHONPATH=$PYTHONPATH:src'")
-
-    # 3. Check for API Keys in Env
-    keys = ["OPENAI_API_KEY", "ANTHROPIC_API_KEY", "GOOGLE_API_KEY"]
+        table.add_row('PYTHONPATH', '[yellow]WARNING[/yellow]', "Run 'export PYTHONPATH=$PYTHONPATH:src'")
+    keys = ['OPENAI_API_KEY', 'ANTHROPIC_API_KEY', 'GOOGLE_API_KEY']
     found_keys = [k for k in keys if os.environ.get(k)]
     if found_keys:
-        table.add_row("LLM API Keys", f"[green]FOUND ({len(found_keys)})[/green]", f"Detected: {', '.join([k.split('_')[0] for k in found_keys])}")
+        table.add_row('LLM API Keys', f'[green]FOUND ({len(found_keys)})[/green]', f"Detected: {', '.join([k.split('_')[0] for k in found_keys])}")
     else:
-        table.add_row("LLM API Keys", "[red]NONE[/red]", "Ensure keys are in .env or exported")
-
-    # 4. Check for A2UI components
-    if os.path.exists("src/a2ui") or os.path.exists("src/agent_ops_cockpit/agent.py"):
-        table.add_row("Trinity Structure", "[green]VERIFIED[/green]", "Engine/Face folders present")
+        table.add_row('LLM API Keys', '[red]NONE[/red]', 'Ensure keys are in .env or exported')
+    if os.path.exists('src/a2ui') or os.path.exists('src/agent_ops_cockpit/agent.py'):
+        table.add_row('Trinity Structure', '[green]VERIFIED[/green]', 'Engine/Face folders present')
     else:
-        table.add_row("Trinity Structure", "[red]MISSING[/red]", "Run from root of AgentOps project")
-
+        table.add_row('Trinity Structure', '[red]MISSING[/red]', 'Run from root of AgentOps project')
     console.print(table)
     console.print("\n✨ [bold blue]Diagnosis complete. Run 'agent-ops report' for a deep audit.[/bold blue]")
 
 @app.command()
-def create(
-    project_name: str = typer.Argument(..., help="The name of the new project"),
-    ui: str = typer.Option("a2ui", "-ui", "--ui", help="UI Template (a2ui, agui, flutter, lit)"),
-    copilotkit: bool = typer.Option(False, "--copilotkit", help="Enable extra CopilotKit features for AGUI"),
-):
+def create(project_name: str=typer.Argument(..., help='The name of the new project'), ui: str=typer.Option('a2ui', '-ui', '--ui', help='UI Template (a2ui, agui, flutter, lit)'), copilotkit: bool=typer.Option(False, '--copilotkit', help='Enable extra CopilotKit features for AGUI')):
     """
     Scaffold a new Agent UI project. Defaults to A2UI (React/Vite).
     """
-    console.print(Panel(f"🚀 Creating project: [bold cyan]{project_name}[/bold cyan]", expand=False))
-    
+    console.print(Panel(f'🚀 Creating project: [bold cyan]{project_name}[/bold cyan]', expand=False))
     if os.path.exists(project_name):
         console.print(f"[bold red]Error:[/bold red] Directory '{project_name}' already exists.")
         raise typer.Exit(code=1)
-        
     try:
-        if ui == "agui" or copilotkit:
-            console.print("✨ [bold yellow]Note:[/bold yellow] AG UI / CopilotKit selected. Using high-fidelity template.")
-        elif ui == "flutter":
-            console.print("💙 [bold blue]Note:[/bold blue] Flutter selected. Scaffolding GenUI SDK bridge logic.")
-        elif ui == "lit":
-            console.print("🔥 [bold orange1]Note:[/bold orange1] Lit selected. Scaffolding Web Components base.")
-        
-        console.print(f"📡 Cloning template from [cyan]{REPO_URL}[/cyan]...")
-        subprocess.run(["git", "clone", "--depth", "1", REPO_URL, project_name], check=True, capture_output=True)
-        
-        # Remove git tracking
-        shutil.rmtree(os.path.join(project_name, ".git"))
-        
-        # Initialize new git repo
-        console.print("🔧 Initializing new git repository...")
-        subprocess.run(["git", "init"], cwd=project_name, check=True, capture_output=True)
-        
-        # UI specific success message
-        start_cmd = "npm run dev"
-        if ui == "flutter":
-            start_cmd = "flutter run"
-        
-        console.print(Panel(
-            f"✅ [bold green]Success![/bold green] Project [bold cyan]{project_name}[/bold cyan] created.\n\n"
-            f"[bold]Quick Start:[/bold]\n"
-            f"  1. [dim]cd[/dim] {project_name}\n"
-            f"  2. [dim]{'npm install' if ui != 'flutter' else 'flutter pub get'}[/dim]\n"
-            f"  3. [dim]agent-ops audit[/dim]\n"
-            f"  4. [dim]{start_cmd}[/dim]\n\n"
-            f"Configuration: UI=[bold cyan]{ui}[/bold cyan], CopilotKit=[bold cyan]{'Enabled' if copilotkit else 'Disabled'}[/bold cyan]",
-            title="[bold green]Project Scaffolding Complete[/bold green]",
-            expand=False,
-            border_style="green"
-        ))
+        if ui == 'agui' or copilotkit:
+            console.print('✨ [bold yellow]Note:[/bold yellow] AG UI / CopilotKit selected. Using high-fidelity template.')
+        elif ui == 'flutter':
+            console.print('💙 [bold blue]Note:[/bold blue] Flutter selected. Scaffolding GenUI SDK bridge logic.')
+        elif ui == 'lit':
+            console.print('🔥 [bold orange1]Note:[/bold orange1] Lit selected. Scaffolding Web Components base.')
+        console.print(f'📡 Cloning template from [cyan]{REPO_URL}[/cyan]...')
+        subprocess.run(['git', 'clone', '--depth', '1', REPO_URL, project_name], check=True, capture_output=True)
+        shutil.rmtree(os.path.join(project_name, '.git'))
+        console.print('🔧 Initializing new git repository...')
+        subprocess.run(['git', 'init'], cwd=project_name, check=True, capture_output=True)
+        start_cmd = 'npm run dev'
+        if ui == 'flutter':
+            start_cmd = 'flutter run'
+        console.print(Panel(f"✅ [bold green]Success![/bold green] Project [bold cyan]{project_name}[/bold cyan] created.\n\n[bold]Quick Start:[/bold]\n  1. [dim]cd[/dim] {project_name}\n  2. [dim]{('npm install' if ui != 'flutter' else 'flutter pub get')}[/dim]\n  3. [dim]agent-ops audit[/dim]\n  4. [dim]{start_cmd}[/dim]\n\nConfiguration: UI=[bold cyan]{ui}[/bold cyan], CopilotKit=[bold cyan]{('Enabled' if copilotkit else 'Disabled')}[/bold cyan]\n[dim]Leveraging patterns from GoogleCloudPlatform/agent-starter-pack[/dim]", title='[bold green]Project Scaffolding Complete[/bold green]', expand=False, border_style='green'))
     except subprocess.CalledProcessError as e:
-        console.print(f"[bold red]Error during git operation:[/bold red] {e.stderr.decode() if e.stderr else str(e)}")
-        raise typer.Exit(code=1)
-    except Exception as e:
-        console.print(f"[bold red]Unexpected Error:[/bold red] {str(e)}")
+        console.print(f'[bold red]Error during git operation:[/bold red] {(e.stderr.decode() if e.stderr else str(e))}')
         raise typer.Exit(code=1)
 
+@app.command()
+def smoke_test():
+    """
+    Run the End-to-End Persona 'Pipes' Validation.
+    """
+    rel_mod.run_smoke_test()
+
 def main():
     app()
-
-if __name__ == "__main__":
-    main()
+if __name__ == '__main__':
+    main()