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

agent_ops_cockpit/optimizer.py

@@ -8,24 +8,24 @@ from rich.table import Table
 from rich.panel import Panel
 from rich.syntax import Syntax
 from packaging import version
-
-# Import the evidence bridge
 try:
     from agent_ops_cockpit.ops.evidence_bridge import get_package_evidence, get_compatibility_report
 except ImportError:
-    # Fallback for local execution
     try:
         from backend.ops.evidence_bridge import get_package_evidence, get_compatibility_report
     except ImportError:
-        # Final fallback
-        def get_package_evidence(pkg): return {}
-        def get_compatibility_report(imports): return []
 
-app = typer.Typer(help="AgentOps Cockpit: The Agent Optimizer CLI")
+        def get_package_evidence(pkg):
+            return {}
+
+        def get_compatibility_report(imports):
+            return []
+app = typer.Typer(help='AgentOps Cockpit: The Agent Optimizer CLI')
 console = Console()
 
 class OptimizationIssue:
-    def __init__(self, id: str, title: str, impact: str, savings: str, description: str, diff: str, package: str = None, fix_pattern: str = None):
+
+    def __init__(self, id: str, title: str, impact: str, savings: str, description: str, diff: str, package: str=None, fix_pattern: str=None):
         self.id = id
         self.title = title
         self.impact = impact
@@ -36,450 +36,227 @@ class OptimizationIssue:
         self.fix_pattern = fix_pattern
         self.evidence = None
 
-def analyze_code(content: str, file_path: str = "agent.py", versions: Dict[str, str] = None) -> List[OptimizationIssue]:
+def analyze_code(content: str, file_path: str='agent.py', versions: Dict[str, str]=None) -> List[OptimizationIssue]:
     issues = []
     content_lower = content.lower()
+    content_no_comments = re.sub('#.*', '', content_lower)
     versions = versions or {}
-
-    # --- SITUATIONAL PLATFORM OPTIMIZATIONS ---
-
-    v_ai = versions.get("google-cloud-aiplatform", "Not Installed")
-    if "google.cloud.aiplatform" in content_lower or "vertexai" in content_lower:
-        if v_ai == "Not Installed":
-             issues.append(OptimizationIssue(
-                "vertex_install", "Install Modern Vertex SDK", "HIGH", "90% cost savings",
-                "You appear to be using Vertex AI logic but the SDK is not in your environment. Install v1.70.0+ to unlock context caching.",
-                "+ # pip install google-cloud-aiplatform>=1.70.0",
-                package="google-cloud-aiplatform"
-            ))
-        elif v_ai != "Unknown":
+    v_ai = versions.get('google-cloud-aiplatform', 'Not Installed')
+    if 'google.cloud.aiplatform' in content_lower or 'vertexai' in content_lower:
+        if v_ai == 'Not Installed':
+            issues.append(OptimizationIssue('vertex_install', 'Install Modern Vertex SDK', 'HIGH', '90% cost savings', 'You appear to be using Vertex AI logic but the SDK is not in your environment. Install v1.70.0+ to unlock context caching.', '+ # pip install google-cloud-aiplatform>=1.70.0', package='google-cloud-aiplatform'))
+        elif v_ai != 'Unknown':
             try:
-                if version.parse(v_ai) < version.parse("1.70.0"):
-                     issues.append(OptimizationIssue(
-                        "vertex_legacy_opt", "Situational Performance (Legacy SDK)", "MEDIUM", "20% cost savings",
-                        f"Your SDK ({v_ai}) lacks native Context Caching. Optimize by using selective prompt pruning before execution.",
-                        "+ from agent_ops_cockpit.ops.cost_optimizer import situational_pruning\n+ pruned = situational_pruning(context)",
-                        package="google-cloud-aiplatform"
-                    ))
-                     issues.append(OptimizationIssue(
-                        "vertex_upgrade_path", "Modernization Path", "HIGH", "90% cost savings",
-                        "Upgrading to 1.70.0+ enables near-instant token reuse via CachingConfig.",
-                        "+ # Upgrade to >1.70.0",
-                        package="google-cloud-aiplatform"
-                    ))
-                elif "cache" not in content_lower:
-                    issues.append(OptimizationIssue(
-                        "context_caching", "Enable Context Caching", "HIGH", "90% cost reduction",
-                        "Large model context detected. Use native CachingConfig.",
-                        "+ cache = vertexai.preview.CachingConfig(ttl=3600)",
-                        package="google-cloud-aiplatform"
-                    ))
+                if version.parse(v_ai) < version.parse('1.70.0'):
+                    issues.append(OptimizationIssue('vertex_legacy_opt', 'Situational Performance (Legacy SDK)', 'MEDIUM', '20% cost savings', f'Your SDK ({v_ai}) lacks native Context Caching. Optimize by using selective prompt pruning before execution.', '+ from agent_ops_cockpit.ops.cost_optimizer import situational_pruning\n+ pruned = situational_pruning(context)', package='google-cloud-aiplatform'))
+                    issues.append(OptimizationIssue('vertex_upgrade_path', 'Modernization Path', 'HIGH', '90% cost savings', 'Upgrading to 1.70.0+ enables near-instant token reuse via CachingConfig.', '+ # Upgrade to >1.70.0', package='google-cloud-aiplatform'))
+                elif 'cache' not in content_lower:
+                    issues.append(OptimizationIssue('context_caching', 'Enable Context Caching', 'HIGH', '90% cost reduction', 'Large model context detected. Use native CachingConfig.', '+ cache = vertexai.preview.CachingConfig(ttl=3600)', package='google-cloud-aiplatform'))
             except Exception:
                 pass
