@leoustc/ai-runner-codex 0.1.1 → 0.1.2

package/AGENTS.md

@@ -141,6 +141,11 @@ For each valid payload, reply to the Hub as:
 { "requestId": "...", "type": "end", "name": "agent-name" }
 ```
 
+Large Codex reply text must be split into multiple `message` frames before the
+matching `end`. Each `message.text` chunk should be at most 8 KiB encoded as
+UTF-8. Include `message.chunk` metadata with `index`, `total`, and `maxBytes`
+when a reply has more than one chunk.
+
 Errors use:
 
 ```json

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@leoustc/ai-runner-codex",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "private": false,
   "description": "Codex app-server backed Agent Runner Hub CLI connector",
   "license": "MIT",

package/src/hub.js

@@ -1,4 +1,4 @@
-import { codexRequest, codexText, heartbeat, isRoomPayload, registration, reply } from "./protocol.js";
+import { codexRequest, codexText, heartbeat, isRoomPayload, messageReplies, registration, reply } from "./protocol.js";
 import { requestCodex } from "./codex.js";
 import { ensureCodexLogin, isCodexAuthError } from "./login.js";
 
@@ -94,10 +94,10 @@ export class HubClient {
     try {
       const response = await this.requestCodexWithLogin(value);
       const text = codexText(response) || "Codex returned no text.";
-      this.send(reply(value, this.config, "message", {
+      this.sendMessages(value, {
         model: this.config.model,
         text,
-      }));
+      });
     } catch (error) {
       this.send(reply(value, this.config, "error", {
         error: error instanceof Error ? error.message : String(error),
@@ -118,10 +118,10 @@ export class HubClient {
       }
 
       await ensureCodexLogin(this.config, this.log, async (text) => {
-        this.send(reply(value, this.config, "message", {
+        this.sendMessages(value, {
           codexLogin: true,
           text,
-        }));
+        });
       });
 
       return await requestCodex(this.config, request);
@@ -210,6 +210,12 @@ export class HubClient {
     }
   }
 
+  sendMessages(payload, message) {
+    for (const value of messageReplies(payload, this.config, message)) {
+      this.send(value);
+    }
+  }
+
   sendNow(value) {
     if (this.socket?.readyState !== WebSocket.OPEN) {
       return false;

package/src/protocol.js

@@ -36,6 +36,30 @@ export function reply(payload, config, type, message) {
   };
 }
 
+export function messageReplies(payload, config, message, maxBytes = 8 * 1024) {
+  if (typeof message?.text !== "string") {
+    return [reply(payload, config, "message", message)];
+  }
+
+  const chunks = splitUtf8Chunks(message.text, maxBytes);
+
+  return chunks.map((chunk, index) =>
+    reply(payload, config, "message", {
+      ...message,
+      text: chunk,
+      ...(chunks.length > 1
+        ? {
+            chunk: {
+              index,
+              total: chunks.length,
+              maxBytes,
+            },
+          }
+        : {}),
+    }),
+  );
+}
+
 export function codexRequest(payload, config) {
   return {
     requestId: payload.requestId,
@@ -80,3 +104,28 @@ function parseMessage(message) {
   }
 }
 
+function splitUtf8Chunks(text, maxBytes) {
+  const encoder = new TextEncoder();
+  const chunks = [];
+  let current = "";
+  let currentBytes = 0;
+
+  for (const character of text) {
+    const characterBytes = encoder.encode(character).byteLength;
+
+    if (current && currentBytes + characterBytes > maxBytes) {
+      chunks.push(current);
+      current = "";
+      currentBytes = 0;
+    }
+
+    current += character;
+    currentBytes += characterBytes;
+  }
+
+  if (current || chunks.length === 0) {
+    chunks.push(current);
+  }
+
+  return chunks;
+}