support api usage for extract with no args (#582)

Author: seanmcguire12

.changeset/petite-donuts-lead.md

@@ -0,0 +1,5 @@
+---
+"@browserbasehq/stagehand": patch
+---
+
+support api usage for extract with no args

lib/StagehandPage.ts

@@ -547,6 +547,9 @@ export class StagehandPage {
 
     // check if user called extract() with no arguments
     if (!instructionOrOptions) {
+      if (this.api) {
+        return this.api.extract<T>({});
+      }
       return this.extractHandler.extract();
     }
 

lib/api.ts

@@ -91,6 +91,12 @@ export class StagehandAPI {
   async extract<T extends z.AnyZodObject>(
     options: ExtractOptions<T>,
   ): Promise<ExtractResult<T>> {
+    if (!options.schema) {
+      return this.execute<ExtractResult<T>>({
+        method: "extract",
+        args: {},
+      });
+    }
     const parsedSchema = zodToJsonSchema(options.schema);
     return this.execute<ExtractResult<T>>({
       method: "extract",

types/stagehand.ts

@@ -98,8 +98,8 @@ export interface ActResult {
 }
 
 export interface ExtractOptions<T extends z.AnyZodObject> {
-  instruction: string;
-  schema: T;
+  instruction?: string;
+  schema?: T;
   modelName?: AvailableModel;
   modelClientOptions?: ClientOptions;
   domSettleTimeoutMs?: number;