@cohortapp/agent-sdk 2.3.0
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/.claude/commands/init-agent.md +104 -0
- package/.claude/commands/init-maestro.md +1187 -0
- package/.claude/settings.json +161 -0
- package/.env.example +216 -0
- package/README.md +632 -0
- package/agents/browser-operator/agent.md +52 -0
- package/agents/calendar-ops/agent.md +50 -0
- package/agents/communications/agent.md +96 -0
- package/agents/decision-log/agent.md +65 -0
- package/agents/desktop-operator/agent.md +59 -0
- package/agents/gmail-operator/agent.md +62 -0
- package/agents/inbound-dispatcher/agent.md +66 -0
- package/agents/inbox-processor/agent.md +39 -0
- package/agents/pmo-execution/agent.md +60 -0
- package/agents/session-spawner/agent.md +64 -0
- package/agents/slack-operator/agent.md +60 -0
- package/agents/whatsapp-operator/agent.md +60 -0
- package/agents/workflow-automation/agent.md +61 -0
- package/archetypes/altitudes/c-suite.yaml +58 -0
- package/archetypes/altitudes/founder.yaml +68 -0
- package/archetypes/altitudes/senior-manager.yaml +63 -0
- package/archetypes/altitudes/svp.yaml +60 -0
- package/archetypes/altitudes/vp.yaml +50 -0
- package/archetypes/archetype.schema.json +77 -0
- package/archetypes/base.yaml +47 -0
- package/archetypes/capabilities/commercial-leader.yaml +159 -0
- package/archetypes/capabilities/compliance-officer.yaml +159 -0
- package/archetypes/capabilities/executive-operator.yaml +169 -0
- package/archetypes/capabilities/finance-leader.yaml +162 -0
- package/archetypes/capabilities/operations-leader.yaml +154 -0
- package/archetypes/capabilities/product-leader.yaml +148 -0
- package/archetypes/capabilities/technical-leader.yaml +146 -0
- package/archetypes/functions/commercial-leader.yaml +62 -0
- package/archetypes/functions/compliance-officer.yaml +64 -0
- package/archetypes/functions/executive-operator.yaml +70 -0
- package/archetypes/functions/finance-leader.yaml +70 -0
- package/archetypes/functions/operations-leader.yaml +62 -0
- package/archetypes/functions/product-leader.yaml +61 -0
- package/archetypes/functions/technical-leader.yaml +57 -0
- package/bin/cohort-mcp.mjs +81 -0
- package/bin/maestro.mjs +3516 -0
- package/bin/maestro.test.mjs +1015 -0
- package/desktop-control/README.md +56 -0
- package/desktop-control/app-profiles/gmail.yaml +120 -0
- package/desktop-control/app-profiles/slack.yaml +315 -0
- package/desktop-control/app-profiles/whatsapp.yaml +107 -0
- package/docs/architecture/agent-topology.md +2239 -0
- package/docs/architecture/archetype-agent-factory.md +110 -0
- package/docs/architecture/collective-memory-and-org-mesh.md +115 -0
- package/docs/architecture/continuous-monitoring.md +221 -0
- package/docs/architecture/mcp-capability-map.md +585 -0
- package/docs/architecture/system-architecture.md +1272 -0
- package/docs/company-context/README.md +40 -0
- package/docs/guides/agent-persona-setup.md +600 -0
- package/docs/guides/agents-observe-setup.md +64 -0
- package/docs/guides/billing-console-keys.md +88 -0
- package/docs/guides/ccxray-diagnostics.md +65 -0
- package/docs/guides/channel-bus.md +127 -0
- package/docs/guides/claude-mem-setup.md +79 -0
- package/docs/guides/claude-pace-setup.md +56 -0
- package/docs/guides/claudraband-sessions.md +98 -0
- package/docs/guides/clawteam-swarm.md +116 -0
- package/docs/guides/code-review-graph-setup.md +86 -0
- package/docs/guides/email-setup.md +431 -0
- package/docs/guides/mac-mini.md +119 -0
- package/docs/guides/media-generation-setup.md +349 -0
- package/docs/guides/model-routing.md +162 -0
- package/docs/guides/observability-otel.md +265 -0
- package/docs/guides/org-onboarding.md +132 -0
- package/docs/guides/outbound-governance-setup.md +437 -0
- package/docs/guides/pdf-generation-setup.md +315 -0
- package/docs/guides/poller-daemon-setup.md +563 -0
- package/docs/guides/rag-context-setup.md +459 -0
- package/docs/guides/self-optimization-pattern.md +82 -0
- package/docs/guides/setup-wizard.md +178 -0
- package/docs/guides/slack-setup.md +350 -0
- package/docs/guides/telegram-setup.md +227 -0
- package/docs/guides/twilio-subaccounts-setup.md +223 -0
- package/docs/guides/verification.md +128 -0
- package/docs/guides/voice-mode.md +188 -0
- package/docs/guides/voice-sms-setup.md +698 -0
- package/docs/guides/webhook-relay-setup.md +349 -0
- package/docs/guides/whatsapp-setup.md +288 -0
- package/docs/prompts/board-pack-cover-template.md +36 -0
- package/docs/prompts/decision-recommendation-template.md +88 -0
- package/docs/prompts/followup-message-template.md +141 -0
- package/docs/prompts/investor-letter-template.md +52 -0
- package/docs/prompts/morning-brief-template.md +82 -0
- package/docs/prompts/presentation-template.md +58 -0
- package/docs/prompts/weekly-strategic-memo-template.md +104 -0
- package/docs/research/hallucinated-tool-output-investigation.md +151 -0
- package/docs/runbooks/backup-restore.md +205 -0
- package/docs/runbooks/cohort-cutover.md +129 -0
- package/docs/runbooks/fleet-operations.md +200 -0
- package/docs/runbooks/incident-response.md +226 -0
- package/docs/runbooks/mac-mini-bootstrap.md +431 -0
- package/docs/runbooks/perpetual-operations.md +509 -0
- package/docs/runbooks/recovery-and-failover.md +260 -0
- package/framework-features.json +267 -0
- package/ingest/README.md +87 -0
- package/lib/action-executor.js +689 -0
- package/lib/action-executor.test.mjs +871 -0
- package/lib/agent-root.mjs +37 -0
- package/lib/archetype.mjs +236 -0
- package/lib/archetype.test.mjs +132 -0
- package/lib/autonomy.mjs +114 -0
- package/lib/autonomy.test.mjs +66 -0
- package/lib/backlog.mjs +358 -0
- package/lib/backlog.test.mjs +266 -0
- package/lib/budget-guard.mjs +279 -0
- package/lib/budget-guard.test.mjs +291 -0
- package/lib/cadence-bus-schedule.test.mjs +194 -0
- package/lib/cadence-bus.mjs +1120 -0
- package/lib/cadence-bus.test.mjs +720 -0
- package/lib/cadences.mjs +205 -0
- package/lib/cadences.test.mjs +125 -0
- package/lib/capability.mjs +154 -0
- package/lib/capability.test.mjs +78 -0
- package/lib/channels/base-adapter.mjs +719 -0
- package/lib/channels/base-adapter.test.mjs +590 -0
- package/lib/channels/channel.mjs +128 -0
- package/lib/channels/channels.test.mjs +371 -0
- package/lib/channels/contract.mjs +215 -0
- package/lib/channels/contract.test.mjs +137 -0
- package/lib/channels/conversation-resolver.mjs +95 -0
- package/lib/channels/gmail/adapter.mjs +87 -0
- package/lib/channels/inbox-item.mjs +255 -0
- package/lib/channels/inbox-item.test.mjs +335 -0
- package/lib/channels/index.mjs +94 -0
- package/lib/channels/orgmail/adapter.mjs +353 -0
- package/lib/channels/orgmail/adapter.test.mjs +311 -0
- package/lib/channels/pairing.mjs +363 -0
- package/lib/channels/pairing.test.mjs +270 -0
- package/lib/channels/registry.mjs +164 -0
- package/lib/channels/slack/adapter.mjs +317 -0
- package/lib/channels/slack-adapter.test.mjs +212 -0
- package/lib/channels/sms/adapter.mjs +43 -0
- package/lib/channels/telegram/adapter.mjs +432 -0
- package/lib/channels/telegram-adapter.test.mjs +306 -0
- package/lib/channels/voice/adapter.mjs +301 -0
- package/lib/channels/voice/adapter.test.mjs +278 -0
- package/lib/channels/whatsapp/adapter-baileys.mjs +587 -0
- package/lib/channels/whatsapp/adapter-baileys.test.mjs +359 -0
- package/lib/channels/whatsapp/adapter-twilio.mjs +65 -0
- package/lib/channels/whatsapp/baileys-typing.test.mjs +154 -0
- package/lib/charter.mjs +256 -0
- package/lib/charter.test.mjs +89 -0
- package/lib/claude-bin.mjs +134 -0
- package/lib/claude-bin.test.mjs +75 -0
- package/lib/collective/capture.mjs +185 -0
- package/lib/collective/capture.test.mjs +121 -0
- package/lib/collective/cards.mjs +201 -0
- package/lib/collective/cards.test.mjs +114 -0
- package/lib/collective/config.mjs +186 -0
- package/lib/collective/config.test.mjs +123 -0
- package/lib/collective/global-config.mjs +113 -0
- package/lib/collective/global-config.test.mjs +75 -0
- package/lib/collective/presence.mjs +201 -0
- package/lib/collective/presence.test.mjs +95 -0
- package/lib/collective/recall.mjs +215 -0
- package/lib/collective/recall.test.mjs +116 -0
- package/lib/comms/send-gate.mjs +554 -0
- package/lib/comms/send-gate.test.mjs +577 -0
- package/lib/comms.mjs +67 -0
- package/lib/comms.test.mjs +41 -0
- package/lib/diagnostics/alerts.mjs +424 -0
- package/lib/diagnostics/alerts.test.mjs +318 -0
- package/lib/diagnostics/backup-freshness.mjs +188 -0
- package/lib/diagnostics/backup-freshness.test.mjs +185 -0
- package/lib/diagnostics/counters.mjs +269 -0
- package/lib/diagnostics/counters.test.mjs +206 -0
- package/lib/diagnostics/events.mjs +188 -0
- package/lib/diagnostics/events.test.mjs +290 -0
- package/lib/diagnostics/otel.mjs +237 -0
- package/lib/diagnostics/otel.test.mjs +196 -0
- package/lib/diagnostics/trace.mjs +216 -0
- package/lib/diagnostics/trace.test.mjs +251 -0
- package/lib/env-compat.mjs +74 -0
- package/lib/env-compat.test.mjs +104 -0
- package/lib/feature-init.mjs +331 -0
- package/lib/fs-atomic.mjs +112 -0
- package/lib/fs-atomic.test.mjs +72 -0
- package/lib/fs-ownership.mjs +111 -0
- package/lib/fs-ownership.test.mjs +158 -0
- package/lib/hooks/bus.mjs +347 -0
- package/lib/hooks/bus.test.mjs +387 -0
- package/lib/index.js +16 -0
- package/lib/learning/config.mjs +106 -0
- package/lib/learning/config.test.mjs +75 -0
- package/lib/learning/counters.mjs +156 -0
- package/lib/learning/counters.test.mjs +69 -0
- package/lib/learning/curator-consolidate.test.mjs +238 -0
- package/lib/learning/curator.mjs +453 -0
- package/lib/learning/curator.test.mjs +106 -0
- package/lib/learning/log.mjs +40 -0
- package/lib/learning/reflect.mjs +534 -0
- package/lib/learning/reflect.test.mjs +0 -0
- package/lib/learning/session-index.mjs +352 -0
- package/lib/learning/session-index.test.mjs +125 -0
- package/lib/learning/skill-writer.mjs +474 -0
- package/lib/learning/skill-writer.test.mjs +210 -0
- package/lib/mcp/server.mjs +328 -0
- package/lib/mcp/server.test.mjs +400 -0
- package/lib/model-router/auth-profiles.mjs +758 -0
- package/lib/model-router/auth-profiles.test.mjs +580 -0
- package/lib/model-router/catalog/anthropic.yaml +153 -0
- package/lib/model-router/catalog/deepseek.yaml +86 -0
- package/lib/model-router/catalog/moonshot.yaml +81 -0
- package/lib/model-router/catalog/qwen.yaml +114 -0
- package/lib/model-router/catalog.mjs +925 -0
- package/lib/model-router/catalog.test.mjs +385 -0
- package/lib/model-router/economics.mjs +564 -0
- package/lib/model-router/economics.test.mjs +344 -0
- package/lib/model-router/failover.mjs +298 -0
- package/lib/model-router/failover.test.mjs +439 -0
- package/lib/model-router/health.mjs +453 -0
- package/lib/model-router/health.test.mjs +338 -0
- package/lib/model-router/integration-coverage.test.mjs +829 -0
- package/lib/model-router/integration.test.mjs +564 -0
- package/lib/model-router/ledger.mjs +402 -0
- package/lib/model-router/ledger.test.mjs +382 -0
- package/lib/model-router/llm-task.mjs +515 -0
- package/lib/model-router/llm-task.test.mjs +392 -0
- package/lib/model-router/org-credentials.mjs +260 -0
- package/lib/model-router/org-credentials.test.mjs +265 -0
- package/lib/model-router/pricing-refresh.mjs +463 -0
- package/lib/model-router/pricing-refresh.test.mjs +286 -0
- package/lib/model-router/reconcile.mjs +429 -0
- package/lib/model-router/reconcile.test.mjs +316 -0
- package/lib/model-router/repair.mjs +471 -0
- package/lib/model-router/repair.test.mjs +180 -0
- package/lib/model-router/resolve.mjs +1206 -0
- package/lib/model-router/spawn.mjs +497 -0
- package/lib/model-router/spawn.test.mjs +425 -0
- package/lib/model-router/taxonomy.mjs +893 -0
- package/lib/model-router/taxonomy.test.mjs +410 -0
- package/lib/model-router.mjs +677 -0
- package/lib/model-router.test.mjs +907 -0
- package/lib/org/activity.mjs +211 -0
- package/lib/org/activity.test.mjs +134 -0
- package/lib/org/approvals.mjs +448 -0
- package/lib/org/approvals.test.mjs +216 -0
- package/lib/org/awareness.mjs +222 -0
- package/lib/org/awareness.test.mjs +159 -0
- package/lib/org/board.mjs +229 -0
- package/lib/org/board.test.mjs +177 -0
- package/lib/org/bootstrap-context.mjs +169 -0
- package/lib/org/bootstrap-context.test.mjs +153 -0
- package/lib/org/client.mjs +1628 -0
- package/lib/org/client.test.mjs +1107 -0
- package/lib/org/cohort-client.mjs +67 -0
- package/lib/org/cohort-client.test.mjs +126 -0
- package/lib/org/cost-sync.mjs +227 -0
- package/lib/org/cost-sync.test.mjs +153 -0
- package/lib/org/doctor.mjs +212 -0
- package/lib/org/doctor.test.mjs +212 -0
- package/lib/org/handoff.mjs +293 -0
- package/lib/org/handoff.test.mjs +269 -0
- package/lib/org/integration-tools.mjs +182 -0
- package/lib/org/integration-tools.test.mjs +160 -0
- package/lib/org/keys.mjs +131 -0
- package/lib/org/keys.test.mjs +92 -0
- package/lib/org/knowledge.mjs +463 -0
- package/lib/org/knowledge.test.mjs +319 -0
- package/lib/org/leases.mjs +335 -0
- package/lib/org/leases.test.mjs +235 -0
- package/lib/org/mesh-integration.test.mjs +127 -0
- package/lib/org/mesh.mjs +459 -0
- package/lib/org/mesh.test.mjs +345 -0
- package/lib/org/messaging.mjs +503 -0
- package/lib/org/messaging.test.mjs +238 -0
- package/lib/org/policy.mjs +345 -0
- package/lib/org/policy.test.mjs +237 -0
- package/lib/org/protocol.checksum +1 -0
- package/lib/org/protocol.checksum.test.mjs +90 -0
- package/lib/org/protocol.mjs +967 -0
- package/lib/org/protocol.test.mjs +264 -0
- package/lib/org/registry.mjs +194 -0
- package/lib/org/registry.test.mjs +100 -0
- package/lib/org/tool-surface-integration.test.mjs +120 -0
- package/lib/org/tool-surface.mjs +2535 -0
- package/lib/org/tool-surface.test.mjs +589 -0
- package/lib/org/ui-parity.mjs +3236 -0
- package/lib/org/ui-parity.test.mjs +348 -0
- package/lib/org/verify.mjs +176 -0
- package/lib/org/verify.test.mjs +194 -0
- package/lib/rag/embed.mjs +188 -0
- package/lib/rag/indexer.mjs +425 -0
- package/lib/rag/rag.test.mjs +505 -0
- package/lib/rag/search.mjs +475 -0
- package/lib/rate-guard.mjs +246 -0
- package/lib/rate-guard.test.mjs +201 -0
- package/lib/render.mjs +112 -0
- package/lib/render.test.mjs +68 -0
- package/lib/resource-governor.mjs +297 -0
- package/lib/resource-governor.test.mjs +262 -0
- package/lib/scheduling/dynamic-jobs.mjs +675 -0
- package/lib/scheduling/dynamic-jobs.test.mjs +344 -0
- package/lib/scheduling/jitter.mjs +0 -0
- package/lib/scheduling/jitter.test.mjs +140 -0
- package/lib/secrets/broker.mjs +315 -0
- package/lib/secrets/broker.test.mjs +280 -0
- package/lib/secrets/providers.mjs +461 -0
- package/lib/secrets/providers.test.mjs +274 -0
- package/lib/security/audit-engine.mjs +684 -0
- package/lib/security/audit-engine.test.mjs +389 -0
- package/lib/security/coerce-args.mjs +552 -0
- package/lib/security/coerce-args.test.mjs +281 -0
- package/lib/security/dangerous-tools.mjs +97 -0
- package/lib/security/dangerous-tools.test.mjs +68 -0
- package/lib/security/external-content.mjs +145 -0
- package/lib/security/external-content.test.mjs +67 -0
- package/lib/security/redact.mjs +592 -0
- package/lib/security/redact.test.mjs +441 -0
- package/lib/security/secret-equal.mjs +73 -0
- package/lib/security/secret-equal.test.mjs +55 -0
- package/lib/session-permissions.mjs +101 -0
- package/lib/session-permissions.test.mjs +100 -0
- package/lib/setup/claude-probe.mjs +74 -0
- package/lib/setup/completeness.mjs +175 -0
- package/lib/setup/completeness.test.mjs +110 -0
- package/lib/setup/context-pack.mjs +173 -0
- package/lib/setup/context-pack.test.mjs +89 -0
- package/lib/setup/enrich.mjs +277 -0
- package/lib/setup/enrich.test.mjs +115 -0
- package/lib/setup/enroll-from-cohort.mjs +441 -0
- package/lib/setup/enroll-from-cohort.test.mjs +233 -0
- package/lib/setup/integration.test.mjs +162 -0
- package/lib/setup/io.mjs +360 -0
- package/lib/setup/io.test.mjs +77 -0
- package/lib/setup/run-generator.mjs +81 -0
- package/lib/setup/runner.mjs +244 -0
- package/lib/setup/runner.test.mjs +132 -0
- package/lib/setup/sections/comms.mjs +173 -0
- package/lib/setup/sections/company.mjs +120 -0
- package/lib/setup/sections/enrich.mjs +138 -0
- package/lib/setup/sections/identity.mjs +182 -0
- package/lib/setup/sections/identity.test.mjs +140 -0
- package/lib/setup/sections/learning.mjs +153 -0
- package/lib/setup/sections/learning.test.mjs +81 -0
- package/lib/setup/sections/messaging.mjs +219 -0
- package/lib/setup/sections/messaging.test.mjs +127 -0
- package/lib/setup/sections/model.mjs +102 -0
- package/lib/setup/sections/operating-model.mjs +78 -0
- package/lib/setup/sections/org.mjs +475 -0
- package/lib/setup/sections/org.test.mjs +313 -0
- package/lib/setup/sections/orgmail.mjs +173 -0
- package/lib/setup/sections/orgmail.test.mjs +118 -0
- package/lib/setup/sections/recovery.mjs +159 -0
- package/lib/setup/sections/recovery.test.mjs +98 -0
- package/lib/setup/sections/tools.mjs +132 -0
- package/lib/setup/sections/verify.mjs +97 -0
- package/lib/setup/sot.mjs +205 -0
- package/lib/setup/sot.test.mjs +81 -0
- package/lib/setup/state.mjs +151 -0
- package/lib/setup/state.test.mjs +92 -0
- package/lib/singleton.js +229 -0
- package/lib/singleton.test.mjs +135 -0
- package/lib/telemetry/alerts.mjs +216 -0
- package/lib/telemetry/alerts.test.mjs +109 -0
- package/lib/telemetry/collect.mjs +512 -0
- package/lib/telemetry/collect.test.mjs +202 -0
- package/lib/tool-definitions-integration.test.mjs +83 -0
- package/lib/tool-definitions.js +738 -0
- package/lib/tool-definitions.test.mjs +437 -0
- package/lib/util/fetch-timeout.mjs +136 -0
- package/lib/util/fetch-timeout.test.mjs +202 -0
- package/lib/util/reconnect.mjs +343 -0
- package/lib/util/reconnect.test.mjs +369 -0
- package/lib/util/unhandled.mjs +205 -0
- package/lib/util/unhandled.test.mjs +216 -0
- package/lib/voice/context-loader.mjs +466 -0
- package/lib/voice/index.mjs +100 -0
- package/lib/voice/openai-realtime.mjs +510 -0
- package/lib/voice/outbound.mjs +542 -0
- package/lib/voice/outbound.test.mjs +69 -0
- package/lib/voice/post-call-brief.mjs +428 -0
- package/lib/voice/provider.mjs +52 -0
- package/lib/voice/session-rotation.mjs +257 -0
- package/lib/voice/stt.mjs +161 -0
- package/lib/voice/stt.test.mjs +226 -0
- package/lib/voice/tool-bridge.mjs +370 -0
- package/lib/voice/tts.mjs +104 -0
- package/lib/voice/twilio-sip-bridge.mjs +288 -0
- package/lib/voice/voice.test.mjs +990 -0
- package/mcp/README.md +80 -0
- package/package.json +151 -0
- package/plugins/maestro-skills/plugin.json +139 -0
- package/plugins/maestro-skills/skills/agents-observe.md +110 -0
- package/plugins/maestro-skills/skills/board-deck.md +68 -0
- package/plugins/maestro-skills/skills/books-close.md +77 -0
- package/plugins/maestro-skills/skills/brand-steward.md +121 -0
- package/plugins/maestro-skills/skills/calendar-plan.md +57 -0
- package/plugins/maestro-skills/skills/call-working-sessions.md +124 -0
- package/plugins/maestro-skills/skills/ccxray-diagnostics.md +91 -0
- package/plugins/maestro-skills/skills/claude-pace.md +61 -0
- package/plugins/maestro-skills/skills/code-review-graph.md +99 -0
- package/plugins/maestro-skills/skills/crm-pipeline.md +65 -0
- package/plugins/maestro-skills/skills/decision-brief.md +89 -0
- package/plugins/maestro-skills/skills/directory-hygiene.md +125 -0
- package/plugins/maestro-skills/skills/draft-comms.md +84 -0
- package/plugins/maestro-skills/skills/evening-wrap.md +53 -0
- package/plugins/maestro-skills/skills/files-find.md +65 -0
- package/plugins/maestro-skills/skills/generative-ui.md +228 -0
- package/plugins/maestro-skills/skills/hiring-triage.md +74 -0
- package/plugins/maestro-skills/skills/inbox-triage.md +61 -0
- package/plugins/maestro-skills/skills/mail-triage.md +86 -0
- package/plugins/maestro-skills/skills/morning-brief.md +54 -0
- package/plugins/maestro-skills/skills/native-artifacts.md +157 -0
- package/plugins/maestro-skills/skills/org-board.md +133 -0
- package/plugins/maestro-skills/skills/org-credential.md +68 -0
- package/plugins/maestro-skills/skills/org-recall.md +81 -0
- package/plugins/maestro-skills/skills/pipeline-review.md +76 -0
- package/plugins/maestro-skills/skills/regulatory-status.md +81 -0
- package/plugins/maestro-skills/skills/router-why.md +78 -0
- package/plugins/maestro-skills/skills/schedule-meeting.md +91 -0
- package/plugins/maestro-skills/skills/session-search.md +71 -0
- package/plugins/maestro-skills/skills/set-reminder.md +93 -0
- package/plugins/maestro-skills/skills/slack-followup.md +64 -0
- package/plugins/maestro-skills/skills/team-activity.md +86 -0
- package/plugins/maestro-skills/skills/weekly-memo.md +70 -0
- package/policies/action-classification.yaml +114 -0
- package/policies/ai-disclosure.yaml +294 -0
- package/policies/communication-style.md +139 -0
- package/policies/information-barriers.yaml +118 -0
- package/policies/prompt-injection-defence.yaml +138 -0
- package/public/assets/icon-dark.png +0 -0
- package/public/assets/icon-dark.svg +9 -0
- package/public/assets/icon-light.svg +9 -0
- package/public/assets/logo-dark.svg +15 -0
- package/public/assets/logo-light.svg +15 -0
- package/scaffold/.mcp.json +7 -0
- package/scaffold/CLAUDE.md +368 -0
- package/scaffold/config/agent.json +55 -0
- package/scaffold/config/agent.ts +76 -0
- package/scaffold/config/agent.ts.example +89 -0
- package/scaffold/config/alerts.yaml +23 -0
- package/scaffold/config/allowlist.yaml.example +25 -0
- package/scaffold/config/caller-id-map.yaml +46 -0
- package/scaffold/config/collective.yaml +49 -0
- package/scaffold/config/company.json +20 -0
- package/scaffold/config/known-agents.json +6 -0
- package/scaffold/config/learning.yaml +55 -0
- package/scaffold/config/model-routing.yaml.example +104 -0
- package/scaffold/config/org.yaml +25 -0
- package/scaffold/config/orgmail.yaml.example +19 -0
- package/scaffold/config/recovery.yaml +72 -0
- package/scaffold/config/secrets.yaml +27 -0
- package/scaffold/config/slack.yaml.example +35 -0
- package/scaffold/config/telegram.yaml.example +38 -0
- package/scaffold/config/voice.yaml.example +89 -0
- package/scaffold/config/whatsapp.yaml.example +39 -0
- package/schedules/README.md +49 -0
- package/schedules/triggers/backlog-executor.md +102 -0
- package/schedules/triggers/brand-steward.md +72 -0
- package/schedules/triggers/daily-evening-wrap.md +159 -0
- package/schedules/triggers/daily-midday-sweep.md +58 -0
- package/schedules/triggers/daily-morning-brief.md +55 -0
- package/schedules/triggers/directory-hygiene.md +81 -0
- package/schedules/triggers/dynamic-jobs.md +40 -0
- package/schedules/triggers/inbox-processor.md +115 -0
- package/schedules/triggers/meeting-action-capture.md +60 -0
- package/schedules/triggers/meeting-prep.md +69 -0
- package/schedules/triggers/messaging-inbound.md +50 -0
- package/schedules/triggers/org-pulse.md +24 -0
- package/schedules/triggers/quarterly-self-assessment.md +54 -0
- package/schedules/triggers/weekly-engineering-health.md +37 -0
- package/schedules/triggers/weekly-execution.md +65 -0
- package/schedules/triggers/weekly-hiring.md +53 -0
- package/schedules/triggers/weekly-priorities.md +38 -0
- package/schedules/triggers/weekly-strategic-memo.md +124 -0
- package/scripts/archive-email.sh +55 -0
- package/scripts/cadence/cadence-status.mjs +36 -0
- package/scripts/cadence/enqueue-cadence-tick.mjs +174 -0
- package/scripts/cadence/enqueue-cadence-tick.test.mjs +187 -0
- package/scripts/cadence/launchd-cadence-wrapper.sh +85 -0
- package/scripts/cadence/launchd-cloud-relay-wrapper.sh +95 -0
- package/scripts/cadence/launchd-socket-mode-wrapper.sh +95 -0
- package/scripts/ci/check-docs-accuracy.mjs +493 -0
- package/scripts/ci/check-docs-accuracy.test.mjs +409 -0
- package/scripts/ci/check-exports-exist.mjs +140 -0
- package/scripts/ci/check-files-exist.mjs +107 -0
- package/scripts/ci/check-no-build-artifacts.mjs +111 -0
- package/scripts/ci/check-no-build-artifacts.test.mjs +71 -0
- package/scripts/ci/check-no-confidential.mjs +198 -0
- package/scripts/ci/check-no-conflict-markers.mjs +169 -0
- package/scripts/ci/check-no-residual-identity.mjs +163 -0
- package/scripts/ci/check-no-residual-identity.test.mjs +89 -0
- package/scripts/ci/check-tarball-fidelity.mjs +205 -0
- package/scripts/ci/check-unresolved-tokens.mjs +83 -0
- package/scripts/ci/check.mjs +109 -0
- package/scripts/ci/check.test.mjs +194 -0
- package/scripts/ci/run-coverage.mjs +82 -0
- package/scripts/ci/run-tests.mjs +71 -0
- package/scripts/cloud-relay/README.md +59 -0
- package/scripts/cloud-relay/index.mjs +233 -0
- package/scripts/cloud-relay/package.json +15 -0
- package/scripts/cloud-relay/railway.json +13 -0
- package/scripts/cloud-relay/voice/README.md +94 -0
- package/scripts/cloud-relay/voice/package-lock.json +39 -0
- package/scripts/cloud-relay/voice/package.json +16 -0
- package/scripts/cloud-relay/voice/railway.json +13 -0
- package/scripts/cloud-relay/voice/server.mjs +532 -0
- package/scripts/collective/hook-runner.mjs +211 -0
- package/scripts/collective/hook-runner.test.mjs +90 -0
- package/scripts/collective/org-pulse.mjs +72 -0
- package/scripts/collective/org-sync.mjs +61 -0
- package/scripts/collective/recall.mjs +45 -0
- package/scripts/collective/who.mjs +30 -0
- package/scripts/comms-monitor.sh +288 -0
- package/scripts/configure-whatsapp-sandbox.sh +201 -0
- package/scripts/continuous-monitor.sh +91 -0
- package/scripts/cost/fleet-digest.mjs +407 -0
- package/scripts/cost/fleet-digest.test.mjs +207 -0
- package/scripts/cost/track-claude-usage.mjs +169 -0
- package/scripts/daemon/agent-daemon.mjs +989 -0
- package/scripts/daemon/agent-daemon.test.mjs +525 -0
- package/scripts/daemon/cadence-consumer-governance.test.mjs +220 -0
- package/scripts/daemon/cadence-consumer.mjs +1080 -0
- package/scripts/daemon/cadence-consumer.test.mjs +770 -0
- package/scripts/daemon/cadence-handlers.mjs +1121 -0
- package/scripts/daemon/cadence-handlers.test.mjs +617 -0
- package/scripts/daemon/classifier.mjs +704 -0
- package/scripts/daemon/classifier.test.mjs +238 -0
- package/scripts/daemon/classify-kind.mjs +54 -0
- package/scripts/daemon/classify-kind.test.mjs +40 -0
- package/scripts/daemon/context-compiler.mjs +605 -0
- package/scripts/daemon/context-compiler.test.mjs +300 -0
- package/scripts/daemon/dispatcher-cooldown.test.mjs +122 -0
- package/scripts/daemon/dispatcher-governance.test.mjs +886 -0
- package/scripts/daemon/dispatcher.mjs +1516 -0
- package/scripts/daemon/health.mjs +72 -0
- package/scripts/daemon/inbox-deferral.mjs +210 -0
- package/scripts/daemon/inbox-deferral.test.mjs +242 -0
- package/scripts/daemon/integration.test.mjs +149 -0
- package/scripts/daemon/launchd-wrapper-generic.sh +96 -0
- package/scripts/daemon/launchd-wrapper-slack-events.sh +37 -0
- package/scripts/daemon/launchd-wrapper.sh +91 -0
- package/scripts/daemon/lib/session-router.mjs +274 -0
- package/scripts/daemon/lib/session-router.test.mjs +295 -0
- package/scripts/daemon/maestro-daemon.mjs +275 -0
- package/scripts/daemon/prompt-builder.mjs +685 -0
- package/scripts/daemon/prompt-builder.test.mjs +213 -0
- package/scripts/daemon/responder.mjs +854 -0
- package/scripts/daemon/session-lock.mjs +721 -0
- package/scripts/daemon/session-lock.test.mjs +252 -0
- package/scripts/daemon/session-outcomes.mjs +640 -0
- package/scripts/daemon/session-outcomes.test.mjs +533 -0
- package/scripts/daemon/typing-registry.mjs +90 -0
- package/scripts/daemon/typing-registry.test.mjs +77 -0
- package/scripts/daemon/voice-webhook-server.mjs +804 -0
- package/scripts/decisions/capture-decision.mjs +116 -0
- package/scripts/disclosure_assessment.py +873 -0
- package/scripts/disclosure_boundaries.py +562 -0
- package/scripts/email-signature-principal.html +52 -0
- package/scripts/email-signature.html +60 -0
- package/scripts/email_quote_thread.py +167 -0
- package/scripts/email_thread_dedup.py +362 -0
- package/scripts/emergency-stop.sh +81 -0
- package/scripts/healthcheck.sh +116 -0
- package/scripts/hooks/block-mcp-cohort-send.sh +15 -0
- package/scripts/hooks/block-mcp-slack-send.sh +7 -0
- package/scripts/hooks/post-action-log.sh +126 -0
- package/scripts/hooks/pre-send-audit.sh +174 -0
- package/scripts/hooks/pre-send-audit.test.mjs +215 -0
- package/scripts/hooks/session-end-log.sh +27 -0
- package/scripts/hooks/session-start-banner.sh +115 -0
- package/scripts/huddle/audio-bridge.mjs +664 -0
- package/scripts/huddle/boot-slack-cdp.sh +102 -0
- package/scripts/huddle/huddle-controller.mjs +942 -0
- package/scripts/huddle/huddle-server.mjs +1229 -0
- package/scripts/huddle/launch-slack.sh +232 -0
- package/scripts/huddle/openai-realtime-bridge.mjs +462 -0
- package/scripts/huddle/package-lock.json +62 -0
- package/scripts/huddle/package.json +22 -0
- package/scripts/huddle/setup-audio.sh +239 -0
- package/scripts/huddle/start-call.mjs +318 -0
- package/scripts/huddle/test-pipeline.mjs +263 -0
- package/scripts/learning/consolidate-skills.mjs +72 -0
- package/scripts/learning/session-search.mjs +125 -0
- package/scripts/llm_email_dedup.py +442 -0
- package/scripts/local-triggers/generate-plists.sh +432 -0
- package/scripts/local-triggers/generate-plists.test.mjs +413 -0
- package/scripts/local-triggers/install-all.sh +49 -0
- package/scripts/local-triggers/plists/.gitkeep +0 -0
- package/scripts/local-triggers/run-trigger.sh +63 -0
- package/scripts/local-triggers/templates/rag-reindex.plist.template +47 -0
- package/scripts/local-triggers/templates/voice-relay-poller.plist.template +54 -0
- package/scripts/local-triggers/templates/voice-tunnel.plist.template +55 -0
- package/scripts/local-triggers/templates/voice-webhook.plist.template +51 -0
- package/scripts/maintenance/backup-to-cloud.sh +124 -0
- package/scripts/maintenance/health-check.sh +377 -0
- package/scripts/media-generation/README.md +105 -0
- package/scripts/media-generation/gemini-image-client.mjs +173 -0
- package/scripts/media-generation/generate-assets.mjs +289 -0
- package/scripts/media-generation/veo-video-client.mjs +219 -0
- package/scripts/org/send-orgmail.mjs +227 -0
- package/scripts/outbound-dedup-cleanup.sh +43 -0
- package/scripts/outbound-dedup.sh +477 -0
- package/scripts/outbound_dedup.py +115 -0
- package/scripts/parse-voice-transcript.mjs +481 -0
- package/scripts/pdf-generation/README.md +63 -0
- package/scripts/pdf-generation/build-document.mjs +247 -0
- package/scripts/pdf-generation/templates/board-pack.latex +136 -0
- package/scripts/pdf-generation/templates/corporate-letter.latex +126 -0
- package/scripts/pdf-generation/templates/memo.latex +114 -0
- package/scripts/poll-slack-events.sh +35 -0
- package/scripts/poller/calendar-poller.mjs +12 -0
- package/scripts/poller/gmail-poller.mjs +192 -0
- package/scripts/poller/imap-client.mjs +289 -0
- package/scripts/poller/inbox-scan-poller.mjs +156 -0
- package/scripts/poller/inbox-scan-poller.test.mjs +231 -0
- package/scripts/poller/index.mjs +73 -0
- package/scripts/poller/intra-session-check.mjs +285 -0
- package/scripts/poller/lib/cloud-relay-dedup.mjs +88 -0
- package/scripts/poller/lib/cloud-relay-dedup.test.mjs +133 -0
- package/scripts/poller/lib/slash-command-handlers.mjs +177 -0
- package/scripts/poller/secondary-gmail-poller.mjs +132 -0
- package/scripts/poller/slack-cloud-relay-client.mjs +368 -0
- package/scripts/poller/slack-poller.mjs +854 -0
- package/scripts/poller/slack-socket-mode.mjs +917 -0
- package/scripts/poller/slack-socket-mode.test.mjs +753 -0
- package/scripts/poller/trigger.mjs +75 -0
- package/scripts/poller/utils.mjs +371 -0
- package/scripts/poller/voice-cloud-relay-client.mjs +179 -0
- package/scripts/poller/voice-poller.mjs +236 -0
- package/scripts/poller-launchd/install.sh +66 -0
- package/scripts/poller-launchd/poller.plist.template +40 -0
- package/scripts/poller-launchd/whatsapp-handler.plist.template +39 -0
- package/scripts/post-interaction-indexer.py +1598 -0
- package/scripts/pre-draft-context.py +994 -0
- package/scripts/pre_draft_lookup.py +258 -0
- package/scripts/rag/build-index.mjs +47 -0
- package/scripts/rag/ingest.mjs +111 -0
- package/scripts/rag/search.mjs +119 -0
- package/scripts/rag-indexer.py +629 -0
- package/scripts/restore-from-backup.sh +248 -0
- package/scripts/restore-from-backup.test.mjs +178 -0
- package/scripts/resume-operations.sh +80 -0
- package/scripts/search-secondary-inbox.py +181 -0
- package/scripts/secondary-inbox-poller.py +437 -0
- package/scripts/self-optimization/compute-metrics.py +398 -0
- package/scripts/send-email-as-principal.py +369 -0
- package/scripts/send-email-threaded.py +392 -0
- package/scripts/send-email-with-attachment.py +377 -0
- package/scripts/send-email.sh +131 -0
- package/scripts/send-sms.sh +175 -0
- package/scripts/send-whatsapp.sh +292 -0
- package/scripts/session-start.sh +106 -0
- package/scripts/setup/boot-claude-session.sh +94 -0
- package/scripts/setup/configure-macos.sh +674 -0
- package/scripts/setup/configure-twilio-sip-trunk.mjs +207 -0
- package/scripts/setup/configure-voice-tunnel.mjs +182 -0
- package/scripts/setup/generate-agent-env.mjs +92 -0
- package/scripts/setup/generate-agent-package-json.mjs +222 -0
- package/scripts/setup/generate-agent-package-json.test.mjs +143 -0
- package/scripts/setup/generate-autonomy.mjs +60 -0
- package/scripts/setup/generate-backlog.mjs +101 -0
- package/scripts/setup/generate-cadences.mjs +92 -0
- package/scripts/setup/generate-capability.mjs +162 -0
- package/scripts/setup/generate-charter.mjs +76 -0
- package/scripts/setup/generate-comms.mjs +59 -0
- package/scripts/setup/generate-company.mjs +92 -0
- package/scripts/setup/init-agent.sh +547 -0
- package/scripts/setup/init-agent.test.mjs +151 -0
- package/scripts/setup/init-archetype.mjs +74 -0
- package/scripts/setup/init-backup.mjs +54 -0
- package/scripts/setup/init-cadence-bus.mjs +60 -0
- package/scripts/setup/init-channel-bus.mjs +46 -0
- package/scripts/setup/init-cost-tracking.mjs +45 -0
- package/scripts/setup/init-decision-capture.mjs +66 -0
- package/scripts/setup/init-known-agents.mjs +57 -0
- package/scripts/setup/init-learning.mjs +70 -0
- package/scripts/setup/init-memory-executive.mjs +45 -0
- package/scripts/setup/init-model-router.mjs +124 -0
- package/scripts/setup/init-rag.mjs +174 -0
- package/scripts/setup/init-session-router.mjs +38 -0
- package/scripts/setup/init-slack-socket-mode.mjs +260 -0
- package/scripts/setup/init-telegram.mjs +165 -0
- package/scripts/setup/init-voice-realtime.mjs +204 -0
- package/scripts/setup/init-whatsapp-baileys.mjs +77 -0
- package/scripts/setup/install-dev-tools.sh +150 -0
- package/scripts/setup/lib/install-plist.mjs +95 -0
- package/scripts/setup/migrate-agent-to-sot.mjs +192 -0
- package/scripts/setup/render-environment-yaml.mjs +133 -0
- package/scripts/slack-events-ctl.sh +177 -0
- package/scripts/slack-events-server.mjs +1045 -0
- package/scripts/slack-react.mjs +89 -0
- package/scripts/slack-responded.sh +232 -0
- package/scripts/slack-send.sh +287 -0
- package/scripts/slack-typing.mjs +196 -0
- package/scripts/slack-upload-v2.py +95 -0
- package/scripts/sms-handler.mjs +450 -0
- package/scripts/spawn-session.sh +120 -0
- package/scripts/sync-protocol.mjs +217 -0
- package/scripts/system-verify.sh +184 -0
- package/scripts/test-email-thread-dedup.py +239 -0
- package/scripts/test-information-barriers.py +484 -0
- package/scripts/test-llm-email-dedup.py +251 -0
- package/scripts/test-pre-draft-integration.py +203 -0
- package/scripts/test-rag-phase2.sh +442 -0
- package/scripts/test-rag-search.sh +251 -0
- package/scripts/test-voice-parser.mjs +316 -0
- package/scripts/user-context-search.py +659 -0
- package/scripts/validate_outbound.py +1504 -0
- package/scripts/watchdog/ai.maestro.memory-watchdog.plist +41 -0
- package/scripts/watchdog/force-reboot.sh +157 -0
- package/scripts/watchdog/memory-watchdog.sh +473 -0
- package/scripts/whatsapp-handler.mjs +538 -0
- package/teams/desktop-operations.yaml +34 -0
- package/teams/executive-office.yaml +27 -0
- package/teams/legal-and-regulatory.yaml +24 -0
- package/teams/platform-and-engineering.yaml +23 -0
- package/teams/strategy-and-growth.yaml +29 -0
- package/workflows/continuous/backlog-executor.yaml +141 -0
- package/workflows/continuous/inbound-monitor.yaml +168 -0
- package/workflows/daily/applicant-triage.yaml +197 -0
- package/workflows/daily/comms-triage.yaml +80 -0
- package/workflows/daily/evening-wrap.yaml +105 -0
- package/workflows/daily/morning-brief.yaml +164 -0
- package/workflows/daily/slack-followup-sweep.yaml +87 -0
- package/workflows/event-driven/README.md +50 -0
- package/workflows/event-driven/agent-failure-investigation.yaml +137 -0
- package/workflows/event-driven/pr-review.yaml +107 -0
- package/workflows/monthly/board-readiness.yaml +76 -0
- package/workflows/quarterly/strategic-scenario-analysis.yaml +85 -0
- package/workflows/session-protocol.md +171 -0
- package/workflows/weekly/engineering-health.yaml +154 -0
- package/workflows/weekly/hiring-review.yaml +169 -0
- package/workflows/weekly/rollup-pipeline-review.yaml +76 -0
- package/workflows/weekly/strategic-memo.yaml +79 -0
|
@@ -0,0 +1,580 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* auth-profiles.test.mjs — the OpenClaw auth-profile rotation store.
|
|
3
|
+
*
|
|
4
|
+
* Verifies: loadAuthProfiles pools keys from config + env lists; nextCredential
|
|
5
|
+
* round-robins over a multi-key pool; markCredentialCoolingDown cools ONE key
|
|
6
|
+
* (the others still serve); a single-key config is unchanged (no rotation, no
|
|
7
|
+
* ledger write); available() counts non-cooling keys; the O_EXCL ledger survives
|
|
8
|
+
* concurrent marks with monotonic strikes (longer cooldown wins).
|
|
9
|
+
*
|
|
10
|
+
* Hermetic: injected clock + a per-test temp stateDir; no real network/clock.
|
|
11
|
+
* Run: node --test lib/model-router/auth-profiles.test.mjs
|
|
12
|
+
*/
|
|
13
|
+
"use strict";
|
|
14
|
+
|
|
15
|
+
import { test } from "node:test";
|
|
16
|
+
import assert from "node:assert/strict";
|
|
17
|
+
import { mkdtempSync, rmSync, existsSync, readFileSync } from "node:fs";
|
|
18
|
+
import { tmpdir } from "node:os";
|
|
19
|
+
import { join } from "node:path";
|
|
20
|
+
|
|
21
|
+
import {
|
|
22
|
+
loadAuthProfiles,
|
|
23
|
+
nextCredential,
|
|
24
|
+
markCredentialCoolingDown,
|
|
25
|
+
markCredentialHealthy,
|
|
26
|
+
available,
|
|
27
|
+
fingerprint,
|
|
28
|
+
authEnvMapFromCatalog,
|
|
29
|
+
_internals,
|
|
30
|
+
} from "./auth-profiles.mjs";
|
|
31
|
+
import { loadCatalog } from "./catalog.mjs";
|
|
32
|
+
import { resolveChain, _internals as resolveInternals } from "./resolve.mjs";
|
|
33
|
+
import { spawnRouted, _internals as spawnInternals } from "./spawn.mjs";
|
|
34
|
+
|
|
35
|
+
function tmpStateDir() {
|
|
36
|
+
return mkdtempSync(join(tmpdir(), "authprof-"));
|
|
37
|
+
}
|
|
38
|
+
|
|
39
|
+
function clockFrom(start) {
|
|
40
|
+
let t = start;
|
|
41
|
+
const now = () => t;
|
|
42
|
+
now.advance = (ms) => { t += ms; };
|
|
43
|
+
now.set = (v) => { t = v; };
|
|
44
|
+
return now;
|
|
45
|
+
}
|
|
46
|
+
|
|
47
|
+
// ---------------------------------------------------------------------------
|
|
48
|
+
// loadAuthProfiles
|
|
49
|
+
// ---------------------------------------------------------------------------
|
|
50
|
+
|
|
51
|
+
test("loadAuthProfiles pools keys from a config doc (keys + env lists)", () => {
|
|
52
|
+
const env = { DEEPSEEK_API_KEY: "sk-env-1", DEEPSEEK_API_KEY_2: "sk-env-2" };
|
|
53
|
+
const configDoc = {
|
|
54
|
+
providers: {
|
|
55
|
+
deepseek: { keys: ["sk-cfg-a", "sk-cfg-b"], env: ["DEEPSEEK_API_KEY"] },
|
|
56
|
+
moonshot: ["sk-moon"],
|
|
57
|
+
},
|
|
58
|
+
};
|
|
59
|
+
const ap = loadAuthProfiles(null, { env, configDoc });
|
|
60
|
+
// config keys first (sk-cfg-a, sk-cfg-b), then the env var named in `env`, then
|
|
61
|
+
// env-list siblings are NOT pulled here (no authEnvByProvider passed).
|
|
62
|
+
assert.deepEqual(ap.keysFor("deepseek"), ["sk-cfg-a", "sk-cfg-b", "sk-env-1"]);
|
|
63
|
+
assert.deepEqual(ap.keysFor("moonshot"), ["sk-moon"]);
|
|
64
|
+
assert.equal(ap.isPooled("deepseek"), true);
|
|
65
|
+
assert.equal(ap.isPooled("moonshot"), false);
|
|
66
|
+
});
|
|
67
|
+
|
|
68
|
+
test("loadAuthProfiles pools numbered env-var siblings via authEnvByProvider", () => {
|
|
69
|
+
const env = {
|
|
70
|
+
ANTHROPIC_API_KEY: "sk-1",
|
|
71
|
+
ANTHROPIC_API_KEY_2: "sk-2",
|
|
72
|
+
ANTHROPIC_API_KEY_3: "sk-3",
|
|
73
|
+
// gap at _4 stops the scan; _5 must be ignored
|
|
74
|
+
ANTHROPIC_API_KEY_5: "sk-5",
|
|
75
|
+
};
|
|
76
|
+
const ap = loadAuthProfiles(null, {
|
|
77
|
+
env,
|
|
78
|
+
configDoc: null,
|
|
79
|
+
authEnvByProvider: { anthropic: "ANTHROPIC_API_KEY" },
|
|
80
|
+
});
|
|
81
|
+
assert.deepEqual(ap.keysFor("anthropic"), ["sk-1", "sk-2", "sk-3"]);
|
|
82
|
+
});
|
|
83
|
+
|
|
84
|
+
test("loadAuthProfiles supports the <BASE>S comma-list env form", () => {
|
|
85
|
+
const env = { DEEPSEEK_API_KEYS: "k1, k2 ,k3" };
|
|
86
|
+
const ap = loadAuthProfiles(null, {
|
|
87
|
+
env,
|
|
88
|
+
configDoc: null,
|
|
89
|
+
authEnvByProvider: { deepseek: "DEEPSEEK_API_KEY" },
|
|
90
|
+
});
|
|
91
|
+
assert.deepEqual(ap.keysFor("deepseek"), ["k1", "k2", "k3"]);
|
|
92
|
+
});
|
|
93
|
+
|
|
94
|
+
test("loadAuthProfiles is empty + never throws with no config and no env", () => {
|
|
95
|
+
const ap = loadAuthProfiles(null, { env: {}, configDoc: null });
|
|
96
|
+
assert.deepEqual(ap.providers, []);
|
|
97
|
+
assert.deepEqual(ap.keysFor("deepseek"), []);
|
|
98
|
+
assert.equal(ap.isPooled("deepseek"), false);
|
|
99
|
+
});
|
|
100
|
+
|
|
101
|
+
test("authEnvMapFromCatalog reads provider->auth_env from a catalog", () => {
|
|
102
|
+
const catalog = {
|
|
103
|
+
providers: new Map([
|
|
104
|
+
["anthropic", { auth_env: null }],
|
|
105
|
+
["deepseek", { auth_env: "DEEPSEEK_API_KEY" }],
|
|
106
|
+
]),
|
|
107
|
+
models: [{ provider: "moonshot", auth_env: "MOONSHOT_API_KEY" }],
|
|
108
|
+
};
|
|
109
|
+
const m = authEnvMapFromCatalog(catalog);
|
|
110
|
+
assert.equal(m.get("deepseek"), "DEEPSEEK_API_KEY");
|
|
111
|
+
assert.equal(m.get("moonshot"), "MOONSHOT_API_KEY");
|
|
112
|
+
assert.equal(m.has("anthropic"), false); // null auth_env dropped
|
|
113
|
+
});
|
|
114
|
+
|
|
115
|
+
// ---------------------------------------------------------------------------
|
|
116
|
+
// nextCredential — round-robin
|
|
117
|
+
// ---------------------------------------------------------------------------
|
|
118
|
+
|
|
119
|
+
test("nextCredential round-robins over a multi-key pool", () => {
|
|
120
|
+
const dir = tmpStateDir();
|
|
121
|
+
try {
|
|
122
|
+
const keys = ["sk-a", "sk-b", "sk-c"];
|
|
123
|
+
const now = clockFrom(1_000_000);
|
|
124
|
+
const deps = { stateDir: dir, now };
|
|
125
|
+
const got = [];
|
|
126
|
+
for (let i = 0; i < 6; i++) {
|
|
127
|
+
got.push(nextCredential("deepseek", { keys, deps }).key);
|
|
128
|
+
}
|
|
129
|
+
// Cursor starts at 0, advances by one each pick → a, b, c, a, b, c.
|
|
130
|
+
assert.deepEqual(got, ["sk-a", "sk-b", "sk-c", "sk-a", "sk-b", "sk-c"]);
|
|
131
|
+
} finally {
|
|
132
|
+
rmSync(dir, { recursive: true, force: true });
|
|
133
|
+
}
|
|
134
|
+
});
|
|
135
|
+
|
|
136
|
+
test("nextCredential is unchanged for a single-key config (no ledger write)", () => {
|
|
137
|
+
const dir = tmpStateDir();
|
|
138
|
+
try {
|
|
139
|
+
const now = clockFrom(2_000_000);
|
|
140
|
+
const deps = { stateDir: dir, now };
|
|
141
|
+
const r1 = nextCredential("moonshot", { keys: ["sk-only"], deps });
|
|
142
|
+
const r2 = nextCredential("moonshot", { keys: ["sk-only"], deps });
|
|
143
|
+
assert.equal(r1.key, "sk-only");
|
|
144
|
+
assert.equal(r2.key, "sk-only");
|
|
145
|
+
assert.equal(r1.poolSize, 1);
|
|
146
|
+
// Single-key path takes no lock + writes no ledger file.
|
|
147
|
+
assert.equal(existsSync(_internals.ledgerPath(deps)), false);
|
|
148
|
+
} finally {
|
|
149
|
+
rmSync(dir, { recursive: true, force: true });
|
|
150
|
+
}
|
|
151
|
+
});
|
|
152
|
+
|
|
153
|
+
test("nextCredential returns null for an empty pool", () => {
|
|
154
|
+
const dir = tmpStateDir();
|
|
155
|
+
try {
|
|
156
|
+
const r = nextCredential("nobody", { keys: [], deps: { stateDir: dir } });
|
|
157
|
+
assert.equal(r.key, null);
|
|
158
|
+
assert.equal(r.poolSize, 0);
|
|
159
|
+
} finally {
|
|
160
|
+
rmSync(dir, { recursive: true, force: true });
|
|
161
|
+
}
|
|
162
|
+
});
|
|
163
|
+
|
|
164
|
+
// ---------------------------------------------------------------------------
|
|
165
|
+
// markCredentialCoolingDown — cools ONE key, others serve
|
|
166
|
+
// ---------------------------------------------------------------------------
|
|
167
|
+
|
|
168
|
+
test("cooling one key removes it from rotation; the others still serve", () => {
|
|
169
|
+
const dir = tmpStateDir();
|
|
170
|
+
try {
|
|
171
|
+
const keys = ["sk-a", "sk-b", "sk-c"];
|
|
172
|
+
const now = clockFrom(5_000_000);
|
|
173
|
+
const deps = { stateDir: dir, now };
|
|
174
|
+
|
|
175
|
+
// Cool sk-b for the default ladder (30s). It must be skipped.
|
|
176
|
+
const mark = markCredentialCoolingDown("deepseek", "sk-b", "rate_limit", null, { deps, now: now() });
|
|
177
|
+
assert.equal(mark.fingerprint, fingerprint("sk-b"));
|
|
178
|
+
assert.equal(mark.strikes, 1);
|
|
179
|
+
assert.equal(mark.coolUntil, now() + 30_000);
|
|
180
|
+
|
|
181
|
+
// available() should report 2 of 3.
|
|
182
|
+
const av = available("deepseek", { keys, deps, now: now() });
|
|
183
|
+
assert.equal(av.count, 2);
|
|
184
|
+
assert.equal(av.poolSize, 3);
|
|
185
|
+
|
|
186
|
+
// nextCredential must never hand out sk-b while it cools.
|
|
187
|
+
const handed = new Set();
|
|
188
|
+
for (let i = 0; i < 6; i++) handed.add(nextCredential("deepseek", { keys, deps, now: now() }).key);
|
|
189
|
+
assert.equal(handed.has("sk-b"), false);
|
|
190
|
+
assert.equal(handed.has("sk-a"), true);
|
|
191
|
+
assert.equal(handed.has("sk-c"), true);
|
|
192
|
+
|
|
193
|
+
// After the cooldown expires, sk-b is back in rotation.
|
|
194
|
+
now.advance(30_001);
|
|
195
|
+
const av2 = available("deepseek", { keys, deps, now: now() });
|
|
196
|
+
assert.equal(av2.count, 3);
|
|
197
|
+
const handed2 = new Set();
|
|
198
|
+
for (let i = 0; i < 6; i++) handed2.add(nextCredential("deepseek", { keys, deps, now: now() }).key);
|
|
199
|
+
assert.equal(handed2.has("sk-b"), true);
|
|
200
|
+
} finally {
|
|
201
|
+
rmSync(dir, { recursive: true, force: true });
|
|
202
|
+
}
|
|
203
|
+
});
|
|
204
|
+
|
|
205
|
+
test("when every key cools, nextCredential reports allCooling + a wake time", () => {
|
|
206
|
+
const dir = tmpStateDir();
|
|
207
|
+
try {
|
|
208
|
+
const keys = ["sk-a", "sk-b"];
|
|
209
|
+
const now = clockFrom(7_000_000);
|
|
210
|
+
const deps = { stateDir: dir, now };
|
|
211
|
+
markCredentialCoolingDown("deepseek", "sk-a", "rate_limit", null, { deps, now: now() });
|
|
212
|
+
markCredentialCoolingDown("deepseek", "sk-b", "auth", null, { deps, now: now() });
|
|
213
|
+
const r = nextCredential("deepseek", { keys, deps, now: now() });
|
|
214
|
+
assert.equal(r.key, null);
|
|
215
|
+
assert.equal(r.allCooling, true);
|
|
216
|
+
assert.equal(r.available, 0);
|
|
217
|
+
assert.ok(r.soonestCoolUntil > now());
|
|
218
|
+
} finally {
|
|
219
|
+
rmSync(dir, { recursive: true, force: true });
|
|
220
|
+
}
|
|
221
|
+
});
|
|
222
|
+
|
|
223
|
+
test("the cooldown ladder steps 30s → 1m → 5m on repeated strikes", () => {
|
|
224
|
+
const dir = tmpStateDir();
|
|
225
|
+
try {
|
|
226
|
+
const now = clockFrom(9_000_000);
|
|
227
|
+
const deps = { stateDir: dir, now };
|
|
228
|
+
const m1 = markCredentialCoolingDown("deepseek", "sk-x", "rate_limit", null, { deps, now: now() });
|
|
229
|
+
const m2 = markCredentialCoolingDown("deepseek", "sk-x", "rate_limit", null, { deps, now: now() });
|
|
230
|
+
const m3 = markCredentialCoolingDown("deepseek", "sk-x", "rate_limit", null, { deps, now: now() });
|
|
231
|
+
const m4 = markCredentialCoolingDown("deepseek", "sk-x", "rate_limit", null, { deps, now: now() });
|
|
232
|
+
assert.equal(m1.coolUntil - now(), 30_000);
|
|
233
|
+
assert.equal(m2.coolUntil - now(), 60_000);
|
|
234
|
+
assert.equal(m3.coolUntil - now(), 300_000);
|
|
235
|
+
assert.equal(m4.coolUntil - now(), 300_000); // sticks at 5m
|
|
236
|
+
assert.deepEqual([m1.strikes, m2.strikes, m3.strikes, m4.strikes], [1, 2, 3, 4]);
|
|
237
|
+
} finally {
|
|
238
|
+
rmSync(dir, { recursive: true, force: true });
|
|
239
|
+
}
|
|
240
|
+
});
|
|
241
|
+
|
|
242
|
+
test("markCredentialHealthy clears a cooled key", () => {
|
|
243
|
+
const dir = tmpStateDir();
|
|
244
|
+
try {
|
|
245
|
+
const now = clockFrom(11_000_000);
|
|
246
|
+
const deps = { stateDir: dir, now };
|
|
247
|
+
const keys = ["sk-a", "sk-b"];
|
|
248
|
+
markCredentialCoolingDown("deepseek", "sk-a", "rate_limit", null, { deps, now: now() });
|
|
249
|
+
assert.equal(available("deepseek", { keys, deps, now: now() }).count, 1);
|
|
250
|
+
markCredentialHealthy("deepseek", "sk-a", { deps });
|
|
251
|
+
assert.equal(available("deepseek", { keys, deps, now: now() }).count, 2);
|
|
252
|
+
} finally {
|
|
253
|
+
rmSync(dir, { recursive: true, force: true });
|
|
254
|
+
}
|
|
255
|
+
});
|
|
256
|
+
|
|
257
|
+
// ---------------------------------------------------------------------------
|
|
258
|
+
// O_EXCL: concurrent marks converge (monotonic strikes, longer cooldown wins)
|
|
259
|
+
// ---------------------------------------------------------------------------
|
|
260
|
+
|
|
261
|
+
test("concurrent marks on one key converge under the O_EXCL lock", async () => {
|
|
262
|
+
const dir = tmpStateDir();
|
|
263
|
+
try {
|
|
264
|
+
const now = clockFrom(13_000_000);
|
|
265
|
+
const deps = { stateDir: dir, now };
|
|
266
|
+
// Fire 8 concurrent cooldown marks on the SAME key.
|
|
267
|
+
await Promise.all(
|
|
268
|
+
Array.from({ length: 8 }, () =>
|
|
269
|
+
Promise.resolve().then(() =>
|
|
270
|
+
markCredentialCoolingDown("deepseek", "sk-hot", "rate_limit", null, { deps, now: now() })
|
|
271
|
+
)
|
|
272
|
+
)
|
|
273
|
+
);
|
|
274
|
+
// The ledger must show exactly 8 strikes (monotonic — no lost updates).
|
|
275
|
+
const led = _internals.readLedgerSafe(deps);
|
|
276
|
+
const entry = _internals.providerEntry(led, "deepseek");
|
|
277
|
+
const cs = _internals.credState(entry, fingerprint("sk-hot"));
|
|
278
|
+
assert.equal(cs.strikes, 8);
|
|
279
|
+
// The persisted file is valid JSON (atomic writes, no torn file).
|
|
280
|
+
const raw = readFileSync(_internals.ledgerPath(deps), "utf-8");
|
|
281
|
+
assert.doesNotThrow(() => JSON.parse(raw));
|
|
282
|
+
} finally {
|
|
283
|
+
rmSync(dir, { recursive: true, force: true });
|
|
284
|
+
}
|
|
285
|
+
});
|
|
286
|
+
|
|
287
|
+
test("a racing shorter cooldown never shortens a longer one", () => {
|
|
288
|
+
const dir = tmpStateDir();
|
|
289
|
+
try {
|
|
290
|
+
const now = clockFrom(15_000_000);
|
|
291
|
+
const deps = { stateDir: dir, now };
|
|
292
|
+
// Explicit long cooldown first, then a short ladder hit — coolUntil must not regress.
|
|
293
|
+
const long = markCredentialCoolingDown("deepseek", "sk-y", "billing", 600_000, { deps, now: now() });
|
|
294
|
+
const short = markCredentialCoolingDown("deepseek", "sk-y", "rate_limit", null, { deps, now: now() });
|
|
295
|
+
assert.equal(long.coolUntil, now() + 600_000);
|
|
296
|
+
assert.equal(short.coolUntil, now() + 600_000); // not shortened to ~1m
|
|
297
|
+
} finally {
|
|
298
|
+
rmSync(dir, { recursive: true, force: true });
|
|
299
|
+
}
|
|
300
|
+
});
|
|
301
|
+
|
|
302
|
+
test("fingerprint is deterministic and never the plaintext", () => {
|
|
303
|
+
const fp = fingerprint("super-secret-key");
|
|
304
|
+
assert.equal(fp, fingerprint("super-secret-key"));
|
|
305
|
+
assert.notEqual(fp, "super-secret-key");
|
|
306
|
+
assert.match(fp, /^[0-9a-f]{16}$/);
|
|
307
|
+
assert.equal(fingerprint(""), "");
|
|
308
|
+
});
|
|
309
|
+
|
|
310
|
+
test("the cooldown ledger never stores plaintext credentials", () => {
|
|
311
|
+
const dir = tmpStateDir();
|
|
312
|
+
try {
|
|
313
|
+
const now = clockFrom(17_000_000);
|
|
314
|
+
const deps = { stateDir: dir, now };
|
|
315
|
+
markCredentialCoolingDown("deepseek", "sk-PLAINTEXT-SECRET", "rate_limit", null, { deps, now: now() });
|
|
316
|
+
const raw = readFileSync(_internals.ledgerPath(deps), "utf-8");
|
|
317
|
+
assert.equal(raw.includes("sk-PLAINTEXT-SECRET"), false);
|
|
318
|
+
assert.equal(raw.includes(fingerprint("sk-PLAINTEXT-SECRET")), true);
|
|
319
|
+
} finally {
|
|
320
|
+
rmSync(dir, { recursive: true, force: true });
|
|
321
|
+
}
|
|
322
|
+
});
|
|
323
|
+
|
|
324
|
+
// ---------------------------------------------------------------------------
|
|
325
|
+
// Wiring: resolve.mjs picks a pooled credential + skips a cooled one;
|
|
326
|
+
// a single-key config is unchanged. spawn.mjs cools the credential on failover.
|
|
327
|
+
// ---------------------------------------------------------------------------
|
|
328
|
+
|
|
329
|
+
// A catalog with a session-capable DeepSeek row (grade A so a tool-use session
|
|
330
|
+
// can pick it) reachable on the anthropic wire (endpoints.anthropic).
|
|
331
|
+
function pooledCatalog() {
|
|
332
|
+
const catalogText = {
|
|
333
|
+
deepseek: [
|
|
334
|
+
"provider: deepseek",
|
|
335
|
+
"auth_env: DEEPSEEK_API_KEY",
|
|
336
|
+
"endpoints: { anthropic: https://api.deepseek.com/anthropic }",
|
|
337
|
+
"models:",
|
|
338
|
+
" - id: deepseek-chat",
|
|
339
|
+
" status: available",
|
|
340
|
+
" tool_reliability: A",
|
|
341
|
+
" context_window: 64000",
|
|
342
|
+
" context_tokens: 60000",
|
|
343
|
+
" harness: { session: true, direct: true }",
|
|
344
|
+
" cost: { input: 0.14, output: 0.28 }",
|
|
345
|
+
].join("\n"),
|
|
346
|
+
anthropic: [
|
|
347
|
+
"provider: anthropic",
|
|
348
|
+
"endpoints: {}",
|
|
349
|
+
"models:",
|
|
350
|
+
" - id: claude-sonnet-4-6",
|
|
351
|
+
" status: available",
|
|
352
|
+
" tool_reliability: A",
|
|
353
|
+
" context_window: 200000",
|
|
354
|
+
" context_tokens: 180000",
|
|
355
|
+
" harness: { session: true, direct: true }",
|
|
356
|
+
" cost: { input: 3, output: 15 }",
|
|
357
|
+
].join("\n"),
|
|
358
|
+
};
|
|
359
|
+
return loadCatalog(undefined, { catalogText, agentRoot: tmpdir() });
|
|
360
|
+
}
|
|
361
|
+
|
|
362
|
+
// A v2 routing config: deepseek-first chain, anthropic fallback, deepseek allowed
|
|
363
|
+
// for public data so the data-class gate passes.
|
|
364
|
+
function pooledConfig() {
|
|
365
|
+
return {
|
|
366
|
+
schema_version: 2,
|
|
367
|
+
defaults: { data_class: "public" },
|
|
368
|
+
backends: { deepseek: { allowed_data_classes: ["public", "internal", "sensitive"] } },
|
|
369
|
+
aliases: {
|
|
370
|
+
cheap: "deepseek/deepseek-chat",
|
|
371
|
+
default: "anthropic/claude-sonnet-4-6",
|
|
372
|
+
},
|
|
373
|
+
routing_policy: [
|
|
374
|
+
{ match: { task_class: "lookup" }, harness: "session", needs_tool_use: false, chain: ["cheap", "default"] },
|
|
375
|
+
{ default: true, chain: ["default"] },
|
|
376
|
+
],
|
|
377
|
+
fallback_to_anthropic: true,
|
|
378
|
+
};
|
|
379
|
+
}
|
|
380
|
+
|
|
381
|
+
test("resolve picks a pooled credential into the session-retarget env", () => {
|
|
382
|
+
const dir = tmpStateDir();
|
|
383
|
+
try {
|
|
384
|
+
const catalog = pooledCatalog();
|
|
385
|
+
const profiles = loadAuthProfiles(null, {
|
|
386
|
+
env: {},
|
|
387
|
+
configDoc: { providers: { deepseek: { keys: ["sk-a", "sk-b", "sk-c"] } } },
|
|
388
|
+
});
|
|
389
|
+
const now = clockFrom(20_000_000);
|
|
390
|
+
const dec = resolveChain(
|
|
391
|
+
{ task_class: "lookup", data_class: "public", needs_tool_use: false },
|
|
392
|
+
{
|
|
393
|
+
config: pooledConfig(),
|
|
394
|
+
catalog,
|
|
395
|
+
env: {}, // NO single DEEPSEEK_API_KEY — the pool supplies the credential
|
|
396
|
+
authProfiles: profiles,
|
|
397
|
+
authProfileDeps: { stateDir: dir, now },
|
|
398
|
+
now, // resolveChain wants a clock FUNCTION (so ctx.now matches the ledger clock)
|
|
399
|
+
}
|
|
400
|
+
);
|
|
401
|
+
assert.equal(dec.chosen.provider, "deepseek");
|
|
402
|
+
// The retarget env carries a pooled key (not undefined despite empty env).
|
|
403
|
+
assert.equal(dec.envForSpawn.ANTHROPIC_BASE_URL, "https://api.deepseek.com/anthropic");
|
|
404
|
+
assert.ok(["sk-a", "sk-b", "sk-c"].includes(dec.envForSpawn.ANTHROPIC_AUTH_TOKEN));
|
|
405
|
+
} finally {
|
|
406
|
+
rmSync(dir, { recursive: true, force: true });
|
|
407
|
+
}
|
|
408
|
+
});
|
|
409
|
+
|
|
410
|
+
test("resolve skips a backend whose entire credential pool is cooled", () => {
|
|
411
|
+
const dir = tmpStateDir();
|
|
412
|
+
try {
|
|
413
|
+
const catalog = pooledCatalog();
|
|
414
|
+
const profiles = loadAuthProfiles(null, {
|
|
415
|
+
env: {},
|
|
416
|
+
configDoc: { providers: { deepseek: { keys: ["sk-a", "sk-b"] } } },
|
|
417
|
+
});
|
|
418
|
+
const now = clockFrom(22_000_000);
|
|
419
|
+
const deps = { stateDir: dir, now };
|
|
420
|
+
// Cool BOTH deepseek keys → the pool is dry; resolve must fall to anthropic.
|
|
421
|
+
markCredentialCoolingDown("deepseek", "sk-a", "rate_limit", null, { deps, now: now() });
|
|
422
|
+
markCredentialCoolingDown("deepseek", "sk-b", "rate_limit", null, { deps, now: now() });
|
|
423
|
+
|
|
424
|
+
const dec = resolveChain(
|
|
425
|
+
{ task_class: "lookup", data_class: "public", needs_tool_use: false },
|
|
426
|
+
{
|
|
427
|
+
config: pooledConfig(),
|
|
428
|
+
catalog,
|
|
429
|
+
env: {},
|
|
430
|
+
authProfiles: profiles,
|
|
431
|
+
authProfileDeps: deps,
|
|
432
|
+
now, // clock FUNCTION
|
|
433
|
+
}
|
|
434
|
+
);
|
|
435
|
+
assert.equal(dec.chosen.provider, "anthropic");
|
|
436
|
+
// deepseek was tried and skipped for missing_credential (pool exhausted).
|
|
437
|
+
assert.ok(dec.tried.some((t) => t.ref === "deepseek/deepseek-chat" && t.reason === "missing_credential"));
|
|
438
|
+
} finally {
|
|
439
|
+
rmSync(dir, { recursive: true, force: true });
|
|
440
|
+
}
|
|
441
|
+
});
|
|
442
|
+
|
|
443
|
+
test("a single-key config is unchanged (no pool, no rotation, env var wins)", () => {
|
|
444
|
+
const dir = tmpStateDir();
|
|
445
|
+
try {
|
|
446
|
+
const catalog = pooledCatalog();
|
|
447
|
+
// Single key in the pool ⇒ NOT pooled ⇒ resolve falls through to env[auth_env]
|
|
448
|
+
// exactly as before. We give a single-key profile AND the env var; both point
|
|
449
|
+
// at the same single-key behavior.
|
|
450
|
+
const profiles = loadAuthProfiles(null, {
|
|
451
|
+
env: {},
|
|
452
|
+
configDoc: { providers: { deepseek: { keys: ["sk-solo"] } } },
|
|
453
|
+
});
|
|
454
|
+
const now = clockFrom(24_000_000);
|
|
455
|
+
const dec = resolveChain(
|
|
456
|
+
{ task_class: "lookup", data_class: "public", needs_tool_use: false },
|
|
457
|
+
{
|
|
458
|
+
config: pooledConfig(),
|
|
459
|
+
catalog,
|
|
460
|
+
env: { DEEPSEEK_API_KEY: "sk-env-solo" },
|
|
461
|
+
authProfiles: profiles,
|
|
462
|
+
authProfileDeps: { stateDir: dir, now },
|
|
463
|
+
now, // clock FUNCTION
|
|
464
|
+
}
|
|
465
|
+
);
|
|
466
|
+
assert.equal(dec.chosen.provider, "deepseek");
|
|
467
|
+
// Single-key pool is not "pooled", so the env var supplies the token verbatim.
|
|
468
|
+
assert.equal(dec.envForSpawn.ANTHROPIC_AUTH_TOKEN, "sk-env-solo");
|
|
469
|
+
// No ledger file written for the single-key path.
|
|
470
|
+
assert.equal(existsSync(_internals.ledgerPath({ stateDir: dir })), false);
|
|
471
|
+
} finally {
|
|
472
|
+
rmSync(dir, { recursive: true, force: true });
|
|
473
|
+
}
|
|
474
|
+
});
|
|
475
|
+
|
|
476
|
+
test("resolve with NO authProfiles wired behaves exactly as today (env var only)", () => {
|
|
477
|
+
const catalog = pooledCatalog();
|
|
478
|
+
const now = clockFrom(26_000_000);
|
|
479
|
+
const dec = resolveChain(
|
|
480
|
+
{ task_class: "lookup", data_class: "public", needs_tool_use: false },
|
|
481
|
+
{
|
|
482
|
+
config: pooledConfig(),
|
|
483
|
+
catalog,
|
|
484
|
+
env: { DEEPSEEK_API_KEY: "sk-classic" },
|
|
485
|
+
now, // clock FUNCTION
|
|
486
|
+
// no authProfiles, no authProfileDeps
|
|
487
|
+
}
|
|
488
|
+
);
|
|
489
|
+
assert.equal(dec.chosen.provider, "deepseek");
|
|
490
|
+
assert.equal(dec.envForSpawn.ANTHROPIC_AUTH_TOKEN, "sk-classic");
|
|
491
|
+
});
|
|
492
|
+
|
|
493
|
+
test("resolveRowCredential helper: pooled vs single-key", () => {
|
|
494
|
+
const dir = tmpStateDir();
|
|
495
|
+
try {
|
|
496
|
+
const now = clockFrom(28_000_000);
|
|
497
|
+
const profiles = loadAuthProfiles(null, {
|
|
498
|
+
env: {},
|
|
499
|
+
configDoc: { providers: { deepseek: { keys: ["sk-a", "sk-b"] } } },
|
|
500
|
+
});
|
|
501
|
+
const row = { provider: "deepseek", auth_env: "DEEPSEEK_API_KEY" };
|
|
502
|
+
const pooledCtx = { env: {}, authProfiles: profiles, authProfileDeps: { stateDir: dir, now }, now: now() };
|
|
503
|
+
const p = resolveInternals.resolveRowCredential(row, pooledCtx);
|
|
504
|
+
assert.equal(p.pooled, true);
|
|
505
|
+
assert.ok(["sk-a", "sk-b"].includes(p.token));
|
|
506
|
+
|
|
507
|
+
// Single env var, no pool ⇒ not pooled, env token verbatim.
|
|
508
|
+
const single = resolveInternals.resolveRowCredential(row, {
|
|
509
|
+
env: { DEEPSEEK_API_KEY: "sk-env" },
|
|
510
|
+
authProfiles: null,
|
|
511
|
+
now: now(),
|
|
512
|
+
});
|
|
513
|
+
assert.equal(single.pooled, false);
|
|
514
|
+
assert.equal(single.token, "sk-env");
|
|
515
|
+
} finally {
|
|
516
|
+
rmSync(dir, { recursive: true, force: true });
|
|
517
|
+
}
|
|
518
|
+
});
|
|
519
|
+
|
|
520
|
+
test("spawnRouted cools the active credential on a rate_limit failover", async () => {
|
|
521
|
+
const dir = tmpStateDir();
|
|
522
|
+
try {
|
|
523
|
+
const catalog = pooledCatalog();
|
|
524
|
+
const profiles = loadAuthProfiles(null, {
|
|
525
|
+
env: {},
|
|
526
|
+
configDoc: { providers: { deepseek: { keys: ["sk-a", "sk-b"] } } },
|
|
527
|
+
});
|
|
528
|
+
const now = clockFrom(30_000_000);
|
|
529
|
+
const deps = { stateDir: dir, now };
|
|
530
|
+
|
|
531
|
+
// First attempt (deepseek) rate-limits; the failover should advance to
|
|
532
|
+
// anthropic AND cool the deepseek credential the first attempt used.
|
|
533
|
+
let calls = 0;
|
|
534
|
+
const result = await spawnRouted(
|
|
535
|
+
{ task_class: "lookup", data_class: "public", needs_tool_use: false },
|
|
536
|
+
{
|
|
537
|
+
config: pooledConfig(),
|
|
538
|
+
catalog,
|
|
539
|
+
env: {},
|
|
540
|
+
authProfiles: profiles,
|
|
541
|
+
authProfileDeps: deps,
|
|
542
|
+
now,
|
|
543
|
+
// a no-op breaker so failover walks deterministically
|
|
544
|
+
breaker: { isOpen: () => ({ open: false }), open: () => {}, close: () => {} },
|
|
545
|
+
spawnImpl: async (candidate) => {
|
|
546
|
+
calls++;
|
|
547
|
+
if (candidate.ref === "deepseek/deepseek-chat") {
|
|
548
|
+
return { ok: false, reason: "rate_limit" };
|
|
549
|
+
}
|
|
550
|
+
return { ok: true, result: "ok" };
|
|
551
|
+
},
|
|
552
|
+
}
|
|
553
|
+
);
|
|
554
|
+
|
|
555
|
+
assert.equal(result.ok, true);
|
|
556
|
+
assert.equal(result.chosen.provider, "anthropic"); // failed over
|
|
557
|
+
// Exactly one deepseek credential is now cooling (the one the attempt used).
|
|
558
|
+
const av = available("deepseek", { profiles, deps, now: now() });
|
|
559
|
+
assert.equal(av.count, 1, "one of the two pooled keys should be cooling");
|
|
560
|
+
assert.equal(av.poolSize, 2);
|
|
561
|
+
} finally {
|
|
562
|
+
rmSync(dir, { recursive: true, force: true });
|
|
563
|
+
}
|
|
564
|
+
});
|
|
565
|
+
|
|
566
|
+
test("buildChildEnv never leaks the internal __credFingerprint marker", () => {
|
|
567
|
+
// The retarget set may carry a private __credFingerprint; only RETARGET_KEYS
|
|
568
|
+
// reach the child env, so the marker is always scrubbed.
|
|
569
|
+
const child = spawnInternals.buildChildEnv(
|
|
570
|
+
{ PATH: "/usr/bin" },
|
|
571
|
+
{
|
|
572
|
+
ANTHROPIC_BASE_URL: "https://api.deepseek.com/anthropic",
|
|
573
|
+
ANTHROPIC_AUTH_TOKEN: "sk-a",
|
|
574
|
+
__credFingerprint: "deadbeefdeadbeef",
|
|
575
|
+
}
|
|
576
|
+
);
|
|
577
|
+
assert.equal(child.ANTHROPIC_AUTH_TOKEN, "sk-a");
|
|
578
|
+
assert.equal(child.__credFingerprint, undefined);
|
|
579
|
+
assert.equal(child.PATH, "/usr/bin");
|
|
580
|
+
});
|