|
1 | 1 | #!/usr/bin/env -S npm run tsn -T
|
2 | 2 |
|
3 |
| -import { BrowserUse } from 'browser-use-sdk'; |
| 3 | +import { BrowserUseClient } from "browser-use-sdk"; |
4 | 4 |
|
5 |
| -import { env, spinner } from './utils'; |
6 |
| -import z from 'zod'; |
| 5 | +import { env, spinner } from "./utils"; |
| 6 | +import z from "zod"; |
7 | 7 |
|
8 | 8 | env();
|
9 | 9 |
|
10 | 10 | // gets API Key from environment variable BROWSER_USE_API_KEY
|
11 |
| -const browseruse = new BrowserUse(); |
| 11 | +const browseruse = new BrowserUseClient({ |
| 12 | + apiKey: process.env.BROWSER_USE_API_KEY!, |
| 13 | + environment: "production", |
| 14 | +}); |
| 15 | + |
| 16 | +// Basic --------------------------------------------------------------------- |
12 | 17 |
|
13 | 18 | async function basic() {
|
14 |
| - let log = 'starting'; |
15 |
| - const stop = spinner(() => log); |
16 |
| - |
17 |
| - // Create Task |
18 |
| - const rsp = await browseruse.tasks.create({ |
19 |
| - task: "What's the weather line in SF and what's the temperature?", |
20 |
| - agentSettings: { llm: 'gemini-2.5-flash' }, |
21 |
| - }); |
22 |
| - |
23 |
| - poll: do { |
24 |
| - // Wait for Task to Finish |
25 |
| - const status = await browseruse.tasks.retrieve(rsp.id); |
26 |
| - |
27 |
| - switch (status.status) { |
28 |
| - case 'started': |
29 |
| - case 'paused': |
30 |
| - case 'stopped': |
31 |
| - log = `agent ${status.status} - live: ${status.session.liveUrl}`; |
32 |
| - |
33 |
| - await new Promise((resolve) => setTimeout(resolve, 2000)); |
34 |
| - break; |
35 |
| - |
36 |
| - case 'finished': |
37 |
| - stop(); |
38 |
| - |
39 |
| - console.log(status.doneOutput); |
40 |
| - break poll; |
41 |
| - } |
42 |
| - } while (true); |
| 19 | + let log = "starting"; |
| 20 | + const stop = spinner(() => log); |
| 21 | + |
| 22 | + // Create Task |
| 23 | + const rsp = await browseruse.tasks.createTask({ |
| 24 | + task: "What's the weather line in SF and what's the temperature?", |
| 25 | + agent: { llm: "gemini-2.5-flash" }, |
| 26 | + }); |
| 27 | + |
| 28 | + poll: do { |
| 29 | + // Wait for Task to Finish |
| 30 | + const status = await browseruse.tasks.getTask(rsp.id); |
| 31 | + |
| 32 | + switch (status.status) { |
| 33 | + case "started": |
| 34 | + case "paused": |
| 35 | + case "stopped": |
| 36 | + log = `agent ${status.status} - live: ${status.session.liveUrl}`; |
| 37 | + |
| 38 | + await new Promise((resolve) => setTimeout(resolve, 2000)); |
| 39 | + break; |
| 40 | + |
| 41 | + case "finished": |
| 42 | + stop(); |
| 43 | + |
| 44 | + console.log(status.output); |
| 45 | + break poll; |
| 46 | + } |
| 47 | + } while (true); |
43 | 48 | }
|
44 | 49 |
|
| 50 | +// Structured ---------------------------------------------------------------- |
| 51 | + |
45 | 52 | // Define Structured Output Schema
|
46 | 53 | const HackerNewsResponse = z.object({
|
47 |
| - title: z.string(), |
48 |
| - url: z.string(), |
49 |
| - score: z.number(), |
| 54 | + title: z.string(), |
| 55 | + url: z.string(), |
| 56 | + score: z.number(), |
50 | 57 | });
|
51 | 58 |
|
52 | 59 | const TaskOutput = z.object({
|
53 |
| - posts: z.array(HackerNewsResponse), |
| 60 | + posts: z.array(HackerNewsResponse), |
54 | 61 | });
|
55 | 62 |
|
56 | 63 | async function structured() {
|
57 |
| - let log = 'starting'; |
58 |
| - const stop = spinner(() => log); |
59 |
| - |
60 |
| - // Create Task |
61 |
| - const rsp = await browseruse.tasks.create({ |
62 |
| - task: 'Extract top 10 Hacker News posts and return the title, url, and score', |
63 |
| - schema: TaskOutput, |
64 |
| - agentSettings: { llm: 'gpt-4.1' }, |
65 |
| - }); |
66 |
| - |
67 |
| - poll: do { |
68 |
| - // Wait for Task to Finish |
69 |
| - const status = await browseruse.tasks.retrieve({ |
70 |
| - taskId: rsp.id, |
71 |
| - schema: TaskOutput, |
| 64 | + let log = "starting"; |
| 65 | + const stop = spinner(() => log); |
| 66 | + |
| 67 | + // Create Task |
| 68 | + const rsp = await browseruse.tasks.createTask({ |
| 69 | + task: "Extract top 10 Hacker News posts and return the title, url, and score", |
| 70 | + schema: TaskOutput, |
| 71 | + agent: { llm: "gpt-4.1" }, |
72 | 72 | });
|
73 | 73 |
|
74 |
| - switch (status.status) { |
75 |
| - case 'started': |
76 |
| - case 'paused': |
77 |
| - case 'stopped': { |
78 |
| - log = `agent ${status.status} ${status.session.liveUrl} | ${status.steps.length} steps`; |
| 74 | + poll: do { |
| 75 | + // Wait for Task to Finish |
| 76 | + const status = await browseruse.tasks.getTask({ |
| 77 | + taskId: rsp.id, |
| 78 | + schema: TaskOutput, |
| 79 | + }); |
79 | 80 |
|
80 |
| - await new Promise((resolve) => setTimeout(resolve, 2000)); |
| 81 | + switch (status.status) { |
| 82 | + case "started": |
| 83 | + case "paused": |
| 84 | + case "stopped": { |
| 85 | + log = `agent ${status.status} ${status.session.liveUrl} | ${status.steps.length} steps`; |
81 | 86 |
|
82 |
| - break; |
83 |
| - } |
| 87 | + await new Promise((resolve) => setTimeout(resolve, 2000)); |
84 | 88 |
|
85 |
| - case 'finished': |
86 |
| - if (status.parsedOutput == null) { |
87 |
| - throw new Error('No output'); |
88 |
| - } |
| 89 | + break; |
| 90 | + } |
89 | 91 |
|
90 |
| - stop(); |
| 92 | + case "finished": |
| 93 | + if (status.parsed == null) { |
| 94 | + throw new Error("No output"); |
| 95 | + } |
91 | 96 |
|
92 |
| - // Print Structured Output |
93 |
| - console.log('Top Hacker News Posts:'); |
| 97 | + stop(); |
94 | 98 |
|
95 |
| - for (const post of status.parsedOutput.posts) { |
96 |
| - console.log(` - ${post.title} (${post.score}) ${post.url}`); |
97 |
| - } |
| 99 | + // Print Structured Output |
| 100 | + console.log("Top Hacker News Posts:"); |
| 101 | + |
| 102 | + for (const post of status.parsed.posts) { |
| 103 | + console.log(` - ${post.title} (${post.score}) ${post.url}`); |
| 104 | + } |
98 | 105 |
|
99 |
| - break poll; |
100 |
| - } |
101 |
| - } while (true); |
| 106 | + break poll; |
| 107 | + } |
| 108 | + } while (true); |
102 | 109 | }
|
103 | 110 |
|
104 | 111 | basic()
|
105 |
| - .then(() => structured()) |
106 |
| - .catch(console.error); |
| 112 | + .then(() => structured()) |
| 113 | + .catch(console.error); |
0 commit comments