Update publish and invoke to allow for PENDING active state

README.md

@@ -4,7 +4,7 @@
 
 The Browserbase Functions SDK lets you define, develop, and deploy serverless browser automation functions on [Browserbase](https://browserbase.com). Each function gets a managed browser session — write your automation logic, test it locally, and publish it to the cloud.
 
-The full documentation can be found on [docs.browserbase.com](https://docs.browserbase.com).
+The full documentation can be found on [docs.browserbase.com](https://docs.browserbase.com/functions/quickstart).
 
 ## Installation
 
@@ -103,7 +103,7 @@ defineFn(
 
 ### Custom Browser Configuration
 
-Pass `sessionConfig` to customize the browser session (uses the same options as the [Browserbase SDK session create params](https://docs.browserbase.com)):
+Pass `sessionConfig` to customize the browser session (uses the same options as the [Browserbase SDK session create params](https://docs.browserbase.com/reference/api/create-a-session)):
 
 ```ts
 import { defineFn } from "@browserbasehq/sdk-functions";

src/cli/invoke/index.ts

@@ -1,13 +1,13 @@
 import chalk from "chalk";
 
-import { loadBaseConfig, apiGet, apiPost, pollUntil } from "../shared/index.js";
-
-export type InvocationStatus =
-  | "PENDING"
-  | "RUNNING"
-  | "COMPLETED"
-  | "FAILED"
-  | "TIMEOUT";
+import {
+  loadBaseConfig,
+  apiGet,
+  apiPost,
+  pollUntil,
+  type InvocationStatus,
+  isTerminalInvocationStatus,
+} from "../shared/index.js";
 
 export interface InvocationResponse {
   id: string;
@@ -32,10 +32,6 @@ export interface InvokeOptions {
   checkStatus?: string;
 }
 
-function isTerminalStatus(status: InvocationStatus): boolean {
-  return status !== "RUNNING" && status !== "PENDING";
-}
-
 function displayInvocationResult(invocation: InvocationResponse): void {
   console.log(chalk.bold.cyan("\n📋 Invocation Details"));
   console.log(chalk.gray("─".repeat(50)));
@@ -163,7 +159,7 @@ export async function invoke(options: InvokeOptions): Promise<void> {
           config,
           `/v1/functions/invocations/${result.data.id}`,
         ),
-      (status) => isTerminalStatus(status.status),
+      (status) => isTerminalInvocationStatus(status.status),
       (status) => status.status,
       {
         intervalMs: 1000,
@@ -183,13 +179,11 @@ export async function invoke(options: InvokeOptions): Promise<void> {
       console.log(chalk.green("✓ Invocation completed successfully"));
     } else if (finalStatus.status === "FAILED") {
       console.error(chalk.red("✗ Invocation failed"));
-    } else if (finalStatus.status === "TIMEOUT") {
-      console.error(chalk.red("✗ Invocation timed out"));
     }
 
     displayInvocationResult(finalStatus);
 
-    if (finalStatus.status === "FAILED" || finalStatus.status === "TIMEOUT") {
+    if (finalStatus.status === "FAILED") {
       process.exit(1);
     }
   } catch (error: unknown) {

src/cli/publish/api-client.ts

@@ -1,7 +1,12 @@
 import chalk from "chalk";
 
 import { type PublishConfig } from "./config.js";
-import { parseErrorResponse, pollUntil } from "../shared/index.js";
+import {
+  parseErrorResponse,
+  pollUntil,
+  type BuildStatus,
+  isTerminalBuildStatus,
+} from "../shared/index.js";
 
 export interface BuildMetadata {
   entrypoint: string;
@@ -14,8 +19,6 @@ export interface UploadResult {
   message?: string;
 }
 
-export type BuildStatus = "RUNNING" | "COMPLETED" | "FAILED";
-
 export interface FunctionCreatedVersion {
   id: string;
   projectId: string;
@@ -212,7 +215,7 @@ export async function pollBuildStatus(
 
   const result = await pollUntil(
     () => getBuildStatus(config, buildId),
-    (status) => status.status !== "RUNNING",
+    (status) => isTerminalBuildStatus(status.status),
     (status) => status.status,
     {
       intervalMs: options?.intervalMs ?? 2000,

src/cli/shared/api-client.ts

@@ -117,6 +117,18 @@ export async function apiPost<T>(
   }
 }
 
+export type BuildStatus = "PENDING" | "RUNNING" | "COMPLETED" | "FAILED";
+
+export type InvocationStatus = "PENDING" | "RUNNING" | "COMPLETED" | "FAILED";
+
+export function isTerminalBuildStatus(status: BuildStatus): boolean {
+  return status !== "PENDING" && status !== "RUNNING";
+}
+
+export function isTerminalInvocationStatus(status: InvocationStatus): boolean {
+  return status !== "PENDING" && status !== "RUNNING";
+}
+
 export interface PollOptions {
   /** Polling interval in milliseconds */
   intervalMs?: number;

src/cli/shared/index.ts

@@ -13,4 +13,8 @@ export {
   apiPost,
   pollUntil,
   type PollOptions,
+  type BuildStatus,
+  type InvocationStatus,
+  isTerminalBuildStatus,
+  isTerminalInvocationStatus,
 } from "./api-client.js";