@machina.ai/cell-cli-core 1.41.1-rc2 → 1.45.1-rc2
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/docs/AFTER_MERGE_PROMPT.md +1 -1
- package/dist/docs/changelogs/index.md +63 -0
- package/dist/docs/changelogs/latest.md +200 -244
- package/dist/docs/changelogs/preview.md +198 -385
- package/dist/docs/cli/auto-memory.md +61 -40
- package/dist/docs/cli/cli-reference.md +2 -1
- package/dist/docs/cli/creating-skills.md +165 -38
- package/dist/docs/cli/custom-commands.md +1 -0
- package/dist/docs/cli/gemini-md.md +0 -3
- package/dist/docs/cli/model-routing.md +3 -3
- package/dist/docs/cli/plan-mode.md +2 -2
- package/dist/docs/cli/settings.md +20 -19
- package/dist/docs/cli/skills-best-practices.md +78 -0
- package/dist/docs/cli/skills.md +98 -176
- package/dist/docs/cli/tutorials/memory-management.md +3 -3
- package/dist/docs/cli/tutorials/session-management.md +13 -0
- package/dist/docs/cli/tutorials/skills-getting-started.md +140 -92
- package/dist/docs/cli/using-agent-skills.md +90 -0
- package/dist/docs/core/gemma-setup.md +83 -0
- package/dist/docs/core/index.md +3 -2
- package/dist/docs/core/local-model-routing.md +14 -7
- package/dist/docs/extensions/reference.md +16 -0
- package/dist/docs/extensions/releasing.md +58 -24
- package/dist/docs/extensions/writing-extensions.md +7 -0
- package/dist/docs/get-started/installation.mdx +2 -2
- package/dist/docs/model-routing-spec.md +683 -0
- package/dist/docs/reference/commands.md +14 -7
- package/dist/docs/reference/configuration.md +264 -88
- package/dist/docs/reference/keyboard-shortcuts.md +24 -1
- package/dist/docs/reference/policy-engine.md +14 -3
- package/dist/docs/reference/tools.md +48 -1
- package/dist/docs/releases.md +2 -2
- package/dist/docs/sidebar.json +24 -2
- package/dist/docs/tools/activate-skill.md +1 -1
- package/dist/docs/tools/mcp-server.md +24 -3
- package/dist/docs/tools/memory.md +10 -13
- package/dist/docs/tools/shell.md +17 -0
- package/dist/package.json +18 -18
- package/dist/src/agent/content-utils.js +6 -1
- package/dist/src/agent/content-utils.js.map +1 -1
- package/dist/src/agent/content-utils.test.js +5 -1
- package/dist/src/agent/content-utils.test.js.map +1 -1
- package/dist/src/agent/event-translator.js +8 -7
- package/dist/src/agent/event-translator.js.map +1 -1
- package/dist/src/agent/event-translator.test.js +2 -2
- package/dist/src/agent/event-translator.test.js.map +1 -1
- package/dist/src/agent/legacy-agent-session.js +5 -1
- package/dist/src/agent/legacy-agent-session.js.map +1 -1
- package/dist/src/agent/legacy-agent-session.test.js +11 -3
- package/dist/src/agent/legacy-agent-session.test.js.map +1 -1
- package/dist/src/agent/tool-display-utils.d.ts +3 -2
- package/dist/src/agent/tool-display-utils.js +3 -2
- package/dist/src/agent/tool-display-utils.js.map +1 -1
- package/dist/src/agent/types.d.ts +33 -3
- package/dist/src/agents/a2aUtils.d.ts +1 -1
- package/dist/src/agents/a2aUtils.js +5 -4
- package/dist/src/agents/a2aUtils.js.map +1 -1
- package/dist/src/agents/a2aUtils.test.js +18 -0
- package/dist/src/agents/a2aUtils.test.js.map +1 -1
- package/dist/src/agents/agent-tool.d.ts +3 -1
- package/dist/src/agents/agent-tool.js +19 -3
- package/dist/src/agents/agent-tool.js.map +1 -1
- package/dist/src/agents/agent-tool.test.js +76 -0
- package/dist/src/agents/agent-tool.test.js.map +1 -1
- package/dist/src/agents/agentLoader.d.ts +127 -22
- package/dist/src/agents/agentLoader.js +20 -0
- package/dist/src/agents/agentLoader.js.map +1 -1
- package/dist/src/agents/agentLoader.test.js +60 -0
- package/dist/src/agents/agentLoader.test.js.map +1 -1
- package/dist/src/agents/auth-provider/types.d.ts +5 -0
- package/dist/src/agents/browser/browserAgentInvocation.js +24 -19
- package/dist/src/agents/browser/browserAgentInvocation.js.map +1 -1
- package/dist/src/agents/browser/snapshotSuperseder.js +11 -8
- package/dist/src/agents/browser/snapshotSuperseder.js.map +1 -1
- package/dist/src/agents/browser/snapshotSuperseder.test.js +6 -1
- package/dist/src/agents/browser/snapshotSuperseder.test.js.map +1 -1
- package/dist/src/agents/generalist-agent.js +8 -1
- package/dist/src/agents/generalist-agent.js.map +1 -1
- package/dist/src/agents/generalist-agent.test.js +24 -0
- package/dist/src/agents/generalist-agent.test.js.map +1 -1
- package/dist/src/agents/local-executor.d.ts +1 -0
- package/dist/src/agents/local-executor.js +76 -45
- package/dist/src/agents/local-executor.js.map +1 -1
- package/dist/src/agents/local-executor.test.js +199 -27
- package/dist/src/agents/local-executor.test.js.map +1 -1
- package/dist/src/agents/local-invocation.d.ts +1 -1
- package/dist/src/agents/local-invocation.js +47 -48
- package/dist/src/agents/local-invocation.js.map +1 -1
- package/dist/src/agents/local-invocation.test.js +13 -15
- package/dist/src/agents/local-invocation.test.js.map +1 -1
- package/dist/src/agents/local-session-invocation.d.ts +51 -0
- package/dist/src/agents/local-session-invocation.js +320 -0
- package/dist/src/agents/local-session-invocation.js.map +1 -0
- package/dist/src/agents/local-session-invocation.test.js +512 -0
- package/dist/src/agents/local-session-invocation.test.js.map +1 -0
- package/dist/src/agents/local-subagent-protocol.d.ts +18 -0
- package/dist/src/agents/local-subagent-protocol.js +357 -0
- package/dist/src/agents/local-subagent-protocol.js.map +1 -0
- package/dist/src/agents/local-subagent-protocol.test.js +676 -0
- package/dist/src/agents/local-subagent-protocol.test.js.map +1 -0
- package/dist/src/agents/registry.d.ts +8 -4
- package/dist/src/agents/registry.js +112 -47
- package/dist/src/agents/registry.js.map +1 -1
- package/dist/src/agents/registry.test.js +9 -20
- package/dist/src/agents/registry.test.js.map +1 -1
- package/dist/src/agents/remote-invocation.js +6 -6
- package/dist/src/agents/remote-invocation.js.map +1 -1
- package/dist/src/agents/remote-invocation.test.js +23 -12
- package/dist/src/agents/remote-invocation.test.js.map +1 -1
- package/dist/src/agents/remote-session-invocation.d.ts +48 -0
- package/dist/src/agents/remote-session-invocation.js +193 -0
- package/dist/src/agents/remote-session-invocation.js.map +1 -0
- package/dist/src/agents/remote-session-invocation.test.d.ts +6 -0
- package/dist/src/agents/remote-session-invocation.test.js +405 -0
- package/dist/src/agents/remote-session-invocation.test.js.map +1 -0
- package/dist/src/agents/remote-subagent-protocol.d.ts +42 -0
- package/dist/src/agents/remote-subagent-protocol.js +348 -0
- package/dist/src/agents/remote-subagent-protocol.js.map +1 -0
- package/dist/src/agents/remote-subagent-protocol.test.d.ts +6 -0
- package/dist/src/agents/remote-subagent-protocol.test.js +652 -0
- package/dist/src/agents/remote-subagent-protocol.test.js.map +1 -0
- package/dist/src/agents/skill-extraction-agent.d.ts +8 -1
- package/dist/src/agents/skill-extraction-agent.js +171 -21
- package/dist/src/agents/skill-extraction-agent.js.map +1 -1
- package/dist/src/agents/skill-extraction-agent.test.js +66 -2
- package/dist/src/agents/skill-extraction-agent.test.js.map +1 -1
- package/dist/src/agents/types.d.ts +38 -2
- package/dist/src/agents/types.js +7 -0
- package/dist/src/agents/types.js.map +1 -1
- package/dist/src/availability/autoRoutingFallback.integration.test.d.ts +6 -0
- package/dist/src/availability/autoRoutingFallback.integration.test.js +288 -0
- package/dist/src/availability/autoRoutingFallback.integration.test.js.map +1 -0
- package/dist/src/availability/fallbackIntegration.test.js +29 -0
- package/dist/src/availability/fallbackIntegration.test.js.map +1 -1
- package/dist/src/availability/modelAvailabilityService.d.ts +6 -6
- package/dist/src/availability/modelAvailabilityService.js +16 -8
- package/dist/src/availability/modelAvailabilityService.js.map +1 -1
- package/dist/src/availability/modelAvailabilityService.test.js +39 -0
- package/dist/src/availability/modelAvailabilityService.test.js.map +1 -1
- package/dist/src/availability/modelPolicy.d.ts +1 -0
- package/dist/src/availability/policyCatalog.d.ts +2 -0
- package/dist/src/availability/policyCatalog.js +38 -9
- package/dist/src/availability/policyCatalog.js.map +1 -1
- package/dist/src/availability/policyCatalog.test.js +5 -4
- package/dist/src/availability/policyCatalog.test.js.map +1 -1
- package/dist/src/availability/policyHelpers.js +42 -27
- package/dist/src/availability/policyHelpers.js.map +1 -1
- package/dist/src/availability/policyHelpers.test.js +47 -7
- package/dist/src/availability/policyHelpers.test.js.map +1 -1
- package/dist/src/availability/testUtils.js +1 -1
- package/dist/src/availability/testUtils.js.map +1 -1
- package/dist/src/code_assist/admin/admin_controls.js +3 -1
- package/dist/src/code_assist/admin/admin_controls.js.map +1 -1
- package/dist/src/code_assist/experiments/flagNames.d.ts +1 -1
- package/dist/src/code_assist/experiments/flagNames.js +1 -1
- package/dist/src/code_assist/experiments/flagNames.js.map +1 -1
- package/dist/src/code_assist/oauth-credential-storage.js +12 -3
- package/dist/src/code_assist/oauth-credential-storage.js.map +1 -1
- package/dist/src/code_assist/oauth-credential-storage.test.js +29 -2
- package/dist/src/code_assist/oauth-credential-storage.test.js.map +1 -1
- package/dist/src/code_assist/oauth2.js +12 -3
- package/dist/src/code_assist/oauth2.js.map +1 -1
- package/dist/src/code_assist/oauth2.test.js +38 -0
- package/dist/src/code_assist/oauth2.test.js.map +1 -1
- package/dist/src/code_assist/setup.d.ts +3 -0
- package/dist/src/code_assist/setup.js +9 -0
- package/dist/src/code_assist/setup.js.map +1 -1
- package/dist/src/code_assist/setup.test.js +10 -1
- package/dist/src/code_assist/setup.test.js.map +1 -1
- package/dist/src/commands/memory.d.ts +83 -2
- package/dist/src/commands/memory.js +479 -28
- package/dist/src/commands/memory.js.map +1 -1
- package/dist/src/commands/memory.test.js +414 -58
- package/dist/src/commands/memory.test.js.map +1 -1
- package/dist/src/config/config.d.ts +61 -37
- package/dist/src/config/config.js +294 -101
- package/dist/src/config/config.js.map +1 -1
- package/dist/src/config/config.test.js +365 -113
- package/dist/src/config/config.test.js.map +1 -1
- package/dist/src/config/defaultModelConfigs.js +185 -61
- package/dist/src/config/defaultModelConfigs.js.map +1 -1
- package/dist/src/config/flashFallback.test.js +31 -0
- package/dist/src/config/flashFallback.test.js.map +1 -1
- package/dist/src/config/models.d.ts +20 -10
- package/dist/src/config/models.js +105 -34
- package/dist/src/config/models.js.map +1 -1
- package/dist/src/config/models.test.js +204 -47
- package/dist/src/config/models.test.js.map +1 -1
- package/dist/src/config/projectRegistry.d.ts +1 -0
- package/dist/src/config/projectRegistry.js +14 -3
- package/dist/src/config/projectRegistry.js.map +1 -1
- package/dist/src/config/projectRegistry.test.js +43 -0
- package/dist/src/config/projectRegistry.test.js.map +1 -1
- package/dist/src/config/scoped-config.d.ts +22 -0
- package/dist/src/config/scoped-config.js +32 -0
- package/dist/src/config/scoped-config.js.map +1 -1
- package/dist/src/config/storage.d.ts +0 -1
- package/dist/src/config/storage.js +0 -3
- package/dist/src/config/storage.js.map +1 -1
- package/dist/src/confirmation-bus/message-bus.d.ts +3 -1
- package/dist/src/confirmation-bus/message-bus.js +14 -5
- package/dist/src/confirmation-bus/message-bus.js.map +1 -1
- package/dist/src/confirmation-bus/message-bus.test.js +34 -0
- package/dist/src/confirmation-bus/message-bus.test.js.map +1 -1
- package/dist/src/context/chatCompressionService.js +7 -5
- package/dist/src/context/chatCompressionService.js.map +1 -1
- package/dist/src/context/chatCompressionService.test.js +1 -1
- package/dist/src/context/chatCompressionService.test.js.map +1 -1
- package/dist/src/context/config/configLoader.js +4 -1
- package/dist/src/context/config/configLoader.js.map +1 -1
- package/dist/src/context/config/profiles.d.ts +10 -0
- package/dist/src/context/config/profiles.js +98 -3
- package/dist/src/context/config/profiles.js.map +1 -1
- package/dist/src/context/config/schema.d.ts +4 -0
- package/dist/src/context/config/schema.js +4 -0
- package/dist/src/context/config/schema.js.map +1 -1
- package/dist/src/context/config/types.d.ts +13 -1
- package/dist/src/context/contextCompressionService.test.js +7 -3
- package/dist/src/context/contextCompressionService.test.js.map +1 -1
- package/dist/src/context/contextManager.barrier.test.js +39 -16
- package/dist/src/context/contextManager.barrier.test.js.map +1 -1
- package/dist/src/context/contextManager.d.ts +25 -28
- package/dist/src/context/contextManager.hotstart.test.d.ts +6 -0
- package/dist/src/context/contextManager.hotstart.test.js +65 -0
- package/dist/src/context/contextManager.hotstart.test.js.map +1 -0
- package/dist/src/context/contextManager.incremental.test.d.ts +6 -0
- package/dist/src/context/contextManager.incremental.test.js +101 -0
- package/dist/src/context/contextManager.incremental.test.js.map +1 -0
- package/dist/src/context/contextManager.js +276 -79
- package/dist/src/context/contextManager.js.map +1 -1
- package/dist/src/context/contextManager.test.d.ts +6 -0
- package/dist/src/context/contextManager.test.js +142 -0
- package/dist/src/context/contextManager.test.js.map +1 -0
- package/dist/src/context/eventBus.d.ts +13 -0
- package/dist/src/context/eventBus.js +12 -0
- package/dist/src/context/eventBus.js.map +1 -1
- package/dist/src/context/graph/behaviorRegistry.d.ts +4 -12
- package/dist/src/context/graph/behaviorRegistry.js.map +1 -1
- package/dist/src/context/graph/builtinBehaviors.d.ts +6 -1
- package/dist/src/context/graph/builtinBehaviors.js +23 -108
- package/dist/src/context/graph/builtinBehaviors.js.map +1 -1
- package/dist/src/context/graph/fromGraph.d.ts +8 -3
- package/dist/src/context/graph/fromGraph.js +46 -30
- package/dist/src/context/graph/fromGraph.js.map +1 -1
- package/dist/src/context/graph/fromGraph.test.d.ts +6 -0
- package/dist/src/context/graph/fromGraph.test.js +186 -0
- package/dist/src/context/graph/fromGraph.test.js.map +1 -0
- package/dist/src/context/graph/mapper.d.ts +8 -10
- package/dist/src/context/graph/mapper.js +11 -19
- package/dist/src/context/graph/mapper.js.map +1 -1
- package/dist/src/context/graph/mapper.test.d.ts +6 -0
- package/dist/src/context/graph/mapper.test.js +101 -0
- package/dist/src/context/graph/mapper.test.js.map +1 -0
- package/dist/src/context/graph/nodeIdService.d.ts +17 -0
- package/dist/src/context/graph/nodeIdService.js +24 -0
- package/dist/src/context/graph/nodeIdService.js.map +1 -0
- package/dist/src/context/graph/render.d.ts +24 -5
- package/dist/src/context/graph/render.js +129 -34
- package/dist/src/context/graph/render.js.map +1 -1
- package/dist/src/context/graph/render.test.d.ts +6 -0
- package/dist/src/context/graph/render.test.js +280 -0
- package/dist/src/context/graph/render.test.js.map +1 -0
- package/dist/src/context/graph/toGraph.d.ts +16 -14
- package/dist/src/context/graph/toGraph.js +180 -202
- package/dist/src/context/graph/toGraph.js.map +1 -1
- package/dist/src/context/graph/toGraph.test.d.ts +6 -0
- package/dist/src/context/graph/toGraph.test.js +116 -0
- package/dist/src/context/graph/toGraph.test.js.map +1 -0
- package/dist/src/context/graph/types.d.ts +36 -73
- package/dist/src/context/graph/types.js +23 -14
- package/dist/src/context/graph/types.js.map +1 -1
- package/dist/src/context/initializer.js +26 -5
- package/dist/src/context/initializer.js.map +1 -1
- package/dist/src/context/pipeline/contextWorkingBuffer.d.ts +5 -8
- package/dist/src/context/pipeline/contextWorkingBuffer.js +105 -35
- package/dist/src/context/pipeline/contextWorkingBuffer.js.map +1 -1
- package/dist/src/context/pipeline/contextWorkingBuffer.test.js +81 -13
- package/dist/src/context/pipeline/contextWorkingBuffer.test.js.map +1 -1
- package/dist/src/context/pipeline/environment.d.ts +4 -0
- package/dist/src/context/pipeline/environmentImpl.d.ts +6 -5
- package/dist/src/context/pipeline/environmentImpl.js +7 -9
- package/dist/src/context/pipeline/environmentImpl.js.map +1 -1
- package/dist/src/context/pipeline/environmentImpl.test.js +5 -1
- package/dist/src/context/pipeline/environmentImpl.test.js.map +1 -1
- package/dist/src/context/pipeline/orchestrator.d.ts +20 -6
- package/dist/src/context/pipeline/orchestrator.js +97 -80
- package/dist/src/context/pipeline/orchestrator.js.map +1 -1
- package/dist/src/context/pipeline/orchestrator.test.js +33 -36
- package/dist/src/context/pipeline/orchestrator.test.js.map +1 -1
- package/dist/src/context/pipeline.d.ts +0 -1
- package/dist/src/context/processors/blobDegradationProcessor.js +43 -84
- package/dist/src/context/processors/blobDegradationProcessor.js.map +1 -1
- package/dist/src/context/processors/blobDegradationProcessor.test.js +33 -37
- package/dist/src/context/processors/blobDegradationProcessor.test.js.map +1 -1
- package/dist/src/context/processors/nodeDistillationProcessor.js +58 -80
- package/dist/src/context/processors/nodeDistillationProcessor.js.map +1 -1
- package/dist/src/context/processors/nodeDistillationProcessor.test.js +21 -15
- package/dist/src/context/processors/nodeDistillationProcessor.test.js.map +1 -1
- package/dist/src/context/processors/nodeTruncationProcessor.js +16 -60
- package/dist/src/context/processors/nodeTruncationProcessor.js.map +1 -1
- package/dist/src/context/processors/nodeTruncationProcessor.test.js +16 -19
- package/dist/src/context/processors/nodeTruncationProcessor.test.js.map +1 -1
- package/dist/src/context/processors/rollingSummaryProcessor.js +12 -25
- package/dist/src/context/processors/rollingSummaryProcessor.js.map +1 -1
- package/dist/src/context/processors/rollingSummaryProcessor.test.js +10 -9
- package/dist/src/context/processors/rollingSummaryProcessor.test.js.map +1 -1
- package/dist/src/context/processors/stateSnapshotAsyncProcessor.d.ts +7 -0
- package/dist/src/context/processors/stateSnapshotAsyncProcessor.js +37 -19
- package/dist/src/context/processors/stateSnapshotAsyncProcessor.js.map +1 -1
- package/dist/src/context/processors/stateSnapshotAsyncProcessor.test.js +35 -10
- package/dist/src/context/processors/stateSnapshotAsyncProcessor.test.js.map +1 -1
- package/dist/src/context/processors/stateSnapshotProcessor.d.ts +2 -0
- package/dist/src/context/processors/stateSnapshotProcessor.js +53 -21
- package/dist/src/context/processors/stateSnapshotProcessor.js.map +1 -1
- package/dist/src/context/processors/stateSnapshotProcessor.test.js +52 -12
- package/dist/src/context/processors/stateSnapshotProcessor.test.js.map +1 -1
- package/dist/src/context/processors/toolMaskingProcessor.js +96 -117
- package/dist/src/context/processors/toolMaskingProcessor.js.map +1 -1
- package/dist/src/context/processors/toolMaskingProcessor.test.js +50 -17
- package/dist/src/context/processors/toolMaskingProcessor.test.js.map +1 -1
- package/dist/src/context/system-tests/hysteresis.test.d.ts +6 -0
- package/dist/src/context/system-tests/hysteresis.test.js +100 -0
- package/dist/src/context/system-tests/hysteresis.test.js.map +1 -0
- package/dist/src/context/system-tests/lifecycle.golden.test.js +107 -72
- package/dist/src/context/system-tests/lifecycle.golden.test.js.map +1 -1
- package/dist/src/context/system-tests/powerUserLifecycle.test.d.ts +6 -0
- package/dist/src/context/system-tests/powerUserLifecycle.test.js +91 -0
- package/dist/src/context/system-tests/powerUserLifecycle.test.js.map +1 -0
- package/dist/src/context/system-tests/simulationHarness.d.ts +2 -5
- package/dist/src/context/system-tests/simulationHarness.js +34 -35
- package/dist/src/context/system-tests/simulationHarness.js.map +1 -1
- package/dist/src/context/testing/contextTestUtils.d.ts +5 -3
- package/dist/src/context/testing/contextTestUtils.js +74 -53
- package/dist/src/context/testing/contextTestUtils.js.map +1 -1
- package/dist/src/context/testing/testProfile.js +1 -0
- package/dist/src/context/testing/testProfile.js.map +1 -1
- package/dist/src/context/toolOutputMaskingService.js +1 -2
- package/dist/src/context/toolOutputMaskingService.js.map +1 -1
- package/dist/src/context/toolOutputMaskingService.test.js +5 -20
- package/dist/src/context/toolOutputMaskingService.test.js.map +1 -1
- package/dist/src/context/utils/adaptiveTokenCalculator.d.ts +70 -0
- package/dist/src/context/utils/adaptiveTokenCalculator.js +138 -0
- package/dist/src/context/utils/adaptiveTokenCalculator.js.map +1 -0
- package/dist/src/context/utils/adaptiveTokenCalculator.test.d.ts +6 -0
- package/dist/src/context/utils/adaptiveTokenCalculator.test.js +129 -0
- package/dist/src/context/utils/adaptiveTokenCalculator.test.js.map +1 -0
- package/dist/src/context/utils/contextTokenCalculator.d.ts +63 -2
- package/dist/src/context/utils/contextTokenCalculator.js +80 -5
- package/dist/src/context/utils/contextTokenCalculator.js.map +1 -1
- package/dist/src/context/utils/contextTokenCalculator.test.d.ts +6 -0
- package/dist/src/context/utils/contextTokenCalculator.test.js +54 -0
- package/dist/src/context/utils/contextTokenCalculator.test.js.map +1 -0
- package/dist/src/context/utils/formatNodesForLlm.d.ts +21 -0
- package/dist/src/context/utils/formatNodesForLlm.js +69 -0
- package/dist/src/context/utils/formatNodesForLlm.js.map +1 -0
- package/dist/src/context/utils/formatNodesForLlm.test.d.ts +6 -0
- package/dist/src/context/utils/formatNodesForLlm.test.js +110 -0
- package/dist/src/context/utils/formatNodesForLlm.test.js.map +1 -0
- package/dist/src/context/utils/invariantChecker.d.ts +11 -0
- package/dist/src/context/utils/invariantChecker.js +36 -0
- package/dist/src/context/utils/invariantChecker.js.map +1 -0
- package/dist/src/context/utils/snapshotGenerator.d.ts +43 -1
- package/dist/src/context/utils/snapshotGenerator.js +332 -33
- package/dist/src/context/utils/snapshotGenerator.js.map +1 -1
- package/dist/src/context/utils/snapshotGenerator.test.d.ts +6 -0
- package/dist/src/context/utils/snapshotGenerator.test.js +362 -0
- package/dist/src/context/utils/snapshotGenerator.test.js.map +1 -0
- package/dist/src/context/utils/tokenCalibration.d.ts +9 -0
- package/dist/src/context/utils/tokenCalibration.js +30 -0
- package/dist/src/context/utils/tokenCalibration.js.map +1 -0
- package/dist/src/core/agentChatHistory.d.ts +29 -14
- package/dist/src/core/agentChatHistory.js +27 -27
- package/dist/src/core/agentChatHistory.js.map +1 -1
- package/dist/src/core/baseLlmClient.d.ts +8 -0
- package/dist/src/core/baseLlmClient.js +23 -3
- package/dist/src/core/baseLlmClient.js.map +1 -1
- package/dist/src/core/baseLlmClient.test.js +27 -23
- package/dist/src/core/baseLlmClient.test.js.map +1 -1
- package/dist/src/core/client.d.ts +7 -5
- package/dist/src/core/client.js +65 -40
- package/dist/src/core/client.js.map +1 -1
- package/dist/src/core/client.test.js +35 -131
- package/dist/src/core/client.test.js.map +1 -1
- package/dist/src/core/contentGenerator.js +46 -21
- package/dist/src/core/contentGenerator.js.map +1 -1
- package/dist/src/core/contentGenerator.test.js +191 -13
- package/dist/src/core/contentGenerator.test.js.map +1 -1
- package/dist/src/core/fakeContentGenerator.d.ts +15 -3
- package/dist/src/core/fakeContentGenerator.js +29 -9
- package/dist/src/core/fakeContentGenerator.js.map +1 -1
- package/dist/src/core/geminiChat.d.ts +18 -7
- package/dist/src/core/geminiChat.js +312 -54
- package/dist/src/core/geminiChat.js.map +1 -1
- package/dist/src/core/geminiChat.test.js +448 -54
- package/dist/src/core/geminiChat.test.js.map +1 -1
- package/dist/src/core/geminiChat_network_retry.test.js +39 -0
- package/dist/src/core/geminiChat_network_retry.test.js.map +1 -1
- package/dist/src/core/localLiteRtLmClient.js +6 -2
- package/dist/src/core/localLiteRtLmClient.js.map +1 -1
- package/dist/src/core/prompts.test.js +12 -7
- package/dist/src/core/prompts.test.js.map +1 -1
- package/dist/src/core/turn.d.ts +7 -2
- package/dist/src/core/turn.js +61 -4
- package/dist/src/core/turn.js.map +1 -1
- package/dist/src/core/turn.test.js +19 -10
- package/dist/src/core/turn.test.js.map +1 -1
- package/dist/src/fallback/handler.js +16 -6
- package/dist/src/fallback/handler.js.map +1 -1
- package/dist/src/fallback/handler.test.js +8 -2
- package/dist/src/fallback/handler.test.js.map +1 -1
- package/dist/src/generated/git-commit.d.ts +2 -2
- package/dist/src/generated/git-commit.js +2 -2
- package/dist/src/hooks/hookEventHandler.js +3 -2
- package/dist/src/hooks/hookEventHandler.js.map +1 -1
- package/dist/src/hooks/hookEventHandler.test.js +80 -0
- package/dist/src/hooks/hookEventHandler.test.js.map +1 -1
- package/dist/src/hooks/hookRunner.test.js +3 -3
- package/dist/src/hooks/hookRunner.test.js.map +1 -1
- package/dist/src/hooks/hookTranslator.js +95 -5
- package/dist/src/hooks/hookTranslator.js.map +1 -1
- package/dist/src/hooks/hookTranslator.test.js +171 -0
- package/dist/src/hooks/hookTranslator.test.js.map +1 -1
- package/dist/src/ide/ide-client.js +5 -3
- package/dist/src/ide/ide-client.js.map +1 -1
- package/dist/src/ide/ide-connection-utils.js +12 -10
- package/dist/src/ide/ide-connection-utils.js.map +1 -1
- package/dist/src/ide/ide-connection-utils.test.js +25 -2
- package/dist/src/ide/ide-connection-utils.test.js.map +1 -1
- package/dist/src/ide/types.d.ts +16 -16
- package/dist/src/index.d.ts +5 -2
- package/dist/src/index.js +5 -3
- package/dist/src/index.js.map +1 -1
- package/dist/src/mcp/oauth-provider.d.ts +8 -0
- package/dist/src/mcp/oauth-provider.js +41 -0
- package/dist/src/mcp/oauth-provider.js.map +1 -1
- package/dist/src/mcp/oauth-token-storage.js +7 -1
- package/dist/src/mcp/oauth-token-storage.js.map +1 -1
- package/dist/src/mcp/oauth-token-storage.test.js +55 -0
- package/dist/src/mcp/oauth-token-storage.test.js.map +1 -1
- package/dist/src/mcp/stored-token-provider.d.ts +27 -0
- package/dist/src/mcp/stored-token-provider.js +76 -0
- package/dist/src/mcp/stored-token-provider.js.map +1 -0
- package/dist/src/mcp/token-storage/keychain-token-storage.js +2 -2
- package/dist/src/mcp/token-storage/keychain-token-storage.js.map +1 -1
- package/dist/src/mcp/token-storage/keychain-token-storage.test.js +14 -1
- package/dist/src/mcp/token-storage/keychain-token-storage.test.js.map +1 -1
- package/dist/src/output/json-formatter.d.ts +1 -1
- package/dist/src/output/json-formatter.js +4 -1
- package/dist/src/output/json-formatter.js.map +1 -1
- package/dist/src/output/json-formatter.test.js +7 -0
- package/dist/src/output/json-formatter.test.js.map +1 -1
- package/dist/src/output/types.d.ts +1 -0
- package/dist/src/output/types.js.map +1 -1
- package/dist/src/policy/config.js +25 -0
- package/dist/src/policy/config.js.map +1 -1
- package/dist/src/policy/config.test.js +80 -0
- package/dist/src/policy/config.test.js.map +1 -1
- package/dist/src/policy/core-tools-mapping.test.js +9 -1
- package/dist/src/policy/core-tools-mapping.test.js.map +1 -1
- package/dist/src/policy/policies/plan.toml +1 -1
- package/dist/src/policy/policies/write.toml +0 -7
- package/dist/src/policy/policy-engine.test.js +0 -8
- package/dist/src/policy/policy-engine.test.js.map +1 -1
- package/dist/src/policy/sandboxPolicyManager.d.ts +20 -20
- package/dist/src/policy/stable-stringify.js +10 -6
- package/dist/src/policy/stable-stringify.js.map +1 -1
- package/dist/src/policy/stable-stringify.test.js +157 -0
- package/dist/src/policy/stable-stringify.test.js.map +1 -0
- package/dist/src/policy/types.d.ts +1 -0
- package/dist/src/policy/types.js.map +1 -1
- package/dist/src/prompts/promptProvider.js +8 -9
- package/dist/src/prompts/promptProvider.js.map +1 -1
- package/dist/src/prompts/promptProvider.test.js +3 -1
- package/dist/src/prompts/promptProvider.test.js.map +1 -1
- package/dist/src/prompts/snippets-memory.test.d.ts +6 -0
- package/dist/src/prompts/{snippets-memory-v2.test.js → snippets-memory.test.js} +5 -23
- package/dist/src/prompts/snippets-memory.test.js.map +1 -0
- package/dist/src/prompts/snippets.d.ts +6 -11
- package/dist/src/prompts/snippets.js +28 -30
- package/dist/src/prompts/snippets.js.map +1 -1
- package/dist/src/prompts/snippets.legacy.d.ts +0 -1
- package/dist/src/prompts/snippets.legacy.js +8 -15
- package/dist/src/prompts/snippets.legacy.js.map +1 -1
- package/dist/src/routing/strategies/approvalModeStrategy.js +5 -3
- package/dist/src/routing/strategies/approvalModeStrategy.js.map +1 -1
- package/dist/src/routing/strategies/approvalModeStrategy.test.js +9 -0
- package/dist/src/routing/strategies/approvalModeStrategy.test.js.map +1 -1
- package/dist/src/routing/strategies/classifierStrategy.js +18 -4
- package/dist/src/routing/strategies/classifierStrategy.js.map +1 -1
- package/dist/src/routing/strategies/classifierStrategy.test.js +77 -1
- package/dist/src/routing/strategies/classifierStrategy.test.js.map +1 -1
- package/dist/src/routing/strategies/defaultStrategy.js +1 -1
- package/dist/src/routing/strategies/defaultStrategy.js.map +1 -1
- package/dist/src/routing/strategies/fallbackStrategy.js +1 -1
- package/dist/src/routing/strategies/fallbackStrategy.js.map +1 -1
- package/dist/src/routing/strategies/gemmaClassifierStrategy.js +7 -1
- package/dist/src/routing/strategies/gemmaClassifierStrategy.js.map +1 -1
- package/dist/src/routing/strategies/gemmaClassifierStrategy.test.js +15 -1
- package/dist/src/routing/strategies/gemmaClassifierStrategy.test.js.map +1 -1
- package/dist/src/routing/strategies/numericalClassifierStrategy.d.ts +1 -0
- package/dist/src/routing/strategies/numericalClassifierStrategy.js +32 -5
- package/dist/src/routing/strategies/numericalClassifierStrategy.js.map +1 -1
- package/dist/src/routing/strategies/numericalClassifierStrategy.test.js +247 -25
- package/dist/src/routing/strategies/numericalClassifierStrategy.test.js.map +1 -1
- package/dist/src/routing/strategies/overrideStrategy.js +1 -1
- package/dist/src/routing/strategies/overrideStrategy.js.map +1 -1
- package/dist/src/sandbox/utils/commandSafety.js +22 -2
- package/dist/src/sandbox/utils/commandSafety.js.map +1 -1
- package/dist/src/sandbox/utils/commandSafety.test.d.ts +6 -0
- package/dist/src/sandbox/utils/commandSafety.test.js +85 -0
- package/dist/src/sandbox/utils/commandSafety.test.js.map +1 -0
- package/dist/src/scheduler/confirmation.test.js +29 -0
- package/dist/src/scheduler/confirmation.test.js.map +1 -1
- package/dist/src/scheduler/scheduler.js +15 -0
- package/dist/src/scheduler/scheduler.js.map +1 -1
- package/dist/src/scheduler/scheduler.test.js +1 -1
- package/dist/src/scheduler/scheduler.test.js.map +1 -1
- package/dist/src/scheduler/scheduler_parallel.test.js +37 -0
- package/dist/src/scheduler/scheduler_parallel.test.js.map +1 -1
- package/dist/src/scheduler/state-manager.js +5 -1
- package/dist/src/scheduler/state-manager.js.map +1 -1
- package/dist/src/scheduler/tool-executor.js +7 -4
- package/dist/src/scheduler/tool-executor.js.map +1 -1
- package/dist/src/scheduler/types.d.ts +5 -1
- package/dist/src/scheduler/types.js.map +1 -1
- package/dist/src/services/chatRecordingService.d.ts +14 -7
- package/dist/src/services/chatRecordingService.js +157 -133
- package/dist/src/services/chatRecordingService.js.map +1 -1
- package/dist/src/services/chatRecordingService.test.js +189 -52
- package/dist/src/services/chatRecordingService.test.js.map +1 -1
- package/dist/src/services/fileDiscoveryService.js +2 -1
- package/dist/src/services/fileDiscoveryService.js.map +1 -1
- package/dist/src/services/fileDiscoveryService.test.js +36 -0
- package/dist/src/services/fileDiscoveryService.test.js.map +1 -1
- package/dist/src/services/gitService.js +43 -4
- package/dist/src/services/gitService.js.map +1 -1
- package/dist/src/services/gitService.test.js +105 -1
- package/dist/src/services/gitService.test.js.map +1 -1
- package/dist/src/services/keychainService.js +14 -5
- package/dist/src/services/keychainService.js.map +1 -1
- package/dist/src/services/memoryPatchUtils.d.ts +93 -0
- package/dist/src/services/memoryPatchUtils.js +310 -2
- package/dist/src/services/memoryPatchUtils.js.map +1 -1
- package/dist/src/services/memoryService.d.ts +2 -0
- package/dist/src/services/memoryService.js +214 -9
- package/dist/src/services/memoryService.js.map +1 -1
- package/dist/src/services/memoryService.test.js +133 -0
- package/dist/src/services/memoryService.test.js.map +1 -1
- package/dist/src/services/modelConfigService.d.ts +3 -0
- package/dist/src/services/modelConfigService.js +22 -13
- package/dist/src/services/modelConfigService.js.map +1 -1
- package/dist/src/services/modelConfigService.test.js +73 -0
- package/dist/src/services/modelConfigService.test.js.map +1 -1
- package/dist/src/services/shellExecutionService.js +88 -41
- package/dist/src/services/shellExecutionService.js.map +1 -1
- package/dist/src/services/shellExecutionService.test.js +22 -9
- package/dist/src/services/shellExecutionService.test.js.map +1 -1
- package/dist/src/services/shellExecutionService.windows.integration.test.d.ts +6 -0
- package/dist/src/services/shellExecutionService.windows.integration.test.js +63 -0
- package/dist/src/services/shellExecutionService.windows.integration.test.js.map +1 -0
- package/dist/src/services/test-data/resolved-aliases-retry.golden.json +85 -7
- package/dist/src/services/test-data/resolved-aliases.golden.json +85 -7
- package/dist/src/services/trackerTypes.d.ts +4 -4
- package/dist/src/skills/skillManager.d.ts +4 -0
- package/dist/src/skills/skillManager.js +6 -0
- package/dist/src/skills/skillManager.js.map +1 -1
- package/dist/src/skills/skillManager.test.js +10 -0
- package/dist/src/skills/skillManager.test.js.map +1 -1
- package/dist/src/telemetry/file-exporters.d.ts +4 -1
- package/dist/src/telemetry/file-exporters.js +21 -3
- package/dist/src/telemetry/file-exporters.js.map +1 -1
- package/dist/src/telemetry/file-exporters.test.js +32 -2
- package/dist/src/telemetry/file-exporters.test.js.map +1 -1
- package/dist/src/telemetry/gcp-exporters.d.ts +3 -0
- package/dist/src/telemetry/gcp-exporters.js +72 -5
- package/dist/src/telemetry/gcp-exporters.js.map +1 -1
- package/dist/src/telemetry/gcp-exporters.test.js +52 -0
- package/dist/src/telemetry/gcp-exporters.test.js.map +1 -1
- package/dist/src/telemetry/heap-snapshot.d.ts +12 -0
- package/dist/src/telemetry/heap-snapshot.js +35 -0
- package/dist/src/telemetry/heap-snapshot.js.map +1 -0
- package/dist/src/telemetry/heap-snapshot.test.d.ts +6 -0
- package/dist/src/telemetry/heap-snapshot.test.js +38 -0
- package/dist/src/telemetry/heap-snapshot.test.js.map +1 -0
- package/dist/src/telemetry/index.d.ts +1 -0
- package/dist/src/telemetry/index.js +1 -0
- package/dist/src/telemetry/index.js.map +1 -1
- package/dist/src/telemetry/loggers.test.js +1 -0
- package/dist/src/telemetry/loggers.test.js.map +1 -1
- package/dist/src/telemetry/memory-monitor.d.ts +5 -0
- package/dist/src/telemetry/memory-monitor.js +8 -0
- package/dist/src/telemetry/memory-monitor.js.map +1 -1
- package/dist/src/telemetry/metrics.js +13 -2
- package/dist/src/telemetry/metrics.js.map +1 -1
- package/dist/src/telemetry/metrics.test.js +61 -1
- package/dist/src/telemetry/metrics.test.js.map +1 -1
- package/dist/src/test-utils/config.js +10 -1
- package/dist/src/test-utils/config.js.map +1 -1
- package/dist/src/tools/ask-user.js +25 -1
- package/dist/src/tools/ask-user.js.map +1 -1
- package/dist/src/tools/ask-user.test.js +46 -1
- package/dist/src/tools/ask-user.test.js.map +1 -1
- package/dist/src/tools/definitions/base-declarations.d.ts +0 -3
- package/dist/src/tools/definitions/base-declarations.js +0 -4
- package/dist/src/tools/definitions/base-declarations.js.map +1 -1
- package/dist/src/tools/definitions/coreTools.d.ts +1 -2
- package/dist/src/tools/definitions/coreTools.js +2 -8
- package/dist/src/tools/definitions/coreTools.js.map +1 -1
- package/dist/src/tools/definitions/coreToolsModelSnapshots.test.js +1 -2
- package/dist/src/tools/definitions/coreToolsModelSnapshots.test.js.map +1 -1
- package/dist/src/tools/definitions/model-family-sets/default-legacy.js +7 -31
- package/dist/src/tools/definitions/model-family-sets/default-legacy.js.map +1 -1
- package/dist/src/tools/definitions/model-family-sets/gemini-3.js +9 -26
- package/dist/src/tools/definitions/model-family-sets/gemini-3.js.map +1 -1
- package/dist/src/tools/definitions/types.d.ts +0 -1
- package/dist/src/tools/edit.js +27 -5
- package/dist/src/tools/edit.js.map +1 -1
- package/dist/src/tools/edit.test.js +37 -0
- package/dist/src/tools/edit.test.js.map +1 -1
- package/dist/src/tools/grep.js +14 -2
- package/dist/src/tools/grep.js.map +1 -1
- package/dist/src/tools/grep.test.js +17 -0
- package/dist/src/tools/grep.test.js.map +1 -1
- package/dist/src/tools/jit-context.d.ts +1 -1
- package/dist/src/tools/jit-context.js +1 -4
- package/dist/src/tools/jit-context.js.map +1 -1
- package/dist/src/tools/jit-context.test.js +1 -13
- package/dist/src/tools/jit-context.test.js.map +1 -1
- package/dist/src/tools/ls.js +5 -0
- package/dist/src/tools/ls.js.map +1 -1
- package/dist/src/tools/mcp-client-manager.js +2 -1
- package/dist/src/tools/mcp-client-manager.js.map +1 -1
- package/dist/src/tools/mcp-client-manager.test.js +29 -0
- package/dist/src/tools/mcp-client-manager.test.js.map +1 -1
- package/dist/src/tools/mcp-client.js +89 -50
- package/dist/src/tools/mcp-client.js.map +1 -1
- package/dist/src/tools/mcp-client.test.js +353 -59
- package/dist/src/tools/mcp-client.test.js.map +1 -1
- package/dist/src/tools/{xcode-mcp-fix-transport.d.ts → mcp-compliance-transport.d.ts} +6 -6
- package/dist/src/tools/{xcode-mcp-fix-transport.js → mcp-compliance-transport.js} +6 -6
- package/dist/src/tools/mcp-compliance-transport.js.map +1 -0
- package/dist/src/tools/mcp-compliance-transport.test.d.ts +6 -0
- package/dist/src/tools/mcp-compliance-transport.test.js +162 -0
- package/dist/src/tools/mcp-compliance-transport.test.js.map +1 -0
- package/dist/src/tools/memoryTool.d.ts +9 -31
- package/dist/src/tools/memoryTool.js +47 -262
- package/dist/src/tools/memoryTool.js.map +1 -1
- package/dist/src/tools/memoryTool.test.js +41 -312
- package/dist/src/tools/memoryTool.test.js.map +1 -1
- package/dist/src/tools/read-file.js +11 -6
- package/dist/src/tools/read-file.js.map +1 -1
- package/dist/src/tools/read-file.test.js +20 -8
- package/dist/src/tools/read-file.test.js.map +1 -1
- package/dist/src/tools/read-many-files.js +2 -2
- package/dist/src/tools/read-many-files.js.map +1 -1
- package/dist/src/tools/ripGrep.d.ts +3 -7
- package/dist/src/tools/ripGrep.js +57 -35
- package/dist/src/tools/ripGrep.js.map +1 -1
- package/dist/src/tools/ripGrep.test.js +197 -276
- package/dist/src/tools/ripGrep.test.js.map +1 -1
- package/dist/src/tools/shell.d.ts +5 -3
- package/dist/src/tools/shell.js +130 -36
- package/dist/src/tools/shell.js.map +1 -1
- package/dist/src/tools/shell.test.js +186 -14
- package/dist/src/tools/shell.test.js.map +1 -1
- package/dist/src/tools/shell_proactive.test.js +1 -0
- package/dist/src/tools/shell_proactive.test.js.map +1 -1
- package/dist/src/tools/tool-names.d.ts +3 -3
- package/dist/src/tools/tool-names.js +4 -5
- package/dist/src/tools/tool-names.js.map +1 -1
- package/dist/src/tools/tool-registry.js +1 -1
- package/dist/src/tools/tool-registry.js.map +1 -1
- package/dist/src/tools/tools.d.ts +6 -0
- package/dist/src/tools/tools.js.map +1 -1
- package/dist/src/tools/topicTool.js +5 -0
- package/dist/src/tools/topicTool.js.map +1 -1
- package/dist/src/tools/write-file.js +13 -0
- package/dist/src/tools/write-file.js.map +1 -1
- package/dist/src/tools/write-file.test.js +8 -0
- package/dist/src/tools/write-file.test.js.map +1 -1
- package/dist/src/utils/atCommandUtils.d.ts +35 -0
- package/dist/src/utils/atCommandUtils.js +163 -0
- package/dist/src/utils/atCommandUtils.js.map +1 -0
- package/dist/src/utils/atCommandUtils.test.d.ts +6 -0
- package/dist/src/utils/atCommandUtils.test.js +292 -0
- package/dist/src/utils/atCommandUtils.test.js.map +1 -0
- package/dist/src/utils/channel.d.ts +8 -0
- package/dist/src/utils/channel.js +21 -10
- package/dist/src/utils/channel.js.map +1 -1
- package/dist/src/utils/cryptoUtils.d.ts +11 -0
- package/dist/src/utils/cryptoUtils.js +19 -0
- package/dist/src/utils/cryptoUtils.js.map +1 -0
- package/dist/src/utils/cryptoUtils.test.d.ts +6 -0
- package/dist/src/utils/cryptoUtils.test.js +31 -0
- package/dist/src/utils/cryptoUtils.test.js.map +1 -0
- package/dist/src/utils/editor.d.ts +29 -3
- package/dist/src/utils/editor.js +94 -3
- package/dist/src/utils/editor.js.map +1 -1
- package/dist/src/utils/editor.test.js +176 -2
- package/dist/src/utils/editor.test.js.map +1 -1
- package/dist/src/utils/environmentContext.d.ts +2 -1
- package/dist/src/utils/environmentContext.js +15 -8
- package/dist/src/utils/environmentContext.js.map +1 -1
- package/dist/src/utils/environmentContext.test.js +4 -14
- package/dist/src/utils/environmentContext.test.js.map +1 -1
- package/dist/src/utils/errors.js +3 -8
- package/dist/src/utils/errors.js.map +1 -1
- package/dist/src/utils/events.d.ts +26 -1
- package/dist/src/utils/events.js +21 -2
- package/dist/src/utils/events.js.map +1 -1
- package/dist/src/utils/events.test.js +39 -0
- package/dist/src/utils/events.test.js.map +1 -1
- package/dist/src/utils/extensionLoader.js +2 -2
- package/dist/src/utils/extensionLoader.js.map +1 -1
- package/dist/src/utils/extensionLoader.test.js +22 -15
- package/dist/src/utils/extensionLoader.test.js.map +1 -1
- package/dist/src/utils/fetch.d.ts +3 -3
- package/dist/src/utils/fetch.js +44 -20
- package/dist/src/utils/fetch.js.map +1 -1
- package/dist/src/utils/fetch.test.js +93 -17
- package/dist/src/utils/fetch.test.js.map +1 -1
- package/dist/src/utils/fileUtils.js +4 -1
- package/dist/src/utils/fileUtils.js.map +1 -1
- package/dist/src/utils/filesearch/fileSearch.js +20 -9
- package/dist/src/utils/filesearch/fileSearch.js.map +1 -1
- package/dist/src/utils/filesearch/ignore.js +4 -1
- package/dist/src/utils/filesearch/ignore.js.map +1 -1
- package/dist/src/utils/generateContentResponseUtilities.d.ts +1 -0
- package/dist/src/utils/generateContentResponseUtilities.js +37 -7
- package/dist/src/utils/generateContentResponseUtilities.js.map +1 -1
- package/dist/src/utils/generateContentResponseUtilities.test.js +33 -0
- package/dist/src/utils/generateContentResponseUtilities.test.js.map +1 -1
- package/dist/src/utils/gitUtils.d.ts +5 -0
- package/dist/src/utils/gitUtils.js +11 -0
- package/dist/src/utils/gitUtils.js.map +1 -1
- package/dist/src/utils/historyHardening.d.ts +37 -0
- package/dist/src/utils/historyHardening.js +332 -0
- package/dist/src/utils/historyHardening.js.map +1 -0
- package/dist/src/utils/historyHardening.test.d.ts +6 -0
- package/dist/src/utils/historyHardening.test.js +317 -0
- package/dist/src/utils/historyHardening.test.js.map +1 -0
- package/dist/src/utils/ignoreFileParser.js +1 -1
- package/dist/src/utils/ignoreFileParser.js.map +1 -1
- package/dist/src/utils/ignorePatterns.js +2 -0
- package/dist/src/utils/ignorePatterns.js.map +1 -1
- package/dist/src/utils/ignorePatterns.test.js +1 -0
- package/dist/src/utils/ignorePatterns.test.js.map +1 -1
- package/dist/src/utils/memoryDiscovery.d.ts +0 -20
- package/dist/src/utils/memoryDiscovery.js +57 -220
- package/dist/src/utils/memoryDiscovery.js.map +1 -1
- package/dist/src/utils/memoryDiscovery.test.js +112 -403
- package/dist/src/utils/memoryDiscovery.test.js.map +1 -1
- package/dist/src/utils/modelUtils.d.ts +14 -0
- package/dist/src/utils/modelUtils.js +17 -0
- package/dist/src/utils/modelUtils.js.map +1 -0
- package/dist/src/utils/modelUtils.test.d.ts +6 -0
- package/dist/src/utils/modelUtils.test.js +23 -0
- package/dist/src/utils/modelUtils.test.js.map +1 -0
- package/dist/src/utils/partUtils.d.ts +26 -1
- package/dist/src/utils/partUtils.js +37 -0
- package/dist/src/utils/partUtils.js.map +1 -1
- package/dist/src/utils/path-validator.d.ts +17 -0
- package/dist/src/utils/path-validator.js +76 -0
- package/dist/src/utils/path-validator.js.map +1 -0
- package/dist/src/utils/path-validator.test.d.ts +6 -0
- package/dist/src/utils/path-validator.test.js +91 -0
- package/dist/src/utils/path-validator.test.js.map +1 -0
- package/dist/src/utils/pathReader.js +12 -0
- package/dist/src/utils/pathReader.js.map +1 -1
- package/dist/src/utils/pathReader.test.js +95 -0
- package/dist/src/utils/pathReader.test.js.map +1 -1
- package/dist/src/utils/paths.d.ts +19 -1
- package/dist/src/utils/paths.js +74 -9
- package/dist/src/utils/paths.js.map +1 -1
- package/dist/src/utils/paths.test.js +111 -1
- package/dist/src/utils/paths.test.js.map +1 -1
- package/dist/src/utils/quotaErrorDetection.js +23 -12
- package/dist/src/utils/quotaErrorDetection.js.map +1 -1
- package/dist/src/utils/ragLogger.d.ts +32 -0
- package/dist/src/utils/ragLogger.js +56 -0
- package/dist/src/utils/ragLogger.js.map +1 -0
- package/dist/src/utils/ragLogger.test.d.ts +6 -0
- package/dist/src/utils/ragLogger.test.js +97 -0
- package/dist/src/utils/ragLogger.test.js.map +1 -0
- package/dist/src/utils/retry.js +8 -3
- package/dist/src/utils/retry.js.map +1 -1
- package/dist/src/utils/retry.test.js +17 -0
- package/dist/src/utils/retry.test.js.map +1 -1
- package/dist/src/utils/safeJsonStringify.js +0 -2
- package/dist/src/utils/safeJsonStringify.js.map +1 -1
- package/dist/src/utils/sessionOperations.d.ts +26 -0
- package/dist/src/utils/sessionOperations.js +177 -8
- package/dist/src/utils/sessionOperations.js.map +1 -1
- package/dist/src/utils/sessionUtils.d.ts +11 -5
- package/dist/src/utils/sessionUtils.js +139 -68
- package/dist/src/utils/sessionUtils.js.map +1 -1
- package/dist/src/utils/sessionUtils.test.js +31 -5
- package/dist/src/utils/sessionUtils.test.js.map +1 -1
- package/dist/src/utils/shell-utils.d.ts +9 -1
- package/dist/src/utils/shell-utils.js +51 -20
- package/dist/src/utils/shell-utils.js.map +1 -1
- package/dist/src/utils/shell-utils.test.js +87 -46
- package/dist/src/utils/shell-utils.test.js.map +1 -1
- package/dist/src/utils/textUtils.d.ts +12 -2
- package/dist/src/utils/textUtils.js +30 -3
- package/dist/src/utils/textUtils.js.map +1 -1
- package/dist/src/utils/textUtils.test.js +96 -1
- package/dist/src/utils/textUtils.test.js.map +1 -1
- package/dist/src/utils/tokenCalculation.d.ts +3 -2
- package/dist/src/utils/tokenCalculation.js +9 -5
- package/dist/src/utils/tokenCalculation.js.map +1 -1
- package/dist/src/utils/tokenCalculation.test.js +15 -0
- package/dist/src/utils/tokenCalculation.test.js.map +1 -1
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/package.json +18 -18
- package/dist/src/context/historyObserver.d.ts +0 -28
- package/dist/src/context/historyObserver.js +0 -63
- package/dist/src/context/historyObserver.js.map +0 -1
- package/dist/src/policy/memory-manager-policy.test.js +0 -80
- package/dist/src/policy/memory-manager-policy.test.js.map +0 -1
- package/dist/src/policy/policies/memory-manager.toml +0 -20
- package/dist/src/prompts/snippets-memory-v2.test.js.map +0 -1
- package/dist/src/tools/xcode-mcp-fix-transport.js.map +0 -1
- package/dist/src/tools/xcode-mcp-fix-transport.test.d.ts +0 -1
- package/dist/src/tools/xcode-mcp-fix-transport.test.js +0 -98
- package/dist/src/tools/xcode-mcp-fix-transport.test.js.map +0 -1
- package/dist/src/utils/systemEncoding.d.ts +0 -40
- package/dist/src/utils/systemEncoding.js +0 -150
- package/dist/src/utils/systemEncoding.js.map +0 -1
- package/dist/src/utils/systemEncoding.test.js +0 -369
- package/dist/src/utils/systemEncoding.test.js.map +0 -1
- /package/dist/src/{policy/memory-manager-policy.test.d.ts → agents/local-session-invocation.test.d.ts} +0 -0
- /package/dist/src/{prompts/snippets-memory-v2.test.d.ts → agents/local-subagent-protocol.test.d.ts} +0 -0
- /package/dist/src/{utils/systemEncoding.test.d.ts → policy/stable-stringify.test.d.ts} +0 -0
|
@@ -6,6 +6,7 @@
|
|
|
6
6
|
import { describe, it, expect } from 'vitest';
|
|
7
7
|
import { createRollingSummaryProcessor } from './rollingSummaryProcessor.js';
|
|
8
8
|
import { createMockProcessArgs, createMockEnvironment, createDummyNode, } from '../testing/contextTestUtils.js';
|
|
9
|
+
import { NodeType } from '../graph/types.js';
|
|
9
10
|
describe('RollingSummaryProcessor', () => {
|
|
10
11
|
it('should initialize with correct default options', () => {
|
|
11
12
|
const env = createMockEnvironment();
|
|
@@ -25,9 +26,9 @@ describe('RollingSummaryProcessor', () => {
|
|
|
25
26
|
});
|
|
26
27
|
const text50 = 'A'.repeat(50);
|
|
27
28
|
const targets = [
|
|
28
|
-
createDummyNode('ep1',
|
|
29
|
-
createDummyNode('ep1',
|
|
30
|
-
createDummyNode('ep1',
|
|
29
|
+
createDummyNode('ep1', NodeType.USER_PROMPT, 50, { payload: { text: text50 } }, 'id1'),
|
|
30
|
+
createDummyNode('ep1', NodeType.AGENT_THOUGHT, 50, { payload: { text: text50 } }, 'id2'),
|
|
31
|
+
createDummyNode('ep1', NodeType.AGENT_YIELD, 50, { payload: { text: text50 } }, 'id3'),
|
|
31
32
|
];
|
|
32
33
|
const result = await processor.process(createMockProcessArgs(targets));
|
|
33
34
|
// 3 nodes at 50 cost each.
|
|
@@ -35,8 +36,8 @@ describe('RollingSummaryProcessor', () => {
|
|
|
35
36
|
// Node id2 adds 50 deficit. Node id3 adds 50 deficit. Total = 100 deficit, which hits the target break point.
|
|
36
37
|
// Thus, id2 and id3 are summarized into a new ROLLING_SUMMARY node.
|
|
37
38
|
expect(result.length).toBe(2);
|
|
38
|
-
expect(result[0].type).toBe(
|
|
39
|
-
expect(result[1].type).toBe(
|
|
39
|
+
expect(result[0].type).toBe(NodeType.USER_PROMPT);
|
|
40
|
+
expect(result[1].type).toBe(NodeType.ROLLING_SUMMARY);
|
|
40
41
|
});
|
|
41
42
|
it('should preserve targets if deficit does not trigger summary', async () => {
|
|
42
43
|
const env = createMockEnvironment();
|
|
@@ -47,14 +48,14 @@ describe('RollingSummaryProcessor', () => {
|
|
|
47
48
|
});
|
|
48
49
|
const text10 = 'A'.repeat(10);
|
|
49
50
|
const targets = [
|
|
50
|
-
createDummyNode('ep1',
|
|
51
|
-
createDummyNode('ep1',
|
|
51
|
+
createDummyNode('ep1', NodeType.USER_PROMPT, 10, { payload: { text: text10 } }, 'id1'),
|
|
52
|
+
createDummyNode('ep1', NodeType.AGENT_THOUGHT, 10, { payload: { text: text10 } }, 'id2'),
|
|
52
53
|
];
|
|
53
54
|
const result = await processor.process(createMockProcessArgs(targets));
|
|
54
55
|
// Deficit accumulator reaches 10. This is < 100 limit, and total summarizable nodes < 2 anyway.
|
|
55
56
|
expect(result.length).toBe(2);
|
|
56
|
-
expect(result[0].type).toBe(
|
|
57
|
-
expect(result[1].type).toBe(
|
|
57
|
+
expect(result[0].type).toBe(NodeType.USER_PROMPT);
|
|
58
|
+
expect(result[1].type).toBe(NodeType.AGENT_THOUGHT);
|
|
58
59
|
});
|
|
59
60
|
});
|
|
60
61
|
//# sourceMappingURL=rollingSummaryProcessor.test.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"rollingSummaryProcessor.test.js","sourceRoot":"","sources":["../../../../src/context/processors/rollingSummaryProcessor.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,6BAA6B,EAAE,MAAM,8BAA8B,CAAC;AAC7E,OAAO,EACL,qBAAqB,EACrB,qBAAqB,EACrB,eAAe,GAChB,MAAM,gCAAgC,CAAC;
|
|
1
|
+
{"version":3,"file":"rollingSummaryProcessor.test.js","sourceRoot":"","sources":["../../../../src/context/processors/rollingSummaryProcessor.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,6BAA6B,EAAE,MAAM,8BAA8B,CAAC;AAC7E,OAAO,EACL,qBAAqB,EACrB,qBAAqB,EACrB,eAAe,GAChB,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,MAAM,SAAS,GAAG,6BAA6B,CAC7C,yBAAyB,EACzB,GAAG,EACH;YACE,MAAM,EAAE,aAAa;SACtB,CACF,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,0EAA0E;QAC1E,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QAEpC,sCAAsC;QACtC,iDAAiD;QACjD,MAAM,SAAS,GAAG,6BAA6B,CAC7C,yBAAyB,EACzB,GAAG,EACH;YACE,MAAM,EAAE,aAAa;YACrB,gBAAgB,EAAE,GAAG;SACtB,CACF,CAAC;QAEF,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC9B,MAAM,OAAO,GAAG;YACd,eAAe,CACb,KAAK,EACL,QAAQ,CAAC,WAAW,EACpB,EAAE,EACF,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAC7B,KAAK,CACN;YACD,eAAe,CACb,KAAK,EACL,QAAQ,CAAC,aAAa,EACtB,EAAE,EACF,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAC7B,KAAK,CACN;YACD,eAAe,CACb,KAAK,EACL,QAAQ,CAAC,WAAW,EACpB,EAAE,EACF,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAC7B,KAAK,CACN;SACF,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC;QAEvE,2BAA2B;QAC3B,oGAAoG;QACpG,8GAA8G;QAC9G,oEAAoE;QACpE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QAEpC,2EAA2E;QAC3E,MAAM,SAAS,GAAG,6BAA6B,CAC7C,yBAAyB,EACzB,GAAG,EACH;YACE,MAAM,EAAE,aAAa;YACrB,gBAAgB,EAAE,GAAG;SACtB,CACF,CAAC;QAEF,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC9B,MAAM,OAAO,GAAG;YACd,eAAe,CACb,KAAK,EACL,QAAQ,CAAC,WAAW,EACpB,EAAE,EACF,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAC7B,KAAK,CACN;YACD,eAAe,CACb,KAAK,EACL,QAAQ,CAAC,aAAa,EACtB,EAAE,EACF,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAC7B,KAAK,CACN;SACF,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC;QAEvE,gGAAgG;QAChG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1,9 +1,16 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2026 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
1
6
|
import type { JSONSchemaType } from 'ajv';
|
|
2
7
|
import type { AsyncContextProcessor } from '../pipeline.js';
|
|
3
8
|
import type { ContextEnvironment } from '../pipeline/environment.js';
|
|
4
9
|
export interface StateSnapshotAsyncProcessorOptions {
|
|
5
10
|
type?: 'accumulate' | 'point-in-time';
|
|
6
11
|
systemInstruction?: string;
|
|
12
|
+
maxSummaryTurns?: number;
|
|
13
|
+
maxStateTokens?: number;
|
|
7
14
|
}
|
|
8
15
|
export declare const StateSnapshotAsyncProcessorOptionsSchema: JSONSchemaType<StateSnapshotAsyncProcessorOptions>;
|
|
9
16
|
export declare function createStateSnapshotAsyncProcessor(id: string, env: ContextEnvironment, options: StateSnapshotAsyncProcessorOptions): AsyncContextProcessor;
|
|
@@ -1,11 +1,6 @@
|
|
|
1
|
-
|
|
2
|
-
* @license
|
|
3
|
-
* Copyright 2026 Google LLC
|
|
4
|
-
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
-
*/
|
|
6
|
-
import { randomUUID } from 'node:crypto';
|
|
7
|
-
import { SnapshotGenerator } from '../utils/snapshotGenerator.js';
|
|
1
|
+
import { SnapshotGenerator, findLatestSnapshotBaseline, } from '../utils/snapshotGenerator.js';
|
|
8
2
|
import { debugLogger } from '../../utils/debugLogger.js';
|
|
3
|
+
import { NodeType } from '../graph/types.js';
|
|
9
4
|
export const StateSnapshotAsyncProcessorOptionsSchema = {
|
|
10
5
|
type: 'object',
|
|
11
6
|
properties: {
|
|
@@ -15,6 +10,8 @@ export const StateSnapshotAsyncProcessorOptionsSchema = {
|
|
|
15
10
|
nullable: true,
|
|
16
11
|
},
|
|
17
12
|
systemInstruction: { type: 'string', nullable: true },
|
|
13
|
+
maxSummaryTurns: { type: 'number', nullable: true },
|
|
14
|
+
maxStateTokens: { type: 'number', nullable: true },
|
|
18
15
|
},
|
|
19
16
|
required: [],
|
|
20
17
|
};
|
|
@@ -27,11 +24,12 @@ export function createStateSnapshotAsyncProcessor(id, env, options) {
|
|
|
27
24
|
if (targets.length === 0)
|
|
28
25
|
return;
|
|
29
26
|
try {
|
|
30
|
-
let nodesToSummarize = [...targets];
|
|
31
27
|
let previousConsumedIds = [];
|
|
32
28
|
const processorType = options.type ?? 'point-in-time';
|
|
29
|
+
const nodesToSummarize = [...targets];
|
|
30
|
+
let previousStateJson = undefined;
|
|
33
31
|
if (processorType === 'accumulate') {
|
|
34
|
-
// Look for the most recent unconsumed accumulate snapshot in the inbox
|
|
32
|
+
// 1. Look for the most recent unconsumed accumulate snapshot in the inbox
|
|
35
33
|
const proposedSnapshots = inbox.getMessages('PROPOSED_SNAPSHOT');
|
|
36
34
|
const accumulateSnapshots = proposedSnapshots.filter((s) => s.payload.type === 'accumulate');
|
|
37
35
|
if (accumulateSnapshots.length > 0) {
|
|
@@ -43,18 +41,37 @@ export function createStateSnapshotAsyncProcessor(id, env, options) {
|
|
|
43
41
|
// because we are effectively "taking" it from the shelf to modify.
|
|
44
42
|
env.inbox.drainConsumed(new Set([latest.id]));
|
|
45
43
|
previousConsumedIds = latest.payload.consumedIds;
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
44
|
+
previousStateJson = latest.payload.newText;
|
|
45
|
+
}
|
|
46
|
+
else {
|
|
47
|
+
// 2. Global Lookback: No draft in inbox, scan the context graph for the last live snapshot
|
|
48
|
+
const baseline = findLatestSnapshotBaseline(targets);
|
|
49
|
+
if (baseline) {
|
|
50
|
+
previousStateJson = baseline.text;
|
|
51
|
+
previousConsumedIds = [...baseline.abstractsIds];
|
|
52
|
+
}
|
|
53
|
+
else {
|
|
54
|
+
debugLogger.log('[StateSnapshotAsyncProcessor] No previous snapshot found in Inbox or Graph. Initializing new Master State baseline in background.');
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
// If the snapshot happens to be inside our summary window, remove it so the LLM doesn't read it as raw transcript
|
|
59
|
+
if (previousStateJson) {
|
|
60
|
+
const summaryIdx = nodesToSummarize.findIndex((n) => n.type === NodeType.SNAPSHOT &&
|
|
61
|
+
n.payload.text === previousStateJson);
|
|
62
|
+
if (summaryIdx !== -1) {
|
|
63
|
+
nodesToSummarize.splice(summaryIdx, 1);
|
|
55
64
|
}
|
|
56
65
|
}
|
|
57
|
-
|
|
66
|
+
if (nodesToSummarize.length === 0)
|
|
67
|
+
return;
|
|
68
|
+
const snapshotText = await generator.synthesizeSnapshot(nodesToSummarize, previousStateJson, {
|
|
69
|
+
maxSummaryTurns: options.maxSummaryTurns,
|
|
70
|
+
maxStateTokens: options.maxStateTokens,
|
|
71
|
+
});
|
|
72
|
+
env.tracer.logEvent('StateSnapshotAsyncProcessor', 'Snapshot Synthesized', {
|
|
73
|
+
snapshotText,
|
|
74
|
+
});
|
|
58
75
|
const newConsumedIds = [
|
|
59
76
|
...previousConsumedIds,
|
|
60
77
|
...targets.map((t) => t.id),
|
|
@@ -64,6 +81,7 @@ export function createStateSnapshotAsyncProcessor(id, env, options) {
|
|
|
64
81
|
newText: snapshotText,
|
|
65
82
|
consumedIds: newConsumedIds,
|
|
66
83
|
type: processorType,
|
|
84
|
+
timestamp: targets[targets.length - 1].timestamp,
|
|
67
85
|
});
|
|
68
86
|
}
|
|
69
87
|
catch (e) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stateSnapshotAsyncProcessor.js","sourceRoot":"","sources":["../../../../src/context/processors/stateSnapshotAsyncProcessor.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"stateSnapshotAsyncProcessor.js","sourceRoot":"","sources":["../../../../src/context/processors/stateSnapshotAsyncProcessor.ts"],"names":[],"mappings":"AAQA,OAAO,EACL,iBAAiB,EACjB,0BAA0B,GAC3B,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAS7C,MAAM,CAAC,MAAM,wCAAwC,GACnD;IACE,IAAI,EAAE,QAAQ;IACd,UAAU,EAAE;QACV,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,CAAC,YAAY,EAAE,eAAe,CAAC;YACrC,QAAQ,EAAE,IAAI;SACf;QACD,iBAAiB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;QACrD,eAAe,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;QACnD,cAAc,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;KACnD;IACD,QAAQ,EAAE,EAAE;CACb,CAAC;AAEJ,MAAM,UAAU,iCAAiC,CAC/C,EAAU,EACV,GAAuB,EACvB,OAA2C;IAE3C,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC,GAAG,CAAC,CAAC;IAE7C,OAAO;QACL,EAAE;QACF,IAAI,EAAE,6BAA6B;QACnC,OAAO,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,EAAe,EAAiB,EAAE;YAChE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO;YAEjC,IAAI,CAAC;gBACH,IAAI,mBAAmB,GAAa,EAAE,CAAC;gBACvC,MAAM,aAAa,GAAG,OAAO,CAAC,IAAI,IAAI,eAAe,CAAC;gBACtD,MAAM,gBAAgB,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC;gBACtC,IAAI,iBAAiB,GAAuB,SAAS,CAAC;gBAEtD,IAAI,aAAa,KAAK,YAAY,EAAE,CAAC;oBACnC,0EAA0E;oBAC1E,MAAM,iBAAiB,GAAG,KAAK,CAAC,WAAW,CAIxC,mBAAmB,CAAC,CAAC;oBACxB,MAAM,mBAAmB,GAAG,iBAAiB,CAAC,MAAM,CAClD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,KAAK,YAAY,CACvC,CAAC;oBAEF,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACnC,+BAA+B;wBAC/B,MAAM,MAAM,GAAG,CAAC,GAAG,mBAAmB,CAAC,CAAC,IAAI,CAC1C,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CACpC,CAAC,CAAC,CAAC,CAAC;wBAEL,uEAAuE;wBACvE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;wBACzB,0EAA0E;wBAC1E,mEAAmE;wBACnE,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;wBAE9C,mBAAmB,GAAG,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;wBACjD,iBAAiB,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;oBAC7C,CAAC;yBAAM,CAAC;wBACN,2FAA2F;wBAC3F,MAAM,QAAQ,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;wBAErD,IAAI,QAAQ,EAAE,CAAC;4BACb,iBAAiB,GAAG,QAAQ,CAAC,IAAI,CAAC;4BAClC,mBAAmB,GAAG,CAAC,GAAG,QAAQ,CAAC,YAAY,CAAC,CAAC;wBACnD,CAAC;6BAAM,CAAC;4BACN,WAAW,CAAC,GAAG,CACb,mIAAmI,CACpI,CAAC;wBACJ,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,kHAAkH;gBAClH,IAAI,iBAAiB,EAAE,CAAC;oBACtB,MAAM,UAAU,GAAG,gBAAgB,CAAC,SAAS,CAC3C,CAAC,CAAC,EAAE,EAAE,CACJ,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,QAAQ;wBAC5B,CAAC,CAAC,OAAO,CAAC,IAAI,KAAK,iBAAiB,CACvC,CAAC;oBACF,IAAI,UAAU,KAAK,CAAC,CAAC,EAAE,CAAC;wBACtB,gBAAgB,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;oBACzC,CAAC;gBACH,CAAC;gBAED,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC;oBAAE,OAAO;gBAE1C,MAAM,YAAY,GAAG,MAAM,SAAS,CAAC,kBAAkB,CACrD,gBAAgB,EAChB,iBAAiB,EACjB;oBACE,eAAe,EAAE,OAAO,CAAC,eAAe;oBACxC,cAAc,EAAE,OAAO,CAAC,cAAc;iBACvC,CACF,CAAC;gBAEF,GAAG,CAAC,MAAM,CAAC,QAAQ,CACjB,6BAA6B,EAC7B,sBAAsB,EACtB;oBACE,YAAY;iBACb,CACF,CAAC;gBAEF,MAAM,cAAc,GAAG;oBACrB,GAAG,mBAAmB;oBACtB,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC5B,CAAC;gBAEF,6FAA6F;gBAC7F,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,mBAAmB,EAAE;oBACrC,OAAO,EAAE,YAAY;oBACrB,WAAW,EAAE,cAAc;oBAC3B,IAAI,EAAE,aAAa;oBACnB,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,SAAS;iBACjD,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,WAAW,CAAC,KAAK,CACf,yDAAyD,EACzD,CAAC,CACF,CAAC;YACJ,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -6,21 +6,22 @@
|
|
|
6
6
|
import { describe, it, expect, vi } from 'vitest';
|
|
7
7
|
import { createStateSnapshotAsyncProcessor } from './stateSnapshotAsyncProcessor.js';
|
|
8
8
|
import { createMockEnvironment, createDummyNode, createMockProcessArgs, } from '../testing/contextTestUtils.js';
|
|
9
|
+
import { NodeType } from '../graph/types.js';
|
|
9
10
|
describe('StateSnapshotAsyncProcessor', () => {
|
|
10
11
|
it('should generate a snapshot and publish it to the inbox', async () => {
|
|
11
12
|
const env = createMockEnvironment();
|
|
12
13
|
// Spy on the publish method
|
|
13
14
|
const publishSpy = vi.spyOn(env.inbox, 'publish');
|
|
14
15
|
const worker = createStateSnapshotAsyncProcessor('StateSnapshotAsyncProcessor', env, { type: 'point-in-time' });
|
|
15
|
-
const nodeA = createDummyNode('ep1',
|
|
16
|
-
const nodeB = createDummyNode('ep1',
|
|
16
|
+
const nodeA = createDummyNode('ep1', NodeType.USER_PROMPT, 50, {}, 'node-A');
|
|
17
|
+
const nodeB = createDummyNode('ep1', NodeType.AGENT_THOUGHT, 60, {}, 'node-B');
|
|
17
18
|
const targets = [nodeA, nodeB];
|
|
18
19
|
await worker.process(createMockProcessArgs(targets, targets, []));
|
|
19
|
-
// Ensure
|
|
20
|
-
expect(env.llmClient.
|
|
20
|
+
// Ensure generateJson was called
|
|
21
|
+
expect(env.llmClient.generateJson).toHaveBeenCalled();
|
|
21
22
|
// Verify it published to the inbox
|
|
22
23
|
expect(publishSpy).toHaveBeenCalledWith('PROPOSED_SNAPSHOT', expect.objectContaining({
|
|
23
|
-
newText: 'Mock LLM summary response',
|
|
24
|
+
newText: '{"active_tasks":[],"discovered_facts":[],"constraints_and_preferences":[],"recent_arc":["Mock LLM summary response"]}',
|
|
24
25
|
consumedIds: ['node-A', 'node-B'],
|
|
25
26
|
type: 'point-in-time',
|
|
26
27
|
}));
|
|
@@ -30,7 +31,7 @@ describe('StateSnapshotAsyncProcessor', () => {
|
|
|
30
31
|
const publishSpy = vi.spyOn(env.inbox, 'publish');
|
|
31
32
|
const drainSpy = vi.spyOn(env.inbox, 'drainConsumed');
|
|
32
33
|
const worker = createStateSnapshotAsyncProcessor('StateSnapshotAsyncProcessor', env, { type: 'accumulate' });
|
|
33
|
-
const nodeC = createDummyNode('ep2',
|
|
34
|
+
const nodeC = createDummyNode('ep2', NodeType.USER_PROMPT, 50, {}, 'node-C');
|
|
34
35
|
const targets = [nodeC];
|
|
35
36
|
const inboxMessages = [
|
|
36
37
|
{
|
|
@@ -51,17 +52,17 @@ describe('StateSnapshotAsyncProcessor', () => {
|
|
|
51
52
|
expect(drainSpy).toHaveBeenCalledWith(expect.any(Set));
|
|
52
53
|
// The new publish should contain ALL consumed IDs (old + new)
|
|
53
54
|
expect(publishSpy).toHaveBeenCalledWith('PROPOSED_SNAPSHOT', expect.objectContaining({
|
|
54
|
-
newText: 'Mock LLM summary response',
|
|
55
|
+
newText: '{"active_tasks":[],"discovered_facts":[],"constraints_and_preferences":[],"recent_arc":["Mock LLM summary response"]}',
|
|
55
56
|
consumedIds: ['node-A', 'node-B', 'node-C'], // Aggregated!
|
|
56
57
|
type: 'accumulate',
|
|
57
58
|
}));
|
|
58
|
-
// Verify the LLM was called with the old snapshot
|
|
59
|
-
expect(env.llmClient.
|
|
59
|
+
// Verify the LLM was called with the old snapshot provided in the prompt
|
|
60
|
+
expect(env.llmClient.generateJson).toHaveBeenCalledWith(expect.objectContaining({
|
|
60
61
|
contents: expect.arrayContaining([
|
|
61
62
|
expect.objectContaining({
|
|
62
63
|
parts: expect.arrayContaining([
|
|
63
64
|
expect.objectContaining({
|
|
64
|
-
text: expect.stringContaining('
|
|
65
|
+
text: expect.stringContaining('CURRENT MASTER STATE'),
|
|
65
66
|
}),
|
|
66
67
|
]),
|
|
67
68
|
}),
|
|
@@ -76,5 +77,29 @@ describe('StateSnapshotAsyncProcessor', () => {
|
|
|
76
77
|
expect(env.llmClient.generateContent).not.toHaveBeenCalled();
|
|
77
78
|
expect(publishSpy).not.toHaveBeenCalled();
|
|
78
79
|
});
|
|
80
|
+
it('should use Global Lookback to find an existing snapshot in the graph when inbox is empty', async () => {
|
|
81
|
+
const env = createMockEnvironment();
|
|
82
|
+
const worker = createStateSnapshotAsyncProcessor('StateSnapshotAsyncProcessor', env, { type: 'accumulate' });
|
|
83
|
+
// Create an old snapshot with existing JSON state
|
|
84
|
+
const oldStateJson = JSON.stringify({
|
|
85
|
+
discovered_facts: ['Global Lookback Async Works!'],
|
|
86
|
+
});
|
|
87
|
+
const oldSnapshot = createDummyNode('ep1', NodeType.SNAPSHOT, 10, { payload: { text: oldStateJson } }, 'old-snap');
|
|
88
|
+
const nodeC = createDummyNode('ep2', NodeType.USER_PROMPT, 50, {}, 'node-C');
|
|
89
|
+
const targets = [oldSnapshot, nodeC];
|
|
90
|
+
const args = createMockProcessArgs(targets, targets, []); // Empty inbox!
|
|
91
|
+
await worker.process(args);
|
|
92
|
+
expect(env.llmClient.generateJson).toHaveBeenCalledWith(expect.objectContaining({
|
|
93
|
+
contents: expect.arrayContaining([
|
|
94
|
+
expect.objectContaining({
|
|
95
|
+
parts: expect.arrayContaining([
|
|
96
|
+
expect.objectContaining({
|
|
97
|
+
text: expect.stringContaining('Global Lookback Async Works!'),
|
|
98
|
+
}),
|
|
99
|
+
]),
|
|
100
|
+
}),
|
|
101
|
+
]),
|
|
102
|
+
}));
|
|
103
|
+
});
|
|
79
104
|
});
|
|
80
105
|
//# sourceMappingURL=stateSnapshotAsyncProcessor.test.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stateSnapshotAsyncProcessor.test.js","sourceRoot":"","sources":["../../../../src/context/processors/stateSnapshotAsyncProcessor.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,iCAAiC,EAAE,MAAM,kCAAkC,CAAC;AACrF,OAAO,EACL,qBAAqB,EACrB,eAAe,EACf,qBAAqB,GACtB,MAAM,gCAAgC,CAAC;
|
|
1
|
+
{"version":3,"file":"stateSnapshotAsyncProcessor.test.js","sourceRoot":"","sources":["../../../../src/context/processors/stateSnapshotAsyncProcessor.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,iCAAiC,EAAE,MAAM,kCAAkC,CAAC;AACrF,OAAO,EACL,qBAAqB,EACrB,eAAe,EACf,qBAAqB,GACtB,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAI7C,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,4BAA4B;QAC5B,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,iCAAiC,CAC9C,6BAA6B,EAC7B,GAAG,EACH,EAAE,IAAI,EAAE,eAAe,EAAE,CAC1B,CAAC;QAEF,MAAM,KAAK,GAAG,eAAe,CAC3B,KAAK,EACL,QAAQ,CAAC,WAAW,EACpB,EAAE,EACF,EAAE,EACF,QAAQ,CACT,CAAC;QACF,MAAM,KAAK,GAAG,eAAe,CAC3B,KAAK,EACL,QAAQ,CAAC,aAAa,EACtB,EAAE,EACF,EAAE,EACF,QAAQ,CACT,CAAC;QAEF,MAAM,OAAO,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QAC/B,MAAM,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAElE,iCAAiC;QACjC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAEtD,mCAAmC;QACnC,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,mBAAmB,EACnB,MAAM,CAAC,gBAAgB,CAAC;YACtB,OAAO,EACL,uHAAuH;YACzH,WAAW,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC;YACjC,IAAI,EAAE,eAAe;SACtB,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAClD,MAAM,QAAQ,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;QAEtD,MAAM,MAAM,GAAG,iCAAiC,CAC9C,6BAA6B,EAC7B,GAAG,EACH,EAAE,IAAI,EAAE,YAAY,EAAE,CACvB,CAAC;QAEF,MAAM,KAAK,GAAG,eAAe,CAC3B,KAAK,EACL,QAAQ,CAAC,WAAW,EACpB,EAAE,EACF,EAAE,EACF,QAAQ,CACT,CAAC;QACF,MAAM,OAAO,GAAG,CAAC,KAAK,CAAC,CAAC;QAExB,MAAM,aAAa,GAAmB;YACpC;gBACE,EAAE,EAAE,SAAS;gBACb,KAAK,EAAE,mBAAmB;gBAC1B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI;gBAC5B,OAAO,EAAE;oBACP,WAAW,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC;oBACjC,OAAO,EAAE,gBAAgB;oBACzB,IAAI,EAAE,YAAY;iBACnB;aACF;SACF,CAAC;QAEF,MAAM,IAAI,GAAG,qBAAqB,CAAC,OAAO,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;QAEpE,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAE3B,mCAAmC;QACnC,MAAM,CACH,IAAI,CAAC,KAA2B,CAAC,cAAc,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,CAClE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACb,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;QAEvD,8DAA8D;QAC9D,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,mBAAmB,EACnB,MAAM,CAAC,gBAAgB,CAAC;YACtB,OAAO,EACL,uHAAuH;YACzH,WAAW,EAAE,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,EAAE,cAAc;YAC3D,IAAI,EAAE,YAAY;SACnB,CAAC,CACH,CAAC;QACF,yEAAyE;QACzE,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACrD,MAAM,CAAC,gBAAgB,CAAC;YACtB,QAAQ,EAAE,MAAM,CAAC,eAAe,CAAC;gBAC/B,MAAM,CAAC,gBAAgB,CAAC;oBACtB,KAAK,EAAE,MAAM,CAAC,eAAe,CAAC;wBAC5B,MAAM,CAAC,gBAAgB,CAAC;4BACtB,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,sBAAsB,CAAC;yBACtD,CAAC;qBACH,CAAC;iBACH,CAAC;aACH,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,iCAAiC,CAC9C,6BAA6B,EAC7B,GAAG,EACH,EAAE,IAAI,EAAE,YAAY,EAAE,CACvB,CAAC;QAEF,MAAM,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAExD,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC7D,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0FAA0F,EAAE,KAAK,IAAI,EAAE;QACxG,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QAEpC,MAAM,MAAM,GAAG,iCAAiC,CAC9C,6BAA6B,EAC7B,GAAG,EACH,EAAE,IAAI,EAAE,YAAY,EAAE,CACvB,CAAC;QAEF,kDAAkD;QAClD,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC;YAClC,gBAAgB,EAAE,CAAC,8BAA8B,CAAC;SACnD,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,eAAe,CACjC,KAAK,EACL,QAAQ,CAAC,QAAQ,EACjB,EAAE,EACF,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EACnC,UAAU,CACX,CAAC;QACF,MAAM,KAAK,GAAG,eAAe,CAC3B,KAAK,EACL,QAAQ,CAAC,WAAW,EACpB,EAAE,EACF,EAAE,EACF,QAAQ,CACT,CAAC;QAEF,MAAM,OAAO,GAAG,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;QACrC,MAAM,IAAI,GAAG,qBAAqB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,eAAe;QAEzE,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAE3B,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACrD,MAAM,CAAC,gBAAgB,CAAC;YACtB,QAAQ,EAAE,MAAM,CAAC,eAAe,CAAC;gBAC/B,MAAM,CAAC,gBAAgB,CAAC;oBACtB,KAAK,EAAE,MAAM,CAAC,eAAe,CAAC;wBAC5B,MAAM,CAAC,gBAAgB,CAAC;4BACtB,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,8BAA8B,CAAC;yBAC9D,CAAC;qBACH,CAAC;iBACH,CAAC;aACH,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -4,6 +4,8 @@ import type { ContextEnvironment } from '../pipeline/environment.js';
|
|
|
4
4
|
export interface StateSnapshotProcessorOptions extends BackstopTargetOptions {
|
|
5
5
|
model?: string;
|
|
6
6
|
systemInstruction?: string;
|
|
7
|
+
maxSummaryTurns?: number;
|
|
8
|
+
maxStateTokens?: number;
|
|
7
9
|
}
|
|
8
10
|
export declare const StateSnapshotProcessorOptionsSchema: JSONSchemaType<StateSnapshotProcessorOptions>;
|
|
9
11
|
export declare function createStateSnapshotProcessor(id: string, env: ContextEnvironment, options: StateSnapshotProcessorOptions): ContextProcessor;
|
|
@@ -3,8 +3,9 @@
|
|
|
3
3
|
* Copyright 2026 Google LLC
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
-
import {
|
|
7
|
-
import {
|
|
6
|
+
import { deriveStableId } from '../../utils/cryptoUtils.js';
|
|
7
|
+
import { NodeType } from '../graph/types.js';
|
|
8
|
+
import { SnapshotGenerator, findLatestSnapshotBaseline, } from '../utils/snapshotGenerator.js';
|
|
8
9
|
import { debugLogger } from '../../utils/debugLogger.js';
|
|
9
10
|
export const StateSnapshotProcessorOptionsSchema = {
|
|
10
11
|
type: 'object',
|
|
@@ -17,6 +18,8 @@ export const StateSnapshotProcessorOptionsSchema = {
|
|
|
17
18
|
freeTokensTarget: { type: 'number', nullable: true },
|
|
18
19
|
model: { type: 'string', nullable: true },
|
|
19
20
|
systemInstruction: { type: 'string', nullable: true },
|
|
21
|
+
maxSummaryTurns: { type: 'number', nullable: true },
|
|
22
|
+
maxStateTokens: { type: 'number', nullable: true },
|
|
20
23
|
},
|
|
21
24
|
required: [],
|
|
22
25
|
};
|
|
@@ -40,19 +43,24 @@ export function createStateSnapshotProcessor(id, env, options) {
|
|
|
40
43
|
// Sort by newest timestamp first (we want the most accumulated snapshot)
|
|
41
44
|
const sorted = [...matchingSnapshots].sort((a, b) => b.timestamp - a.timestamp);
|
|
42
45
|
for (const proposed of sorted) {
|
|
43
|
-
const { consumedIds, newText } = proposed.payload;
|
|
46
|
+
const { consumedIds, newText, timestamp } = proposed.payload;
|
|
44
47
|
// Verify all consumed IDs still exist sequentially in targets
|
|
45
48
|
const targetIds = new Set(targets.map((t) => t.id));
|
|
46
49
|
const isValid = consumedIds.every((id) => targetIds.has(id));
|
|
47
50
|
if (isValid) {
|
|
51
|
+
env.tracer.logEvent('StateSnapshotProcessor', 'Snapshot Spliced from Inbox', {
|
|
52
|
+
snapshotText: newText,
|
|
53
|
+
});
|
|
54
|
+
debugLogger.log(`[StateSnapshotProcessor] Successfully spliced PROPOSED_SNAPSHOT from Inbox into Graph. Consumed ${consumedIds.length} nodes.`);
|
|
48
55
|
// If valid, apply it!
|
|
49
|
-
const newId =
|
|
56
|
+
const newId = deriveStableId(consumedIds);
|
|
50
57
|
const snapshotNode = {
|
|
51
58
|
id: newId,
|
|
52
|
-
|
|
53
|
-
type:
|
|
54
|
-
timestamp: Date.now(),
|
|
55
|
-
|
|
59
|
+
turnId: newId,
|
|
60
|
+
type: NodeType.SNAPSHOT,
|
|
61
|
+
timestamp: timestamp ?? Date.now(),
|
|
62
|
+
role: 'user',
|
|
63
|
+
payload: { text: newText },
|
|
56
64
|
abstractsIds: consumedIds,
|
|
57
65
|
};
|
|
58
66
|
// Remove the consumed nodes and insert the snapshot at the earliest index
|
|
@@ -68,6 +76,9 @@ export function createStateSnapshotProcessor(id, env, options) {
|
|
|
68
76
|
inbox.consume(proposed.id);
|
|
69
77
|
return returnedNodes;
|
|
70
78
|
}
|
|
79
|
+
else {
|
|
80
|
+
debugLogger.log(`[StateSnapshotProcessor] Rejected PROPOSED_SNAPSHOT from Inbox because one or more target IDs were missing from the current graph window.`);
|
|
81
|
+
}
|
|
71
82
|
}
|
|
72
83
|
}
|
|
73
84
|
// 2. The Synchronous Backstop (The Slow Path)
|
|
@@ -85,11 +96,6 @@ export function createStateSnapshotProcessor(id, env, options) {
|
|
|
85
96
|
const nodesToSummarize = [];
|
|
86
97
|
// Scan oldest to newest
|
|
87
98
|
for (const node of targets) {
|
|
88
|
-
if (node.id === targets[0].id && node.type === 'USER_PROMPT') {
|
|
89
|
-
// Keep system prompt if it's the very first node
|
|
90
|
-
// In a real system, system prompt is protected, but we double check
|
|
91
|
-
continue;
|
|
92
|
-
}
|
|
93
99
|
nodesToSummarize.push(node);
|
|
94
100
|
deficitAccumulator += env.tokenCalculator.getTokenCost(node);
|
|
95
101
|
if (deficitAccumulator >= targetTokensToRemove)
|
|
@@ -97,18 +103,44 @@ export function createStateSnapshotProcessor(id, env, options) {
|
|
|
97
103
|
}
|
|
98
104
|
if (nodesToSummarize.length < 2)
|
|
99
105
|
return targets; // Not enough context
|
|
106
|
+
let previousStateJson = undefined;
|
|
107
|
+
let baselineIdToConsume = undefined;
|
|
108
|
+
// Global Lookback: Find the absolute most recent snapshot anywhere in the active context
|
|
109
|
+
const baseline = findLatestSnapshotBaseline(targets);
|
|
110
|
+
if (baseline) {
|
|
111
|
+
previousStateJson = baseline.text;
|
|
112
|
+
// If the snapshot happens to be inside our summary window, remove it so the LLM doesn't read it as raw transcript
|
|
113
|
+
const summaryIdx = nodesToSummarize.findIndex((n) => n.id === baseline.id);
|
|
114
|
+
if (summaryIdx !== -1) {
|
|
115
|
+
baselineIdToConsume = baseline.id;
|
|
116
|
+
nodesToSummarize.splice(summaryIdx, 1);
|
|
117
|
+
}
|
|
118
|
+
}
|
|
119
|
+
else {
|
|
120
|
+
debugLogger.log('[StateSnapshotProcessor] No previous snapshot found in context graph. Initializing new Master State baseline.');
|
|
121
|
+
}
|
|
100
122
|
try {
|
|
101
|
-
const snapshotText = await generator.synthesizeSnapshot(nodesToSummarize,
|
|
102
|
-
|
|
123
|
+
const snapshotText = await generator.synthesizeSnapshot(nodesToSummarize, previousStateJson, {
|
|
124
|
+
maxSummaryTurns: options.maxSummaryTurns,
|
|
125
|
+
maxStateTokens: options.maxStateTokens,
|
|
126
|
+
});
|
|
127
|
+
env.tracer.logEvent('StateSnapshotProcessor', 'Snapshot Synthesized', {
|
|
128
|
+
snapshotText,
|
|
129
|
+
});
|
|
130
|
+
const consumedIds = nodesToSummarize.map((n) => n.id);
|
|
131
|
+
if (baselineIdToConsume && !consumedIds.includes(baselineIdToConsume)) {
|
|
132
|
+
consumedIds.push(baselineIdToConsume);
|
|
133
|
+
}
|
|
134
|
+
const newId = deriveStableId(consumedIds);
|
|
103
135
|
const snapshotNode = {
|
|
104
136
|
id: newId,
|
|
105
|
-
|
|
106
|
-
type:
|
|
107
|
-
timestamp:
|
|
108
|
-
|
|
109
|
-
|
|
137
|
+
turnId: newId,
|
|
138
|
+
type: NodeType.SNAPSHOT,
|
|
139
|
+
timestamp: nodesToSummarize[nodesToSummarize.length - 1].timestamp,
|
|
140
|
+
role: 'user',
|
|
141
|
+
payload: { text: snapshotText },
|
|
142
|
+
abstractsIds: [...consumedIds],
|
|
110
143
|
};
|
|
111
|
-
const consumedIds = nodesToSummarize.map((n) => n.id);
|
|
112
144
|
const returnedNodes = targets.filter((t) => !consumedIds.includes(t.id));
|
|
113
145
|
const firstRemovedIdx = targets.findIndex((t) => consumedIds.includes(t.id));
|
|
114
146
|
if (firstRemovedIdx !== -1) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stateSnapshotProcessor.js","sourceRoot":"","sources":["../../../../src/context/processors/stateSnapshotProcessor.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"stateSnapshotProcessor.js","sourceRoot":"","sources":["../../../../src/context/processors/stateSnapshotProcessor.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAQ5D,OAAO,EAAoC,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC/E,OAAO,EACL,iBAAiB,EACjB,0BAA0B,GAC3B,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AASzD,MAAM,CAAC,MAAM,mCAAmC,GAC9C;IACE,IAAI,EAAE,QAAQ;IACd,UAAU,EAAE;QACV,MAAM,EAAE;YACN,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,CAAC,aAAa,EAAE,aAAa,EAAE,KAAK,CAAC;YAC3C,QAAQ,EAAE,IAAI;SACf;QACD,gBAAgB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;QACpD,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;QACzC,iBAAiB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;QACrD,eAAe,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;QACnD,cAAc,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;KACnD;IACD,QAAQ,EAAE,EAAE;CACb,CAAC;AAEJ,MAAM,UAAU,4BAA4B,CAC1C,EAAU,EACV,GAAuB,EACvB,OAAsC;IAEtC,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC,GAAG,CAAC,CAAC;IAE7C,OAAO;QACL,EAAE;QACF,IAAI,EAAE,wBAAwB;QAC9B,OAAO,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,EAAe,EAAE,EAAE;YACjD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,OAAO,CAAC;YACjB,CAAC;YAED,kGAAkG;YAClG,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,IAAI,KAAK,CAAC;YACzC,MAAM,YAAY,GAChB,QAAQ,KAAK,aAAa,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,YAAY,CAAC;YAE9D,0DAA0D;YAC1D,MAAM,iBAAiB,GAAG,KAAK,CAAC,WAAW,CAKxC,mBAAmB,CAAC,CAAC;YAExB,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACjC,+DAA+D;gBAC/D,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,MAAM,CAChD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,KAAK,YAAY,CACvC,CAAC;gBAEF,yEAAyE;gBACzE,MAAM,MAAM,GAAG,CAAC,GAAG,iBAAiB,CAAC,CAAC,IAAI,CACxC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CACpC,CAAC;gBAEF,KAAK,MAAM,QAAQ,IAAI,MAAM,EAAE,CAAC;oBAC9B,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,QAAQ,CAAC,OAAO,CAAC;oBAE7D,8DAA8D;oBAC9D,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;oBACpD,MAAM,OAAO,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;oBAE7D,IAAI,OAAO,EAAE,CAAC;wBACZ,GAAG,CAAC,MAAM,CAAC,QAAQ,CACjB,wBAAwB,EACxB,6BAA6B,EAC7B;4BACE,YAAY,EAAE,OAAO;yBACtB,CACF,CAAC;wBACF,WAAW,CAAC,GAAG,CACb,mGAAmG,WAAW,CAAC,MAAM,SAAS,CAC/H,CAAC;wBACF,sBAAsB;wBACtB,MAAM,KAAK,GAAG,cAAc,CAAC,WAAW,CAAC,CAAC;wBAE1C,MAAM,YAAY,GAAa;4BAC7B,EAAE,EAAE,KAAK;4BACT,MAAM,EAAE,KAAK;4BACb,IAAI,EAAE,QAAQ,CAAC,QAAQ;4BACvB,SAAS,EAAE,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE;4BAClC,IAAI,EAAE,MAAM;4BACZ,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE;4BAC1B,YAAY,EAAE,WAAW;yBAC1B,CAAC;wBAEF,0EAA0E;wBAC1E,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAClC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CACnC,CAAC;wBACF,MAAM,eAAe,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAC9C,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAC3B,CAAC;wBAEF,IAAI,eAAe,KAAK,CAAC,CAAC,EAAE,CAAC;4BAC3B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;4BACzC,aAAa,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,EAAE,YAAY,CAAC,CAAC;wBAC7C,CAAC;6BAAM,CAAC;4BACN,aAAa,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;wBACtC,CAAC;wBAED,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;wBAC3B,OAAO,aAAa,CAAC;oBACvB,CAAC;yBAAM,CAAC;wBACN,WAAW,CAAC,GAAG,CACb,2IAA2I,CAC5I,CAAC;oBACJ,CAAC;gBACH,CAAC;YACH,CAAC;YAED,8CAA8C;YAC9C,IAAI,oBAAoB,GAAG,CAAC,CAAC;YAE7B,IAAI,QAAQ,KAAK,aAAa,EAAE,CAAC;gBAC/B,oBAAoB,GAAG,QAAQ,CAAC,CAAC,yEAAyE;YAC5G,CAAC;iBAAM,IAAI,QAAQ,KAAK,aAAa,EAAE,CAAC;gBACtC,oBAAoB,GAAG,OAAO,CAAC,gBAAgB,IAAI,QAAQ,CAAC;YAC9D,CAAC;iBAAM,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;gBAC9B,oBAAoB,GAAG,QAAQ,CAAC;YAClC,CAAC;YAED,IAAI,kBAAkB,GAAG,CAAC,CAAC;YAC3B,MAAM,gBAAgB,GAAmB,EAAE,CAAC;YAE5C,wBAAwB;YACxB,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;gBAC3B,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC5B,kBAAkB,IAAI,GAAG,CAAC,eAAe,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;gBAE7D,IAAI,kBAAkB,IAAI,oBAAoB;oBAAE,MAAM;YACxD,CAAC;YAED,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC;gBAAE,OAAO,OAAO,CAAC,CAAC,qBAAqB;YAEtE,IAAI,iBAAiB,GAAuB,SAAS,CAAC;YACtD,IAAI,mBAAmB,GAAuB,SAAS,CAAC;YAExD,yFAAyF;YACzF,MAAM,QAAQ,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAC;YAErD,IAAI,QAAQ,EAAE,CAAC;gBACb,iBAAiB,GAAG,QAAQ,CAAC,IAAI,CAAC;gBAClC,kHAAkH;gBAClH,MAAM,UAAU,GAAG,gBAAgB,CAAC,SAAS,CAC3C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,QAAQ,CAAC,EAAE,CAC5B,CAAC;gBACF,IAAI,UAAU,KAAK,CAAC,CAAC,EAAE,CAAC;oBACtB,mBAAmB,GAAG,QAAQ,CAAC,EAAE,CAAC;oBAClC,gBAAgB,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,GAAG,CACb,+GAA+G,CAChH,CAAC;YACJ,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,YAAY,GAAG,MAAM,SAAS,CAAC,kBAAkB,CACrD,gBAAgB,EAChB,iBAAiB,EACjB;oBACE,eAAe,EAAE,OAAO,CAAC,eAAe;oBACxC,cAAc,EAAE,OAAO,CAAC,cAAc;iBACvC,CACF,CAAC;gBAEF,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,EAAE,sBAAsB,EAAE;oBACpE,YAAY;iBACb,CAAC,CAAC;gBAEH,MAAM,WAAW,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACtD,IAAI,mBAAmB,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;oBACtE,WAAW,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;gBACxC,CAAC;gBACD,MAAM,KAAK,GAAG,cAAc,CAAC,WAAW,CAAC,CAAC;gBAE1C,MAAM,YAAY,GAAa;oBAC7B,EAAE,EAAE,KAAK;oBACT,MAAM,EAAE,KAAK;oBACb,IAAI,EAAE,QAAQ,CAAC,QAAQ;oBACvB,SAAS,EAAE,gBAAgB,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,SAAS;oBAClE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE;oBAC/B,YAAY,EAAE,CAAC,GAAG,WAAW,CAAC;iBAC/B,CAAC;gBAEF,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAClC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CACnC,CAAC;gBACF,MAAM,eAAe,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAC9C,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAC3B,CAAC;gBAEF,IAAI,eAAe,KAAK,CAAC,CAAC,EAAE,CAAC;oBAC3B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;oBACzC,aAAa,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,EAAE,YAAY,CAAC,CAAC;gBAC7C,CAAC;qBAAM,CAAC;oBACN,aAAa,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;gBACtC,CAAC;gBAED,OAAO,aAAa,CAAC;YACvB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,WAAW,CAAC,KAAK,CAAC,6CAA6C,EAAE,CAAC,CAAC,CAAC;gBACpE,OAAO,OAAO,CAAC;YACjB,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -6,13 +6,14 @@
|
|
|
6
6
|
import { describe, it, expect } from 'vitest';
|
|
7
7
|
import { createStateSnapshotProcessor } from './stateSnapshotProcessor.js';
|
|
8
8
|
import { createMockEnvironment, createDummyNode, createMockProcessArgs, } from '../testing/contextTestUtils.js';
|
|
9
|
+
import { NodeType } from '../graph/types.js';
|
|
9
10
|
describe('StateSnapshotProcessor', () => {
|
|
10
11
|
it('should ignore if budget is satisfied', async () => {
|
|
11
12
|
const env = createMockEnvironment();
|
|
12
13
|
const processor = createStateSnapshotProcessor('StateSnapshotProcessor', env, {
|
|
13
14
|
target: 'incremental',
|
|
14
15
|
});
|
|
15
|
-
const targets = [createDummyNode('ep1',
|
|
16
|
+
const targets = [createDummyNode('ep1', NodeType.USER_PROMPT)];
|
|
16
17
|
const result = await processor.process(createMockProcessArgs(targets));
|
|
17
18
|
expect(result).toBe(targets); // Strict equality
|
|
18
19
|
});
|
|
@@ -21,9 +22,9 @@ describe('StateSnapshotProcessor', () => {
|
|
|
21
22
|
const processor = createStateSnapshotProcessor('StateSnapshotProcessor', env, {
|
|
22
23
|
target: 'incremental',
|
|
23
24
|
});
|
|
24
|
-
const nodeA = createDummyNode('ep1',
|
|
25
|
-
const nodeB = createDummyNode('ep1',
|
|
26
|
-
const nodeC = createDummyNode('ep2',
|
|
25
|
+
const nodeA = createDummyNode('ep1', NodeType.USER_PROMPT, 50, {}, 'node-A');
|
|
26
|
+
const nodeB = createDummyNode('ep1', NodeType.AGENT_THOUGHT, 60, {}, 'node-B');
|
|
27
|
+
const nodeC = createDummyNode('ep2', NodeType.USER_PROMPT, 50, {}, 'node-C');
|
|
27
28
|
const targets = [nodeA, nodeB, nodeC];
|
|
28
29
|
// The async background pipeline created a snapshot of A and B
|
|
29
30
|
const messages = [
|
|
@@ -42,7 +43,7 @@ describe('StateSnapshotProcessor', () => {
|
|
|
42
43
|
const result = await processor.process(processArgs);
|
|
43
44
|
// Should remove A and B, insert Snapshot, keep C
|
|
44
45
|
expect(result.length).toBe(2);
|
|
45
|
-
expect(result[0].type).toBe(
|
|
46
|
+
expect(result[0].type).toBe(NodeType.SNAPSHOT);
|
|
46
47
|
expect(result[1].id).toBe('node-C');
|
|
47
48
|
// Should consume the message
|
|
48
49
|
expect(processArgs.inbox.getConsumedIds().has('msg-1')).toBe(true);
|
|
@@ -54,7 +55,7 @@ describe('StateSnapshotProcessor', () => {
|
|
|
54
55
|
});
|
|
55
56
|
// Make deficit 0 so we don't fall through to the sync backstop and fail the test that way
|
|
56
57
|
// node-A is MISSING (user deleted it)
|
|
57
|
-
const nodeB = createDummyNode('ep1',
|
|
58
|
+
const nodeB = createDummyNode('ep1', NodeType.AGENT_THOUGHT, 60, {}, 'node-B');
|
|
58
59
|
const targets = [nodeB];
|
|
59
60
|
const messages = [
|
|
60
61
|
{
|
|
@@ -77,15 +78,54 @@ describe('StateSnapshotProcessor', () => {
|
|
|
77
78
|
it('should fall back to sync backstop if inbox is empty', async () => {
|
|
78
79
|
const env = createMockEnvironment();
|
|
79
80
|
const processor = createStateSnapshotProcessor('StateSnapshotProcessor', env, { target: 'max' }); // Summarize all
|
|
80
|
-
const nodeA = createDummyNode('ep1',
|
|
81
|
-
const nodeB = createDummyNode('ep1',
|
|
82
|
-
const nodeC = createDummyNode('ep2',
|
|
81
|
+
const nodeA = createDummyNode('ep1', NodeType.USER_PROMPT, 50, {}, 'node-A');
|
|
82
|
+
const nodeB = createDummyNode('ep1', NodeType.AGENT_THOUGHT, 60, {}, 'node-B');
|
|
83
|
+
const nodeC = createDummyNode('ep2', NodeType.USER_PROMPT, 50, {}, 'node-C');
|
|
83
84
|
const targets = [nodeA, nodeB, nodeC];
|
|
84
85
|
const result = await processor.process(createMockProcessArgs(targets));
|
|
85
86
|
// Should synthesize a new snapshot synchronously
|
|
86
|
-
expect(env.llmClient.
|
|
87
|
-
expect(result.length).toBe(
|
|
88
|
-
expect(result[
|
|
87
|
+
expect(env.llmClient.generateJson).toHaveBeenCalled();
|
|
88
|
+
expect(result.length).toBe(1); // nodeA is no longer protected, so everything is snapshotted
|
|
89
|
+
expect(result[0].type).toBe(NodeType.SNAPSHOT);
|
|
90
|
+
});
|
|
91
|
+
it('should use Global Lookback to find an existing snapshot in the graph as the baseline', async () => {
|
|
92
|
+
const env = createMockEnvironment();
|
|
93
|
+
const processor = createStateSnapshotProcessor('StateSnapshotProcessor', env, { target: 'incremental' });
|
|
94
|
+
// Create an old snapshot with existing JSON state
|
|
95
|
+
const oldStateJson = JSON.stringify({
|
|
96
|
+
discovered_facts: ['Global Lookback Works!'],
|
|
97
|
+
});
|
|
98
|
+
const oldSnapshot = createDummyNode('ep1', NodeType.SNAPSHOT, 10, { payload: { text: oldStateJson } }, 'old-snap');
|
|
99
|
+
const nodeA = createDummyNode('ep2', NodeType.USER_PROMPT, 50, {}, 'node-A');
|
|
100
|
+
// targets array contains the snapshot
|
|
101
|
+
const targets = [oldSnapshot, nodeA];
|
|
102
|
+
await processor.process(createMockProcessArgs(targets));
|
|
103
|
+
// The SnapshotGenerator should have been called with the oldStateJson as the baseline
|
|
104
|
+
expect(env.llmClient.generateJson).toHaveBeenCalledWith(expect.objectContaining({
|
|
105
|
+
contents: expect.arrayContaining([
|
|
106
|
+
expect.objectContaining({
|
|
107
|
+
parts: expect.arrayContaining([
|
|
108
|
+
expect.objectContaining({
|
|
109
|
+
text: expect.stringContaining('Global Lookback Works!'),
|
|
110
|
+
}),
|
|
111
|
+
]),
|
|
112
|
+
}),
|
|
113
|
+
]),
|
|
114
|
+
}));
|
|
115
|
+
});
|
|
116
|
+
it('should garbage collect the old baseline snapshot from the live graph when creating a new sync snapshot', async () => {
|
|
117
|
+
const env = createMockEnvironment();
|
|
118
|
+
const processor = createStateSnapshotProcessor('StateSnapshotProcessor', env, { target: 'incremental' });
|
|
119
|
+
const oldSnapshot = createDummyNode('ep1', NodeType.SNAPSHOT, 10, { payload: { text: '{}' } }, 'old-snap');
|
|
120
|
+
const nodeA = createDummyNode('ep2', NodeType.USER_PROMPT, 50, {}, 'node-A');
|
|
121
|
+
// The processor summarizes these 2 nodes
|
|
122
|
+
const result = await processor.process(createMockProcessArgs([oldSnapshot, nodeA]));
|
|
123
|
+
// It should have replaced BOTH the old snapshot and the new node with ONE new snapshot
|
|
124
|
+
expect(result.length).toBe(1);
|
|
125
|
+
expect(result[0].type).toBe(NodeType.SNAPSHOT);
|
|
126
|
+
expect(result[0].id).not.toBe('old-snap');
|
|
127
|
+
expect(result[0].abstractsIds).toContain('old-snap');
|
|
128
|
+
expect(result[0].abstractsIds).toContain('node-A');
|
|
89
129
|
});
|
|
90
130
|
});
|
|
91
131
|
//# sourceMappingURL=stateSnapshotProcessor.test.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stateSnapshotProcessor.test.js","sourceRoot":"","sources":["../../../../src/context/processors/stateSnapshotProcessor.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,4BAA4B,EAAE,MAAM,6BAA6B,CAAC;AAC3E,OAAO,EACL,qBAAqB,EACrB,eAAe,EACf,qBAAqB,GACtB,MAAM,gCAAgC,CAAC;
|
|
1
|
+
{"version":3,"file":"stateSnapshotProcessor.test.js","sourceRoot":"","sources":["../../../../src/context/processors/stateSnapshotProcessor.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,4BAA4B,EAAE,MAAM,6BAA6B,CAAC;AAC3E,OAAO,EACL,qBAAqB,EACrB,eAAe,EACf,qBAAqB,GACtB,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAG7C,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,MAAM,SAAS,GAAG,4BAA4B,CAC5C,wBAAwB,EACxB,GAAG,EACH;YACE,MAAM,EAAE,aAAa;SACtB,CACF,CAAC;QACF,MAAM,OAAO,GAAG,CAAC,eAAe,CAAC,KAAK,EAAE,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,kBAAkB;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,MAAM,SAAS,GAAG,4BAA4B,CAC5C,wBAAwB,EACxB,GAAG,EACH;YACE,MAAM,EAAE,aAAa;SACtB,CACF,CAAC;QAEF,MAAM,KAAK,GAAG,eAAe,CAC3B,KAAK,EACL,QAAQ,CAAC,WAAW,EACpB,EAAE,EACF,EAAE,EACF,QAAQ,CACT,CAAC;QACF,MAAM,KAAK,GAAG,eAAe,CAC3B,KAAK,EACL,QAAQ,CAAC,aAAa,EACtB,EAAE,EACF,EAAE,EACF,QAAQ,CACT,CAAC;QACF,MAAM,KAAK,GAAG,eAAe,CAC3B,KAAK,EACL,QAAQ,CAAC,WAAW,EACpB,EAAE,EACF,EAAE,EACF,QAAQ,CACT,CAAC;QAEF,MAAM,OAAO,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QAEtC,8DAA8D;QAC9D,MAAM,QAAQ,GAAG;YACf;gBACE,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE,mBAAmB;gBAC1B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,OAAO,EAAE;oBACP,WAAW,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC;oBACjC,OAAO,EAAE,sBAAsB;oBAC/B,IAAI,EAAE,eAAe;iBACtB;aACF;SACF,CAAC;QAEF,MAAM,WAAW,GAAG,qBAAqB,CAAC,OAAO,EAAE,EAAE,EAAE,QAAQ,CAAC,CAAC;QACjE,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAEpD,iDAAiD;QACjD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAEpC,6BAA6B;QAC7B,MAAM,CACH,WAAW,CAAC,KAA2B,CAAC,cAAc,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,CACvE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;QAC/F,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,MAAM,SAAS,GAAG,4BAA4B,CAC5C,wBAAwB,EACxB,GAAG,EACH;YACE,MAAM,EAAE,aAAa;SACtB,CACF,CAAC;QACF,0FAA0F;QAE1F,sCAAsC;QACtC,MAAM,KAAK,GAAG,eAAe,CAC3B,KAAK,EACL,QAAQ,CAAC,aAAa,EACtB,EAAE,EACF,EAAE,EACF,QAAQ,CACT,CAAC;QACF,MAAM,OAAO,GAAG,CAAC,KAAK,CAAC,CAAC;QAExB,MAAM,QAAQ,GAAG;YACf;gBACE,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE,mBAAmB;gBAC1B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,OAAO,EAAE;oBACP,WAAW,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC;oBACjC,OAAO,EAAE,sBAAsB;iBAChC;aACF;SACF,CAAC;QAEF,MAAM,WAAW,GAAG,qBAAqB,CAAC,OAAO,EAAE,EAAE,EAAE,QAAQ,CAAC,CAAC;QACjE,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAEpD,sEAAsE;QACtE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,CACH,WAAW,CAAC,KAA2B,CAAC,cAAc,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,CACvE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,MAAM,SAAS,GAAG,4BAA4B,CAC5C,wBAAwB,EACxB,GAAG,EACH,EAAE,MAAM,EAAE,KAAK,EAAE,CAClB,CAAC,CAAC,gBAAgB;QAEnB,MAAM,KAAK,GAAG,eAAe,CAC3B,KAAK,EACL,QAAQ,CAAC,WAAW,EACpB,EAAE,EACF,EAAE,EACF,QAAQ,CACT,CAAC;QACF,MAAM,KAAK,GAAG,eAAe,CAC3B,KAAK,EACL,QAAQ,CAAC,aAAa,EACtB,EAAE,EACF,EAAE,EACF,QAAQ,CACT,CAAC;QACF,MAAM,KAAK,GAAG,eAAe,CAC3B,KAAK,EACL,QAAQ,CAAC,WAAW,EACpB,EAAE,EACF,EAAE,EACF,QAAQ,CACT,CAAC;QACF,MAAM,OAAO,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC;QAEvE,iDAAiD;QACjD,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,6DAA6D;QAC5F,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sFAAsF,EAAE,KAAK,IAAI,EAAE;QACpG,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,MAAM,SAAS,GAAG,4BAA4B,CAC5C,wBAAwB,EACxB,GAAG,EACH,EAAE,MAAM,EAAE,aAAa,EAAE,CAC1B,CAAC;QAEF,kDAAkD;QAClD,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC;YAClC,gBAAgB,EAAE,CAAC,wBAAwB,CAAC;SAC7C,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,eAAe,CACjC,KAAK,EACL,QAAQ,CAAC,QAAQ,EACjB,EAAE,EACF,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EACnC,UAAU,CACX,CAAC;QACF,MAAM,KAAK,GAAG,eAAe,CAC3B,KAAK,EACL,QAAQ,CAAC,WAAW,EACpB,EAAE,EACF,EAAE,EACF,QAAQ,CACT,CAAC;QAEF,sCAAsC;QACtC,MAAM,OAAO,GAAG,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;QAErC,MAAM,SAAS,CAAC,OAAO,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC;QAExD,sFAAsF;QACtF,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACrD,MAAM,CAAC,gBAAgB,CAAC;YACtB,QAAQ,EAAE,MAAM,CAAC,eAAe,CAAC;gBAC/B,MAAM,CAAC,gBAAgB,CAAC;oBACtB,KAAK,EAAE,MAAM,CAAC,eAAe,CAAC;wBAC5B,MAAM,CAAC,gBAAgB,CAAC;4BACtB,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,wBAAwB,CAAC;yBACxD,CAAC;qBACH,CAAC;iBACH,CAAC;aACH,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wGAAwG,EAAE,KAAK,IAAI,EAAE;QACtH,MAAM,GAAG,GAAG,qBAAqB,EAAE,CAAC;QACpC,MAAM,SAAS,GAAG,4BAA4B,CAC5C,wBAAwB,EACxB,GAAG,EACH,EAAE,MAAM,EAAE,aAAa,EAAE,CAC1B,CAAC;QAEF,MAAM,WAAW,GAAG,eAAe,CACjC,KAAK,EACL,QAAQ,CAAC,QAAQ,EACjB,EAAE,EACF,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,EAC3B,UAAU,CACX,CAAC;QACF,MAAM,KAAK,GAAG,eAAe,CAC3B,KAAK,EACL,QAAQ,CAAC,WAAW,EACpB,EAAE,EACF,EAAE,EACF,QAAQ,CACT,CAAC;QAEF,yCAAyC;QACzC,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CACpC,qBAAqB,CAAC,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC,CAC5C,CAAC;QAEF,uFAAuF;QACvF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|