@illuma-ai/agents 1.0.96 → 1.0.98

package/src/graphs/contextManagement.e2e.test.ts

@@ -596,30 +596,38 @@ describe('Pre-invocation utilization gate', () => {
     expect(emergency.length).toBeLessThan(2000); // Emergency summaries are compact
   });
 
-  it('injects delegation hint at >70% utilization for agents with task tool', () => {
-    const utilization = 75;
-    const hasTaskTool = true;
-
-    if (utilization > 70 && hasTaskTool) {
-      const delegationHint = new HumanMessage({
-        content:
-          '[System] Context window is at 75% capacity. Consider delegating complex sub-tasks ' +
-          'to the task tool to maintain context availability.',
-      });
-      expect(delegationHint.content).toContain('75%');
-      expect(delegationHint.content).toContain('task tool');
+  it('does NOT inject token budget hints at any utilization level', () => {
+    // Token budget hints were removed to prevent LLM voluntary bail-out.
+    // Context overflow is handled mechanically by pruning + auto-continuation.
+    // See: docs/context-overflow-architecture.md
+    const utilizationLevels = [50, 70, 85, 95, 101];
+    for (const utilization of utilizationLevels) {
+      const messages = buildConversation(10, 200);
+      // No message should contain raw token numbers or budget percentages
+      for (const msg of messages) {
+        const content =
+          typeof msg.content === 'string'
+            ? msg.content
+            : JSON.stringify(msg.content);
+        expect(content).not.toMatch(/CONTEXT BUDGET/);
+        expect(content).not.toMatch(/\d+ of \d+ tokens/);
+      }
     }
   });
 
-  it('does not inject delegation hint below 70%', () => {
-    const utilization = 65;
-    let delegationInjected = false;
-
-    if (utilization > 70) {
-      delegationInjected = true;
+  it('post-prune note does not contain token numbers', () => {
+    // After pruning, a context note is injected but it must not
+    // expose any token counts or budget percentages to the LLM
+    const { buildPostPruneNote } = require('@/utils/contextPressure');
+    const noteWithSummary = buildPostPruneNote(10, true);
+    const noteWithout = buildPostPruneNote(10, false);
+    for (const note of [noteWithSummary, noteWithout]) {
+      expect(note).not.toBeNull();
+      expect(note).not.toMatch(/\d+%/);
+      expect(note).not.toMatch(/\d+ of \d+ tokens/);
+      expect(note).not.toMatch(/BUDGET/i);
+      expect(note).toContain('task');
     }
-
-    expect(delegationInjected).toBe(false);
   });
 });
 

package/src/specs/agent-handoffs-bedrock.integration.test.ts

@@ -20,7 +20,7 @@ config({
 import { HumanMessage, ToolMessage } from '@langchain/core/messages';
 import type { RunnableConfig } from '@langchain/core/runnables';
 import type * as t from '@/types';
-import { Providers, Constants, GraphEvents } from '@/common';
+import { Providers, Constants, GraphEvents, EdgeType } from '@/common';
 import { StandardGraph } from '@/graphs/Graph';
 import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
 import { ToolEndHandler, ModelEndHandler } from '@/events';
@@ -118,13 +118,13 @@ describeIf('Agent Handoff E2E with Bedrock', () => {
         {
           from: 'supervisor_abc123',
           to: 'agent_W47hBnn2RoVZEOy5595GC',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           // No description - should auto-generate from agent name + description
         },
         {
           from: 'supervisor_abc123',
           to: 'agent_X92kLmn4TpQR8vw3221HD',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           // No description
         },
       ];
@@ -203,8 +203,8 @@ describeIf('Agent Handoff E2E with Bedrock', () => {
       ];
 
       const edges: t.GraphEdge[] = [
-        { from: 'router', to: 'sales_agent', edgeType: 'handoff' },
-        { from: 'router', to: 'support_agent', edgeType: 'handoff' },
+        { from: 'router', to: 'sales_agent', edgeType: EdgeType.HANDOFF },
+        { from: 'router', to: 'support_agent', edgeType: EdgeType.HANDOFF },
       ];
 
       const { contentParts: _contentParts, aggregateContent } =
@@ -321,8 +321,8 @@ describeIf('Agent Handoff E2E with Bedrock', () => {
       ];
 
       const edges: t.GraphEdge[] = [
-        { from: 'router', to: 'sales_agent', edgeType: 'handoff' },
-        { from: 'router', to: 'support_agent', edgeType: 'handoff' },
+        { from: 'router', to: 'sales_agent', edgeType: EdgeType.HANDOFF },
+        { from: 'router', to: 'support_agent', edgeType: EdgeType.HANDOFF },
       ];
 
       const { contentParts: _contentParts, aggregateContent } =

package/src/specs/agent-handoffs.test.ts

@@ -4,7 +4,7 @@ import { HumanMessage, ToolMessage } from '@langchain/core/messages';
 import type { ToolCall } from '@langchain/core/messages/tool';
 import type { RunnableConfig } from '@langchain/core/runnables';
 import type * as t from '@/types';
-import { Providers, Constants } from '@/common';
+import { Providers, Constants, EdgeType } from '@/common';
 import { StandardGraph } from '@/graphs/Graph';
 import { Run } from '@/run';
 
@@ -94,7 +94,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent B',
         },
       ];
@@ -128,7 +128,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent B when needed',
         },
       ];
@@ -192,7 +192,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
       ];
 
@@ -223,13 +223,13 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent B',
         },
         {
           from: 'agent_b',
           to: 'agent_a',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent A',
         },
       ];
@@ -268,12 +268,12 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
         {
           from: 'agent_b',
           to: 'agent_a',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
       ];
 
@@ -334,13 +334,13 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent B',
         },
         {
           from: 'agent_b',
           to: 'agent_c',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent C',
         },
       ];