-
-    # OpenAI
-    openai_v = versions.get("openai", "Not Installed")
-    if "openai" in content_lower:
-        if openai_v != "Not Installed" and version.parse(openai_v) < version.parse("1.0.0"):
-             issues.append(OptimizationIssue(
-                "openai_legacy", "Found Legacy OpenAI SDK", "HIGH", "40% latency reduction",
-                f"You are on {openai_v}. Transitioning to the v1.0.0+ Client pattern enables modern streaming and improved error handling.",
-                "+ from openai import OpenAI\n+ client = OpenAI()",
-                package="openai"
-            ))
-        elif "prompt_cache" not in content_lower:
-             issues.append(OptimizationIssue(
-                "openai_caching", "OpenAI Prompt Caching", "MEDIUM", "50% latency reduction",
-                "OpenAI automatically caches repeated input prefixes. Ensure your system prompt is first.",
-                "+ # Ensure system prompt is first\n+ messages = [{'role': 'system', ...}]",
-                package="openai"
-            ))
-
-    # Anthropic
-    if ("anthropic" in content_lower or "claude" in content_lower) and "orchestra" not in content_lower:
-        issues.append(OptimizationIssue(
-            "anthropic_orchestration", "Anthropic Orchestration Pattern", "HIGH", "30% reliability boost",
-            "Claude performs best with an Orchestrator-Subagent pattern for complex tasks.",
-            "+ # Use orchestrator to delegate sub-tasks",
-            package="anthropic"
-        ))
-
-    # Microsoft
-    if ("autogen" in content_lower or "microsoft" in content_lower) and "workflow" not in content_lower:
-        issues.append(OptimizationIssue(
-            "ms_workflows", "Microsoft Agent Workflows", "MEDIUM", "40% consistency boost",
-            "Using graph-based repeatable workflows ensures enterprise reliability.",
-            "+ # Define a repeatable graph-based flow",
-            package="pyautogen"
-        ))
-
-    # AWS
-    if ("bedrock" in content_lower or "boto3" in content_lower) and "actiongroup" not in content_lower:
-        issues.append(OptimizationIssue(
-            "aws_action_groups", "AWS Bedrock Action Groups", "HIGH", "50% tool reliability",
-            "Standardize tool execution via Bedrock Action Group schemas.",
-            "+ # Define Bedrock Action Group",
-            package="aws-sdk"
-        ))
-
-    # CopilotKit
-    if "copilotkit" in content_lower and "usecopilotstate" not in content_lower:
-        issues.append(OptimizationIssue(
-            "copilot_state", "CopilotKit Shared State", "MEDIUM", "60% UI responsiveness",
-            "Ensure the Face remains aligned with the Engine via shared state sync.",
-            "+ # Use shared state for UI alignment",
-            package="@copilotkit/react-core"
-        ))
-
-    # Routing
-    if "pro" in content_lower and "flash" not in content_lower:
-        issues.append(OptimizationIssue(
-            "model_routing", "Smart Model Routing", "HIGH", "70% cost savings",
-            "Route simple queries to Flash models to minimize consumption.",
-            "+ if is_simple(q): model = 'gemini-1.5-flash'",
-            package="google-cloud-aiplatform"
-        ))
-
-    # Infrastructure (Cloud Run + GKE)
-    if "cloud run" in content_lower and "cpu_boost" not in content_lower:
-        issues.append(OptimizationIssue(
-            "cr_startup_boost", "Cloud Run Startup Boost", "HIGH", "50% latency reduction",
-            "Enable Startup CPU Boost to reduce cold-start latency for Python agents.",
-            "+ startup_cpu_boost: true",
-            package="google-cloud-run"
-        ))
-    if ("gke" in content_lower or "kubernetes" in content_lower) and "identity" not in content_lower:
-        issues.append(OptimizationIssue(
-            "gke_identity", "GKE Workload Identity", "HIGH", "100% security baseline",
-            "Use Workload Identity for secure service-to-service communication.",
-            "+ # Use Workload Identity",
-            package="google-cloud-gke"
-        ))
-
-    # Language Specific (Go + Node)
-    if file_path.endswith(".go") and "goroutine" not in content_lower:
-        issues.append(OptimizationIssue(
-            "go_concurrency", "Go Native Concurrency", "HIGH", "80% throughput boost",
-            "Leveraging Goroutines for parallel tool execution is a Go best practice.",
-            "+ go func() { tool.execute() }()",
-            package="golang"
-        ))
-    if (file_path.endswith(".ts") or file_path.endswith(".js") or "axios" in content_lower) and "fetch" not in content_lower:
-        issues.append(OptimizationIssue(
-            "node_native_fetch", "Native Fetch API", "MEDIUM", "20% bundle reduction",
-            "Node 20+ supports native fetch, reducing dependency on heavy libraries like axios.",
-            "+ const res = await fetch(url);",
-            package="nodejs"
-        ))
-
-    lg_v = versions.get("langgraph", "Not Installed")
-    if "langgraph" in content_lower:
-        if lg_v != "Not Installed" and lg_v != "Unknown":
+    openai_v = versions.get('openai', 'Not Installed')
+    if 'openai' in content_lower:
+        if openai_v != 'Not Installed' and version.parse(openai_v) < version.parse('1.0.0'):
+            issues.append(OptimizationIssue('openai_legacy', 'Found Legacy OpenAI SDK', 'HIGH', '40% latency reduction', f'You are on {openai_v}. Transitioning to the v1.0.0+ Client pattern enables modern streaming and improved error handling.', '+ from openai import OpenAI\n+ client = OpenAI()', package='openai'))
+        elif 'prompt_cache' not in content_lower:
+            issues.append(OptimizationIssue('openai_caching', 'OpenAI Prompt Caching', 'MEDIUM', '50% latency reduction', 'OpenAI automatically caches repeated input prefixes. Ensure your system prompt is first.', "+ # Ensure system prompt is first\n+ messages = [{'role': 'system', ...}]", package='openai'))
+    if ('anthropic' in content_lower or 'claude' in content_lower) and 'orchestra' not in content_lower:
+        issues.append(OptimizationIssue('anthropic_orchestration', 'Anthropic Orchestration Pattern', 'HIGH', '30% reliability boost', 'Claude performs best with an Orchestrator-Subagent pattern for complex tasks.', '+ # Use orchestrator to delegate sub-tasks', package='anthropic'))
+    if ('autogen' in content_lower or 'microsoft' in content_lower) and 'workflow' not in content_lower:
+        issues.append(OptimizationIssue('ms_workflows', 'Microsoft Agent Workflows', 'MEDIUM', '40% consistency boost', 'Using graph-based repeatable workflows ensures enterprise reliability.', '+ # Define a repeatable graph-based flow', package='pyautogen'))
+    if ('bedrock' in content_lower or 'boto3' in content_lower) and 'actiongroup' not in content_lower:
+        issues.append(OptimizationIssue('aws_action_groups', 'AWS Bedrock Action Groups', 'HIGH', '50% tool reliability', 'Standardize tool execution via Bedrock Action Group schemas.', '+ # Define Bedrock Action Group', package='aws-sdk'))
+    if 'copilotkit' in content_lower and 'usecopilotstate' not in content_lower:
+        issues.append(OptimizationIssue('copilot_state', 'CopilotKit Shared State', 'MEDIUM', '60% UI responsiveness', 'Ensure the Face remains aligned with the Engine via shared state sync.', '+ # Use shared state for UI alignment', package='@copilotkit/react-core'))
+    if 'pro' in content_lower and 'flash' not in content_lower:
+        issues.append(OptimizationIssue('model_routing', 'Smart Model Routing', 'HIGH', '70% cost savings', 'Route simple queries to Flash models to minimize consumption.', "+ if is_simple(q): model = 'gemini-1.5-flash'", package='google-cloud-aiplatform'))
+    if 'cloud run' in content_lower and 'cpu_boost' not in content_lower:
+        issues.append(OptimizationIssue('cr_startup_boost', 'Cloud Run Startup Boost', 'HIGH', '50% latency reduction', 'Enable Startup CPU Boost to reduce cold-start latency for Python agents.', '+ startup_cpu_boost: true', package='google-cloud-run'))
+    if ('gke' in content_lower or 'kubernetes' in content_lower) and 'identity' not in content_lower:
+        issues.append(OptimizationIssue('gke_identity', 'GKE Workload Identity', 'HIGH', '100% security baseline', 'Use Workload Identity for secure service-to-service communication.', '+ # Use Workload Identity', package='google-cloud-gke'))
+    if file_path.endswith('.go') and 'goroutine' not in content_lower:
+        issues.append(OptimizationIssue('go_concurrency', 'Go Native Concurrency', 'HIGH', '80% throughput boost', 'Leveraging Goroutines for parallel tool execution is a Go best practice.', '+ go func() { tool.execute() }()', package='golang'))
+    if (file_path.endswith('.ts') or file_path.endswith('.js') or 'axios' in content_lower) and 'fetch' not in content_lower:
+        issues.append(OptimizationIssue('node_native_fetch', 'Native Fetch API', 'MEDIUM', '20% bundle reduction', 'Node 20+ supports native fetch, reducing dependency on heavy libraries like axios.', '+ const res = await fetch(url);', package='nodejs'))
+    lg_v = versions.get('langgraph', 'Not Installed')
+    if 'langgraph' in content_lower:
+        if lg_v != 'Not Installed' and lg_v != 'Unknown':
             try:
