fix model

Author: maticzav

examples/retrieve.ts

@@ -61,7 +61,7 @@ async function structured() {
   const rsp = await browseruse.tasks.create({
     task: 'Extract top 10 Hacker News posts and return the title, url, and score',
     schema: TaskOutput,
-    agentSettings: { llm: 'gemini-2.5-flash' },
+    agentSettings: { llm: 'gpt-4.1' },
   });
 
   poll: do {

examples/run.ts

@@ -41,7 +41,7 @@ async function structured() {
   const rsp = await browseruse.tasks.run({
     task: 'Search for the top 10 Hacker News posts and return the title, url, and score',
     schema: TaskOutput,
-    agentSettings: { llm: 'gemini-2.5-flash' },
+    agentSettings: { llm: 'gpt-4.1' },
   });
 
   const posts = rsp.parsedOutput?.posts;

examples/stream.ts

@@ -25,6 +25,10 @@ async function basic() {
     console.log(
       `Basic: ${msg.data.status} ${msg.data.session.liveUrl} ${msg.data.steps[msg.data.steps.length - 1]?.nextGoal}`,
     );
+
+    if (msg.data.status === 'finished') {
+      console.log(`Basic: ${msg.data.doneOutput}`);
+    }
   }
 
   console.log('\nBasic: Stream completed');
@@ -50,7 +54,7 @@ async function structured() {
   const task = await browseruse.tasks.create({
     task: 'Extract top 10 Hacker News posts and return the title, url, and score',
     schema: TaskOutput,
-    agentSettings: { llm: 'gemini-2.5-flash' },
+    agentSettings: { llm: 'gpt-4.1' },
   });
 
   const stream = browseruse.tasks.stream({

src/lib/stream.ts

@@ -1,86 +1,14 @@
-import type { TaskView, TaskStepView } from '../resources/tasks';
-import { type DeepMutable, ExhaustiveSwitchCheck } from './types';
-
-export type ReducerEvent = TaskView | null;
-
-export type BrowserState = Readonly<{
-  taskId: string;
-  sessionId: string;
-
-  liveUrl: string | null;
-  steps: ReadonlyArray<TaskStepView>;
-  doneOutput: string | null;
-}> | null;
-
-type BrowserAction = {
-  kind: 'status';
-  status: TaskView;
-};
-
-export function reducer(state: BrowserState, action: BrowserAction): [BrowserState, ReducerEvent] {
-  switch (action.kind) {
-    case 'status': {
-      // INIT
-
-      if (state == null) {
-        const liveUrl = action.status.session.liveUrl ?? null;
-        const doneOutput = action.status.doneOutput ?? null;
-
-        const state: BrowserState = {
-          taskId: action.status.id,
-          sessionId: action.status.sessionId,
-          liveUrl: liveUrl,
-          steps: action.status.steps,
-          doneOutput: doneOutput,
-        };
-
-        return [state, action.status];
-      }
-
-      // UPDATE
-
-      const liveUrl = action.status.session.liveUrl ?? state.liveUrl;
-      const doneOutput = action.status.doneOutput ?? state.doneOutput;
-
-      const steps: TaskStepView[] = [...state.steps];
-      if (action.status.steps != null) {
-        const newSteps = action.status.steps.slice(state.steps.length);
-
-        for (const step of newSteps) {
-          steps.push(step);
-        }
-      }
-
-      const newState: DeepMutable<BrowserState> = {
-        ...state,
-        liveUrl,
-        steps,
-        doneOutput,
-      };
-
-      // CHANGES
-
-      if (
-        (state.liveUrl == null && liveUrl != null) ||
-        state.steps.length !== steps.length ||
-        state.doneOutput != doneOutput
-      ) {
-        const update: ReducerEvent = {
-          ...action.status,
-          steps: newState.steps,
-          session: {
-            ...action.status.session,
-            liveUrl: newState.liveUrl,
-          },
-          doneOutput: newState.doneOutput,
-        };
-
-        return [newState, update];
-      }
-
-      return [newState, null];
-    }
-    default:
-      throw new ExhaustiveSwitchCheck(action.kind);
-  }
+import { createHash } from 'crypto';
+import stringify from 'fast-json-stable-stringify';
+
+import type { TaskView } from '../resources';
+
+/**
+ * Hashes the task view to detect changes.
+ * Uses fast-json-stable-stringify for deterministic JSON, then SHA-256.
+ */
+export function getTaskViewHash(view: TaskView): string {
+  const dump = stringify(view);
+  const hash = createHash('sha256').update(dump).digest('hex');
+  return hash;
 }

src/resources/tasks.ts

@@ -12,7 +12,7 @@ import {
   type TaskCreateParamsWithSchema,
   type TaskViewWithSchema,
 } from '../lib/parse';
-import { BrowserState, reducer } from '../lib/stream';
+import { getTaskViewHash } from '../lib/stream';
 
 export class Tasks extends APIResource {
   /**
@@ -131,29 +131,26 @@ export class Tasks extends APIResource {
     config: { interval: number },
     options?: RequestOptions,
   ): AsyncGenerator<{ event: 'status'; data: TaskView }> {
-    const tick: { current: number } = { current: 0 };
-    const state: { current: BrowserState } = { current: null };
+    const hash: { current: string | null } = { current: null };
 
     poll: do {
       if (options?.signal?.aborted) {
         break poll;
       }
 
-      tick.current++;
+      const res = await this.retrieve(taskId);
 
-      const status = await this.retrieve(taskId);
+      const resHash = getTaskViewHash(res);
 
-      const [newState, event] = reducer(state.current, { kind: 'status', status });
+      if (hash.current == null || resHash !== hash.current) {
+        hash.current = resHash;
 
-      if (event != null) {
-        yield { event: 'status', data: event };
-
-        if (event.status === 'finished') {
-          break;
-        }
+        yield { event: 'status', data: res };
       }
 
-      state.current = newState;
+      if (res.status === 'finished') {
+        break poll;
+      }
 
       await new Promise((resolve) => setTimeout(resolve, config.interval));
     } while (true);