netra-zen 1.0.1__py3-none-any.whl → 1.0.2__py3-none-any.whl

{netra_zen-1.0.1.dist-info → netra_zen-1.0.2.dist-info}/RECORD

@@ -1,15 +1,15 @@
-zen_orchestrator.py,sha256=QApF_yEo7C8CUq_p8GGYJT2gPwMCLIMLSNIdFZFuxcI,144907
+zen_orchestrator.py,sha256=Q4bEnsyMYPJV6R9PutC6vrfxGiaOizlFjSW6qJDLrDI,148686
 agent_interface/__init__.py,sha256=OsbOKzElHsxhVgak87oOx_u46QNgKmz-Reis-plAMwk,525
 agent_interface/base_agent.py,sha256=GNskG9VaZgno7X24lQTpFdxUoQE0yJHLh0UPFJvOPn4,11098
-netra_zen-1.0.1.dist-info/licenses/LICENSE.md,sha256=PZrP0UDn58i4LjV4zijIQTnsQPvWm4zq9Fet9i7qgwI,80
+netra_zen-1.0.2.dist-info/licenses/LICENSE.md,sha256=t6LtOzAE2hgIIv5WbaN0wOcU3QCnGtAkMGNclHrKTOs,79
 token_budget/__init__.py,sha256=_2tmi72DGNtbYcZ-rGIxVKMytdkHFjzJaWz8bDhYACc,33
 token_budget/budget_manager.py,sha256=VRWxKcGDtgJfIRh-ztYQ4-wuhBvddVJJnyoGfxCBlv0,9567
 token_budget/models.py,sha256=14xFTk2-R1Ql0F9WLDof7vADrKC_5Fj7EE7UmZdoG00,2384
 token_budget/visualization.py,sha256=SaNnZ15edHXtjDCA5Yfu7w3AztCZRYsYCPGBqzapupY,719
 token_transparency/__init__.py,sha256=go86Rg4_VYAPLw3myVpLe1s1PbMu1llDTw1wfomP1lA,453
 token_transparency/claude_pricing_engine.py,sha256=9zWQJS3HJEs_lljil-xT1cUvG-Jf3ykNAninJFyfNSM,12630
-netra_zen-1.0.1.dist-info/METADATA,sha256=ixxjb8v_zQKUIIoMXOPhfSXdrawdMMPWsOxAUBu_xCE,30220
-netra_zen-1.0.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-netra_zen-1.0.1.dist-info/entry_points.txt,sha256=oDehCnPGZezG0m9ZWspxjHLHyQ3eERX87eojR4ljaRo,45
-netra_zen-1.0.1.dist-info/top_level.txt,sha256=dHz-hgh_dfiiOUrPf8wW80fA31rfEYDFmA6fpu67Yjk,65
-netra_zen-1.0.1.dist-info/RECORD,,
+netra_zen-1.0.2.dist-info/METADATA,sha256=3iv4Hq5DNLRAMivXVCwx_jieqevK_G3d1OU0Tbksl-g,29442
+netra_zen-1.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+netra_zen-1.0.2.dist-info/entry_points.txt,sha256=oDehCnPGZezG0m9ZWspxjHLHyQ3eERX87eojR4ljaRo,45
+netra_zen-1.0.2.dist-info/top_level.txt,sha256=dHz-hgh_dfiiOUrPf8wW80fA31rfEYDFmA6fpu67Yjk,65
+netra_zen-1.0.2.dist-info/RECORD,,

{netra_zen-1.0.1.dist-info → netra_zen-1.0.2.dist-info}/licenses/LICENSE.md

@@ -1 +1 @@
-© Netra Inc. All rights reserved. Use is subject to Netra's Terms of Service.
+© Netra Inc. All rights reserved. Use is subject to Netra's Terms of Service.

zen_orchestrator.py