-                if version.parse(lg_v) < version.parse("0.1.0"):
-                     issues.append(OptimizationIssue(
-                        "langgraph_legacy", "Situational Stability (Legacy LangGraph)", "HIGH", "Stability Boost",
-                        f"You are on {lg_v}. Older versions lack the hardened StateGraph compilation. Upgrade is recommended.",
-                        "+ # Consider upgrading for better persistence",
-                        package="langgraph"
-                    ))
+                if version.parse(lg_v) < version.parse('0.1.0'):
+                    issues.append(OptimizationIssue('langgraph_legacy', 'Situational Stability (Legacy LangGraph)', 'HIGH', 'Stability Boost', f'You are on {lg_v}. Older versions lack the hardened StateGraph compilation. Upgrade is recommended.', '+ # Consider upgrading for better persistence', package='langgraph'))
             except Exception:
                 pass
-        
-        if "persistence" not in content_lower and "checkpointer" not in content_lower:
-            issues.append(OptimizationIssue(
-                "langgraph_persistence", "LangGraph Persistence", "HIGH", "100% state recovery",
-                "A checkpointer is mandatory for reliable long-running agents.",
-                "+ graph.compile(checkpointer=checkpointer)",
-                package="langgraph"
-            ))
-        if "recursion" not in content_lower:
-            issues.append(OptimizationIssue(
-                "langgraph_recursion", "Recursion Limits", "MEDIUM", "Safety Guardrail",
-                "Set recursion limits to prevent expensive infinite loops in cyclic graphs.",
-                "+ graph.invoke(..., config={'recursion_limit': 50})",
-                package="langgraph"
-            ))
-
-    # --- ARCHITECTURAL OPTIMIZATIONS ---
-
-    # Large system instructions
-    large_string_pattern = re.compile(r'"""[\s\S]{200,}"""|\'\'\'[\s\S]{200,}\'\'\'')
-    if large_string_pattern.search(content) and "cache" not in content_lower:
-        issues.append(OptimizationIssue(
-            "context_caching", "Enable Context Caching", "HIGH", "90% cost reduction",
-            "Large static system instructions detected. Use context caching.",
-            "+ cache = vertexai.preview.CachingConfig(ttl=3600)",
-            package="google-cloud-aiplatform"
-        ))
-
-    # Missing semantic cache
-    if "hive_mind" not in content_lower and "cache" not in content_lower:
-         issues.append(OptimizationIssue(
-            "semantic_caching", "Implement Semantic Caching", "HIGH", "40-60% savings",
-            "No caching layer detected. Adding a semantic cache reduces LLM costs.",
-            "+ @hive_mind(cache=global_cache)",
-            package="google-adk"
-        ))
-
-    # --- BEST PRACTICE OPTIMIZATIONS ---
-
-    # Prompt Externalization
-    if large_string_pattern.search(content):
-        issues.append(OptimizationIssue(
-            "external_prompts", "Externalize System Prompts", "MEDIUM", "Architectural Debt Reduction",
-            "Keeping large system prompts in code makes them hard to version and test. Move them to 'system_prompt.md' and load dynamically.",
-            "+ with open('system_prompt.md', 'r') as f:\n+     SYSTEM_PROMPT = f.read()"
-        ))
-
-    # Resiliency / Retries
-    if "retry" not in content_lower and "tenacity" not in content_lower:
-        issues.append(OptimizationIssue(
-            "resiliency_retries", "Implement Exponential Backoff", "HIGH", "99.9% Reliability",
-            "Your agent calls external APIs/DBs but has no retry logic. Use 'tenacity' to handle transient failures.",
-            "+ @retry(wait=wait_exponential(multiplier=1, min=4, max=10), stop=stop_after_attempt(3))",
-            package="tenacity"
-        ))
-
-    # Session Management
-    if "session" not in content_lower and "conversation_id" not in content_lower:
-         issues.append(OptimizationIssue(
-            "session_management", "Add Session Tracking", "MEDIUM", "User Continuity",
-            "No session tracking detected. Agents in production need a 'conversation_id' to maintain multi-turn context.",
-            "+ def chat(q: str, conversation_id: str = None):"
-        ))
-
-    # Pinecone Optimization
-    if "pinecone" in content_lower:
-        if "grpc" not in content_lower:
-            issues.append(OptimizationIssue(
-                "pinecone_grpc", "Pinecone High-Perf (gRPC)", "MEDIUM", "40% latency reduction",
-                "You are using the standard Pinecone client. Switching to pinecone[grpc] enables low-latency streaming for large vector retrievals.",
-                "+ from pinecone.grpc import PineconeGRPC as Pinecone\n+ pc = Pinecone(api_key='...')"
-            ))
-        if "namespace" not in content_lower:
-            issues.append(OptimizationIssue(
-                "pinecone_isolation", "Pinecone Namespace Isolation", "MEDIUM", "RAG Accuracy Boost",
-                "No namespaces detected. Use namespaces to isolate user data or document segments for more accurate retrieval.",
-                "+ index.query(..., namespace='customer-a')"
-            ))
-
-    # Google Cloud Database Optimizations
+        if 'persistence' not in content_lower and 'checkpointer' not in content_lower:
+            issues.append(OptimizationIssue('langgraph_persistence', 'LangGraph Persistence', 'HIGH', '100% state recovery', 'A checkpointer is mandatory for reliable long-running agents.', '+ graph.compile(checkpointer=checkpointer)', package='langgraph'))
+        if 'recursion' not in content_lower:
+            issues.append(OptimizationIssue('langgraph_recursion', 'Recursion Limits', 'MEDIUM', 'Safety Guardrail', 'Set recursion limits to prevent expensive infinite loops in cyclic graphs.', "+ graph.invoke(..., config={'recursion_limit': 50})", package='langgraph'))
+    docstrings = re.findall('"""([\\s\\S]*?)"""|\\\'\\\'\\\'([\\s\\S]*?)\\\'\\\'\\\'', content)
+    has_large_prompt = any((len(d[0] or d[1]) > 200 for d in docstrings))
+    if has_large_prompt and 'cache' not in content_lower:
+        issues.append(OptimizationIssue('context_caching', 'Enable Context Caching', 'HIGH', '90% cost reduction', 'Large static system instructions detected. Use context caching.', '+ cache = vertexai.preview.CachingConfig(ttl=3600)', package='google-cloud-aiplatform'))
+    if 'hive_mind' not in content_lower and 'cache' not in content_lower:
+        issues.append(OptimizationIssue('semantic_caching', 'Implement Semantic Caching', 'HIGH', '40-60% savings', 'No caching layer detected. Adding a semantic cache reduces LLM costs.', '+ @hive_mind(cache=global_cache)', package='google-adk'))
+    if has_large_prompt:
+        issues.append(OptimizationIssue('external_prompts', 'Externalize System Prompts', 'MEDIUM', 'Architectural Debt Reduction', "Keeping large system prompts in code makes them hard to version and test. Move them to 'system_prompt.md' and load dynamically.", "+ with open('system_prompt.md', 'r') as f:\n+     SYSTEM_PROMPT = f.read()"))
+    if 'retry' not in content_lower and 'tenacity' not in content_lower:
+        issues.append(OptimizationIssue('resiliency_retries', 'Implement Exponential Backoff', 'HIGH', '99.9% Reliability', "Your agent calls external APIs/DBs but has no retry logic. Use 'tenacity' to handle transient failures.", '+ @retry(wait=wait_exponential(multiplier=1, min=4, max=10), stop=stop_after_attempt(3))', package='tenacity'))
+    if 'session' not in content_lower and 'conversation_id' not in content_lower:
+        issues.append(OptimizationIssue('session_management', 'Add Session Tracking', 'MEDIUM', 'User Continuity', "No session tracking detected. Agents in production need a 'conversation_id' to maintain multi-turn context.", '+ def chat(q: str, conversation_id: str = None):'))
+    if 'pinecone' in content_lower:
+        if 'grpc' not in content_lower:
+            issues.append(OptimizationIssue('pinecone_grpc', 'Pinecone High-Perf (gRPC)', 'MEDIUM', '40% latency reduction', 'You are using the standard Pinecone client. Switching to pinecone[grpc] enables low-latency streaming for large vector retrievals.', "+ from pinecone.grpc import PineconeGRPC as Pinecone\n+ pc = Pinecone(api_key='...')"))
+        if 'namespace' not in content_lower:
+            issues.append(OptimizationIssue('pinecone_isolation', 'Pinecone Namespace Isolation', 'MEDIUM', 'RAG Accuracy Boost', 'No namespaces detected. Use namespaces to isolate user data or document segments for more accurate retrieval.', "+ index.query(..., namespace='customer-a')"))
+    if 'alloydb' in content_no_comments:
+        if 'columnar' not in content_no_comments:
+            issues.append(OptimizationIssue('alloydb_columnar', 'AlloyDB Columnar Engine', 'HIGH', '100x Query Speedup', 'AlloyDB detected. Enable the Columnar Engine for analytical and AI-driven vector queries.', '+ # Enable AlloyDB Columnar Engine for vector scaling'))
+    if 'bigquery' in content_no_comments or 'bq' in content_no_comments:
+        if 'vector_search' not in content_no_comments:
+            issues.append(OptimizationIssue('bq_vector_search', 'BigQuery Vector Search', 'HIGH', 'FinOps: Serverless RAG', 'BigQuery detected. Use BQ Vector Search for cost-effective RAG over massive datasets without moving data to a separate DB.', '+ SELECT * FROM VECTOR_SEARCH(TABLE my_dataset.embeddings, ...)'))
+    if 'cloudsql' in content_lower or 'psycopg2' in content_lower or 'sqlalchemy' in content_lower:
+        if 'cloud-sql-connector' not in content_lower:
+            issues.append(OptimizationIssue('cloudsql_connector', 'Cloud SQL Python Connector', 'MEDIUM', '100% Secure Auth', 'Using raw drivers detected. Use the official Cloud SQL Python Connector for IAM-based authentication and automatic encryption.', '+ from google.cloud.sql.connector import Connector\n+ connector = Connector()'))
+    if 'firestore' in content_lower:
+        if 'vector' not in content_lower:
+            issues.append(OptimizationIssue('firestore_vector', 'Firestore Vector Search (Native)', 'HIGH', 'Real-time RAG', 'Firestore detected. Use native Vector Search and KNN queries for high-concurrency mobile/web agent retrieval.', "+ collection.find_nearest(vector_field='embedding', ...)"))
+    if 'oci' in content_lower or 'oracle' in content_lower:
+        if 'resource_principal' not in content_lower:
+            issues.append(OptimizationIssue('oci_auth', 'OCI Resource Principals', 'HIGH', '100% Secure Auth', 'Using static config/keys detected on OCI. Use Resource Principals for secure, credential-less access from OCI compute.', '+ auth = oci.auth.signers.get_resource_principals_signer()'))
+    if 'crewai' in content_lower or 'crew(' in content_lower:
+        if 'manager_agent' not in content_lower and 'hierarchical' not in content_lower:
+            issues.append(OptimizationIssue('crewai_manager', 'Use Hierarchical Manager', 'MEDIUM', '30% Coordination Boost', 'Your crew uses sequential execution. For complex tasks, a Manager Agent improves task handoffs and reasoning.', '+ crew = Crew(..., process=Process.hierarchical, manager_agent=manager)'))
     
