@relaymessenger/pi 0.1.4-staging.4 → 0.1.4-staging.5

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import Relay from "@relaymessenger/sdk";
+import Relay, { type MessagePart } from "@relaymessenger/sdk";
 export interface PiChannelOptions {
     readonly agentToken: string;
     readonly baseURL?: string;
@@ -16,6 +16,21 @@ export interface PiProcess {
     readonly stdout: AsyncIterable<string>;
     readonly kill: () => void;
 }
+/**
+ * The prompt pi is given for one message: the words, then how to answer.
+ * This process sends pi's final text for it, and the same buttons rules
+ * every other runtime carries.
+ */
+export declare const piPrompt: (message: string) => string;
+/**
+ * The messages an answer becomes: text in chunks the API takes, and the
+ * buttons its fenced block asked for on the last one. A block pi wrote that
+ * cannot be read stays in the words, so nothing the person was told is lost.
+ */
+export declare const answerMessages: (answer: string) => {
+    parts: MessagePart[];
+    error?: string;
+}[];
 export declare class PiChannel {
     #private;
     constructor(options: PiChannelOptions);

package/dist/index.js

@@ -1,6 +1,6 @@
 import { spawn } from "node:child_process";
 import { createInterface } from "node:readline";
-import Relay, {} from "@relaymessenger/sdk";
+import Relay, { BUTTONS_BLOCK_INSTRUCTION, BUTTONS_GUIDANCE, splitButtons, } from "@relaymessenger/sdk";
 class ChildPiProcess {
     #child;
     constructor(command, args) { this.#child = spawn(command, [...args], { stdio: ["pipe", "pipe", "pipe"] }); this.#child.stderr.resume(); }
@@ -12,11 +12,34 @@ const textFromEvent = (event) => {
     if (event.event_type !== "message.received" || event.data.direction !== "inbound")
         return null;
     return event.data.parts
-        .flatMap((part) => part.type === "text" || part.type === "link"
-        ? [part.value]
-        : part.type === "button_reply" ? [part.label] : [])
+        .flatMap((part) => part.type === "text" || part.type === "link" ? [part.value] : [])
         .join("\n").trim() || null;
 };
+/**
+ * The prompt pi is given for one message: the words, then how to answer.
+ * This process sends pi's final text for it, and the same buttons rules
+ * every other runtime carries.
+ */
+export const piPrompt = (message) => `${message}\n\nWrite your answer as your final message. Relay sends that answer to the chat for you, so do not send it yourself. Write chat text. Inline Markdown draws: bold, italic, strikethrough, code, links. Headings, lists and code fences show as written.\n\n${BUTTONS_BLOCK_INSTRUCTION} ${BUTTONS_GUIDANCE}`;
+/**
+ * The messages an answer becomes: text in chunks the API takes, and the
+ * buttons its fenced block asked for on the last one. A block pi wrote that
+ * cannot be read stays in the words, so nothing the person was told is lost.
+ */
+export const answerMessages = (answer) => {
+    const { text, buttons, error } = splitButtons(answer);
+    const chunks = text.match(/[\s\S]{1,10000}/gu) ?? [];
+    const messages = chunks.map((chunk) => ({ parts: [{ type: "text", value: chunk }] }));
+    if (buttons) {
+        if (messages.length === 0)
+            messages.push({ parts: [buttons] });
+        else
+            messages[messages.length - 1].parts.push(buttons);
+    }
+    if (error && messages[0])
+        messages[0].error = error;
+    return messages;
+};
 class ChatSession {
     process;
     lines;
@@ -126,7 +149,7 @@ export class PiChannel {
             this.#sessions.set(data.chat.id, session);
         }
         const timeout = this.#options.rpcTimeoutMs ?? 60_000;
-        await session.command("prompt", { message }, timeout, signal);
+        await session.command("prompt", { message: piPrompt(message) }, timeout, signal);
         if (!session.settled) {
             while (!session.settled)
                 await session.read(timeout, signal);
@@ -135,9 +158,11 @@ export class PiChannel {
         const answer = response.data?.text?.trim();
         if (!answer)
             throw new Error("Pi returned no final text answer");
-        const chunks = answer.match(/[\s\S]{1,10000}/gu) ?? [];
-        for (const [index, chunk] of chunks.entries())
-            await this.#relay.chats.messages.send(data.chat.id, { message: { parts: [{ type: "text", value: chunk }], idempotency_key: `pi-${event.event_id}-${index}` } });
+        const messages = answerMessages(answer);
+        if (messages[0]?.error)
+            console.error(`Relay: the buttons block in pi's answer was left as text: ${messages[0].error}.`);
+        for (const [index, message] of messages.entries())
+            await this.#relay.chats.messages.send(data.chat.id, { message: { parts: message.parts, idempotency_key: `pi-${event.event_id}-${index}` } });
     }
 }
 export const runPiChannel = (options, signal) => new PiChannel(options).run(signal);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@relaymessenger/pi",
-  "version": "0.1.4-staging.4",
+  "version": "0.1.4-staging.5",
   "description": "Relay Messenger channel for Pi over Relay v1 WebSocket delivery.",
   "type": "module",
   "main": "./dist/index.js",
@@ -34,7 +34,7 @@
     "test": "vitest run"
   },
   "dependencies": {
-    "@relaymessenger/sdk": "0.3.6-staging.4"
+    "@relaymessenger/sdk": "0.3.6-staging.5"
   },
   "devDependencies": {
     "@mariozechner/pi-coding-agent": "0.73.1",