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

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,22 @@ 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 and link
+ * rules every other runtime carries.
+ */
+export declare const piPrompt: (message: string) => string;
+/**
+ * The messages an answer becomes: each link written alone on a line as its
+ * own message, text in chunks the API takes, and the buttons its fenced block
+ * asked for under the last words. 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, LINK_LINE_INSTRUCTION, answerMessages as splitAnswer, } 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,38 @@ 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 and link
+ * 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} ${LINK_LINE_INSTRUCTION} ${BUTTONS_GUIDANCE}`;
+/**
+ * The messages an answer becomes: each link written alone on a line as its
+ * own message, text in chunks the API takes, and the buttons its fenced block
+ * asked for under the last words. 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 split = splitAnswer(answer);
+    const messages = [];
+    for (const [first, ...rest] of split.messages) {
+        if (first?.type !== "text" || first.value.length <= 10_000) {
+            messages.push({ parts: first ? [first, ...rest] : rest });
+            continue;
+        }
+        const chunks = first.value.match(/[\s\S]{1,10000}/gu) ?? [];
+        for (const [index, chunk] of chunks.entries()) {
+            messages.push({ parts: [{ type: "text", value: chunk }, ...(index === chunks.length - 1 ? rest : [])] });
+        }
+    }
+    if (split.error && messages[0])
+        messages[0].error = split.error;
+    return messages;
+};
 class ChatSession {
     process;
     lines;
@@ -126,7 +153,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 +162,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.6",
   "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.6"
   },
   "devDependencies": {
     "@mariozechner/pi-coding-agent": "0.73.1",