-    # AlloyDB
-    if "alloydb" in content_lower:
-        if "columnar" not in content_lower:
-            issues.append(OptimizationIssue(
-                "alloydb_columnar", "AlloyDB Columnar Engine", "HIGH", "100x Query Speedup",
-                "AlloyDB detected. Enable the Columnar Engine for analytical and AI-driven vector queries.",
-                "+ # Enable AlloyDB Columnar Engine for vector scaling"
-            ))
+    # v1.2 Principal SME Extras
+    if 'rag' in content_lower or 'retriev' in content_lower:
+        if 'chunk' not in content_lower and 'atomic' not in content_lower:
+            issues.append(OptimizationIssue('atomic_rag', 'Implement Atomic RAG', 'HIGH', '30% Token Savings', "You appear to be using RAG but no 'chunking' or 'atomic retrieval' logic was detected. Sending full documents kills margins.", "+ docs = vector_db.search(query, chunk_limit=5)"))
             
-    # BigQuery
-    if "bigquery" in content_lower or "bq" in content_lower:
-        if "vector_search" not in content_lower:
-            issues.append(OptimizationIssue(
-                "bq_vector_search", "BigQuery Vector Search", "HIGH", "FinOps: Serverless RAG",
-                "BigQuery detected. Use BQ Vector Search for cost-effective RAG over massive datasets without moving data to a separate DB.",
-                "+ SELECT * FROM VECTOR_SEARCH(TABLE my_dataset.embeddings, ...)"
-            ))
+    if 'model' in content_lower and 'router' not in content_lower and 'is_simple' not in content_lower:
+        issues.append(OptimizationIssue('tiered_orchestration', 'Implement Tiered Orchestration', 'HIGH', '70% Cost Savings', "No model routing detected. Use a 'Router Agent' to decide if a query needs a Pro model or a Flash model.", "+ if is_simple(query): model = 'gemini-1.5-flash'"))
 
