From 2a11fefcaaa61e3598e380f68b7061cf550beb86 Mon Sep 17 00:00:00 2001
From: Dominik Jain <dominik.jain@outlook.com>
Date: Thu, 5 Jun 2025 00:23:04 +0200
Subject: [PATCH] Add None check (mypy)

---
 src/serena/mcp.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/serena/mcp.py b/src/serena/mcp.py
index abc9cb5..a6c471b 100644
--- a/src/serena/mcp.py
+++ b/src/serena/mcp.py
@@ -160,7 +160,8 @@ def create_mcp_server_and_agent(
         nonlocal agent
         mark_used(mcp_server)
         yield
-        agent.language_server.stop()
+        if agent.language_server is not None:
+            agent.language_server.stop()
 
     mcp_settings = Settings(lifespan=server_lifespan, host=host, port=port)
     mcp = FastMCP(**mcp_settings.model_dump())