takos-control 1.0.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/package.json +169 -0
- package/src/__tests__/db-runtime-contracts.test.ts +57 -0
- package/src/adapters/analytics-engine-binding.ts +104 -0
- package/src/adapters/dynamo-kv-store.ts +284 -0
- package/src/adapters/firestore-kv-store.ts +287 -0
- package/src/adapters/gcs-object-store.ts +506 -0
- package/src/adapters/openai-binding.ts +83 -0
- package/src/adapters/pgvector-store.ts +236 -0
- package/src/adapters/pubsub-queue.ts +127 -0
- package/src/adapters/r2-compat-types.ts +54 -0
- package/src/adapters/s3-object-store.ts +539 -0
- package/src/adapters/sqs-queue.ts +133 -0
- package/src/adapters/workflow-binding.ts +131 -0
- package/src/application/services/actions/actions-env.ts +49 -0
- package/src/application/services/actions/actions-execution.ts +156 -0
- package/src/application/services/actions/actions-triggers.ts +224 -0
- package/src/application/services/actions/index.ts +18 -0
- package/src/application/services/activitypub/remote-install.ts +170 -0
- package/src/application/services/activitypub/remote-store-client.ts +435 -0
- package/src/application/services/activitypub/store-registry.ts +341 -0
- package/src/application/services/activitypub/store-subscription.ts +240 -0
- package/src/application/services/activitypub/stores.ts +389 -0
- package/src/application/services/agent/agent-models.ts +55 -0
- package/src/application/services/agent/delegation.ts +328 -0
- package/src/application/services/agent/execute-run.ts +46 -0
- package/src/application/services/agent/index.ts +46 -0
- package/src/application/services/agent/langgraph-agent.ts +282 -0
- package/src/application/services/agent/langgraph-checkpointer.ts +556 -0
- package/src/application/services/agent/langgraph-graph.ts +357 -0
- package/src/application/services/agent/langgraph-runner.ts +326 -0
- package/src/application/services/agent/langgraph-tools.ts +157 -0
- package/src/application/services/agent/llm-manager.ts +51 -0
- package/src/application/services/agent/llm.ts +106 -0
- package/src/application/services/agent/memory-manager.ts +84 -0
- package/src/application/services/agent/message-persistence.ts +151 -0
- package/src/application/services/agent/message-utils.ts +167 -0
- package/src/application/services/agent/model-catalog.ts +71 -0
- package/src/application/services/agent/official-skills.ts +437 -0
- package/src/application/services/agent/prompt-assets.generated.ts +23 -0
- package/src/application/services/agent/prompt-budget.ts +140 -0
- package/src/application/services/agent/prompt-builder.ts +72 -0
- package/src/application/services/agent/prompts/core.md +9 -0
- package/src/application/services/agent/prompts/general-workflow.md +6 -0
- package/src/application/services/agent/prompts/modes/assistant.md +4 -0
- package/src/application/services/agent/prompts/modes/default.md +8 -0
- package/src/application/services/agent/prompts/modes/implementer.md +5 -0
- package/src/application/services/agent/prompts/modes/planner.md +4 -0
- package/src/application/services/agent/prompts/modes/researcher.md +5 -0
- package/src/application/services/agent/prompts/modes/reviewer.md +4 -0
- package/src/application/services/agent/prompts/response-guidelines.md +20 -0
- package/src/application/services/agent/prompts/runtime-tool-catalog-empty.md +3 -0
- package/src/application/services/agent/prompts/runtime-tool-catalog-header.md +3 -0
- package/src/application/services/agent/prompts/runtime-tool-catalog-selective.md +8 -0
- package/src/application/services/agent/prompts/skills/planning-structurer.en.md +1 -0
- package/src/application/services/agent/prompts/skills/planning-structurer.ja.md +1 -0
- package/src/application/services/agent/prompts/skills/repo-app-operator.en.md +1 -0
- package/src/application/services/agent/prompts/skills/repo-app-operator.ja.md +1 -0
- package/src/application/services/agent/prompts/skills/research-brief.en.md +1 -0
- package/src/application/services/agent/prompts/skills/research-brief.ja.md +1 -0
- package/src/application/services/agent/prompts/skills/slides-author.en.md +1 -0
- package/src/application/services/agent/prompts/skills/slides-author.ja.md +1 -0
- package/src/application/services/agent/prompts/skills/writing-draft.en.md +1 -0
- package/src/application/services/agent/prompts/skills/writing-draft.ja.md +1 -0
- package/src/application/services/agent/prompts/tool-runtime-rules.md +6 -0
- package/src/application/services/agent/providers/llm-providers.ts +484 -0
- package/src/application/services/agent/remote-tool-executor.ts +78 -0
- package/src/application/services/agent/run-lifecycle.ts +59 -0
- package/src/application/services/agent/runner-config.ts +77 -0
- package/src/application/services/agent/runner-events.ts +169 -0
- package/src/application/services/agent/runner-history.ts +368 -0
- package/src/application/services/agent/runner-types.ts +73 -0
- package/src/application/services/agent/runner.ts +682 -0
- package/src/application/services/agent/security/injection-detector.ts +145 -0
- package/src/application/services/agent/session-closer.ts +438 -0
- package/src/application/services/agent/simple-loop.ts +367 -0
- package/src/application/services/agent/skill-contracts.ts +21 -0
- package/src/application/services/agent/skill-loader.ts +319 -0
- package/src/application/services/agent/skill-plan.ts +67 -0
- package/src/application/services/agent/skill-resolution.ts +328 -0
- package/src/application/services/agent/skill-scoring.ts +261 -0
- package/src/application/services/agent/skill-templates.ts +51 -0
- package/src/application/services/agent/skills.ts +54 -0
- package/src/application/services/agent/thread-context.ts +472 -0
- package/src/application/services/agent/workflow-pr.ts +150 -0
- package/src/application/services/agent/workflow-review.ts +116 -0
- package/src/application/services/agent/workflow-session.ts +156 -0
- package/src/application/services/agent/workflow-types.ts +133 -0
- package/src/application/services/agent/workflow.ts +264 -0
- package/src/application/services/billing/billing-accounts.ts +165 -0
- package/src/application/services/billing/billing-plans.ts +252 -0
- package/src/application/services/billing/billing-run-usage.ts +89 -0
- package/src/application/services/billing/billing-types.ts +94 -0
- package/src/application/services/billing/billing-usage.ts +370 -0
- package/src/application/services/billing/billing.ts +64 -0
- package/src/application/services/billing/stripe.ts +331 -0
- package/src/application/services/cloudflare/api-client.ts +195 -0
- package/src/application/services/cloudflare/resources.ts +134 -0
- package/src/application/services/common-env/audit.ts +65 -0
- package/src/application/services/common-env/crypto.ts +184 -0
- package/src/application/services/common-env/db-helpers.ts +16 -0
- package/src/application/services/common-env/index.ts +4 -0
- package/src/application/services/common-env/link-state.ts +70 -0
- package/src/application/services/common-env/maintenance.ts +36 -0
- package/src/application/services/common-env/manual-link-ops.ts +385 -0
- package/src/application/services/common-env/orchestrator.ts +146 -0
- package/src/application/services/common-env/reconcile-jobs.ts +529 -0
- package/src/application/services/common-env/reconciler.ts +103 -0
- package/src/application/services/common-env/repository.ts +185 -0
- package/src/application/services/common-env/service-link-ops.ts +144 -0
- package/src/application/services/common-env/service.ts +355 -0
- package/src/application/services/common-env/space-env-ops.ts +245 -0
- package/src/application/services/common-env/takos-builtins.ts +386 -0
- package/src/application/services/deployment/artifacts.ts +105 -0
- package/src/application/services/deployment/deployment-artifacts.ts +143 -0
- package/src/application/services/deployment/execute.ts +364 -0
- package/src/application/services/deployment/group-deploy-manifest.ts +71 -0
- package/src/application/services/deployment/group-deploy-types.ts +116 -0
- package/src/application/services/deployment/group-deploy.ts +349 -0
- package/src/application/services/deployment/index.ts +7 -0
- package/src/application/services/deployment/models.ts +154 -0
- package/src/application/services/deployment/provider.ts +358 -0
- package/src/application/services/deployment/resource-provisioner.ts +175 -0
- package/src/application/services/deployment/rollback-orchestrator.ts +260 -0
- package/src/application/services/deployment/rollback.ts +110 -0
- package/src/application/services/deployment/routing.ts +311 -0
- package/src/application/services/deployment/service.ts +325 -0
- package/src/application/services/deployment/state.ts +85 -0
- package/src/application/services/deployment/store.ts +386 -0
- package/src/application/services/deployment/wrangler-config-gen.ts +182 -0
- package/src/application/services/execution/embeddings.ts +457 -0
- package/src/application/services/execution/run-creation.ts +138 -0
- package/src/application/services/execution/run-events.ts +127 -0
- package/src/application/services/execution/runtime-request-handler.ts +173 -0
- package/src/application/services/execution/sql-validation.ts +202 -0
- package/src/application/services/execution/workflow-engine-converters.ts +75 -0
- package/src/application/services/execution/workflow-engine-types.ts +77 -0
- package/src/application/services/execution/workflow-engine.ts +104 -0
- package/src/application/services/execution/workflow-job-scheduler.ts +344 -0
- package/src/application/services/execution/workflow-run-lifecycle.ts +389 -0
- package/src/application/services/execution/workflow-storage.ts +83 -0
- package/src/application/services/git-smart/client/fetch-pack.ts +167 -0
- package/src/application/services/git-smart/client/fetch-refs.ts +134 -0
- package/src/application/services/git-smart/client/index.ts +10 -0
- package/src/application/services/git-smart/core/commit-index.ts +436 -0
- package/src/application/services/git-smart/core/merge.ts +102 -0
- package/src/application/services/git-smart/core/object-store.ts +235 -0
- package/src/application/services/git-smart/core/object.ts +238 -0
- package/src/application/services/git-smart/core/readable-commit.ts +59 -0
- package/src/application/services/git-smart/core/refs.ts +364 -0
- package/src/application/services/git-smart/core/sha1.ts +40 -0
- package/src/application/services/git-smart/core/tree-ops.ts +228 -0
- package/src/application/services/git-smart/git-objects.ts +122 -0
- package/src/application/services/git-smart/index.ts +135 -0
- package/src/application/services/git-smart/operations.ts +154 -0
- package/src/application/services/git-smart/protocol/capabilities.ts +28 -0
- package/src/application/services/git-smart/protocol/packfile-reader.ts +368 -0
- package/src/application/services/git-smart/protocol/packfile-writer.ts +129 -0
- package/src/application/services/git-smart/protocol/pkt-line.ts +97 -0
- package/src/application/services/git-smart/smart-http/info-refs.ts +67 -0
- package/src/application/services/git-smart/smart-http/receive-pack.ts +485 -0
- package/src/application/services/git-smart/smart-http/upload-pack.ts +72 -0
- package/src/application/services/identity/auth-utils.ts +432 -0
- package/src/application/services/identity/locale.ts +19 -0
- package/src/application/services/identity/membership-resolver.ts +28 -0
- package/src/application/services/identity/principals.ts +76 -0
- package/src/application/services/identity/profile-activity.ts +248 -0
- package/src/application/services/identity/response-formatters.ts +88 -0
- package/src/application/services/identity/session.ts +221 -0
- package/src/application/services/identity/shortcut-groups.ts +421 -0
- package/src/application/services/identity/shortcuts.ts +216 -0
- package/src/application/services/identity/space-access.ts +128 -0
- package/src/application/services/identity/space-crud.ts +494 -0
- package/src/application/services/identity/space-members.ts +177 -0
- package/src/application/services/identity/space-models.ts +50 -0
- package/src/application/services/identity/spaces.ts +29 -0
- package/src/application/services/identity/takos-access-tokens.ts +149 -0
- package/src/application/services/identity/user-cache.ts +66 -0
- package/src/application/services/identity/user-settings.ts +146 -0
- package/src/application/services/maintenance/backup-maintenance.ts +453 -0
- package/src/application/services/maintenance/custom-domain-maintenance.ts +415 -0
- package/src/application/services/maintenance/index.ts +31 -0
- package/src/application/services/maintenance/resource-orphan-gc.ts +66 -0
- package/src/application/services/maintenance/session-maintenance.ts +50 -0
- package/src/application/services/maintenance/snapshot-maintenance.ts +147 -0
- package/src/application/services/memory/consolidation.ts +365 -0
- package/src/application/services/memory/extractor.ts +320 -0
- package/src/application/services/memory/index.ts +17 -0
- package/src/application/services/memory/llm-parser.ts +44 -0
- package/src/application/services/memory/memories.ts +392 -0
- package/src/application/services/memory-graph/activation.ts +60 -0
- package/src/application/services/memory-graph/claim-store.ts +306 -0
- package/src/application/services/memory-graph/graph-models.ts +119 -0
- package/src/application/services/memory-graph/memory-graph-runtime.ts +220 -0
- package/src/application/services/memory-graph/observer.ts +158 -0
- package/src/application/services/memory-graph/overlay.ts +101 -0
- package/src/application/services/notifications/notification-models.ts +58 -0
- package/src/application/services/notifications/service.ts +522 -0
- package/src/application/services/oauth/audit.ts +50 -0
- package/src/application/services/oauth/authorization.ts +286 -0
- package/src/application/services/oauth/client.ts +346 -0
- package/src/application/services/oauth/consent.ts +244 -0
- package/src/application/services/oauth/device.ts +295 -0
- package/src/application/services/oauth/pkce.ts +60 -0
- package/src/application/services/oauth/scopes.ts +61 -0
- package/src/application/services/oauth/token.ts +555 -0
- package/src/application/services/offload/index.ts +31 -0
- package/src/application/services/offload/messages.ts +67 -0
- package/src/application/services/offload/run-events.ts +128 -0
- package/src/application/services/offload/usage-client.ts +39 -0
- package/src/application/services/offload/usage-events.ts +100 -0
- package/src/application/services/platform/app-deployments.ts +397 -0
- package/src/application/services/platform/capabilities.ts +288 -0
- package/src/application/services/platform/custom-domains/access.ts +93 -0
- package/src/application/services/platform/custom-domains/cloudflare.ts +64 -0
- package/src/application/services/platform/custom-domains/dns.ts +55 -0
- package/src/application/services/platform/custom-domains/domain-crud.ts +230 -0
- package/src/application/services/platform/custom-domains/domain-models.ts +71 -0
- package/src/application/services/platform/custom-domains/domain-verification.ts +345 -0
- package/src/application/services/platform/custom-domains.ts +27 -0
- package/src/application/services/platform/desired-state-types.ts +111 -0
- package/src/application/services/platform/env-state-resolution.ts +249 -0
- package/src/application/services/platform/infra.ts +246 -0
- package/src/application/services/platform/mcp/crud.ts +343 -0
- package/src/application/services/platform/mcp/crypto.ts +63 -0
- package/src/application/services/platform/mcp/mcp-models.ts +119 -0
- package/src/application/services/platform/mcp/oauth.ts +375 -0
- package/src/application/services/platform/mcp/validation.ts +69 -0
- package/src/application/services/platform/mcp.ts +51 -0
- package/src/application/services/platform/resource-bindings.ts +172 -0
- package/src/application/services/platform/rollout-health.ts +48 -0
- package/src/application/services/platform/rollout.ts +358 -0
- package/src/application/services/platform/runtime-config.ts +155 -0
- package/src/application/services/platform/ui-extensions.ts +190 -0
- package/src/application/services/platform/worker-desired-state.ts +397 -0
- package/src/application/services/platform/workers.ts +426 -0
- package/src/application/services/platform/workflow-artifacts.ts +204 -0
- package/src/application/services/pull-requests/ai-review.ts +362 -0
- package/src/application/services/pull-requests/event-tasks.ts +67 -0
- package/src/application/services/pull-requests/index.ts +25 -0
- package/src/application/services/pull-requests/merge-resolution.ts +501 -0
- package/src/application/services/r2/orphaned-object-gc.ts +268 -0
- package/src/application/services/resources/access.ts +180 -0
- package/src/application/services/resources/bindings.ts +185 -0
- package/src/application/services/resources/format.ts +76 -0
- package/src/application/services/resources/index.ts +32 -0
- package/src/application/services/resources/lifecycle.ts +92 -0
- package/src/application/services/resources/store.ts +387 -0
- package/src/application/services/routing/cache.ts +176 -0
- package/src/application/services/routing/phase.ts +20 -0
- package/src/application/services/routing/resolver.ts +227 -0
- package/src/application/services/routing/routing-models.ts +46 -0
- package/src/application/services/routing/service.ts +298 -0
- package/src/application/services/routing/sharding.ts +41 -0
- package/src/application/services/run-notifier/client.ts +32 -0
- package/src/application/services/run-notifier/index.ts +20 -0
- package/src/application/services/run-notifier/run-events-contract.ts +114 -0
- package/src/application/services/run-notifier/run-failure-events.ts +62 -0
- package/src/application/services/run-notifier/run-notifier-payload.ts +28 -0
- package/src/application/services/runs/create-thread-run-store.ts +527 -0
- package/src/application/services/runs/create-thread-run-validation.ts +88 -0
- package/src/application/services/runs/run-serialization.ts +102 -0
- package/src/application/services/seed-repositories.ts +37 -0
- package/src/application/services/source/__tests__/app-manifest-template.test.ts +211 -0
- package/src/application/services/source/app-manifest-bundle.ts +427 -0
- package/src/application/services/source/app-manifest-parser.ts +375 -0
- package/src/application/services/source/app-manifest-template.ts +93 -0
- package/src/application/services/source/app-manifest-types.ts +241 -0
- package/src/application/services/source/app-manifest-validation.ts +212 -0
- package/src/application/services/source/app-manifest.ts +38 -0
- package/src/application/services/source/apps.ts +130 -0
- package/src/application/services/source/explore-catalog.ts +446 -0
- package/src/application/services/source/explore-packages.ts +612 -0
- package/src/application/services/source/explore-repos.ts +189 -0
- package/src/application/services/source/explore-types.ts +155 -0
- package/src/application/services/source/explore.ts +20 -0
- package/src/application/services/source/external-import-utils.ts +126 -0
- package/src/application/services/source/external-import.ts +430 -0
- package/src/application/services/source/fork.ts +349 -0
- package/src/application/services/source/git.ts +556 -0
- package/src/application/services/source/info-units.ts +383 -0
- package/src/application/services/source/official-packages.ts +50 -0
- package/src/application/services/source/repo-release-assets.ts +55 -0
- package/src/application/services/source/repos.ts +231 -0
- package/src/application/services/source/search.ts +247 -0
- package/src/application/services/source/skill-search.ts +315 -0
- package/src/application/services/source/skills.ts +584 -0
- package/src/application/services/source/source-exploration.ts +221 -0
- package/src/application/services/source/space-storage.ts +484 -0
- package/src/application/services/sync/git-sync-types.ts +35 -0
- package/src/application/services/sync/git-sync.ts +228 -0
- package/src/application/services/sync/index.ts +6 -0
- package/src/application/services/sync/models.ts +65 -0
- package/src/application/services/sync/runtime-session.ts +444 -0
- package/src/application/services/sync/session-files.ts +370 -0
- package/src/application/services/sync/snapshot-cleanup.ts +290 -0
- package/src/application/services/sync/snapshot-compressor.ts +79 -0
- package/src/application/services/sync/snapshot-storage.ts +136 -0
- package/src/application/services/sync/snapshot.ts +426 -0
- package/src/application/services/threads/thread-export.ts +140 -0
- package/src/application/services/threads/thread-history.ts +422 -0
- package/src/application/services/threads/thread-search.ts +299 -0
- package/src/application/services/threads/thread-service.ts +394 -0
- package/src/application/services/threads/thread-shares.ts +200 -0
- package/src/application/services/threads/thread-timeline.ts +44 -0
- package/src/application/services/wfp/assets.ts +218 -0
- package/src/application/services/wfp/bindings.ts +176 -0
- package/src/application/services/wfp/client.ts +199 -0
- package/src/application/services/wfp/d1.ts +133 -0
- package/src/application/services/wfp/index.ts +23 -0
- package/src/application/services/wfp/kv.ts +38 -0
- package/src/application/services/wfp/orchestrator.ts +339 -0
- package/src/application/services/wfp/queues.ts +77 -0
- package/src/application/services/wfp/r2.ts +131 -0
- package/src/application/services/wfp/service.ts +341 -0
- package/src/application/services/wfp/vectorize.ts +48 -0
- package/src/application/services/wfp/wfp-contracts.ts +102 -0
- package/src/application/services/wfp/worker-metadata.ts +54 -0
- package/src/application/services/wfp/workers.ts +307 -0
- package/src/application/services/workflow-runs/commands.ts +354 -0
- package/src/application/services/workflow-runs/read-model.ts +202 -0
- package/src/application/services/workflow-runs/stream.ts +54 -0
- package/src/application/tools/builtin/agent.ts +383 -0
- package/src/application/tools/builtin/artifact.ts +163 -0
- package/src/application/tools/builtin/browser/definitions.ts +167 -0
- package/src/application/tools/builtin/browser/handler-action.ts +76 -0
- package/src/application/tools/builtin/browser/handler-close.ts +31 -0
- package/src/application/tools/builtin/browser/handler-extract.ts +42 -0
- package/src/application/tools/builtin/browser/handler-goto.ts +41 -0
- package/src/application/tools/builtin/browser/handler-html.ts +38 -0
- package/src/application/tools/builtin/browser/handler-open.ts +68 -0
- package/src/application/tools/builtin/browser/handler-screenshot.ts +33 -0
- package/src/application/tools/builtin/browser/session.ts +51 -0
- package/src/application/tools/builtin/browser.ts +27 -0
- package/src/application/tools/builtin/container/availability.ts +65 -0
- package/src/application/tools/builtin/container/definitions.ts +119 -0
- package/src/application/tools/builtin/container/handler-commit.ts +182 -0
- package/src/application/tools/builtin/container/handler-create-repository.ts +54 -0
- package/src/application/tools/builtin/container/handler-start.ts +236 -0
- package/src/application/tools/builtin/container/handler-status.ts +108 -0
- package/src/application/tools/builtin/container/handler-stop.ts +64 -0
- package/src/application/tools/builtin/container/session.ts +116 -0
- package/src/application/tools/builtin/container.ts +39 -0
- package/src/application/tools/builtin/deploy.ts +65 -0
- package/src/application/tools/builtin/discovery.ts +142 -0
- package/src/application/tools/builtin/file/definitions.ts +220 -0
- package/src/application/tools/builtin/file/file-operations.ts +112 -0
- package/src/application/tools/builtin/file/handler-copy.ts +77 -0
- package/src/application/tools/builtin/file/handler-delete.ts +34 -0
- package/src/application/tools/builtin/file/handler-list.ts +26 -0
- package/src/application/tools/builtin/file/handler-mkdir.ts +25 -0
- package/src/application/tools/builtin/file/handler-read.ts +27 -0
- package/src/application/tools/builtin/file/handler-rename.ts +85 -0
- package/src/application/tools/builtin/file/handler-write-binary.ts +63 -0
- package/src/application/tools/builtin/file/handler-write.ts +41 -0
- package/src/application/tools/builtin/file/limits.ts +114 -0
- package/src/application/tools/builtin/file/session.ts +86 -0
- package/src/application/tools/builtin/file.ts +54 -0
- package/src/application/tools/builtin/index.ts +9 -0
- package/src/application/tools/builtin/info-unit.ts +403 -0
- package/src/application/tools/builtin/mcp.ts +232 -0
- package/src/application/tools/builtin/memory-graph.ts +97 -0
- package/src/application/tools/builtin/memory.ts +247 -0
- package/src/application/tools/builtin/platform/deployment-history.ts +185 -0
- package/src/application/tools/builtin/platform/deployments.ts +298 -0
- package/src/application/tools/builtin/platform/domains.ts +264 -0
- package/src/application/tools/builtin/platform/worker-settings.ts +482 -0
- package/src/application/tools/builtin/platform.ts +63 -0
- package/src/application/tools/builtin/registry.ts +133 -0
- package/src/application/tools/builtin/repo.ts +135 -0
- package/src/application/tools/builtin/runtime-tool-executor.ts +259 -0
- package/src/application/tools/builtin/space-app-deployments.ts +136 -0
- package/src/application/tools/builtin/space-common-env.ts +125 -0
- package/src/application/tools/builtin/space-files.ts +414 -0
- package/src/application/tools/builtin/space-skills.ts +482 -0
- package/src/application/tools/builtin/space-source.ts +193 -0
- package/src/application/tools/builtin/storage/d1.ts +309 -0
- package/src/application/tools/builtin/storage/kv.ts +201 -0
- package/src/application/tools/builtin/storage/r2.ts +344 -0
- package/src/application/tools/builtin/storage/resources.ts +288 -0
- package/src/application/tools/builtin/storage/validators.ts +35 -0
- package/src/application/tools/builtin/storage.ts +64 -0
- package/src/application/tools/builtin/web.ts +492 -0
- package/src/application/tools/candidate-selector.ts +131 -0
- package/src/application/tools/capabilities.ts +51 -0
- package/src/application/tools/capability-registry.ts +79 -0
- package/src/application/tools/capability-types.ts +44 -0
- package/src/application/tools/circuit-breaker.ts +149 -0
- package/src/application/tools/descriptor-builder.ts +185 -0
- package/src/application/tools/executor-setup.ts +195 -0
- package/src/application/tools/executor-utils.ts +21 -0
- package/src/application/tools/executor.ts +399 -0
- package/src/application/tools/idempotency.ts +137 -0
- package/src/application/tools/index.ts +61 -0
- package/src/application/tools/loaders/mcp-tools.ts +261 -0
- package/src/application/tools/mcp-client.ts +116 -0
- package/src/application/tools/namespace-map.ts +133 -0
- package/src/application/tools/resolver.ts +125 -0
- package/src/application/tools/tool-circuit-breaker.ts +84 -0
- package/src/application/tools/tool-definitions.ts +140 -0
- package/src/application/tools/tool-error-classifier.ts +115 -0
- package/src/application/tools/tool-permission.ts +103 -0
- package/src/application/tools/tool-policy-helpers.ts +107 -0
- package/src/application/tools/tool-policy-types.ts +98 -0
- package/src/application/tools/tool-policy.ts +579 -0
- package/src/dispatch.ts +183 -0
- package/src/index.ts +3 -0
- package/src/infra/db/client.ts +28 -0
- package/src/infra/db/index.ts +174 -0
- package/src/infra/db/schema-accounts.ts +198 -0
- package/src/infra/db/schema-agents.ts +305 -0
- package/src/infra/db/schema-auth.ts +58 -0
- package/src/infra/db/schema-billing.ts +120 -0
- package/src/infra/db/schema-oauth.ts +194 -0
- package/src/infra/db/schema-platform.ts +437 -0
- package/src/infra/db/schema-repos.ts +348 -0
- package/src/infra/db/schema-services.ts +74 -0
- package/src/infra/db/schema-workers.ts +359 -0
- package/src/infra/db/schema-workflows.ts +114 -0
- package/src/infra/db/schema.ts +172 -0
- package/src/local-platform/adapters/local.redis.test.ts +145 -0
- package/src/local-platform/bootstrap.test.ts +1374 -0
- package/src/local-platform/bootstrap.ts +1 -0
- package/src/local-platform/cloudflare-containers-shim.mjs +28 -0
- package/src/local-platform/cloudflare-workers-shim.mjs +8 -0
- package/src/local-platform/container-backend.ts +65 -0
- package/src/local-platform/d1-migrations.ts +338 -0
- package/src/local-platform/d1-prepared-statement.ts +104 -0
- package/src/local-platform/d1-shared.ts +131 -0
- package/src/local-platform/d1-sql-rewrite.ts +386 -0
- package/src/local-platform/docker-container-backend.ts +205 -0
- package/src/local-platform/execution-context.ts +24 -0
- package/src/local-platform/executor-control-rpc.ts +398 -0
- package/src/local-platform/fetch-server.ts +8 -0
- package/src/local-platform/in-memory-bindings.test.ts +49 -0
- package/src/local-platform/in-memory-bindings.ts +91 -0
- package/src/local-platform/in-memory-d1.ts +126 -0
- package/src/local-platform/in-memory-kv.ts +84 -0
- package/src/local-platform/in-memory-queue.ts +23 -0
- package/src/local-platform/in-memory-r2.ts +272 -0
- package/src/local-platform/k8s-container-backend.ts +364 -0
- package/src/local-platform/load-adapter.ts +42 -0
- package/src/local-platform/miniflare-bindings.ts +353 -0
- package/src/local-platform/miniflare-registry.ts +351 -0
- package/src/local-platform/node-fetch-server.ts +70 -0
- package/src/local-platform/node-resolve-loader.mjs +97 -0
- package/src/local-platform/oci-orchestrator-node.ts +23 -0
- package/src/local-platform/oci-orchestrator.test.ts +110 -0
- package/src/local-platform/oci-orchestrator.ts +529 -0
- package/src/local-platform/persistent-bindings.test.ts +173 -0
- package/src/local-platform/persistent-bindings.ts +6 -0
- package/src/local-platform/persistent-d1.ts +171 -0
- package/src/local-platform/persistent-durable-objects.ts +57 -0
- package/src/local-platform/persistent-kv.ts +121 -0
- package/src/local-platform/persistent-queue.ts +59 -0
- package/src/local-platform/persistent-r2.test.ts +72 -0
- package/src/local-platform/persistent-r2.ts +376 -0
- package/src/local-platform/persistent-shared.ts +27 -0
- package/src/local-platform/public-runtime-contract.test.ts +211 -0
- package/src/local-platform/queue-runtime.ts +85 -0
- package/src/local-platform/redis-bindings.ts +185 -0
- package/src/local-platform/routing-store.ts +118 -0
- package/src/local-platform/run-smoke-proxyless.ts +67 -0
- package/src/local-platform/run-smoke.ts +176 -0
- package/src/local-platform/runtime-env.ts +114 -0
- package/src/local-platform/runtime-gateway-stubs.ts +168 -0
- package/src/local-platform/runtime-host-fetch.ts +154 -0
- package/src/local-platform/runtime-http.ts +63 -0
- package/src/local-platform/runtime-types.ts +96 -0
- package/src/local-platform/runtime.ts +114 -0
- package/src/local-platform/tenant-binding-polyfills.ts +203 -0
- package/src/local-platform/tenant-binding-rpc.ts +216 -0
- package/src/local-platform/tenant-resource-limits.ts +32 -0
- package/src/local-platform/tenant-worker-runtime.ts +179 -0
- package/src/local-platform/url-registry.ts +75 -0
- package/src/local-platform/worker.test.ts +78 -0
- package/src/local-platform/worker.ts +186 -0
- package/src/node-platform/env-builder.ts +317 -0
- package/src/node-platform/index.ts +7 -0
- package/src/node-platform/resolvers/ai-resolver.ts +39 -0
- package/src/node-platform/resolvers/bucket-resolver.ts +89 -0
- package/src/node-platform/resolvers/db-resolver.ts +17 -0
- package/src/node-platform/resolvers/dispatch-resolver.ts +119 -0
- package/src/node-platform/resolvers/durable-object-resolver.ts +17 -0
- package/src/node-platform/resolvers/env-helpers.ts +47 -0
- package/src/node-platform/resolvers/kv-resolver.ts +39 -0
- package/src/node-platform/resolvers/queue-resolver.ts +99 -0
- package/src/node-platform/resolvers/routing-resolver.ts +103 -0
- package/src/platform/accessors.ts +100 -0
- package/src/platform/adapters/node.ts +236 -0
- package/src/platform/adapters/shared.ts +161 -0
- package/src/platform/adapters/workers.ts +133 -0
- package/src/platform/context.ts +26 -0
- package/src/platform/index.ts +43 -0
- package/src/platform/platform-config.ts +170 -0
- package/src/platform/providers/cloudflare/pdf-render.ts +30 -0
- package/src/platform/providers/cloudflare/resources.ts +19 -0
- package/src/platform/providers/cloudflare/wfp.ts +22 -0
- package/src/platform/providers/node/pdf-render.ts +123 -0
- package/src/runtime/container-hosts/browser-session-host.ts +359 -0
- package/src/runtime/container-hosts/browser-session-types.ts +33 -0
- package/src/runtime/container-hosts/container-runtime.ts +52 -0
- package/src/runtime/container-hosts/d1-raw.ts +22 -0
- package/src/runtime/container-hosts/executor-auth.ts +161 -0
- package/src/runtime/container-hosts/executor-control-rpc.ts +449 -0
- package/src/runtime/container-hosts/executor-dispatch.ts +84 -0
- package/src/runtime/container-hosts/executor-host.ts +447 -0
- package/src/runtime/container-hosts/executor-proxy-config.ts +38 -0
- package/src/runtime/container-hosts/executor-proxy-handlers.ts +427 -0
- package/src/runtime/container-hosts/executor-run-state.ts +389 -0
- package/src/runtime/container-hosts/executor-utils.ts +269 -0
- package/src/runtime/container-hosts/proxy-token-manager.ts +188 -0
- package/src/runtime/container-hosts/runtime-host.ts +241 -0
- package/src/runtime/durable-objects/do-header-utils.ts +160 -0
- package/src/runtime/durable-objects/git-push-lock.ts +94 -0
- package/src/runtime/durable-objects/notification-notifier.ts +257 -0
- package/src/runtime/durable-objects/rate-limiter.ts +268 -0
- package/src/runtime/durable-objects/routing.ts +339 -0
- package/src/runtime/durable-objects/run-notifier.ts +555 -0
- package/src/runtime/durable-objects/session.ts +167 -0
- package/src/runtime/executor-proxy-api.ts +211 -0
- package/src/runtime/indexer/handlers.ts +133 -0
- package/src/runtime/indexer/index.ts +103 -0
- package/src/runtime/queues/deploy-jobs.ts +89 -0
- package/src/runtime/queues/parallel-steps.ts +433 -0
- package/src/runtime/queues/workflow-dlq.ts +132 -0
- package/src/runtime/queues/workflow-events.ts +34 -0
- package/src/runtime/queues/workflow-expressions.ts +69 -0
- package/src/runtime/queues/workflow-job-handler.ts +193 -0
- package/src/runtime/queues/workflow-job-phases.ts +335 -0
- package/src/runtime/queues/workflow-jobs.ts +43 -0
- package/src/runtime/queues/workflow-runner.ts +87 -0
- package/src/runtime/queues/workflow-runtime-client.ts +180 -0
- package/src/runtime/queues/workflow-secrets.ts +78 -0
- package/src/runtime/queues/workflow-steps.ts +62 -0
- package/src/runtime/queues/workflow-types.ts +163 -0
- package/src/runtime/runner/cron-handler.ts +78 -0
- package/src/runtime/runner/index.ts +20 -0
- package/src/runtime/runner/queue-handler.ts +247 -0
- package/src/runtime/runner/runner-constants.ts +7 -0
- package/src/runtime/worker/egress.ts +375 -0
- package/src/runtime/worker/env.ts +63 -0
- package/src/runtime/worker/index.ts +11 -0
- package/src/runtime/worker/runtime-factory.ts +111 -0
- package/src/server/middleware/auth.ts +210 -0
- package/src/server/middleware/billing.ts +62 -0
- package/src/server/middleware/body-size.ts +59 -0
- package/src/server/middleware/cache.ts +192 -0
- package/src/server/middleware/content-type.ts +59 -0
- package/src/server/middleware/git-auth.ts +103 -0
- package/src/server/middleware/oauth-auth.ts +204 -0
- package/src/server/middleware/param-validation.ts +59 -0
- package/src/server/middleware/plan-gates.ts +71 -0
- package/src/server/middleware/space-access.ts +125 -0
- package/src/server/middleware/static-assets.ts +25 -0
- package/src/server/middleware/trust-tier.ts +42 -0
- package/src/server/middleware/turnstile.ts +50 -0
- package/src/server/routes/activitypub-store/activitypub-queries.ts +234 -0
- package/src/server/routes/activitypub-store/routes.ts +512 -0
- package/src/server/routes/agent-tasks-handlers.ts +211 -0
- package/src/server/routes/agent-tasks.ts +352 -0
- package/src/server/routes/api.ts +399 -0
- package/src/server/routes/app-deployments.ts +201 -0
- package/src/server/routes/apps.ts +393 -0
- package/src/server/routes/auth/cli.ts +226 -0
- package/src/server/routes/auth/external.ts +366 -0
- package/src/server/routes/auth/html.ts +398 -0
- package/src/server/routes/auth/link.ts +158 -0
- package/src/server/routes/auth/provisioning.ts +210 -0
- package/src/server/routes/auth/session.ts +277 -0
- package/src/server/routes/auth-api.ts +155 -0
- package/src/server/routes/billing/routes.ts +494 -0
- package/src/server/routes/billing/stripe.ts +128 -0
- package/src/server/routes/browser-sessions.ts +258 -0
- package/src/server/routes/common-env/handlers.ts +23 -0
- package/src/server/routes/custom-domains.ts +132 -0
- package/src/server/routes/explore/explore-filters.ts +262 -0
- package/src/server/routes/explore/index.ts +1 -0
- package/src/server/routes/explore/packages.ts +392 -0
- package/src/server/routes/explore/repos.ts +218 -0
- package/src/server/routes/explore/routes.ts +14 -0
- package/src/server/routes/explore/users.ts +139 -0
- package/src/server/routes/git.ts +218 -0
- package/src/server/routes/index/graph.ts +127 -0
- package/src/server/routes/index/handlers.ts +210 -0
- package/src/server/routes/index/index-context.ts +80 -0
- package/src/server/routes/index/index.ts +50 -0
- package/src/server/routes/index/jobs.ts +159 -0
- package/src/server/routes/mcp.ts +343 -0
- package/src/server/routes/me.ts +408 -0
- package/src/server/routes/memories.ts +366 -0
- package/src/server/routes/notifications-sse.ts +54 -0
- package/src/server/routes/notifications.ts +158 -0
- package/src/server/routes/oauth/authorize.ts +311 -0
- package/src/server/routes/oauth/device.ts +344 -0
- package/src/server/routes/oauth/introspect.ts +82 -0
- package/src/server/routes/oauth/register.ts +185 -0
- package/src/server/routes/oauth/request-utils.ts +94 -0
- package/src/server/routes/oauth/revoke.ts +69 -0
- package/src/server/routes/oauth/routes.ts +255 -0
- package/src/server/routes/oauth/token.ts +409 -0
- package/src/server/routes/oauth/userinfo.ts +101 -0
- package/src/server/routes/oauth-consent-api.ts +468 -0
- package/src/server/routes/profiles/api.ts +14 -0
- package/src/server/routes/profiles/block-follow-helpers.ts +166 -0
- package/src/server/routes/profiles/block-mute.ts +127 -0
- package/src/server/routes/profiles/dto.ts +42 -0
- package/src/server/routes/profiles/follow.ts +449 -0
- package/src/server/routes/profiles/index.ts +12 -0
- package/src/server/routes/profiles/profile-crud.ts +276 -0
- package/src/server/routes/profiles/profile-queries.ts +245 -0
- package/src/server/routes/profiles/register.ts +20 -0
- package/src/server/routes/profiles/repo.ts +382 -0
- package/src/server/routes/profiles/view.ts +213 -0
- package/src/server/routes/public-share.ts +143 -0
- package/src/server/routes/pull-requests/comments.ts +159 -0
- package/src/server/routes/pull-requests/diff.ts +309 -0
- package/src/server/routes/pull-requests/dto.ts +238 -0
- package/src/server/routes/pull-requests/git-store.ts +2 -0
- package/src/server/routes/pull-requests/index.ts +12 -0
- package/src/server/routes/pull-requests/merge-handlers.ts +266 -0
- package/src/server/routes/pull-requests/merge.ts +423 -0
- package/src/server/routes/pull-requests/read-model.ts +201 -0
- package/src/server/routes/pull-requests/reviews.ts +259 -0
- package/src/server/routes/pull-requests/routes.ts +305 -0
- package/src/server/routes/pull-requests/workflow-trigger.ts +61 -0
- package/src/server/routes/reminders.ts +173 -0
- package/src/server/routes/repos/actions/artifacts.ts +90 -0
- package/src/server/routes/repos/actions/jobs.ts +137 -0
- package/src/server/routes/repos/actions/logs.ts +74 -0
- package/src/server/routes/repos/actions/runs.ts +174 -0
- package/src/server/routes/repos/actions/secrets.ts +145 -0
- package/src/server/routes/repos/external-import.ts +158 -0
- package/src/server/routes/repos/forks.ts +196 -0
- package/src/server/routes/repos/git-advanced.ts +509 -0
- package/src/server/routes/repos/git-commits.ts +344 -0
- package/src/server/routes/repos/git-files.ts +218 -0
- package/src/server/routes/repos/git-refs.ts +203 -0
- package/src/server/routes/repos/git-shared.ts +57 -0
- package/src/server/routes/repos/git-write-operations.ts +217 -0
- package/src/server/routes/repos/git.ts +12 -0
- package/src/server/routes/repos/index.ts +30 -0
- package/src/server/routes/repos/release-assets.ts +299 -0
- package/src/server/routes/repos/release-crud.ts +419 -0
- package/src/server/routes/repos/release-shared.ts +59 -0
- package/src/server/routes/repos/releases.ts +8 -0
- package/src/server/routes/repos/repo-helpers.ts +168 -0
- package/src/server/routes/repos/routes.ts +338 -0
- package/src/server/routes/repos/shared.ts +110 -0
- package/src/server/routes/repos/stars.ts +200 -0
- package/src/server/routes/repos/sync.ts +500 -0
- package/src/server/routes/repos/workflows.ts +388 -0
- package/src/server/routes/resources/access.ts +109 -0
- package/src/server/routes/resources/bindings.ts +157 -0
- package/src/server/routes/resources/d1.ts +294 -0
- package/src/server/routes/resources/index.ts +16 -0
- package/src/server/routes/resources/r2.ts +181 -0
- package/src/server/routes/resources/routes.ts +407 -0
- package/src/server/routes/resources/tokens.ts +365 -0
- package/src/server/routes/rpc-types.ts +87 -0
- package/src/server/routes/runs/access.ts +34 -0
- package/src/server/routes/runs/create.ts +42 -0
- package/src/server/routes/runs/list.ts +120 -0
- package/src/server/routes/runs/observation.ts +161 -0
- package/src/server/routes/runs/routes.ts +274 -0
- package/src/server/routes/runs/sse.ts +62 -0
- package/src/server/routes/search.ts +220 -0
- package/src/server/routes/seed-repositories.ts +17 -0
- package/src/server/routes/sessions/auth.ts +33 -0
- package/src/server/routes/sessions/heartbeat.ts +120 -0
- package/src/server/routes/sessions/index.ts +57 -0
- package/src/server/routes/sessions/lifecycle.ts +337 -0
- package/src/server/routes/sessions/session-mappers.ts +49 -0
- package/src/server/routes/setup.ts +111 -0
- package/src/server/routes/shared/route-auth.ts +245 -0
- package/src/server/routes/shortcuts.ts +335 -0
- package/src/server/routes/skills.ts +443 -0
- package/src/server/routes/smart-http.ts +380 -0
- package/src/server/routes/spaces/common-env.ts +92 -0
- package/src/server/routes/spaces/members.ts +326 -0
- package/src/server/routes/spaces/repositories.ts +69 -0
- package/src/server/routes/spaces/routes.ts +386 -0
- package/src/server/routes/spaces/storage-downloads.ts +257 -0
- package/src/server/routes/spaces/storage-management.ts +354 -0
- package/src/server/routes/spaces/storage-operations.ts +46 -0
- package/src/server/routes/spaces/storage-uploads.ts +195 -0
- package/src/server/routes/spaces/storage.ts +10 -0
- package/src/server/routes/spaces/store-registry.ts +413 -0
- package/src/server/routes/spaces/stores.ts +139 -0
- package/src/server/routes/thread-messages.ts +141 -0
- package/src/server/routes/thread-shares.ts +113 -0
- package/src/server/routes/threads.ts +450 -0
- package/src/server/routes/well-known.ts +120 -0
- package/src/server/routes/workers/deployments.ts +339 -0
- package/src/server/routes/workers/index.ts +12 -0
- package/src/server/routes/workers/routes.ts +305 -0
- package/src/server/routes/workers/settings-bindings.ts +184 -0
- package/src/server/routes/workers/settings-common-env.ts +272 -0
- package/src/server/routes/workers/settings-config.ts +95 -0
- package/src/server/routes/workers/settings-env-vars.ts +107 -0
- package/src/server/routes/workers/settings.ts +14 -0
- package/src/server/routes/workers/slug.ts +149 -0
- package/src/server/routes/zod-validator.ts +24 -0
- package/src/shared/config/index.ts +38 -0
- package/src/shared/config/limits.ts +104 -0
- package/src/shared/config/timeouts.ts +71 -0
- package/src/shared/constants/app.ts +62 -0
- package/src/shared/constants/dns.ts +12 -0
- package/src/shared/constants/index.ts +11 -0
- package/src/shared/constants/roles.ts +45 -0
- package/src/shared/types/bindings.ts +71 -0
- package/src/shared/types/drizzle-helpers.ts +5 -0
- package/src/shared/types/env.ts +151 -0
- package/src/shared/types/index.ts +172 -0
- package/src/shared/types/models.ts +506 -0
- package/src/shared/types/oauth.ts +216 -0
- package/src/shared/types/queue-messages.ts +140 -0
- package/src/shared/types/routing.ts +54 -0
- package/src/shared/utils/content-type.ts +39 -0
- package/src/shared/utils/crypto.ts +181 -0
- package/src/shared/utils/date-utils.ts +16 -0
- package/src/shared/utils/db-guards.ts +39 -0
- package/src/shared/utils/db-transaction.ts +157 -0
- package/src/shared/utils/device-auth-rate-limit.ts +40 -0
- package/src/shared/utils/domain-validation.ts +428 -0
- package/src/shared/utils/encoding-utils.ts +77 -0
- package/src/shared/utils/error-response.ts +51 -0
- package/src/shared/utils/gzip.ts +67 -0
- package/src/shared/utils/hash.ts +36 -0
- package/src/shared/utils/http-response.ts +37 -0
- package/src/shared/utils/index.ts +65 -0
- package/src/shared/utils/lcs-diff.ts +65 -0
- package/src/shared/utils/logger.ts +220 -0
- package/src/shared/utils/naming-utils.ts +16 -0
- package/src/shared/utils/path-validation.ts +124 -0
- package/src/shared/utils/rate-limiter.ts +149 -0
- package/src/shared/utils/service-client.ts +109 -0
- package/src/shared/utils/sliding-window.ts +103 -0
- package/src/shared/utils/spa-fallback.ts +26 -0
- package/src/shared/utils/token-bucket.ts +106 -0
- package/src/shared/utils/unified-diff.ts +109 -0
- package/src/shared/utils/url-utils.ts +18 -0
- package/src/shared/utils/validate-env.ts +126 -0
- package/src/shared/utils/with-timeout.ts +32 -0
- package/src/shared/utils/zip-stream.ts +226 -0
- package/src/web.ts +505 -0
- package/src/worker-emulation/redis-durable-object.ts +246 -0
- package/src/worker-emulation/sse-notifier.ts +316 -0
|
@@ -0,0 +1,484 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Multi-Model Provider Abstraction
|
|
3
|
+
* Supports OpenAI, Anthropic Claude, and Google Gemini
|
|
4
|
+
*/
|
|
5
|
+
|
|
6
|
+
import type { AgentMessage, AgentTool, ToolCall } from '../agent-models';
|
|
7
|
+
import { logError } from '../../../../shared/utils/logger';
|
|
8
|
+
import {
|
|
9
|
+
DEFAULT_MODEL_ID,
|
|
10
|
+
getModelProvider,
|
|
11
|
+
type ModelProvider,
|
|
12
|
+
} from '../model-catalog';
|
|
13
|
+
export { DEFAULT_MODEL_ID } from '../model-catalog';
|
|
14
|
+
export type { ModelProvider } from '../model-catalog';
|
|
15
|
+
|
|
16
|
+
/** Truncate and redact LLM API error bodies to prevent API key / secret leakage. */
|
|
17
|
+
function sanitizeLlmError(body: string, maxLen = 500): string {
|
|
18
|
+
return body
|
|
19
|
+
.slice(0, maxLen)
|
|
20
|
+
.replace(/sk-[A-Za-z0-9_-]{10,}/g, 'sk-***')
|
|
21
|
+
.replace(/key-[A-Za-z0-9_-]{10,}/g, 'key-***')
|
|
22
|
+
.replace(/Bearer\s+[A-Za-z0-9_-]+/gi, 'Bearer ***');
|
|
23
|
+
}
|
|
24
|
+
|
|
25
|
+
export interface ModelConfig {
|
|
26
|
+
provider: ModelProvider;
|
|
27
|
+
model: string;
|
|
28
|
+
apiKey: string;
|
|
29
|
+
maxTokens?: number;
|
|
30
|
+
temperature?: number;
|
|
31
|
+
}
|
|
32
|
+
|
|
33
|
+
export interface LLMResponse {
|
|
34
|
+
content: string;
|
|
35
|
+
toolCalls?: ToolCall[];
|
|
36
|
+
stopReason: 'stop' | 'tool_calls' | 'length';
|
|
37
|
+
usage: {
|
|
38
|
+
inputTokens: number;
|
|
39
|
+
outputTokens: number;
|
|
40
|
+
};
|
|
41
|
+
}
|
|
42
|
+
|
|
43
|
+
export interface LLMProvider {
|
|
44
|
+
chat(messages: AgentMessage[], tools?: AgentTool[], signal?: AbortSignal): Promise<LLMResponse>;
|
|
45
|
+
}
|
|
46
|
+
|
|
47
|
+
// ============================================================================
|
|
48
|
+
// OpenAI Provider
|
|
49
|
+
// ============================================================================
|
|
50
|
+
|
|
51
|
+
interface OpenAIMessage {
|
|
52
|
+
role: 'system' | 'user' | 'assistant' | 'tool';
|
|
53
|
+
content: string | null;
|
|
54
|
+
tool_calls?: { id: string; type: 'function'; function: { name: string; arguments: string } }[];
|
|
55
|
+
tool_call_id?: string;
|
|
56
|
+
}
|
|
57
|
+
|
|
58
|
+
class OpenAIProvider implements LLMProvider {
|
|
59
|
+
constructor(private config: ModelConfig) {}
|
|
60
|
+
|
|
61
|
+
private convertTools(tools: AgentTool[]) {
|
|
62
|
+
return tools.map(tool => ({
|
|
63
|
+
type: 'function' as const,
|
|
64
|
+
function: {
|
|
65
|
+
name: tool.name,
|
|
66
|
+
description: tool.description,
|
|
67
|
+
parameters: {
|
|
68
|
+
type: 'object',
|
|
69
|
+
properties: tool.parameters.properties,
|
|
70
|
+
required: tool.parameters.required,
|
|
71
|
+
},
|
|
72
|
+
},
|
|
73
|
+
}));
|
|
74
|
+
}
|
|
75
|
+
|
|
76
|
+
private convertMessages(messages: AgentMessage[]): OpenAIMessage[] {
|
|
77
|
+
return messages.map(msg => {
|
|
78
|
+
if (msg.role === 'system' || msg.role === 'user') {
|
|
79
|
+
return { role: msg.role, content: msg.content };
|
|
80
|
+
}
|
|
81
|
+
if (msg.role === 'assistant') {
|
|
82
|
+
const openaiMsg: OpenAIMessage = { role: 'assistant', content: msg.content || null };
|
|
83
|
+
if (msg.tool_calls?.length) {
|
|
84
|
+
openaiMsg.tool_calls = msg.tool_calls.map(tc => ({
|
|
85
|
+
id: tc.id,
|
|
86
|
+
type: 'function' as const,
|
|
87
|
+
function: {
|
|
88
|
+
name: tc.name,
|
|
89
|
+
arguments: typeof tc.arguments === 'string' ? tc.arguments : JSON.stringify(tc.arguments || {}),
|
|
90
|
+
},
|
|
91
|
+
}));
|
|
92
|
+
}
|
|
93
|
+
return openaiMsg;
|
|
94
|
+
}
|
|
95
|
+
if (msg.role === 'tool') {
|
|
96
|
+
return { role: 'tool', content: msg.content, tool_call_id: msg.tool_call_id ?? '' };
|
|
97
|
+
}
|
|
98
|
+
return { role: msg.role as OpenAIMessage['role'], content: msg.content };
|
|
99
|
+
});
|
|
100
|
+
}
|
|
101
|
+
|
|
102
|
+
private isReasoningModel(): boolean {
|
|
103
|
+
const m = this.config.model;
|
|
104
|
+
return /^o[0-9]/.test(m) || m.includes('o1') || m.includes('o3') || m.includes('gpt-5');
|
|
105
|
+
}
|
|
106
|
+
|
|
107
|
+
async chat(messages: AgentMessage[], tools?: AgentTool[], signal?: AbortSignal): Promise<LLMResponse> {
|
|
108
|
+
const requestBody: Record<string, unknown> = {
|
|
109
|
+
model: this.config.model,
|
|
110
|
+
max_completion_tokens: this.config.maxTokens || 4096,
|
|
111
|
+
messages: this.convertMessages(messages),
|
|
112
|
+
};
|
|
113
|
+
|
|
114
|
+
// Reasoning models (o1, o3, etc.) only support temperature=1
|
|
115
|
+
if (!this.isReasoningModel()) {
|
|
116
|
+
requestBody.temperature = this.config.temperature ?? 1;
|
|
117
|
+
}
|
|
118
|
+
|
|
119
|
+
if (tools?.length) {
|
|
120
|
+
requestBody.tools = this.convertTools(tools);
|
|
121
|
+
}
|
|
122
|
+
|
|
123
|
+
const response = await fetch('https://api.openai.com/v1/chat/completions', {
|
|
124
|
+
method: 'POST',
|
|
125
|
+
headers: {
|
|
126
|
+
'Content-Type': 'application/json',
|
|
127
|
+
'Authorization': `Bearer ${this.config.apiKey}`,
|
|
128
|
+
},
|
|
129
|
+
body: JSON.stringify(requestBody),
|
|
130
|
+
signal,
|
|
131
|
+
});
|
|
132
|
+
|
|
133
|
+
if (!response.ok) {
|
|
134
|
+
const error = sanitizeLlmError(await response.text());
|
|
135
|
+
throw new Error(`OpenAI API error: ${response.status} - ${error}`);
|
|
136
|
+
}
|
|
137
|
+
|
|
138
|
+
const data = await response.json() as {
|
|
139
|
+
choices: { message: { content: string | null; tool_calls?: { id: string; function: { name: string; arguments: string } }[] }; finish_reason: string }[];
|
|
140
|
+
usage: { prompt_tokens: number; completion_tokens: number };
|
|
141
|
+
};
|
|
142
|
+
|
|
143
|
+
if (!data.choices || data.choices.length === 0) {
|
|
144
|
+
throw new Error('OpenAI API returned empty choices array');
|
|
145
|
+
}
|
|
146
|
+
|
|
147
|
+
const choice = data.choices[0];
|
|
148
|
+
const toolCalls: ToolCall[] = [];
|
|
149
|
+
if (choice.message.tool_calls) {
|
|
150
|
+
for (const tc of choice.message.tool_calls) {
|
|
151
|
+
try {
|
|
152
|
+
toolCalls.push({ id: tc.id, name: tc.function.name, arguments: JSON.parse(tc.function.arguments) });
|
|
153
|
+
} catch (e) {
|
|
154
|
+
logError('Failed to parse tool call', e, { module: 'services/agent/providers' });
|
|
155
|
+
}
|
|
156
|
+
}
|
|
157
|
+
}
|
|
158
|
+
|
|
159
|
+
return {
|
|
160
|
+
content: choice.message.content || '',
|
|
161
|
+
toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
|
|
162
|
+
stopReason: choice.finish_reason as LLMResponse['stopReason'],
|
|
163
|
+
usage: { inputTokens: data.usage.prompt_tokens, outputTokens: data.usage.completion_tokens },
|
|
164
|
+
};
|
|
165
|
+
}
|
|
166
|
+
}
|
|
167
|
+
|
|
168
|
+
// ============================================================================
|
|
169
|
+
// Anthropic Provider
|
|
170
|
+
// ============================================================================
|
|
171
|
+
|
|
172
|
+
interface AnthropicMessage {
|
|
173
|
+
role: 'user' | 'assistant';
|
|
174
|
+
content: string | { type: string; text?: string; tool_use_id?: string; content?: string; id?: string; name?: string; input?: unknown }[];
|
|
175
|
+
}
|
|
176
|
+
|
|
177
|
+
class AnthropicProvider implements LLMProvider {
|
|
178
|
+
constructor(private config: ModelConfig) {}
|
|
179
|
+
|
|
180
|
+
private convertTools(tools: AgentTool[]) {
|
|
181
|
+
return tools.map(tool => ({
|
|
182
|
+
name: tool.name,
|
|
183
|
+
description: tool.description,
|
|
184
|
+
input_schema: {
|
|
185
|
+
type: 'object',
|
|
186
|
+
properties: tool.parameters.properties,
|
|
187
|
+
required: tool.parameters.required,
|
|
188
|
+
},
|
|
189
|
+
}));
|
|
190
|
+
}
|
|
191
|
+
|
|
192
|
+
private convertMessages(messages: AgentMessage[]): { system: string; messages: AnthropicMessage[] } {
|
|
193
|
+
let systemPrompt = '';
|
|
194
|
+
const anthropicMessages: AnthropicMessage[] = [];
|
|
195
|
+
|
|
196
|
+
for (const msg of messages) {
|
|
197
|
+
if (msg.role === 'system') {
|
|
198
|
+
systemPrompt += (systemPrompt ? '\n\n' : '') + msg.content;
|
|
199
|
+
continue;
|
|
200
|
+
}
|
|
201
|
+
|
|
202
|
+
if (msg.role === 'user') {
|
|
203
|
+
anthropicMessages.push({ role: 'user', content: msg.content });
|
|
204
|
+
} else if (msg.role === 'assistant') {
|
|
205
|
+
if (msg.tool_calls?.length) {
|
|
206
|
+
const content: { type: string; id?: string; name?: string; input?: unknown; text?: string }[] = [];
|
|
207
|
+
if (msg.content) {
|
|
208
|
+
content.push({ type: 'text', text: msg.content });
|
|
209
|
+
}
|
|
210
|
+
for (const tc of msg.tool_calls) {
|
|
211
|
+
content.push({
|
|
212
|
+
type: 'tool_use',
|
|
213
|
+
id: tc.id,
|
|
214
|
+
name: tc.name,
|
|
215
|
+
input: tc.arguments,
|
|
216
|
+
});
|
|
217
|
+
}
|
|
218
|
+
anthropicMessages.push({ role: 'assistant', content });
|
|
219
|
+
} else {
|
|
220
|
+
anthropicMessages.push({ role: 'assistant', content: msg.content });
|
|
221
|
+
}
|
|
222
|
+
} else if (msg.role === 'tool') {
|
|
223
|
+
// Tool results need to be in a user message
|
|
224
|
+
const lastMsg = anthropicMessages[anthropicMessages.length - 1];
|
|
225
|
+
if (lastMsg?.role === 'user' && Array.isArray(lastMsg.content)) {
|
|
226
|
+
(lastMsg.content as { type: string; tool_use_id?: string; content?: string }[]).push({
|
|
227
|
+
type: 'tool_result',
|
|
228
|
+
tool_use_id: msg.tool_call_id ?? '',
|
|
229
|
+
content: msg.content,
|
|
230
|
+
});
|
|
231
|
+
} else {
|
|
232
|
+
anthropicMessages.push({
|
|
233
|
+
role: 'user',
|
|
234
|
+
content: [{ type: 'tool_result', tool_use_id: msg.tool_call_id ?? '', content: msg.content }],
|
|
235
|
+
});
|
|
236
|
+
}
|
|
237
|
+
}
|
|
238
|
+
}
|
|
239
|
+
|
|
240
|
+
return { system: systemPrompt, messages: anthropicMessages };
|
|
241
|
+
}
|
|
242
|
+
|
|
243
|
+
async chat(messages: AgentMessage[], tools?: AgentTool[], signal?: AbortSignal): Promise<LLMResponse> {
|
|
244
|
+
const { system, messages: anthropicMessages } = this.convertMessages(messages);
|
|
245
|
+
|
|
246
|
+
const requestBody: Record<string, unknown> = {
|
|
247
|
+
model: this.config.model,
|
|
248
|
+
max_tokens: this.config.maxTokens || 4096,
|
|
249
|
+
messages: anthropicMessages,
|
|
250
|
+
};
|
|
251
|
+
|
|
252
|
+
if (system) {
|
|
253
|
+
requestBody.system = system;
|
|
254
|
+
}
|
|
255
|
+
|
|
256
|
+
if (tools?.length) {
|
|
257
|
+
requestBody.tools = this.convertTools(tools);
|
|
258
|
+
}
|
|
259
|
+
|
|
260
|
+
const response = await fetch('https://api.anthropic.com/v1/messages', {
|
|
261
|
+
method: 'POST',
|
|
262
|
+
headers: {
|
|
263
|
+
'Content-Type': 'application/json',
|
|
264
|
+
'x-api-key': this.config.apiKey,
|
|
265
|
+
'anthropic-version': '2023-06-01',
|
|
266
|
+
},
|
|
267
|
+
body: JSON.stringify(requestBody),
|
|
268
|
+
signal,
|
|
269
|
+
});
|
|
270
|
+
|
|
271
|
+
if (!response.ok) {
|
|
272
|
+
const error = sanitizeLlmError(await response.text());
|
|
273
|
+
throw new Error(`Anthropic API error: ${response.status} - ${error}`);
|
|
274
|
+
}
|
|
275
|
+
|
|
276
|
+
const data = await response.json() as {
|
|
277
|
+
content: { type: string; text?: string; id?: string; name?: string; input?: unknown }[];
|
|
278
|
+
stop_reason: string;
|
|
279
|
+
usage: { input_tokens: number; output_tokens: number };
|
|
280
|
+
};
|
|
281
|
+
|
|
282
|
+
if (!data.content || !Array.isArray(data.content)) {
|
|
283
|
+
throw new Error('Anthropic API returned invalid content structure');
|
|
284
|
+
}
|
|
285
|
+
|
|
286
|
+
let textContent = '';
|
|
287
|
+
const toolCalls: ToolCall[] = [];
|
|
288
|
+
|
|
289
|
+
for (const block of data.content) {
|
|
290
|
+
if (block.type === 'text') {
|
|
291
|
+
textContent += block.text || '';
|
|
292
|
+
} else if (block.type === 'tool_use') {
|
|
293
|
+
toolCalls.push({
|
|
294
|
+
id: block.id ?? '',
|
|
295
|
+
name: block.name ?? '',
|
|
296
|
+
arguments: block.input as Record<string, unknown>,
|
|
297
|
+
});
|
|
298
|
+
}
|
|
299
|
+
}
|
|
300
|
+
|
|
301
|
+
return {
|
|
302
|
+
content: textContent,
|
|
303
|
+
toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
|
|
304
|
+
stopReason: data.stop_reason === 'tool_use' ? 'tool_calls' : (data.stop_reason as LLMResponse['stopReason']),
|
|
305
|
+
usage: { inputTokens: data.usage.input_tokens, outputTokens: data.usage.output_tokens },
|
|
306
|
+
};
|
|
307
|
+
}
|
|
308
|
+
}
|
|
309
|
+
|
|
310
|
+
// ============================================================================
|
|
311
|
+
// Google Gemini Provider
|
|
312
|
+
// ============================================================================
|
|
313
|
+
|
|
314
|
+
interface GeminiContent {
|
|
315
|
+
role: 'user' | 'model';
|
|
316
|
+
parts: { text?: string; functionCall?: { name: string; args: unknown }; functionResponse?: { name: string; response: unknown } }[];
|
|
317
|
+
}
|
|
318
|
+
|
|
319
|
+
class GoogleProvider implements LLMProvider {
|
|
320
|
+
constructor(private config: ModelConfig) {}
|
|
321
|
+
|
|
322
|
+
private convertTools(tools: AgentTool[]) {
|
|
323
|
+
return [{
|
|
324
|
+
functionDeclarations: tools.map(tool => ({
|
|
325
|
+
name: tool.name,
|
|
326
|
+
description: tool.description,
|
|
327
|
+
parameters: {
|
|
328
|
+
type: 'OBJECT',
|
|
329
|
+
properties: Object.fromEntries(
|
|
330
|
+
Object.entries(tool.parameters.properties).map(([key, value]: [string, { type?: string; description?: string }]) => [
|
|
331
|
+
key,
|
|
332
|
+
{ type: (value.type || 'string').toUpperCase(), description: value.description },
|
|
333
|
+
])
|
|
334
|
+
),
|
|
335
|
+
required: tool.parameters.required,
|
|
336
|
+
},
|
|
337
|
+
})),
|
|
338
|
+
}];
|
|
339
|
+
}
|
|
340
|
+
|
|
341
|
+
private convertMessages(messages: AgentMessage[]): { systemInstruction: string; contents: GeminiContent[] } {
|
|
342
|
+
let systemInstruction = '';
|
|
343
|
+
const contents: GeminiContent[] = [];
|
|
344
|
+
|
|
345
|
+
for (const msg of messages) {
|
|
346
|
+
if (msg.role === 'system') {
|
|
347
|
+
systemInstruction += (systemInstruction ? '\n\n' : '') + msg.content;
|
|
348
|
+
continue;
|
|
349
|
+
}
|
|
350
|
+
|
|
351
|
+
if (msg.role === 'user') {
|
|
352
|
+
contents.push({ role: 'user', parts: [{ text: msg.content }] });
|
|
353
|
+
} else if (msg.role === 'assistant') {
|
|
354
|
+
const parts: GeminiContent['parts'] = [];
|
|
355
|
+
if (msg.content) {
|
|
356
|
+
parts.push({ text: msg.content });
|
|
357
|
+
}
|
|
358
|
+
if (msg.tool_calls?.length) {
|
|
359
|
+
for (const tc of msg.tool_calls) {
|
|
360
|
+
parts.push({ functionCall: { name: tc.name, args: tc.arguments } });
|
|
361
|
+
}
|
|
362
|
+
}
|
|
363
|
+
if (parts.length) {
|
|
364
|
+
contents.push({ role: 'model', parts });
|
|
365
|
+
}
|
|
366
|
+
} else if (msg.role === 'tool') {
|
|
367
|
+
// Find the corresponding tool call name
|
|
368
|
+
const prevMsg = contents[contents.length - 1];
|
|
369
|
+
const functionCall = prevMsg?.parts.find(p => p.functionCall);
|
|
370
|
+
const name = functionCall?.functionCall?.name || 'unknown';
|
|
371
|
+
|
|
372
|
+
contents.push({
|
|
373
|
+
role: 'user',
|
|
374
|
+
parts: [{ functionResponse: { name, response: { result: msg.content } } }],
|
|
375
|
+
});
|
|
376
|
+
}
|
|
377
|
+
}
|
|
378
|
+
|
|
379
|
+
return { systemInstruction, contents };
|
|
380
|
+
}
|
|
381
|
+
|
|
382
|
+
async chat(messages: AgentMessage[], tools?: AgentTool[], signal?: AbortSignal): Promise<LLMResponse> {
|
|
383
|
+
const { systemInstruction, contents } = this.convertMessages(messages);
|
|
384
|
+
|
|
385
|
+
const requestBody: Record<string, unknown> = {
|
|
386
|
+
contents,
|
|
387
|
+
generationConfig: {
|
|
388
|
+
maxOutputTokens: this.config.maxTokens || 4096,
|
|
389
|
+
temperature: this.config.temperature ?? 1,
|
|
390
|
+
},
|
|
391
|
+
};
|
|
392
|
+
|
|
393
|
+
if (systemInstruction) {
|
|
394
|
+
requestBody.systemInstruction = { parts: [{ text: systemInstruction }] };
|
|
395
|
+
}
|
|
396
|
+
|
|
397
|
+
if (tools?.length) {
|
|
398
|
+
requestBody.tools = this.convertTools(tools);
|
|
399
|
+
}
|
|
400
|
+
|
|
401
|
+
const url = `https://generativelanguage.googleapis.com/v1beta/models/${this.config.model}:generateContent`;
|
|
402
|
+
|
|
403
|
+
const response = await fetch(url, {
|
|
404
|
+
method: 'POST',
|
|
405
|
+
headers: {
|
|
406
|
+
'Content-Type': 'application/json',
|
|
407
|
+
'x-goog-api-key': this.config.apiKey,
|
|
408
|
+
},
|
|
409
|
+
body: JSON.stringify(requestBody),
|
|
410
|
+
signal,
|
|
411
|
+
});
|
|
412
|
+
|
|
413
|
+
if (!response.ok) {
|
|
414
|
+
const error = sanitizeLlmError(await response.text());
|
|
415
|
+
throw new Error(`Google API error: ${response.status} - ${error}`);
|
|
416
|
+
}
|
|
417
|
+
|
|
418
|
+
const data = await response.json() as {
|
|
419
|
+
candidates: { content: { parts: { text?: string; functionCall?: { name: string; args: unknown } }[] }; finishReason: string }[];
|
|
420
|
+
usageMetadata: { promptTokenCount: number; candidatesTokenCount: number };
|
|
421
|
+
};
|
|
422
|
+
|
|
423
|
+
if (!data.candidates || data.candidates.length === 0) {
|
|
424
|
+
throw new Error('Google API returned empty candidates array');
|
|
425
|
+
}
|
|
426
|
+
|
|
427
|
+
const candidate = data.candidates[0];
|
|
428
|
+
|
|
429
|
+
if (!candidate.content || !candidate.content.parts) {
|
|
430
|
+
throw new Error('Google API returned invalid candidate content structure');
|
|
431
|
+
}
|
|
432
|
+
let textContent = '';
|
|
433
|
+
const toolCalls: ToolCall[] = [];
|
|
434
|
+
|
|
435
|
+
for (const part of candidate.content.parts) {
|
|
436
|
+
if (part.text) {
|
|
437
|
+
textContent += part.text;
|
|
438
|
+
} else if (part.functionCall) {
|
|
439
|
+
toolCalls.push({
|
|
440
|
+
id: `call_${crypto.randomUUID()}`,
|
|
441
|
+
name: part.functionCall.name,
|
|
442
|
+
arguments: part.functionCall.args as Record<string, unknown>,
|
|
443
|
+
});
|
|
444
|
+
}
|
|
445
|
+
}
|
|
446
|
+
|
|
447
|
+
const finishReason = candidate.finishReason === 'STOP' ? 'stop' :
|
|
448
|
+
toolCalls.length > 0 ? 'tool_calls' : 'stop';
|
|
449
|
+
|
|
450
|
+
return {
|
|
451
|
+
content: textContent,
|
|
452
|
+
toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
|
|
453
|
+
stopReason: finishReason as LLMResponse['stopReason'],
|
|
454
|
+
usage: {
|
|
455
|
+
inputTokens: data.usageMetadata?.promptTokenCount || 0,
|
|
456
|
+
outputTokens: data.usageMetadata?.candidatesTokenCount || 0,
|
|
457
|
+
},
|
|
458
|
+
};
|
|
459
|
+
}
|
|
460
|
+
}
|
|
461
|
+
|
|
462
|
+
// ============================================================================
|
|
463
|
+
// Factory
|
|
464
|
+
// ============================================================================
|
|
465
|
+
|
|
466
|
+
export function createProvider(config: ModelConfig): LLMProvider {
|
|
467
|
+
switch (config.provider) {
|
|
468
|
+
case 'openai':
|
|
469
|
+
return new OpenAIProvider(config);
|
|
470
|
+
case 'anthropic':
|
|
471
|
+
return new AnthropicProvider(config);
|
|
472
|
+
case 'google':
|
|
473
|
+
return new GoogleProvider(config);
|
|
474
|
+
default:
|
|
475
|
+
throw new Error(`Unknown provider: ${config.provider}`);
|
|
476
|
+
}
|
|
477
|
+
}
|
|
478
|
+
|
|
479
|
+
/**
|
|
480
|
+
* Get provider from model ID
|
|
481
|
+
*/
|
|
482
|
+
export function getProviderFromModel(modelId: string): ModelProvider {
|
|
483
|
+
return getModelProvider(modelId);
|
|
484
|
+
}
|
|
@@ -0,0 +1,78 @@
|
|
|
1
|
+
import type { ToolObserver } from '../memory-graph/graph-models';
|
|
2
|
+
import type { ToolCall, ToolDefinition, ToolResult } from '../../tools/tool-definitions';
|
|
3
|
+
import type { ToolExecutorLike } from '../../tools/executor';
|
|
4
|
+
|
|
5
|
+
type RemoteToolCatalog = {
|
|
6
|
+
tools: ToolDefinition[];
|
|
7
|
+
mcpFailedServers: string[];
|
|
8
|
+
};
|
|
9
|
+
|
|
10
|
+
export interface RemoteToolExecutorIo {
|
|
11
|
+
getToolCatalog(input: { runId: string }): Promise<RemoteToolCatalog>;
|
|
12
|
+
executeTool(input: { runId: string; toolCall: ToolCall }): Promise<ToolResult>;
|
|
13
|
+
cleanupToolExecutor(input: { runId: string }): Promise<void>;
|
|
14
|
+
}
|
|
15
|
+
|
|
16
|
+
export class RemoteToolExecutor implements ToolExecutorLike {
|
|
17
|
+
private readonly io: RemoteToolExecutorIo;
|
|
18
|
+
private readonly runId: string;
|
|
19
|
+
private readonly tools: ToolDefinition[];
|
|
20
|
+
private readonly failedServers: string[];
|
|
21
|
+
private observer: ToolObserver | null = null;
|
|
22
|
+
|
|
23
|
+
private constructor(runId: string, catalog: RemoteToolCatalog, io: RemoteToolExecutorIo) {
|
|
24
|
+
this.io = io;
|
|
25
|
+
this.runId = runId;
|
|
26
|
+
this.tools = catalog.tools;
|
|
27
|
+
this.failedServers = catalog.mcpFailedServers;
|
|
28
|
+
}
|
|
29
|
+
|
|
30
|
+
static async create(runId: string, io: RemoteToolExecutorIo): Promise<RemoteToolExecutor> {
|
|
31
|
+
const catalog = await io.getToolCatalog({ runId });
|
|
32
|
+
return new RemoteToolExecutor(runId, {
|
|
33
|
+
tools: Array.isArray(catalog.tools) ? catalog.tools : [],
|
|
34
|
+
mcpFailedServers: Array.isArray(catalog.mcpFailedServers) ? catalog.mcpFailedServers : [],
|
|
35
|
+
}, io);
|
|
36
|
+
}
|
|
37
|
+
|
|
38
|
+
getAvailableTools(): ToolDefinition[] {
|
|
39
|
+
return this.tools;
|
|
40
|
+
}
|
|
41
|
+
|
|
42
|
+
get mcpFailedServers(): string[] {
|
|
43
|
+
return this.failedServers;
|
|
44
|
+
}
|
|
45
|
+
|
|
46
|
+
setObserver(observer: ToolObserver): void {
|
|
47
|
+
this.observer = observer;
|
|
48
|
+
}
|
|
49
|
+
|
|
50
|
+
async execute(toolCall: ToolCall): Promise<ToolResult> {
|
|
51
|
+
const startedAt = Date.now();
|
|
52
|
+
const result = await this.io.executeTool({
|
|
53
|
+
runId: this.runId,
|
|
54
|
+
toolCall,
|
|
55
|
+
});
|
|
56
|
+
|
|
57
|
+
if (this.observer) {
|
|
58
|
+
try {
|
|
59
|
+
this.observer.observe({
|
|
60
|
+
toolName: toolCall.name,
|
|
61
|
+
arguments: toolCall.arguments,
|
|
62
|
+
result: result.error ? '' : result.output,
|
|
63
|
+
error: result.error,
|
|
64
|
+
timestamp: startedAt,
|
|
65
|
+
duration: Date.now() - startedAt,
|
|
66
|
+
});
|
|
67
|
+
} catch {
|
|
68
|
+
// Best-effort observation.
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
|
|
72
|
+
return result;
|
|
73
|
+
}
|
|
74
|
+
|
|
75
|
+
async cleanup(): Promise<void> {
|
|
76
|
+
await this.io.cleanupToolExecutor({ runId: this.runId });
|
|
77
|
+
}
|
|
78
|
+
}
|
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
import type { RunStatus } from '../../../shared/types';
|
|
2
|
+
import type { RunTerminalPayload } from '../run-notifier';
|
|
3
|
+
import type { AgentEvent } from './agent-models';
|
|
4
|
+
import { logError } from '../../../shared/utils/logger';
|
|
5
|
+
|
|
6
|
+
// --- RunCancelledError (formerly services/agent/errors.ts) ---
|
|
7
|
+
|
|
8
|
+
export class RunCancelledError extends Error {
|
|
9
|
+
constructor(message = 'Run cancelled') {
|
|
10
|
+
super(message);
|
|
11
|
+
this.name = 'RunCancelledError';
|
|
12
|
+
}
|
|
13
|
+
}
|
|
14
|
+
|
|
15
|
+
// --- Run reset policy (formerly services/agent/run-reset-policy.ts) ---
|
|
16
|
+
|
|
17
|
+
export function shouldResetRunToQueuedOnContainerError(status: RunStatus | null | undefined): boolean {
|
|
18
|
+
return status === 'running';
|
|
19
|
+
}
|
|
20
|
+
|
|
21
|
+
export interface RunLifecycleDeps {
|
|
22
|
+
updateRunStatus: (status: RunStatus, output?: string, error?: string) => Promise<void>;
|
|
23
|
+
emitEvent: (type: AgentEvent['type'], data: Record<string, unknown>) => Promise<void>;
|
|
24
|
+
buildTerminalEventPayload: (
|
|
25
|
+
status: 'completed' | 'failed' | 'cancelled',
|
|
26
|
+
details?: Record<string, unknown>
|
|
27
|
+
) => RunTerminalPayload;
|
|
28
|
+
autoCloseSession: (status: 'completed' | 'failed') => Promise<void>;
|
|
29
|
+
enqueuePostRunJobs: () => Promise<void>;
|
|
30
|
+
sanitizeErrorMessage: (error: string) => string;
|
|
31
|
+
}
|
|
32
|
+
|
|
33
|
+
export async function handleSuccessfulRunCompletion(deps: RunLifecycleDeps): Promise<void> {
|
|
34
|
+
await deps.enqueuePostRunJobs();
|
|
35
|
+
// Auto-close session if still open after successful completion
|
|
36
|
+
// Agent can still explicitly call container_commit/container_stop for control
|
|
37
|
+
await deps.autoCloseSession('completed');
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
export async function handleCancelledRun(deps: RunLifecycleDeps): Promise<void> {
|
|
41
|
+
await deps.updateRunStatus('cancelled', undefined, 'Run cancelled');
|
|
42
|
+
await deps.emitEvent('cancelled', deps.buildTerminalEventPayload('cancelled'));
|
|
43
|
+
await deps.autoCloseSession('failed');
|
|
44
|
+
await deps.enqueuePostRunJobs();
|
|
45
|
+
}
|
|
46
|
+
|
|
47
|
+
export async function handleFailedRun(deps: RunLifecycleDeps, error: unknown): Promise<void> {
|
|
48
|
+
const rawErrorMessage = String(error);
|
|
49
|
+
const errorMessage = deps.sanitizeErrorMessage(rawErrorMessage);
|
|
50
|
+
logError('Agent error', rawErrorMessage, { module: 'services/agent/run-lifecycle' }); // Full error for internal logs
|
|
51
|
+
|
|
52
|
+
await deps.updateRunStatus('failed', undefined, errorMessage);
|
|
53
|
+
await deps.emitEvent(
|
|
54
|
+
'error',
|
|
55
|
+
deps.buildTerminalEventPayload('failed', { error: errorMessage })
|
|
56
|
+
);
|
|
57
|
+
await deps.autoCloseSession('failed');
|
|
58
|
+
await deps.enqueuePostRunJobs();
|
|
59
|
+
}
|
|
@@ -0,0 +1,77 @@
|
|
|
1
|
+
import type { AgentConfig } from './agent-models';
|
|
2
|
+
import type { Env } from '../../../shared/types';
|
|
3
|
+
import { BUILTIN_TOOLS } from '../../tools/builtin';
|
|
4
|
+
import { SYSTEM_PROMPTS } from './prompt-builder';
|
|
5
|
+
import { logWarn } from '../../../shared/utils/logger';
|
|
6
|
+
import { parseIntValue, parseFloatValue } from 'takos-common/env-parse';
|
|
7
|
+
import {
|
|
8
|
+
AGENT_ITERATION_TIMEOUT_MS,
|
|
9
|
+
AGENT_TOTAL_TIMEOUT_MS,
|
|
10
|
+
AGENT_TOOL_EXECUTION_TIMEOUT_MS,
|
|
11
|
+
AGENT_LANGGRAPH_TIMEOUT_MS,
|
|
12
|
+
} from '../../../shared/config/timeouts';
|
|
13
|
+
|
|
14
|
+
const DEFAULT_MAX_ITERATIONS = 10000;
|
|
15
|
+
const DEFAULT_TEMPERATURE = 0.5;
|
|
16
|
+
// Default timeouts — these apply when running in CF Workers (15-min Queue consumer limit).
|
|
17
|
+
// When running inside a CF Container (executor), AGENT_TOTAL_TIMEOUT env var is set
|
|
18
|
+
// to 86400000 (24h) and the 15-min cap is not enforced.
|
|
19
|
+
export const DEFAULT_ITERATION_TIMEOUT = AGENT_ITERATION_TIMEOUT_MS; // 2 min per LLM call
|
|
20
|
+
export const DEFAULT_TOTAL_TIMEOUT = AGENT_TOTAL_TIMEOUT_MS; // 15 min total (CF Workers Queue limit)
|
|
21
|
+
const DEFAULT_TOOL_EXECUTION_TIMEOUT = AGENT_TOOL_EXECUTION_TIMEOUT_MS; // 5 min per tool (e.g. build commands)
|
|
22
|
+
const DEFAULT_LANGGRAPH_TIMEOUT = AGENT_LANGGRAPH_TIMEOUT_MS; // 15 min for LangGraph (CF Workers)
|
|
23
|
+
|
|
24
|
+
export function getTimeoutConfig(env?: Env): {
|
|
25
|
+
iterationTimeout: number;
|
|
26
|
+
totalTimeout: number;
|
|
27
|
+
toolExecutionTimeout: number;
|
|
28
|
+
langGraphTimeout: number;
|
|
29
|
+
} {
|
|
30
|
+
const warn = (msg: string) => logWarn(msg, { module: 'services/agent/runner-config' });
|
|
31
|
+
|
|
32
|
+
const MIN_TIMEOUT = 1000;
|
|
33
|
+
// In CF Container mode, AGENT_TOTAL_TIMEOUT env var sets a higher limit (up to 24h).
|
|
34
|
+
// In CF Workers Queue mode, cap at 15 min (Queue consumer hard limit).
|
|
35
|
+
const MAX_TIMEOUT = env?.AGENT_TOTAL_TIMEOUT
|
|
36
|
+
? Math.min(parseIntValue('AGENT_TOTAL_TIMEOUT', env.AGENT_TOTAL_TIMEOUT, 900000, { min: MIN_TIMEOUT, max: 86400000, warn }), 86400000)
|
|
37
|
+
: 900000; // default cap: 15 min (CF Workers Queue consumer limit)
|
|
38
|
+
|
|
39
|
+
const parseOpts = { min: MIN_TIMEOUT, max: MAX_TIMEOUT, warn };
|
|
40
|
+
|
|
41
|
+
return {
|
|
42
|
+
iterationTimeout: parseIntValue('AGENT_ITERATION_TIMEOUT', env?.AGENT_ITERATION_TIMEOUT, DEFAULT_ITERATION_TIMEOUT, parseOpts),
|
|
43
|
+
totalTimeout: parseIntValue('AGENT_TOTAL_TIMEOUT', env?.AGENT_TOTAL_TIMEOUT, DEFAULT_TOTAL_TIMEOUT, parseOpts),
|
|
44
|
+
toolExecutionTimeout: parseIntValue('TOOL_EXECUTION_TIMEOUT', env?.TOOL_EXECUTION_TIMEOUT, DEFAULT_TOOL_EXECUTION_TIMEOUT, parseOpts),
|
|
45
|
+
langGraphTimeout: parseIntValue('LANGGRAPH_TIMEOUT', env?.LANGGRAPH_TIMEOUT, DEFAULT_LANGGRAPH_TIMEOUT, parseOpts),
|
|
46
|
+
};
|
|
47
|
+
}
|
|
48
|
+
|
|
49
|
+
export function getAgentConfig(agentType: string, env?: Env): AgentConfig {
|
|
50
|
+
const warn = (msg: string) => logWarn(msg, { module: 'services/agent/runner-config' });
|
|
51
|
+
|
|
52
|
+
const systemPrompt = SYSTEM_PROMPTS[agentType] || SYSTEM_PROMPTS.default;
|
|
53
|
+
|
|
54
|
+
const tools = BUILTIN_TOOLS.map((tool) => ({
|
|
55
|
+
name: tool.name,
|
|
56
|
+
description: tool.description,
|
|
57
|
+
parameters: tool.parameters,
|
|
58
|
+
}));
|
|
59
|
+
|
|
60
|
+
const maxIterations = parseIntValue('MAX_AGENT_ITERATIONS', env?.MAX_AGENT_ITERATIONS, DEFAULT_MAX_ITERATIONS, { min: 1, warn });
|
|
61
|
+
|
|
62
|
+
const temperature = parseFloatValue('AGENT_TEMPERATURE', env?.AGENT_TEMPERATURE, DEFAULT_TEMPERATURE, { min: 0, max: 1, warn });
|
|
63
|
+
|
|
64
|
+
const rateLimitRaw = env?.AGENT_RATE_LIMIT
|
|
65
|
+
? parseIntValue('AGENT_RATE_LIMIT', env.AGENT_RATE_LIMIT, 0, { min: 1, warn })
|
|
66
|
+
: undefined;
|
|
67
|
+
const rateLimit = rateLimitRaw && rateLimitRaw > 0 ? rateLimitRaw : undefined;
|
|
68
|
+
|
|
69
|
+
return {
|
|
70
|
+
type: agentType,
|
|
71
|
+
systemPrompt,
|
|
72
|
+
tools,
|
|
73
|
+
maxIterations,
|
|
74
|
+
temperature,
|
|
75
|
+
rateLimit,
|
|
76
|
+
};
|
|
77
|
+
}
|