-    # Cloud SQL
-    if "cloudsql" in content_lower or "psycopg2" in content_lower or "sqlalchemy" in content_lower:
-        if "cloud-sql-connector" not in content_lower:
-            issues.append(OptimizationIssue(
-                "cloudsql_connector", "Cloud SQL Python Connector", "MEDIUM", "100% Secure Auth",
-                "Using raw drivers detected. Use the official Cloud SQL Python Connector for IAM-based authentication and automatic encryption.",
-                "+ from google.cloud.sql.connector import Connector\n+ connector = Connector()"
-            ))
+    if any(phrase in content_lower for phrase in ["you are a helpful assistant", "very good at", "please help me"]):
+        issues.append(OptimizationIssue('prompt_compression', 'Token Density: Redundant English', 'MEDIUM', '15% Token Savings', "Identified 'filler' tokens in system instructions. Compressing 'You are a helpful assistant who is very good at coding' to 'Expert coder' reduces baseline cost.", "- You are a helpful assistant...\n+ Expert coder"))
 
-    # Firestore
-    if "firestore" in content_lower:
-        if "vector" not in content_lower:
-            issues.append(OptimizationIssue(
-                "firestore_vector", "Firestore Vector Search (Native)", "HIGH", "Real-time RAG",
-                "Firestore detected. Use native Vector Search and KNN queries for high-concurrency mobile/web agent retrieval.",
-                "+ collection.find_nearest(vector_field='embedding', ...)"
-            ))
-
-    # Oracle OCI Optimizations
-    if "oci" in content_lower or "oracle" in content_lower:
-        if "resource_principal" not in content_lower:
-            issues.append(OptimizationIssue(
-                "oci_auth", "OCI Resource Principals", "HIGH", "100% Secure Auth",
-                "Using static config/keys detected on OCI. Use Resource Principals for secure, credential-less access from OCI compute.",
-                "+ auth = oci.auth.signers.get_resource_principals_signer()"
-            ))
-
-    # CrewAI Optimizations
-    if "crewai" in content_lower or "crew(" in content_lower:
-        if "manager_agent" not in content_lower and "hierarchical" not in content_lower:
-             issues.append(OptimizationIssue(
-                "crewai_manager", "Use Hierarchical Manager", "MEDIUM", "30% Coordination Boost",
-                "Your crew uses sequential execution. For complex tasks, a Manager Agent improves task handoffs and reasoning.",
-                "+ crew = Crew(..., process=Process.hierarchical, manager_agent=manager)"
-            ))
+    if 'model' in content_lower and 'retry' not in content_lower and 'tenacity' not in content_lower:
+        issues.append(OptimizationIssue('quota_management', 'Quota Management: Missing Backoff', 'HIGH', 'Resiliency & ROI', "High-volume model calls detected without Exponential Backoff. Failed requests due to rate-limiting represent wasted compute and broken ROI.", "+ @retry(wait=wait_exponential(multiplier=1, max=10))"))
 
     return issues
 
 def estimate_savings(token_count: int, issues: List[OptimizationIssue]) -> Dict[str, Any]:
     baseline_cost_per_m = 10.0
