traw 0.1.1 → 0.1.3

package/.github/workflows/release-traw.yml

@@ -47,9 +47,13 @@ jobs:
             ```bash
             # npm
             npm install traw
+            # npm global
+            npm install -g traw
 
             # bun
             bun add traw
+            # bun global
+            bun add -g traw
             ```
 
             ### Usage

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "traw",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "module": "src/index.ts",
   "type": "module",
   "bin": {

package/src/agent.ts

@@ -1,6 +1,7 @@
 import type { Action, AgentConfig, AgentStep, ChatMessage, PageState } from "./types"
 import { BrowserController } from "./browser"
 import { MoClient } from "./mo-client"
+import { log } from "./log"
 
 const systemPrompt = `You are a browser automation agent. You see the page state and decide what to do next.
 
@@ -66,11 +67,11 @@ export class Agent {
   }
 
   async run(goal: string): Promise<AgentStep[]> {
-    console.log(`\n[goal] ${goal}\n`)
+    log.info("goal", goal)
 
-    console.log("[planning]...")
+    log.info("planning", "...")
     this.plan = await this.createPlan(goal)
-    console.log("\n" + this.plan + "\n")
+    log.plan(this.plan)
 
     await this.browser.launch()
     await this.browser.execute({
@@ -89,17 +90,17 @@ export class Agent {
       for (let step = 0; step < this.config.maxSteps; step++) {
         const state = await this.browser.getState()
 
-        console.log(`\n--- step ${step + 1} ---`)
-        console.log(`url: ${state.url}`)
-        console.log(`title: ${state.title}`)
+        log.step(step + 1)
+        log.dim("url", state.url)
+        log.dim("title", state.title)
 
         const decision = await this.think(state)
 
-        console.log(`thought: ${decision.thought}`)
-        console.log(`action: ${decision.action.type} - ${decision.action.reason}`)
+        log.thought(decision.thought)
+        log.action(decision.action.type, decision.action.reason)
 
         const result = await this.browser.execute(decision.action)
-        console.log(`result: ${result}`)
+        log.result(result)
 
         this.history.push({
           timestamp: Date.now(),
@@ -109,7 +110,7 @@ export class Agent {
         })
 
         if (decision.action.type === "done") {
-          console.log(`\n[complete]`)
+          log.done()
           break
         }
 
@@ -118,7 +119,7 @@ export class Agent {
     } finally {
       const videoPath = await this.browser.close()
       if (videoPath) {
-        console.log(`[video] ${videoPath}`)
+        log.info("video", videoPath)
       }
     }
 

package/src/index.ts

@@ -1,12 +1,13 @@
 #!/usr/bin/env bun
 import { Agent } from "./agent"
 import type { AgentConfig } from "./types"
+import { log } from "./log"
 
 const defaultConfig: AgentConfig = {
   moUrl: "http://localhost:8080",
   model: "glm-4.7",
   headless: false,
-  recordVideo: true,
+  recordVideo: false,
   maxSteps: 20,
 }
 
@@ -34,8 +35,8 @@ async function main() {
       config.headless = true
       continue
     }
-    if (arg === "--no-video") {
-      config.recordVideo = false
+    if (arg === "--video") {
+      config.recordVideo = true
       continue
     }
     if (arg.startsWith("--steps=")) {
@@ -53,49 +54,48 @@ async function main() {
 
   const goal = goalParts.join(" ")
   if (!goal) {
-    console.error("[error] provide a goal: bun run traw run \"your goal\"")
+    log.error("provide a goal: bun run traw run \"your goal\"")
     process.exit(1)
   }
 
-  console.log("[traw] starting agent...")
-  console.log(`  mo: ${config.moUrl}`)
-  console.log(`  headless: ${config.headless}`)
-  console.log(`  video: ${config.recordVideo}`)
-  console.log(`  max steps: ${config.maxSteps}`)
+  log.info("traw", "starting agent...")
+  log.dim("  mo", config.moUrl)
+  log.dim("  headless", String(config.headless))
+  log.dim("  video", String(config.recordVideo))
+  log.dim("  max steps", String(config.maxSteps))
 
   const agent = new Agent(config)
 
   try {
     const history = await agent.run(goal)
 
-    console.log("\n[done] steps:", history.length)
+    log.done(`steps: ${history.length}`)
     if (history.length > 0) {
       const last = history[history.length - 1]
-      console.log("  final:", last.action.type, "-", last.action.reason)
+      log.dim("  final", `${last.action.type} - ${last.action.reason}`)
     }
   } catch (err: any) {
-    console.error("\n[error]", err.message)
+    log.error(err.message)
     process.exit(1)
   }
 }
 
 function printHelp() {
   console.log(`
-🌐 traw - AI browser agent
+traw - AI browser agent
 
 Usage:
   traw run "your goal here"
-  traw run                    # interactive mode (coming soon)
 
 Options:
   --headless    run without visible browser
-  --no-video    disable video recording
+  --video       enable video recording
   --steps=N     max steps (default: 20)
   --mo=URL      mo server url (default: http://localhost:8080)
 
 Examples:
   traw run "find the weather in Moscow"
-  traw run "search for bun.js documentation"
+  traw run --video "search for bun.js documentation"
 `)
 }
 

package/src/log.ts

@@ -0,0 +1,50 @@
+// ansi color codes for terminal output
+
+const c = {
+  reset: "\x1b[0m",
+  dim: "\x1b[2m",
+  cyan: "\x1b[36m",
+  green: "\x1b[32m",
+  yellow: "\x1b[33m",
+  red: "\x1b[31m",
+  magenta: "\x1b[35m",
+  blue: "\x1b[34m",
+}
+
+export const log = {
+  info: (tag: string, msg: string) => {
+    console.log(`${c.cyan}[${tag}]${c.reset} ${msg}`)
+  },
+
+  step: (n: number) => {
+    console.log(`\n${c.magenta}--- step ${n} ---${c.reset}`)
+  },
+
+  dim: (label: string, value: string) => {
+    console.log(`${c.dim}${label}:${c.reset} ${value}`)
+  },
+
+  thought: (msg: string) => {
+    console.log(`${c.yellow}thought:${c.reset} ${msg}`)
+  },
+
+  action: (type: string, reason: string) => {
+    console.log(`${c.blue}action:${c.reset} ${type} - ${reason}`)
+  },
+
+  result: (msg: string) => {
+    console.log(`${c.green}result:${c.reset} ${msg}`)
+  },
+
+  done: (msg?: string) => {
+    console.log(`\n${c.green}[done]${c.reset}${msg ? " " + msg : ""}`)
+  },
+
+  error: (msg: string) => {
+    console.error(`${c.red}[error]${c.reset} ${msg}`)
+  },
+
+  plan: (text: string) => {
+    console.log(`\n${c.dim}${text}${c.reset}\n`)
+  },
+}