@illuma-ai/agents 1.0.89 → 1.0.93

package/src/messages/prune.ts

@@ -41,11 +41,11 @@ function addThinkingBlock(
   const content: MessageContentComplex[] = Array.isArray(message.content)
     ? (message.content as MessageContentComplex[])
     : [
-      {
-        type: ContentTypes.TEXT,
-        text: message.content,
-      },
-    ];
+        {
+          type: ContentTypes.TEXT,
+          text: message.content,
+        },
+      ];
   /** Edge case, the message already has the thinking block */
   if (content[0].type === thinkingBlock.type) {
     return message;
@@ -94,6 +94,45 @@ export type PruningResult = {
  * @param options Configuration options for processing messages
  * @returns Object containing the message context, remaining tokens, messages not included, and summary index
  */
+/**
+ * Builds an emergency context from first and last message pairs when normal pruning
+ * removes all AI messages. Preserves conversation continuity by keeping boundaries.
+ */
+function buildEmergencyContext(
+  messages: BaseMessage[],
+  endIndex: number
+): BaseMessage[] {
+  const result: BaseMessage[] = [];
+  const instructions = endIndex > 0 ? messages[0] : undefined;
+  if (instructions) {
+    result.push(instructions);
+  }
+
+  /** Find first user+AI pair after system message */
+  for (let i = endIndex; i < messages.length - 1; i++) {
+    const current = messages[i];
+    const next = messages[i + 1];
+    if (current.getType() === 'human' && next.getType() === 'ai') {
+      result.push(current, next);
+      break;
+    }
+  }
+
+  /** Find last user+AI pair */
+  for (let i = messages.length - 1; i > endIndex; i--) {
+    const current = messages[i];
+    const prev = messages[i - 1];
+    if (current.getType() === 'ai' && prev.getType() === 'human') {
+      if (!result.includes(prev)) {
+        result.push(prev, current);
+      }
+      break;
+    }
+  }
+
+  return result;
+}
+
 export function getMessagesWithinTokenLimit({
   messages: _messages,
   maxContextTokens,
@@ -292,9 +331,13 @@ export function getMessagesWithinTokenLimit({
   }
 
   if (assistantIndex === -1) {
-    throw new Error(
-      'The payload is malformed. There is a thinking sequence but no "AI" messages to append thinking blocks to.'
-    );
+    /** Emergency fallback: keep first and last message pairs instead of crashing */
+    const emergencyContext = buildEmergencyContext(_messages, endIndex);
+    return {
+      remainingContextTokens: 0,
+      context: emergencyContext,
+      messagesToRefine: _messages.filter((m) => !emergencyContext.includes(m)),
+    };
   }
 
   thinkingStartIndex = originalLength - 1 - assistantIndex;
@@ -536,25 +579,30 @@ export function createPruneMessages(factoryParams: PruneMessagesFactoryParams) {
 
     lastTurnStartIndex = params.messages.length;
     if (lastCutOffIndex === 0 && totalTokens <= factoryParams.maxTokens) {
-      return { context: params.messages, indexTokenCountMap, messagesToRefine: [] };
+      return {
+        context: params.messages,
+        indexTokenCountMap,
+        messagesToRefine: [],
+      };
     }
 
-    const { context, thinkingStartIndex, messagesToRefine } = getMessagesWithinTokenLimit({
-      maxContextTokens: factoryParams.maxTokens,
-      messages: params.messages,
-      indexTokenCountMap,
-      startType: params.startType,
-      thinkingEnabled: factoryParams.thinkingEnabled,
-      tokenCounter: factoryParams.tokenCounter,
-      reasoningType:
-        factoryParams.provider === Providers.BEDROCK
-          ? ContentTypes.REASONING_CONTENT
-          : ContentTypes.THINKING,
-      thinkingStartIndex:
-        factoryParams.thinkingEnabled === true
-          ? runThinkingStartIndex
-          : undefined,
-    });
+    const { context, thinkingStartIndex, messagesToRefine } =
+      getMessagesWithinTokenLimit({
+        maxContextTokens: factoryParams.maxTokens,
+        messages: params.messages,
+        indexTokenCountMap,
+        startType: params.startType,
+        thinkingEnabled: factoryParams.thinkingEnabled,
+        tokenCounter: factoryParams.tokenCounter,
+        reasoningType:
+          factoryParams.provider === Providers.BEDROCK
+            ? ContentTypes.REASONING_CONTENT
+            : ContentTypes.THINKING,
+        thinkingStartIndex:
+          factoryParams.thinkingEnabled === true
+            ? runThinkingStartIndex
+            : undefined,
+      });
     runThinkingStartIndex = thinkingStartIndex ?? -1;
     /** The index is the first value of `context`, index relative to `params.messages` */
     lastCutOffIndex = Math.max(
@@ -566,3 +614,17 @@ export function createPruneMessages(factoryParams: PruneMessagesFactoryParams) {
     return { context, indexTokenCountMap, messagesToRefine };
   };
 }
+
+/** Calculates context utilization as a percentage (0-100) */
+export function getContextUtilization(
+  indexTokenCountMap: Record<string, number | undefined>,
+  instructionTokens: number,
+  maxContextTokens: number
+): number {
+  if (maxContextTokens <= 0) return 0;
+  let totalTokens = instructionTokens;
+  for (const key in indexTokenCountMap) {
+    totalTokens += indexTokenCountMap[key] ?? 0;
+  }
+  return (totalTokens / maxContextTokens) * 100;
+}

package/src/messages/reducer.js

@@ -0,0 +1,60 @@
+import { coerceMessageLikeToMessage, } from '@langchain/core/messages';
+import { v4 } from 'uuid';
+export const REMOVE_ALL_MESSAGES = '__remove_all__';
+/**
+ * Prebuilt reducer that combines returned messages.
+ * Can handle standard messages and special modifiers like {@link RemoveMessage}
+ * instances.
+ */
+export function messagesStateReducer(left, right) {
+    const leftArray = Array.isArray(left) ? left : [left];
+    const rightArray = Array.isArray(right) ? right : [right];
+    // coerce to message
+    const leftMessages = leftArray.map(coerceMessageLikeToMessage);
+    const rightMessages = rightArray.map(coerceMessageLikeToMessage);
+    // assign missing ids
+    for (const m of leftMessages) {
+        if (m.id === null || m.id === undefined) {
+            m.id = v4();
+            m.lc_kwargs.id = m.id;
+        }
+    }
+    let removeAllIdx;
+    for (let i = 0; i < rightMessages.length; i += 1) {
+        const m = rightMessages[i];
+        if (m.id === null || m.id === undefined) {
+            m.id = v4();
+            m.lc_kwargs.id = m.id;
+        }
+        if (m.getType() === 'remove' && m.id === REMOVE_ALL_MESSAGES) {
+            removeAllIdx = i;
+        }
+    }
+    if (removeAllIdx != null)
+        return rightMessages.slice(removeAllIdx + 1);
+    // merge
+    const merged = [...leftMessages];
+    const mergedById = new Map(merged.map((m, i) => [m.id, i]));
+    const idsToRemove = new Set();
+    for (const m of rightMessages) {
+        const existingIdx = mergedById.get(m.id);
+        if (existingIdx !== undefined) {
+            if (m.getType() === 'remove') {
+                idsToRemove.add(m.id);
+            }
+            else {
+                idsToRemove.delete(m.id);
+                merged[existingIdx] = m;
+            }
+        }
+        else {
+            if (m.getType() === 'remove') {
+                throw new Error(`Attempting to delete a message with an ID that doesn't exist ('${m.id}')`);
+            }
+            mergedById.set(m.id, merged.length);
+            merged.push(m);
+        }
+    }
+    return merged.filter((m) => !idsToRemove.has(m.id));
+}
+//# sourceMappingURL=reducer.js.map

package/src/messages/reducer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reducer.js","sourceRoot":"","sources":["reducer.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,0BAA0B,GAC3B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AAE1B,MAAM,CAAC,MAAM,mBAAmB,GAAG,gBAAgB,CAAC;AAOpD;;;;GAIG;AACH,MAAM,UAAU,oBAAoB,CAClC,IAAc,EACd,KAAe;IAEf,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACtD,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IAC1D,oBAAoB;IACpB,MAAM,YAAY,GAAI,SAA+B,CAAC,GAAG,CACvD,0BAA0B,CAC3B,CAAC;IACF,MAAM,aAAa,GAAI,UAAgC,CAAC,GAAG,CACzD,0BAA0B,CAC3B,CAAC;IACF,qBAAqB;IACrB,KAAK,MAAM,CAAC,IAAI,YAAY,EAAE,CAAC;QAC7B,IAAI,CAAC,CAAC,EAAE,KAAK,IAAI,IAAI,CAAC,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;YACxC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC;YACZ,CAAC,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;QACxB,CAAC;IACH,CAAC;IAED,IAAI,YAAgC,CAAC;IACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QACjD,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QAC3B,IAAI,CAAC,CAAC,EAAE,KAAK,IAAI,IAAI,CAAC,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;YACxC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC;YACZ,CAAC,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;QACxB,CAAC;QAED,IAAI,CAAC,CAAC,OAAO,EAAE,KAAK,QAAQ,IAAI,CAAC,CAAC,EAAE,KAAK,mBAAmB,EAAE,CAAC;YAC7D,YAAY,GAAG,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;IAED,IAAI,YAAY,IAAI,IAAI;QAAE,OAAO,aAAa,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC;IAEvE,QAAQ;IACR,MAAM,MAAM,GAAG,CAAC,GAAG,YAAY,CAAC,CAAC;IACjC,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5D,MAAM,WAAW,GAAG,IAAI,GAAG,EAAE,CAAC;IAC9B,KAAK,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;QAC9B,MAAM,WAAW,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACzC,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YAC9B,IAAI,CAAC,CAAC,OAAO,EAAE,KAAK,QAAQ,EAAE,CAAC;gBAC7B,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACxB,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACzB,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,CAAC,OAAO,EAAE,KAAK,QAAQ,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CACb,kEAAkE,CAAC,CAAC,EAAE,IAAI,CAC3E,CAAC;YACJ,CAAC;YACD,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YACpC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AACtD,CAAC"}

package/src/messages/shiftIndexTokenCountMap.test.js

@@ -0,0 +1,63 @@
+import { shiftIndexTokenCountMap } from './format';
+describe('shiftIndexTokenCountMap', () => {
+    it('should add a system message token count at index 0 and shift all other indices', () => {
+        const originalMap = {
+            0: 10,
+            1: 20,
+            2: 30
+        };
+        const systemMessageTokenCount = 15;
+        const result = shiftIndexTokenCountMap(originalMap, systemMessageTokenCount);
+        // Check that the system message token count is at index 0
+        expect(result[0]).toBe(15);
+        // Check that all other indices are shifted by 1
+        expect(result[1]).toBe(10);
+        expect(result[2]).toBe(20);
+        expect(result[3]).toBe(30);
+        // Check that the original map is not modified
+        expect(originalMap[0]).toBe(10);
+        expect(originalMap[1]).toBe(20);
+        expect(originalMap[2]).toBe(30);
+    });
+    it('should handle an empty map', () => {
+        const emptyMap = {};
+        const systemMessageTokenCount = 15;
+        const result = shiftIndexTokenCountMap(emptyMap, systemMessageTokenCount);
+        // Check that only the system message token count is in the result
+        expect(Object.keys(result).length).toBe(1);
+        expect(result[0]).toBe(15);
+    });
+    it('should handle non-sequential indices', () => {
+        const nonSequentialMap = {
+            0: 10,
+            2: 20,
+            5: 30
+        };
+        const systemMessageTokenCount = 15;
+        const result = shiftIndexTokenCountMap(nonSequentialMap, systemMessageTokenCount);
+        // Check that the system message token count is at index 0
+        expect(result[0]).toBe(15);
+        // Check that all other indices are shifted by 1
+        expect(result[1]).toBe(10);
+        expect(result[3]).toBe(20);
+        expect(result[6]).toBe(30);
+    });
+    it('should handle string keys', () => {
+        // TypeScript will convert string keys to numbers when accessing the object
+        const mapWithStringKeys = {
+            '0': 10,
+            '1': 20,
+            '2': 30
+        };
+        const systemMessageTokenCount = 15;
+        // Cast to Record<number, number> to match the function signature
+        const result = shiftIndexTokenCountMap(mapWithStringKeys, systemMessageTokenCount);
+        // Check that the system message token count is at index 0
+        expect(result[0]).toBe(15);
+        // Check that all other indices are shifted by 1
+        expect(result[1]).toBe(10);
+        expect(result[2]).toBe(20);
+        expect(result[3]).toBe(30);
+    });
+});
+//# sourceMappingURL=shiftIndexTokenCountMap.test.js.map

package/src/messages/shiftIndexTokenCountMap.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shiftIndexTokenCountMap.test.js","sourceRoot":"","sources":["shiftIndexTokenCountMap.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AAEnD,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;QACxF,MAAM,WAAW,GAA2B;YAC1C,CAAC,EAAE,EAAE;YACL,CAAC,EAAE,EAAE;YACL,CAAC,EAAE,EAAE;SACN,CAAC;QAEF,MAAM,uBAAuB,GAAG,EAAE,CAAC;QAEnC,MAAM,MAAM,GAAG,uBAAuB,CAAC,WAAW,EAAE,uBAAuB,CAAC,CAAC;QAE7E,0DAA0D;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE3B,gDAAgD;QAChD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE3B,8CAA8C;QAC9C,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,QAAQ,GAA2B,EAAE,CAAC;QAC5C,MAAM,uBAAuB,GAAG,EAAE,CAAC;QAEnC,MAAM,MAAM,GAAG,uBAAuB,CAAC,QAAQ,EAAE,uBAAuB,CAAC,CAAC;QAE1E,kEAAkE;QAClE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,gBAAgB,GAA2B;YAC/C,CAAC,EAAE,EAAE;YACL,CAAC,EAAE,EAAE;YACL,CAAC,EAAE,EAAE;SACN,CAAC;QAEF,MAAM,uBAAuB,GAAG,EAAE,CAAC;QAEnC,MAAM,MAAM,GAAG,uBAAuB,CAAC,gBAAgB,EAAE,uBAAuB,CAAC,CAAC;QAElF,0DAA0D;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE3B,gDAAgD;QAChD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,2EAA2E;QAC3E,MAAM,iBAAiB,GAA2B;YAChD,GAAG,EAAE,EAAE;YACP,GAAG,EAAE,EAAE;YACP,GAAG,EAAE,EAAE;SACR,CAAC;QAEF,MAAM,uBAAuB,GAAG,EAAE,CAAC;QAEnC,iEAAiE;QACjE,MAAM,MAAM,GAAG,uBAAuB,CAAC,iBAAsD,EAAE,uBAAuB,CAAC,CAAC;QAExH,0DAA0D;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE3B,gDAAgD;QAChD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/src/messages/summarize.js

@@ -0,0 +1,146 @@
+import { HumanMessage } from '@langchain/core/messages';
+const MAX_MESSAGE_CHARS = 4000;
+const MAX_TRUNCATED_CHARS = 500;
+const SIMPLE_MAX_TOKENS = 512;
+export const FULL_SUMMARY_TEMPLATE = `Analyze this conversation and produce a structured summary covering each section below. Be precise and preserve actionable details.
+
+## 1. Task Overview & User Intent
+What the user originally asked for and their high-level goal.
+
+## 2. Technical Context
+Languages, frameworks, libraries, APIs, and architectural patterns involved.
+
+## 3. Files & Code State
+All file paths mentioned, their current state, and any pending modifications.
+
+## 4. Problem Resolution History
+Problems encountered, debugging steps taken, and their outcomes.
+
+## 5. Agent Workflow State
+{agent_workflow}
+
+## 6. Tool Results Summary
+Tools called, their inputs, and key outputs (omit verbose raw data).
+
+## 7. Progress Tracking
+What has been completed, what remains, and any blockers.
+
+## 8. Active Working State
+Variables, configurations, and runtime state that must be preserved.
+
+## 9. Continuation Plan
+Immediate next steps and the order in which they should be executed.
+
+## 10. Critical Context
+Any constraints, warnings, edge cases, or decisions that must not be lost.
+
+Conversation:
+{conversation}
+
+Structured Summary:`;
+export const SIMPLE_SUMMARY_TEMPLATE = 'Summarize this conversation preserving: user\'s original request, key decisions, current task state, and any file paths mentioned.\n\nConversation:\n{conversation}\n\nSummary:';
+export function formatMessagesForSummary(messages) {
+    const parts = [];
+    for (const message of messages) {
+        const role = message.getType();
+        const raw = typeof message.content === 'string'
+            ? message.content
+            : JSON.stringify(message.content);
+        const content = raw.length > MAX_MESSAGE_CHARS
+            ? raw.slice(0, MAX_MESSAGE_CHARS) + '...'
+            : raw;
+        parts.push(`${role}: ${content}`);
+    }
+    return parts.join('\n');
+}
+export function buildFullSummaryPrompt(conversation, config) {
+    let agentWorkflow = 'N/A (single-agent conversation)';
+    if (config?.isMultiAgent === true && config.agentWorkflowState != null) {
+        const { currentAgentId, agentChain, pendingAgents } = config.agentWorkflowState;
+        agentWorkflow = [
+            `Current agent: ${currentAgentId}`,
+            `Agent chain: ${agentChain.join(' -> ')}`,
+            `Pending agents: ${pendingAgents.length > 0 ? pendingAgents.join(', ') : 'none'}`,
+        ].join('\n');
+    }
+    return FULL_SUMMARY_TEMPLATE
+        .replace('{agent_workflow}', agentWorkflow)
+        .replace('{conversation}', conversation);
+}
+export function buildSimpleSummaryPrompt(conversation) {
+    return SIMPLE_SUMMARY_TEMPLATE.replace('{conversation}', conversation);
+}
+export function createEmergencySummary(messages) {
+    let firstUserContent = '';
+    let lastAIContent = '';
+    const toolNames = new Set();
+    for (const message of messages) {
+        const type = message.getType();
+        const content = typeof message.content === 'string'
+            ? message.content
+            : JSON.stringify(message.content);
+        if (type === 'human' && firstUserContent.length === 0) {
+            firstUserContent = content.slice(0, MAX_TRUNCATED_CHARS);
+        }
+        if (type === 'ai') {
+            lastAIContent = content.slice(0, MAX_TRUNCATED_CHARS);
+        }
+        if (type === 'tool') {
+            const name = message.name;
+            if (name != null && name.length > 0) {
+                toolNames.add(name);
+            }
+        }
+    }
+    const toolList = toolNames.size > 0 ? [...toolNames].join(', ') : 'none';
+    return [
+        '[Emergency Context Summary]',
+        `Original request: ${firstUserContent}`,
+        `Last response: ${lastAIContent}`,
+        `Tools used: ${toolList}`,
+        `Messages compacted: ${messages.length}`,
+    ].join('\n');
+}
+export function validateSummarySize(summary, budget, tokenCounter) {
+    const count = tokenCounter(new HumanMessage(summary));
+    return count <= budget;
+}
+export async function summarize(messages, callback, config) {
+    const messageCount = messages.length;
+    const maxOutputTokens = config?.maxOutputTokens ?? 1024;
+    // Tier 1: Full structured summary
+    try {
+        const conversation = formatMessagesForSummary(messages);
+        const prompt = buildFullSummaryPrompt(conversation, config);
+        const result = await callback(prompt, maxOutputTokens);
+        if (result != null && result.length > 0) {
+            if (config?.summaryBudget != null && config.summaryBudget > 0 && config.tokenCounter != null) {
+                if (validateSummarySize(result, config.summaryBudget, config.tokenCounter)) {
+                    return { summary: result, tier: 'full', messagesCompacted: messageCount };
+                }
+            }
+            else {
+                return { summary: result, tier: 'full', messagesCompacted: messageCount };
+            }
+        }
+    }
+    catch {
+        // Fall through to Tier 2
+    }
+    // Tier 2: Simple summary
+    try {
+        const conversation = formatMessagesForSummary(messages);
+        const prompt = buildSimpleSummaryPrompt(conversation);
+        const result = await callback(prompt, SIMPLE_MAX_TOKENS);
+        if (result != null && result.length > 0) {
+            return { summary: result, tier: 'simple', messagesCompacted: messageCount };
+        }
+    }
+    catch {
+        // Fall through to Tier 3
+    }
+    // Tier 3: Emergency (no LLM)
+    const summary = createEmergencySummary(messages);
+    return { summary, tier: 'emergency', messagesCompacted: messageCount };
+}
+//# sourceMappingURL=summarize.js.map

package/src/messages/summarize.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"summarize.js","sourceRoot":"","sources":["summarize.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAe,MAAM,0BAA0B,CAAC;AA6BrE,MAAM,iBAAiB,GAAG,IAAI,CAAC;AAC/B,MAAM,mBAAmB,GAAG,GAAG,CAAC;AAChC,MAAM,iBAAiB,GAAG,GAAG,CAAC;AAE9B,MAAM,CAAC,MAAM,qBAAqB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;oBAmCjB,CAAC;AAErB,MAAM,CAAC,MAAM,uBAAuB,GAClC,iLAAiL,CAAC;AAEpL,MAAM,UAAU,wBAAwB,CAAC,QAAuB;IAC9D,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;QAC/B,MAAM,GAAG,GAAG,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;YAC7C,CAAC,CAAC,OAAO,CAAC,OAAO;YACjB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACpC,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,GAAG,iBAAiB;YAC5C,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,iBAAiB,CAAC,GAAG,KAAK;YACzC,CAAC,CAAC,GAAG,CAAC;QACR,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,KAAK,OAAO,EAAE,CAAC,CAAC;IACpC,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,YAAoB,EACpB,MAAwB;IAExB,IAAI,aAAa,GAAG,iCAAiC,CAAC;IACtD,IAAI,MAAM,EAAE,YAAY,KAAK,IAAI,IAAI,MAAM,CAAC,kBAAkB,IAAI,IAAI,EAAE,CAAC;QACvE,MAAM,EAAE,cAAc,EAAE,UAAU,EAAE,aAAa,EAAE,GAAG,MAAM,CAAC,kBAAkB,CAAC;QAChF,aAAa,GAAG;YACd,kBAAkB,cAAc,EAAE;YAClC,gBAAgB,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE;YACzC,mBAAmB,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE;SAClF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACf,CAAC;IAED,OAAO,qBAAqB;SACzB,OAAO,CAAC,kBAAkB,EAAE,aAAa,CAAC;SAC1C,OAAO,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;AAC7C,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,YAAoB;IAC3D,OAAO,uBAAuB,CAAC,OAAO,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;AACzE,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,QAAuB;IAC5D,IAAI,gBAAgB,GAAG,EAAE,CAAC;IAC1B,IAAI,aAAa,GAAG,EAAE,CAAC;IACvB,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IAEpC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;QAC/B,MAAM,OAAO,GAAG,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;YACjD,CAAC,CAAC,OAAO,CAAC,OAAO;YACjB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAEpC,IAAI,IAAI,KAAK,OAAO,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtD,gBAAgB,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,mBAAmB,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAClB,aAAa,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,mBAAmB,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;YACpB,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;YAC1B,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IACzE,OAAO;QACL,6BAA6B;QAC7B,qBAAqB,gBAAgB,EAAE;QACvC,kBAAkB,aAAa,EAAE;QACjC,eAAe,QAAQ,EAAE;QACzB,uBAAuB,QAAQ,CAAC,MAAM,EAAE;KACzC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,OAAe,EACf,MAAc,EACd,YAA0B;IAE1B,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC;IACtD,OAAO,KAAK,IAAI,MAAM,CAAC;AACzB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,SAAS,CAC7B,QAAuB,EACvB,QAA4E,EAC5E,MAAwB;IAExB,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC;IACrC,MAAM,eAAe,GAAG,MAAM,EAAE,eAAe,IAAI,IAAI,CAAC;IAExD,kCAAkC;IAClC,IAAI,CAAC;QACH,MAAM,YAAY,GAAG,wBAAwB,CAAC,QAAQ,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,sBAAsB,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;QAEvD,IAAI,MAAM,IAAI,IAAI,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxC,IAAI,MAAM,EAAE,aAAa,IAAI,IAAI,IAAI,MAAM,CAAC,aAAa,GAAG,CAAC,IAAI,MAAM,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;gBAC7F,IAAI,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,aAAa,EAAE,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;oBAC3E,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,iBAAiB,EAAE,YAAY,EAAE,CAAC;gBAC5E,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,iBAAiB,EAAE,YAAY,EAAE,CAAC;YAC5E,CAAC;QACH,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,yBAAyB;IAC3B,CAAC;IAED,yBAAyB;IACzB,IAAI,CAAC;QACH,MAAM,YAAY,GAAG,wBAAwB,CAAC,QAAQ,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,wBAAwB,CAAC,YAAY,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC;QAEzD,IAAI,MAAM,IAAI,IAAI,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,iBAAiB,EAAE,YAAY,EAAE,CAAC;QAC9E,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,yBAAyB;IAC3B,CAAC;IAED,6BAA6B;IAC7B,MAAM,OAAO,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;IACjD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,iBAAiB,EAAE,YAAY,EAAE,CAAC;AACzE,CAAC"}