-    monthly_requests = 10000 
-    current_cost = (token_count / 1_000_000) * baseline_cost_per_m * monthly_requests
-    
+    monthly_requests = 10000
+    current_cost = token_count / 1000000 * baseline_cost_per_m * monthly_requests
     total_savings_pct = 0.0
     for issue in issues:
-        if "90%" in issue.savings:
-            total_savings_pct += 0.45  # Context Caching / Modern SDK
-        elif "70%" in issue.savings:
-            total_savings_pct += 0.35  # Smart Routing (Pro -> Flash)
-        elif "50%" in issue.savings:
-            total_savings_pct += 0.20  # Infrastructure / Startup Boost
-        elif "40-60%" in issue.savings:
-            total_savings_pct += 0.25  # Semantic Caching (Hive Mind)
+        if '90%' in issue.savings:
+            total_savings_pct += 0.45
+        elif '70%' in issue.savings:
+            total_savings_pct += 0.35
+        elif '50%' in issue.savings:
+            total_savings_pct += 0.2
+        elif '40-60%' in issue.savings:
+            total_savings_pct += 0.25
         else:
-            total_savings_pct += 0.05  # Standard Best Practices
-
+            total_savings_pct += 0.05
     projected_savings = current_cost * min(total_savings_pct, 0.85)
