@friendlyrobot/discord-pi-agent 0.4.2 → 0.4.3

package/dist/index.js

@@ -520,13 +520,16 @@ async function handleCommand(input, agentService, promptQueue) {
 }
 
 // src/message-chunker.ts
+import { Lexer as Lexer2 } from "marked";
 var DISCORD_MESSAGE_LIMIT = 2000;
 var SAFE_MESSAGE_LIMIT = 1900;
 function chunkMessage(text) {
   if (text.length <= SAFE_MESSAGE_LIMIT) {
     return [text];
   }
+  const codeBlockRanges = getCodeBlockRanges(text);
   const chunks = [];
+  let offset = 0;
   let remaining = text;
   while (remaining.length > SAFE_MESSAGE_LIMIT) {
     const candidate = remaining.slice(0, SAFE_MESSAGE_LIMIT);
@@ -534,15 +537,41 @@ function chunkMessage(text) {
 
 `), candidate.lastIndexOf(`
 `), candidate.lastIndexOf(" "));
-    const boundary = splitIndex > 0 ? splitIndex : SAFE_MESSAGE_LIMIT;
+    const relBoundary = splitIndex > 0 ? splitIndex : SAFE_MESSAGE_LIMIT;
+    const absBoundary = adjustBoundaryForCodeBlock(offset + relBoundary, codeBlockRanges);
+    const boundary = absBoundary - offset;
     chunks.push(remaining.slice(0, boundary).trim());
     remaining = remaining.slice(boundary).trim();
+    offset = absBoundary;
   }
   if (remaining.length > 0) {
     chunks.push(remaining);
   }
   return chunks.filter((chunk) => chunk.length > 0).map((chunk) => chunk.slice(0, DISCORD_MESSAGE_LIMIT));
 }
+function getCodeBlockRanges(text) {
+  const tokens = Lexer2.lex(text);
+  const ranges = [];
+  let pos = 0;
+  for (const token of tokens) {
+    if (token.type === "code") {
+      ranges.push({ start: pos, end: pos + token.raw.length });
+    }
+    pos += token.raw.length;
+  }
+  return ranges;
+}
+function adjustBoundaryForCodeBlock(absBoundary, codeBlockRanges) {
+  for (const range of codeBlockRanges) {
+    if (absBoundary > range.start && absBoundary < range.end) {
+      if (range.end <= DISCORD_MESSAGE_LIMIT) {
+        return range.end;
+      }
+      return absBoundary;
+    }
+  }
+  return absBoundary;
+}
 
 // src/discord-client.ts
 async function startDiscordClient(config, agentService, promptQueue) {

package/dist/message-chunker.test.d.ts

@@ -0,0 +1 @@
+export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@friendlyrobot/discord-pi-agent",
-  "version": "0.4.2",
+  "version": "0.4.3",
   "description": "Reusable Discord gateway bridge for persistent pi agent sessions",
   "license": "MIT",
   "type": "module",