netra-zen 1.1.2__py3-none-any.whl → 1.2.0__py3-none-any.whl

{netra_zen-1.1.2.dist-info → netra_zen-1.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: netra-zen
-Version: 1.1.2
+Version: 1.2.0
 Summary: Multi-instance Claude orchestrator for parallel task execution
 Home-page: https://github.com/netra-systems/zen
 Author:  Systems

{netra_zen-1.1.2.dist-info → netra_zen-1.2.0.dist-info}/RECORD

@@ -1,10 +1,10 @@
 zen_orchestrator.py,sha256=hKeP2dW6eJgEhtUHYylnhIgP_HgaxrDAwCnlS7mKkgU,156167
 agent_interface/__init__.py,sha256=OsbOKzElHsxhVgak87oOx_u46QNgKmz-Reis-plAMwk,525
 agent_interface/base_agent.py,sha256=GNskG9VaZgno7X24lQTpFdxUoQE0yJHLh0UPFJvOPn4,11098
-netra_zen-1.1.2.dist-info/licenses/LICENSE.md,sha256=PZrP0UDn58i4LjV4zijIQTnsQPvWm4zq9Fet9i7qgwI,80
+netra_zen-1.2.0.dist-info/licenses/LICENSE.md,sha256=PZrP0UDn58i4LjV4zijIQTnsQPvWm4zq9Fet9i7qgwI,80
 scripts/__init__.py,sha256=r6jX6e9SaisREml6B7uDoV-_rzIW0_yQXMm6OhNgbIw,52
 scripts/__main__.py,sha256=NSqyN47EijNf7m_8QsoxeYevKYgH2sNtNPFZcwKWMs0,160
-scripts/agent_cli.py,sha256=Ut_7iou5M5Meeb9Ulvn_zZZrEGmpmDi0vDJAEP_fUtA,340749
+scripts/agent_cli.py,sha256=ATQIWoVx7GthFV1WeiV6Co8PzoXYXhlJjOerU-JocJ0,345625
 scripts/agent_logs.py,sha256=ppseAtUCvS2N-iEDSUQh84I9Ov-gBEw5ElxYajNdpJs,11218
 scripts/bump_version.py,sha256=fjABzzRVXJ00CbYMpUIUMwcOHwafLYtFL6NvUga-i6M,4183
 scripts/demo_log_collection.py,sha256=5XladS8j4lz-jJdVzSeDSxopChqHKX7HLNJUzSWR40c,5623
@@ -23,8 +23,8 @@ zen/telemetry/__init__.py,sha256=zyH7YcK_eWxwaQZW0MRefu1bX5hEgfRljr_dsqNU-tw,452
 zen/telemetry/apex_telemetry.py,sha256=FAkiHXOuzZFL-51oYetTC4wbjCL5UH-qOghwcaaOVgE,9685
 zen/telemetry/embedded_credentials.py,sha256=z-j_TfuVIz3nX-Vvh4O6_iDQhtteEyQgfc-xSslVbXU,1716
 zen/telemetry/manager.py,sha256=Rdbpnjbjel9xZEJyvLqy2M-4amvkr80abRiWnqHghIQ,9980
-netra_zen-1.1.2.dist-info/METADATA,sha256=-7DFDTlxlMX9NnptWNy-LW09GsKyYGiPC5cyuANh3Gk,46283
-netra_zen-1.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-netra_zen-1.1.2.dist-info/entry_points.txt,sha256=oDehCnPGZezG0m9ZWspxjHLHyQ3eERX87eojR4ljaRo,45
-netra_zen-1.1.2.dist-info/top_level.txt,sha256=OhiyXmoXftBijCF6ck-RS1dN2NBJv9wdd7kBG1Es7zA,77
-netra_zen-1.1.2.dist-info/RECORD,,
+netra_zen-1.2.0.dist-info/METADATA,sha256=sb-1CUEfEqK3rmHbqxqrNeMi4t5d_hZ34xhBoPT1a90,46283
+netra_zen-1.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+netra_zen-1.2.0.dist-info/entry_points.txt,sha256=oDehCnPGZezG0m9ZWspxjHLHyQ3eERX87eojR4ljaRo,45
+netra_zen-1.2.0.dist-info/top_level.txt,sha256=OhiyXmoXftBijCF6ck-RS1dN2NBJv9wdd7kBG1Es7zA,77
+netra_zen-1.2.0.dist-info/RECORD,,

scripts/agent_cli.py

@@ -2883,22 +2883,49 @@ class WebSocketClient:
                         self.connected = True
                         return True
                     else:
-                        # Handshake failed - backend might be using old reactive model
+                        # Handshake failed - server not ready to process messages
                         self.debug.debug_print(
-                            "WARNING: No proactive handshake received from server",
+                            "WARNING: Server handshake not completed - server not ready",
                             DebugLevel.BASIC,
                             style="yellow"
                         )
                         self.debug.debug_print(
-                            "Server may be using pre-2025-10-09 architecture (reactive handshake)",
+                            "Server still completing lifecycle phases (Initialize → Authenticate → Handshake → Prepare → Processing)",
                             DebugLevel.VERBOSE,
                             style="yellow"
                         )
 
-                        # Still mark as connected for backward compatibility
-                        # Old backends might work without the proactive handshake
-                        self.connected = True
-                        return True
+                        # Wait briefly for server to complete its phases and retry
+                        safe_console_print("⏳ Waiting for server to complete initialization...", style="yellow",
+                                         json_mode=self.config.json_mode, ci_mode=self.config.ci_mode)
+                        await asyncio.sleep(3.0)  # Give server time to reach PROCESSING phase
+
+                        # Retry handshake once more
+                        self.debug.debug_print(
+                            "Retrying handshake after delay...",
+                            DebugLevel.VERBOSE,
+                            style="cyan"
+                        )
+                        handshake_success = await self._perform_handshake()
+                        if handshake_success:
+                            safe_console_print(f"✅ Connected with thread ID: {self.current_thread_id}", style="green",
+                                             json_mode=self.config.json_mode, ci_mode=self.config.ci_mode)
+                            self.connected = True
+                            return True
+                        else:
+                            # Server still not ready - fail the connection
+                            self.debug.debug_print(
+                                "ERROR: Server not ready after retry",
+                                DebugLevel.BASIC,
+                                style="red"
+                            )
+                            safe_console_print("❌ Server not ready to process messages. Please try again.", style="red",
+                                             json_mode=self.config.json_mode, ci_mode=self.config.ci_mode)
+
+                            # Close WebSocket and fail gracefully
+                            if self.ws:
+                                await self.ws.close()
+                            return False
             except Exception as e:
                 self.debug.log_connection_attempt(method_name, self.config.ws_url, success=False, error=str(e))
                 self.debug.debug_print(
@@ -3027,7 +3054,8 @@ class WebSocketClient:
 
             # Server enters HANDSHAKING phase after authentication
             # and proactively sends handshake_response
-            handshake_timeout = 10.0  # Wait up to 10 seconds
+            # Use configured timeout or default to 10 seconds
+            handshake_timeout = self.handshake_timeout if hasattr(self, 'handshake_timeout') and self.handshake_timeout else 10.0
             start_time = asyncio.get_event_loop().time()
 
             try:
@@ -3274,14 +3302,64 @@ class WebSocketClient:
             )
 
         # CRITICAL: Send acknowledgment with the SAME thread_id
+        # Per WebSocket Client Lifecycle Guide, use "handshake_acknowledged"
         ack_message = {
-            "type": "session_acknowledged",
+            "type": "handshake_acknowledged",
             "thread_id": backend_thread_id,  # Echo back the same ID
             "timestamp": datetime.now(timezone.utc).isoformat()
         }
 
         await self.ws.send(json.dumps(ack_message))
 
+        # Per WebSocket Client Lifecycle Guide, wait for handshake_complete
+        # and then add delay for server to enter Phase 5 (Processing)
+        self.debug.debug_print(
+            "Waiting for handshake_complete confirmation...",
+            DebugLevel.VERBOSE,
+            style="cyan"
+        )
+
+        # Wait briefly for handshake_complete message
+        try:
+            complete_msg = await asyncio.wait_for(self.ws.recv(), timeout=2.0)
+            complete_data = json.loads(complete_msg)
+            if complete_data.get('type') == 'handshake_complete':
+                self.debug.debug_print(
+                    "✅ Received handshake_complete - Server at Phase 4 (Ready)",
+                    DebugLevel.VERBOSE,
+                    style="green"
+                )
+
+                # CRITICAL: Add delay for server to enter Phase 5 (Processing)
+                # Per documentation, 500ms is recommended
+                self.debug.debug_print(
+                    "Waiting 500ms for server to enter Phase 5 (Processing)...",
+                    DebugLevel.VERBOSE,
+                    style="cyan"
+                )
+                await asyncio.sleep(0.5)
+
+                self.debug.debug_print(
+                    "✅ Server should now be in Phase 5 (Processing) - ready for messages",
+                    DebugLevel.VERBOSE,
+                    style="green"
+                )
+        except asyncio.TimeoutError:
+            # If no handshake_complete, still add a delay to be safe
+            self.debug.debug_print(
+                "No handshake_complete received, adding safety delay",
+                DebugLevel.VERBOSE,
+                style="yellow"
+            )
+            await asyncio.sleep(0.5)
+        except Exception as e:
+            self.debug.debug_print(
+                f"Error waiting for handshake_complete: {e}",
+                DebugLevel.VERBOSE,
+                style="yellow"
+            )
+            await asyncio.sleep(0.5)
+
         return True
 
     def _get_platform_cache_path(self) -> Path:
@@ -3622,6 +3700,27 @@ class WebSocketClient:
         if not self.ws:
             raise RuntimeError("WebSocket not connected")
 
+        # Check if connection handshake is fully complete
+        if not self.connected:
+            self.debug.debug_print(
+                "ERROR: Connection not ready - handshake not complete",
+                DebugLevel.BASIC,
+                style="red"
+            )
+            safe_console_print(
+                "\n❌ ERROR: Cannot send message - server not ready",
+                style="red",
+                json_mode=self.config.json_mode,
+                ci_mode=self.config.ci_mode
+            )
+            safe_console_print(
+                "The server is still completing its initialization phases.",
+                style="yellow",
+                json_mode=self.config.json_mode,
+                ci_mode=self.config.ci_mode
+            )
+            raise RuntimeError("Connection not ready - handshake incomplete. Server needs to reach Processing phase.")
+
         # Generate run_id
         self.run_id = f"cli_{datetime.now().strftime('%Y%m%d_%H%M%S')}_{os.getpid()}"