-    
-    return {
-        "current_monthly": current_cost,
-        "projected_savings": projected_savings,
-        "new_monthly": current_cost - projected_savings
-    }
+    return {'current_monthly': current_cost, 'projected_savings': projected_savings, 'new_monthly': current_cost - projected_savings}
 
 @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"),
-    apply_fix: bool = typer.Option(False, "--apply", "--fix", help="Automatically apply recommended fixes"),
-    quick: bool = typer.Option(False, "--quick", "-q", help="Skip live evidence fetching for faster execution")
-):
-    console.print(Panel.fit("🔍 [bold blue]GCP AGENT OPS: OPTIMIZER AUDIT[/bold blue]", border_style="blue"))
-    if quick:
-        console.print("[dim]⚡ Running in Quick Mode (skipping live evidence fetches)[/dim]")
-    console.print(f"Target: [yellow]{file_path}[/yellow]")
-    
+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'), apply_fix: bool=typer.Option(False, '--apply', '--fix', help='Automatically apply recommended fixes'), quick: bool=typer.Option(False, '--quick', '-q', help='Skip live evidence fetching for faster execution')):
+    console.print(Panel.fit('🔍 [bold blue]GCP AGENT OPS: OPTIMIZER AUDIT[/bold blue]', border_style='blue'))
     if not os.path.exists(file_path):
-        console.print(f"❌ [red]Error: File {file_path} not found.[/red]")
+        console.print(f'❌ [red]Error: Path {file_path} not found.[/red]')
         raise typer.Exit(1)
-
+    if os.path.isdir(file_path):
+        from agent_ops_cockpit.ops.discovery import DiscoveryEngine
+        discovery = DiscoveryEngine(file_path)
+        file_path = discovery.find_agent_brain()
+        
+        if not os.path.exists(file_path):
+            console.print(f'❌ [red]Error: No python entry point found in {discovery.root_path}[/red]')
+            raise typer.Exit(1)
+    console.print(f'Target: [yellow]{file_path}[/yellow]')
     with open(file_path, 'r') as f:
         content = f.read()
-    
-    # Heuristic: Find all imported packages
-    imports = re.findall(r"(?:from|import)\s+([\w\.-]+)", content)
-    
+    imports = re.findall('(?:from|import)\\s+([\\w\\.-]+)', content)
     from agent_ops_cockpit.ops.evidence_bridge import get_installed_version
-    package_versions = { pkg: get_installed_version(pkg) for pkg in ["google-cloud-aiplatform", "openai", "anthropic", "langgraph", "crewai"] }
-    
-    token_estimate = len(content.split()) * 1.5 
-    console.print(f"📊 Token Metrics: ~[bold]{token_estimate:.0f}[/bold] prompt tokens detected.")
-    
+    package_versions = {pkg: get_installed_version(pkg) for pkg in ['google-cloud-aiplatform', 'openai', 'anthropic', 'langgraph', 'crewai']}
+    token_estimate = len(content.split()) * 1.5
+    console.print(f'📊 Token Metrics: ~[bold]{token_estimate:.0f}[/bold] prompt tokens detected.')
     issues = analyze_code(content, file_path, versions=package_versions)
-    # Inject live evidence (skip in quick mode)
     if not quick:
         for issue in issues:
             if issue.package:
                 issue.evidence = get_package_evidence(issue.package)
-
-    # --- CROSS-PACKAGE VALIDATION ---
     comp_reports = get_compatibility_report(imports)
-    
     if comp_reports:
-        console.print("\n[bold yellow]🧩 Cross-Package Validation:[/bold yellow]")
+        console.print('\n[bold yellow]🧩 Cross-Package Validation:[/bold yellow]')
         for report in comp_reports:
-            if report["type"] == "INCOMPATIBLE":
+            if report['type'] == 'INCOMPATIBLE':
                 console.print(f"❌ [bold red]Conflict Detected:[/bold red] {report['component']} + {report['conflict_with']}")
                 console.print(f"   [dim]{report['reason']}[/dim]")
-            elif report["type"] == "SYNERGY":
+            elif report['type'] == 'SYNERGY':
                 console.print(f"✅ [bold green]Synergy Verified:[/bold green] {report['component']} is optimally paired.")
-
     if not issues:
-        console.print("\n[bold green]✅ No immediate code-level optimizations found. Your agent is lean![/bold green]")
+        console.print('\n[bold green]✅ No immediate code-level optimizations found. Your agent is lean![/bold green]')
         if not comp_reports:
-             return
+            return
         else:
-             raise typer.Exit(0)
-
+            raise typer.Exit(0)
     savings = estimate_savings(token_estimate, issues)
-    finops_panel = Panel(
-        f"💰 [bold]FinOps Projection (Est. 10k req/mo)[/bold]\n"
-        f"Current Monthly Spend: [red]${savings['current_monthly']:.2f}[/red]\n"
-        f"Projected Savings: [green]${savings['projected_savings']:.2f}[/green]\n"
-        f"New Monthly Spend: [blue]${savings['new_monthly']:.2f}[/blue]",
-        title="[bold yellow]Financial Optimization[/bold yellow]",
-        border_style="yellow"
-    )
+    finops_panel = Panel(f"💰 [bold]FinOps Projection (Est. 10k req/mo)[/bold]\nCurrent Monthly Spend: [red]${savings['current_monthly']:.2f}[/red]\nProjected Savings: [green]${savings['projected_savings']:.2f}[/green]\nNew Monthly Spend: [blue]${savings['new_monthly']:.2f}[/blue]", title='[bold yellow]Financial Optimization[/bold yellow]', border_style='yellow')
     console.print(finops_panel)
