Method connect

ms_agent/tools/code/code_executor.py:203–314

Initialize sandbox manager and create sandbox instance with automatic port retry

(self)

            raise ValueError(f'Unknown sandbox type: {self.sandbox_type}')

    async def connect(self) -> None:
        """Initialize sandbox manager and create sandbox instance with automatic port retry"""
        from ms_enclave.sandbox.model import SandboxType

        if self._initialized:
            logger.debug('Sandbox already initialized')
            return

        try:
            logger.info('Initializing sandbox manager...')

            # Create manager using factory
            self.manager = await SandboxManagerFactory.create_manager(
                self.config)
            await self.manager.start()

            logger.info('Creating sandbox instance...')

            # Try to create sandbox with port retry logic
            retry_count = 0
            max_retries = self._max_port_retries if self._port_retry_enabled else 1
            last_error = None

            while retry_count < max_retries:
                try:
                    self.sandbox_id = await self.manager.create_sandbox(
                        sandbox_type=self.sandbox_type,
                        config=self.sandbox_config)

                    logger.info(f'Sandbox created: {self.sandbox_id}')

                    # Wait for sandbox to be ready
                    await self._wait_for_sandbox_ready()

                    self._initialized = True
                    logger.info('Sandbox is ready for code execution')
                    return

                except Exception as e:
                    error_msg = str(e).lower()
                    last_error = e

                    # Check if it's a port conflict error
                    is_port_conflict = any(keyword in error_msg
                                           for keyword in [
                                               'address already in use',
                                               'port is already allocated',
                                               'bind: address already in use',
                                               'port already in use'
                                           ])

                    if is_port_conflict and self._port_retry_enabled and retry_count < (
                            max_retries - 1):
                        retry_count += 1
                        logger.warning(
                            f'Port conflict detected (attempt {retry_count}/{max_retries}): {e}'
                        )