traw 0.2.2 → 0.2.4

package/package.json

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

package/src/agent/agent.ts

@@ -72,9 +72,13 @@ export class Agent {
         log.step(step + 1, this.config.maxSteps, state.url)
 
         const thinkStart = Date.now()
+        let decision: { thought: string; action: Action }
         log.receiveStart()
-        const decision = await this.think(state)
-        log.receiveStop()
+        try {
+          decision = await this.think(state)
+        } finally {
+          log.receiveStop()
+        }
         this.aiTime += Date.now() - thinkStart
 
         log.thought(decision.thought)

package/src/api/mo-client.ts

@@ -22,7 +22,7 @@ export class MoClient {
     this.thinking = opts.thinking
   }
 
-  async chat(messages: ChatMessage[]): Promise<string> {
+  async chat(messages: ChatMessage[], timeoutMs = 300000): Promise<string> {
     const headers: Record<string, string> = {
       "Content-Type": "application/json",
     }
@@ -31,16 +31,30 @@ export class MoClient {
       headers["Authorization"] = `Bearer ${this.apiKey}`
     }
 
-    const resp = await fetch(`${this.url}/v1/chat/completions`, {
-      method: "POST",
-      headers,
-      body: JSON.stringify({
-        model: this.model,
-        messages,
-        stream: false,
-        thinking: this.thinking,
-      }),
-    })
+    const controller = new AbortController()
+    const timeoutId = setTimeout(() => controller.abort(), timeoutMs)
+
+    let resp: Response
+    try {
+      resp = await fetch(`${this.url}/v1/chat/completions`, {
+        method: "POST",
+        headers,
+        signal: controller.signal,
+        body: JSON.stringify({
+          model: this.model,
+          messages,
+          stream: false,
+          thinking: this.thinking,
+        }),
+      })
+    } catch (err: any) {
+      clearTimeout(timeoutId)
+      if (err.name === "AbortError") {
+        throw new Error(`mo timeout: no response in ${timeoutMs / 1000}s`)
+      }
+      throw err
+    }
+    clearTimeout(timeoutId)
 
     if (!resp.ok) {
       const body = await resp.text()

package/src/browser/controller.ts

@@ -74,8 +74,6 @@ export class BrowserController {
 
       const walk = (node: Element, depth: number) => {
         const el = node as HTMLElement
-        if (el.offsetParent === null && el.tagName !== "BODY" && el.tagName !== "HTML") return
-
         const tag = el.tagName.toLowerCase()
         const indent = "  ".repeat(depth)
 
@@ -99,7 +97,7 @@ export class BrowserController {
           if (href && tag === "a") attrs.push(`href="${esc(href.slice(0, 80))}"`)
           
           const val = (el as HTMLInputElement).value
-          if (val) attrs.push(`value="${esc(val.slice(0, 40))}"`)
+          if (val) attrs.push(`value="${esc(val)}"`)
           
           if ((el as any).disabled) attrs.push(`disabled="true"`)
           if ((el as any).checked) attrs.push(`checked="true"`)
@@ -108,7 +106,7 @@ export class BrowserController {
           if (el.getAttribute("aria-expanded")) attrs.push(`expanded="${el.getAttribute("aria-expanded")}"`)
           if (el.getAttribute("aria-selected") === "true") attrs.push(`selected="true"`)
 
-          const text = el.textContent?.trim().slice(0, 60) || ""
+          const text = el.textContent?.trim() || ""
           const ariaLabel = el.getAttribute("aria-label")
           const placeholder = (el as HTMLInputElement).placeholder
           const label = esc(ariaLabel || text || placeholder || "")
@@ -123,7 +121,6 @@ export class BrowserController {
               .map(n => n.textContent?.trim())
               .join(" ")
               .trim()
-              .slice(0, 120)
 
             if (directText.length > 2) {
               out.push(`${indent}<${tag}>${esc(directText)}</${tag}>`)