@meetopenbot/pi 0.1.1 → 0.1.3

package/dist/runtime.js

@@ -1,17 +1,12 @@
 import { resolveConfig } from './config.js';
-import { remapPiError } from './credits.js';
 import { getOrCreatePiSession } from './session.js';
 import { runPiPrompt } from './stream.js';
-export async function runPiTurn({ prompt, handlerCtx, context }) {
+export async function* runPiTurn({ prompt, handlerCtx, context, }) {
     const config = resolveConfig(context, handlerCtx.state.channelDetails?.cwd);
     const { session } = await getOrCreatePiSession({
         config,
         state: handlerCtx.state,
         storage: context.storage,
     });
-    const text = await runPiPrompt(session, prompt, { streaming: session.isStreaming });
-    if (text.startsWith('Pi error: ')) {
-        return `Pi error: ${remapPiError(text.slice('Pi error: '.length))}`;
-    }
-    return text;
+    yield* runPiPrompt(session, prompt, { streaming: session.isStreaming });
 }

package/dist/stream.js

@@ -1,36 +1,53 @@
+import { createTextReplyBuffer, textReply, yieldFromBackgroundQueue, } from '@meetopenbot/plugin-sdk';
+import { remapPiError } from './credits.js';
 const isAssistantMessage = (message) => message.role === 'assistant';
-/** Run a Pi prompt and return the final assistant text. */
-export async function runPiPrompt(session, prompt, options) {
-    let assistantText = '';
-    let result = '';
-    let wake;
-    let finished = false;
-    let error;
-    const wakeUp = () => {
-        wake?.();
-        wake = undefined;
-    };
-    const onEvent = (event) => {
-        if (event.type === 'message_update') {
-            const assistantEvent = event.assistantMessageEvent;
-            if (assistantEvent.type === 'text_delta')
-                assistantText += assistantEvent.delta;
-            return;
-        }
-        if (event.type === 'agent_end') {
-            const lastAssistant = [...event.messages].reverse().find(isAssistantMessage);
-            if (lastAssistant?.stopReason === 'error' && lastAssistant.errorMessage?.trim()) {
-                result = `Pi error: ${lastAssistant.errorMessage.trim()}`;
-            }
-            else {
-                result = assistantText;
-            }
-            finished = true;
-            wakeUp();
+export const classifyPiEvent = (event) => {
+    if (event.type === 'tool_execution_start') {
+        return {
+            type: 'tool_call',
+            id: event.toolCallId,
+            name: event.toolName,
+            input: event.args,
+        };
+    }
+    if (event.type === 'tool_execution_end') {
+        return {
+            type: 'tool_result',
+            id: event.toolCallId,
+            name: event.toolName,
+            output: event.result,
+            error: event.isError,
+        };
+    }
+    if (event.type === 'message_update') {
+        const assistantEvent = event.assistantMessageEvent;
+        if (assistantEvent.type === 'text_delta')
+            return { type: 'delta', text: assistantEvent.delta };
+        return { type: 'skip' };
+    }
+    if (event.type === 'message_end')
+        return { type: 'flush' };
+    return { type: 'skip' };
+};
+function* consumePiEvent(event, replies) {
+    if (event.type === 'agent_end') {
+        yield* replies.end();
+        const lastAssistant = [...event.messages].reverse().find(isAssistantMessage);
+        if (lastAssistant?.stopReason === 'error' && lastAssistant.errorMessage?.trim()) {
+            yield textReply(`Pi error: ${remapPiError(lastAssistant.errorMessage.trim())}`);
         }
-    };
-    const unsubscribe = session.subscribe(onEvent);
-    const run = (async () => {
+        return;
+    }
+    yield* replies.apply(classifyPiEvent(event));
+}
+/** Run a Pi prompt and yield each completed assistant turn and tool widget. */
+export async function* runPiPrompt(session, prompt, options) {
+    const replies = createTextReplyBuffer({ groupId: 'pi:tools' });
+    yield* yieldFromBackgroundQueue(async (enqueue) => {
+        const unsubscribe = session.subscribe((event) => {
+            for (const item of consumePiEvent(event, replies))
+                enqueue(item);
+        });
         try {
             if (options?.streaming && session.isStreaming) {
                 await session.followUp(prompt);
@@ -39,30 +56,10 @@ export async function runPiPrompt(session, prompt, options) {
                 await session.prompt(prompt);
             }
         }
-        catch (err) {
-            error = err instanceof Error ? err : new Error(String(err));
-        }
         finally {
-            finished = true;
-            wakeUp();
+            unsubscribe();
+            for (const item of replies.end())
+                enqueue(item);
         }
-    })();
-    try {
-        while (!finished) {
-            await new Promise((resolve) => {
-                if (finished) {
-                    resolve();
-                    return;
-                }
-                wake = resolve;
-            });
-        }
-        await run;
-        if (error)
-            throw error;
-        return result.trim();
-    }
-    finally {
-        unsubscribe();
-    }
+    });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@meetopenbot/pi",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "OpenBot agent plugin powered by the Pi coding agent SDK.",
   "type": "module",
   "main": "./dist/index.js",
@@ -21,7 +21,7 @@
   "dependencies": {
     "@earendil-works/pi-ai": "^0.79.1",
     "@earendil-works/pi-coding-agent": "^0.79.1",
-    "@meetopenbot/plugin-sdk": "^0.3.0"
+    "@meetopenbot/plugin-sdk": "^0.5.0"
   },
   "devDependencies": {
     "@types/node": "^25.9.2",