handle close session (#528)

Author: seanmcguire12

Diff for: .changeset/pretty-parrots-wave.md

@@ -0,0 +1,5 @@
+---
+"@browserbasehq/stagehand": patch
+---
+
+handle attempt to close session that has already been closed when using the api

Diff for: lib/api.ts

@@ -112,9 +112,10 @@ export class StagehandAPI {
     });
   }
 
-  async end(): Promise<void> {
-    return this.execute<void>({
-      method: "end",
+  async end(): Promise<Response> {
+    const url = `/sessions/${this.sessionId}/end`;
+    return await this.request(url, {
+      method: "POST",
     });
   }
 

Diff for: lib/index.ts

@@ -32,6 +32,7 @@ import { scriptContent } from "./dom/build/scriptContent";
 import { LLMClient } from "./llm/LLMClient";
 import { LLMProvider } from "./llm/LLMProvider";
 import { logLineToString, isRunningInBun } from "./utils";
+import { ApiResponse, ErrorResponse } from "@/types/api";
 
 dotenv.config({ path: ".env" });
 
@@ -711,7 +712,20 @@ export class Stagehand {
 
   async close(): Promise<void> {
     if (this.apiClient) {
-      await this.apiClient.end();
+      const response = await this.apiClient.end();
+      const body: ApiResponse<unknown> = await response.json();
+      if (!body.success) {
+        if (response.status == 409) {
+          this.log({
+            category: "close",
+            message:
+              "Warning: attempted to end a session that is not currently active",
+            level: 0,
+          });
+        } else {
+          throw new Error((body as ErrorResponse).message);
+        }
+      }
       return;
     } else {
       await this.context.close();