@illuma-ai/agents 1.0.90 → 1.0.93
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/cjs/agents/AgentContext.cjs +98 -49
- package/dist/cjs/agents/AgentContext.cjs.map +1 -1
- package/dist/cjs/common/constants.cjs +25 -0
- package/dist/cjs/common/constants.cjs.map +1 -0
- package/dist/cjs/common/enum.cjs +30 -0
- package/dist/cjs/common/enum.cjs.map +1 -1
- package/dist/cjs/events.cjs +9 -4
- package/dist/cjs/events.cjs.map +1 -1
- package/dist/cjs/graphs/Graph.cjs +397 -92
- package/dist/cjs/graphs/Graph.cjs.map +1 -1
- package/dist/cjs/graphs/MultiAgentGraph.cjs +223 -92
- package/dist/cjs/graphs/MultiAgentGraph.cjs.map +1 -1
- package/dist/cjs/instrumentation.cjs +30 -14
- package/dist/cjs/instrumentation.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/index.cjs +43 -11
- package/dist/cjs/llm/anthropic/index.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/types.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/utils/message_inputs.cjs +10 -7
- package/dist/cjs/llm/anthropic/utils/message_inputs.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/utils/message_outputs.cjs +32 -0
- package/dist/cjs/llm/anthropic/utils/message_outputs.cjs.map +1 -1
- package/dist/cjs/llm/anthropic/utils/tools.cjs.map +1 -1
- package/dist/cjs/llm/bedrock/index.cjs +129 -101
- package/dist/cjs/llm/bedrock/index.cjs.map +1 -1
- package/dist/cjs/llm/bedrock/utils/message_inputs.cjs +489 -0
- package/dist/cjs/llm/bedrock/utils/message_inputs.cjs.map +1 -0
- package/dist/cjs/llm/bedrock/utils/message_outputs.cjs +176 -0
- package/dist/cjs/llm/bedrock/utils/message_outputs.cjs.map +1 -0
- package/dist/cjs/llm/fake.cjs.map +1 -1
- package/dist/cjs/llm/google/index.cjs.map +1 -1
- package/dist/cjs/llm/google/utils/common.cjs.map +1 -1
- package/dist/cjs/llm/openai/index.cjs +1 -1
- package/dist/cjs/llm/openai/index.cjs.map +1 -1
- package/dist/cjs/llm/openai/utils/index.cjs.map +1 -1
- package/dist/cjs/llm/openrouter/index.cjs +59 -5
- package/dist/cjs/llm/openrouter/index.cjs.map +1 -1
- package/dist/cjs/llm/providers.cjs.map +1 -1
- package/dist/cjs/llm/text.cjs.map +1 -1
- package/dist/cjs/llm/vertexai/index.cjs +80 -2
- package/dist/cjs/llm/vertexai/index.cjs.map +1 -1
- package/dist/cjs/main.cjs +60 -27
- package/dist/cjs/main.cjs.map +1 -1
- package/dist/cjs/messages/cache.cjs +131 -108
- package/dist/cjs/messages/cache.cjs.map +1 -1
- package/dist/cjs/messages/content.cjs.map +1 -1
- package/dist/cjs/messages/core.cjs +3 -0
- package/dist/cjs/messages/core.cjs.map +1 -1
- package/dist/cjs/messages/format.cjs +265 -47
- package/dist/cjs/messages/format.cjs.map +1 -1
- package/dist/cjs/messages/ids.cjs.map +1 -1
- package/dist/cjs/messages/prune.cjs +55 -2
- package/dist/cjs/messages/prune.cjs.map +1 -1
- package/dist/cjs/messages/summarize.cjs +170 -0
- package/dist/cjs/messages/summarize.cjs.map +1 -0
- package/dist/cjs/messages/tools.cjs.map +1 -1
- package/dist/cjs/run.cjs +87 -30
- package/dist/cjs/run.cjs.map +1 -1
- package/dist/cjs/schemas/validate.cjs.map +1 -1
- package/dist/cjs/splitStream.cjs.map +1 -1
- package/dist/cjs/stream.cjs +59 -25
- package/dist/cjs/stream.cjs.map +1 -1
- package/dist/cjs/tools/AskUser.cjs +131 -0
- package/dist/cjs/tools/AskUser.cjs.map +1 -0
- package/dist/cjs/tools/BrowserTools.cjs +11 -7
- package/dist/cjs/tools/BrowserTools.cjs.map +1 -1
- package/dist/cjs/tools/Calculator.cjs.map +1 -1
- package/dist/cjs/tools/CodeExecutor.cjs +46 -4
- package/dist/cjs/tools/CodeExecutor.cjs.map +1 -1
- package/dist/cjs/tools/ProgrammaticToolCalling.cjs +36 -53
- package/dist/cjs/tools/ProgrammaticToolCalling.cjs.map +1 -1
- package/dist/cjs/tools/StreamingToolCallBuffer.cjs +208 -0
- package/dist/cjs/tools/StreamingToolCallBuffer.cjs.map +1 -0
- package/dist/cjs/tools/ToolNode.cjs +333 -30
- package/dist/cjs/tools/ToolNode.cjs.map +1 -1
- package/dist/cjs/tools/ToolSearch.cjs +66 -30
- package/dist/cjs/tools/ToolSearch.cjs.map +1 -1
- package/dist/cjs/tools/handlers.cjs +94 -8
- package/dist/cjs/tools/handlers.cjs.map +1 -1
- package/dist/cjs/tools/schema.cjs.map +1 -1
- package/dist/cjs/tools/search/content.cjs.map +1 -1
- package/dist/cjs/tools/search/firecrawl.cjs.map +1 -1
- package/dist/cjs/tools/search/format.cjs.map +1 -1
- package/dist/cjs/tools/search/highlights.cjs.map +1 -1
- package/dist/cjs/tools/search/rerankers.cjs.map +1 -1
- package/dist/cjs/tools/search/schema.cjs.map +1 -1
- package/dist/cjs/tools/search/search.cjs +1 -0
- package/dist/cjs/tools/search/search.cjs.map +1 -1
- package/dist/cjs/tools/search/serper-scraper.cjs.map +1 -1
- package/dist/cjs/tools/search/tool.cjs.map +1 -1
- package/dist/cjs/tools/search/utils.cjs.map +1 -1
- package/dist/cjs/types/graph.cjs +1 -1
- package/dist/cjs/types/graph.cjs.map +1 -1
- package/dist/cjs/utils/contextAnalytics.cjs +23 -6
- package/dist/cjs/utils/contextAnalytics.cjs.map +1 -1
- package/dist/cjs/utils/events.cjs.map +1 -1
- package/dist/cjs/utils/graph.cjs.map +1 -1
- package/dist/cjs/utils/handlers.cjs.map +1 -1
- package/dist/cjs/utils/llm.cjs.map +1 -1
- package/dist/cjs/utils/misc.cjs.map +1 -1
- package/dist/cjs/utils/run.cjs +3 -1
- package/dist/cjs/utils/run.cjs.map +1 -1
- package/dist/cjs/utils/schema.cjs.map +1 -1
- package/dist/cjs/utils/title.cjs.map +1 -1
- package/dist/cjs/utils/tokens.cjs +33 -58
- package/dist/cjs/utils/tokens.cjs.map +1 -1
- package/dist/cjs/utils/toolCallContinuation.cjs +55 -0
- package/dist/cjs/utils/toolCallContinuation.cjs.map +1 -0
- package/dist/cjs/utils/toonFormat.cjs.map +1 -1
- package/dist/esm/agents/AgentContext.mjs +98 -49
- package/dist/esm/agents/AgentContext.mjs.map +1 -1
- package/dist/esm/common/constants.mjs +22 -0
- package/dist/esm/common/constants.mjs.map +1 -0
- package/dist/esm/common/enum.mjs +31 -1
- package/dist/esm/common/enum.mjs.map +1 -1
- package/dist/esm/events.mjs +9 -4
- package/dist/esm/events.mjs.map +1 -1
- package/dist/esm/graphs/Graph.mjs +393 -88
- package/dist/esm/graphs/Graph.mjs.map +1 -1
- package/dist/esm/graphs/MultiAgentGraph.mjs +224 -93
- package/dist/esm/graphs/MultiAgentGraph.mjs.map +1 -1
- package/dist/esm/instrumentation.mjs +30 -14
- package/dist/esm/instrumentation.mjs.map +1 -1
- package/dist/esm/llm/anthropic/index.mjs +43 -11
- package/dist/esm/llm/anthropic/index.mjs.map +1 -1
- package/dist/esm/llm/anthropic/types.mjs.map +1 -1
- package/dist/esm/llm/anthropic/utils/message_inputs.mjs +10 -7
- package/dist/esm/llm/anthropic/utils/message_inputs.mjs.map +1 -1
- package/dist/esm/llm/anthropic/utils/message_outputs.mjs +32 -0
- package/dist/esm/llm/anthropic/utils/message_outputs.mjs.map +1 -1
- package/dist/esm/llm/anthropic/utils/tools.mjs.map +1 -1
- package/dist/esm/llm/bedrock/index.mjs +128 -101
- package/dist/esm/llm/bedrock/index.mjs.map +1 -1
- package/dist/esm/llm/bedrock/utils/message_inputs.mjs +484 -0
- package/dist/esm/llm/bedrock/utils/message_inputs.mjs.map +1 -0
- package/dist/esm/llm/bedrock/utils/message_outputs.mjs +171 -0
- package/dist/esm/llm/bedrock/utils/message_outputs.mjs.map +1 -0
- package/dist/esm/llm/fake.mjs.map +1 -1
- package/dist/esm/llm/google/index.mjs.map +1 -1
- package/dist/esm/llm/google/utils/common.mjs.map +1 -1
- package/dist/esm/llm/openai/index.mjs +1 -1
- package/dist/esm/llm/openai/index.mjs.map +1 -1
- package/dist/esm/llm/openai/utils/index.mjs.map +1 -1
- package/dist/esm/llm/openrouter/index.mjs +59 -5
- package/dist/esm/llm/openrouter/index.mjs.map +1 -1
- package/dist/esm/llm/providers.mjs.map +1 -1
- package/dist/esm/llm/text.mjs.map +1 -1
- package/dist/esm/llm/vertexai/index.mjs +80 -2
- package/dist/esm/llm/vertexai/index.mjs.map +1 -1
- package/dist/esm/main.mjs +8 -3
- package/dist/esm/main.mjs.map +1 -1
- package/dist/esm/messages/cache.mjs +131 -108
- package/dist/esm/messages/cache.mjs.map +1 -1
- package/dist/esm/messages/content.mjs.map +1 -1
- package/dist/esm/messages/core.mjs +4 -1
- package/dist/esm/messages/core.mjs.map +1 -1
- package/dist/esm/messages/format.mjs +267 -49
- package/dist/esm/messages/format.mjs.map +1 -1
- package/dist/esm/messages/ids.mjs.map +1 -1
- package/dist/esm/messages/prune.mjs +56 -4
- package/dist/esm/messages/prune.mjs.map +1 -1
- package/dist/esm/messages/summarize.mjs +161 -0
- package/dist/esm/messages/summarize.mjs.map +1 -0
- package/dist/esm/messages/tools.mjs.map +1 -1
- package/dist/esm/run.mjs +88 -31
- package/dist/esm/run.mjs.map +1 -1
- package/dist/esm/schemas/validate.mjs.map +1 -1
- package/dist/esm/splitStream.mjs.map +1 -1
- package/dist/esm/stream.mjs +60 -26
- package/dist/esm/stream.mjs.map +1 -1
- package/dist/esm/tools/AskUser.mjs +125 -0
- package/dist/esm/tools/AskUser.mjs.map +1 -0
- package/dist/esm/tools/BrowserTools.mjs +11 -7
- package/dist/esm/tools/BrowserTools.mjs.map +1 -1
- package/dist/esm/tools/Calculator.mjs.map +1 -1
- package/dist/esm/tools/CodeExecutor.mjs +46 -4
- package/dist/esm/tools/CodeExecutor.mjs.map +1 -1
- package/dist/esm/tools/ProgrammaticToolCalling.mjs +37 -54
- package/dist/esm/tools/ProgrammaticToolCalling.mjs.map +1 -1
- package/dist/esm/tools/StreamingToolCallBuffer.mjs +206 -0
- package/dist/esm/tools/StreamingToolCallBuffer.mjs.map +1 -0
- package/dist/esm/tools/ToolNode.mjs +333 -30
- package/dist/esm/tools/ToolNode.mjs.map +1 -1
- package/dist/esm/tools/ToolSearch.mjs +66 -30
- package/dist/esm/tools/ToolSearch.mjs.map +1 -1
- package/dist/esm/tools/handlers.mjs +95 -9
- package/dist/esm/tools/handlers.mjs.map +1 -1
- package/dist/esm/tools/schema.mjs.map +1 -1
- package/dist/esm/tools/search/content.mjs.map +1 -1
- package/dist/esm/tools/search/firecrawl.mjs.map +1 -1
- package/dist/esm/tools/search/format.mjs.map +1 -1
- package/dist/esm/tools/search/highlights.mjs.map +1 -1
- package/dist/esm/tools/search/rerankers.mjs.map +1 -1
- package/dist/esm/tools/search/schema.mjs.map +1 -1
- package/dist/esm/tools/search/search.mjs +1 -0
- package/dist/esm/tools/search/search.mjs.map +1 -1
- package/dist/esm/tools/search/serper-scraper.mjs.map +1 -1
- package/dist/esm/tools/search/tool.mjs.map +1 -1
- package/dist/esm/tools/search/utils.mjs.map +1 -1
- package/dist/esm/types/graph.mjs +1 -1
- package/dist/esm/types/graph.mjs.map +1 -1
- package/dist/esm/utils/contextAnalytics.mjs +23 -6
- package/dist/esm/utils/contextAnalytics.mjs.map +1 -1
- package/dist/esm/utils/events.mjs.map +1 -1
- package/dist/esm/utils/graph.mjs.map +1 -1
- package/dist/esm/utils/handlers.mjs.map +1 -1
- package/dist/esm/utils/llm.mjs.map +1 -1
- package/dist/esm/utils/misc.mjs.map +1 -1
- package/dist/esm/utils/run.mjs +3 -1
- package/dist/esm/utils/run.mjs.map +1 -1
- package/dist/esm/utils/schema.mjs.map +1 -1
- package/dist/esm/utils/title.mjs.map +1 -1
- package/dist/esm/utils/tokens.mjs +33 -59
- package/dist/esm/utils/tokens.mjs.map +1 -1
- package/dist/esm/utils/toolCallContinuation.mjs +52 -0
- package/dist/esm/utils/toolCallContinuation.mjs.map +1 -0
- package/dist/esm/utils/toonFormat.mjs.map +1 -1
- package/dist/types/agents/AgentContext.d.ts +14 -7
- package/dist/types/common/constants.d.ts +18 -0
- package/dist/types/common/enum.d.ts +28 -0
- package/dist/types/common/index.d.ts +1 -0
- package/dist/types/events.d.ts +10 -3
- package/dist/types/graphs/Graph.d.ts +37 -0
- package/dist/types/index.d.ts +4 -0
- package/dist/types/llm/anthropic/index.d.ts +7 -1
- package/dist/types/llm/anthropic/types.d.ts +5 -2
- package/dist/types/llm/anthropic/utils/message_outputs.d.ts +1 -1
- package/dist/types/llm/bedrock/index.d.ts +40 -33
- package/dist/types/llm/bedrock/utils/message_outputs.d.ts +1 -1
- package/dist/types/llm/google/index.d.ts +2 -3
- package/dist/types/llm/openrouter/index.d.ts +21 -1
- package/dist/types/llm/vertexai/index.d.ts +3 -2
- package/dist/types/messages/cache.d.ts +1 -1
- package/dist/types/messages/index.d.ts +1 -0
- package/dist/types/messages/prune.d.ts +2 -7
- package/dist/types/messages/summarize.d.ts +33 -0
- package/dist/types/run.d.ts +6 -0
- package/dist/types/tools/AskUser.d.ts +408 -0
- package/dist/types/tools/BrowserTools.d.ts +2 -2
- package/dist/types/tools/CodeExecutor.d.ts +2 -2
- package/dist/types/tools/StreamingToolCallBuffer.d.ts +106 -0
- package/dist/types/tools/ToolNode.d.ts +55 -3
- package/dist/types/tools/ToolSearch.d.ts +9 -5
- package/dist/types/tools/handlers.d.ts +2 -2
- package/dist/types/types/graph.d.ts +9 -2
- package/dist/types/types/llm.d.ts +8 -3
- package/dist/types/types/run.d.ts +2 -0
- package/dist/types/types/tools.d.ts +20 -2
- package/dist/types/utils/contextAnalytics.d.ts +5 -4
- package/dist/types/utils/index.d.ts +1 -0
- package/dist/types/utils/tokens.d.ts +6 -19
- package/dist/types/utils/toolCallContinuation.d.ts +30 -0
- package/package.json +15 -8
- package/src/agents/AgentContext.js +782 -0
- package/src/agents/AgentContext.js.map +1 -0
- package/src/agents/AgentContext.test.js +421 -0
- package/src/agents/AgentContext.test.js.map +1 -0
- package/src/agents/AgentContext.ts +132 -64
- package/src/agents/__tests__/AgentContext.test.js +678 -0
- package/src/agents/__tests__/AgentContext.test.js.map +1 -0
- package/src/agents/__tests__/AgentContext.test.ts +25 -4
- package/src/agents/__tests__/resolveStructuredOutputMode.test.js +117 -0
- package/src/agents/__tests__/resolveStructuredOutputMode.test.js.map +1 -0
- package/src/common/__tests__/enum.test.ts +135 -0
- package/src/common/constants.ts +21 -0
- package/src/common/enum.js +192 -0
- package/src/common/enum.js.map +1 -0
- package/src/common/enum.ts +30 -0
- package/src/common/index.js +3 -0
- package/src/common/index.js.map +1 -0
- package/src/common/index.ts +2 -1
- package/src/events.js +166 -0
- package/src/events.js.map +1 -0
- package/src/events.ts +11 -14
- package/src/graphs/Graph.js +1857 -0
- package/src/graphs/Graph.js.map +1 -0
- package/src/graphs/Graph.ts +580 -162
- package/src/graphs/MultiAgentGraph.js +1092 -0
- package/src/graphs/MultiAgentGraph.js.map +1 -0
- package/src/graphs/MultiAgentGraph.ts +331 -112
- package/src/graphs/__tests__/adaptive-thinking.test.ts +369 -0
- package/src/graphs/__tests__/graph-direct-tool-names.test.ts +210 -0
- package/src/graphs/__tests__/multi-agent-edges.test.ts +237 -0
- package/src/graphs/__tests__/structured-output.integration.test.js +624 -0
- package/src/graphs/__tests__/structured-output.integration.test.js.map +1 -0
- package/src/graphs/__tests__/structured-output.test.js +144 -0
- package/src/graphs/__tests__/structured-output.test.js.map +1 -0
- package/src/graphs/contextManagement.e2e.test.js +718 -0
- package/src/graphs/contextManagement.e2e.test.js.map +1 -0
- package/src/graphs/contextManagement.e2e.test.ts +990 -0
- package/src/graphs/contextManagement.test.js +485 -0
- package/src/graphs/contextManagement.test.js.map +1 -0
- package/src/graphs/contextManagement.test.ts +625 -0
- package/src/graphs/handoffValidation.test.js +276 -0
- package/src/graphs/handoffValidation.test.js.map +1 -0
- package/src/graphs/handoffValidation.test.ts +353 -0
- package/src/graphs/index.js +3 -0
- package/src/graphs/index.js.map +1 -0
- package/src/index.js +28 -0
- package/src/index.js.map +1 -0
- package/src/index.ts +13 -0
- package/src/instrumentation.js +21 -0
- package/src/instrumentation.js.map +1 -0
- package/src/instrumentation.ts +38 -17
- package/src/llm/anthropic/index.js +319 -0
- package/src/llm/anthropic/index.js.map +1 -0
- package/src/llm/anthropic/index.ts +68 -15
- package/src/llm/anthropic/llm.spec.ts +402 -0
- package/src/llm/anthropic/types.js +46 -0
- package/src/llm/anthropic/types.js.map +1 -0
- package/src/llm/anthropic/types.ts +8 -2
- package/src/llm/anthropic/utils/message_inputs.js +627 -0
- package/src/llm/anthropic/utils/message_inputs.js.map +1 -0
- package/src/llm/anthropic/utils/message_inputs.ts +16 -33
- package/src/llm/anthropic/utils/message_outputs.js +290 -0
- package/src/llm/anthropic/utils/message_outputs.js.map +1 -0
- package/src/llm/anthropic/utils/message_outputs.ts +40 -1
- package/src/llm/anthropic/utils/output_parsers.js +89 -0
- package/src/llm/anthropic/utils/output_parsers.js.map +1 -0
- package/src/llm/anthropic/utils/tools.js +25 -0
- package/src/llm/anthropic/utils/tools.js.map +1 -0
- package/src/llm/bedrock/__tests__/bedrock-caching.test.js +392 -0
- package/src/llm/bedrock/__tests__/bedrock-caching.test.js.map +1 -0
- package/src/llm/bedrock/__tests__/bedrock-caching.test.ts +24 -40
- package/src/llm/bedrock/index.js +303 -0
- package/src/llm/bedrock/index.js.map +1 -0
- package/src/llm/bedrock/index.ts +171 -134
- package/src/llm/bedrock/llm.spec.ts +395 -52
- package/src/llm/bedrock/types.js +2 -0
- package/src/llm/bedrock/types.js.map +1 -0
- package/src/llm/bedrock/utils/index.js +6 -0
- package/src/llm/bedrock/utils/index.js.map +1 -0
- package/src/llm/bedrock/utils/message_inputs.js +463 -0
- package/src/llm/bedrock/utils/message_inputs.js.map +1 -0
- package/src/llm/bedrock/utils/message_inputs.ts +30 -5
- package/src/llm/bedrock/utils/message_outputs.js +269 -0
- package/src/llm/bedrock/utils/message_outputs.js.map +1 -0
- package/src/llm/bedrock/utils/message_outputs.ts +70 -22
- package/src/llm/fake.js +92 -0
- package/src/llm/fake.js.map +1 -0
- package/src/llm/google/index.js +215 -0
- package/src/llm/google/index.js.map +1 -0
- package/src/llm/google/index.ts +2 -3
- package/src/llm/google/types.js +12 -0
- package/src/llm/google/types.js.map +1 -0
- package/src/llm/google/utils/common.js +670 -0
- package/src/llm/google/utils/common.js.map +1 -0
- package/src/llm/google/utils/tools.js +111 -0
- package/src/llm/google/utils/tools.js.map +1 -0
- package/src/llm/google/utils/zod_to_genai_parameters.js +47 -0
- package/src/llm/google/utils/zod_to_genai_parameters.js.map +1 -0
- package/src/llm/openai/index.js +1033 -0
- package/src/llm/openai/index.js.map +1 -0
- package/src/llm/openai/types.js +2 -0
- package/src/llm/openai/types.js.map +1 -0
- package/src/llm/openai/utils/index.js +756 -0
- package/src/llm/openai/utils/index.js.map +1 -0
- package/src/llm/openai/utils/isReasoningModel.test.js +79 -0
- package/src/llm/openai/utils/isReasoningModel.test.js.map +1 -0
- package/src/llm/openrouter/index.js +261 -0
- package/src/llm/openrouter/index.js.map +1 -0
- package/src/llm/openrouter/index.ts +117 -6
- package/src/llm/openrouter/reasoning.test.js +181 -0
- package/src/llm/openrouter/reasoning.test.js.map +1 -0
- package/src/llm/openrouter/reasoning.test.ts +207 -0
- package/src/llm/providers.js +36 -0
- package/src/llm/providers.js.map +1 -0
- package/src/llm/text.js +65 -0
- package/src/llm/text.js.map +1 -0
- package/src/llm/vertexai/index.js +402 -0
- package/src/llm/vertexai/index.js.map +1 -0
- package/src/llm/vertexai/index.ts +115 -5
- package/src/llm/vertexai/llm.spec.ts +114 -0
- package/src/messages/__tests__/tools.test.js +392 -0
- package/src/messages/__tests__/tools.test.js.map +1 -0
- package/src/messages/cache.js +404 -0
- package/src/messages/cache.js.map +1 -0
- package/src/messages/cache.test.js +1167 -0
- package/src/messages/cache.test.js.map +1 -0
- package/src/messages/cache.test.ts +178 -16
- package/src/messages/cache.ts +152 -147
- package/src/messages/content.js +48 -0
- package/src/messages/content.js.map +1 -0
- package/src/messages/content.test.js +314 -0
- package/src/messages/content.test.js.map +1 -0
- package/src/messages/core.js +359 -0
- package/src/messages/core.js.map +1 -0
- package/src/messages/core.ts +5 -0
- package/src/messages/ensureThinkingBlock.test.js +997 -0
- package/src/messages/ensureThinkingBlock.test.js.map +1 -0
- package/src/messages/ensureThinkingBlock.test.ts +751 -10
- package/src/messages/format.js +973 -0
- package/src/messages/format.js.map +1 -0
- package/src/messages/format.ts +334 -57
- package/src/messages/formatAgentMessages.test.js +2278 -0
- package/src/messages/formatAgentMessages.test.js.map +1 -0
- package/src/messages/formatAgentMessages.test.ts +1175 -1
- package/src/messages/formatAgentMessages.tools.test.js +362 -0
- package/src/messages/formatAgentMessages.tools.test.js.map +1 -0
- package/src/messages/formatMessage.test.js +608 -0
- package/src/messages/formatMessage.test.js.map +1 -0
- package/src/messages/ids.js +18 -0
- package/src/messages/ids.js.map +1 -0
- package/src/messages/index.js +9 -0
- package/src/messages/index.js.map +1 -0
- package/src/messages/index.ts +1 -0
- package/src/messages/labelContentByAgent.test.js +725 -0
- package/src/messages/labelContentByAgent.test.js.map +1 -0
- package/src/messages/prune.js +438 -0
- package/src/messages/prune.js.map +1 -0
- package/src/messages/prune.ts +87 -25
- package/src/messages/reducer.js +60 -0
- package/src/messages/reducer.js.map +1 -0
- package/src/messages/shiftIndexTokenCountMap.test.js +63 -0
- package/src/messages/shiftIndexTokenCountMap.test.js.map +1 -0
- package/src/messages/summarize.js +146 -0
- package/src/messages/summarize.js.map +1 -0
- package/src/messages/summarize.test.js +332 -0
- package/src/messages/summarize.test.js.map +1 -0
- package/src/messages/summarize.test.ts +466 -0
- package/src/messages/summarize.ts +222 -0
- package/src/messages/tools.js +90 -0
- package/src/messages/tools.js.map +1 -0
- package/src/mockStream.js +81 -0
- package/src/mockStream.js.map +1 -0
- package/src/prompts/collab.js +7 -0
- package/src/prompts/collab.js.map +1 -0
- package/src/prompts/index.js +3 -0
- package/src/prompts/index.js.map +1 -0
- package/src/prompts/taskmanager.js +58 -0
- package/src/prompts/taskmanager.js.map +1 -0
- package/src/run.js +427 -0
- package/src/run.js.map +1 -0
- package/src/run.ts +101 -33
- package/src/schemas/index.js +3 -0
- package/src/schemas/index.js.map +1 -0
- package/src/schemas/schema-preparation.test.js +370 -0
- package/src/schemas/schema-preparation.test.js.map +1 -0
- package/src/schemas/validate.js +314 -0
- package/src/schemas/validate.js.map +1 -0
- package/src/schemas/validate.test.js +264 -0
- package/src/schemas/validate.test.js.map +1 -0
- package/src/scripts/abort.js +127 -0
- package/src/scripts/abort.js.map +1 -0
- package/src/scripts/ant_web_search.js +130 -0
- package/src/scripts/ant_web_search.js.map +1 -0
- package/src/scripts/ant_web_search.ts +1 -0
- package/src/scripts/ant_web_search_edge_case.js +133 -0
- package/src/scripts/ant_web_search_edge_case.js.map +1 -0
- package/src/scripts/ant_web_search_edge_case.ts +1 -0
- package/src/scripts/ant_web_search_error_edge_case.js +119 -0
- package/src/scripts/ant_web_search_error_edge_case.js.map +1 -0
- package/src/scripts/ant_web_search_error_edge_case.ts +1 -0
- package/src/scripts/args.js +41 -0
- package/src/scripts/args.js.map +1 -0
- package/src/scripts/bedrock-cache-debug.js +186 -0
- package/src/scripts/bedrock-cache-debug.js.map +1 -0
- package/src/scripts/bedrock-cache-debug.ts +250 -0
- package/src/scripts/bedrock-content-aggregation-test.js +195 -0
- package/src/scripts/bedrock-content-aggregation-test.js.map +1 -0
- package/src/scripts/bedrock-content-aggregation-test.ts +266 -0
- package/src/scripts/bedrock-merge-test.js +80 -0
- package/src/scripts/bedrock-merge-test.js.map +1 -0
- package/src/scripts/bedrock-merge-test.ts +107 -0
- package/src/scripts/bedrock-parallel-tools-test.js +150 -0
- package/src/scripts/bedrock-parallel-tools-test.js.map +1 -0
- package/src/scripts/bedrock-parallel-tools-test.ts +204 -0
- package/src/scripts/caching.js +106 -0
- package/src/scripts/caching.js.map +1 -0
- package/src/scripts/caching.ts +1 -0
- package/src/scripts/cli.js +152 -0
- package/src/scripts/cli.js.map +1 -0
- package/src/scripts/cli2.js +119 -0
- package/src/scripts/cli2.js.map +1 -0
- package/src/scripts/cli3.js +163 -0
- package/src/scripts/cli3.js.map +1 -0
- package/src/scripts/cli4.js +165 -0
- package/src/scripts/cli4.js.map +1 -0
- package/src/scripts/cli5.js +165 -0
- package/src/scripts/cli5.js.map +1 -0
- package/src/scripts/code_exec.js +171 -0
- package/src/scripts/code_exec.js.map +1 -0
- package/src/scripts/code_exec.ts +1 -0
- package/src/scripts/code_exec_files.js +180 -0
- package/src/scripts/code_exec_files.js.map +1 -0
- package/src/scripts/code_exec_files.ts +1 -0
- package/src/scripts/code_exec_multi_session.js +185 -0
- package/src/scripts/code_exec_multi_session.js.map +1 -0
- package/src/scripts/code_exec_multi_session.ts +9 -13
- package/src/scripts/code_exec_ptc.js +265 -0
- package/src/scripts/code_exec_ptc.js.map +1 -0
- package/src/scripts/code_exec_ptc.ts +1 -0
- package/src/scripts/code_exec_session.js +217 -0
- package/src/scripts/code_exec_session.js.map +1 -0
- package/src/scripts/code_exec_session.ts +1 -0
- package/src/scripts/code_exec_simple.js +120 -0
- package/src/scripts/code_exec_simple.js.map +1 -0
- package/src/scripts/code_exec_simple.ts +1 -0
- package/src/scripts/content.js +111 -0
- package/src/scripts/content.js.map +1 -0
- package/src/scripts/content.ts +1 -0
- package/src/scripts/empty_input.js +125 -0
- package/src/scripts/empty_input.js.map +1 -0
- package/src/scripts/handoff-test.js +96 -0
- package/src/scripts/handoff-test.js.map +1 -0
- package/src/scripts/image.js +138 -0
- package/src/scripts/image.js.map +1 -0
- package/src/scripts/image.ts +3 -1
- package/src/scripts/memory.js +83 -0
- package/src/scripts/memory.js.map +1 -0
- package/src/scripts/memory.ts +16 -6
- package/src/scripts/multi-agent-chain.js +271 -0
- package/src/scripts/multi-agent-chain.js.map +1 -0
- package/src/scripts/multi-agent-chain.ts +1 -0
- package/src/scripts/multi-agent-conditional.js +185 -0
- package/src/scripts/multi-agent-conditional.js.map +1 -0
- package/src/scripts/multi-agent-conditional.ts +1 -0
- package/src/scripts/multi-agent-document-review-chain.js +171 -0
- package/src/scripts/multi-agent-document-review-chain.js.map +1 -0
- package/src/scripts/multi-agent-document-review-chain.ts +1 -0
- package/src/scripts/multi-agent-hybrid-flow.js +264 -0
- package/src/scripts/multi-agent-hybrid-flow.js.map +1 -0
- package/src/scripts/multi-agent-hybrid-flow.ts +1 -0
- package/src/scripts/multi-agent-parallel-start.js +214 -0
- package/src/scripts/multi-agent-parallel-start.js.map +1 -0
- package/src/scripts/multi-agent-parallel-start.ts +4 -4
- package/src/scripts/multi-agent-parallel.js +346 -0
- package/src/scripts/multi-agent-parallel.js.map +1 -0
- package/src/scripts/multi-agent-parallel.ts +1 -0
- package/src/scripts/multi-agent-sequence.js +184 -0
- package/src/scripts/multi-agent-sequence.js.map +1 -0
- package/src/scripts/multi-agent-sequence.ts +4 -4
- package/src/scripts/multi-agent-supervisor.js +324 -0
- package/src/scripts/multi-agent-supervisor.js.map +1 -0
- package/src/scripts/multi-agent-supervisor.ts +1 -0
- package/src/scripts/multi-agent-test.js +147 -0
- package/src/scripts/multi-agent-test.js.map +1 -0
- package/src/scripts/multi-agent-test.ts +1 -0
- package/src/scripts/parallel-asymmetric-tools-test.js +202 -0
- package/src/scripts/parallel-asymmetric-tools-test.js.map +1 -0
- package/src/scripts/parallel-asymmetric-tools-test.ts +1 -0
- package/src/scripts/parallel-full-metadata-test.js +176 -0
- package/src/scripts/parallel-full-metadata-test.js.map +1 -0
- package/src/scripts/parallel-full-metadata-test.ts +1 -0
- package/src/scripts/parallel-tools-test.js +256 -0
- package/src/scripts/parallel-tools-test.js.map +1 -0
- package/src/scripts/parallel-tools-test.ts +1 -0
- package/src/scripts/poc-multi-agent-comprehensive.ts +1222 -0
- package/src/scripts/programmatic_exec.js +277 -0
- package/src/scripts/programmatic_exec.js.map +1 -0
- package/src/scripts/programmatic_exec_agent.js +168 -0
- package/src/scripts/programmatic_exec_agent.js.map +1 -0
- package/src/scripts/programmatic_exec_agent.ts +1 -0
- package/src/scripts/search.js +118 -0
- package/src/scripts/search.js.map +1 -0
- package/src/scripts/search.ts +1 -0
- package/src/scripts/sequential-full-metadata-test.js +143 -0
- package/src/scripts/sequential-full-metadata-test.js.map +1 -0
- package/src/scripts/sequential-full-metadata-test.ts +1 -0
- package/src/scripts/simple.js +174 -0
- package/src/scripts/simple.js.map +1 -0
- package/src/scripts/simple.ts +2 -1
- package/src/scripts/single-agent-metadata-test.js +152 -0
- package/src/scripts/single-agent-metadata-test.js.map +1 -0
- package/src/scripts/single-agent-metadata-test.ts +4 -6
- package/src/scripts/stream.js +113 -0
- package/src/scripts/stream.js.map +1 -0
- package/src/scripts/stream.ts +1 -0
- package/src/scripts/test-custom-prompt-key.js +132 -0
- package/src/scripts/test-custom-prompt-key.js.map +1 -0
- package/src/scripts/test-handoff-input.js +143 -0
- package/src/scripts/test-handoff-input.js.map +1 -0
- package/src/scripts/test-handoff-preamble.js +227 -0
- package/src/scripts/test-handoff-preamble.js.map +1 -0
- package/src/scripts/test-handoff-preamble.ts +1 -0
- package/src/scripts/test-handoff-steering.js +353 -0
- package/src/scripts/test-handoff-steering.js.map +1 -0
- package/src/scripts/test-handoff-steering.ts +430 -0
- package/src/scripts/test-multi-agent-list-handoff.js +318 -0
- package/src/scripts/test-multi-agent-list-handoff.js.map +1 -0
- package/src/scripts/test-multi-agent-list-handoff.ts +1 -0
- package/src/scripts/test-parallel-agent-labeling.js +253 -0
- package/src/scripts/test-parallel-agent-labeling.js.map +1 -0
- package/src/scripts/test-parallel-agent-labeling.ts +2 -0
- package/src/scripts/test-parallel-handoffs.js +229 -0
- package/src/scripts/test-parallel-handoffs.js.map +1 -0
- package/src/scripts/test-parallel-handoffs.ts +1 -0
- package/src/scripts/test-thinking-handoff-bedrock.js +132 -0
- package/src/scripts/test-thinking-handoff-bedrock.js.map +1 -0
- package/src/scripts/test-thinking-handoff-bedrock.ts +1 -0
- package/src/scripts/test-thinking-handoff.js +132 -0
- package/src/scripts/test-thinking-handoff.js.map +1 -0
- package/src/scripts/test-thinking-handoff.ts +1 -0
- package/src/scripts/test-thinking-to-thinking-handoff-bedrock.js +140 -0
- package/src/scripts/test-thinking-to-thinking-handoff-bedrock.js.map +1 -0
- package/src/scripts/test-thinking-to-thinking-handoff-bedrock.ts +166 -0
- package/src/scripts/test-tool-before-handoff-role-order.js +223 -0
- package/src/scripts/test-tool-before-handoff-role-order.js.map +1 -0
- package/src/scripts/test-tool-before-handoff-role-order.ts +276 -0
- package/src/scripts/test-tools-before-handoff.js +187 -0
- package/src/scripts/test-tools-before-handoff.js.map +1 -0
- package/src/scripts/test-tools-before-handoff.ts +4 -8
- package/src/scripts/test_code_api.js +263 -0
- package/src/scripts/test_code_api.js.map +1 -0
- package/src/scripts/thinking-bedrock.js +128 -0
- package/src/scripts/thinking-bedrock.js.map +1 -0
- package/src/scripts/thinking-bedrock.ts +1 -0
- package/src/scripts/thinking-vertexai.js +130 -0
- package/src/scripts/thinking-vertexai.js.map +1 -0
- package/src/scripts/thinking-vertexai.ts +168 -0
- package/src/scripts/thinking.js +134 -0
- package/src/scripts/thinking.js.map +1 -0
- package/src/scripts/thinking.ts +1 -0
- package/src/scripts/tool_search.js +114 -0
- package/src/scripts/tool_search.js.map +1 -0
- package/src/scripts/tools.js +125 -0
- package/src/scripts/tools.js.map +1 -0
- package/src/scripts/tools.ts +5 -19
- package/src/specs/agent-handoffs-bedrock.integration.test.js +280 -0
- package/src/specs/agent-handoffs-bedrock.integration.test.js.map +1 -0
- package/src/specs/agent-handoffs-bedrock.integration.test.ts +412 -375
- package/src/specs/agent-handoffs.test.js +924 -0
- package/src/specs/agent-handoffs.test.js.map +1 -0
- package/src/specs/agent-handoffs.test.ts +152 -39
- package/src/specs/anthropic.simple.test.js +287 -0
- package/src/specs/anthropic.simple.test.js.map +1 -0
- package/src/specs/anthropic.simple.test.ts +7 -4
- package/src/specs/azure.simple.test.js +381 -0
- package/src/specs/azure.simple.test.js.map +1 -0
- package/src/specs/azure.simple.test.ts +143 -5
- package/src/specs/cache.simple.test.js +282 -0
- package/src/specs/cache.simple.test.js.map +1 -0
- package/src/specs/cache.simple.test.ts +9 -2
- package/src/specs/custom-event-await.test.js +148 -0
- package/src/specs/custom-event-await.test.js.map +1 -0
- package/src/specs/custom-event-await.test.ts +215 -0
- package/src/specs/deepseek.simple.test.js +189 -0
- package/src/specs/deepseek.simple.test.js.map +1 -0
- package/src/specs/deepseek.simple.test.ts +4 -2
- package/src/specs/emergency-prune.test.js +308 -0
- package/src/specs/emergency-prune.test.js.map +1 -0
- package/src/specs/moonshot.simple.test.js +237 -0
- package/src/specs/moonshot.simple.test.js.map +1 -0
- package/src/specs/moonshot.simple.test.ts +6 -2
- package/src/specs/observability.integration.test.js +1337 -0
- package/src/specs/observability.integration.test.js.map +1 -0
- package/src/specs/observability.integration.test.ts +2223 -0
- package/src/specs/openai.simple.test.js +233 -0
- package/src/specs/openai.simple.test.js.map +1 -0
- package/src/specs/openai.simple.test.ts +4 -2
- package/src/specs/openrouter.simple.test.js +202 -0
- package/src/specs/openrouter.simple.test.js.map +1 -0
- package/src/specs/openrouter.simple.test.ts +165 -4
- package/src/specs/prune.test.js +733 -0
- package/src/specs/prune.test.js.map +1 -0
- package/src/specs/prune.test.ts +1 -0
- package/src/specs/reasoning.test.js +144 -0
- package/src/specs/reasoning.test.js.map +1 -0
- package/src/specs/reasoning.test.ts +2 -2
- package/src/specs/spec.utils.js +4 -0
- package/src/specs/spec.utils.js.map +1 -0
- package/src/specs/thinking-handoff.test.js +486 -0
- package/src/specs/thinking-handoff.test.js.map +1 -0
- package/src/specs/thinking-handoff.test.ts +3 -2
- package/src/specs/thinking-prune.test.js +600 -0
- package/src/specs/thinking-prune.test.js.map +1 -0
- package/src/specs/token-distribution-edge-case.test.js +246 -0
- package/src/specs/token-distribution-edge-case.test.js.map +1 -0
- package/src/specs/token-memoization.test.js +32 -0
- package/src/specs/token-memoization.test.js.map +1 -0
- package/src/specs/token-memoization.test.ts +14 -5
- package/src/specs/tokens.test.js +49 -0
- package/src/specs/tokens.test.js.map +1 -0
- package/src/specs/tokens.test.ts +64 -0
- package/src/specs/tool-error.test.js +139 -0
- package/src/specs/tool-error.test.js.map +1 -0
- package/src/specs/tool-error.test.ts +2 -2
- package/src/splitStream.js +204 -0
- package/src/splitStream.js.map +1 -0
- package/src/splitStream.test.js +504 -0
- package/src/splitStream.test.js.map +1 -0
- package/src/stream.js +650 -0
- package/src/stream.js.map +1 -0
- package/src/stream.test.js +225 -0
- package/src/stream.test.js.map +1 -0
- package/src/stream.test.ts +25 -15
- package/src/stream.ts +82 -32
- package/src/test/mockTools.js +340 -0
- package/src/test/mockTools.js.map +1 -0
- package/src/tools/AskUser.ts +159 -0
- package/src/tools/BrowserTools.js +245 -0
- package/src/tools/BrowserTools.js.map +1 -0
- package/src/tools/BrowserTools.ts +12 -8
- package/src/tools/Calculator.js +38 -0
- package/src/tools/Calculator.js.map +1 -0
- package/src/tools/Calculator.test.js +225 -0
- package/src/tools/Calculator.test.js.map +1 -0
- package/src/tools/CodeExecutor.js +233 -0
- package/src/tools/CodeExecutor.js.map +1 -0
- package/src/tools/CodeExecutor.selfhealing.test.ts +435 -0
- package/src/tools/CodeExecutor.ts +62 -4
- package/src/tools/ProgrammaticToolCalling.js +602 -0
- package/src/tools/ProgrammaticToolCalling.js.map +1 -0
- package/src/tools/ProgrammaticToolCalling.ts +40 -52
- package/src/tools/StreamingToolCallBuffer.js +179 -0
- package/src/tools/StreamingToolCallBuffer.js.map +1 -0
- package/src/tools/StreamingToolCallBuffer.ts +218 -0
- package/src/tools/ToolNode.js +930 -0
- package/src/tools/ToolNode.js.map +1 -0
- package/src/tools/ToolNode.ts +454 -41
- package/src/tools/ToolSearch.js +904 -0
- package/src/tools/ToolSearch.js.map +1 -0
- package/src/tools/ToolSearch.ts +84 -33
- package/src/tools/__tests__/AskUser.test.ts +537 -0
- package/src/tools/__tests__/BrowserTools.test.js +306 -0
- package/src/tools/__tests__/BrowserTools.test.js.map +1 -0
- package/src/tools/__tests__/BrowserTools.test.ts +131 -6
- package/src/tools/__tests__/ProgrammaticToolCalling.integration.test.js +276 -0
- package/src/tools/__tests__/ProgrammaticToolCalling.integration.test.js.map +1 -0
- package/src/tools/__tests__/ProgrammaticToolCalling.test.js +807 -0
- package/src/tools/__tests__/ProgrammaticToolCalling.test.js.map +1 -0
- package/src/tools/__tests__/StreamingToolCallBuffer.test.js +175 -0
- package/src/tools/__tests__/StreamingToolCallBuffer.test.js.map +1 -0
- package/src/tools/__tests__/StreamingToolCallBuffer.test.ts +263 -0
- package/src/tools/__tests__/ToolApproval.test.js +675 -0
- package/src/tools/__tests__/ToolApproval.test.js.map +1 -0
- package/src/tools/__tests__/ToolApproval.test.ts +194 -20
- package/src/tools/__tests__/ToolNode.hitl.test.ts +267 -0
- package/src/tools/__tests__/ToolNode.recovery.test.js +200 -0
- package/src/tools/__tests__/ToolNode.recovery.test.js.map +1 -0
- package/src/tools/__tests__/ToolNode.recovery.test.ts +276 -0
- package/src/tools/__tests__/ToolNode.session.test.js +319 -0
- package/src/tools/__tests__/ToolNode.session.test.js.map +1 -0
- package/src/tools/__tests__/ToolNode.session.test.ts +465 -0
- package/src/tools/__tests__/ToolSearch.integration.test.js +125 -0
- package/src/tools/__tests__/ToolSearch.integration.test.js.map +1 -0
- package/src/tools/__tests__/ToolSearch.test.js +812 -0
- package/src/tools/__tests__/ToolSearch.test.js.map +1 -0
- package/src/tools/__tests__/ToolSearch.test.ts +78 -5
- package/src/tools/__tests__/handlers.test.js +799 -0
- package/src/tools/__tests__/handlers.test.js.map +1 -0
- package/src/tools/__tests__/handlers.test.ts +1100 -0
- package/src/tools/__tests__/truncation-recovery.integration.test.js +362 -0
- package/src/tools/__tests__/truncation-recovery.integration.test.js.map +1 -0
- package/src/tools/__tests__/truncation-recovery.integration.test.ts +560 -0
- package/src/tools/handlers.js +306 -0
- package/src/tools/handlers.js.map +1 -0
- package/src/tools/handlers.ts +119 -16
- package/src/tools/schema.js +25 -0
- package/src/tools/schema.js.map +1 -0
- package/src/tools/search/anthropic.js +34 -0
- package/src/tools/search/anthropic.js.map +1 -0
- package/src/tools/search/content.js +116 -0
- package/src/tools/search/content.js.map +1 -0
- package/src/tools/search/content.test.js +133 -0
- package/src/tools/search/content.test.js.map +1 -0
- package/src/tools/search/firecrawl.js +173 -0
- package/src/tools/search/firecrawl.js.map +1 -0
- package/src/tools/search/format.js +198 -0
- package/src/tools/search/format.js.map +1 -0
- package/src/tools/search/highlights.js +241 -0
- package/src/tools/search/highlights.js.map +1 -0
- package/src/tools/search/index.js +3 -0
- package/src/tools/search/index.js.map +1 -0
- package/src/tools/search/jina-reranker.test.js +106 -0
- package/src/tools/search/jina-reranker.test.js.map +1 -0
- package/src/tools/search/rerankers.js +165 -0
- package/src/tools/search/rerankers.js.map +1 -0
- package/src/tools/search/schema.js +102 -0
- package/src/tools/search/schema.js.map +1 -0
- package/src/tools/search/search.js +561 -0
- package/src/tools/search/search.js.map +1 -0
- package/src/tools/search/serper-scraper.js +126 -0
- package/src/tools/search/serper-scraper.js.map +1 -0
- package/src/tools/search/test.js +129 -0
- package/src/tools/search/test.js.map +1 -0
- package/src/tools/search/tool.js +453 -0
- package/src/tools/search/tool.js.map +1 -0
- package/src/tools/search/types.js +2 -0
- package/src/tools/search/types.js.map +1 -0
- package/src/tools/search/utils.js +59 -0
- package/src/tools/search/utils.js.map +1 -0
- package/src/types/graph.js +24 -0
- package/src/types/graph.js.map +1 -0
- package/src/types/graph.test.js +192 -0
- package/src/types/graph.test.js.map +1 -0
- package/src/types/graph.ts +26 -6
- package/src/types/index.js +7 -0
- package/src/types/index.js.map +1 -0
- package/src/types/llm.js +2 -0
- package/src/types/llm.js.map +1 -0
- package/src/types/llm.ts +8 -3
- package/src/types/messages.js +2 -0
- package/src/types/messages.js.map +1 -0
- package/src/types/run.js +2 -0
- package/src/types/run.js.map +1 -0
- package/src/types/run.ts +2 -0
- package/src/types/stream.js +2 -0
- package/src/types/stream.js.map +1 -0
- package/src/types/tools.js +2 -0
- package/src/types/tools.js.map +1 -0
- package/src/types/tools.ts +21 -2
- package/src/utils/contextAnalytics.js +79 -0
- package/src/utils/contextAnalytics.js.map +1 -0
- package/src/utils/contextAnalytics.test.js +166 -0
- package/src/utils/contextAnalytics.test.js.map +1 -0
- package/src/utils/contextAnalytics.test.ts +222 -0
- package/src/utils/contextAnalytics.ts +27 -9
- package/src/utils/events.js +26 -0
- package/src/utils/events.js.map +1 -0
- package/src/utils/graph.js +11 -0
- package/src/utils/graph.js.map +1 -0
- package/src/utils/handlers.js +65 -0
- package/src/utils/handlers.js.map +1 -0
- package/src/utils/index.js +10 -0
- package/src/utils/index.js.map +1 -0
- package/src/utils/index.ts +1 -0
- package/src/utils/llm.js +21 -0
- package/src/utils/llm.js.map +1 -0
- package/src/utils/llmConfig.js +205 -0
- package/src/utils/llmConfig.js.map +1 -0
- package/src/utils/llmConfig.ts +5 -5
- package/src/utils/logging.js +37 -0
- package/src/utils/logging.js.map +1 -0
- package/src/utils/misc.js +51 -0
- package/src/utils/misc.js.map +1 -0
- package/src/utils/run.js +69 -0
- package/src/utils/run.js.map +1 -0
- package/src/utils/run.ts +108 -106
- package/src/utils/schema.js +21 -0
- package/src/utils/schema.js.map +1 -0
- package/src/utils/title.js +119 -0
- package/src/utils/title.js.map +1 -0
- package/src/utils/tokens.js +92 -0
- package/src/utils/tokens.js.map +1 -0
- package/src/utils/tokens.ts +118 -142
- package/src/utils/toolCallContinuation.ts +55 -0
- package/src/utils/toonFormat.js +379 -0
- package/src/utils/toonFormat.js.map +1 -0
|
@@ -1,8 +1,27 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
|
-
var
|
|
3
|
+
var aiTokenizer = require('ai-tokenizer');
|
|
4
4
|
var _enum = require('../common/enum.cjs');
|
|
5
5
|
|
|
6
|
+
const tokenizers = {};
|
|
7
|
+
async function getTokenizer(encoding = 'o200k_base') {
|
|
8
|
+
const cached = tokenizers[encoding];
|
|
9
|
+
if (cached) {
|
|
10
|
+
return cached;
|
|
11
|
+
}
|
|
12
|
+
const data = encoding === 'claude'
|
|
13
|
+
? await import('ai-tokenizer/encoding/claude')
|
|
14
|
+
: await import('ai-tokenizer/encoding/o200k_base');
|
|
15
|
+
const instance = new aiTokenizer.Tokenizer(data);
|
|
16
|
+
tokenizers[encoding] = instance;
|
|
17
|
+
return instance;
|
|
18
|
+
}
|
|
19
|
+
function encodingForModel(model) {
|
|
20
|
+
if (model.toLowerCase().includes('claude')) {
|
|
21
|
+
return 'claude';
|
|
22
|
+
}
|
|
23
|
+
return 'o200k_base';
|
|
24
|
+
}
|
|
6
25
|
function getTokenCountForMessage(message, getTokenCount) {
|
|
7
26
|
const tokensPerMessage = 3;
|
|
8
27
|
const processValue = (value) => {
|
|
@@ -50,76 +69,32 @@ function getTokenCountForMessage(message, getTokenCount) {
|
|
|
50
69
|
processValue(message.content);
|
|
51
70
|
return numTokens;
|
|
52
71
|
}
|
|
53
|
-
let encoderPromise;
|
|
54
|
-
let tokenCounterPromise;
|
|
55
|
-
/**
|
|
56
|
-
* Simple character-based token estimation (~4 chars per token).
|
|
57
|
-
* Used as fallback when tiktoken encoder fails to load.
|
|
58
|
-
*/
|
|
59
|
-
const estimateTokens = (text) => Math.ceil(text.length / 4);
|
|
60
|
-
async function getSharedEncoder() {
|
|
61
|
-
if (encoderPromise) {
|
|
62
|
-
return encoderPromise;
|
|
63
|
-
}
|
|
64
|
-
// Use bundled tokenizer data - no external network calls
|
|
65
|
-
encoderPromise = Promise.resolve().then(() => {
|
|
66
|
-
try {
|
|
67
|
-
return jsTiktoken.getEncoding('o200k_base');
|
|
68
|
-
}
|
|
69
|
-
catch {
|
|
70
|
-
// Fallback to null - will use character estimation
|
|
71
|
-
return null;
|
|
72
|
-
}
|
|
73
|
-
});
|
|
74
|
-
return encoderPromise;
|
|
75
|
-
}
|
|
76
72
|
/**
|
|
77
|
-
* Creates a
|
|
78
|
-
*
|
|
79
|
-
* This ensures token counting never blocks LLM calls.
|
|
73
|
+
* Creates a token counter function using the specified encoding.
|
|
74
|
+
* Lazily loads the encoding data on first use via dynamic import.
|
|
80
75
|
*/
|
|
81
|
-
const createTokenCounter = async () => {
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
tokenCounterPromise = (async () => {
|
|
86
|
-
const enc = await getSharedEncoder();
|
|
87
|
-
const countTokens = enc
|
|
88
|
-
? (text) => enc.encode(text).length
|
|
89
|
-
: estimateTokens;
|
|
90
|
-
return (message) => getTokenCountForMessage(message, countTokens);
|
|
91
|
-
})();
|
|
92
|
-
return tokenCounterPromise;
|
|
76
|
+
const createTokenCounter = async (encoding = 'o200k_base') => {
|
|
77
|
+
const tok = await getTokenizer(encoding);
|
|
78
|
+
const countTokens = (text) => tok.count(text);
|
|
79
|
+
return (message) => getTokenCountForMessage(message, countTokens);
|
|
93
80
|
};
|
|
94
|
-
/**
|
|
95
|
-
* Utility to manage the token encoder lifecycle explicitly.
|
|
96
|
-
* Useful for applications that need fine-grained control over resource management.
|
|
97
|
-
*/
|
|
81
|
+
/** Utility to manage the token encoder lifecycle explicitly. */
|
|
98
82
|
const TokenEncoderManager = {
|
|
99
|
-
/**
|
|
100
|
-
* Pre-initializes the encoder. This can be called during app startup
|
|
101
|
-
* to avoid lazy loading delays later.
|
|
102
|
-
*/
|
|
103
83
|
async initialize() {
|
|
104
|
-
|
|
84
|
+
// No-op: ai-tokenizer is synchronously initialized from bundled data.
|
|
105
85
|
},
|
|
106
|
-
/**
|
|
107
|
-
* Clears the cached encoder and token counter.
|
|
108
|
-
* Useful for testing or when you need to force a fresh reload.
|
|
109
|
-
*/
|
|
110
86
|
reset() {
|
|
111
|
-
|
|
112
|
-
|
|
87
|
+
for (const key of Object.keys(tokenizers)) {
|
|
88
|
+
delete tokenizers[key];
|
|
89
|
+
}
|
|
113
90
|
},
|
|
114
|
-
/**
|
|
115
|
-
* Checks if the encoder has been initialized.
|
|
116
|
-
*/
|
|
117
91
|
isInitialized() {
|
|
118
|
-
return
|
|
92
|
+
return Object.keys(tokenizers).length > 0;
|
|
119
93
|
},
|
|
120
94
|
};
|
|
121
95
|
|
|
122
96
|
exports.TokenEncoderManager = TokenEncoderManager;
|
|
123
97
|
exports.createTokenCounter = createTokenCounter;
|
|
98
|
+
exports.encodingForModel = encodingForModel;
|
|
124
99
|
exports.getTokenCountForMessage = getTokenCountForMessage;
|
|
125
100
|
//# sourceMappingURL=tokens.cjs.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"tokens.cjs","sources":["../../../src/utils/tokens.ts"],"sourcesContent":["import {
|
|
1
|
+
{"version":3,"file":"tokens.cjs","sources":["../../../src/utils/tokens.ts"],"sourcesContent":["import { Tokenizer } from 'ai-tokenizer';\r\nimport type { BaseMessage } from '@langchain/core/messages';\r\nimport { ContentTypes } from '@/common/enum';\r\n\r\nexport type EncodingName = 'o200k_base' | 'claude';\r\n\r\nconst tokenizers: Partial<Record<EncodingName, Tokenizer>> = {};\r\n\r\nasync function getTokenizer(\r\n encoding: EncodingName = 'o200k_base'\r\n): Promise<Tokenizer> {\r\n const cached = tokenizers[encoding];\r\n if (cached) {\r\n return cached;\r\n }\r\n const data =\r\n encoding === 'claude'\r\n ? await import('ai-tokenizer/encoding/claude')\r\n : await import('ai-tokenizer/encoding/o200k_base');\r\n const instance = new Tokenizer(data);\r\n tokenizers[encoding] = instance;\r\n return instance;\r\n}\r\n\r\nexport function encodingForModel(model: string): EncodingName {\r\n if (model.toLowerCase().includes('claude')) {\r\n return 'claude';\r\n }\r\n return 'o200k_base';\r\n}\r\n\r\nexport function getTokenCountForMessage(\r\n message: BaseMessage,\r\n getTokenCount: (text: string) => number\r\n): number {\r\n const tokensPerMessage = 3;\r\n\r\n const processValue = (value: unknown): void => {\r\n if (Array.isArray(value)) {\r\n for (const item of value) {\r\n if (\r\n !item ||\r\n !item.type ||\r\n item.type === ContentTypes.ERROR ||\r\n item.type === ContentTypes.IMAGE_URL\r\n ) {\r\n continue;\r\n }\r\n\r\n if (item.type === ContentTypes.TOOL_CALL && item.tool_call != null) {\r\n const toolName = item.tool_call?.name || '';\r\n if (toolName != null && toolName && typeof toolName === 'string') {\r\n numTokens += getTokenCount(toolName);\r\n }\r\n\r\n const args = item.tool_call?.args || '';\r\n if (args != null && args && typeof args === 'string') {\r\n numTokens += getTokenCount(args);\r\n }\r\n\r\n const output = item.tool_call?.output || '';\r\n if (output != null && output && typeof output === 'string') {\r\n numTokens += getTokenCount(output);\r\n }\r\n continue;\r\n }\r\n\r\n const nestedValue = item[item.type];\r\n\r\n if (!nestedValue) {\r\n continue;\r\n }\r\n\r\n processValue(nestedValue);\r\n }\r\n } else if (typeof value === 'string') {\r\n numTokens += getTokenCount(value);\r\n } else if (typeof value === 'number') {\r\n numTokens += getTokenCount(value.toString());\r\n } else if (typeof value === 'boolean') {\r\n numTokens += getTokenCount(value.toString());\r\n }\r\n };\r\n\r\n let numTokens = tokensPerMessage;\r\n processValue(message.content);\r\n return numTokens;\r\n}\r\n\r\n/**\r\n * Creates a token counter function using the specified encoding.\r\n * Lazily loads the encoding data on first use via dynamic import.\r\n */\r\nexport const createTokenCounter = async (\r\n encoding: EncodingName = 'o200k_base'\r\n): Promise<(message: BaseMessage) => number> => {\r\n const tok = await getTokenizer(encoding);\r\n const countTokens = (text: string): number => tok.count(text);\r\n return (message: BaseMessage): number =>\r\n getTokenCountForMessage(message, countTokens);\r\n};\r\n\r\n/** Utility to manage the token encoder lifecycle explicitly. */\r\nexport const TokenEncoderManager = {\r\n async initialize(): Promise<void> {\r\n // No-op: ai-tokenizer is synchronously initialized from bundled data.\r\n },\r\n\r\n reset(): void {\r\n for (const key of Object.keys(tokenizers)) {\r\n delete tokenizers[key as EncodingName];\r\n }\r\n },\r\n\r\n isInitialized(): boolean {\r\n return Object.keys(tokenizers).length > 0;\r\n },\r\n};\r\n"],"names":["Tokenizer","ContentTypes"],"mappings":";;;;;AAMA,MAAM,UAAU,GAA6C,EAAE;AAE/D,eAAe,YAAY,CACzB,QAAA,GAAyB,YAAY,EAAA;AAErC,IAAA,MAAM,MAAM,GAAG,UAAU,CAAC,QAAQ,CAAC;IACnC,IAAI,MAAM,EAAE;AACV,QAAA,OAAO,MAAM;IACf;AACA,IAAA,MAAM,IAAI,GACR,QAAQ,KAAK;AACX,UAAE,MAAM,OAAO,8BAA8B;AAC7C,UAAE,MAAM,OAAO,kCAAkC,CAAC;AACtD,IAAA,MAAM,QAAQ,GAAG,IAAIA,qBAAS,CAAC,IAAI,CAAC;AACpC,IAAA,UAAU,CAAC,QAAQ,CAAC,GAAG,QAAQ;AAC/B,IAAA,OAAO,QAAQ;AACjB;AAEM,SAAU,gBAAgB,CAAC,KAAa,EAAA;IAC5C,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE;AAC1C,QAAA,OAAO,QAAQ;IACjB;AACA,IAAA,OAAO,YAAY;AACrB;AAEM,SAAU,uBAAuB,CACrC,OAAoB,EACpB,aAAuC,EAAA;IAEvC,MAAM,gBAAgB,GAAG,CAAC;AAE1B,IAAA,MAAM,YAAY,GAAG,CAAC,KAAc,KAAU;AAC5C,QAAA,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;AACxB,YAAA,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;AACxB,gBAAA,IACE,CAAC,IAAI;oBACL,CAAC,IAAI,CAAC,IAAI;AACV,oBAAA,IAAI,CAAC,IAAI,KAAKC,kBAAY,CAAC,KAAK;AAChC,oBAAA,IAAI,CAAC,IAAI,KAAKA,kBAAY,CAAC,SAAS,EACpC;oBACA;gBACF;AAEA,gBAAA,IAAI,IAAI,CAAC,IAAI,KAAKA,kBAAY,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,EAAE;oBAClE,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,EAAE,IAAI,IAAI,EAAE;oBAC3C,IAAI,QAAQ,IAAI,IAAI,IAAI,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE;AAChE,wBAAA,SAAS,IAAI,aAAa,CAAC,QAAQ,CAAC;oBACtC;oBAEA,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,EAAE,IAAI,IAAI,EAAE;oBACvC,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE;AACpD,wBAAA,SAAS,IAAI,aAAa,CAAC,IAAI,CAAC;oBAClC;oBAEA,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,MAAM,IAAI,EAAE;oBAC3C,IAAI,MAAM,IAAI,IAAI,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE;AAC1D,wBAAA,SAAS,IAAI,aAAa,CAAC,MAAM,CAAC;oBACpC;oBACA;gBACF;gBAEA,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;gBAEnC,IAAI,CAAC,WAAW,EAAE;oBAChB;gBACF;gBAEA,YAAY,CAAC,WAAW,CAAC;YAC3B;QACF;AAAO,aAAA,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;AACpC,YAAA,SAAS,IAAI,aAAa,CAAC,KAAK,CAAC;QACnC;AAAO,aAAA,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;YACpC,SAAS,IAAI,aAAa,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC9C;AAAO,aAAA,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE;YACrC,SAAS,IAAI,aAAa,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC9C;AACF,IAAA,CAAC;IAED,IAAI,SAAS,GAAG,gBAAgB;AAChC,IAAA,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC;AAC7B,IAAA,OAAO,SAAS;AAClB;AAEA;;;AAGG;MACU,kBAAkB,GAAG,OAChC,QAAA,GAAyB,YAAY,KACQ;AAC7C,IAAA,MAAM,GAAG,GAAG,MAAM,YAAY,CAAC,QAAQ,CAAC;AACxC,IAAA,MAAM,WAAW,GAAG,CAAC,IAAY,KAAa,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;IAC7D,OAAO,CAAC,OAAoB,KAC1B,uBAAuB,CAAC,OAAO,EAAE,WAAW,CAAC;AACjD;AAEA;AACO,MAAM,mBAAmB,GAAG;AACjC,IAAA,MAAM,UAAU,GAAA;;IAEhB,CAAC;IAED,KAAK,GAAA;QACH,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE;AACzC,YAAA,OAAO,UAAU,CAAC,GAAmB,CAAC;QACxC;IACF,CAAC;IAED,aAAa,GAAA;QACX,OAAO,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC;IAC3C,CAAC;;;;;;;;"}
|
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
|
|
3
|
+
var _enum = require('../common/enum.cjs');
|
|
4
|
+
|
|
5
|
+
/**
|
|
6
|
+
* toolCallContinuation — Utilities for detecting and classifying LLM finish
|
|
7
|
+
* reasons across providers. Used to detect max_tokens truncation and normalize
|
|
8
|
+
* stop reason extraction from provider-specific metadata formats.
|
|
9
|
+
*
|
|
10
|
+
* @module
|
|
11
|
+
*/
|
|
12
|
+
/**
|
|
13
|
+
* Extract the finish/stop reason from an AIMessageChunk's response_metadata.
|
|
14
|
+
* Centralizes provider-specific metadata access patterns.
|
|
15
|
+
*
|
|
16
|
+
* Provider formats:
|
|
17
|
+
* - OpenAI/Azure: `response_metadata.finish_reason`
|
|
18
|
+
* - Anthropic direct: `response_metadata.stop_reason`
|
|
19
|
+
* - Bedrock invoke: `response_metadata.stopReason`
|
|
20
|
+
* - Bedrock streaming: `response_metadata.messageStop.stopReason`
|
|
21
|
+
* - VertexAI/Google: `response_metadata.finishReason`
|
|
22
|
+
*
|
|
23
|
+
* @param chunk - The final accumulated AIMessageChunk
|
|
24
|
+
* @returns The finish reason string, or undefined if not found
|
|
25
|
+
*/
|
|
26
|
+
function extractFinishReason(chunk) {
|
|
27
|
+
if (!chunk || !('response_metadata' in chunk))
|
|
28
|
+
return undefined;
|
|
29
|
+
const meta = chunk.response_metadata;
|
|
30
|
+
if (!meta)
|
|
31
|
+
return undefined;
|
|
32
|
+
// Bedrock streaming nests stopReason inside messageStop: { stopReason: '...' }
|
|
33
|
+
const messageStop = meta.messageStop;
|
|
34
|
+
return (meta.finish_reason ?? // OpenAI/Azure
|
|
35
|
+
meta.stop_reason ?? // Anthropic direct API
|
|
36
|
+
meta.stopReason ?? // Bedrock invoke (non-streaming)
|
|
37
|
+
messageStop?.stopReason ?? // Bedrock streaming
|
|
38
|
+
meta.finishReason // VertexAI/Google
|
|
39
|
+
);
|
|
40
|
+
}
|
|
41
|
+
/**
|
|
42
|
+
* Check if a finish reason indicates the response was truncated by token limits.
|
|
43
|
+
*
|
|
44
|
+
* @param reason - The finish reason from extractFinishReason()
|
|
45
|
+
* @returns true if the response was cut short by max_tokens
|
|
46
|
+
*/
|
|
47
|
+
function isMaxTokensFinish(reason) {
|
|
48
|
+
if (!reason)
|
|
49
|
+
return false;
|
|
50
|
+
return reason === _enum.FinishReasons.MAX_TOKENS || reason === _enum.FinishReasons.LENGTH;
|
|
51
|
+
}
|
|
52
|
+
|
|
53
|
+
exports.extractFinishReason = extractFinishReason;
|
|
54
|
+
exports.isMaxTokensFinish = isMaxTokensFinish;
|
|
55
|
+
//# sourceMappingURL=toolCallContinuation.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"toolCallContinuation.cjs","sources":["../../../src/utils/toolCallContinuation.ts"],"sourcesContent":["/**\n * toolCallContinuation — Utilities for detecting and classifying LLM finish\n * reasons across providers. Used to detect max_tokens truncation and normalize\n * stop reason extraction from provider-specific metadata formats.\n *\n * @module\n */\n\nimport type { AIMessageChunk } from '@langchain/core/messages';\nimport { FinishReasons } from '@/common';\n\n/**\n * Extract the finish/stop reason from an AIMessageChunk's response_metadata.\n * Centralizes provider-specific metadata access patterns.\n *\n * Provider formats:\n * - OpenAI/Azure: `response_metadata.finish_reason`\n * - Anthropic direct: `response_metadata.stop_reason`\n * - Bedrock invoke: `response_metadata.stopReason`\n * - Bedrock streaming: `response_metadata.messageStop.stopReason`\n * - VertexAI/Google: `response_metadata.finishReason`\n *\n * @param chunk - The final accumulated AIMessageChunk\n * @returns The finish reason string, or undefined if not found\n */\nexport function extractFinishReason(\n chunk: AIMessageChunk | undefined\n): string | undefined {\n if (!chunk || !('response_metadata' in chunk)) return undefined;\n\n const meta = chunk.response_metadata as Record<string, unknown>;\n if (!meta) return undefined;\n\n // Bedrock streaming nests stopReason inside messageStop: { stopReason: '...' }\n const messageStop = meta.messageStop as Record<string, unknown> | undefined;\n\n return (\n (meta.finish_reason as string | undefined) ?? // OpenAI/Azure\n (meta.stop_reason as string | undefined) ?? // Anthropic direct API\n (meta.stopReason as string | undefined) ?? // Bedrock invoke (non-streaming)\n (messageStop?.stopReason as string | undefined) ?? // Bedrock streaming\n (meta.finishReason as string | undefined) // VertexAI/Google\n );\n}\n\n/**\n * Check if a finish reason indicates the response was truncated by token limits.\n *\n * @param reason - The finish reason from extractFinishReason()\n * @returns true if the response was cut short by max_tokens\n */\nexport function isMaxTokensFinish(reason: string | undefined): boolean {\n if (!reason) return false;\n return reason === FinishReasons.MAX_TOKENS || reason === FinishReasons.LENGTH;\n}\n"],"names":["FinishReasons"],"mappings":";;;;AAAA;;;;;;AAMG;AAKH;;;;;;;;;;;;;AAaG;AACG,SAAU,mBAAmB,CACjC,KAAiC,EAAA;IAEjC,IAAI,CAAC,KAAK,IAAI,EAAE,mBAAmB,IAAI,KAAK,CAAC;AAAE,QAAA,OAAO,SAAS;AAE/D,IAAA,MAAM,IAAI,GAAG,KAAK,CAAC,iBAA4C;AAC/D,IAAA,IAAI,CAAC,IAAI;AAAE,QAAA,OAAO,SAAS;;AAG3B,IAAA,MAAM,WAAW,GAAG,IAAI,CAAC,WAAkD;AAE3E,IAAA,QACG,IAAI,CAAC,aAAoC;QACzC,IAAI,CAAC,WAAkC;QACvC,IAAI,CAAC,UAAiC;QACtC,WAAW,EAAE,UAAiC;QAC9C,IAAI,CAAC,YAAmC;;AAE7C;AAEA;;;;;AAKG;AACG,SAAU,iBAAiB,CAAC,MAA0B,EAAA;AAC1D,IAAA,IAAI,CAAC,MAAM;AAAE,QAAA,OAAO,KAAK;IACzB,OAAO,MAAM,KAAKA,mBAAa,CAAC,UAAU,IAAI,MAAM,KAAKA,mBAAa,CAAC,MAAM;AAC/E;;;;;"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"toonFormat.cjs","sources":["../../../src/utils/toonFormat.ts"],"sourcesContent":["/**\n * TOON Format Utility\n *\n * Provides JSON to TOON conversion for token-efficient encoding.\n * TOON format can reduce token count by 30-60% for JSON data.\n *\n * Uses the official @toon-format/toon library when available.\n */\n\n// Dynamic import holder for the TOON library (ESM)\nlet toonEncode: ((data: unknown) => string) | null = null;\nlet toonLoadAttempted = false;\nlet toonLoadPromise: Promise<boolean> | null = null;\n\n/**\n * Lazily loads the TOON library\n * Tries both ESM dynamic import and CommonJS require for compatibility\n */\nasync function loadToonLibrary(): Promise<boolean> {\n if (toonLoadAttempted) return toonEncode !== null;\n toonLoadAttempted = true;\n\n // Try ESM dynamic import first\n try {\n const mod = await import('@toon-format/toon');\n toonEncode = mod.encode;\n return true;\n } catch {\n // ESM import failed, try CommonJS require\n try {\n // eslint-disable-next-line @typescript-eslint/no-require-imports\n const mod = require('@toon-format/toon');\n toonEncode = mod.encode;\n return true;\n } catch {\n // Library not available - that's OK, we'll return original content\n return false;\n }\n }\n}\n\n/**\n * Ensures the TOON library is loaded. Call this before using jsonToToon in tests.\n */\nexport async function ensureToonLoaded(): Promise<boolean> {\n if (!toonLoadPromise) {\n toonLoadPromise = loadToonLibrary();\n }\n return toonLoadPromise;\n}\n\n// Start loading immediately\nensureToonLoaded();\n\n/**\n * Check if a string appears to be in TOON format.\n * TOON format characteristics:\n * - Uses key: value syntax (like YAML)\n * - Uses array notation like [3] or {fields}\n * - Does NOT start with { or [\n * - Often has patterns like \"name[N]:\" or \"name{fields}:\"\n *\n * @param str - String to check\n * @returns true if string appears to be TOON format\n */\nexport function isToonFormat(str: string): boolean {\n if (typeof str !== 'string' || str.length === 0) return false;\n\n const trimmed = str.trim();\n\n // TOON doesn't start with JSON brackets\n if (trimmed.startsWith('{') || trimmed.startsWith('[')) return false;\n\n // Check for TOON-specific patterns:\n // 1. Key-value with colon (but not URL-like patterns)\n // 2. Array notation like \"items[3]:\" or \"data[10]:\"\n // 3. Object schema notation like \"items{id,name}:\"\n const toonPatterns = [\n /^\\w+:$/m, // Simple key: at start of line\n /^\\w+\\[\\d+\\]:/m, // Array notation: items[3]:\n /^\\w+\\{[^}]+\\}:/m, // Schema notation: items{id,name}:\n /^\\w+\\[\\d+\\]\\{[^}]+\\}:/m, // Combined: items[3]{id,name}:\n ];\n\n // Must match at least one TOON pattern\n const hasToonPattern = toonPatterns.some((pattern) => pattern.test(trimmed));\n\n // Additional check: TOON typically has multiple lines with consistent indentation\n const lines = trimmed.split('\\n').filter((l) => l.trim());\n const hasMultipleKeyValueLines =\n lines.filter((l) => /^\\s*\\w+.*:/.test(l)).length >= 2;\n\n return hasToonPattern || hasMultipleKeyValueLines;\n}\n\n/**\n * Extract the first valid JSON object or array from a string.\n * Handles pure JSON or JSON embedded in text (e.g., tool responses).\n *\n * @param str - The string to extract JSON from\n * @returns Object with found status, parsed JSON, and indices\n */\nexport function extractFirstJson(str: string): {\n found: boolean;\n parsed: unknown;\n startIndex: number;\n endIndex: number;\n} {\n if (typeof str !== 'string' || str.length === 0) {\n return { found: false, parsed: null, startIndex: -1, endIndex: -1 };\n }\n\n // Find the first { or [ character\n const objStart = str.indexOf('{');\n const arrStart = str.indexOf('[');\n\n if (objStart === -1 && arrStart === -1) {\n return { found: false, parsed: null, startIndex: -1, endIndex: -1 };\n }\n\n let startIndex: number;\n if (objStart === -1) {\n startIndex = arrStart;\n } else if (arrStart === -1) {\n startIndex = objStart;\n } else {\n startIndex = Math.min(objStart, arrStart);\n }\n\n // Find the matching closing bracket using a stack-based approach\n // Properly handle strings to avoid counting brackets inside strings\n let depth = 0;\n let inString = false;\n let escapeNext = false;\n\n for (let i = startIndex; i < str.length; i++) {\n const char = str[i];\n\n if (escapeNext) {\n escapeNext = false;\n continue;\n }\n\n if (char === '\\\\' && inString) {\n escapeNext = true;\n continue;\n }\n\n if (char === '\"') {\n inString = !inString;\n continue;\n }\n\n if (inString) continue;\n\n if (char === '{' || char === '[') depth++;\n if (char === '}' || char === ']') depth--;\n\n if (depth === 0) {\n // Found matching bracket, try to parse\n const jsonStr = str.substring(startIndex, i + 1);\n try {\n const parsed = JSON.parse(jsonStr);\n return { found: true, parsed, startIndex, endIndex: i };\n } catch {\n // Not valid JSON at this bracket level, continue searching\n // Reset and look for the next JSON start after this position\n const nextResult = extractFirstJson(str.substring(i + 1));\n if (nextResult.found) {\n return {\n found: true,\n parsed: nextResult.parsed,\n startIndex: i + 1 + nextResult.startIndex,\n endIndex: i + 1 + nextResult.endIndex,\n };\n }\n return { found: false, parsed: null, startIndex: -1, endIndex: -1 };\n }\n }\n }\n\n return { found: false, parsed: null, startIndex: -1, endIndex: -1 };\n}\n\n/**\n * Convert JSON content to TOON format for token efficiency.\n * Extracts JSON from string if embedded, converts to TOON.\n * Returns original string if:\n * - Already in TOON format\n * - Not JSON\n * - TOON conversion fails or is larger\n *\n * @param str - The string containing JSON to convert\n * @returns Object with conversion status, result, and reduction percentage\n */\nexport function jsonToToon(str: string): {\n converted: boolean;\n result: string;\n reduction: number;\n alreadyToon?: boolean;\n error?: string;\n} {\n if (!toonEncode || typeof str !== 'string') {\n return {\n converted: false,\n result: str,\n reduction: 0,\n error: !toonEncode ? 'TOON library not loaded' : undefined,\n };\n }\n\n // Check if already in TOON format - skip conversion\n if (isToonFormat(str)) {\n return { converted: false, result: str, reduction: 0, alreadyToon: true };\n }\n\n try {\n const { found, parsed, startIndex, endIndex } = extractFirstJson(str);\n\n if (!found || !parsed) {\n return {\n converted: false,\n result: str,\n reduction: 0,\n error: 'No JSON found in content',\n };\n }\n\n // Check if this JSON structure would benefit from TOON\n // Text-heavy content (emails, documents) won't compress well\n if (!isToonBeneficial(parsed)) {\n return {\n converted: false,\n result: str,\n reduction: 0,\n error:\n 'Content is text-heavy (>70% string values), TOON not beneficial',\n };\n }\n\n const toonResult = toonEncode(parsed);\n\n // Preserve text before and after the JSON block\n const textBefore = str.substring(0, startIndex);\n const textAfter = str.substring(endIndex + 1);\n\n // Build the full result: textBefore + TOON + textAfter\n const fullResult = textBefore + toonResult + textAfter;\n\n if (fullResult.length < str.length) {\n const reduction = Math.round(\n ((str.length - fullResult.length) / str.length) * 100\n );\n return { converted: true, result: fullResult, reduction };\n } else {\n // TOON output was larger or same size - not beneficial\n return {\n converted: false,\n result: str,\n reduction: 0,\n error: `TOON output not smaller (${fullResult.length} >= ${str.length})`,\n };\n }\n } catch (err) {\n // TOON encoding or extraction failed - log error for debugging\n const errorMsg = err instanceof Error ? err.message : String(err);\n return {\n converted: false,\n result: str,\n reduction: 0,\n error: `TOON conversion error: ${errorMsg}`,\n };\n }\n}\n\n/**\n * Check if TOON library is loaded and available\n */\nexport function isToonAvailable(): boolean {\n return toonEncode !== null;\n}\n\n/**\n * Result of processing tool output\n */\nexport interface ProcessToolOutputResult {\n /** The processed content string */\n content: string;\n /** Whether TOON conversion was applied */\n toonConverted: boolean;\n /** Whether content was truncated */\n truncated: boolean;\n /** Percentage reduction from TOON (0 if not converted) */\n reduction: number;\n /** Whether input was already in TOON format */\n alreadyToon: boolean;\n /** Original content length */\n originalLength: number;\n /** Estimated original tokens (~4 chars per token) */\n originalTokens: number;\n /** Final content length */\n finalLength: number;\n /** Estimated final tokens */\n finalTokens: number;\n /** Error message if TOON conversion failed (for debugging) */\n toonError?: string;\n}\n\n/**\n * Options for processing tool output\n */\nexport interface ProcessToolOutputOptions {\n /** Maximum output length in characters (default: 100000) */\n maxLength?: number;\n /** Whether to apply TOON conversion (default: true) */\n enableToon?: boolean;\n /** Minimum content size to attempt TOON conversion (default: 1000) */\n minSizeForToon?: number;\n /** Minimum reduction % to accept TOON result (default: 10) */\n minReductionPercent?: number;\n}\n\n/**\n * Analyze JSON structure to determine if TOON conversion would be beneficial.\n * Text-heavy content (emails, documents) doesn't compress well with TOON.\n * Structured data (API responses, metadata) compresses much better.\n *\n * @param parsed - Parsed JSON object\n * @returns true if TOON conversion is likely beneficial\n */\nfunction isToonBeneficial(parsed: unknown): boolean {\n if (!parsed || typeof parsed !== 'object') return false;\n\n const jsonStr = JSON.stringify(parsed);\n const totalLength = jsonStr.length;\n\n // Count characters in string values (text content)\n let textContentLength = 0;\n\n function countTextContent(obj: unknown): void {\n if (typeof obj === 'string') {\n // Count string length (this is text content that TOON can't compress)\n textContentLength += obj.length;\n } else if (Array.isArray(obj)) {\n obj.forEach(countTextContent);\n } else if (obj && typeof obj === 'object') {\n Object.values(obj).forEach(countTextContent);\n }\n }\n\n countTextContent(parsed);\n\n // Calculate ratio of text content to total JSON\n const textRatio = textContentLength / totalLength;\n\n // If more than 70% is text content, TOON won't help much\n // TOON compresses structure (field names, brackets), not text\n return textRatio < 0.7;\n}\n\n/**\n * Process tool output: apply TOON conversion if beneficial, then truncate if needed.\n * This is the main entry point for processing any tool output (regular or MCP).\n *\n * Flow:\n * 1. Check if already TOON format → skip conversion\n * 2. Try TOON conversion if content is JSON and large enough\n * 3. Truncate if still exceeds maxLength (with smart break points)\n *\n * @param content - The tool output content\n * @param options - Processing options\n * @returns Processed content and metadata\n */\nexport function processToolOutput(\n content: string,\n options: ProcessToolOutputOptions = {}\n): ProcessToolOutputResult {\n const {\n maxLength = 100000,\n enableToon = true,\n minSizeForToon = 1000,\n minReductionPercent = 10, // Increased from 5% - only apply TOON when clearly beneficial\n } = options;\n\n const originalLength = content.length;\n const originalTokens = Math.ceil(originalLength / 4);\n\n let result = content;\n let toonConverted = false;\n let alreadyToon = false;\n let reduction = 0;\n let truncated = false;\n let toonError: string | undefined;\n\n // Step 1: Check if already TOON format\n if (isToonFormat(content)) {\n alreadyToon = true;\n }\n // Step 2: Apply TOON conversion if enabled and content is large enough\n else if (enableToon && content.length > minSizeForToon) {\n const toonResult = jsonToToon(content);\n if (toonResult.alreadyToon) {\n alreadyToon = true;\n } else if (\n toonResult.converted &&\n toonResult.reduction >= minReductionPercent\n ) {\n result = toonResult.result;\n toonConverted = true;\n reduction = toonResult.reduction;\n } else if (toonResult.error) {\n // Track error for debugging\n toonError = toonResult.error;\n } else if (\n toonResult.converted &&\n toonResult.reduction < minReductionPercent\n ) {\n // TOON converted but reduction was too small to be worth it\n toonError = `TOON reduction ${toonResult.reduction}% below threshold ${minReductionPercent}%`;\n } else if (!toonResult.converted) {\n // Conversion failed without explicit error - investigate\n toonError =\n toonResult.error || 'TOON conversion returned false with no error';\n }\n }\n\n // Step 3: Truncate if still too long (with smart break points)\n if (result.length > maxLength) {\n let truncatedStr = result.substring(0, maxLength);\n\n // Try to find a clean break point\n if (toonConverted || alreadyToon) {\n // For TOON format, break at newline for cleaner output\n const lastNewline = truncatedStr.lastIndexOf('\\n');\n if (lastNewline > maxLength * 0.7) {\n truncatedStr = truncatedStr.substring(0, lastNewline);\n }\n } else {\n // For JSON, try to find a clean JSON break point\n const lastCompleteItem = truncatedStr.lastIndexOf('},');\n const lastArrayItem = truncatedStr.lastIndexOf('],');\n const breakPoint = Math.max(lastCompleteItem, lastArrayItem);\n if (breakPoint > maxLength * 0.5) {\n truncatedStr = truncatedStr.substring(0, breakPoint + 1);\n }\n }\n\n // Build truncation message\n const truncationInfo = toonConverted\n ? `Original ${originalLength.toLocaleString()} chars → TOON ${result.length.toLocaleString()} chars (${reduction}% saved). ` +\n `Still exceeds ${maxLength.toLocaleString()} char limit.`\n : `Original output was ${originalLength.toLocaleString()} characters (~${originalTokens.toLocaleString()} tokens).`;\n\n result =\n truncatedStr +\n `\\n\\n[OUTPUT_TRUNCATED: ${truncationInfo} Please use more specific queries or smaller date ranges.]`;\n truncated = true;\n }\n\n const finalLength = result.length;\n const finalTokens = Math.ceil(finalLength / 4);\n\n return {\n content: result,\n toonConverted,\n truncated,\n reduction,\n alreadyToon,\n originalLength,\n originalTokens,\n finalLength,\n finalTokens,\n toonError,\n };\n}\n"],"names":[],"mappings":";;AAAA;;;;;;;AAOG;AAEH;AACA,IAAI,UAAU,GAAuC,IAAI;AACzD,IAAI,iBAAiB,GAAG,KAAK;AAC7B,IAAI,eAAe,GAA4B,IAAI;AAEnD;;;AAGG;AACH,eAAe,eAAe,GAAA;AAC5B,IAAA,IAAI,iBAAiB;QAAE,OAAO,UAAU,KAAK,IAAI;IACjD,iBAAiB,GAAG,IAAI;;AAGxB,IAAA,IAAI;AACF,QAAA,MAAM,GAAG,GAAG,MAAM,OAAO,mBAAmB,CAAC;AAC7C,QAAA,UAAU,GAAG,GAAG,CAAC,MAAM;AACvB,QAAA,OAAO,IAAI;;AACX,IAAA,MAAM;;AAEN,QAAA,IAAI;;AAEF,YAAA,MAAM,GAAG,GAAG,OAAO,CAAC,mBAAmB,CAAC;AACxC,YAAA,UAAU,GAAG,GAAG,CAAC,MAAM;AACvB,YAAA,OAAO,IAAI;;AACX,QAAA,MAAM;;AAEN,YAAA,OAAO,KAAK;;;AAGlB;AAEA;;AAEG;AACI,eAAe,gBAAgB,GAAA;IACpC,IAAI,CAAC,eAAe,EAAE;QACpB,eAAe,GAAG,eAAe,EAAE;;AAErC,IAAA,OAAO,eAAe;AACxB;AAEA;AACA,gBAAgB,EAAE;AAElB;;;;;;;;;;AAUG;AACG,SAAU,YAAY,CAAC,GAAW,EAAA;IACtC,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;AAAE,QAAA,OAAO,KAAK;AAE7D,IAAA,MAAM,OAAO,GAAG,GAAG,CAAC,IAAI,EAAE;;AAG1B,IAAA,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;AAAE,QAAA,OAAO,KAAK;;;;;AAMpE,IAAA,MAAM,YAAY,GAAG;AACnB,QAAA,SAAS;AACT,QAAA,eAAe;AACf,QAAA,iBAAiB;AACjB,QAAA,wBAAwB;KACzB;;AAGD,IAAA,MAAM,cAAc,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;;IAG5E,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IACzD,MAAM,wBAAwB,GAC5B,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC;IAEvD,OAAO,cAAc,IAAI,wBAAwB;AACnD;AAEA;;;;;;AAMG;AACG,SAAU,gBAAgB,CAAC,GAAW,EAAA;IAM1C,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE;AAC/C,QAAA,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;;;IAIrE,MAAM,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC;IACjC,MAAM,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC;IAEjC,IAAI,QAAQ,KAAK,EAAE,IAAI,QAAQ,KAAK,EAAE,EAAE;AACtC,QAAA,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;;AAGrE,IAAA,IAAI,UAAkB;AACtB,IAAA,IAAI,QAAQ,KAAK,EAAE,EAAE;QACnB,UAAU,GAAG,QAAQ;;AAChB,SAAA,IAAI,QAAQ,KAAK,EAAE,EAAE;QAC1B,UAAU,GAAG,QAAQ;;SAChB;QACL,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC;;;;IAK3C,IAAI,KAAK,GAAG,CAAC;IACb,IAAI,QAAQ,GAAG,KAAK;IACpB,IAAI,UAAU,GAAG,KAAK;AAEtB,IAAA,KAAK,IAAI,CAAC,GAAG,UAAU,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;AAC5C,QAAA,MAAM,IAAI,GAAG,GAAG,CAAC,CAAC,CAAC;QAEnB,IAAI,UAAU,EAAE;YACd,UAAU,GAAG,KAAK;YAClB;;AAGF,QAAA,IAAI,IAAI,KAAK,IAAI,IAAI,QAAQ,EAAE;YAC7B,UAAU,GAAG,IAAI;YACjB;;AAGF,QAAA,IAAI,IAAI,KAAK,GAAG,EAAE;YAChB,QAAQ,GAAG,CAAC,QAAQ;YACpB;;AAGF,QAAA,IAAI,QAAQ;YAAE;AAEd,QAAA,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG;AAAE,YAAA,KAAK,EAAE;AACzC,QAAA,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG;AAAE,YAAA,KAAK,EAAE;AAEzC,QAAA,IAAI,KAAK,KAAK,CAAC,EAAE;;AAEf,YAAA,MAAM,OAAO,GAAG,GAAG,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC,GAAG,CAAC,CAAC;AAChD,YAAA,IAAI;gBACF,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;AAClC,gBAAA,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,EAAE;;AACvD,YAAA,MAAM;;;AAGN,gBAAA,MAAM,UAAU,GAAG,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;AACzD,gBAAA,IAAI,UAAU,CAAC,KAAK,EAAE;oBACpB,OAAO;AACL,wBAAA,KAAK,EAAE,IAAI;wBACX,MAAM,EAAE,UAAU,CAAC,MAAM;AACzB,wBAAA,UAAU,EAAE,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,UAAU;AACzC,wBAAA,QAAQ,EAAE,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,QAAQ;qBACtC;;AAEH,gBAAA,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;;;;AAKzE,IAAA,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;AACrE;AAEA;;;;;;;;;;AAUG;AACG,SAAU,UAAU,CAAC,GAAW,EAAA;IAOpC,IAAI,CAAC,UAAU,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE;QAC1C,OAAO;AACL,YAAA,SAAS,EAAE,KAAK;AAChB,YAAA,MAAM,EAAE,GAAG;AACX,YAAA,SAAS,EAAE,CAAC;YACZ,KAAK,EAAE,CAAC,UAAU,GAAG,yBAAyB,GAAG,SAAS;SAC3D;;;AAIH,IAAA,IAAI,YAAY,CAAC,GAAG,CAAC,EAAE;AACrB,QAAA,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE;;AAG3E,IAAA,IAAI;AACF,QAAA,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,gBAAgB,CAAC,GAAG,CAAC;AAErE,QAAA,IAAI,CAAC,KAAK,IAAI,CAAC,MAAM,EAAE;YACrB,OAAO;AACL,gBAAA,SAAS,EAAE,KAAK;AAChB,gBAAA,MAAM,EAAE,GAAG;AACX,gBAAA,SAAS,EAAE,CAAC;AACZ,gBAAA,KAAK,EAAE,0BAA0B;aAClC;;;;AAKH,QAAA,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE;YAC7B,OAAO;AACL,gBAAA,SAAS,EAAE,KAAK;AAChB,gBAAA,MAAM,EAAE,GAAG;AACX,gBAAA,SAAS,EAAE,CAAC;AACZ,gBAAA,KAAK,EACH,iEAAiE;aACpE;;AAGH,QAAA,MAAM,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC;;QAGrC,MAAM,UAAU,GAAG,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,UAAU,CAAC;QAC/C,MAAM,SAAS,GAAG,GAAG,CAAC,SAAS,CAAC,QAAQ,GAAG,CAAC,CAAC;;AAG7C,QAAA,MAAM,UAAU,GAAG,UAAU,GAAG,UAAU,GAAG,SAAS;QAEtD,IAAI,UAAU,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,EAAE;YAClC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAC1B,CAAC,CAAC,GAAG,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CACtD;YACD,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE;;aACpD;;YAEL,OAAO;AACL,gBAAA,SAAS,EAAE,KAAK;AAChB,gBAAA,MAAM,EAAE,GAAG;AACX,gBAAA,SAAS,EAAE,CAAC;gBACZ,KAAK,EAAE,4BAA4B,UAAU,CAAC,MAAM,CAAO,IAAA,EAAA,GAAG,CAAC,MAAM,CAAG,CAAA,CAAA;aACzE;;;IAEH,OAAO,GAAG,EAAE;;AAEZ,QAAA,MAAM,QAAQ,GAAG,GAAG,YAAY,KAAK,GAAG,GAAG,CAAC,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC;QACjE,OAAO;AACL,YAAA,SAAS,EAAE,KAAK;AAChB,YAAA,MAAM,EAAE,GAAG;AACX,YAAA,SAAS,EAAE,CAAC;YACZ,KAAK,EAAE,CAA0B,uBAAA,EAAA,QAAQ,CAAE,CAAA;SAC5C;;AAEL;AAEA;;AAEG;SACa,eAAe,GAAA;IAC7B,OAAO,UAAU,KAAK,IAAI;AAC5B;AA0CA;;;;;;;AAOG;AACH,SAAS,gBAAgB,CAAC,MAAe,EAAA;AACvC,IAAA,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;AAAE,QAAA,OAAO,KAAK;IAEvD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;AACtC,IAAA,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM;;IAGlC,IAAI,iBAAiB,GAAG,CAAC;IAEzB,SAAS,gBAAgB,CAAC,GAAY,EAAA;AACpC,QAAA,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE;;AAE3B,YAAA,iBAAiB,IAAI,GAAG,CAAC,MAAM;;AAC1B,aAAA,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;AAC7B,YAAA,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAC;;AACxB,aAAA,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE;YACzC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC;;;IAIhD,gBAAgB,CAAC,MAAM,CAAC;;AAGxB,IAAA,MAAM,SAAS,GAAG,iBAAiB,GAAG,WAAW;;;IAIjD,OAAO,SAAS,GAAG,GAAG;AACxB;AAEA;;;;;;;;;;;;AAYG;SACa,iBAAiB,CAC/B,OAAe,EACf,UAAoC,EAAE,EAAA;AAEtC,IAAA,MAAM,EACJ,SAAS,GAAG,MAAM,EAClB,UAAU,GAAG,IAAI,EACjB,cAAc,GAAG,IAAI,EACrB,mBAAmB,GAAG,EAAE;AACzB,MAAA,GAAG,OAAO;AAEX,IAAA,MAAM,cAAc,GAAG,OAAO,CAAC,MAAM;IACrC,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;IAEpD,IAAI,MAAM,GAAG,OAAO;IACpB,IAAI,aAAa,GAAG,KAAK;IACzB,IAAI,WAAW,GAAG,KAAK;IACvB,IAAI,SAAS,GAAG,CAAC;IACjB,IAAI,SAAS,GAAG,KAAK;AACrB,IAAA,IAAI,SAA6B;;AAGjC,IAAA,IAAI,YAAY,CAAC,OAAO,CAAC,EAAE;QACzB,WAAW,GAAG,IAAI;;;SAGf,IAAI,UAAU,IAAI,OAAO,CAAC,MAAM,GAAG,cAAc,EAAE;AACtD,QAAA,MAAM,UAAU,GAAG,UAAU,CAAC,OAAO,CAAC;AACtC,QAAA,IAAI,UAAU,CAAC,WAAW,EAAE;YAC1B,WAAW,GAAG,IAAI;;aACb,IACL,UAAU,CAAC,SAAS;AACpB,YAAA,UAAU,CAAC,SAAS,IAAI,mBAAmB,EAC3C;AACA,YAAA,MAAM,GAAG,UAAU,CAAC,MAAM;YAC1B,aAAa,GAAG,IAAI;AACpB,YAAA,SAAS,GAAG,UAAU,CAAC,SAAS;;AAC3B,aAAA,IAAI,UAAU,CAAC,KAAK,EAAE;;AAE3B,YAAA,SAAS,GAAG,UAAU,CAAC,KAAK;;aACvB,IACL,UAAU,CAAC,SAAS;AACpB,YAAA,UAAU,CAAC,SAAS,GAAG,mBAAmB,EAC1C;;YAEA,SAAS,GAAG,kBAAkB,UAAU,CAAC,SAAS,CAAqB,kBAAA,EAAA,mBAAmB,GAAG;;AACxF,aAAA,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE;;YAEhC,SAAS;AACP,gBAAA,UAAU,CAAC,KAAK,IAAI,8CAA8C;;;;AAKxE,IAAA,IAAI,MAAM,CAAC,MAAM,GAAG,SAAS,EAAE;QAC7B,IAAI,YAAY,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,SAAS,CAAC;;AAGjD,QAAA,IAAI,aAAa,IAAI,WAAW,EAAE;;YAEhC,MAAM,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC;AAClD,YAAA,IAAI,WAAW,GAAG,SAAS,GAAG,GAAG,EAAE;gBACjC,YAAY,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,WAAW,CAAC;;;aAElD;;YAEL,MAAM,gBAAgB,GAAG,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC;YACvD,MAAM,aAAa,GAAG,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC;YACpD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,gBAAgB,EAAE,aAAa,CAAC;AAC5D,YAAA,IAAI,UAAU,GAAG,SAAS,GAAG,GAAG,EAAE;gBAChC,YAAY,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,UAAU,GAAG,CAAC,CAAC;;;;QAK5D,MAAM,cAAc,GAAG;AACrB,cAAE,CAAA,SAAA,EAAY,cAAc,CAAC,cAAc,EAAE,CAAA,cAAA,EAAiB,MAAM,CAAC,MAAM,CAAC,cAAc,EAAE,CAAA,QAAA,EAAW,SAAS,CAAY,UAAA,CAAA;AAC1H,gBAAA,CAAA,cAAA,EAAiB,SAAS,CAAC,cAAc,EAAE,CAAc,YAAA;AAC3D,cAAE,CAAA,oBAAA,EAAuB,cAAc,CAAC,cAAc,EAAE,CAAiB,cAAA,EAAA,cAAc,CAAC,cAAc,EAAE,CAAA,SAAA,CAAW;QAErH,MAAM;YACJ,YAAY;gBACZ,CAA0B,uBAAA,EAAA,cAAc,4DAA4D;QACtG,SAAS,GAAG,IAAI;;AAGlB,IAAA,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM;IACjC,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC;IAE9C,OAAO;AACL,QAAA,OAAO,EAAE,MAAM;QACf,aAAa;QACb,SAAS;QACT,SAAS;QACT,WAAW;QACX,cAAc;QACd,cAAc;QACd,WAAW;QACX,WAAW;QACX,SAAS;KACV;AACH;;;;;;;;;"}
|
|
1
|
+
{"version":3,"file":"toonFormat.cjs","sources":["../../../src/utils/toonFormat.ts"],"sourcesContent":["/**\n * TOON Format Utility\n *\n * Provides JSON to TOON conversion for token-efficient encoding.\n * TOON format can reduce token count by 30-60% for JSON data.\n *\n * Uses the official @toon-format/toon library when available.\n */\n\n// Dynamic import holder for the TOON library (ESM)\nlet toonEncode: ((data: unknown) => string) | null = null;\nlet toonLoadAttempted = false;\nlet toonLoadPromise: Promise<boolean> | null = null;\n\n/**\n * Lazily loads the TOON library\n * Tries both ESM dynamic import and CommonJS require for compatibility\n */\nasync function loadToonLibrary(): Promise<boolean> {\n if (toonLoadAttempted) return toonEncode !== null;\n toonLoadAttempted = true;\n\n // Try ESM dynamic import first\n try {\n const mod = await import('@toon-format/toon');\n toonEncode = mod.encode;\n return true;\n } catch {\n // ESM import failed, try CommonJS require\n try {\n // eslint-disable-next-line @typescript-eslint/no-require-imports\n const mod = require('@toon-format/toon');\n toonEncode = mod.encode;\n return true;\n } catch {\n // Library not available - that's OK, we'll return original content\n return false;\n }\n }\n}\n\n/**\n * Ensures the TOON library is loaded. Call this before using jsonToToon in tests.\n */\nexport async function ensureToonLoaded(): Promise<boolean> {\n if (!toonLoadPromise) {\n toonLoadPromise = loadToonLibrary();\n }\n return toonLoadPromise;\n}\n\n// Start loading immediately\nensureToonLoaded();\n\n/**\n * Check if a string appears to be in TOON format.\n * TOON format characteristics:\n * - Uses key: value syntax (like YAML)\n * - Uses array notation like [3] or {fields}\n * - Does NOT start with { or [\n * - Often has patterns like \"name[N]:\" or \"name{fields}:\"\n *\n * @param str - String to check\n * @returns true if string appears to be TOON format\n */\nexport function isToonFormat(str: string): boolean {\n if (typeof str !== 'string' || str.length === 0) return false;\n\n const trimmed = str.trim();\n\n // TOON doesn't start with JSON brackets\n if (trimmed.startsWith('{') || trimmed.startsWith('[')) return false;\n\n // Check for TOON-specific patterns:\n // 1. Key-value with colon (but not URL-like patterns)\n // 2. Array notation like \"items[3]:\" or \"data[10]:\"\n // 3. Object schema notation like \"items{id,name}:\"\n const toonPatterns = [\n /^\\w+:$/m, // Simple key: at start of line\n /^\\w+\\[\\d+\\]:/m, // Array notation: items[3]:\n /^\\w+\\{[^}]+\\}:/m, // Schema notation: items{id,name}:\n /^\\w+\\[\\d+\\]\\{[^}]+\\}:/m, // Combined: items[3]{id,name}:\n ];\n\n // Must match at least one TOON pattern\n const hasToonPattern = toonPatterns.some((pattern) => pattern.test(trimmed));\n\n // Additional check: TOON typically has multiple lines with consistent indentation\n const lines = trimmed.split('\\n').filter((l) => l.trim());\n const hasMultipleKeyValueLines =\n lines.filter((l) => /^\\s*\\w+.*:/.test(l)).length >= 2;\n\n return hasToonPattern || hasMultipleKeyValueLines;\n}\n\n/**\n * Extract the first valid JSON object or array from a string.\n * Handles pure JSON or JSON embedded in text (e.g., tool responses).\n *\n * @param str - The string to extract JSON from\n * @returns Object with found status, parsed JSON, and indices\n */\nexport function extractFirstJson(str: string): {\n found: boolean;\n parsed: unknown;\n startIndex: number;\n endIndex: number;\n} {\n if (typeof str !== 'string' || str.length === 0) {\n return { found: false, parsed: null, startIndex: -1, endIndex: -1 };\n }\n\n // Find the first { or [ character\n const objStart = str.indexOf('{');\n const arrStart = str.indexOf('[');\n\n if (objStart === -1 && arrStart === -1) {\n return { found: false, parsed: null, startIndex: -1, endIndex: -1 };\n }\n\n let startIndex: number;\n if (objStart === -1) {\n startIndex = arrStart;\n } else if (arrStart === -1) {\n startIndex = objStart;\n } else {\n startIndex = Math.min(objStart, arrStart);\n }\n\n // Find the matching closing bracket using a stack-based approach\n // Properly handle strings to avoid counting brackets inside strings\n let depth = 0;\n let inString = false;\n let escapeNext = false;\n\n for (let i = startIndex; i < str.length; i++) {\n const char = str[i];\n\n if (escapeNext) {\n escapeNext = false;\n continue;\n }\n\n if (char === '\\\\' && inString) {\n escapeNext = true;\n continue;\n }\n\n if (char === '\"') {\n inString = !inString;\n continue;\n }\n\n if (inString) continue;\n\n if (char === '{' || char === '[') depth++;\n if (char === '}' || char === ']') depth--;\n\n if (depth === 0) {\n // Found matching bracket, try to parse\n const jsonStr = str.substring(startIndex, i + 1);\n try {\n const parsed = JSON.parse(jsonStr);\n return { found: true, parsed, startIndex, endIndex: i };\n } catch {\n // Not valid JSON at this bracket level, continue searching\n // Reset and look for the next JSON start after this position\n const nextResult = extractFirstJson(str.substring(i + 1));\n if (nextResult.found) {\n return {\n found: true,\n parsed: nextResult.parsed,\n startIndex: i + 1 + nextResult.startIndex,\n endIndex: i + 1 + nextResult.endIndex,\n };\n }\n return { found: false, parsed: null, startIndex: -1, endIndex: -1 };\n }\n }\n }\n\n return { found: false, parsed: null, startIndex: -1, endIndex: -1 };\n}\n\n/**\n * Convert JSON content to TOON format for token efficiency.\n * Extracts JSON from string if embedded, converts to TOON.\n * Returns original string if:\n * - Already in TOON format\n * - Not JSON\n * - TOON conversion fails or is larger\n *\n * @param str - The string containing JSON to convert\n * @returns Object with conversion status, result, and reduction percentage\n */\nexport function jsonToToon(str: string): {\n converted: boolean;\n result: string;\n reduction: number;\n alreadyToon?: boolean;\n error?: string;\n} {\n if (!toonEncode || typeof str !== 'string') {\n return {\n converted: false,\n result: str,\n reduction: 0,\n error: !toonEncode ? 'TOON library not loaded' : undefined,\n };\n }\n\n // Check if already in TOON format - skip conversion\n if (isToonFormat(str)) {\n return { converted: false, result: str, reduction: 0, alreadyToon: true };\n }\n\n try {\n const { found, parsed, startIndex, endIndex } = extractFirstJson(str);\n\n if (!found || !parsed) {\n return {\n converted: false,\n result: str,\n reduction: 0,\n error: 'No JSON found in content',\n };\n }\n\n // Check if this JSON structure would benefit from TOON\n // Text-heavy content (emails, documents) won't compress well\n if (!isToonBeneficial(parsed)) {\n return {\n converted: false,\n result: str,\n reduction: 0,\n error:\n 'Content is text-heavy (>70% string values), TOON not beneficial',\n };\n }\n\n const toonResult = toonEncode(parsed);\n\n // Preserve text before and after the JSON block\n const textBefore = str.substring(0, startIndex);\n const textAfter = str.substring(endIndex + 1);\n\n // Build the full result: textBefore + TOON + textAfter\n const fullResult = textBefore + toonResult + textAfter;\n\n if (fullResult.length < str.length) {\n const reduction = Math.round(\n ((str.length - fullResult.length) / str.length) * 100\n );\n return { converted: true, result: fullResult, reduction };\n } else {\n // TOON output was larger or same size - not beneficial\n return {\n converted: false,\n result: str,\n reduction: 0,\n error: `TOON output not smaller (${fullResult.length} >= ${str.length})`,\n };\n }\n } catch (err) {\n // TOON encoding or extraction failed - log error for debugging\n const errorMsg = err instanceof Error ? err.message : String(err);\n return {\n converted: false,\n result: str,\n reduction: 0,\n error: `TOON conversion error: ${errorMsg}`,\n };\n }\n}\n\n/**\n * Check if TOON library is loaded and available\n */\nexport function isToonAvailable(): boolean {\n return toonEncode !== null;\n}\n\n/**\n * Result of processing tool output\n */\nexport interface ProcessToolOutputResult {\n /** The processed content string */\n content: string;\n /** Whether TOON conversion was applied */\n toonConverted: boolean;\n /** Whether content was truncated */\n truncated: boolean;\n /** Percentage reduction from TOON (0 if not converted) */\n reduction: number;\n /** Whether input was already in TOON format */\n alreadyToon: boolean;\n /** Original content length */\n originalLength: number;\n /** Estimated original tokens (~4 chars per token) */\n originalTokens: number;\n /** Final content length */\n finalLength: number;\n /** Estimated final tokens */\n finalTokens: number;\n /** Error message if TOON conversion failed (for debugging) */\n toonError?: string;\n}\n\n/**\n * Options for processing tool output\n */\nexport interface ProcessToolOutputOptions {\n /** Maximum output length in characters (default: 100000) */\n maxLength?: number;\n /** Whether to apply TOON conversion (default: true) */\n enableToon?: boolean;\n /** Minimum content size to attempt TOON conversion (default: 1000) */\n minSizeForToon?: number;\n /** Minimum reduction % to accept TOON result (default: 10) */\n minReductionPercent?: number;\n}\n\n/**\n * Analyze JSON structure to determine if TOON conversion would be beneficial.\n * Text-heavy content (emails, documents) doesn't compress well with TOON.\n * Structured data (API responses, metadata) compresses much better.\n *\n * @param parsed - Parsed JSON object\n * @returns true if TOON conversion is likely beneficial\n */\nfunction isToonBeneficial(parsed: unknown): boolean {\n if (!parsed || typeof parsed !== 'object') return false;\n\n const jsonStr = JSON.stringify(parsed);\n const totalLength = jsonStr.length;\n\n // Count characters in string values (text content)\n let textContentLength = 0;\n\n function countTextContent(obj: unknown): void {\n if (typeof obj === 'string') {\n // Count string length (this is text content that TOON can't compress)\n textContentLength += obj.length;\n } else if (Array.isArray(obj)) {\n obj.forEach(countTextContent);\n } else if (obj && typeof obj === 'object') {\n Object.values(obj).forEach(countTextContent);\n }\n }\n\n countTextContent(parsed);\n\n // Calculate ratio of text content to total JSON\n const textRatio = textContentLength / totalLength;\n\n // If more than 70% is text content, TOON won't help much\n // TOON compresses structure (field names, brackets), not text\n return textRatio < 0.7;\n}\n\n/**\n * Process tool output: apply TOON conversion if beneficial, then truncate if needed.\n * This is the main entry point for processing any tool output (regular or MCP).\n *\n * Flow:\n * 1. Check if already TOON format → skip conversion\n * 2. Try TOON conversion if content is JSON and large enough\n * 3. Truncate if still exceeds maxLength (with smart break points)\n *\n * @param content - The tool output content\n * @param options - Processing options\n * @returns Processed content and metadata\n */\nexport function processToolOutput(\n content: string,\n options: ProcessToolOutputOptions = {}\n): ProcessToolOutputResult {\n const {\n maxLength = 100000,\n enableToon = true,\n minSizeForToon = 1000,\n minReductionPercent = 10, // Increased from 5% - only apply TOON when clearly beneficial\n } = options;\n\n const originalLength = content.length;\n const originalTokens = Math.ceil(originalLength / 4);\n\n let result = content;\n let toonConverted = false;\n let alreadyToon = false;\n let reduction = 0;\n let truncated = false;\n let toonError: string | undefined;\n\n // Step 1: Check if already TOON format\n if (isToonFormat(content)) {\n alreadyToon = true;\n }\n // Step 2: Apply TOON conversion if enabled and content is large enough\n else if (enableToon && content.length > minSizeForToon) {\n const toonResult = jsonToToon(content);\n if (toonResult.alreadyToon) {\n alreadyToon = true;\n } else if (\n toonResult.converted &&\n toonResult.reduction >= minReductionPercent\n ) {\n result = toonResult.result;\n toonConverted = true;\n reduction = toonResult.reduction;\n } else if (toonResult.error) {\n // Track error for debugging\n toonError = toonResult.error;\n } else if (\n toonResult.converted &&\n toonResult.reduction < minReductionPercent\n ) {\n // TOON converted but reduction was too small to be worth it\n toonError = `TOON reduction ${toonResult.reduction}% below threshold ${minReductionPercent}%`;\n } else if (!toonResult.converted) {\n // Conversion failed without explicit error - investigate\n toonError =\n toonResult.error || 'TOON conversion returned false with no error';\n }\n }\n\n // Step 3: Truncate if still too long (with smart break points)\n if (result.length > maxLength) {\n let truncatedStr = result.substring(0, maxLength);\n\n // Try to find a clean break point\n if (toonConverted || alreadyToon) {\n // For TOON format, break at newline for cleaner output\n const lastNewline = truncatedStr.lastIndexOf('\\n');\n if (lastNewline > maxLength * 0.7) {\n truncatedStr = truncatedStr.substring(0, lastNewline);\n }\n } else {\n // For JSON, try to find a clean JSON break point\n const lastCompleteItem = truncatedStr.lastIndexOf('},');\n const lastArrayItem = truncatedStr.lastIndexOf('],');\n const breakPoint = Math.max(lastCompleteItem, lastArrayItem);\n if (breakPoint > maxLength * 0.5) {\n truncatedStr = truncatedStr.substring(0, breakPoint + 1);\n }\n }\n\n // Build truncation message\n const truncationInfo = toonConverted\n ? `Original ${originalLength.toLocaleString()} chars → TOON ${result.length.toLocaleString()} chars (${reduction}% saved). ` +\n `Still exceeds ${maxLength.toLocaleString()} char limit.`\n : `Original output was ${originalLength.toLocaleString()} characters (~${originalTokens.toLocaleString()} tokens).`;\n\n result =\n truncatedStr +\n `\\n\\n[OUTPUT_TRUNCATED: ${truncationInfo} Please use more specific queries or smaller date ranges.]`;\n truncated = true;\n }\n\n const finalLength = result.length;\n const finalTokens = Math.ceil(finalLength / 4);\n\n return {\n content: result,\n toonConverted,\n truncated,\n reduction,\n alreadyToon,\n originalLength,\n originalTokens,\n finalLength,\n finalTokens,\n toonError,\n };\n}\n"],"names":[],"mappings":";;AAAA;;;;;;;AAOG;AAEH;AACA,IAAI,UAAU,GAAuC,IAAI;AACzD,IAAI,iBAAiB,GAAG,KAAK;AAC7B,IAAI,eAAe,GAA4B,IAAI;AAEnD;;;AAGG;AACH,eAAe,eAAe,GAAA;AAC5B,IAAA,IAAI,iBAAiB;QAAE,OAAO,UAAU,KAAK,IAAI;IACjD,iBAAiB,GAAG,IAAI;;AAGxB,IAAA,IAAI;AACF,QAAA,MAAM,GAAG,GAAG,MAAM,OAAO,mBAAmB,CAAC;AAC7C,QAAA,UAAU,GAAG,GAAG,CAAC,MAAM;AACvB,QAAA,OAAO,IAAI;IACb;AAAE,IAAA,MAAM;;AAEN,QAAA,IAAI;;AAEF,YAAA,MAAM,GAAG,GAAG,OAAO,CAAC,mBAAmB,CAAC;AACxC,YAAA,UAAU,GAAG,GAAG,CAAC,MAAM;AACvB,YAAA,OAAO,IAAI;QACb;AAAE,QAAA,MAAM;;AAEN,YAAA,OAAO,KAAK;QACd;IACF;AACF;AAEA;;AAEG;AACI,eAAe,gBAAgB,GAAA;IACpC,IAAI,CAAC,eAAe,EAAE;QACpB,eAAe,GAAG,eAAe,EAAE;IACrC;AACA,IAAA,OAAO,eAAe;AACxB;AAEA;AACA,gBAAgB,EAAE;AAElB;;;;;;;;;;AAUG;AACG,SAAU,YAAY,CAAC,GAAW,EAAA;IACtC,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;AAAE,QAAA,OAAO,KAAK;AAE7D,IAAA,MAAM,OAAO,GAAG,GAAG,CAAC,IAAI,EAAE;;AAG1B,IAAA,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;AAAE,QAAA,OAAO,KAAK;;;;;AAMpE,IAAA,MAAM,YAAY,GAAG;AACnB,QAAA,SAAS;AACT,QAAA,eAAe;AACf,QAAA,iBAAiB;AACjB,QAAA,wBAAwB;KACzB;;AAGD,IAAA,MAAM,cAAc,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;;IAG5E,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IACzD,MAAM,wBAAwB,GAC5B,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC;IAEvD,OAAO,cAAc,IAAI,wBAAwB;AACnD;AAEA;;;;;;AAMG;AACG,SAAU,gBAAgB,CAAC,GAAW,EAAA;IAM1C,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE;AAC/C,QAAA,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;IACrE;;IAGA,MAAM,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC;IACjC,MAAM,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC;IAEjC,IAAI,QAAQ,KAAK,EAAE,IAAI,QAAQ,KAAK,EAAE,EAAE;AACtC,QAAA,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;IACrE;AAEA,IAAA,IAAI,UAAkB;AACtB,IAAA,IAAI,QAAQ,KAAK,EAAE,EAAE;QACnB,UAAU,GAAG,QAAQ;IACvB;AAAO,SAAA,IAAI,QAAQ,KAAK,EAAE,EAAE;QAC1B,UAAU,GAAG,QAAQ;IACvB;SAAO;QACL,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC3C;;;IAIA,IAAI,KAAK,GAAG,CAAC;IACb,IAAI,QAAQ,GAAG,KAAK;IACpB,IAAI,UAAU,GAAG,KAAK;AAEtB,IAAA,KAAK,IAAI,CAAC,GAAG,UAAU,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;AAC5C,QAAA,MAAM,IAAI,GAAG,GAAG,CAAC,CAAC,CAAC;QAEnB,IAAI,UAAU,EAAE;YACd,UAAU,GAAG,KAAK;YAClB;QACF;AAEA,QAAA,IAAI,IAAI,KAAK,IAAI,IAAI,QAAQ,EAAE;YAC7B,UAAU,GAAG,IAAI;YACjB;QACF;AAEA,QAAA,IAAI,IAAI,KAAK,GAAG,EAAE;YAChB,QAAQ,GAAG,CAAC,QAAQ;YACpB;QACF;AAEA,QAAA,IAAI,QAAQ;YAAE;AAEd,QAAA,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG;AAAE,YAAA,KAAK,EAAE;AACzC,QAAA,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG;AAAE,YAAA,KAAK,EAAE;AAEzC,QAAA,IAAI,KAAK,KAAK,CAAC,EAAE;;AAEf,YAAA,MAAM,OAAO,GAAG,GAAG,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC,GAAG,CAAC,CAAC;AAChD,YAAA,IAAI;gBACF,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;AAClC,gBAAA,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,EAAE;YACzD;AAAE,YAAA,MAAM;;;AAGN,gBAAA,MAAM,UAAU,GAAG,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;AACzD,gBAAA,IAAI,UAAU,CAAC,KAAK,EAAE;oBACpB,OAAO;AACL,wBAAA,KAAK,EAAE,IAAI;wBACX,MAAM,EAAE,UAAU,CAAC,MAAM;AACzB,wBAAA,UAAU,EAAE,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,UAAU;AACzC,wBAAA,QAAQ,EAAE,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,QAAQ;qBACtC;gBACH;AACA,gBAAA,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;YACrE;QACF;IACF;AAEA,IAAA,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;AACrE;AAEA;;;;;;;;;;AAUG;AACG,SAAU,UAAU,CAAC,GAAW,EAAA;IAOpC,IAAI,CAAC,UAAU,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE;QAC1C,OAAO;AACL,YAAA,SAAS,EAAE,KAAK;AAChB,YAAA,MAAM,EAAE,GAAG;AACX,YAAA,SAAS,EAAE,CAAC;YACZ,KAAK,EAAE,CAAC,UAAU,GAAG,yBAAyB,GAAG,SAAS;SAC3D;IACH;;AAGA,IAAA,IAAI,YAAY,CAAC,GAAG,CAAC,EAAE;AACrB,QAAA,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE;IAC3E;AAEA,IAAA,IAAI;AACF,QAAA,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,gBAAgB,CAAC,GAAG,CAAC;AAErE,QAAA,IAAI,CAAC,KAAK,IAAI,CAAC,MAAM,EAAE;YACrB,OAAO;AACL,gBAAA,SAAS,EAAE,KAAK;AAChB,gBAAA,MAAM,EAAE,GAAG;AACX,gBAAA,SAAS,EAAE,CAAC;AACZ,gBAAA,KAAK,EAAE,0BAA0B;aAClC;QACH;;;AAIA,QAAA,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE;YAC7B,OAAO;AACL,gBAAA,SAAS,EAAE,KAAK;AAChB,gBAAA,MAAM,EAAE,GAAG;AACX,gBAAA,SAAS,EAAE,CAAC;AACZ,gBAAA,KAAK,EACH,iEAAiE;aACpE;QACH;AAEA,QAAA,MAAM,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC;;QAGrC,MAAM,UAAU,GAAG,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,UAAU,CAAC;QAC/C,MAAM,SAAS,GAAG,GAAG,CAAC,SAAS,CAAC,QAAQ,GAAG,CAAC,CAAC;;AAG7C,QAAA,MAAM,UAAU,GAAG,UAAU,GAAG,UAAU,GAAG,SAAS;QAEtD,IAAI,UAAU,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,EAAE;YAClC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAC1B,CAAC,CAAC,GAAG,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CACtD;YACD,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE;QAC3D;aAAO;;YAEL,OAAO;AACL,gBAAA,SAAS,EAAE,KAAK;AAChB,gBAAA,MAAM,EAAE,GAAG;AACX,gBAAA,SAAS,EAAE,CAAC;gBACZ,KAAK,EAAE,4BAA4B,UAAU,CAAC,MAAM,CAAA,IAAA,EAAO,GAAG,CAAC,MAAM,CAAA,CAAA,CAAG;aACzE;QACH;IACF;IAAE,OAAO,GAAG,EAAE;;AAEZ,QAAA,MAAM,QAAQ,GAAG,GAAG,YAAY,KAAK,GAAG,GAAG,CAAC,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC;QACjE,OAAO;AACL,YAAA,SAAS,EAAE,KAAK;AAChB,YAAA,MAAM,EAAE,GAAG;AACX,YAAA,SAAS,EAAE,CAAC;YACZ,KAAK,EAAE,CAAA,uBAAA,EAA0B,QAAQ,CAAA,CAAE;SAC5C;IACH;AACF;AAEA;;AAEG;SACa,eAAe,GAAA;IAC7B,OAAO,UAAU,KAAK,IAAI;AAC5B;AA0CA;;;;;;;AAOG;AACH,SAAS,gBAAgB,CAAC,MAAe,EAAA;AACvC,IAAA,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;AAAE,QAAA,OAAO,KAAK;IAEvD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;AACtC,IAAA,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM;;IAGlC,IAAI,iBAAiB,GAAG,CAAC;IAEzB,SAAS,gBAAgB,CAAC,GAAY,EAAA;AACpC,QAAA,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE;;AAE3B,YAAA,iBAAiB,IAAI,GAAG,CAAC,MAAM;QACjC;AAAO,aAAA,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE;AAC7B,YAAA,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAC;QAC/B;AAAO,aAAA,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE;YACzC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC;QAC9C;IACF;IAEA,gBAAgB,CAAC,MAAM,CAAC;;AAGxB,IAAA,MAAM,SAAS,GAAG,iBAAiB,GAAG,WAAW;;;IAIjD,OAAO,SAAS,GAAG,GAAG;AACxB;AAEA;;;;;;;;;;;;AAYG;SACa,iBAAiB,CAC/B,OAAe,EACf,UAAoC,EAAE,EAAA;AAEtC,IAAA,MAAM,EACJ,SAAS,GAAG,MAAM,EAClB,UAAU,GAAG,IAAI,EACjB,cAAc,GAAG,IAAI,EACrB,mBAAmB,GAAG,EAAE;AACzB,MAAA,GAAG,OAAO;AAEX,IAAA,MAAM,cAAc,GAAG,OAAO,CAAC,MAAM;IACrC,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;IAEpD,IAAI,MAAM,GAAG,OAAO;IACpB,IAAI,aAAa,GAAG,KAAK;IACzB,IAAI,WAAW,GAAG,KAAK;IACvB,IAAI,SAAS,GAAG,CAAC;IACjB,IAAI,SAAS,GAAG,KAAK;AACrB,IAAA,IAAI,SAA6B;;AAGjC,IAAA,IAAI,YAAY,CAAC,OAAO,CAAC,EAAE;QACzB,WAAW,GAAG,IAAI;IACpB;;SAEK,IAAI,UAAU,IAAI,OAAO,CAAC,MAAM,GAAG,cAAc,EAAE;AACtD,QAAA,MAAM,UAAU,GAAG,UAAU,CAAC,OAAO,CAAC;AACtC,QAAA,IAAI,UAAU,CAAC,WAAW,EAAE;YAC1B,WAAW,GAAG,IAAI;QACpB;aAAO,IACL,UAAU,CAAC,SAAS;AACpB,YAAA,UAAU,CAAC,SAAS,IAAI,mBAAmB,EAC3C;AACA,YAAA,MAAM,GAAG,UAAU,CAAC,MAAM;YAC1B,aAAa,GAAG,IAAI;AACpB,YAAA,SAAS,GAAG,UAAU,CAAC,SAAS;QAClC;AAAO,aAAA,IAAI,UAAU,CAAC,KAAK,EAAE;;AAE3B,YAAA,SAAS,GAAG,UAAU,CAAC,KAAK;QAC9B;aAAO,IACL,UAAU,CAAC,SAAS;AACpB,YAAA,UAAU,CAAC,SAAS,GAAG,mBAAmB,EAC1C;;YAEA,SAAS,GAAG,kBAAkB,UAAU,CAAC,SAAS,CAAA,kBAAA,EAAqB,mBAAmB,GAAG;QAC/F;AAAO,aAAA,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE;;YAEhC,SAAS;AACP,gBAAA,UAAU,CAAC,KAAK,IAAI,8CAA8C;QACtE;IACF;;AAGA,IAAA,IAAI,MAAM,CAAC,MAAM,GAAG,SAAS,EAAE;QAC7B,IAAI,YAAY,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,SAAS,CAAC;;AAGjD,QAAA,IAAI,aAAa,IAAI,WAAW,EAAE;;YAEhC,MAAM,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC;AAClD,YAAA,IAAI,WAAW,GAAG,SAAS,GAAG,GAAG,EAAE;gBACjC,YAAY,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,WAAW,CAAC;YACvD;QACF;aAAO;;YAEL,MAAM,gBAAgB,GAAG,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC;YACvD,MAAM,aAAa,GAAG,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC;YACpD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,gBAAgB,EAAE,aAAa,CAAC;AAC5D,YAAA,IAAI,UAAU,GAAG,SAAS,GAAG,GAAG,EAAE;gBAChC,YAAY,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,UAAU,GAAG,CAAC,CAAC;YAC1D;QACF;;QAGA,MAAM,cAAc,GAAG;AACrB,cAAE,CAAA,SAAA,EAAY,cAAc,CAAC,cAAc,EAAE,CAAA,cAAA,EAAiB,MAAM,CAAC,MAAM,CAAC,cAAc,EAAE,CAAA,QAAA,EAAW,SAAS,CAAA,UAAA,CAAY;AAC1H,gBAAA,CAAA,cAAA,EAAiB,SAAS,CAAC,cAAc,EAAE,CAAA,YAAA;AAC7C,cAAE,CAAA,oBAAA,EAAuB,cAAc,CAAC,cAAc,EAAE,CAAA,cAAA,EAAiB,cAAc,CAAC,cAAc,EAAE,CAAA,SAAA,CAAW;QAErH,MAAM;YACJ,YAAY;gBACZ,CAAA,uBAAA,EAA0B,cAAc,4DAA4D;QACtG,SAAS,GAAG,IAAI;IAClB;AAEA,IAAA,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM;IACjC,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC;IAE9C,OAAO;AACL,QAAA,OAAO,EAAE,MAAM;QACf,aAAa;QACb,SAAS;QACT,SAAS;QACT,WAAW;QACX,cAAc;QACd,cAAc;QACd,WAAW;QACX,WAAW;QACX,SAAS;KACV;AACH;;;;;;;;;"}
|
|
@@ -14,14 +14,15 @@ class AgentContext {
|
|
|
14
14
|
* Create an AgentContext from configuration with token accounting initialization
|
|
15
15
|
*/
|
|
16
16
|
static fromConfig(agentConfig, tokenCounter, indexTokenCountMap) {
|
|
17
|
-
const { agentId, name, description, provider, clientOptions, tools, toolMap, toolEnd, toolRegistry, toolDefinitions, instructions, additional_instructions, streamBuffer, maxContextTokens, reasoningKey, useLegacyContent, dynamicContext, structuredOutput: structuredOutputCamel, structured_output: structuredOutputSnake, discoveredTools, summarizeCallback, } = agentConfig;
|
|
17
|
+
const { agentId, name, description, provider, clientOptions, tools, toolMap, toolEnd, toolRegistry, toolDefinitions, instructions, additional_instructions, streamBuffer, maxContextTokens, reasoningKey, useLegacyContent, dynamicContext, structuredOutput: structuredOutputCamel, structured_output: structuredOutputSnake, discoveredTools, summarizeCallback, persistedSummary, } = agentConfig;
|
|
18
18
|
// Normalize structured output: support both camelCase and snake_case inputs
|
|
19
19
|
// Priority: structuredOutput (camelCase) > structured_output (snake_case with enabled check)
|
|
20
20
|
let structuredOutput;
|
|
21
21
|
if (structuredOutputCamel) {
|
|
22
22
|
structuredOutput = structuredOutputCamel;
|
|
23
23
|
}
|
|
24
|
-
else if (structuredOutputSnake?.enabled
|
|
24
|
+
else if (structuredOutputSnake?.enabled === true &&
|
|
25
|
+
structuredOutputSnake.schema != null) {
|
|
25
26
|
// Convert snake_case input to StructuredOutputConfig
|
|
26
27
|
structuredOutput = {
|
|
27
28
|
schema: structuredOutputSnake.schema,
|
|
@@ -53,16 +54,16 @@ class AgentContext {
|
|
|
53
54
|
dynamicContext,
|
|
54
55
|
structuredOutput,
|
|
55
56
|
discoveredTools,
|
|
57
|
+
summarizeCallback,
|
|
58
|
+
persistedSummary,
|
|
56
59
|
});
|
|
57
|
-
if (summarizeCallback) {
|
|
58
|
-
agentContext.summarizeCallback = summarizeCallback;
|
|
59
|
-
}
|
|
60
60
|
if (tokenCounter) {
|
|
61
61
|
// Initialize system runnable BEFORE async tool token calculation
|
|
62
62
|
// This ensures system message tokens are in instructionTokens before
|
|
63
63
|
// updateTokenMapWithInstructions is called
|
|
64
64
|
agentContext.initializeSystemRunnable();
|
|
65
65
|
const tokenMap = indexTokenCountMap || {};
|
|
66
|
+
agentContext.baseIndexTokenCountMap = { ...tokenMap };
|
|
66
67
|
agentContext.indexTokenCountMap = tokenMap;
|
|
67
68
|
agentContext.tokenCalculationPromise = agentContext
|
|
68
69
|
.calculateInstructionTokens(tokenCounter)
|
|
@@ -75,6 +76,7 @@ class AgentContext {
|
|
|
75
76
|
});
|
|
76
77
|
}
|
|
77
78
|
else if (indexTokenCountMap) {
|
|
79
|
+
agentContext.baseIndexTokenCountMap = { ...indexTokenCountMap };
|
|
78
80
|
agentContext.indexTokenCountMap = indexTokenCountMap;
|
|
79
81
|
}
|
|
80
82
|
return agentContext;
|
|
@@ -91,6 +93,8 @@ class AgentContext {
|
|
|
91
93
|
clientOptions;
|
|
92
94
|
/** Token count map indexed by message position */
|
|
93
95
|
indexTokenCountMap = {};
|
|
96
|
+
/** Canonical pre-run token map used to restore token accounting on reset */
|
|
97
|
+
baseIndexTokenCountMap = {};
|
|
94
98
|
/** Maximum context tokens for this agent */
|
|
95
99
|
maxContextTokens;
|
|
96
100
|
/** Current usage metadata for this agent */
|
|
@@ -107,6 +111,8 @@ class AgentContext {
|
|
|
107
111
|
lastStreamCall;
|
|
108
112
|
/** Tools available to this agent */
|
|
109
113
|
tools;
|
|
114
|
+
/** Graph-managed tools (e.g., handoff tools created by MultiAgentGraph) that bypass event-driven dispatch */
|
|
115
|
+
graphTools;
|
|
110
116
|
/** Tool map for this agent */
|
|
111
117
|
toolMap;
|
|
112
118
|
/**
|
|
@@ -137,6 +143,8 @@ class AgentContext {
|
|
|
137
143
|
lastToken;
|
|
138
144
|
/** Token type switch state */
|
|
139
145
|
tokenTypeSwitch;
|
|
146
|
+
/** Tracks how many reasoning→text transitions have occurred (ensures unique post-reasoning step keys) */
|
|
147
|
+
reasoningTransitionCount = 0;
|
|
140
148
|
/** Current token type being processed */
|
|
141
149
|
currentTokenType = ContentTypes.TEXT;
|
|
142
150
|
/** Whether tools should end the workflow */
|
|
@@ -175,12 +183,11 @@ class AgentContext {
|
|
|
175
183
|
* instead of streaming text.
|
|
176
184
|
*/
|
|
177
185
|
structuredOutput;
|
|
178
|
-
/**
|
|
179
|
-
* Optional callback for summarizing messages that were pruned from context.
|
|
180
|
-
* When provided, discarded messages are summarized instead of silently dropped.
|
|
181
|
-
*/
|
|
186
|
+
/** Optional callback to summarize discarded messages during context pruning */
|
|
182
187
|
summarizeCallback;
|
|
183
|
-
|
|
188
|
+
/** Pre-existing summary loaded from persistent storage, injected into context on new turns */
|
|
189
|
+
persistedSummary;
|
|
190
|
+
constructor({ agentId, name, description, provider, clientOptions, maxContextTokens, streamBuffer, tokenCounter, tools, toolMap, toolRegistry, toolDefinitions, instructions, additionalInstructions, dynamicContext, reasoningKey, toolEnd, instructionTokens, useLegacyContent, structuredOutput, discoveredTools, summarizeCallback, persistedSummary, }) {
|
|
184
191
|
this.agentId = agentId;
|
|
185
192
|
this.name = name;
|
|
186
193
|
this.description = description;
|
|
@@ -197,6 +204,8 @@ class AgentContext {
|
|
|
197
204
|
this.additionalInstructions = additionalInstructions;
|
|
198
205
|
this.dynamicContext = dynamicContext;
|
|
199
206
|
this.structuredOutput = structuredOutput;
|
|
207
|
+
this.summarizeCallback = summarizeCallback;
|
|
208
|
+
this.persistedSummary = persistedSummary;
|
|
200
209
|
if (reasoningKey) {
|
|
201
210
|
this.reasoningKey = reasoningKey;
|
|
202
211
|
}
|
|
@@ -219,7 +228,10 @@ class AgentContext {
|
|
|
219
228
|
* and return a validated JSON response.
|
|
220
229
|
*/
|
|
221
230
|
get isStructuredOutputMode() {
|
|
222
|
-
|
|
231
|
+
// Runtime safety: schema can be null/undefined via API despite type saying required
|
|
232
|
+
return (this.structuredOutput != null &&
|
|
233
|
+
// eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
|
|
234
|
+
this.structuredOutput.schema != null);
|
|
223
235
|
}
|
|
224
236
|
/**
|
|
225
237
|
* Gets the structured output schema with normalized defaults.
|
|
@@ -231,15 +243,19 @@ class AgentContext {
|
|
|
231
243
|
}
|
|
232
244
|
const schema = { ...this.structuredOutput.schema };
|
|
233
245
|
// Ensure type is set
|
|
234
|
-
if (
|
|
246
|
+
if (schema.type == null && schema.properties != null) {
|
|
235
247
|
schema.type = 'object';
|
|
236
248
|
}
|
|
237
249
|
// Add title from config name
|
|
238
|
-
if (this.structuredOutput.name
|
|
250
|
+
if (this.structuredOutput.name != null &&
|
|
251
|
+
this.structuredOutput.name !== '' &&
|
|
252
|
+
schema.title == null) {
|
|
239
253
|
schema.title = this.structuredOutput.name;
|
|
240
254
|
}
|
|
241
255
|
// Add description from config
|
|
242
|
-
if (this.structuredOutput.description
|
|
256
|
+
if (this.structuredOutput.description != null &&
|
|
257
|
+
this.structuredOutput.description !== '' &&
|
|
258
|
+
schema.description == null) {
|
|
243
259
|
schema.description = this.structuredOutput.description;
|
|
244
260
|
}
|
|
245
261
|
// Enable strict mode by default
|
|
@@ -462,30 +478,6 @@ class AgentContext {
|
|
|
462
478
|
};
|
|
463
479
|
}
|
|
464
480
|
}
|
|
465
|
-
// Handle Bedrock prompt caching (Converse API)
|
|
466
|
-
// Adds cachePoint block after text content for system message caching
|
|
467
|
-
// NOTE: Both Claude and Nova models support cachePoint in system and messages
|
|
468
|
-
// (Nova does NOT support cachePoint in tools - that check is in bedrock/index.ts)
|
|
469
|
-
if (this.provider === Providers.BEDROCK) {
|
|
470
|
-
const bedrockOptions = this.clientOptions;
|
|
471
|
-
const modelId = bedrockOptions?.model?.toLowerCase() ?? '';
|
|
472
|
-
const supportsCaching = modelId.includes('claude') ||
|
|
473
|
-
modelId.includes('anthropic') ||
|
|
474
|
-
modelId.includes('nova');
|
|
475
|
-
if (bedrockOptions?.promptCache === true && supportsCaching) {
|
|
476
|
-
finalInstructions = {
|
|
477
|
-
content: [
|
|
478
|
-
{
|
|
479
|
-
type: 'text',
|
|
480
|
-
text: instructionsString,
|
|
481
|
-
},
|
|
482
|
-
{
|
|
483
|
-
cachePoint: { type: 'default' },
|
|
484
|
-
},
|
|
485
|
-
],
|
|
486
|
-
};
|
|
487
|
-
}
|
|
488
|
-
}
|
|
489
481
|
const systemMessage = new SystemMessage(finalInstructions);
|
|
490
482
|
// Update token counts (subtract old, add new)
|
|
491
483
|
if (this.tokenCounter) {
|
|
@@ -508,14 +500,30 @@ class AgentContext {
|
|
|
508
500
|
this.cachedSystemRunnable = undefined;
|
|
509
501
|
this.systemRunnableStale = true;
|
|
510
502
|
this.lastToken = undefined;
|
|
511
|
-
this.indexTokenCountMap = {};
|
|
503
|
+
this.indexTokenCountMap = { ...this.baseIndexTokenCountMap };
|
|
512
504
|
this.currentUsage = undefined;
|
|
513
505
|
this.pruneMessages = undefined;
|
|
514
506
|
this.lastStreamCall = undefined;
|
|
515
507
|
this.tokenTypeSwitch = undefined;
|
|
508
|
+
this.reasoningTransitionCount = 0;
|
|
516
509
|
this.currentTokenType = ContentTypes.TEXT;
|
|
517
510
|
this.discoveredToolNames.clear();
|
|
518
511
|
this.handoffContext = undefined;
|
|
512
|
+
if (this.tokenCounter) {
|
|
513
|
+
this.initializeSystemRunnable();
|
|
514
|
+
const baseTokenMap = { ...this.baseIndexTokenCountMap };
|
|
515
|
+
this.indexTokenCountMap = baseTokenMap;
|
|
516
|
+
this.tokenCalculationPromise = this.calculateInstructionTokens(this.tokenCounter)
|
|
517
|
+
.then(() => {
|
|
518
|
+
this.updateTokenMapWithInstructions(baseTokenMap);
|
|
519
|
+
})
|
|
520
|
+
.catch((err) => {
|
|
521
|
+
console.error('Error calculating instruction tokens:', err);
|
|
522
|
+
});
|
|
523
|
+
}
|
|
524
|
+
else {
|
|
525
|
+
this.tokenCalculationPromise = undefined;
|
|
526
|
+
}
|
|
519
527
|
}
|
|
520
528
|
/**
|
|
521
529
|
* Update the token count map with instruction tokens
|
|
@@ -544,21 +552,51 @@ class AgentContext {
|
|
|
544
552
|
*/
|
|
545
553
|
async calculateInstructionTokens(tokenCounter) {
|
|
546
554
|
let toolTokens = 0;
|
|
547
|
-
|
|
555
|
+
// Track names to avoid double-counting when a tool appears in both
|
|
556
|
+
// this.tools (bound StructuredTool instances) and this.toolDefinitions
|
|
557
|
+
// (MCP / event-driven schemas).
|
|
558
|
+
const countedToolNames = new Set();
|
|
559
|
+
// Reset per-tool breakdown
|
|
560
|
+
this.toolsDetail = [];
|
|
561
|
+
// Count tokens for bound tools (StructuredTool instances with .schema)
|
|
548
562
|
if (this.tools && this.tools.length > 0) {
|
|
549
563
|
for (const tool of this.tools) {
|
|
550
564
|
const genericTool = tool;
|
|
551
565
|
if (genericTool.schema != null &&
|
|
552
566
|
typeof genericTool.schema === 'object') {
|
|
553
|
-
const
|
|
554
|
-
const
|
|
567
|
+
const toolName = genericTool.name ?? '';
|
|
568
|
+
const jsonSchema = toJsonSchema(genericTool.schema, toolName, genericTool.description ?? '');
|
|
555
569
|
const tokens = tokenCounter(new SystemMessage(JSON.stringify(jsonSchema)));
|
|
570
|
+
if (toolName) {
|
|
571
|
+
countedToolNames.add(toolName);
|
|
572
|
+
}
|
|
556
573
|
// Track per-tool breakdown
|
|
557
|
-
this.toolsDetail.push({ name: toolName, tokens });
|
|
574
|
+
this.toolsDetail.push({ name: toolName || 'unknown', tokens });
|
|
558
575
|
toolTokens += tokens;
|
|
559
576
|
}
|
|
560
577
|
}
|
|
561
578
|
}
|
|
579
|
+
// Count tokens for tool definitions (MCP / event-driven tools).
|
|
580
|
+
// These are sent to the provider API as tool schemas alongside bound tools.
|
|
581
|
+
// Both can be populated simultaneously (graph tools + MCP tools).
|
|
582
|
+
if (this.toolDefinitions && this.toolDefinitions.length > 0) {
|
|
583
|
+
for (const def of this.toolDefinitions) {
|
|
584
|
+
if (countedToolNames.has(def.name)) {
|
|
585
|
+
continue; // Already counted via this.tools
|
|
586
|
+
}
|
|
587
|
+
const schema = {
|
|
588
|
+
name: def.name,
|
|
589
|
+
description: def.description ?? '',
|
|
590
|
+
parameters: def.parameters ?? {},
|
|
591
|
+
};
|
|
592
|
+
const defTokens = tokenCounter(new SystemMessage(JSON.stringify(schema)));
|
|
593
|
+
this.toolsDetail.push({
|
|
594
|
+
name: def.name || 'unknown',
|
|
595
|
+
tokens: defTokens,
|
|
596
|
+
});
|
|
597
|
+
toolTokens += defTokens;
|
|
598
|
+
}
|
|
599
|
+
}
|
|
562
600
|
// Store total tool tokens for breakdown reporting
|
|
563
601
|
this.toolTokensTotal = toolTokens;
|
|
564
602
|
// Add tool tokens to existing instruction tokens (which may already include system message tokens)
|
|
@@ -689,15 +727,18 @@ class AgentContext {
|
|
|
689
727
|
return this.getEventDrivenToolsForBinding();
|
|
690
728
|
}
|
|
691
729
|
/** Traditional mode: filter actual tool instances */
|
|
692
|
-
|
|
693
|
-
|
|
730
|
+
const filtered = !this.tools || !this.toolRegistry
|
|
731
|
+
? this.tools
|
|
732
|
+
: this.filterToolsForBinding(this.tools);
|
|
733
|
+
if (this.graphTools && this.graphTools.length > 0) {
|
|
734
|
+
return [...(filtered ?? []), ...this.graphTools];
|
|
694
735
|
}
|
|
695
|
-
return
|
|
736
|
+
return filtered;
|
|
696
737
|
}
|
|
697
|
-
/** Creates schema-only tools from toolDefinitions for event-driven mode */
|
|
738
|
+
/** Creates schema-only tools from toolDefinitions for event-driven mode, merged with native tools */
|
|
698
739
|
getEventDrivenToolsForBinding() {
|
|
699
740
|
if (!this.toolDefinitions) {
|
|
700
|
-
return [];
|
|
741
|
+
return this.graphTools ?? [];
|
|
701
742
|
}
|
|
702
743
|
const defsToInclude = this.toolDefinitions.filter((def) => {
|
|
703
744
|
const allowedCallers = def.allowed_callers ?? ['direct'];
|
|
@@ -710,7 +751,15 @@ class AgentContext {
|
|
|
710
751
|
}
|
|
711
752
|
return true;
|
|
712
753
|
});
|
|
713
|
-
|
|
754
|
+
const schemaTools = createSchemaOnlyTools(defsToInclude);
|
|
755
|
+
const allTools = [...schemaTools];
|
|
756
|
+
if (this.graphTools && this.graphTools.length > 0) {
|
|
757
|
+
allTools.push(...this.graphTools);
|
|
758
|
+
}
|
|
759
|
+
if (this.tools && this.tools.length > 0) {
|
|
760
|
+
allTools.push(...this.tools);
|
|
761
|
+
}
|
|
762
|
+
return allTools;
|
|
714
763
|
}
|
|
715
764
|
/** Filters tool instances for binding based on registry config */
|
|
716
765
|
filterToolsForBinding(tools) {
|