fixed deprecated stuff in the sdk

Author: gregpr07

src/wrapper/api/BrowserUseTasks.ts

@@ -69,25 +69,25 @@ export class BrowserUseTasks extends Tasks {
     }
 
     public getTask<T extends ZodType>(
-        request: { taskId: string; schema: T },
+        request: { task_id: string; schema: T },
         requestOptions?: Tasks.RequestOptions,
     ): core.HttpResponsePromise<TaskViewWithSchema<T>>;
     public getTask(
-        taskId: string,
+        request: BrowserUse.GetTaskTasksTaskIdGetRequest,
         requestOptions?: Tasks.RequestOptions,
     ): core.HttpResponsePromise<BrowserUse.TaskView>;
     public getTask(
-        request: { taskId: string; schema: ZodType } | string,
+        request: { task_id: string; schema?: ZodType },
         requestOptions?: Tasks.RequestOptions,
     ): core.HttpResponsePromise<BrowserUse.TaskView | TaskViewWithSchema<ZodType>> {
-        if (typeof request === "string") {
+        if (!request.schema) {
             return super.getTask(request, requestOptions);
         }
 
-        const { taskId, schema } = request;
+        const { task_id, schema } = request;
 
         const promise: Promise<core.WithRawResponse<TaskViewWithSchema<ZodType>>> = super
-            .getTask(taskId, requestOptions)
+            .getTask({ task_id }, requestOptions)
             .withRawResponse()
             .then((res) => {
                 const parsed = parseStructuredTaskOutput(res.data, schema);

src/wrapper/lib/parse.ts

@@ -145,7 +145,7 @@ export function wrapCreateTaskResponse(
                 break poll;
             }
 
-            const res = await client.getTask(taskId);
+            const res = await client.getTask({ task_id: taskId });
 
             const resHash = getTaskViewHash(res);
 
@@ -158,13 +158,12 @@ export function wrapCreateTaskResponse(
             switch (res.status) {
                 case "finished":
                 case "stopped":
-                case "paused":
-                    break poll;
+
                 case "started":
                     await new Promise((resolve) => setTimeout(resolve, intervalMs));
                     break;
                 default:
-                    throw new ExhaustiveSwitchCheck(res.status);
+                    throw new ExhaustiveSwitchCheck(res.status as never);
             }
         } while (true);
     }
@@ -241,24 +240,10 @@ export function wrapCreateTaskResponse(
             switch (msg.data.status) {
                 case "finished":
                 case "stopped":
-                case "paused": {
-                    if (schema != null) {
-                        const parsed: TaskViewWithSchema<ZodType> = parseStructuredTaskOutput<ZodType>(
-                            msg.data,
-                            schema,
-                        );
-
-                        return parsed;
-                    } else {
-                        const result: TaskView = msg.data;
-
-                        return result;
-                    }
-                }
                 case "started":
                     break;
                 default:
-                    throw new ExhaustiveSwitchCheck(msg.data.status);
+                    throw new ExhaustiveSwitchCheck(msg.data.status as never);
             }
         }