@@ -2,43 +2,41 @@
 """
 Usage Examples:
 
-  zen -h  # Help
-
 Direct Command Execution:
-  zen "/single-command-in-claude-commands"  # Execute single command directly
-  zen "/analyze-code" --workspace ~/my-project
-  zen "/debug-issue" --instance-name "debug-session"
-  zen "/optimize-performance" --session-id "perf-1"
-  zen "/generate-docs" --clear-history --compact-history
+  python zen_orchestrator.py "/help"  # Execute single command directly
+  python zen_orchestrator.py "/analyze-code" --workspace ~/my-project
+  python zen_orchestrator.py "/debug-issue" --instance-name "debug-session"
+  python zen_orchestrator.py "/optimize-performance" --session-id "perf-1"
+  python zen_orchestrator.py "/generate-docs" --clear-history --compact-history
 
 Configuration File Mode:
-  zen --config config.json
-  zen --config config.json --workspace ~/my-project
+  python zen_orchestrator.py --config config.json
+  python zen_orchestrator.py --config config.json --workspace ~/my-project
 
 Default Instances Mode:
-  zen --dry-run  # Auto-detects workspace from project root
-  zen --workspace ~/my-project --dry-run  # Override workspace
-  zen --startup-delay 2.0  # 2 second delay between launches
-  zen --startup-delay 0.5  # 0.5 second delay between launches
-  zen --max-line-length 1000  # Longer output lines
-  zen --status-report-interval 60  # Status reports every 60s
-  zen --quiet  # Minimal output, errors only
+  python zen_orchestrator.py --dry-run  # Auto-detects workspace from project root
+  python zen_orchestrator.py --workspace ~/my-project --dry-run  # Override workspace
+  python zen_orchestrator.py --startup-delay 2.0  # 2 second delay between launches
+  python zen_orchestrator.py --startup-delay 0.5  # 0.5 second delay between launches
+  python zen_orchestrator.py --max-line-length 1000  # Longer output lines
+  python zen_orchestrator.py --status-report-interval 60  # Status reports every 60s
+  python zen_orchestrator.py --quiet  # Minimal output, errors only
 
 Command Discovery:
-  zen --list-commands  # Show all available commands
-  zen --inspect-command "/analyze-code"  # Inspect specific command
+  python zen_orchestrator.py --list-commands  # Show all available commands
+  python zen_orchestrator.py --inspect-command "/analyze-code"  # Inspect specific command
 
 Scheduling:
-  zen "/analyze-code" --start-at "2h"  # Start 2 hours from now
-  zen "/debug-issue" --start-at "30m"  # Start in 30 minutes
-  zen "/optimize" --start-at "1am"  # Start at 1 AM (today or tomorrow)
-  zen "/review-code" --start-at "14:30"  # Start at 2:30 PM (today or tomorrow)
-  zen "/generate-docs" --start-at "10:30pm"  # Start at 10:30 PM (today or tomorrow)
+  python zen_orchestrator.py "/analyze-code" --start-at "2h"  # Start 2 hours from now
+  python zen_orchestrator.py "/debug-issue" --start-at "30m"  # Start in 30 minutes
+  python zen_orchestrator.py "/optimize" --start-at "1am"  # Start at 1 AM (today or tomorrow)
+  python zen_orchestrator.py "/review-code" --start-at "14:30"  # Start at 2:30 PM (today or tomorrow)
+  python zen_orchestrator.py "/generate-docs" --start-at "10:30pm"  # Start at 10:30 PM (today or tomorrow)
 
 Precedence Rules:
-  1. Direct command (highest) - zen "/command"
-  2. Config file (medium) - zen --config file.json    # expected default usage pattern
-  3. Default instances (lowest) - zen
+  1. Direct command (highest) - python zen_orchestrator.py "/command"
+  2. Config file (medium) - python zen_orchestrator.py --config file.json
+  3. Default instances (lowest) - python zen_orchestrator.py
 """
 
 import asyncio
