fix: Improve session management in mcp_tools.py by adding error handling and cleanup for client sessions

Author: onuratakan

src/upsonicai/server/tools/server/mcp_tools.py

@@ -26,25 +26,31 @@ class ListToolsRequest(BaseRequestMCP):
     pass
 
 
+# Global variable to store the session
+session_store = None
+
 async def get_session(command: str, args: list, env: dict):
     print("env", env)
     print("args", args)
     print("command", command)
 
-    if env:
-        env = get_default_environment()
-        env.update(env) if env else None
-
     server_params = StdioServerParameters(
         command=command,  # Executable
         args=args,  # Optional command line arguments
         env=env,  # Environment variables
     )
     async with stdio_client(server_params) as (read, write):
         async with ClientSession(read, write) as session:
-            await session.initialize()
-            yield session
-            await session.close()
+            try:
+                print("Initializing session...")
+                await session.initialize()
+                print("Session initialized.")
+                yield session
+            except Exception as e:
+                print(f"Error in session: {e}")
+                raise
+            finally:
+                print("Session cleanup.")
 
 
 @app.post(f"{prefix}/tools")