await session cleanup

Author: Kylejeong2

package.json

@@ -28,7 +28,14 @@
     "prettier:check": "prettier --check .",
     "prettier:fix": "prettier --write .",
     "clean": "rm -rf dist",
-    "prepublishOnly": "pnpm clean && pnpm build"
+    "prepublishOnly": "pnpm clean && pnpm build",
+    "test": "bash tests/run-tests.sh",
+    "test:minimal": "bash tests/run-tests.sh --type minimal",
+    "test:advanced": "bash tests/run-tests.sh --type advanced --llm",
+    "test:all": "bash tests/run-tests.sh --type all",
+    "test:debug": "bash tests/run-tests.sh --debug",
+    "test:install": "npm install @mcpvals chalk",
+    "test:runner": "tsx tests/run-evals.ts"
   },
   "lint-staged": {
     "*.{js,jsx,ts,tsx,json,css,scss,md}": [
@@ -47,6 +54,8 @@
   },
   "devDependencies": {
     "@eslint/js": "^9.29.0",
+    "@mcpvals": "^0.1.0",
+    "chalk": "^5.3.0",
     "eslint": "^9.29.0",
     "eslint-plugin-react": "^7.37.5",
     "globals": "^16.2.0",

src/sessionManager.ts

@@ -324,9 +324,15 @@ export async function getSession(
  * This is called after a browser is closed to ensure proper cleanup.
  * @param sessionId The session ID to clean up
  */
-export function cleanupSession(sessionId: string): void {
+export async function cleanupSession(sessionId: string): Promise<void> {
   process.stderr.write(`[SessionManager] Cleaning up session: ${sessionId}\n`);
 
+  // Get the session to close it gracefully
+  const session = browsers.get(sessionId);
+  if (session) {
+    await closeBrowserGracefully(session, sessionId);
+  }
+
   // Remove from browsers map
   browsers.delete(sessionId);
 

src/tools/session.ts

@@ -161,7 +161,7 @@ async function handleCloseSession(context: Context): Promise<ToolResult> {
         );
 
         // Clean up the session from tracking
-        cleanupSession(previousSessionId);
+        await cleanupSession(previousSessionId);
 
         if (browserbaseSessionId) {
           process.stderr.write(