@@ -79,6 +77,16 @@ except ImportError as e:
     ClaudePricingEngine = None
     TokenUsageData = None
 
+# Add CLI extensions imports
+try:
+    from cli_extensions import handle_example_commands
+except ImportError as e:
+    # Graceful fallback if CLI extensions are not available
+    handle_example_commands = None
+
+# NetraOptimizer database functionality has been removed for security
+# Local token metrics are preserved without database persistence
+
 # Setup logging
 logging.basicConfig(
     level=logging.INFO,
@@ -107,7 +115,8 @@ def determine_log_level(args) -> LogLevel:
 @dataclass
 class InstanceConfig:
     """Configuration for a Claude Code instance"""
-    command: str
+    command: Optional[str] = None  # For slash commands like /help
+    prompt: Optional[str] = None   # For raw prompts like "What are the available commands?"
     name: Optional[str] = None
     description: Optional[str] = None
     allowed_tools: List[str] = None
@@ -121,10 +130,35 @@ class InstanceConfig:
 
     def __post_init__(self):
         """Set defaults after initialization"""
+        # Validate that either command or prompt is provided
+        if not self.command and not self.prompt:
+            raise ValueError("Either 'command' or 'prompt' must be provided")
+
+        # If both are provided, prioritize command over prompt
+        if self.command and self.prompt:
+            # Use command, but log that prompt is being ignored
+            pass  # This is valid - command takes precedence
+
+        # If only prompt is provided, treat it as the command for execution
+        elif self.prompt and not self.command:
+            self.command = self.prompt
+
+        # Set default name
         if self.name is None:
-            self.name = self.command
+            if self.command and self.command.startswith('/'):
+                self.name = self.command
+            else:
+                # For prompts, create a shorter name
+                display_text = self.prompt or self.command
+                self.name = display_text[:30] + "..." if len(display_text) > 30 else display_text
+
+        # Set default description
         if self.description is None:
-            self.description = f"Execute {self.command}"
+            if self.command and self.command.startswith('/'):
+                self.description = f"Execute {self.command}"
+            else:
+                self.description = f"Execute prompt: {(self.prompt or self.command)[:50]}..."
+
         # Set permission mode if not explicitly set
         if self.permission_mode is None:
             # Default to bypassPermissions for all platforms to avoid approval prompts
@@ -180,7 +214,7 @@ class ClaudeInstanceOrchestrator:
 
     def __init__(self, workspace_dir: Path, max_console_lines: int = 5, startup_delay: float = 1.0,
                  max_line_length: int = 500, status_report_interval: int = 30,
-                 quiet: bool = False,
+                 use_cloud_sql: bool = False, quiet: bool = False,
                  overall_token_budget: Optional[int] = None,
                  overall_cost_budget: Optional[float] = None,
                  budget_type: str = "tokens",
@@ -199,9 +233,11 @@ class ClaudeInstanceOrchestrator:
         self.status_report_interval = status_report_interval  # Seconds between status reports
         self.last_status_report = time.time()
         self.status_report_task = None  # For the rolling status report task
+        self.use_cloud_sql = use_cloud_sql
         self.quiet = quiet
         self.log_level = log_level
         self.batch_id = str(uuid4())  # Generate batch ID for this orchestration run
+        # Database client removed for security - local metrics only
         self.optimizer = None
 
         # Initialize budget manager if any budget settings are provided
@@ -242,6 +278,12 @@ class ClaudeInstanceOrchestrator:
         else:
             logger.debug("Token budget tracking disabled (no budget specified)")
 
+        # Configure CloudSQL if requested
+        # CloudSQL functionality available with Netra Apex
+        # Token metrics are now handled locally without database persistence
+        if use_cloud_sql:
+            logger.warning("CloudSQL functionality has been disabled. Token metrics will be displayed locally only.")
+            logger.info("For data persistence, consider upgrading to Netra Apex.")
 
     def log_at_level(self, level: LogLevel, message: str, log_func=None):
         """Log message only if current log level permits."""
@@ -257,10 +299,17 @@ class ClaudeInstanceOrchestrator:
 
     def add_instance(self, config: InstanceConfig):
         """Add a new instance configuration"""
-        # Validate slash command exists
-        if not self.validate_command(config.command):
-            logger.warning(f"Command '{config.command}' not found in available commands")
-            logger.info(f"Available commands: {', '.join(self.discover_available_commands())}")
+        # Determine if this is a slash command or raw prompt
+        is_slash_command = config.command and config.command.startswith('/')
+        is_raw_prompt = config.prompt and not is_slash_command
+
+        if is_slash_command:
+            # Validate slash command exists
+            if not self.validate_command(config.command):
+                logger.warning(f"Command '{config.command}' not found in available commands")
+                logger.info(f"Available commands: {', '.join(self.discover_available_commands())}")
+        elif is_raw_prompt:
+            logger.info(f"Using raw prompt: {config.prompt[:50]}{'...' if len(config.prompt) > 50 else ''}")
 
         self.instances[config.name] = config
         self.statuses[config.name] = InstanceStatus(name=config.name)
@@ -2321,7 +2370,8 @@ async def main():
                        help="Seconds between rolling status reports (default: 5)")
     parser.add_argument("--start-at", type=str, default=None,
                        help="Schedule orchestration to start at specific time. Examples: '2h' (2 hours from now), '30m' (30 minutes), '14:30' (2:30 PM today), '1am' (1 AM today/tomorrow)")
-
+    parser.add_argument("--use-cloud-sql", action="store_true",
+                       help="Save metrics to CloudSQL database (NetraOptimizer integration)")
 
     # Direct command options
     parser.add_argument("--instance-name", type=str, help="Instance name for direct command execution")
@@ -2404,6 +2454,11 @@ async def main():
     
     logger.info(f"Using workspace: {workspace}")
 
+    # Handle CLI extension commands
+    if handle_example_commands and handle_example_commands(args):
+        return
+
+
     # Load instance configurations with direct command precedence
     direct_instance = create_direct_instance(args, workspace)
 
@@ -2492,6 +2547,7 @@ async def main():
         startup_delay=args.startup_delay,
         max_line_length=args.max_line_length,
         status_report_interval=args.status_report_interval,
+        use_cloud_sql=args.use_cloud_sql,
         quiet=args.quiet,
         overall_token_budget=final_overall_budget,
         overall_cost_budget=final_overall_cost_budget,
@@ -2719,6 +2775,9 @@ async def main():
 
     # Run all instances
     logger.info("Starting Claude Code instance orchestration")
+    if args.use_cloud_sql:
+        logger.info(f"Batch ID: {orchestrator.batch_id}")
+        logger.info("Metrics will be saved to CloudSQL")
     start_time = time.time()
 
     results = await orchestrator.run_all_instances(args.timeout)
@@ -2850,13 +2909,20 @@ async def main():
 
     # For detailed data access
     print("\n" + "="*80)
-    print("🚀 Looking for more?")
+    print("🚀 NEED DETAILED TOKEN ANALYTICS & DATA ACCESS?")
     print("="*80)
-    print("Explore Zen with Apex for the most effective AI Ops value for production AI.")
+    print("For comprehensive token usage analytics, historical data,")
+    print("and advanced reporting features, upgrade to Netra Apex.")
     print("")
     print("🌐 Learn more: https://netrasystems.ai/")
     print("="*80)
 
+    # Show CloudSQL info if enabled
+    if args.use_cloud_sql:
+        print(f"\n📊 Local metrics displayed above")
+        print(f"   Batch ID: {orchestrator.batch_id}")
+        print(f"   Database persistence disabled for security")
+
     # Exit with appropriate code
     sys.exit(0 if summary['failed'] == 0 else 1)