-
     applied = 0
     rejected = 0
     fixed_content = content
-
     for opt in issues:
-        console.print(f"\n[bold white on blue] --- [{opt.impact} IMPACT] {opt.title} --- [/bold white on blue]")
-        console.print(f"Benefit: [green]{opt.savings}[/green]")
-        console.print(f"Reason: {opt.description}")
-        
-        if opt.evidence and "error" not in opt.evidence:
+        console.print(f'\n[bold white on blue] --- [{opt.impact} IMPACT] {opt.title} --- [/bold white on blue]')
+        console.print(f'Benefit: [green]{opt.savings}[/green]')
+        console.print(f'Reason: {opt.description}')
+        if opt.evidence and 'error' not in opt.evidence:
             ev = opt.evidence
-            ev_title = "[dim]SDK Citation & Evidence[/dim]"
-            
-            # Highlight if an upgrade is required for maximum efficiency
-            if ev.get("upgrade_required"):
-                console.print("🚨 [bold yellow]URGENT UPGRADE RECOMMENDED[/bold yellow]")
+            ev_title = '[dim]SDK Citation & Evidence[/dim]'
+            if ev.get('upgrade_required'):
+                console.print('🚨 [bold yellow]URGENT UPGRADE RECOMMENDED[/bold yellow]')
                 console.print(f"   Current: {ev['installed_version']} | Required for optimization: >={ev['min_optimized_version']}")
-                ev_title = "[bold red]UPGRADE REQUIRED Evidence[/bold red]"
-
-            ev_panel = Panel(
-                f"🔗 [bold]Source[/bold]: {ev['source_url']}\n"
-                f"📅 [bold]Latest Release[/bold]: {ev['release_date'][:10]}\n"
-                f"📝 [bold]Note[/bold]: {ev['best_practice_context']}",
-                title=ev_title,
-                border_style="red" if ev.get("upgrade_required") else "dim"
-            )
+                ev_title = '[bold red]UPGRADE REQUIRED Evidence[/bold red]'
+            ev_panel = Panel(f"🔗 [bold]Source[/bold]: {ev['source_url']}\n📅 [bold]Latest Release[/bold]: {ev['release_date'][:10]}\n📝 [bold]Note[/bold]: {ev['best_practice_context']}", title=ev_title, border_style='red' if ev.get('upgrade_required') else 'dim')
             console.print(ev_panel)
-            # Orchestrator parsing
             console.print(f"SOURCE: {opt.title} | {ev['source_url']} | {ev['best_practice_context'].replace('\\n', ' ')}")
-
-        syntax = Syntax(opt.diff, "python", theme="monokai", line_numbers=False)
+        syntax = Syntax(opt.diff, 'python', theme='monokai', line_numbers=False)
         console.print(syntax)
-        
-        # Output ACTION: for report generation
-        console.print(f"ACTION: {file_path}:1 | Optimization: {opt.title} | {opt.description} (Est. {opt.savings})")
-        
+        console.print(f'ACTION: {file_path}:1 | Optimization: {opt.title} | {opt.description} (Est. {opt.savings})')
         do_apply = False
         if apply_fix:
             do_apply = True
         elif interactive:
-            do_apply = typer.confirm("\nDo you want to apply this code-level optimization?", default=True)
-        
+            do_apply = typer.confirm('\nDo you want to apply this code-level optimization?', default=True)
         if do_apply:
-            console.print("✅ [APPROVED] applying fix...")
+            console.print('✅ [APPROVED] applying fix...')
             if opt.fix_pattern:
                 fixed_content = opt.fix_pattern + fixed_content
             applied += 1
         else:
-            console.print("❌ [REJECTED] skipping optimization.")
+            console.print('❌ [REJECTED] skipping optimization.')
             rejected += 1
-
     if applied > 0:
         with open(file_path, 'w') as f:
             f.write(fixed_content)
-        console.print(f"\n✨ [bold green]Applied {applied} optimizations to {file_path}![/bold green]")
-    
-    summary_table = Table(title="🎯 AUDIT SUMMARY")
-    summary_table.add_column("Category", style="cyan")
-    summary_table.add_column("Count", style="magenta")
-    summary_table.add_row("Optimizations Applied", str(applied))
-    summary_table.add_row("Optimizations Rejected", str(rejected))
+        console.print(f'\n✨ [bold green]Applied {applied} optimizations to {file_path}![/bold green]')
+    summary_table = Table(title='🎯 AUDIT SUMMARY')
+    summary_table.add_column('Category', style='cyan')
+    summary_table.add_column('Count', style='magenta')
+    summary_table.add_row('Optimizations Applied', str(applied))
+    summary_table.add_row('Optimizations Rejected', str(rejected))
     console.print(summary_table)
-
-    # CI/CD Enforcement: Fail if high-impact issues remain in non-interactive mode
-    if not interactive and any(opt.impact == "HIGH" for opt in issues):
-        console.print("\n[bold red]❌ HIGH IMPACT issues detected. Optimization required for production.[/bold red]")
+    if not interactive and any((opt.impact == 'HIGH' for opt in issues)):
+        console.print('\n[bold red]❌ HIGH IMPACT issues detected. Optimization required for production.[/bold red]')
         raise typer.Exit(code=1)
 
-if __name__ == "__main__":
+@app.command()
+def version():
+    """Show the version of the Optimizer module."""
+    console.print('[bold cyan]v1.3.0[/bold cyan]')
+
+if __name__ == '__main__':
     app()