traw 0.1.2 → 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.2",
+  "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,6 +1,7 @@
 #!/usr/bin/env bun
 import { Agent } from "./agent"
 import type { AgentConfig } from "./types"
+import { log } from "./log"
 
 const defaultConfig: AgentConfig = {
   moUrl: "http://localhost:8080",
@@ -53,28 +54,28 @@ 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)
   }
 }

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`)
+  },
+}