improvements

Author: maticzav

examples/structured-output.ts

@@ -18,16 +18,17 @@ const TaskOutput = z.object({
 });
 
 async function main() {
+  let log = 'starting';
+  const stop = spinner(() => log);
+
+  // Create Task
   const rsp = await browseruse.tasks.createWithStructuredOutput({
     task: 'Extract top 10 Hacker News posts and return the title, url, and score',
     structuredOutputJson: TaskOutput,
   });
 
-  let latestStatusText = 'starting';
-
-  const stop = spinner(() => latestStatusText);
-
   poll: do {
+    // Wait for Task to Finish
     const status = await browseruse.tasks.retrieveWithStructuredOutput(rsp.id, {
       structuredOutputJson: TaskOutput,
     });
@@ -42,14 +43,15 @@ async function main() {
         const lastGoal = lastGoalDescription ? `, last: ${lastGoalDescription}` : '';
         const liveUrl = status.sessionLiveUrl ? `, live: ${status.sessionLiveUrl}` : '';
 
-        latestStatusText = `agent ${status.status} (${steps}${lastGoal}${liveUrl}) `;
+        log = `agent ${status.status} (${steps}${lastGoal}${liveUrl}) `;
 
         await new Promise((resolve) => setTimeout(resolve, 2000));
       }
 
       case 'finished':
         stop();
 
+        // Print Structured Output
         console.log('TOP POSTS:');
 
         for (const post of status.doneOutput!.posts) {

src/lib/parse.ts

@@ -1,16 +1,16 @@
-import z from 'zod';
+import z, { type ZodType } from 'zod';
 import type { TaskCreateParams, TaskRetrieveParams, TaskView } from '../resources/tasks';
 
 // RUN
 
-export type RunTaskCreateParamsWithStructuredOutput<T extends z.ZodTypeAny> = Omit<
+export type RunTaskCreateParamsWithStructuredOutput<T extends ZodType> = Omit<
   TaskCreateParams,
   'structuredOutputJson'
 > & {
   structuredOutputJson: T;
 };
 
-export function stringifyStructuredOutput<T extends z.ZodTypeAny>(
+export function stringifyStructuredOutput<T extends ZodType>(
   req: RunTaskCreateParamsWithStructuredOutput<T>,
 ): TaskCreateParams {
   return {
@@ -21,19 +21,19 @@ export function stringifyStructuredOutput<T extends z.ZodTypeAny>(
 
 // RETRIEVE
 
-export type GetTaskStatusParamsWithStructuredOutput<T extends z.ZodTypeAny> = Omit<
+export type GetTaskStatusParamsWithStructuredOutput<T extends ZodType> = Omit<
   TaskRetrieveParams,
   'statusOnly'
 > & {
   statusOnly?: false;
   structuredOutputJson: T;
 };
 
-export type TaskViewWithStructuredOutput<T extends z.ZodTypeAny> = Omit<TaskView, 'doneOutput'> & {
+export type TaskViewWithStructuredOutput<T extends ZodType> = Omit<TaskView, 'doneOutput'> & {
   doneOutput: z.output<T> | null;
 };
 
-export function parseStructuredTaskOutput<T extends z.ZodTypeAny>(
+export function parseStructuredTaskOutput<T extends ZodType>(
   res: TaskView,
   body: GetTaskStatusParamsWithStructuredOutput<T>,
 ): TaskViewWithStructuredOutput<T> {

src/resources/tasks.ts

@@ -1,6 +1,6 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-import z from 'zod';
+import type { ZodType } from 'zod';
 
 import { APIResource } from '../core/resource';
 import * as TasksAPI from './tasks';
@@ -10,7 +10,7 @@ import { path } from '../internal/utils/path';
 import {
   parseStructuredTaskOutput,
   stringifyStructuredOutput,
-  TaskViewWithStructuredOutput,
+  type TaskViewWithStructuredOutput,
   type GetTaskStatusParamsWithStructuredOutput,
   type RunTaskCreateParamsWithStructuredOutput,
 } from '../lib/parse';
@@ -23,7 +23,7 @@ export class Tasks extends APIResource {
     return this._client.post('/tasks', { body, ...options });
   }
 
-  createWithStructuredOutput<T extends z.ZodTypeAny>(
+  createWithStructuredOutput<T extends ZodType>(
     body: RunTaskCreateParamsWithStructuredOutput<T>,
     options?: RequestOptions,
   ): APIPromise<TaskViewWithStructuredOutput<T>> {
@@ -43,7 +43,7 @@ export class Tasks extends APIResource {
     return this._client.get(path`/tasks/${taskID}`, { query, ...options });
   }
 
-  retrieveWithStructuredOutput<T extends z.ZodTypeAny>(
+  retrieveWithStructuredOutput<T extends ZodType>(
     taskID: string,
     query: GetTaskStatusParamsWithStructuredOutput<T>,
     options?: RequestOptions,