From e08d36633b3aff59e4e455234bbbc47b0a5b7677 Mon Sep 17 00:00:00 2001 From: Michael Panchenko Date: Sat, 29 Mar 2025 14:22:01 +0100 Subject: [PATCH] Start server in with-context --- src/serena/mcp.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/serena/mcp.py b/src/serena/mcp.py index 77a75ea..de45654 100644 --- a/src/serena/mcp.py +++ b/src/serena/mcp.py @@ -79,11 +79,11 @@ async def server_lifespan(mcp_server: FastMCP) -> AsyncIterator[SerenaMCPRequest config = MultilspyConfig(code_language=language) logger = MultilspyLogger() language_server = SyncLanguageServer.create(config, logger, project_root) - language_server.start() try: - yield SerenaMCPRequestContext( - language_server=language_server, project_root=project_root, project_config=project_config, prompt_factory=PromptFactory() - ) + with language_server.start_server(): + yield SerenaMCPRequestContext( + language_server=language_server, project_root=project_root, project_config=project_config, prompt_factory=PromptFactory() + ) finally: language_server.stop()