@roj-ai/sdk 0.1.3 → 0.1.5
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/bootstrap.js +191 -0
- package/dist/bootstrap.js.map +1 -0
- package/dist/builtin-events.js +8 -0
- package/dist/builtin-events.js.map +1 -0
- package/dist/bun-platform/fs.js +39 -0
- package/dist/bun-platform/fs.js.map +1 -0
- package/dist/bun-platform/index.js +18 -0
- package/dist/bun-platform/index.js.map +1 -0
- package/dist/bun-platform/process.js +21 -0
- package/dist/bun-platform/process.js.map +1 -0
- package/dist/config.js +54 -0
- package/dist/config.js.map +1 -0
- package/dist/config.test.js +155 -0
- package/dist/config.test.js.map +1 -0
- package/dist/core/agent-loop.integration.test.js +414 -0
- package/dist/core/agent-loop.integration.test.js.map +1 -0
- package/dist/core/agents/agent-config.test.js +194 -0
- package/dist/core/agents/agent-config.test.js.map +1 -0
- package/dist/core/agents/agent-roles.js +25 -0
- package/dist/core/agents/agent-roles.js.map +1 -0
- package/dist/core/agents/agent-shutdown.test.js +180 -0
- package/dist/core/agents/agent-shutdown.test.js.map +1 -0
- package/dist/core/agents/agent.js +1205 -0
- package/dist/core/agents/agent.js.map +1 -0
- package/dist/core/agents/agent.test.js +313 -0
- package/dist/core/agents/agent.test.js.map +1 -0
- package/dist/core/agents/communicator.js +13 -0
- package/dist/core/agents/communicator.js.map +1 -0
- package/dist/core/agents/config.js +5 -0
- package/dist/core/agents/config.js.map +1 -0
- package/dist/core/agents/context.js +2 -0
- package/dist/core/agents/context.js.map +1 -0
- package/dist/core/agents/debounce.js +74 -0
- package/dist/core/agents/debounce.js.map +1 -0
- package/dist/core/agents/handler-events.test.js +115 -0
- package/dist/core/agents/handler-events.test.js.map +1 -0
- package/dist/core/agents/index.js +2 -0
- package/dist/core/agents/index.js.map +1 -0
- package/dist/core/agents/response-sanitizer.js +46 -0
- package/dist/core/agents/response-sanitizer.js.map +1 -0
- package/dist/core/agents/response-sanitizer.test.js +101 -0
- package/dist/core/agents/response-sanitizer.test.js.map +1 -0
- package/dist/core/agents/retry.js +105 -0
- package/dist/core/agents/retry.js.map +1 -0
- package/dist/core/agents/schema.js +39 -0
- package/dist/core/agents/schema.js.map +1 -0
- package/dist/core/agents/state.js +90 -0
- package/dist/core/agents/state.js.map +1 -0
- package/dist/core/context/state.js +23 -0
- package/dist/core/context/state.js.map +1 -0
- package/dist/core/errors.js +38 -0
- package/dist/core/errors.js.map +1 -0
- package/dist/core/event-sourcing.integration.test.js +154 -0
- package/dist/core/event-sourcing.integration.test.js.map +1 -0
- package/dist/core/events/base-event-store.js +201 -0
- package/dist/core/events/base-event-store.js.map +1 -0
- package/dist/core/events/event-store.js +26 -0
- package/dist/core/events/event-store.js.map +1 -0
- package/dist/core/events/file.js +320 -0
- package/dist/core/events/file.js.map +1 -0
- package/dist/core/events/file.test.js +284 -0
- package/dist/core/events/file.test.js.map +1 -0
- package/dist/core/events/index.js +3 -0
- package/dist/core/events/index.js.map +1 -0
- package/dist/core/events/memory.js +101 -0
- package/dist/core/events/memory.js.map +1 -0
- package/dist/core/events/memory.test.js +502 -0
- package/dist/core/events/memory.test.js.map +1 -0
- package/dist/core/events/metadata-utils.js +107 -0
- package/dist/core/events/metadata-utils.js.map +1 -0
- package/dist/core/events/test-helpers.js +15 -0
- package/dist/core/events/test-helpers.js.map +1 -0
- package/dist/core/events/types.js +21 -0
- package/dist/core/events/types.js.map +1 -0
- package/dist/core/file-store/file-store.js +250 -0
- package/dist/core/file-store/file-store.js.map +1 -0
- package/dist/core/file-store/types.js +7 -0
- package/dist/core/file-store/types.js.map +1 -0
- package/dist/core/image/image-processor.js +106 -0
- package/dist/core/image/image-processor.js.map +1 -0
- package/dist/core/image/image-processor.test.js +171 -0
- package/dist/core/image/image-processor.test.js.map +1 -0
- package/dist/core/image/index.js +4 -0
- package/dist/core/image/index.js.map +1 -0
- package/dist/core/image/noop-resizer.js +6 -0
- package/dist/core/image/noop-resizer.js.map +1 -0
- package/dist/core/image/types.js +2 -0
- package/dist/core/image/types.js.map +1 -0
- package/dist/core/image/vips-resizer.js +100 -0
- package/dist/core/image/vips-resizer.js.map +1 -0
- package/dist/core/image/vips-resizer.test.js +324 -0
- package/dist/core/image/vips-resizer.test.js.map +1 -0
- package/dist/core/llm/anthropic.js +396 -0
- package/dist/core/llm/anthropic.js.map +1 -0
- package/dist/core/llm/anthropic.test.js +434 -0
- package/dist/core/llm/anthropic.test.js.map +1 -0
- package/dist/core/llm/cache-breakpoints.js +37 -0
- package/dist/core/llm/cache-breakpoints.js.map +1 -0
- package/dist/core/llm/cache-live.test.js +137 -0
- package/dist/core/llm/cache-live.test.js.map +1 -0
- package/dist/core/llm/index.js +9 -0
- package/dist/core/llm/index.js.map +1 -0
- package/dist/core/llm/llm-log-types.js +12 -0
- package/dist/core/llm/llm-log-types.js.map +1 -0
- package/dist/core/llm/logger.js +241 -0
- package/dist/core/llm/logger.js.map +1 -0
- package/dist/core/llm/logger.test.js +228 -0
- package/dist/core/llm/logger.test.js.map +1 -0
- package/dist/core/llm/logging-provider.js +49 -0
- package/dist/core/llm/logging-provider.js.map +1 -0
- package/dist/core/llm/middleware.js +114 -0
- package/dist/core/llm/middleware.js.map +1 -0
- package/dist/core/llm/mock.js +186 -0
- package/dist/core/llm/mock.js.map +1 -0
- package/dist/core/llm/mock.test.js +318 -0
- package/dist/core/llm/mock.test.js.map +1 -0
- package/dist/core/llm/openrouter-mapping.test.js +125 -0
- package/dist/core/llm/openrouter-mapping.test.js.map +1 -0
- package/dist/core/llm/openrouter.js +298 -0
- package/dist/core/llm/openrouter.js.map +1 -0
- package/dist/core/llm/openrouter.test.js +377 -0
- package/dist/core/llm/openrouter.test.js.map +1 -0
- package/dist/core/llm/provider-integration.test.js +350 -0
- package/dist/core/llm/provider-integration.test.js.map +1 -0
- package/dist/core/llm/provider.js +18 -0
- package/dist/core/llm/provider.js.map +1 -0
- package/dist/core/llm/routing-provider.js +52 -0
- package/dist/core/llm/routing-provider.js.map +1 -0
- package/dist/core/llm/routing-provider.test.js +94 -0
- package/dist/core/llm/routing-provider.test.js.map +1 -0
- package/dist/core/llm/schema.js +31 -0
- package/dist/core/llm/schema.js.map +1 -0
- package/dist/core/llm/snapshot-fetch.js +122 -0
- package/dist/core/llm/snapshot-fetch.js.map +1 -0
- package/dist/core/llm/snapshot-middleware.js +142 -0
- package/dist/core/llm/snapshot-middleware.js.map +1 -0
- package/dist/core/llm/snapshot-middleware.test.js +144 -0
- package/dist/core/llm/snapshot-middleware.test.js.map +1 -0
- package/dist/core/llm/state.js +48 -0
- package/dist/core/llm/state.js.map +1 -0
- package/dist/core/llm/tokens.js +40 -0
- package/dist/core/llm/tokens.js.map +1 -0
- package/dist/core/multi-agent.integration.test.js +298 -0
- package/dist/core/multi-agent.integration.test.js.map +1 -0
- package/dist/core/plugin-hooks.integration.test.js +344 -0
- package/dist/core/plugin-hooks.integration.test.js.map +1 -0
- package/dist/core/plugins/hook-types.js +5 -0
- package/dist/core/plugins/hook-types.js.map +1 -0
- package/dist/core/plugins/index.js +5 -0
- package/dist/core/plugins/index.js.map +1 -0
- package/dist/core/plugins/plugin-builder.js +321 -0
- package/dist/core/plugins/plugin-builder.js.map +1 -0
- package/dist/core/preset/config.js +54 -0
- package/dist/core/preset/config.js.map +1 -0
- package/dist/core/preset/index.js +6 -0
- package/dist/core/preset/index.js.map +1 -0
- package/dist/core/preset/preset-builder.js +63 -0
- package/dist/core/preset/preset-builder.js.map +1 -0
- package/dist/core/session-lifecycle.integration.test.js +159 -0
- package/dist/core/session-lifecycle.integration.test.js.map +1 -0
- package/dist/core/sessions/apply-event.js +41 -0
- package/dist/core/sessions/apply-event.js.map +1 -0
- package/dist/core/sessions/context.js +2 -0
- package/dist/core/sessions/context.js.map +1 -0
- package/dist/core/sessions/fork-utils.js +42 -0
- package/dist/core/sessions/fork-utils.js.map +1 -0
- package/dist/core/sessions/fork-utils.test.js +129 -0
- package/dist/core/sessions/fork-utils.test.js.map +1 -0
- package/dist/core/sessions/reducer.js +55 -0
- package/dist/core/sessions/reducer.js.map +1 -0
- package/dist/core/sessions/schema.js +66 -0
- package/dist/core/sessions/schema.js.map +1 -0
- package/dist/core/sessions/session-environment.js +2 -0
- package/dist/core/sessions/session-environment.js.map +1 -0
- package/dist/core/sessions/session-manager.js +650 -0
- package/dist/core/sessions/session-manager.js.map +1 -0
- package/dist/core/sessions/session-store.js +118 -0
- package/dist/core/sessions/session-store.js.map +1 -0
- package/dist/core/sessions/session.js +675 -0
- package/dist/core/sessions/session.js.map +1 -0
- package/dist/core/sessions/session.test.js +1095 -0
- package/dist/core/sessions/session.test.js.map +1 -0
- package/dist/core/sessions/state.js +377 -0
- package/dist/core/sessions/state.js.map +1 -0
- package/dist/core/system.js +66 -0
- package/dist/core/system.js.map +1 -0
- package/dist/core/tools/context.js +2 -0
- package/dist/core/tools/context.js.map +1 -0
- package/dist/core/tools/definition.js +4 -0
- package/dist/core/tools/definition.js.map +1 -0
- package/dist/core/tools/executor.js +82 -0
- package/dist/core/tools/executor.js.map +1 -0
- package/dist/core/tools/executor.test.js +143 -0
- package/dist/core/tools/executor.test.js.map +1 -0
- package/dist/core/tools/index.js +4 -0
- package/dist/core/tools/index.js.map +1 -0
- package/dist/core/tools/schema.js +20 -0
- package/dist/core/tools/schema.js.map +1 -0
- package/dist/core/tools/state.js +29 -0
- package/dist/core/tools/state.js.map +1 -0
- package/dist/index.js +70 -0
- package/dist/index.js.map +1 -0
- package/dist/lib/json/index.js +5 -0
- package/dist/lib/json/index.js.map +1 -0
- package/dist/lib/logger/console.js +147 -0
- package/dist/lib/logger/console.js.map +1 -0
- package/dist/lib/logger/console.test.js +258 -0
- package/dist/lib/logger/console.test.js.map +1 -0
- package/dist/lib/logger/file.js +54 -0
- package/dist/lib/logger/file.js.map +1 -0
- package/dist/lib/logger/index.js +4 -0
- package/dist/lib/logger/index.js.map +1 -0
- package/dist/lib/logger/logger.js +28 -0
- package/dist/lib/logger/logger.js.map +1 -0
- package/dist/lib/logger/ring-buffer.js +61 -0
- package/dist/lib/logger/ring-buffer.js.map +1 -0
- package/dist/lib/logger/tee.js +43 -0
- package/dist/lib/logger/tee.js.map +1 -0
- package/dist/lib/mime.js +22 -0
- package/dist/lib/mime.js.map +1 -0
- package/dist/lib/never.js +4 -0
- package/dist/lib/never.js.map +1 -0
- package/dist/lib/utils/hash.js +35 -0
- package/dist/lib/utils/hash.js.map +1 -0
- package/dist/lib/utils/result.js +21 -0
- package/dist/lib/utils/result.js.map +1 -0
- package/dist/platform/fs.js +8 -0
- package/dist/platform/fs.js.map +1 -0
- package/dist/platform/index.js +9 -0
- package/dist/platform/index.js.map +1 -0
- package/dist/platform/process.js +8 -0
- package/dist/platform/process.js.map +1 -0
- package/dist/plugins/agent-status/plugin.js +77 -0
- package/dist/plugins/agent-status/plugin.js.map +1 -0
- package/dist/plugins/agents/agents.integration.test.js +683 -0
- package/dist/plugins/agents/agents.integration.test.js.map +1 -0
- package/dist/plugins/agents/index.js +2 -0
- package/dist/plugins/agents/index.js.map +1 -0
- package/dist/plugins/agents/plugin.js +199 -0
- package/dist/plugins/agents/plugin.js.map +1 -0
- package/dist/plugins/context-compact/context-compact.integration.test.js +174 -0
- package/dist/plugins/context-compact/context-compact.integration.test.js.map +1 -0
- package/dist/plugins/context-compact/context-compactor.js +238 -0
- package/dist/plugins/context-compact/context-compactor.js.map +1 -0
- package/dist/plugins/context-compact/context-compactor.test.js +763 -0
- package/dist/plugins/context-compact/context-compactor.test.js.map +1 -0
- package/dist/plugins/context-compact/history-offloader.js +42 -0
- package/dist/plugins/context-compact/history-offloader.js.map +1 -0
- package/dist/plugins/context-compact/history-offloader.test.js +77 -0
- package/dist/plugins/context-compact/history-offloader.test.js.map +1 -0
- package/dist/plugins/context-compact/index.js +4 -0
- package/dist/plugins/context-compact/index.js.map +1 -0
- package/dist/plugins/context-compact/plugin.js +37 -0
- package/dist/plugins/context-compact/plugin.js.map +1 -0
- package/dist/plugins/filesystem/filesystem.integration.test.js +411 -0
- package/dist/plugins/filesystem/filesystem.integration.test.js.map +1 -0
- package/dist/plugins/filesystem/helpers.js +170 -0
- package/dist/plugins/filesystem/helpers.js.map +1 -0
- package/dist/plugins/filesystem/index.js +3 -0
- package/dist/plugins/filesystem/index.js.map +1 -0
- package/dist/plugins/filesystem/listing.js +247 -0
- package/dist/plugins/filesystem/listing.js.map +1 -0
- package/dist/plugins/filesystem/plugin.js +364 -0
- package/dist/plugins/filesystem/plugin.js.map +1 -0
- package/dist/plugins/filesystem/schema.js +2 -0
- package/dist/plugins/filesystem/schema.js.map +1 -0
- package/dist/plugins/git-status/index.js +2 -0
- package/dist/plugins/git-status/index.js.map +1 -0
- package/dist/plugins/git-status/plugin.js +144 -0
- package/dist/plugins/git-status/plugin.js.map +1 -0
- package/dist/plugins/limits-guard/config.js +5 -0
- package/dist/plugins/limits-guard/config.js.map +1 -0
- package/dist/plugins/limits-guard/index.js +3 -0
- package/dist/plugins/limits-guard/index.js.map +1 -0
- package/dist/plugins/limits-guard/limit-guard.js +125 -0
- package/dist/plugins/limits-guard/limit-guard.js.map +1 -0
- package/dist/plugins/limits-guard/limit-guard.test.js +121 -0
- package/dist/plugins/limits-guard/limit-guard.test.js.map +1 -0
- package/dist/plugins/limits-guard/limits-guard.integration.test.js +378 -0
- package/dist/plugins/limits-guard/limits-guard.integration.test.js.map +1 -0
- package/dist/plugins/limits-guard/plugin.js +240 -0
- package/dist/plugins/limits-guard/plugin.js.map +1 -0
- package/dist/plugins/llm-debug/index.js +2 -0
- package/dist/plugins/llm-debug/index.js.map +1 -0
- package/dist/plugins/llm-debug/llm-debug.integration.test.js +157 -0
- package/dist/plugins/llm-debug/llm-debug.integration.test.js.map +1 -0
- package/dist/plugins/llm-debug/plugin.js +148 -0
- package/dist/plugins/llm-debug/plugin.js.map +1 -0
- package/dist/plugins/logs/index.js +2 -0
- package/dist/plugins/logs/index.js.map +1 -0
- package/dist/plugins/logs/plugin.js +38 -0
- package/dist/plugins/logs/plugin.js.map +1 -0
- package/dist/plugins/mailbox/helpers.js +66 -0
- package/dist/plugins/mailbox/helpers.js.map +1 -0
- package/dist/plugins/mailbox/index.js +9 -0
- package/dist/plugins/mailbox/index.js.map +1 -0
- package/dist/plugins/mailbox/mailbox.integration.test.js +605 -0
- package/dist/plugins/mailbox/mailbox.integration.test.js.map +1 -0
- package/dist/plugins/mailbox/plugin.js +204 -0
- package/dist/plugins/mailbox/plugin.js.map +1 -0
- package/dist/plugins/mailbox/prompts.js +93 -0
- package/dist/plugins/mailbox/prompts.js.map +1 -0
- package/dist/plugins/mailbox/query.js +38 -0
- package/dist/plugins/mailbox/query.js.map +1 -0
- package/dist/plugins/mailbox/schema.js +32 -0
- package/dist/plugins/mailbox/schema.js.map +1 -0
- package/dist/plugins/mailbox/state.js +41 -0
- package/dist/plugins/mailbox/state.js.map +1 -0
- package/dist/plugins/resources/index.js +4 -0
- package/dist/plugins/resources/index.js.map +1 -0
- package/dist/plugins/resources/manifest.js +20 -0
- package/dist/plugins/resources/manifest.js.map +1 -0
- package/dist/plugins/resources/plugin.js +171 -0
- package/dist/plugins/resources/plugin.js.map +1 -0
- package/dist/plugins/resources/post-inject.js +32 -0
- package/dist/plugins/resources/post-inject.js.map +1 -0
- package/dist/plugins/resources/state.js +16 -0
- package/dist/plugins/resources/state.js.map +1 -0
- package/dist/plugins/result-eviction/index.js +2 -0
- package/dist/plugins/result-eviction/index.js.map +1 -0
- package/dist/plugins/result-eviction/plugin.js +43 -0
- package/dist/plugins/result-eviction/plugin.js.map +1 -0
- package/dist/plugins/result-eviction/result-eviction.integration.test.js +217 -0
- package/dist/plugins/result-eviction/result-eviction.integration.test.js.map +1 -0
- package/dist/plugins/services/plugin.js +453 -0
- package/dist/plugins/services/plugin.js.map +1 -0
- package/dist/plugins/services/port-pool.js +70 -0
- package/dist/plugins/services/port-pool.js.map +1 -0
- package/dist/plugins/services/prompt.js +40 -0
- package/dist/plugins/services/prompt.js.map +1 -0
- package/dist/plugins/services/schema.js +9 -0
- package/dist/plugins/services/schema.js.map +1 -0
- package/dist/plugins/services/service.js +470 -0
- package/dist/plugins/services/service.js.map +1 -0
- package/dist/plugins/services/services.integration.test.js +485 -0
- package/dist/plugins/services/services.integration.test.js.map +1 -0
- package/dist/plugins/session-lifecycle/index.js +2 -0
- package/dist/plugins/session-lifecycle/index.js.map +1 -0
- package/dist/plugins/session-lifecycle/plugin.js +273 -0
- package/dist/plugins/session-lifecycle/plugin.js.map +1 -0
- package/dist/plugins/session-lifecycle/session-lifecycle.integration.test.js +498 -0
- package/dist/plugins/session-lifecycle/session-lifecycle.integration.test.js.map +1 -0
- package/dist/plugins/session-state/plugin.js +159 -0
- package/dist/plugins/session-state/plugin.js.map +1 -0
- package/dist/plugins/session-stats/index.js +3 -0
- package/dist/plugins/session-stats/index.js.map +1 -0
- package/dist/plugins/session-stats/plugin.js +81 -0
- package/dist/plugins/session-stats/plugin.js.map +1 -0
- package/dist/plugins/shell/executor.js +339 -0
- package/dist/plugins/shell/executor.js.map +1 -0
- package/dist/plugins/shell/index.js +6 -0
- package/dist/plugins/shell/index.js.map +1 -0
- package/dist/plugins/shell/plugin.js +66 -0
- package/dist/plugins/shell/plugin.js.map +1 -0
- package/dist/plugins/shell/shell.integration.test.js +234 -0
- package/dist/plugins/shell/shell.integration.test.js.map +1 -0
- package/dist/plugins/shell/shell.test.js +236 -0
- package/dist/plugins/shell/shell.test.js.map +1 -0
- package/dist/plugins/skills/discovery.js +205 -0
- package/dist/plugins/skills/discovery.js.map +1 -0
- package/dist/plugins/skills/discovery.test.js +312 -0
- package/dist/plugins/skills/discovery.test.js.map +1 -0
- package/dist/plugins/skills/index.js +12 -0
- package/dist/plugins/skills/index.js.map +1 -0
- package/dist/plugins/skills/plugin.js +293 -0
- package/dist/plugins/skills/plugin.js.map +1 -0
- package/dist/plugins/skills/prompts.js +70 -0
- package/dist/plugins/skills/prompts.js.map +1 -0
- package/dist/plugins/skills/schema.js +18 -0
- package/dist/plugins/skills/schema.js.map +1 -0
- package/dist/plugins/skills/skills.integration.test.js +475 -0
- package/dist/plugins/skills/skills.integration.test.js.map +1 -0
- package/dist/plugins/snapshotting/index.js +3 -0
- package/dist/plugins/snapshotting/index.js.map +1 -0
- package/dist/plugins/snapshotting/jj-snapshotter.js +106 -0
- package/dist/plugins/snapshotting/jj-snapshotter.js.map +1 -0
- package/dist/plugins/snapshotting/plugin.js +28 -0
- package/dist/plugins/snapshotting/plugin.js.map +1 -0
- package/dist/plugins/snapshotting/snapshotter.js +2 -0
- package/dist/plugins/snapshotting/snapshotter.js.map +1 -0
- package/dist/plugins/todo/index.js +7 -0
- package/dist/plugins/todo/index.js.map +1 -0
- package/dist/plugins/todo/plugin.js +319 -0
- package/dist/plugins/todo/plugin.js.map +1 -0
- package/dist/plugins/todo/prompts.js +54 -0
- package/dist/plugins/todo/prompts.js.map +1 -0
- package/dist/plugins/todo/schema.js +18 -0
- package/dist/plugins/todo/schema.js.map +1 -0
- package/dist/plugins/todo/todo.integration.test.js +605 -0
- package/dist/plugins/todo/todo.integration.test.js.map +1 -0
- package/dist/plugins/uploads/index.js +8 -0
- package/dist/plugins/uploads/index.js.map +1 -0
- package/dist/plugins/uploads/plugin.js +346 -0
- package/dist/plugins/uploads/plugin.js.map +1 -0
- package/dist/plugins/uploads/preprocessor.js +44 -0
- package/dist/plugins/uploads/preprocessor.js.map +1 -0
- package/dist/plugins/uploads/preprocessors/image-classifier.js +127 -0
- package/dist/plugins/uploads/preprocessors/image-classifier.js.map +1 -0
- package/dist/plugins/uploads/preprocessors/index.js +7 -0
- package/dist/plugins/uploads/preprocessors/index.js.map +1 -0
- package/dist/plugins/uploads/preprocessors/markitdown-preprocessor.js +204 -0
- package/dist/plugins/uploads/preprocessors/markitdown-preprocessor.js.map +1 -0
- package/dist/plugins/uploads/preprocessors/zip-preprocessor.js +172 -0
- package/dist/plugins/uploads/preprocessors/zip-preprocessor.js.map +1 -0
- package/dist/plugins/uploads/schema.js +20 -0
- package/dist/plugins/uploads/schema.js.map +1 -0
- package/dist/plugins/uploads/state.js +22 -0
- package/dist/plugins/uploads/state.js.map +1 -0
- package/dist/plugins/uploads/uploads.integration.test.js +496 -0
- package/dist/plugins/uploads/uploads.integration.test.js.map +1 -0
- package/dist/plugins/user-chat/index.js +5 -0
- package/dist/plugins/user-chat/index.js.map +1 -0
- package/dist/plugins/user-chat/plugin.js +544 -0
- package/dist/plugins/user-chat/plugin.js.map +1 -0
- package/dist/plugins/user-chat/prompts.js +29 -0
- package/dist/plugins/user-chat/prompts.js.map +1 -0
- package/dist/plugins/user-chat/schema.js +46 -0
- package/dist/plugins/user-chat/schema.js.map +1 -0
- package/dist/plugins/user-chat/user-chat.integration.test.js +668 -0
- package/dist/plugins/user-chat/user-chat.integration.test.js.map +1 -0
- package/dist/plugins/workers/context.js +143 -0
- package/dist/plugins/workers/context.js.map +1 -0
- package/dist/plugins/workers/definition.js +30 -0
- package/dist/plugins/workers/definition.js.map +1 -0
- package/dist/plugins/workers/index.js +7 -0
- package/dist/plugins/workers/index.js.map +1 -0
- package/dist/plugins/workers/plugin.js +578 -0
- package/dist/plugins/workers/plugin.js.map +1 -0
- package/dist/plugins/workers/worker.js +18 -0
- package/dist/plugins/workers/worker.js.map +1 -0
- package/dist/plugins/workers/workers.integration.test.js +629 -0
- package/dist/plugins/workers/workers.integration.test.js.map +1 -0
- package/dist/prompts/base.js +239 -0
- package/dist/prompts/base.js.map +1 -0
- package/dist/prompts/builder.js +131 -0
- package/dist/prompts/builder.js.map +1 -0
- package/dist/prompts/index.js +20 -0
- package/dist/prompts/index.js.map +1 -0
- package/dist/prompts/macros.js +26 -0
- package/dist/prompts/macros.js.map +1 -0
- package/dist/prompts/macros.test.js +80 -0
- package/dist/prompts/macros.test.js.map +1 -0
- package/dist/testing/bootstrap-for-testing.js +28 -0
- package/dist/testing/bootstrap-for-testing.js.map +1 -0
- package/dist/testing/index.js +7 -0
- package/dist/testing/index.js.map +1 -0
- package/dist/testing/node-platform.js +65 -0
- package/dist/testing/node-platform.js.map +1 -0
- package/dist/testing/notification-collector.js +82 -0
- package/dist/testing/notification-collector.js.map +1 -0
- package/dist/testing/preset-helpers.js +37 -0
- package/dist/testing/preset-helpers.js.map +1 -0
- package/dist/testing/test-harness.js +226 -0
- package/dist/testing/test-harness.js.map +1 -0
- package/dist/testing/test-harness.test.js +51 -0
- package/dist/testing/test-harness.test.js.map +1 -0
- package/dist/testing/wait-helpers.js +64 -0
- package/dist/testing/wait-helpers.js.map +1 -0
- package/dist/transport/adapter/client-adapter.js +64 -0
- package/dist/transport/adapter/client-adapter.js.map +1 -0
- package/dist/transport/adapter/index.js +24 -0
- package/dist/transport/adapter/index.js.map +1 -0
- package/dist/transport/adapter/server-adapter.js +73 -0
- package/dist/transport/adapter/server-adapter.js.map +1 -0
- package/dist/transport/adapter/types.js +8 -0
- package/dist/transport/adapter/types.js.map +1 -0
- package/dist/transport/http/app.js +86 -0
- package/dist/transport/http/app.js.map +1 -0
- package/dist/transport/http/index.js +6 -0
- package/dist/transport/http/index.js.map +1 -0
- package/dist/transport/http/middleware/bearer-auth.js +33 -0
- package/dist/transport/http/middleware/bearer-auth.js.map +1 -0
- package/dist/transport/http/middleware/error-handler.js +56 -0
- package/dist/transport/http/middleware/error-handler.js.map +1 -0
- package/dist/transport/http/routes/files.js +237 -0
- package/dist/transport/http/routes/files.js.map +1 -0
- package/dist/transport/http/routes/resources.js +77 -0
- package/dist/transport/http/routes/resources.js.map +1 -0
- package/dist/transport/http/routes/rpc.integration.test.js +189 -0
- package/dist/transport/http/routes/rpc.integration.test.js.map +1 -0
- package/dist/transport/http/routes/rpc.js +110 -0
- package/dist/transport/http/routes/rpc.js.map +1 -0
- package/dist/transport/http/routes/rpc.test.js +316 -0
- package/dist/transport/http/routes/rpc.test.js.map +1 -0
- package/dist/transport/http/routes/upload.js +205 -0
- package/dist/transport/http/routes/upload.js.map +1 -0
- package/dist/transport/rpc/index.js +7 -0
- package/dist/transport/rpc/index.js.map +1 -0
- package/dist/transport/rpc/methods.js +8 -0
- package/dist/transport/rpc/methods.js.map +1 -0
- package/dist/user-config.js +14 -0
- package/dist/user-config.js.map +1 -0
- package/package.json +47 -57
|
@@ -0,0 +1,378 @@
|
|
|
1
|
+
import { describe, expect, it } from 'bun:test';
|
|
2
|
+
import { AgentId } from '../../core/agents/schema.js';
|
|
3
|
+
import { agentEvents } from '../../core/agents/state.js';
|
|
4
|
+
import { MockLLMProvider } from '../../core/llm/mock.js';
|
|
5
|
+
import { selectPluginState } from '../../core/sessions/reducer.js';
|
|
6
|
+
import { ToolCallId } from '../../core/tools/schema.js';
|
|
7
|
+
import { createMultiAgentPreset, createTestPreset, TestHarness } from '../../testing/index.js';
|
|
8
|
+
import { limitsGuardPlugin } from './plugin.js';
|
|
9
|
+
function createLimitsHarness(options) {
|
|
10
|
+
return new TestHarness({ ...options, systemPlugins: [limitsGuardPlugin] });
|
|
11
|
+
}
|
|
12
|
+
/**
|
|
13
|
+
* Wait for a specific agent to reach 'paused' status (or timeout).
|
|
14
|
+
* Polls the TestSession's state directly (live getter).
|
|
15
|
+
*/
|
|
16
|
+
async function waitForAgentPaused(session, agentId, timeoutMs = 10000) {
|
|
17
|
+
const deadline = Date.now() + timeoutMs;
|
|
18
|
+
while (Date.now() < deadline) {
|
|
19
|
+
const agentState = session.state.agents.get(agentId);
|
|
20
|
+
if (agentState?.status === 'paused')
|
|
21
|
+
return;
|
|
22
|
+
await new Promise(r => setTimeout(r, 10));
|
|
23
|
+
}
|
|
24
|
+
throw new Error(`waitForAgentPaused timed out after ${timeoutMs}ms for agent ${agentId}`);
|
|
25
|
+
}
|
|
26
|
+
/**
|
|
27
|
+
* Wait for all agents to be either idle (pending with no work) or paused.
|
|
28
|
+
*/
|
|
29
|
+
async function waitForAllSettled(session, timeoutMs = 10000) {
|
|
30
|
+
const deadline = Date.now() + timeoutMs;
|
|
31
|
+
outer: while (Date.now() < deadline) {
|
|
32
|
+
for (const [, agentState] of session.state.agents) {
|
|
33
|
+
if (agentState.status === 'paused')
|
|
34
|
+
continue;
|
|
35
|
+
if (agentState.status === 'pending' && agentState.pendingToolCalls.length === 0 && agentState.pendingToolResults.length === 0) {
|
|
36
|
+
continue;
|
|
37
|
+
}
|
|
38
|
+
await new Promise(r => setTimeout(r, 10));
|
|
39
|
+
continue outer;
|
|
40
|
+
}
|
|
41
|
+
// Double-check after a brief delay
|
|
42
|
+
await new Promise(r => setTimeout(r, 10));
|
|
43
|
+
for (const [, agentState] of session.state.agents) {
|
|
44
|
+
if (agentState.status === 'paused')
|
|
45
|
+
continue;
|
|
46
|
+
if (agentState.status === 'pending' && agentState.pendingToolCalls.length === 0 && agentState.pendingToolResults.length === 0) {
|
|
47
|
+
continue;
|
|
48
|
+
}
|
|
49
|
+
continue outer;
|
|
50
|
+
}
|
|
51
|
+
return;
|
|
52
|
+
}
|
|
53
|
+
throw new Error(`waitForAllSettled timed out after ${timeoutMs}ms`);
|
|
54
|
+
}
|
|
55
|
+
describe('limits-guard plugin', () => {
|
|
56
|
+
// =========================================================================
|
|
57
|
+
// hard limits
|
|
58
|
+
// =========================================================================
|
|
59
|
+
describe('hard limits', () => {
|
|
60
|
+
it('agent exceeding inference hard limit → agent_paused event → agent stops', async () => {
|
|
61
|
+
let inferenceCount = 0;
|
|
62
|
+
const harness = createLimitsHarness({
|
|
63
|
+
presets: [createTestPreset({
|
|
64
|
+
orchestratorSystem: 'Test agent.',
|
|
65
|
+
orchestratorPlugins: [limitsGuardPlugin.configureAgent({ limits: { maxTurns: 3 } })],
|
|
66
|
+
})],
|
|
67
|
+
mockHandler: () => {
|
|
68
|
+
inferenceCount++;
|
|
69
|
+
return {
|
|
70
|
+
content: null,
|
|
71
|
+
toolCalls: [{ id: ToolCallId(`tc${inferenceCount}`), name: 'tell_user', input: { message: `Turn ${inferenceCount}` } }],
|
|
72
|
+
finishReason: 'stop',
|
|
73
|
+
metrics: MockLLMProvider.defaultMetrics(),
|
|
74
|
+
};
|
|
75
|
+
},
|
|
76
|
+
});
|
|
77
|
+
const session = await harness.createSession('test');
|
|
78
|
+
const entryAgentId = session.getEntryAgentId();
|
|
79
|
+
await session.sendMessage('Start');
|
|
80
|
+
await waitForAgentPaused(session, entryAgentId);
|
|
81
|
+
const entryAgent = session.state.agents.get(entryAgentId);
|
|
82
|
+
expect(entryAgent.status).toBe('paused');
|
|
83
|
+
const pausedEvents = await session.getEventsByType(agentEvents, 'agent_paused');
|
|
84
|
+
expect(pausedEvents.length).toBeGreaterThanOrEqual(1);
|
|
85
|
+
const pauseEvent = pausedEvents.find(e => e.agentId === entryAgentId);
|
|
86
|
+
expect(pauseEvent).toBeDefined();
|
|
87
|
+
expect(pauseEvent.reason).toBe('handler');
|
|
88
|
+
await harness.shutdown();
|
|
89
|
+
});
|
|
90
|
+
it('agent exceeding tool call hard limit → agent paused', async () => {
|
|
91
|
+
let callNum = 0;
|
|
92
|
+
const harness = createLimitsHarness({
|
|
93
|
+
presets: [createTestPreset({
|
|
94
|
+
orchestratorSystem: 'Test agent.',
|
|
95
|
+
orchestratorPlugins: [limitsGuardPlugin.configureAgent({ limits: { maxToolCalls: 3, maxTurns: 100 } })],
|
|
96
|
+
})],
|
|
97
|
+
mockHandler: () => {
|
|
98
|
+
callNum++;
|
|
99
|
+
return {
|
|
100
|
+
content: null,
|
|
101
|
+
toolCalls: [
|
|
102
|
+
{ id: ToolCallId(`tc${callNum}a`), name: 'tell_user', input: { message: `A${callNum}` } },
|
|
103
|
+
{ id: ToolCallId(`tc${callNum}b`), name: 'tell_user', input: { message: `B${callNum}` } },
|
|
104
|
+
],
|
|
105
|
+
finishReason: 'stop',
|
|
106
|
+
metrics: MockLLMProvider.defaultMetrics(),
|
|
107
|
+
};
|
|
108
|
+
},
|
|
109
|
+
});
|
|
110
|
+
const session = await harness.createSession('test');
|
|
111
|
+
const entryAgentId = session.getEntryAgentId();
|
|
112
|
+
await session.sendMessage('Start');
|
|
113
|
+
await waitForAgentPaused(session, entryAgentId);
|
|
114
|
+
const entryAgent = session.state.agents.get(entryAgentId);
|
|
115
|
+
expect(entryAgent.status).toBe('paused');
|
|
116
|
+
const counters = selectPluginState(session.state, 'agentLimits')?.get(entryAgentId);
|
|
117
|
+
expect(counters).toBeDefined();
|
|
118
|
+
expect(counters.toolCallCount).toBeGreaterThanOrEqual(3);
|
|
119
|
+
await harness.shutdown();
|
|
120
|
+
});
|
|
121
|
+
it('agent exceeding spawned agent limit → agent paused', async () => {
|
|
122
|
+
let orchestratorCalls = 0;
|
|
123
|
+
const harness = createLimitsHarness({
|
|
124
|
+
presets: [createMultiAgentPreset([
|
|
125
|
+
{ name: 'worker', system: 'Worker agent.', tools: [], agents: [] },
|
|
126
|
+
], { orchestratorSystem: 'Orchestrator agent.' })],
|
|
127
|
+
mockHandler: (request) => {
|
|
128
|
+
if (request.systemPrompt.includes('Orchestrator')) {
|
|
129
|
+
orchestratorCalls++;
|
|
130
|
+
return {
|
|
131
|
+
content: null,
|
|
132
|
+
toolCalls: [
|
|
133
|
+
{ id: ToolCallId(`tc${orchestratorCalls}a`), name: 'start_worker', input: { message: `Task ${orchestratorCalls}a` } },
|
|
134
|
+
{ id: ToolCallId(`tc${orchestratorCalls}b`), name: 'start_worker', input: { message: `Task ${orchestratorCalls}b` } },
|
|
135
|
+
],
|
|
136
|
+
finishReason: 'stop',
|
|
137
|
+
metrics: MockLLMProvider.defaultMetrics(),
|
|
138
|
+
};
|
|
139
|
+
}
|
|
140
|
+
return { content: 'Worker done', toolCalls: [], finishReason: 'stop', metrics: MockLLMProvider.defaultMetrics() };
|
|
141
|
+
},
|
|
142
|
+
});
|
|
143
|
+
const session = await harness.createSession('test');
|
|
144
|
+
const orchestratorId = session.getEntryAgentId();
|
|
145
|
+
await session.sendMessage('Start');
|
|
146
|
+
await waitForAgentPaused(session, orchestratorId, 15000);
|
|
147
|
+
const orch = session.state.agents.get(orchestratorId);
|
|
148
|
+
expect(orch.status).toBe('paused');
|
|
149
|
+
const counters = selectPluginState(session.state, 'agentLimits')?.get(orchestratorId);
|
|
150
|
+
expect(counters).toBeDefined();
|
|
151
|
+
expect(counters.spawnedAgentCount).toBeGreaterThanOrEqual(10);
|
|
152
|
+
await harness.shutdown();
|
|
153
|
+
});
|
|
154
|
+
});
|
|
155
|
+
// =========================================================================
|
|
156
|
+
// soft limits (status display)
|
|
157
|
+
// =========================================================================
|
|
158
|
+
describe('soft limits (status display)', () => {
|
|
159
|
+
it('approaching limit (80%) → status contains warning message', async () => {
|
|
160
|
+
let inferenceCount = 0;
|
|
161
|
+
let sawWarning = false;
|
|
162
|
+
const harness = createLimitsHarness({
|
|
163
|
+
presets: [createTestPreset({
|
|
164
|
+
orchestratorSystem: 'Test agent.',
|
|
165
|
+
orchestratorPlugins: [limitsGuardPlugin.configureAgent({ limits: { maxTurns: 5 } })],
|
|
166
|
+
})],
|
|
167
|
+
mockHandler: (request) => {
|
|
168
|
+
inferenceCount++;
|
|
169
|
+
const messagesStr = JSON.stringify(request.messages);
|
|
170
|
+
if (messagesStr.includes('Approaching maxTurns limit')) {
|
|
171
|
+
sawWarning = true;
|
|
172
|
+
}
|
|
173
|
+
if (inferenceCount < 5) {
|
|
174
|
+
return {
|
|
175
|
+
content: null,
|
|
176
|
+
toolCalls: [{ id: ToolCallId(`tc${inferenceCount}`), name: 'tell_user', input: { message: `Turn ${inferenceCount}` } }],
|
|
177
|
+
finishReason: 'stop',
|
|
178
|
+
metrics: MockLLMProvider.defaultMetrics(),
|
|
179
|
+
};
|
|
180
|
+
}
|
|
181
|
+
return { content: 'Done', toolCalls: [], finishReason: 'stop', metrics: MockLLMProvider.defaultMetrics() };
|
|
182
|
+
},
|
|
183
|
+
});
|
|
184
|
+
const session = await harness.createSession('test');
|
|
185
|
+
const entryAgentId = session.getEntryAgentId();
|
|
186
|
+
await session.sendMessage('Start');
|
|
187
|
+
// Agent will be paused at turn 5 or finish at turn 5 if last response has no tools
|
|
188
|
+
await waitForAllSettled(session);
|
|
189
|
+
expect(sawWarning).toBe(true);
|
|
190
|
+
await harness.shutdown();
|
|
191
|
+
});
|
|
192
|
+
});
|
|
193
|
+
// =========================================================================
|
|
194
|
+
// counter tracking
|
|
195
|
+
// =========================================================================
|
|
196
|
+
describe('counter tracking', () => {
|
|
197
|
+
it('each inference increments inferenceCount', async () => {
|
|
198
|
+
let callCount = 0;
|
|
199
|
+
const harness = createLimitsHarness({
|
|
200
|
+
presets: [createTestPreset({
|
|
201
|
+
orchestratorSystem: 'Test agent.',
|
|
202
|
+
orchestratorPlugins: [limitsGuardPlugin.configureAgent({ limits: { maxTurns: 100 } })],
|
|
203
|
+
})],
|
|
204
|
+
mockHandler: () => {
|
|
205
|
+
callCount++;
|
|
206
|
+
if (callCount <= 2) {
|
|
207
|
+
return {
|
|
208
|
+
content: null,
|
|
209
|
+
toolCalls: [{ id: ToolCallId(`tc${callCount}`), name: 'tell_user', input: { message: `Msg ${callCount}` } }],
|
|
210
|
+
finishReason: 'stop',
|
|
211
|
+
metrics: MockLLMProvider.defaultMetrics(),
|
|
212
|
+
};
|
|
213
|
+
}
|
|
214
|
+
return { content: 'Done', toolCalls: [], finishReason: 'stop', metrics: MockLLMProvider.defaultMetrics() };
|
|
215
|
+
},
|
|
216
|
+
});
|
|
217
|
+
const session = await harness.createSession('test');
|
|
218
|
+
await session.sendAndWaitForIdle('Start');
|
|
219
|
+
const counters = selectPluginState(session.state, 'agentLimits')?.get(session.getEntryAgentId());
|
|
220
|
+
expect(counters).toBeDefined();
|
|
221
|
+
expect(counters.inferenceCount).toBe(3); // 2 with tools + 1 final
|
|
222
|
+
await harness.shutdown();
|
|
223
|
+
});
|
|
224
|
+
it('each tool call increments toolCallCount', async () => {
|
|
225
|
+
let callCount = 0;
|
|
226
|
+
const harness = createLimitsHarness({
|
|
227
|
+
presets: [createTestPreset({
|
|
228
|
+
orchestratorSystem: 'Test agent.',
|
|
229
|
+
orchestratorPlugins: [limitsGuardPlugin.configureAgent({ limits: { maxTurns: 100 } })],
|
|
230
|
+
})],
|
|
231
|
+
mockHandler: () => {
|
|
232
|
+
callCount++;
|
|
233
|
+
if (callCount === 1) {
|
|
234
|
+
return {
|
|
235
|
+
content: null,
|
|
236
|
+
toolCalls: [
|
|
237
|
+
{ id: ToolCallId('tc1'), name: 'tell_user', input: { message: 'A' } },
|
|
238
|
+
{ id: ToolCallId('tc2'), name: 'tell_user', input: { message: 'B' } },
|
|
239
|
+
],
|
|
240
|
+
finishReason: 'stop',
|
|
241
|
+
metrics: MockLLMProvider.defaultMetrics(),
|
|
242
|
+
};
|
|
243
|
+
}
|
|
244
|
+
return { content: 'Done', toolCalls: [], finishReason: 'stop', metrics: MockLLMProvider.defaultMetrics() };
|
|
245
|
+
},
|
|
246
|
+
});
|
|
247
|
+
const session = await harness.createSession('test');
|
|
248
|
+
await session.sendAndWaitForIdle('Start');
|
|
249
|
+
const counters = selectPluginState(session.state, 'agentLimits')?.get(session.getEntryAgentId());
|
|
250
|
+
expect(counters).toBeDefined();
|
|
251
|
+
expect(counters.toolCallCount).toBe(2);
|
|
252
|
+
await harness.shutdown();
|
|
253
|
+
});
|
|
254
|
+
it('each agent spawn increments parent spawnedAgentCount', async () => {
|
|
255
|
+
let orchestratorCalls = 0;
|
|
256
|
+
const harness = createLimitsHarness({
|
|
257
|
+
presets: [createMultiAgentPreset([
|
|
258
|
+
{ name: 'worker', system: 'Worker agent.', tools: [], agents: [] },
|
|
259
|
+
], { orchestratorSystem: 'Orchestrator agent.' })],
|
|
260
|
+
mockHandler: (request) => {
|
|
261
|
+
if (request.systemPrompt.includes('Orchestrator')) {
|
|
262
|
+
orchestratorCalls++;
|
|
263
|
+
if (orchestratorCalls === 1) {
|
|
264
|
+
return {
|
|
265
|
+
content: null,
|
|
266
|
+
toolCalls: [
|
|
267
|
+
{ id: ToolCallId('tc1'), name: 'start_worker', input: { message: 'Task 1' } },
|
|
268
|
+
{ id: ToolCallId('tc2'), name: 'start_worker', input: { message: 'Task 2' } },
|
|
269
|
+
],
|
|
270
|
+
finishReason: 'stop',
|
|
271
|
+
metrics: MockLLMProvider.defaultMetrics(),
|
|
272
|
+
};
|
|
273
|
+
}
|
|
274
|
+
return { content: 'Done', toolCalls: [], finishReason: 'stop', metrics: MockLLMProvider.defaultMetrics() };
|
|
275
|
+
}
|
|
276
|
+
return { content: 'Worker done', toolCalls: [], finishReason: 'stop', metrics: MockLLMProvider.defaultMetrics() };
|
|
277
|
+
},
|
|
278
|
+
});
|
|
279
|
+
const session = await harness.createSession('test');
|
|
280
|
+
await session.sendAndWaitForIdle('Start', { timeoutMs: 10000 });
|
|
281
|
+
const orchestratorId = session.getEntryAgentId();
|
|
282
|
+
const counters = selectPluginState(session.state, 'agentLimits')?.get(orchestratorId);
|
|
283
|
+
expect(counters).toBeDefined();
|
|
284
|
+
expect(counters.spawnedAgentCount).toBe(2);
|
|
285
|
+
await harness.shutdown();
|
|
286
|
+
});
|
|
287
|
+
it('each mailbox message increments sender messagesSentCount', async () => {
|
|
288
|
+
let orchestratorCalls = 0;
|
|
289
|
+
let workerCalls = 0;
|
|
290
|
+
const harness = createLimitsHarness({
|
|
291
|
+
presets: [createMultiAgentPreset([
|
|
292
|
+
{ name: 'worker', system: 'Worker agent.', tools: [], agents: [] },
|
|
293
|
+
], { orchestratorSystem: 'Orchestrator agent.' })],
|
|
294
|
+
mockHandler: (request) => {
|
|
295
|
+
if (request.systemPrompt.includes('Orchestrator')) {
|
|
296
|
+
orchestratorCalls++;
|
|
297
|
+
if (orchestratorCalls === 1) {
|
|
298
|
+
return {
|
|
299
|
+
content: null,
|
|
300
|
+
toolCalls: [{ id: ToolCallId('tc1'), name: 'start_worker', input: { message: 'Task' } }],
|
|
301
|
+
finishReason: 'stop',
|
|
302
|
+
metrics: MockLLMProvider.defaultMetrics(),
|
|
303
|
+
};
|
|
304
|
+
}
|
|
305
|
+
return { content: 'Done', toolCalls: [], finishReason: 'stop', metrics: MockLLMProvider.defaultMetrics() };
|
|
306
|
+
}
|
|
307
|
+
workerCalls++;
|
|
308
|
+
if (workerCalls === 1) {
|
|
309
|
+
return {
|
|
310
|
+
content: null,
|
|
311
|
+
toolCalls: [{ id: ToolCallId('tc2'), name: 'send_message', input: { to: 'parent', message: 'Report' } }],
|
|
312
|
+
finishReason: 'stop',
|
|
313
|
+
metrics: MockLLMProvider.defaultMetrics(),
|
|
314
|
+
};
|
|
315
|
+
}
|
|
316
|
+
return { content: 'Worker done', toolCalls: [], finishReason: 'stop', metrics: MockLLMProvider.defaultMetrics() };
|
|
317
|
+
},
|
|
318
|
+
});
|
|
319
|
+
const session = await harness.createSession('test');
|
|
320
|
+
await session.sendAndWaitForIdle('Start', { timeoutMs: 10000 });
|
|
321
|
+
const workerCounters = selectPluginState(session.state, 'agentLimits')?.get(AgentId('worker_1'));
|
|
322
|
+
expect(workerCounters).toBeDefined();
|
|
323
|
+
expect(workerCounters.messagesSentCount).toBeGreaterThanOrEqual(1);
|
|
324
|
+
await harness.shutdown();
|
|
325
|
+
});
|
|
326
|
+
});
|
|
327
|
+
// =========================================================================
|
|
328
|
+
// resume resets
|
|
329
|
+
// =========================================================================
|
|
330
|
+
describe('resume resets', () => {
|
|
331
|
+
it('agent_resumed → pattern-based counters reset, cumulative counters preserved', async () => {
|
|
332
|
+
const harness = createLimitsHarness({
|
|
333
|
+
presets: [createTestPreset({
|
|
334
|
+
orchestratorSystem: 'Test agent.',
|
|
335
|
+
orchestratorPlugins: [limitsGuardPlugin.configureAgent({
|
|
336
|
+
limits: { maxRepeatedResponses: 2, maxTurns: 100 },
|
|
337
|
+
})],
|
|
338
|
+
})],
|
|
339
|
+
// Always return the same text → triggers maxRepeatedResponses
|
|
340
|
+
mockHandler: () => ({
|
|
341
|
+
content: 'I am stuck',
|
|
342
|
+
toolCalls: [],
|
|
343
|
+
finishReason: 'stop',
|
|
344
|
+
metrics: MockLLMProvider.defaultMetrics(),
|
|
345
|
+
}),
|
|
346
|
+
});
|
|
347
|
+
const session = await harness.createSession('test');
|
|
348
|
+
const entryAgentId = session.getEntryAgentId();
|
|
349
|
+
// 1st identical response → ok → idle
|
|
350
|
+
await session.sendAndWaitForIdle('Message 1');
|
|
351
|
+
// 2nd identical response → maxRepeatedResponses = 2 → paused
|
|
352
|
+
await session.sendMessage('Message 2');
|
|
353
|
+
await waitForAgentPaused(session, entryAgentId);
|
|
354
|
+
expect(session.state.agents.get(entryAgentId).status).toBe('paused');
|
|
355
|
+
// Resume the agent
|
|
356
|
+
const resumeResult = await session.callPluginMethod('agents.resume', {
|
|
357
|
+
agentId: String(entryAgentId),
|
|
358
|
+
});
|
|
359
|
+
expect(resumeResult.ok).toBe(true);
|
|
360
|
+
// All counters should be reset after resume
|
|
361
|
+
const countersAfterResume = selectPluginState(session.state, 'agentLimits')?.get(entryAgentId);
|
|
362
|
+
expect(countersAfterResume).toBeDefined();
|
|
363
|
+
expect(countersAfterResume.inferenceCount).toBe(0);
|
|
364
|
+
expect(countersAfterResume.toolCallCount).toBe(0);
|
|
365
|
+
expect(countersAfterResume.recentResponseHashes).toEqual([]);
|
|
366
|
+
// Agent resumes with reset counters. Wait for it to settle (no work → idle).
|
|
367
|
+
await session.waitForIdle();
|
|
368
|
+
// Send 2 more messages to hit the limit again
|
|
369
|
+
await session.sendAndWaitForIdle('Message 3');
|
|
370
|
+
await session.sendMessage('Message 4');
|
|
371
|
+
await waitForAgentPaused(session, entryAgentId);
|
|
372
|
+
const resumedEvents = await session.getEventsByType(agentEvents, 'agent_resumed');
|
|
373
|
+
expect(resumedEvents.filter(e => e.agentId === entryAgentId)).toHaveLength(1);
|
|
374
|
+
await harness.shutdown();
|
|
375
|
+
});
|
|
376
|
+
});
|
|
377
|
+
});
|
|
378
|
+
//# sourceMappingURL=limits-guard.integration.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"limits-guard.integration.test.js","sourceRoot":"","sources":["../../../src/plugins/limits-guard/limits-guard.integration.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,UAAU,CAAA;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAA;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAA;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAA;AACpD,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAA;AAC9D,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAA;AACnD,OAAO,EAAE,sBAAsB,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AAE1F,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAA;AAE/C,SAAS,mBAAmB,CAAC,OAA4E;IACxG,OAAO,IAAI,WAAW,CAAC,EAAE,GAAG,OAAO,EAAE,aAAa,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAA;AAC3E,CAAC;AAID;;;GAGG;AACH,KAAK,UAAU,kBAAkB,CAAC,OAAoB,EAAE,OAAgB,EAAE,SAAS,GAAG,KAAK;IAC1F,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAA;IACvC,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC9B,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;QACpD,IAAI,UAAU,EAAE,MAAM,KAAK,QAAQ;YAAE,OAAM;QAC3C,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAA;IAC1C,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,sCAAsC,SAAS,gBAAgB,OAAO,EAAE,CAAC,CAAA;AAC1F,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,iBAAiB,CAAC,OAAoB,EAAE,SAAS,GAAG,KAAK;IACvE,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAA;IACvC,KAAK,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;QACrC,KAAK,MAAM,CAAC,EAAE,UAAU,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;YACnD,IAAI,UAAU,CAAC,MAAM,KAAK,QAAQ;gBAAE,SAAQ;YAC5C,IAAI,UAAU,CAAC,MAAM,KAAK,SAAS,IAAI,UAAU,CAAC,gBAAgB,CAAC,MAAM,KAAK,CAAC,IAAI,UAAU,CAAC,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC/H,SAAQ;YACT,CAAC;YACD,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAA;YACzC,SAAS,KAAK,CAAA;QACf,CAAC;QACD,mCAAmC;QACnC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAA;QACzC,KAAK,MAAM,CAAC,EAAE,UAAU,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;YACnD,IAAI,UAAU,CAAC,MAAM,KAAK,QAAQ;gBAAE,SAAQ;YAC5C,IAAI,UAAU,CAAC,MAAM,KAAK,SAAS,IAAI,UAAU,CAAC,gBAAgB,CAAC,MAAM,KAAK,CAAC,IAAI,UAAU,CAAC,kBAAkB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC/H,SAAQ;YACT,CAAC;YACD,SAAS,KAAK,CAAA;QACf,CAAC;QACD,OAAM;IACP,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,qCAAqC,SAAS,IAAI,CAAC,CAAA;AACpE,CAAC;AAED,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACpC,4EAA4E;IAC5E,cAAc;IACd,4EAA4E;IAE5E,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;YACxF,IAAI,cAAc,GAAG,CAAC,CAAA;YAEtB,MAAM,OAAO,GAAG,mBAAmB,CAAC;gBACnC,OAAO,EAAE,CAAC,gBAAgB,CAAC;wBAC1B,kBAAkB,EAAE,aAAa;wBACjC,mBAAmB,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;qBACpF,CAAC,CAAC;gBACH,WAAW,EAAE,GAAG,EAAE;oBACjB,cAAc,EAAE,CAAA;oBAChB,OAAO;wBACN,OAAO,EAAE,IAAI;wBACb,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,UAAU,CAAC,KAAK,cAAc,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,cAAc,EAAE,EAAE,EAAE,CAAC;wBACvH,YAAY,EAAE,MAAM;wBACpB,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE;qBACzC,CAAA;gBACF,CAAC;aACD,CAAC,CAAA;YAEF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAA;YACnD,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,EAAG,CAAA;YAC/C,MAAM,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAA;YAClC,MAAM,kBAAkB,CAAC,OAAO,EAAE,YAAY,CAAC,CAAA;YAE/C,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,YAAY,CAAE,CAAA;YAC1D,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;YAExC,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,WAAW,EAAE,cAAc,CAAC,CAAA;YAC/E,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;YACrD,MAAM,UAAU,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,YAAY,CAAC,CAAA;YACrE,MAAM,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAA;YAChC,MAAM,CAAC,UAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;YAE1C,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAA;QACzB,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACpE,IAAI,OAAO,GAAG,CAAC,CAAA;YAEf,MAAM,OAAO,GAAG,mBAAmB,CAAC;gBACnC,OAAO,EAAE,CAAC,gBAAgB,CAAC;wBAC1B,kBAAkB,EAAE,aAAa;wBACjC,mBAAmB,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;qBACvG,CAAC,CAAC;gBACH,WAAW,EAAE,GAAG,EAAE;oBACjB,OAAO,EAAE,CAAA;oBACT,OAAO;wBACN,OAAO,EAAE,IAAI;wBACb,SAAS,EAAE;4BACV,EAAE,EAAE,EAAE,UAAU,CAAC,KAAK,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,OAAO,EAAE,EAAE,EAAE;4BACzF,EAAE,EAAE,EAAE,UAAU,CAAC,KAAK,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,OAAO,EAAE,EAAE,EAAE;yBACzF;wBACD,YAAY,EAAE,MAAM;wBACpB,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE;qBACzC,CAAA;gBACF,CAAC;aACD,CAAC,CAAA;YAEF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAA;YACnD,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,EAAG,CAAA;YAC/C,MAAM,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAA;YAClC,MAAM,kBAAkB,CAAC,OAAO,EAAE,YAAY,CAAC,CAAA;YAE/C,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,YAAY,CAAE,CAAA;YAC1D,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;YAExC,MAAM,QAAQ,GAAG,iBAAiB,CAA8B,OAAO,CAAC,KAAK,EAAE,aAAa,CAAC,EAAE,GAAG,CAAC,YAAY,CAAC,CAAA;YAChH,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;YAC9B,MAAM,CAAC,QAAS,CAAC,aAAa,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;YAEzD,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAA;QACzB,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YACnE,IAAI,iBAAiB,GAAG,CAAC,CAAA;YAEzB,MAAM,OAAO,GAAG,mBAAmB,CAAC;gBACnC,OAAO,EAAE,CAAC,sBAAsB,CAAC;wBAChC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;qBAClE,EAAE,EAAE,kBAAkB,EAAE,qBAAqB,EAAE,CAAC,CAAC;gBAClD,WAAW,EAAE,CAAC,OAAO,EAAE,EAAE;oBACxB,IAAI,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;wBACnD,iBAAiB,EAAE,CAAA;wBACnB,OAAO;4BACN,OAAO,EAAE,IAAI;4BACb,SAAS,EAAE;gCACV,EAAE,EAAE,EAAE,UAAU,CAAC,KAAK,iBAAiB,GAAG,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,iBAAiB,GAAG,EAAE,EAAE;gCACrH,EAAE,EAAE,EAAE,UAAU,CAAC,KAAK,iBAAiB,GAAG,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,iBAAiB,GAAG,EAAE,EAAE;6BACrH;4BACD,YAAY,EAAE,MAAM;4BACpB,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE;yBACzC,CAAA;oBACF,CAAC;oBACD,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE,EAAE,CAAA;gBAClH,CAAC;aACD,CAAC,CAAA;YAEF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAA;YACnD,MAAM,cAAc,GAAG,OAAO,CAAC,eAAe,EAAG,CAAA;YACjD,MAAM,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAA;YAClC,MAAM,kBAAkB,CAAC,OAAO,EAAE,cAAc,EAAE,KAAK,CAAC,CAAA;YAExD,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,cAAc,CAAE,CAAA;YACtD,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;YAElC,MAAM,QAAQ,GAAG,iBAAiB,CAA8B,OAAO,CAAC,KAAK,EAAE,aAAa,CAAC,EAAE,GAAG,CAAC,cAAc,CAAC,CAAA;YAClH,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;YAC9B,MAAM,CAAC,QAAS,CAAC,iBAAiB,CAAC,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAA;YAE9D,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAA;QACzB,CAAC,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,4EAA4E;IAC5E,+BAA+B;IAC/B,4EAA4E;IAE5E,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YAC1E,IAAI,cAAc,GAAG,CAAC,CAAA;YACtB,IAAI,UAAU,GAAG,KAAK,CAAA;YAEtB,MAAM,OAAO,GAAG,mBAAmB,CAAC;gBACnC,OAAO,EAAE,CAAC,gBAAgB,CAAC;wBAC1B,kBAAkB,EAAE,aAAa;wBACjC,mBAAmB,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;qBACpF,CAAC,CAAC;gBACH,WAAW,EAAE,CAAC,OAAO,EAAE,EAAE;oBACxB,cAAc,EAAE,CAAA;oBAChB,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;oBACpD,IAAI,WAAW,CAAC,QAAQ,CAAC,4BAA4B,CAAC,EAAE,CAAC;wBACxD,UAAU,GAAG,IAAI,CAAA;oBAClB,CAAC;oBACD,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;wBACxB,OAAO;4BACN,OAAO,EAAE,IAAI;4BACb,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,UAAU,CAAC,KAAK,cAAc,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,cAAc,EAAE,EAAE,EAAE,CAAC;4BACvH,YAAY,EAAE,MAAM;4BACpB,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE;yBACzC,CAAA;oBACF,CAAC;oBACD,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE,EAAE,CAAA;gBAC3G,CAAC;aACD,CAAC,CAAA;YAEF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAA;YACnD,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,EAAG,CAAA;YAC/C,MAAM,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAA;YAClC,mFAAmF;YACnF,MAAM,iBAAiB,CAAC,OAAO,CAAC,CAAA;YAEhC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAE7B,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAA;QACzB,CAAC,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,4EAA4E;IAC5E,mBAAmB;IACnB,4EAA4E;IAE5E,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACzD,IAAI,SAAS,GAAG,CAAC,CAAA;YAEjB,MAAM,OAAO,GAAG,mBAAmB,CAAC;gBACnC,OAAO,EAAE,CAAC,gBAAgB,CAAC;wBAC1B,kBAAkB,EAAE,aAAa;wBACjC,mBAAmB,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;qBACtF,CAAC,CAAC;gBACH,WAAW,EAAE,GAAG,EAAE;oBACjB,SAAS,EAAE,CAAA;oBACX,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;wBACpB,OAAO;4BACN,OAAO,EAAE,IAAI;4BACb,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,UAAU,CAAC,KAAK,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,SAAS,EAAE,EAAE,EAAE,CAAC;4BAC5G,YAAY,EAAE,MAAM;4BACpB,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE;yBACzC,CAAA;oBACF,CAAC;oBACD,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE,EAAE,CAAA;gBAC3G,CAAC;aACD,CAAC,CAAA;YAEF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAA;YACnD,MAAM,OAAO,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAA;YAEzC,MAAM,QAAQ,GAAG,iBAAiB,CAA8B,OAAO,CAAC,KAAK,EAAE,aAAa,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,eAAe,EAAG,CAAC,CAAA;YAC9H,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;YAC9B,MAAM,CAAC,QAAS,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA,CAAC,yBAAyB;YAElE,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAA;QACzB,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACxD,IAAI,SAAS,GAAG,CAAC,CAAA;YAEjB,MAAM,OAAO,GAAG,mBAAmB,CAAC;gBACnC,OAAO,EAAE,CAAC,gBAAgB,CAAC;wBAC1B,kBAAkB,EAAE,aAAa;wBACjC,mBAAmB,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;qBACtF,CAAC,CAAC;gBACH,WAAW,EAAE,GAAG,EAAE;oBACjB,SAAS,EAAE,CAAA;oBACX,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;wBACrB,OAAO;4BACN,OAAO,EAAE,IAAI;4BACb,SAAS,EAAE;gCACV,EAAE,EAAE,EAAE,UAAU,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE;gCACrE,EAAE,EAAE,EAAE,UAAU,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE;6BACrE;4BACD,YAAY,EAAE,MAAM;4BACpB,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE;yBACzC,CAAA;oBACF,CAAC;oBACD,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE,EAAE,CAAA;gBAC3G,CAAC;aACD,CAAC,CAAA;YAEF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAA;YACnD,MAAM,OAAO,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAA;YAEzC,MAAM,QAAQ,GAAG,iBAAiB,CAA8B,OAAO,CAAC,KAAK,EAAE,aAAa,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,eAAe,EAAG,CAAC,CAAA;YAC9H,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;YAC9B,MAAM,CAAC,QAAS,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;YAEvC,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAA;QACzB,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACrE,IAAI,iBAAiB,GAAG,CAAC,CAAA;YAEzB,MAAM,OAAO,GAAG,mBAAmB,CAAC;gBACnC,OAAO,EAAE,CAAC,sBAAsB,CAAC;wBAChC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;qBAClE,EAAE,EAAE,kBAAkB,EAAE,qBAAqB,EAAE,CAAC,CAAC;gBAClD,WAAW,EAAE,CAAC,OAAO,EAAE,EAAE;oBACxB,IAAI,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;wBACnD,iBAAiB,EAAE,CAAA;wBACnB,IAAI,iBAAiB,KAAK,CAAC,EAAE,CAAC;4BAC7B,OAAO;gCACN,OAAO,EAAE,IAAI;gCACb,SAAS,EAAE;oCACV,EAAE,EAAE,EAAE,UAAU,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;oCAC7E,EAAE,EAAE,EAAE,UAAU,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;iCAC7E;gCACD,YAAY,EAAE,MAAM;gCACpB,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE;6BACzC,CAAA;wBACF,CAAC;wBACD,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE,EAAE,CAAA;oBAC3G,CAAC;oBACD,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE,EAAE,CAAA;gBAClH,CAAC;aACD,CAAC,CAAA;YAEF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAA;YACnD,MAAM,OAAO,CAAC,kBAAkB,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAA;YAE/D,MAAM,cAAc,GAAG,OAAO,CAAC,eAAe,EAAG,CAAA;YACjD,MAAM,QAAQ,GAAG,iBAAiB,CAA8B,OAAO,CAAC,KAAK,EAAE,aAAa,CAAC,EAAE,GAAG,CAAC,cAAc,CAAC,CAAA;YAClH,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAA;YAC9B,MAAM,CAAC,QAAS,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;YAE3C,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAA;QACzB,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACzE,IAAI,iBAAiB,GAAG,CAAC,CAAA;YACzB,IAAI,WAAW,GAAG,CAAC,CAAA;YAEnB,MAAM,OAAO,GAAG,mBAAmB,CAAC;gBACnC,OAAO,EAAE,CAAC,sBAAsB,CAAC;wBAChC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,eAAe,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;qBAClE,EAAE,EAAE,kBAAkB,EAAE,qBAAqB,EAAE,CAAC,CAAC;gBAClD,WAAW,EAAE,CAAC,OAAO,EAAE,EAAE;oBACxB,IAAI,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;wBACnD,iBAAiB,EAAE,CAAA;wBACnB,IAAI,iBAAiB,KAAK,CAAC,EAAE,CAAC;4BAC7B,OAAO;gCACN,OAAO,EAAE,IAAI;gCACb,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,UAAU,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC;gCACxF,YAAY,EAAE,MAAM;gCACpB,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE;6BACzC,CAAA;wBACF,CAAC;wBACD,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE,EAAE,CAAA;oBAC3G,CAAC;oBACD,WAAW,EAAE,CAAA;oBACb,IAAI,WAAW,KAAK,CAAC,EAAE,CAAC;wBACvB,OAAO;4BACN,OAAO,EAAE,IAAI;4BACb,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,UAAU,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC;4BACxG,YAAY,EAAE,MAAM;4BACpB,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE;yBACzC,CAAA;oBACF,CAAC;oBACD,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE,EAAE,CAAA;gBAClH,CAAC;aACD,CAAC,CAAA;YAEF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAA;YACnD,MAAM,OAAO,CAAC,kBAAkB,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAA;YAE/D,MAAM,cAAc,GAAG,iBAAiB,CAA8B,OAAO,CAAC,KAAK,EAAE,aAAa,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAA;YAC7H,MAAM,CAAC,cAAc,CAAC,CAAC,WAAW,EAAE,CAAA;YACpC,MAAM,CAAC,cAAe,CAAC,iBAAiB,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;YAEnE,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAA;QACzB,CAAC,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,4EAA4E;IAC5E,gBAAgB;IAChB,4EAA4E;IAE5E,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;YAC5F,MAAM,OAAO,GAAG,mBAAmB,CAAC;gBACnC,OAAO,EAAE,CAAC,gBAAgB,CAAC;wBAC1B,kBAAkB,EAAE,aAAa;wBACjC,mBAAmB,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC;gCACtD,MAAM,EAAE,EAAE,oBAAoB,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,EAAE;6BAClD,CAAC,CAAC;qBACH,CAAC,CAAC;gBACH,8DAA8D;gBAC9D,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;oBACnB,OAAO,EAAE,YAAY;oBACrB,SAAS,EAAE,EAAE;oBACb,YAAY,EAAE,MAAM;oBACpB,OAAO,EAAE,eAAe,CAAC,cAAc,EAAE;iBACzC,CAAC;aACF,CAAC,CAAA;YAEF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAA;YACnD,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,EAAG,CAAA;YAE/C,qCAAqC;YACrC,MAAM,OAAO,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAA;YAC7C,6DAA6D;YAC7D,MAAM,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAA;YACtC,MAAM,kBAAkB,CAAC,OAAO,EAAE,YAAY,CAAC,CAAA;YAE/C,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,YAAY,CAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;YAErE,mBAAmB;YACnB,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,gBAAgB,CAAC,eAAe,EAAE;gBACpE,OAAO,EAAE,MAAM,CAAC,YAAY,CAAC;aAC7B,CAAC,CAAA;YACF,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAElC,4CAA4C;YAC5C,MAAM,mBAAmB,GAAG,iBAAiB,CAA8B,OAAO,CAAC,KAAK,EAAE,aAAa,CAAC,EAAE,GAAG,CAAC,YAAY,CAAC,CAAA;YAC3H,MAAM,CAAC,mBAAmB,CAAC,CAAC,WAAW,EAAE,CAAA;YACzC,MAAM,CAAC,mBAAoB,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;YACnD,MAAM,CAAC,mBAAoB,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;YAClD,MAAM,CAAC,mBAAoB,CAAC,oBAAoB,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;YAE7D,6EAA6E;YAC7E,MAAM,OAAO,CAAC,WAAW,EAAE,CAAA;YAE3B,8CAA8C;YAC9C,MAAM,OAAO,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAA;YAC7C,MAAM,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAA;YACtC,MAAM,kBAAkB,CAAC,OAAO,EAAE,YAAY,CAAC,CAAA;YAE/C,MAAM,aAAa,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,WAAW,EAAE,eAAe,CAAC,CAAA;YACjF,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAE7E,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAA;QACzB,CAAC,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1,240 @@
|
|
|
1
|
+
import { agentEvents } from '../../core/agents/state.js';
|
|
2
|
+
import { llmEvents } from '../../core/llm/state.js';
|
|
3
|
+
import { definePlugin } from '../../core/plugins/plugin-builder.js';
|
|
4
|
+
import { selectPluginState } from '../../core/sessions/reducer.js';
|
|
5
|
+
import { toolEvents } from '../../core/tools/state.js';
|
|
6
|
+
import { responseFingerprint, toolCallFingerprint } from '../../lib/utils/hash.js';
|
|
7
|
+
import { mailboxEvents } from '../../plugins/mailbox/state.js';
|
|
8
|
+
import { checkLimits, countConsecutiveTailDuplicates, resolveAgentLimits } from './limit-guard.js';
|
|
9
|
+
export const createAgentCounters = () => ({
|
|
10
|
+
inferenceCount: 0,
|
|
11
|
+
toolCallCount: 0,
|
|
12
|
+
spawnedAgentCount: 0,
|
|
13
|
+
messagesSentCount: 0,
|
|
14
|
+
consecutiveToolFailures: {},
|
|
15
|
+
recentToolCallHashes: [],
|
|
16
|
+
recentResponseHashes: [],
|
|
17
|
+
});
|
|
18
|
+
/**
|
|
19
|
+
* Extract agent counters from session state (for external consumers).
|
|
20
|
+
*/
|
|
21
|
+
export function selectAgentCounters(sessionState, agentId) {
|
|
22
|
+
return selectPluginState(sessionState, 'agentLimits')?.get(agentId) ?? createAgentCounters();
|
|
23
|
+
}
|
|
24
|
+
// ============================================================================
|
|
25
|
+
// Limit warning event
|
|
26
|
+
// ============================================================================
|
|
27
|
+
import z from 'zod/v4';
|
|
28
|
+
import { agentIdSchema } from '../../core/agents/schema.js';
|
|
29
|
+
import { createEventsFactory } from '../../core/events/types.js';
|
|
30
|
+
export const limitsEvents = createEventsFactory({
|
|
31
|
+
events: {
|
|
32
|
+
limit_warning: z.object({
|
|
33
|
+
agentId: agentIdSchema,
|
|
34
|
+
limitName: z.string(),
|
|
35
|
+
currentValue: z.number(),
|
|
36
|
+
hardLimit: z.number(),
|
|
37
|
+
message: z.string(),
|
|
38
|
+
}),
|
|
39
|
+
},
|
|
40
|
+
});
|
|
41
|
+
// ============================================================================
|
|
42
|
+
// Helper
|
|
43
|
+
// ============================================================================
|
|
44
|
+
/**
|
|
45
|
+
* Resolve tool name from agent state for tool_completed/tool_failed events.
|
|
46
|
+
* Must run BEFORE core reducer (which clears executingToolCall).
|
|
47
|
+
*/
|
|
48
|
+
function resolveToolName(sessionState, agentId, toolCallId) {
|
|
49
|
+
const agent = sessionState.agents.get(agentId);
|
|
50
|
+
if (!agent)
|
|
51
|
+
return 'unknown';
|
|
52
|
+
return agent.executingToolCall?.toolName
|
|
53
|
+
?? agent.pendingToolCalls.find((tc) => tc.id === toolCallId)?.name
|
|
54
|
+
?? 'unknown';
|
|
55
|
+
}
|
|
56
|
+
export const limitsGuardPlugin = definePlugin('limits-guard')
|
|
57
|
+
.events([agentEvents, llmEvents, toolEvents, mailboxEvents])
|
|
58
|
+
.state({
|
|
59
|
+
key: 'agentLimits',
|
|
60
|
+
initial: () => new Map(),
|
|
61
|
+
reduce: (limits, event, sessionState) => {
|
|
62
|
+
switch (event.type) {
|
|
63
|
+
case 'agent_spawned': {
|
|
64
|
+
const newLimits = new Map(limits);
|
|
65
|
+
newLimits.set(event.agentId, createAgentCounters());
|
|
66
|
+
// Increment parent's spawnedAgentCount
|
|
67
|
+
if (event.parentId) {
|
|
68
|
+
const parentCounters = newLimits.get(event.parentId);
|
|
69
|
+
if (parentCounters) {
|
|
70
|
+
newLimits.set(event.parentId, {
|
|
71
|
+
...parentCounters,
|
|
72
|
+
spawnedAgentCount: parentCounters.spawnedAgentCount + 1,
|
|
73
|
+
});
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
return newLimits;
|
|
77
|
+
}
|
|
78
|
+
case 'mailbox_message': {
|
|
79
|
+
// Increment sender's messagesSentCount if sender is an agent
|
|
80
|
+
const senderAgentId = typeof event.message.from === 'string' && event.message.from !== 'user'
|
|
81
|
+
? event.message.from
|
|
82
|
+
: null;
|
|
83
|
+
if (!senderAgentId)
|
|
84
|
+
return limits;
|
|
85
|
+
const senderCounters = limits.get(senderAgentId);
|
|
86
|
+
if (!senderCounters)
|
|
87
|
+
return limits;
|
|
88
|
+
const newLimits = new Map(limits);
|
|
89
|
+
newLimits.set(senderAgentId, {
|
|
90
|
+
...senderCounters,
|
|
91
|
+
messagesSentCount: senderCounters.messagesSentCount + 1,
|
|
92
|
+
});
|
|
93
|
+
return newLimits;
|
|
94
|
+
}
|
|
95
|
+
case 'inference_completed': {
|
|
96
|
+
const counters = limits.get(event.agentId);
|
|
97
|
+
if (!counters)
|
|
98
|
+
return limits;
|
|
99
|
+
const hasToolCalls = event.response.toolCalls.length > 0;
|
|
100
|
+
let newRecentResponseHashes = counters.recentResponseHashes;
|
|
101
|
+
if (!hasToolCalls) {
|
|
102
|
+
const isWaiting = /^\s*WAITING\s*$/.test(event.response.content ?? '');
|
|
103
|
+
if (!isWaiting) {
|
|
104
|
+
const hash = responseFingerprint(event.response.content);
|
|
105
|
+
newRecentResponseHashes = [...newRecentResponseHashes, hash].slice(-10);
|
|
106
|
+
}
|
|
107
|
+
}
|
|
108
|
+
const newLimits = new Map(limits);
|
|
109
|
+
newLimits.set(event.agentId, {
|
|
110
|
+
...counters,
|
|
111
|
+
inferenceCount: counters.inferenceCount + 1,
|
|
112
|
+
recentResponseHashes: newRecentResponseHashes,
|
|
113
|
+
});
|
|
114
|
+
return newLimits;
|
|
115
|
+
}
|
|
116
|
+
case 'tool_started': {
|
|
117
|
+
const counters = limits.get(event.agentId);
|
|
118
|
+
if (!counters)
|
|
119
|
+
return limits;
|
|
120
|
+
const fingerprint = toolCallFingerprint(event.toolName, event.input);
|
|
121
|
+
const newLimits = new Map(limits);
|
|
122
|
+
newLimits.set(event.agentId, {
|
|
123
|
+
...counters,
|
|
124
|
+
toolCallCount: counters.toolCallCount + 1,
|
|
125
|
+
recentToolCallHashes: [...counters.recentToolCallHashes, fingerprint].slice(-20),
|
|
126
|
+
});
|
|
127
|
+
return newLimits;
|
|
128
|
+
}
|
|
129
|
+
case 'tool_completed': {
|
|
130
|
+
const counters = limits.get(event.agentId);
|
|
131
|
+
if (!counters)
|
|
132
|
+
return limits;
|
|
133
|
+
// Resolve tool name from agent state (must run before core clears executingToolCall)
|
|
134
|
+
const toolName = resolveToolName(sessionState, event.agentId, event.toolCallId);
|
|
135
|
+
const { [toolName]: _, ...restFailures } = counters.consecutiveToolFailures;
|
|
136
|
+
const newLimits = new Map(limits);
|
|
137
|
+
newLimits.set(event.agentId, {
|
|
138
|
+
...counters,
|
|
139
|
+
consecutiveToolFailures: restFailures,
|
|
140
|
+
});
|
|
141
|
+
return newLimits;
|
|
142
|
+
}
|
|
143
|
+
case 'tool_failed': {
|
|
144
|
+
const counters = limits.get(event.agentId);
|
|
145
|
+
if (!counters)
|
|
146
|
+
return limits;
|
|
147
|
+
// Resolve tool name from agent state (must run before core clears executingToolCall)
|
|
148
|
+
const toolName = resolveToolName(sessionState, event.agentId, event.toolCallId);
|
|
149
|
+
const currentEntry = counters.consecutiveToolFailures[toolName];
|
|
150
|
+
const newLimits = new Map(limits);
|
|
151
|
+
newLimits.set(event.agentId, {
|
|
152
|
+
...counters,
|
|
153
|
+
consecutiveToolFailures: {
|
|
154
|
+
...counters.consecutiveToolFailures,
|
|
155
|
+
[toolName]: { count: (currentEntry?.count ?? 0) + 1, lastError: event.error },
|
|
156
|
+
},
|
|
157
|
+
});
|
|
158
|
+
return newLimits;
|
|
159
|
+
}
|
|
160
|
+
case 'agent_resumed': {
|
|
161
|
+
const counters = limits.get(event.agentId);
|
|
162
|
+
if (!counters)
|
|
163
|
+
return limits;
|
|
164
|
+
const newLimits = new Map(limits);
|
|
165
|
+
newLimits.set(event.agentId, {
|
|
166
|
+
...counters,
|
|
167
|
+
inferenceCount: 0,
|
|
168
|
+
toolCallCount: 0,
|
|
169
|
+
spawnedAgentCount: 0,
|
|
170
|
+
messagesSentCount: 0,
|
|
171
|
+
consecutiveToolFailures: {},
|
|
172
|
+
recentToolCallHashes: [],
|
|
173
|
+
recentResponseHashes: [],
|
|
174
|
+
});
|
|
175
|
+
return newLimits;
|
|
176
|
+
}
|
|
177
|
+
default:
|
|
178
|
+
return limits;
|
|
179
|
+
}
|
|
180
|
+
},
|
|
181
|
+
})
|
|
182
|
+
.agentConfig()
|
|
183
|
+
.hook('afterInference', async (ctx) => {
|
|
184
|
+
const resolvedLimits = resolveAgentLimits(ctx.pluginAgentConfig?.limits);
|
|
185
|
+
const counters = ctx.pluginState.get(ctx.agentId);
|
|
186
|
+
if (!counters)
|
|
187
|
+
return null;
|
|
188
|
+
// The inference_completed event hasn't been emitted yet, so project
|
|
189
|
+
// what the counters will be after it's processed by the reducer.
|
|
190
|
+
const hasToolCalls = ctx.response.toolCalls.length > 0;
|
|
191
|
+
const projected = {
|
|
192
|
+
...counters,
|
|
193
|
+
inferenceCount: counters.inferenceCount + 1,
|
|
194
|
+
recentResponseHashes: hasToolCalls
|
|
195
|
+
? counters.recentResponseHashes
|
|
196
|
+
: [...counters.recentResponseHashes, responseFingerprint(ctx.response.content)].slice(-10),
|
|
197
|
+
};
|
|
198
|
+
const limitCheck = checkLimits(projected, resolvedLimits);
|
|
199
|
+
if (limitCheck.status === 'hard_limit') {
|
|
200
|
+
return { action: 'pause', reason: limitCheck.reason };
|
|
201
|
+
}
|
|
202
|
+
return null;
|
|
203
|
+
})
|
|
204
|
+
.status((ctx) => {
|
|
205
|
+
const resolvedLimits = resolveAgentLimits(ctx.pluginAgentConfig?.limits);
|
|
206
|
+
const counters = ctx.pluginState.get(ctx.agentId);
|
|
207
|
+
if (!counters)
|
|
208
|
+
return null;
|
|
209
|
+
const parts = [];
|
|
210
|
+
// Consecutive failures of the same tool — surface from count=2, well before the
|
|
211
|
+
// hard limit. The agent sees what it tried, how many times, and with what error,
|
|
212
|
+
// so it can change strategy rather than blindly retry.
|
|
213
|
+
for (const [toolName, entry] of Object.entries(counters.consecutiveToolFailures)) {
|
|
214
|
+
if (entry.count >= 2) {
|
|
215
|
+
const lastError = entry.lastError.length > 600
|
|
216
|
+
? entry.lastError.slice(0, 600) + '… [truncated]'
|
|
217
|
+
: entry.lastError;
|
|
218
|
+
parts.push(`⚠️ Tool "${toolName}" has failed ${entry.count} times in a row (hard limit: ${resolvedLimits.maxConsecutiveToolFailures}). `
|
|
219
|
+
+ `Last error: ${lastError}\n`
|
|
220
|
+
+ `Do NOT retry the same call again. Diagnose the root cause, change approach, or stop and report what you tried.`);
|
|
221
|
+
}
|
|
222
|
+
}
|
|
223
|
+
// Same tool call with identical arguments repeated — even successful calls
|
|
224
|
+
// that produce no new information are a waste. Warn from 2 repeats.
|
|
225
|
+
const repeatedCalls = countConsecutiveTailDuplicates(counters.recentToolCallHashes);
|
|
226
|
+
if (repeatedCalls >= 2) {
|
|
227
|
+
parts.push(`⚠️ You have called the same tool with identical arguments ${repeatedCalls} times in a row `
|
|
228
|
+
+ `(hard limit: ${resolvedLimits.maxRepeatedToolCalls}). `
|
|
229
|
+
+ `Either the call is idempotent (stop repeating it) or your approach isn't working (change strategy).`);
|
|
230
|
+
}
|
|
231
|
+
// Existing counter-based soft warnings (maxTurns, maxToolCalls, etc.)
|
|
232
|
+
const limitCheck = checkLimits(counters, resolvedLimits);
|
|
233
|
+
if (limitCheck.status === 'soft_warning') {
|
|
234
|
+
parts.push(`⚠️ Approaching ${limitCheck.limitName} limit: ${limitCheck.currentValue}/${limitCheck.hardLimit}. `
|
|
235
|
+
+ `Wrap up your current task or you will be stopped.`);
|
|
236
|
+
}
|
|
237
|
+
return parts.length > 0 ? parts.join('\n\n') : null;
|
|
238
|
+
})
|
|
239
|
+
.build();
|
|
240
|
+
//# sourceMappingURL=plugin.js.map
|