@botconnector/bccli 0.4.1 → 0.4.2

package/dist/{chunk-QZKLZELZ.js → chunk-OK2USAXX.js}

@@ -1,12 +1,36 @@
 #!/usr/bin/env node
 
 // src/tokenBudget.ts
+var IMAGE_TOKEN_COST = 1e3;
 var estimateTokens = (text) => Math.ceil(text.length / 4);
-var estimateMessages = (messages) => estimateTokens(JSON.stringify(messages));
+function isImageAttachmentMessage(m) {
+  if (m.role !== "user" || !Array.isArray(m.content)) return false;
+  const hasImage = m.content.some((p) => p.type === "image_url");
+  const firstText = m.content.find((p) => p.type === "text");
+  return hasImage && !!firstText && firstText.type === "text" && firstText.text.startsWith("Image read from ");
+}
+function isUserTurnBoundary(m) {
+  return m.role === "user" && !isImageAttachmentMessage(m);
+}
+function estimateMessages(messages) {
+  let total = 0;
+  for (const m of messages) {
+    if (m.role === "user" && Array.isArray(m.content)) {
+      for (const part of m.content) {
+        if (part.type === "image_url") total += IMAGE_TOKEN_COST;
+        else total += estimateTokens(part.text);
+      }
+      total += 8;
+    } else {
+      total += estimateTokens(JSON.stringify(m));
+    }
+  }
+  return total;
+}
 function trimToFit(messages, maxTokens) {
   let out = messages;
   while (estimateMessages(out) > maxTokens) {
-    const boundary = out.findIndex((m, i) => i > 0 && m.role === "user");
+    const boundary = out.findIndex((m, i) => i > 0 && isUserTurnBoundary(m));
     if (boundary === -1) return out;
     out = out.slice(boundary);
   }
@@ -16,13 +40,13 @@ function splitKeepRecent(messages, keepTokens) {
   if (messages.length === 0) return [[], []];
   let start = messages.length;
   for (let i = messages.length - 1; i >= 0; i--) {
-    if (i > 0 && messages[i].role !== "user") continue;
+    if (i > 0 && !isUserTurnBoundary(messages[i])) continue;
     if (estimateMessages(messages.slice(i)) <= keepTokens) start = i;
     else break;
   }
   if (start === messages.length) {
     for (let i = messages.length - 1; i >= 0; i--) {
-      if (messages[i].role === "user") {
+      if (isUserTurnBoundary(messages[i])) {
         start = i;
         break;
       }