@@ -395,19 +395,19 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'router',
           to: 'agent_a',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent A for task A',
         },
         {
           from: 'router',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent B for task B',
         },
         {
           from: 'router',
           to: 'agent_c',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent C for task C',
         },
       ];
@@ -449,13 +449,13 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'router',
           to: 'agent_a',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent A',
         },
         {
           from: 'router',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent B',
         },
       ];
@@ -519,7 +519,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent B with instructions',
           prompt: 'Provide specific instructions for agent B',
           promptKey: 'instructions',
@@ -551,7 +551,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           prompt: 'Instructions for handoff',
           // promptKey not specified, should default to 'instructions'
         },
@@ -581,7 +581,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent B',
           prompt: 'Additional context for agent B',
           promptKey: 'context',
@@ -638,7 +638,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_a',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Self-handoff (should be allowed but unusual)',
         },
       ];
@@ -724,7 +724,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to agent B',
         },
       ];
@@ -759,12 +759,12 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
         {
           from: 'agent_b',
           to: 'agent_c',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
       ];
 
@@ -806,12 +806,12 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_c',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
         {
           from: 'agent_b',
           to: 'agent_c',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
       ];
 
@@ -853,7 +853,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'router',
           to: 'data_analyst',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to data analyst',
           prompt: 'Instructions for the analyst about what to analyze',
           promptKey: 'instructions',
@@ -939,7 +939,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'flight_assistant',
           to: 'hotel_assistant',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to hotel booking',
         },
       ];
@@ -970,7 +970,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_with_underscores',
           to: 'AgentWithCamelCase',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
       ];
 
@@ -1007,7 +1007,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'supervisor',
           to: 'data_analyst',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           // No description provided - should auto-generate from agent name + description
         },
       ];
@@ -1042,7 +1042,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'supervisor',
           to: 'writer',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
       ];
 
@@ -1075,7 +1075,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'supervisor',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Custom handoff description that takes priority',
         },
       ];
@@ -1118,9 +1118,9 @@ describe('Agent Handoffs Tests', () => {
       ];
 
       const edges: t.GraphEdge[] = [
-        { from: 'router', to: 'sales', edgeType: 'handoff' },
-        { from: 'router', to: 'support', edgeType: 'handoff' },
-        { from: 'router', to: 'billing', edgeType: 'handoff' },
+        { from: 'router', to: 'sales', edgeType: EdgeType.HANDOFF },
+        { from: 'router', to: 'support', edgeType: EdgeType.HANDOFF },
+        { from: 'router', to: 'billing', edgeType: EdgeType.HANDOFF },
       ];
 
       const run = await Run.create(createTestConfig(agents, edges));
@@ -1176,7 +1176,7 @@ describe('Agent Handoffs Tests', () => {
       ];
 
       const edges: t.GraphEdge[] = [
-        { from: 'agent_a', to: 'agent_b', edgeType: 'handoff' },
+        { from: 'agent_a', to: 'agent_b', edgeType: EdgeType.HANDOFF },
       ];
 
       const run = await Run.create(createTestConfig(agents, edges));
@@ -1215,7 +1215,7 @@ describe('Agent Handoffs Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to B',
         },
       ];
@@ -1255,7 +1255,7 @@ describe('Agent Handoffs Tests', () => {
       ];
 
       const edges: t.GraphEdge[] = [
-        { from: 'agent_a', to: 'agent_b', edgeType: 'handoff' },
+        { from: 'agent_a', to: 'agent_b', edgeType: EdgeType.HANDOFF },
       ];
 
       const run = await Run.create(createTestConfig(agents, edges));

package/src/specs/thinking-handoff.test.ts

@@ -3,7 +3,7 @@ import { HumanMessage, ToolMessage } from '@langchain/core/messages';
 import type { ToolCall } from '@langchain/core/messages/tool';
 import type { RunnableConfig } from '@langchain/core/runnables';
 import type * as t from '@/types';
-import { Providers, Constants } from '@/common';
+import { Providers, Constants, EdgeType } from '@/common';
 import { StandardGraph } from '@/graphs/Graph';
 import { Run } from '@/run';
 
@@ -77,7 +77,7 @@ describe('Thinking-Enabled Agent Handoff Tests', () => {
         {
           from: 'supervisor',
           to: 'specialist',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to specialist for detailed analysis',
         },
       ];
@@ -166,7 +166,7 @@ describe('Thinking-Enabled Agent Handoff Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
       ];
 
@@ -223,7 +223,7 @@ describe('Thinking-Enabled Agent Handoff Tests', () => {
         {
           from: 'coordinator',
           to: 'analyst',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to analyst for deep analysis',
         },
       ];
@@ -297,7 +297,7 @@ describe('Thinking-Enabled Agent Handoff Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
       ];
 
@@ -360,7 +360,7 @@ describe('Thinking-Enabled Agent Handoff Tests', () => {
         {
           from: 'supervisor',
           to: 'bedrock_specialist',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
           description: 'Transfer to Bedrock specialist',
         },
       ];
@@ -450,12 +450,12 @@ describe('Thinking-Enabled Agent Handoff Tests', () => {
         {
           from: 'router',
           to: 'processor',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
         {
           from: 'processor',
           to: 'reviewer',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
       ];
 
@@ -538,7 +538,7 @@ describe('Thinking-Enabled Agent Handoff Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
       ];
 
@@ -603,7 +603,7 @@ describe('Thinking-Enabled Agent Handoff Tests', () => {
         {
           from: 'agent_a',
           to: 'agent_b',
-          edgeType: 'handoff',
+          edgeType: EdgeType.HANDOFF,
         },
       ];