@meetopenbot/stagehand 1.0.3 → 1.0.5

package/dist/index.js

@@ -3,6 +3,7 @@ import { defineAgentPlugin } from "@meetopenbot/plugin-sdk";
 
 // src/agent.ts
 import { Stagehand } from "@browserbasehq/stagehand";
+import { textReply as textReply2 } from "@meetopenbot/plugin-sdk";
 
 // src/config.ts
 import {
@@ -30,14 +31,70 @@ function resolveStagehandConfig(config) {
   };
 }
 
+// src/stream.ts
+import {
+  createTextReplyBuffer,
+  textReply
+} from "@meetopenbot/plugin-sdk";
+var classifyStagehandPart = (part) => {
+  if (part.type === "text-delta") {
+    const text = part.text ?? part.delta ?? "";
+    return text ? { type: "delta", text } : { type: "skip" };
+  }
+  if (part.type === "tool-call" && part.toolCallId) {
+    return {
+      type: "tool_call",
+      id: part.toolCallId,
+      name: part.toolName ?? "tool",
+      input: part.input
+    };
+  }
+  if ((part.type === "tool-result" || part.type === "tool-error") && part.toolCallId) {
+    return {
+      type: "tool_result",
+      id: part.toolCallId,
+      name: part.toolName,
+      input: part.input,
+      output: part.type === "tool-error" ? part.error : part.output,
+      error: part.type === "tool-error"
+    };
+  }
+  if (part.type === "step-finish" || part.type === "finish") {
+    return { type: "flush" };
+  }
+  return { type: "skip" };
+};
+async function* runStagehandStream(fullStream) {
+  const replies = createTextReplyBuffer({ groupId: "stagehand:tools" });
+  for await (const part of fullStream) {
+    yield* replies.apply(classifyStagehandPart(part));
+  }
+  yield* replies.end();
+}
+function finalStagehandReply(args) {
+  const items = [];
+  if (!args.yielded && args.message?.trim()) items.push(textReply(args.message.trim()));
+  if (args.output && typeof args.output === "object" && Object.keys(args.output).length > 0) {
+    items.push(textReply(JSON.stringify(args.output, null, 2)));
+  }
+  return items;
+}
+
 // src/agent.ts
-async function runStagehandTurn({ prompt, context }) {
+async function* runStagehandTurn({
+  prompt,
+  context
+}) {
   const config = resolveStagehandConfig(context.config);
   if (config.requestedEnv === "BROWSERBASE" && (!config.browserbaseApiKey || !config.browserbaseProjectId)) {
-    return "Stagehand Browserbase setup is incomplete. Set `BROWSERBASE_API_KEY` and `BROWSERBASE_PROJECT_ID` in workspace settings.";
+    yield textReply2(
+      "Stagehand Browserbase setup is incomplete. Set `BROWSERBASE_API_KEY` and `BROWSERBASE_PROJECT_ID` in workspace settings."
+    );
+    return;
   }
   if (config.requestedEnv === "LOCAL" && !config.apiKey) {
-    return "Stagehand local setup is incomplete. Set `OPENAI_API_KEY` in workspace settings.";
+    yield textReply2("Stagehand local setup is incomplete. Set `OPENAI_API_KEY` in workspace settings.");
+    return;
   }
   let stagehand = null;
   try {
@@ -58,15 +115,19 @@ async function runStagehandTurn({ prompt, context }) {
       instruction: prompt,
       maxSteps: config.maxSteps
     });
-    for await (const _part of streamResult.fullStream) {
+    let yielded = false;
+    for await (const item of runStagehandStream(streamResult.fullStream)) {
+      yielded = true;
+      yield item;
     }
     const result = await streamResult.result;
-    const parts = [];
-    if (result.message) parts.push(result.message);
-    if (result.output && Object.keys(result.output).length > 0) {
-      parts.push(JSON.stringify(result.output, null, 2));
+    for (const item of finalStagehandReply({
+      yielded,
+      message: result.message,
+      output: result.output
+    })) {
+      yield item;
     }
-    return parts.join("\n\n");
   } finally {
     if (stagehand) await stagehand.close().catch(() => {
     });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@meetopenbot/stagehand",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "type": "module",
   "description": "Stagehand AI browser automation plugin for OpenBot",
   "main": "./dist/index.js",
@@ -24,7 +24,7 @@
     "@browserbasehq/stagehand": "^3.3.0",
     "playwright": "^1.52.0",
     "zod": "^4.3.5",
-    "@meetopenbot/plugin-sdk": "^0.3.0"
+    "@meetopenbot/plugin-sdk": "^0.5.0"
   },
   "devDependencies": {
     "@types/node": "^20.10.1",