Method initialize

ms_agent/tools/code_server/lsp_code_server.py:147–209

Initialize the LSP server and wait for it to be ready

(self)

        return {}

    async def initialize(self):
        """Initialize the LSP server and wait for it to be ready"""
        response = await self.send_request(
            'initialize', {
                'processId':
                os.getpid(),
                'rootUri':
                self.workspace_dir.as_uri(),
                'rootPath':
                str(self.workspace_dir),
                'workspaceFolders': [{
                    'uri': self.workspace_dir.as_uri(),
                    'name': self.workspace_dir.name
                }],
                'capabilities': {
                    'textDocument': {
                        'publishDiagnostics': {},
                        'synchronization': {
                            'didOpen': True,
                            'didChange': True,
                            'didClose': True
                        }
                    }
                }
            })

        if 'result' in response:
            await self.send_notification('initialized', {})

            # CRITICAL: Wait for server to be fully ready
            # Read and discard any startup messages
            await asyncio.sleep(
                1.0)  # Give server time to complete initialization

            await self.send_notification(
                'workspace/didChangeConfiguration', {
                    'settings': {
                        'python': {
                            'pythonPath': sys.executable,
                        },
                        'pyright': {
                            'extraPaths': [str(self.workspace_dir)]
                        },
                    }
                })

            # Consume any pending messages (like "starting" notifications)
            try:
                for _ in range(10):
                    try:
                        await asyncio.wait_for(
                            self._read_message(), timeout=2.0)
                    except asyncio.TimeoutError:
                        break
            except Exception as e:
                logger.error(f'Cleared startup messages: {e}')

            self.initialized = True