fixed complete task sdk

Author: gregpr07

examples/retrieve.ts

@@ -28,11 +28,11 @@ async function basic() {
 
     poll: do {
         // Wait for Task to Finish
-        const status = await browseruse.tasks.getTask(rsp.id);
+        const status = await browseruse.tasks.getTask({ task_id: rsp.id });
 
         switch (status.status) {
             case "started":
-            case "paused":
+            case "created":
             case "stopped":
                 log = `agent ${status.status} ${status.sessionId} ${status.steps.length} steps`;
 
@@ -73,13 +73,13 @@ async function structured() {
     poll: do {
         // Wait for Task to Finish
         const status = await browseruse.tasks.getTask({
-            taskId: rsp.id,
+            task_id: rsp.id,
             schema: TaskOutput,
         });
 
         switch (status.status) {
             case "started":
-            case "paused":
+            case "created":
             case "stopped": {
                 log = `agent ${status.status} ${status.sessionId} | ${status.steps.length} steps`;
 

examples/run.ts

@@ -22,6 +22,11 @@ async function basic() {
         task: "What's the weather in SF and what's the temperature?",
     });
 
+
+    for await (const msg of rsp.watch()) {
+        console.log(msg);
+    }
+
     const result = await rsp.complete();
 
     console.log(`Basic: ${result.output}`);

src/wrapper/lib/parse.ts

@@ -157,7 +157,9 @@ export function wrapCreateTaskResponse(
             switch (res.status) {
                 case "finished":
                 case "stopped":
+                    break poll;
 
+                case "created":
                 case "started":
                     await new Promise((resolve) => setTimeout(resolve, intervalMs));
                     break;
@@ -236,14 +238,15 @@ export function wrapCreateTaskResponse(
         options?: RequestOptions,
     ): Promise<TaskViewWithSchema<ZodType> | TaskView> {
         for await (const msg of _watch(response.id, config, options)) {
-            switch (msg.data.status) {
-                case "finished":
-                case "stopped":
-                case "started":
-                    break;
-                // default:
-                // throw new ExhaustiveSwitchCheck(msg.data.status as never);
+            // Return on terminal statuses
+            if (msg.data.status === "finished" || msg.data.status === "stopped") {
+                if (schema != null) {
+                    return parseStructuredTaskOutput<ZodType>(msg.data, schema);
+                } else {
+                    return msg.data;
+                }
             }
+            // Continue polling for any other status (created, started, or future statuses)
         }
 
         throw new Error("Stream ended before the task finished!");