swayrouter 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/.env.example +43 -0
- package/CHANGELOG.md +132 -0
- package/LICENSE +30 -0
- package/README.id.md +633 -0
- package/README.md +628 -0
- package/cli/cli.js +388 -0
- package/cli/package.json +4 -0
- package/cli/src/cli/commands/xaiVideo.js +280 -0
- package/cli/src/cli/tray/SwayAvatar.svg +20 -0
- package/cli/src/cli/tray/autostart.js +351 -0
- package/cli/src/cli/tray/icon.ico +0 -0
- package/cli/src/cli/tray/icon.png +0 -0
- package/cli/src/cli/tray/tray.js +258 -0
- package/cli/src/cli/tray/tray.ps1 +129 -0
- package/cli/src/cli/tray/trayHost.js +157 -0
- package/cli/src/cli/tray/trayWin.js +83 -0
- package/cli/src/cli/utils/display.js +123 -0
- package/cli/src/cli/utils/input.js +138 -0
- package/cli/src/runtime/discover.js +153 -0
- package/cli/src/runtime/lifecycle.js +333 -0
- package/cli/src/runtime/runMenu.js +250 -0
- package/cli/src/runtime/uiAdapters.js +12 -0
- package/cli/src/runtime/update.js +130 -0
- package/dashboard/dist/assets/CliTools-Ch3gKuNw.js +24 -0
- package/dashboard/dist/assets/Combo-BBCldjWi.js +1 -0
- package/dashboard/dist/assets/ConsoleLog-wywAYcPT.js +6 -0
- package/dashboard/dist/assets/Dashboard-DxzyF41b.js +1 -0
- package/dashboard/dist/assets/Header-C_9Su9BF.js +1 -0
- package/dashboard/dist/assets/Keys-D0ufgcsc.js +1 -0
- package/dashboard/dist/assets/Login-BeKpipS_.js +1 -0
- package/dashboard/dist/assets/ManageApiKey-CVXsmW2B.js +1 -0
- package/dashboard/dist/assets/OAuthCallback-D_UZYcF6.js +1 -0
- package/dashboard/dist/assets/Provider-BU7JfTqm.js +4 -0
- package/dashboard/dist/assets/ProviderModelAccordion-BAifkfVR.js +1 -0
- package/dashboard/dist/assets/Proxy-CU15VJ--.js +1 -0
- package/dashboard/dist/assets/QuotaMonitor-HzGK7bcc.js +1 -0
- package/dashboard/dist/assets/Settings-Dt4y2dC0.js +1 -0
- package/dashboard/dist/assets/Setup-DSDUwU7N.js +1 -0
- package/dashboard/dist/assets/StatusBadge-1nAOyHU2.js +1 -0
- package/dashboard/dist/assets/SwayChat-Bup7up6D.js +10 -0
- package/dashboard/dist/assets/Usage-eWnrPQ_v.js +4 -0
- package/dashboard/dist/assets/accordion-DOOPDDgt.js +1 -0
- package/dashboard/dist/assets/admin-extras-api-CCdu5TFM.js +1 -0
- package/dashboard/dist/assets/api-keys-api-ifAC2Ok3.js +1 -0
- package/dashboard/dist/assets/card-CC0Omjcv.js +1 -0
- package/dashboard/dist/assets/chatStudio-C0zPz_Z_.js +2 -0
- package/dashboard/dist/assets/checkbox-7SZBrmbN.js +1 -0
- package/dashboard/dist/assets/connections-api-_N54UD3S.js +1 -0
- package/dashboard/dist/assets/copy-button-qBvM6H9z.js +1 -0
- package/dashboard/dist/assets/currency-api-SV9CBf5e.js +1 -0
- package/dashboard/dist/assets/flip-button-BGgkjtl0.js +1 -0
- package/dashboard/dist/assets/frame-cUiti6Lq.js +1 -0
- package/dashboard/dist/assets/icons-vendor-CUx7SB5V.js +541 -0
- package/dashboard/dist/assets/index-B6j-9zHy.js +10 -0
- package/dashboard/dist/assets/index-C-_l0yYz.css +1 -0
- package/dashboard/dist/assets/live-mode-CHShLY4E.js +1 -0
- package/dashboard/dist/assets/markdown-vendor-BY8e1gQ8.js +29 -0
- package/dashboard/dist/assets/motion-vendor-g5fpGzmJ.js +9 -0
- package/dashboard/dist/assets/oauth-callback-CqYQzyd7.js +1 -0
- package/dashboard/dist/assets/query-vendor-DVevaSVS.js +1 -0
- package/dashboard/dist/assets/react-vendor-xnYrwzsL.js +61 -0
- package/dashboard/dist/assets/ripple-button-l8aMiVVG.js +1 -0
- package/dashboard/dist/assets/segmented-CqmEnQjZ.js +1 -0
- package/dashboard/dist/assets/select-D8pUtMXp.js +1 -0
- package/dashboard/dist/assets/skeleton-7uJDiIXe.js +1 -0
- package/dashboard/dist/assets/table-D0M60Tu1.js +1 -0
- package/dashboard/dist/assets/tabs-C6YkuulG.js +1 -0
- package/dashboard/dist/assets/tunnel-api-DDPsBaDV.js +1 -0
- package/dashboard/dist/assets/ui-vendor-BuM9f2NY.js +17 -0
- package/dashboard/dist/image2/SwayAvatar.svg +20 -0
- package/dashboard/dist/index.html +32 -0
- package/dashboard/dist/logo/avatar.svg +1 -0
- package/dashboard/dist/logo/sway.svg +20 -0
- package/dashboard/dist/logo/swaychat.svg +20 -0
- package/dashboard/dist/providers/SwayCustom.svg +19 -0
- package/dashboard/dist/providers/alicode.svg +12 -0
- package/dashboard/dist/providers/anthropic.svg +11 -0
- package/dashboard/dist/providers/antigravity.svg +3 -0
- package/dashboard/dist/providers/azure.svg +5 -0
- package/dashboard/dist/providers/baidu.svg +3 -0
- package/dashboard/dist/providers/blackbox.svg +5 -0
- package/dashboard/dist/providers/byteplus.svg +3 -0
- package/dashboard/dist/providers/cerebras.svg +4 -0
- package/dashboard/dist/providers/chutes.svg +4 -0
- package/dashboard/dist/providers/claude.svg +3 -0
- package/dashboard/dist/providers/cline.svg +11 -0
- package/dashboard/dist/providers/cloudflare-ai.svg +4 -0
- package/dashboard/dist/providers/codebuddy.svg +11 -0
- package/dashboard/dist/providers/codex.svg +10 -0
- package/dashboard/dist/providers/commandcode.svg +3 -0
- package/dashboard/dist/providers/cursor.svg +3 -0
- package/dashboard/dist/providers/deepseek.svg +10 -0
- package/dashboard/dist/providers/featherless.svg +3 -0
- package/dashboard/dist/providers/fireworks.svg +3 -0
- package/dashboard/dist/providers/gemini.svg +3 -0
- package/dashboard/dist/providers/geminicli.svg +11 -0
- package/dashboard/dist/providers/github.svg +3 -0
- package/dashboard/dist/providers/grok.svg +3 -0
- package/dashboard/dist/providers/groq.svg +3 -0
- package/dashboard/dist/providers/hermesagent.svg +1 -0
- package/dashboard/dist/providers/kilocode.svg +15 -0
- package/dashboard/dist/providers/kimchi.svg +3 -0
- package/dashboard/dist/providers/kimi.svg +4 -0
- package/dashboard/dist/providers/kiro.svg +10 -0
- package/dashboard/dist/providers/meta.svg +3 -0
- package/dashboard/dist/providers/mimo.svg +10 -0
- package/dashboard/dist/providers/minimax.svg +3 -0
- package/dashboard/dist/providers/mistral.svg +3 -0
- package/dashboard/dist/providers/morph.svg +3 -0
- package/dashboard/dist/providers/nebius.svg +4 -0
- package/dashboard/dist/providers/nvidia.svg +3 -0
- package/dashboard/dist/providers/ollama.svg +3 -0
- package/dashboard/dist/providers/omp.svg +5 -0
- package/dashboard/dist/providers/openai.svg +3 -0
- package/dashboard/dist/providers/opencode.svg +3 -0
- package/dashboard/dist/providers/openrouter.svg +3 -0
- package/dashboard/dist/providers/perplexity.svg +3 -0
- package/dashboard/dist/providers/poolside.svg +36 -0
- package/dashboard/dist/providers/qoder.svg +11 -0
- package/dashboard/dist/providers/sumopod.svg +3 -0
- package/dashboard/dist/providers/tencent.svg +5 -0
- package/dashboard/dist/providers/together.svg +12 -0
- package/dashboard/dist/providers/trae.svg +3 -0
- package/dashboard/dist/providers/venice.svg +4 -0
- package/dashboard/dist/providers/vercel.svg +3 -0
- package/dashboard/dist/providers/vertex.svg +17 -0
- package/dashboard/dist/providers/windsurf.svg +10 -0
- package/dashboard/dist/providers/zai.svg +3 -0
- package/dashboard/tsconfig.json +13 -0
- package/package.json +87 -0
- package/src/config/appConstants.js +153 -0
- package/src/config/defaultThinkingSignature.js +7 -0
- package/src/config/errorConfig.js +117 -0
- package/src/config/grokCli.js +9 -0
- package/src/config/kiroConstants.js +235 -0
- package/src/config/mediaConfig.js +20 -0
- package/src/config/ollamaModels.js +18 -0
- package/src/config/providerModels.js +109 -0
- package/src/config/providers.js +17 -0
- package/src/config/runtimeConfig.js +84 -0
- package/src/dashboardGuard.js +254 -0
- package/src/executors/antigravity.js +389 -0
- package/src/executors/azure.js +49 -0
- package/src/executors/base.js +225 -0
- package/src/executors/codebuddy-cn.js +23 -0
- package/src/executors/codebuddy-intl.js +60 -0
- package/src/executors/codex.js +436 -0
- package/src/executors/commandcode.js +83 -0
- package/src/executors/cursor.js +1075 -0
- package/src/executors/default.js +294 -0
- package/src/executors/gemini-cli.js +88 -0
- package/src/executors/github.js +393 -0
- package/src/executors/grok-cli.js +510 -0
- package/src/executors/grok-web.js +342 -0
- package/src/executors/index.js +86 -0
- package/src/executors/kimchi.js +116 -0
- package/src/executors/kiro.js +1155 -0
- package/src/executors/ollama-local.js +14 -0
- package/src/executors/opencode-go.js +47 -0
- package/src/executors/opencode.js +31 -0
- package/src/executors/qoder.js +433 -0
- package/src/executors/trae.js +318 -0
- package/src/executors/vertex.js +151 -0
- package/src/executors/windsurf.js +555 -0
- package/src/executors/xiaomi-tokenplan.js +16 -0
- package/src/handlers/chatCore/nonStreamingHandler.js +370 -0
- package/src/handlers/chatCore/requestDetail.js +130 -0
- package/src/handlers/chatCore/sseToJsonHandler.js +318 -0
- package/src/handlers/chatCore/streamingHandler.js +250 -0
- package/src/handlers/chatCore.js +652 -0
- package/src/lib/auth/dashboardSession.js +83 -0
- package/src/lib/auth/loginLimiter.js +99 -0
- package/src/lib/cache/ttlCache.js +153 -0
- package/src/lib/consoleLogBuffer.js +127 -0
- package/src/lib/dataDir.js +45 -0
- package/src/lib/db/adapters/betterSqliteAdapter.js +36 -0
- package/src/lib/db/adapters/bunSqliteAdapter.js +47 -0
- package/src/lib/db/adapters/nodeSqliteAdapter.js +65 -0
- package/src/lib/db/adapters/runtimeHelpers.js +23 -0
- package/src/lib/db/adapters/sqljsAdapter.js +103 -0
- package/src/lib/db/backup.js +75 -0
- package/src/lib/db/driver.js +110 -0
- package/src/lib/db/helpers/jsonCol.js +18 -0
- package/src/lib/db/helpers/kvStore.js +48 -0
- package/src/lib/db/helpers/metaStore.js +32 -0
- package/src/lib/db/index.js +473 -0
- package/src/lib/db/migrate.js +267 -0
- package/src/lib/db/migrations/001-initial.js +12 -0
- package/src/lib/db/migrations/002-prune-headroom-pxpipe-settings.js +32 -0
- package/src/lib/db/migrations/004-request-details-trace-id.js +12 -0
- package/src/lib/db/migrations/005-request-details-payload-capture.js +11 -0
- package/src/lib/db/migrations/007-headroom-readd-settings.js +35 -0
- package/src/lib/db/migrations/008-account-model-locks.js +22 -0
- package/src/lib/db/migrations/009-warp-pool-lifecycle.js +5 -0
- package/src/lib/db/migrations/010-api-key-file-permissions.js +7 -0
- package/src/lib/db/migrations/011-remove-upstream-user-agent-profile.js +41 -0
- package/src/lib/db/migrations/012-remove-headroom-settings.js +33 -0
- package/src/lib/db/migrations/013-default-fill-first-routing.js +20 -0
- package/src/lib/db/migrations/014-normalize-fill-first-routing.js +37 -0
- package/src/lib/db/migrations/015-remove-ip-ban-tables.js +8 -0
- package/src/lib/db/migrations/016-prune-legacy-codebuddy-intl-models.js +25 -0
- package/src/lib/db/migrations/index.js +20 -0
- package/src/lib/db/nineRouterMigration.js +785 -0
- package/src/lib/db/paths.js +23 -0
- package/src/lib/db/repos/accountModelLocksRepo.js +74 -0
- package/src/lib/db/repos/aliasRepo.js +108 -0
- package/src/lib/db/repos/apiKeysRepo.js +204 -0
- package/src/lib/db/repos/combosRepo.js +85 -0
- package/src/lib/db/repos/connectionsRepo.js +522 -0
- package/src/lib/db/repos/disabledModelsRepo.js +63 -0
- package/src/lib/db/repos/nodesRepo.js +127 -0
- package/src/lib/db/repos/pricingRepo.js +108 -0
- package/src/lib/db/repos/proxyPoolsRepo.js +104 -0
- package/src/lib/db/repos/requestDetailsRepo.js +345 -0
- package/src/lib/db/repos/settingsRepo.js +135 -0
- package/src/lib/db/repos/usageRepo.js +1335 -0
- package/src/lib/db/schema.js +179 -0
- package/src/lib/db/version.js +32 -0
- package/src/lib/db/writeBehind.js +174 -0
- package/src/lib/disabledModelsDb.js +3 -0
- package/src/lib/env.ts +175 -0
- package/src/lib/errors/publicError.js +18 -0
- package/src/lib/grokBuildConfig.js +243 -0
- package/src/lib/guardRequest.js +51 -0
- package/src/lib/localDb.js +21 -0
- package/src/lib/mcp/stdioSseBridge.js +174 -0
- package/src/lib/network/connectionProxy.js +200 -0
- package/src/lib/network/proxyTest.js +158 -0
- package/src/lib/network/proxyUrl.js +76 -0
- package/src/lib/network/relay.js +71 -0
- package/src/lib/oauth/constants/oauth.js +152 -0
- package/src/lib/oauth/constants/xai.js +43 -0
- package/src/lib/oauth/kiroExternalIdp.js +167 -0
- package/src/lib/oauth/providerHelpers.js +90 -0
- package/src/lib/oauth/providers/_shared.js +12 -0
- package/src/lib/oauth/providers/antigravity.js +126 -0
- package/src/lib/oauth/providers/claude.js +60 -0
- package/src/lib/oauth/providers/cline.js +62 -0
- package/src/lib/oauth/providers/clinepass.js +62 -0
- package/src/lib/oauth/providers/codebuddy-cn.js +75 -0
- package/src/lib/oauth/providers/codebuddy-intl.js +73 -0
- package/src/lib/oauth/providers/codex.js +72 -0
- package/src/lib/oauth/providers/cursor.js +18 -0
- package/src/lib/oauth/providers/gemini-cli.js +84 -0
- package/src/lib/oauth/providers/github.js +96 -0
- package/src/lib/oauth/providers/grok-cli.js +123 -0
- package/src/lib/oauth/providers/index.js +233 -0
- package/src/lib/oauth/providers/kilocode.js +60 -0
- package/src/lib/oauth/providers/kimchi.js +79 -0
- package/src/lib/oauth/providers/kimi.js +79 -0
- package/src/lib/oauth/providers/kiro.js +147 -0
- package/src/lib/oauth/providers/qoder.js +90 -0
- package/src/lib/oauth/providers/trae.js +244 -0
- package/src/lib/oauth/providers/windsurf.js +123 -0
- package/src/lib/oauth/providers/xai.js +95 -0
- package/src/lib/oauth/providers.js +16 -0
- package/src/lib/oauth/services/cursor.js +132 -0
- package/src/lib/oauth/services/kiro.js +351 -0
- package/src/lib/oauth/services/oauth.js +135 -0
- package/src/lib/oauth/services/qoder.js +151 -0
- package/src/lib/oauth/services/xai.js +204 -0
- package/src/lib/oauth/utils/ideDetect.js +55 -0
- package/src/lib/oauth/utils/pkce.js +25 -0
- package/src/lib/oauth/utils/server.js +732 -0
- package/src/lib/oauth/utils/ui.js +43 -0
- package/src/lib/providerNormalization.js +40 -0
- package/src/lib/qoder/constants.js +1 -0
- package/src/lib/requestDetailsDb.js +3 -0
- package/src/lib/resourceProfile.ts +124 -0
- package/src/lib/runtimeSecrets.ts +130 -0
- package/src/lib/security/payloadCapture.js +87 -0
- package/src/lib/tunnel/cloudflare/cloudflared.js +436 -0
- package/src/lib/tunnel/cloudflare/config.js +8 -0
- package/src/lib/tunnel/cloudflare/healthCheck.js +37 -0
- package/src/lib/tunnel/cloudflare/manager.js +149 -0
- package/src/lib/tunnel/cloudflare/pid.js +31 -0
- package/src/lib/tunnel/cloudflare/publicUrl.js +41 -0
- package/src/lib/tunnel/cloudflare/requirements.js +12 -0
- package/src/lib/tunnel/index.js +26 -0
- package/src/lib/tunnel/shared/dnsResolver.js +16 -0
- package/src/lib/tunnel/shared/internetCheck.js +30 -0
- package/src/lib/tunnel/shared/state.js +45 -0
- package/src/lib/tunnel/shared/watchdogConfig.js +6 -0
- package/src/lib/update/installer.js +52 -0
- package/src/lib/update/version.js +116 -0
- package/src/lib/usageDb.js +6 -0
- package/src/lib/usageRefresh.js +88 -0
- package/src/lifecycle/appLifecycle.ts +99 -0
- package/src/models.js +36 -0
- package/src/next/headers.js +12 -0
- package/src/next/server.js +74 -0
- package/src/observability/metrics.js +439 -0
- package/src/observability/tracing.js +107 -0
- package/src/open-sse-index.js +70 -0
- package/src/providers/capabilities.js +299 -0
- package/src/providers/display.ts +142 -0
- package/src/providers/index.js +27 -0
- package/src/providers/models/helpers.js +19 -0
- package/src/providers/models/namePatterns.js +27 -0
- package/src/providers/models/schema.js +32 -0
- package/src/providers/pricing.js +248 -0
- package/src/providers/registry/agentrouter.js +52 -0
- package/src/providers/registry/alicode-intl.js +32 -0
- package/src/providers/registry/alicode.js +33 -0
- package/src/providers/registry/alims-intl.js +30 -0
- package/src/providers/registry/anthropic.js +29 -0
- package/src/providers/registry/antigravity.js +139 -0
- package/src/providers/registry/azure.js +23 -0
- package/src/providers/registry/baidu.js +35 -0
- package/src/providers/registry/blackbox.js +41 -0
- package/src/providers/registry/byteplus.js +36 -0
- package/src/providers/registry/cerebras.js +33 -0
- package/src/providers/registry/chutes.js +26 -0
- package/src/providers/registry/claude.js +80 -0
- package/src/providers/registry/cline.js +53 -0
- package/src/providers/registry/clinepass.js +62 -0
- package/src/providers/registry/cloudflare-ai.js +84 -0
- package/src/providers/registry/codebuddy-cn.js +95 -0
- package/src/providers/registry/codebuddy-intl.js +91 -0
- package/src/providers/registry/codex.js +145 -0
- package/src/providers/registry/commandcode.js +45 -0
- package/src/providers/registry/cursor.js +61 -0
- package/src/providers/registry/deepseek.js +55 -0
- package/src/providers/registry/featherless.js +36 -0
- package/src/providers/registry/fireworks.js +34 -0
- package/src/providers/registry/gemini-cli.js +59 -0
- package/src/providers/registry/gemini.js +76 -0
- package/src/providers/registry/github.js +131 -0
- package/src/providers/registry/glm-cn.js +35 -0
- package/src/providers/registry/glm.js +58 -0
- package/src/providers/registry/grok-cli.js +87 -0
- package/src/providers/registry/grok-web.js +41 -0
- package/src/providers/registry/groq.js +38 -0
- package/src/providers/registry/index.js +77 -0
- package/src/providers/registry/kilo-gateway.js +34 -0
- package/src/providers/registry/kilocode.js +48 -0
- package/src/providers/registry/kimchi.js +49 -0
- package/src/providers/registry/kimi.js +82 -0
- package/src/providers/registry/kiro.js +132 -0
- package/src/providers/registry/meta.js +34 -0
- package/src/providers/registry/minimax-cn.js +90 -0
- package/src/providers/registry/minimax.js +90 -0
- package/src/providers/registry/mistral.js +36 -0
- package/src/providers/registry/morph.js +31 -0
- package/src/providers/registry/nebius.js +26 -0
- package/src/providers/registry/nvidia.js +55 -0
- package/src/providers/registry/ollama-local.js +18 -0
- package/src/providers/registry/ollama.js +38 -0
- package/src/providers/registry/openai.js +114 -0
- package/src/providers/registry/opencode-go.js +41 -0
- package/src/providers/registry/opencode.js +26 -0
- package/src/providers/registry/openrouter.js +36 -0
- package/src/providers/registry/perplexity.js +56 -0
- package/src/providers/registry/poolside.js +30 -0
- package/src/providers/registry/qoder.js +58 -0
- package/src/providers/registry/sumopod.js +84 -0
- package/src/providers/registry/tencent.js +29 -0
- package/src/providers/registry/together.js +38 -0
- package/src/providers/registry/trae.js +71 -0
- package/src/providers/registry/venice.js +69 -0
- package/src/providers/registry/vercel-ai-gateway.js +37 -0
- package/src/providers/registry/vertex-partner.js +31 -0
- package/src/providers/registry/vertex.js +33 -0
- package/src/providers/registry/windsurf.js +129 -0
- package/src/providers/registry/xai.js +46 -0
- package/src/providers/registry/xiaomi-mimo.js +66 -0
- package/src/providers/registry/xiaomi-tokenplan.js +58 -0
- package/src/providers/registryEntry.js +3 -0
- package/src/providers/schema.js +33 -0
- package/src/providers/shared.js +114 -0
- package/src/providers/thinkingLevels.js +49 -0
- package/src/routes/api/auth/login/route.js +82 -0
- package/src/routes/api/auth/logout/route.js +9 -0
- package/src/routes/api/auth/reset-password/route.js +12 -0
- package/src/routes/api/auth/status/route.js +33 -0
- package/src/routes/api/auth/verify-password/route.js +39 -0
- package/src/routes/api/chat/tools/curl/route.js +100 -0
- package/src/routes/api/chat/tools/router/route.js +215 -0
- package/src/routes/api/chat/tools/search/route.js +362 -0
- package/src/routes/api/chat/tools/shell/route.js +228 -0
- package/src/routes/api/cli-tools/all-statuses/route.js +37 -0
- package/src/routes/api/cli-tools/claude-settings/route.js +261 -0
- package/src/routes/api/cli-tools/codex-settings/route.js +220 -0
- package/src/routes/api/cli-tools/cowork-mcp-registry/route.js +80 -0
- package/src/routes/api/cli-tools/cowork-mcp-tools/route.js +96 -0
- package/src/routes/api/cli-tools/cowork-settings/route.js +380 -0
- package/src/routes/api/cli-tools/grok-build-settings/route.js +152 -0
- package/src/routes/api/cli-tools/hermes-settings/route.js +169 -0
- package/src/routes/api/cli-tools/kilo-settings/route.js +131 -0
- package/src/routes/api/cli-tools/model-mappings/route.js +43 -0
- package/src/routes/api/cli-tools/omp-settings/route.js +266 -0
- package/src/routes/api/cli-tools/opencode-settings/route.js +243 -0
- package/src/routes/api/combos/[id]/route.js +73 -0
- package/src/routes/api/combos/route.js +43 -0
- package/src/routes/api/health/live/route.js +7 -0
- package/src/routes/api/health/ready/route.js +19 -0
- package/src/routes/api/health/route.js +17 -0
- package/src/routes/api/inflight/stream/route.js +24 -0
- package/src/routes/api/init/route.js +3 -0
- package/src/routes/api/keys/[id]/route.js +71 -0
- package/src/routes/api/keys/route.js +41 -0
- package/src/routes/api/logs/route.js +19 -0
- package/src/routes/api/mcp/[plugin]/message/route.js +23 -0
- package/src/routes/api/mcp/[plugin]/sse/route.js +35 -0
- package/src/routes/api/models/availability/route.js +104 -0
- package/src/routes/api/models/custom/route.js +58 -0
- package/src/routes/api/models/disabled/route.js +47 -0
- package/src/routes/api/models/route.js +64 -0
- package/src/routes/api/models/test/ping.js +236 -0
- package/src/routes/api/models/test/route.js +14 -0
- package/src/routes/api/oauth/[provider]/[action]/route.js +491 -0
- package/src/routes/api/oauth/codex/bulk-import/route.js +105 -0
- package/src/routes/api/oauth/codex/import-token/route.js +82 -0
- package/src/routes/api/oauth/cursor/auto-import/route.js +250 -0
- package/src/routes/api/oauth/cursor/import/route.js +86 -0
- package/src/routes/api/oauth/kiro/api-key/route.js +60 -0
- package/src/routes/api/oauth/kiro/auto-import/route.js +120 -0
- package/src/routes/api/oauth/kiro/import/route.js +72 -0
- package/src/routes/api/oauth/kiro/import-cli-proxy/route.js +42 -0
- package/src/routes/api/oauth/kiro/social-authorize/route.js +40 -0
- package/src/routes/api/oauth/kiro/social-exchange/route.js +62 -0
- package/src/routes/api/pricing/route.js +96 -0
- package/src/routes/api/provider-nodes/[id]/route.js +106 -0
- package/src/routes/api/provider-nodes/route.js +110 -0
- package/src/routes/api/provider-nodes/validate/route.js +197 -0
- package/src/routes/api/providers/[id]/models/route.js +716 -0
- package/src/routes/api/providers/[id]/route.js +195 -0
- package/src/routes/api/providers/[id]/test/route.js +27 -0
- package/src/routes/api/providers/[id]/test/testUtils.js +912 -0
- package/src/routes/api/providers/[id]/test-models/route.js +211 -0
- package/src/routes/api/providers/catalog/route.js +58 -0
- package/src/routes/api/providers/client/route.js +133 -0
- package/src/routes/api/providers/counts/route.js +43 -0
- package/src/routes/api/providers/kilo/free-models/route.js +58 -0
- package/src/routes/api/providers/route.js +361 -0
- package/src/routes/api/providers/suggested-models/filters.js +24 -0
- package/src/routes/api/providers/suggested-models/route.js +34 -0
- package/src/routes/api/providers/test-batch/route.js +145 -0
- package/src/routes/api/providers/validate/route.js +633 -0
- package/src/routes/api/proxy-pools/[id]/route.js +129 -0
- package/src/routes/api/proxy-pools/[id]/test/route.js +64 -0
- package/src/routes/api/proxy-pools/cloudflare-deploy/route.js +183 -0
- package/src/routes/api/proxy-pools/route.js +85 -0
- package/src/routes/api/proxy-pools/vercel-deploy/route.js +181 -0
- package/src/routes/api/settings/currency/route.js +29 -0
- package/src/routes/api/settings/database/path/route.js +8 -0
- package/src/routes/api/settings/database/route.js +89 -0
- package/src/routes/api/settings/migrate-9router/route.js +81 -0
- package/src/routes/api/settings/require-login/route.js +14 -0
- package/src/routes/api/settings/route.js +186 -0
- package/src/routes/api/shutdown/route.js +25 -0
- package/src/routes/api/tags/route.js +17 -0
- package/src/routes/api/translator/console-logs/route.js +28 -0
- package/src/routes/api/translator/console-logs/stream/route.js +95 -0
- package/src/routes/api/tunnel/disable/route.js +18 -0
- package/src/routes/api/tunnel/enable/route.js +32 -0
- package/src/routes/api/tunnel/status/route.js +23 -0
- package/src/routes/api/update/route.js +42 -0
- package/src/routes/api/usage/[connectionId]/codex-reset-credits/route.js +169 -0
- package/src/routes/api/usage/[connectionId]/route.js +222 -0
- package/src/routes/api/usage/chart/route.js +27 -0
- package/src/routes/api/usage/history/route.js +94 -0
- package/src/routes/api/usage/logs/route.js +14 -0
- package/src/routes/api/usage/providers/route.js +36 -0
- package/src/routes/api/usage/request-details/route.js +61 -0
- package/src/routes/api/usage/request-logs/route.js +15 -0
- package/src/routes/api/usage/stats/route.js +23 -0
- package/src/routes/api/usage/stream/route.js +78 -0
- package/src/routes/api/v1/api/chat/route.js +36 -0
- package/src/routes/api/v1/chat/completions/route.js +28 -0
- package/src/routes/api/v1/images/edits/route.js +70 -0
- package/src/routes/api/v1/images/generations/route.js +50 -0
- package/src/routes/api/v1/images/imageValidation.js +45 -0
- package/src/routes/api/v1/messages/count_tokens/route.js +87 -0
- package/src/routes/api/v1/messages/route.js +26 -0
- package/src/routes/api/v1/models/route.js +440 -0
- package/src/routes/api/v1/responses/compact/route.js +34 -0
- package/src/routes/api/v1/responses/route.js +26 -0
- package/src/routes/api/v1/route.js +1 -0
- package/src/routes/api/version/route.js +11 -0
- package/src/routes/table.js +1141 -0
- package/src/rtk/applyFilter.js +13 -0
- package/src/rtk/autodetect.js +104 -0
- package/src/rtk/caveman.js +6 -0
- package/src/rtk/cavemanPrompts.js +107 -0
- package/src/rtk/constants.js +51 -0
- package/src/rtk/filters/buildOutput.js +120 -0
- package/src/rtk/filters/dedupLog.js +43 -0
- package/src/rtk/filters/find.js +47 -0
- package/src/rtk/filters/gitDiff.js +90 -0
- package/src/rtk/filters/gitLog.js +88 -0
- package/src/rtk/filters/gitStatus.js +101 -0
- package/src/rtk/filters/grep.js +45 -0
- package/src/rtk/filters/ls.js +73 -0
- package/src/rtk/filters/readNumbered.js +22 -0
- package/src/rtk/filters/searchList.js +49 -0
- package/src/rtk/filters/smartTruncate.js +13 -0
- package/src/rtk/filters/tree.js +28 -0
- package/src/rtk/index.js +143 -0
- package/src/rtk/ponytail.js +6 -0
- package/src/rtk/ponytailPrompt.js +49 -0
- package/src/rtk/systemInject.js +89 -0
- package/src/server.ts +474 -0
- package/src/services/accountFallback.js +241 -0
- package/src/services/accountPool.js +389 -0
- package/src/services/capacityAdapter.js +140 -0
- package/src/services/cliModelMapping.js +84 -0
- package/src/services/clinepassModels.js +60 -0
- package/src/services/codebuddyModels.js +250 -0
- package/src/services/combo.js +482 -0
- package/src/services/copilotModels.js +126 -0
- package/src/services/currencyRate.js +56 -0
- package/src/services/cursorModels.js +197 -0
- package/src/services/grokCliModels.js +130 -0
- package/src/services/kimchiModels.js +175 -0
- package/src/services/kiroModels.js +246 -0
- package/src/services/model.js +83 -0
- package/src/services/oauthCredentialManager.js +153 -0
- package/src/services/projectId.js +240 -0
- package/src/services/provider.js +145 -0
- package/src/services/qoderModels.js +274 -0
- package/src/services/tokenRefresh/dedup.js +70 -0
- package/src/services/tokenRefresh/providers.js +626 -0
- package/src/services/tokenRefresh.js +274 -0
- package/src/services/usage/claude.js +134 -0
- package/src/services/usage/codebuddy-cn.js +212 -0
- package/src/services/usage/codex.js +193 -0
- package/src/services/usage/deepseek.js +103 -0
- package/src/services/usage/github.js +94 -0
- package/src/services/usage/google.js +218 -0
- package/src/services/usage/grok-cli.js +368 -0
- package/src/services/usage/grokCliQuotaFrame.js +170 -0
- package/src/services/usage/kimi.js +185 -0
- package/src/services/usage/kiro.js +172 -0
- package/src/services/usage/minimax.js +215 -0
- package/src/services/usage/misc.js +264 -0
- package/src/services/usage/shared.js +59 -0
- package/src/services/usage.js +56 -0
- package/src/shared/clineAuth.js +37 -0
- package/src/shared/constants/config.js +80 -0
- package/src/shared/constants/coworkPlugins.js +69 -0
- package/src/shared/constants/models.js +41 -0
- package/src/shared/constants/providers.js +128 -0
- package/src/shared/constants/providersDisplay.js +1 -0
- package/src/shared/qoder/constants.js +50 -0
- package/src/shared/qoder/cosy.js +134 -0
- package/src/shared/qoder/encoding.js +36 -0
- package/src/shared/services/initializeApp.js +226 -0
- package/src/shared/services/quotaAutoPing.js +310 -0
- package/src/shared/utils/apiKey.js +65 -0
- package/src/shared/utils/clineAuth.js +1 -0
- package/src/shared/utils/index.js +3 -0
- package/src/shared/utils/machineId.js +60 -0
- package/src/shared/utils/ssrfGuard.js +280 -0
- package/src/sse/handlers/chat.js +426 -0
- package/src/sse/oauthLease.js +32 -0
- package/src/sse/recovery.js +235 -0
- package/src/sse/services/auth.js +489 -0
- package/src/sse/services/backgroundTokenRefresh.js +173 -0
- package/src/sse/services/model.js +129 -0
- package/src/sse/services/requestCancellation.js +4 -0
- package/src/sse/services/tokenRefresh.js +264 -0
- package/src/sse/utils/logger.js +142 -0
- package/src/staticServe.ts +95 -0
- package/src/transformer/streamToJsonConverter.js +87 -0
- package/src/translator/canonical.js +81 -0
- package/src/translator/concerns/chunk.js +9 -0
- package/src/translator/concerns/compatFallback.js +147 -0
- package/src/translator/concerns/finishReason.js +59 -0
- package/src/translator/concerns/image.js +105 -0
- package/src/translator/concerns/json.js +4 -0
- package/src/translator/concerns/kiroConversation.js +428 -0
- package/src/translator/concerns/message.js +5 -0
- package/src/translator/concerns/modality.js +134 -0
- package/src/translator/concerns/paramSupport.js +97 -0
- package/src/translator/concerns/prefetch.js +86 -0
- package/src/translator/concerns/reasoning.js +18 -0
- package/src/translator/concerns/thinking.js +40 -0
- package/src/translator/concerns/thinkingUnified.js +319 -0
- package/src/translator/concerns/toolCall.js +134 -0
- package/src/translator/concerns/usage.js +61 -0
- package/src/translator/detection.js +42 -0
- package/src/translator/formats/claude.js +323 -0
- package/src/translator/formats/gemini.js +366 -0
- package/src/translator/formats/maxTokens.js +19 -0
- package/src/translator/formats/openai.js +116 -0
- package/src/translator/formats/responsesApi.js +117 -0
- package/src/translator/formats.js +28 -0
- package/src/translator/index.js +237 -0
- package/src/translator/request/antigravity-to-openai.js +217 -0
- package/src/translator/request/claude-to-kiro.js +288 -0
- package/src/translator/request/claude-to-openai.js +199 -0
- package/src/translator/request/gemini-to-openai.js +142 -0
- package/src/translator/request/openai-responses.js +373 -0
- package/src/translator/request/openai-to-claude.js +260 -0
- package/src/translator/request/openai-to-commandcode.js +161 -0
- package/src/translator/request/openai-to-cursor.js +175 -0
- package/src/translator/request/openai-to-gemini.js +422 -0
- package/src/translator/request/openai-to-kiro.js +360 -0
- package/src/translator/request/openai-to-ollama.js +152 -0
- package/src/translator/request/openai-to-vertex.js +36 -0
- package/src/translator/response/claude-to-openai.js +185 -0
- package/src/translator/response/commandcode-to-openai.js +164 -0
- package/src/translator/response/cursor-to-openai.js +18 -0
- package/src/translator/response/gemini-to-openai.js +127 -0
- package/src/translator/response/kiro-to-claude.js +230 -0
- package/src/translator/response/kiro-to-openai.js +136 -0
- package/src/translator/response/ollama-to-openai.js +106 -0
- package/src/translator/response/openai-responses.js +534 -0
- package/src/translator/response/openai-to-antigravity.js +107 -0
- package/src/translator/response/openai-to-claude.js +231 -0
- package/src/translator/schema/blocks.js +40 -0
- package/src/translator/schema/defaults.js +3 -0
- package/src/translator/schema/finishReasons.js +22 -0
- package/src/translator/schema/index.js +7 -0
- package/src/translator/schema/roles.js +12 -0
- package/src/transport/requestBoundary.ts +223 -0
- package/src/transport/routerAdmission.ts +380 -0
- package/src/types/provider.ts +59 -0
- package/src/utils/bypassHandler.js +272 -0
- package/src/utils/claudeCloaking.js +10 -0
- package/src/utils/claudeSignature.js +35 -0
- package/src/utils/clientDetector.js +183 -0
- package/src/utils/cursorChecksum.js +104 -0
- package/src/utils/cursorProtobuf.js +801 -0
- package/src/utils/debugLog.js +12 -0
- package/src/utils/error.js +120 -0
- package/src/utils/inflight.js +127 -0
- package/src/utils/kiroSessionReplay.js +139 -0
- package/src/utils/ollamaTransform.js +83 -0
- package/src/utils/proxyFetch.js +308 -0
- package/src/utils/reasoningContentInjector.js +71 -0
- package/src/utils/requestLogger.js +272 -0
- package/src/utils/responsesStreamHelpers.js +46 -0
- package/src/utils/sessionManager.js +205 -0
- package/src/utils/sse.js +13 -0
- package/src/utils/sseConstants.js +21 -0
- package/src/utils/stream.js +511 -0
- package/src/utils/streamHandler.js +256 -0
- package/src/utils/streamHelpers.js +113 -0
- package/src/utils/toolDeduper.js +44 -0
- package/src/utils/usageTracking.js +329 -0
- package/tsconfig.json +45 -0
|
@@ -0,0 +1,389 @@
|
|
|
1
|
+
import crypto from "crypto";
|
|
2
|
+
import { BaseExecutor } from "./base.js";
|
|
3
|
+
import { PROVIDERS } from "../config/providers.js";
|
|
4
|
+
import { OAUTH_ENDPOINTS, ANTIGRAVITY_HEADERS } from "../config/appConstants.js";
|
|
5
|
+
import { HTTP_STATUS } from "../config/runtimeConfig.js";
|
|
6
|
+
import { resolveSessionId } from "../utils/sessionManager.js";
|
|
7
|
+
import { proxyAwareFetch } from "../utils/proxyFetch.js";
|
|
8
|
+
import { cleanJSONSchemaForAntigravity } from "../translator/formats/gemini.js";
|
|
9
|
+
import { DEFAULT_THINKING_AG_SIGNATURE } from "../config/defaultThinkingSignature.js";
|
|
10
|
+
|
|
11
|
+
function sanitizeFunctionName(name) {
|
|
12
|
+
if (!name) return "_unknown";
|
|
13
|
+
let s = name.replace(/[^a-zA-Z0-9_.:\-]/g, "_");
|
|
14
|
+
if (!/^[a-zA-Z_]/.test(s)) s = "_" + s;
|
|
15
|
+
return s.substring(0, 64);
|
|
16
|
+
}
|
|
17
|
+
|
|
18
|
+
const MAX_RETRY_AFTER_MS = 10000;
|
|
19
|
+
const ANTIGRAVITY_TRANSIENT_RETRY_MAX_MS = 15000;
|
|
20
|
+
const MAX_ANTIGRAVITY_OUTPUT_TOKENS = 64000;
|
|
21
|
+
const ANTIGRAVITY_IDE_REQUEST_ID_RE = /^agent\/[^/]+\/\d+\/[^/]+\/\d+$/;
|
|
22
|
+
|
|
23
|
+
const ANTIGRAVITY_TRANSIENT_ERROR_PATTERNS = [
|
|
24
|
+
/high\s+traffic/i,
|
|
25
|
+
/agent\s+(execution\s+)?terminated\s+due\s+to\s+error/i,
|
|
26
|
+
/capacity/i,
|
|
27
|
+
/temporarily\s+unavailable/i,
|
|
28
|
+
/timeout/i,
|
|
29
|
+
/stream\s+(ended|closed|terminated|interrupted)/i,
|
|
30
|
+
/empty\s+response/i,
|
|
31
|
+
];
|
|
32
|
+
|
|
33
|
+
const ANTIGRAVITY_TRANSIENT_STATUSES = new Set([
|
|
34
|
+
HTTP_STATUS.SERVER_ERROR,
|
|
35
|
+
HTTP_STATUS.BAD_GATEWAY,
|
|
36
|
+
HTTP_STATUS.SERVICE_UNAVAILABLE,
|
|
37
|
+
HTTP_STATUS.GATEWAY_TIMEOUT,
|
|
38
|
+
]);
|
|
39
|
+
|
|
40
|
+
const ANTIGRAVITY_REQUEST_BLACKLIST = [
|
|
41
|
+
"output_config",
|
|
42
|
+
"thinking",
|
|
43
|
+
"reasoning_effort",
|
|
44
|
+
"reasoning",
|
|
45
|
+
"enable_thinking",
|
|
46
|
+
"thinking_budget",
|
|
47
|
+
"thinkingConfig",
|
|
48
|
+
];
|
|
49
|
+
|
|
50
|
+
const stripBlacklisted = obj => {
|
|
51
|
+
for (const key of ANTIGRAVITY_REQUEST_BLACKLIST) delete obj[key];
|
|
52
|
+
};
|
|
53
|
+
|
|
54
|
+
const IMAGE_MODEL_PATTERNS = [
|
|
55
|
+
/image/i,
|
|
56
|
+
/imagen/i,
|
|
57
|
+
/image-generation/i,
|
|
58
|
+
];
|
|
59
|
+
|
|
60
|
+
function isImageModel(model) {
|
|
61
|
+
if (!model) return false;
|
|
62
|
+
return IMAGE_MODEL_PATTERNS.some(p => p.test(model));
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
function parseImageConfig(model) {
|
|
66
|
+
const config = { aspectRatio: "1:1" };
|
|
67
|
+
const resMatch = model.match(/(\d+)x(\d+)$/);
|
|
68
|
+
if (resMatch) {
|
|
69
|
+
const w = parseInt(resMatch[1]);
|
|
70
|
+
const h = parseInt(resMatch[2]);
|
|
71
|
+
if (w <= 16 && h <= 16) {
|
|
72
|
+
config.aspectRatio = `${w}:${h}`;
|
|
73
|
+
} else {
|
|
74
|
+
|
|
75
|
+
const gcd = (a, b) => b ? gcd(b, a % b) : a;
|
|
76
|
+
const d = gcd(w, h);
|
|
77
|
+
config.aspectRatio = `${w/d}:${h/d}`;
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
return config;
|
|
81
|
+
}
|
|
82
|
+
|
|
83
|
+
function uuidFromSeed(seed) {
|
|
84
|
+
const bytes = crypto.createHash("sha256").update(String(seed || "antigravity")).digest().subarray(0, 16);
|
|
85
|
+
bytes[6] = (bytes[6] & 0x0f) | 0x50;
|
|
86
|
+
bytes[8] = (bytes[8] & 0x3f) | 0x80;
|
|
87
|
+
const hex = bytes.toString("hex");
|
|
88
|
+
return `${hex.slice(0, 8)}-${hex.slice(8, 12)}-${hex.slice(12, 16)}-${hex.slice(16, 20)}-${hex.slice(20)}`;
|
|
89
|
+
}
|
|
90
|
+
|
|
91
|
+
function buildIdeRequestId({ body, request, credentials, model, requestType }) {
|
|
92
|
+
if (ANTIGRAVITY_IDE_REQUEST_ID_RE.test(body?.requestId || "")) {
|
|
93
|
+
return body.requestId;
|
|
94
|
+
}
|
|
95
|
+
|
|
96
|
+
const sessionId = request?.sessionId || body?.request?.sessionId || credentials?._clientSessionId || credentials?.connectionId || credentials?.email || "anonymous";
|
|
97
|
+
const conversationId = uuidFromSeed(`antigravity:conversation:${sessionId}`);
|
|
98
|
+
const trajectoryId = uuidFromSeed(`antigravity:trajectory:${sessionId}:${model}:${requestType}`);
|
|
99
|
+
const contentCount = Array.isArray(request?.contents) ? request.contents.length : 1;
|
|
100
|
+
const step = Math.max(1, contentCount * 2 - 1);
|
|
101
|
+
return `agent/${conversationId}/${Date.now()}/${trajectoryId}/${step}`;
|
|
102
|
+
}
|
|
103
|
+
|
|
104
|
+
export class AntigravityExecutor extends BaseExecutor {
|
|
105
|
+
constructor() {
|
|
106
|
+
super("antigravity", PROVIDERS.antigravity);
|
|
107
|
+
}
|
|
108
|
+
|
|
109
|
+
buildUrl(model, stream, urlIndex = 0) {
|
|
110
|
+
const baseUrls = this.getBaseUrls();
|
|
111
|
+
const baseUrl = baseUrls[urlIndex] || baseUrls[0];
|
|
112
|
+
|
|
113
|
+
const forceNonStream = isImageModel(model);
|
|
114
|
+
const action = (stream && !forceNonStream) ? "streamGenerateContent?alt=sse" : "generateContent";
|
|
115
|
+
return `${baseUrl}/v1internal:${action}`;
|
|
116
|
+
}
|
|
117
|
+
|
|
118
|
+
buildHeaders(credentials, stream = true, sessionId = null) {
|
|
119
|
+
return {
|
|
120
|
+
"Content-Type": "application/json",
|
|
121
|
+
"Authorization": `Bearer ${credentials.accessToken}`,
|
|
122
|
+
"User-Agent": this.config.headers?.["User-Agent"] || ANTIGRAVITY_HEADERS["User-Agent"],
|
|
123
|
+
};
|
|
124
|
+
}
|
|
125
|
+
|
|
126
|
+
transformRequest(model, body, stream, credentials) {
|
|
127
|
+
const projectId = credentials?.projectId || this.generateProjectId();
|
|
128
|
+
|
|
129
|
+
if (stream !== true) delete body.stream_options;
|
|
130
|
+
|
|
131
|
+
if (isImageModel(model)) {
|
|
132
|
+
const imageConfig = parseImageConfig(model);
|
|
133
|
+
|
|
134
|
+
const cleanModel = model.replace(/-(\d+)x(\d+)$/, "");
|
|
135
|
+
|
|
136
|
+
const contents = [];
|
|
137
|
+
const srcContents = body.request?.contents || body.contents || [];
|
|
138
|
+
for (const c of srcContents) {
|
|
139
|
+
const textParts = (c.parts || []).filter(p => p.text !== undefined).map(p => ({ text: p.text }));
|
|
140
|
+
if (textParts.length > 0) {
|
|
141
|
+
contents.push({ role: c.role || "user", parts: textParts });
|
|
142
|
+
}
|
|
143
|
+
}
|
|
144
|
+
|
|
145
|
+
const sessionId = resolveSessionId({
|
|
146
|
+
headers: credentials?.rawHeaders,
|
|
147
|
+
body,
|
|
148
|
+
connectionId: credentials?.email || credentials?.connectionId,
|
|
149
|
+
scope: "antigravity",
|
|
150
|
+
});
|
|
151
|
+
|
|
152
|
+
this._lastSessionId = sessionId;
|
|
153
|
+
const request = {
|
|
154
|
+
contents,
|
|
155
|
+
generationConfig: {
|
|
156
|
+
temperature: 1.0,
|
|
157
|
+
topP: 0.95,
|
|
158
|
+
topK: 40,
|
|
159
|
+
maxOutputTokens: 8192,
|
|
160
|
+
imageConfig,
|
|
161
|
+
},
|
|
162
|
+
sessionId,
|
|
163
|
+
|
|
164
|
+
};
|
|
165
|
+
|
|
166
|
+
return {
|
|
167
|
+
project: projectId,
|
|
168
|
+
model: cleanModel,
|
|
169
|
+
userAgent: "antigravity",
|
|
170
|
+
requestType: "image_gen",
|
|
171
|
+
requestId: buildIdeRequestId({ body, request, credentials, model: cleanModel, requestType: "image_gen" }),
|
|
172
|
+
request,
|
|
173
|
+
};
|
|
174
|
+
}
|
|
175
|
+
|
|
176
|
+
const contents = body.request?.contents?.map(c => {
|
|
177
|
+
let role = c.role;
|
|
178
|
+
|
|
179
|
+
if (c.parts?.some(p => p.functionResponse)) {
|
|
180
|
+
role = "user";
|
|
181
|
+
}
|
|
182
|
+
|
|
183
|
+
const parts = c.parts?.filter(p => {
|
|
184
|
+
if (p.thought && !p.functionCall) return false;
|
|
185
|
+
if (p.thoughtSignature && !p.functionCall && !p.text) return false;
|
|
186
|
+
return true;
|
|
187
|
+
});
|
|
188
|
+
|
|
189
|
+
const needsBackfill = parts?.some(p => p.functionCall && !p.thoughtSignature) ?? false;
|
|
190
|
+
if (role !== c.role || parts?.length !== c.parts?.length || needsBackfill) {
|
|
191
|
+
return {
|
|
192
|
+
...c, role,
|
|
193
|
+
parts: needsBackfill
|
|
194
|
+
? parts.map(p => (p.functionCall && !p.thoughtSignature)
|
|
195
|
+
? { ...p, thoughtSignature: DEFAULT_THINKING_AG_SIGNATURE }
|
|
196
|
+
: p)
|
|
197
|
+
: parts,
|
|
198
|
+
};
|
|
199
|
+
}
|
|
200
|
+
return c;
|
|
201
|
+
});
|
|
202
|
+
|
|
203
|
+
let tools = body.request?.tools;
|
|
204
|
+
|
|
205
|
+
if (tools && tools.length > 0) {
|
|
206
|
+
|
|
207
|
+
const seenToolNames = new Set();
|
|
208
|
+
const allDeclarations = [];
|
|
209
|
+
for (const group of tools) {
|
|
210
|
+
for (const fn of group.functionDeclarations || []) {
|
|
211
|
+
const name = sanitizeFunctionName(fn.name);
|
|
212
|
+
if (seenToolNames.has(name)) continue;
|
|
213
|
+
seenToolNames.add(name);
|
|
214
|
+
allDeclarations.push({
|
|
215
|
+
...fn,
|
|
216
|
+
name,
|
|
217
|
+
parameters: fn.parameters
|
|
218
|
+
? cleanJSONSchemaForAntigravity(structuredClone(fn.parameters))
|
|
219
|
+
: { type: "object", properties: { reason: { type: "string", description: "Brief explanation" } }, required: ["reason"] }
|
|
220
|
+
});
|
|
221
|
+
}
|
|
222
|
+
}
|
|
223
|
+
tools = allDeclarations.length > 0 ? [{ functionDeclarations: allDeclarations }] : [];
|
|
224
|
+
}
|
|
225
|
+
|
|
226
|
+
const { tools: _originalTools, toolConfig: _originalToolConfig, ...requestWithoutTools } = body.request || {};
|
|
227
|
+
stripBlacklisted(requestWithoutTools);
|
|
228
|
+
const generationConfig = { ...(requestWithoutTools.generationConfig || {}) };
|
|
229
|
+
if (generationConfig.maxOutputTokens > MAX_ANTIGRAVITY_OUTPUT_TOKENS) {
|
|
230
|
+
generationConfig.maxOutputTokens = MAX_ANTIGRAVITY_OUTPUT_TOKENS;
|
|
231
|
+
}
|
|
232
|
+
|
|
233
|
+
const transformedRequest = {
|
|
234
|
+
...requestWithoutTools,
|
|
235
|
+
generationConfig,
|
|
236
|
+
...(contents && { contents }),
|
|
237
|
+
...(tools && { tools }),
|
|
238
|
+
sessionId: body.request?.sessionId || resolveSessionId({ headers: credentials?.rawHeaders, body, connectionId: credentials?.email || credentials?.connectionId, scope: "antigravity" }),
|
|
239
|
+
safetySettings: undefined,
|
|
240
|
+
...(tools?.length > 0 && { toolConfig: { functionCallingConfig: { mode: "VALIDATED" } } })
|
|
241
|
+
};
|
|
242
|
+
|
|
243
|
+
stripBlacklisted(body);
|
|
244
|
+
|
|
245
|
+
this._lastSessionId = transformedRequest.sessionId;
|
|
246
|
+
|
|
247
|
+
return {
|
|
248
|
+
...body,
|
|
249
|
+
project: projectId,
|
|
250
|
+
model: body.model || model,
|
|
251
|
+
userAgent: "antigravity",
|
|
252
|
+
requestType: "agent",
|
|
253
|
+
requestId: buildIdeRequestId({ body, request: transformedRequest, credentials, model, requestType: "agent" }),
|
|
254
|
+
request: transformedRequest
|
|
255
|
+
};
|
|
256
|
+
}
|
|
257
|
+
|
|
258
|
+
async refreshCredentials(credentials, log, proxyOptions = null) {
|
|
259
|
+
if (!credentials.refreshToken) return null;
|
|
260
|
+
|
|
261
|
+
try {
|
|
262
|
+
const response = await proxyAwareFetch(OAUTH_ENDPOINTS.google.token, {
|
|
263
|
+
method: "POST",
|
|
264
|
+
headers: { "Content-Type": "application/x-www-form-urlencoded", "Accept": "application/json" },
|
|
265
|
+
body: new URLSearchParams({
|
|
266
|
+
grant_type: "refresh_token",
|
|
267
|
+
refresh_token: credentials.refreshToken,
|
|
268
|
+
client_id: this.config.clientId,
|
|
269
|
+
client_secret: this.config.clientSecret
|
|
270
|
+
})
|
|
271
|
+
}, proxyOptions);
|
|
272
|
+
|
|
273
|
+
if (!response.ok) return null;
|
|
274
|
+
|
|
275
|
+
const tokens = await response.json();
|
|
276
|
+
log?.info?.("TOKEN", "Antigravity refreshed");
|
|
277
|
+
|
|
278
|
+
return {
|
|
279
|
+
accessToken: tokens.access_token,
|
|
280
|
+
refreshToken: tokens.refresh_token || credentials.refreshToken,
|
|
281
|
+
expiresIn: tokens.expires_in,
|
|
282
|
+
projectId: credentials.projectId
|
|
283
|
+
};
|
|
284
|
+
} catch (error) {
|
|
285
|
+
log?.error?.("TOKEN", `Antigravity refresh error: ${error.message}`);
|
|
286
|
+
return null;
|
|
287
|
+
}
|
|
288
|
+
}
|
|
289
|
+
|
|
290
|
+
generateProjectId() {
|
|
291
|
+
const adj = ["useful", "bright", "swift", "calm", "bold"][Math.floor(Math.random() * 5)];
|
|
292
|
+
const noun = ["fuze", "wave", "spark", "flow", "core"][Math.floor(Math.random() * 5)];
|
|
293
|
+
return `${adj}-${noun}-${crypto.randomUUID().slice(0, 5)}`;
|
|
294
|
+
}
|
|
295
|
+
|
|
296
|
+
generateSessionId() {
|
|
297
|
+
return crypto.randomUUID() + Date.now().toString();
|
|
298
|
+
}
|
|
299
|
+
|
|
300
|
+
parseRetryHeaders(headers) {
|
|
301
|
+
if (!headers?.get) return null;
|
|
302
|
+
|
|
303
|
+
const retryAfter = headers.get('retry-after');
|
|
304
|
+
if (retryAfter) {
|
|
305
|
+
const seconds = parseInt(retryAfter, 10);
|
|
306
|
+
if (!isNaN(seconds) && seconds > 0) return seconds * 1000;
|
|
307
|
+
|
|
308
|
+
const date = new Date(retryAfter);
|
|
309
|
+
if (!isNaN(date.getTime())) {
|
|
310
|
+
const diff = date.getTime() - Date.now();
|
|
311
|
+
return diff > 0 ? diff : null;
|
|
312
|
+
}
|
|
313
|
+
}
|
|
314
|
+
|
|
315
|
+
const resetAfter = headers.get('x-ratelimit-reset-after');
|
|
316
|
+
if (resetAfter) {
|
|
317
|
+
const seconds = parseInt(resetAfter, 10);
|
|
318
|
+
if (!isNaN(seconds) && seconds > 0) return seconds * 1000;
|
|
319
|
+
}
|
|
320
|
+
|
|
321
|
+
const resetTimestamp = headers.get('x-ratelimit-reset');
|
|
322
|
+
if (resetTimestamp) {
|
|
323
|
+
const ts = parseInt(resetTimestamp, 10) * 1000;
|
|
324
|
+
const diff = ts - Date.now();
|
|
325
|
+
return diff > 0 ? diff : null;
|
|
326
|
+
}
|
|
327
|
+
|
|
328
|
+
return null;
|
|
329
|
+
}
|
|
330
|
+
|
|
331
|
+
parseRetryFromErrorMessage(errorMessage) {
|
|
332
|
+
if (!errorMessage || typeof errorMessage !== "string") return null;
|
|
333
|
+
|
|
334
|
+
const match = errorMessage.match(/reset after (\d+h)?(\d+m)?(\d+s)?/i);
|
|
335
|
+
if (!match) return null;
|
|
336
|
+
|
|
337
|
+
let totalMs = 0;
|
|
338
|
+
if (match[1]) totalMs += parseInt(match[1]) * 3600 * 1000;
|
|
339
|
+
if (match[2]) totalMs += parseInt(match[2]) * 60 * 1000;
|
|
340
|
+
if (match[3]) totalMs += parseInt(match[3]) * 1000;
|
|
341
|
+
|
|
342
|
+
return totalMs > 0 ? totalMs : null;
|
|
343
|
+
}
|
|
344
|
+
|
|
345
|
+
extractErrorMessage(errorJson, bodyText = "") {
|
|
346
|
+
return [
|
|
347
|
+
errorJson?.error?.message,
|
|
348
|
+
errorJson?.message,
|
|
349
|
+
errorJson?.error,
|
|
350
|
+
bodyText,
|
|
351
|
+
].filter(Boolean).map(v => typeof v === "string" ? v : JSON.stringify(v)).join("\n");
|
|
352
|
+
}
|
|
353
|
+
|
|
354
|
+
isTransientAntigravityError(status, message) {
|
|
355
|
+
if (status === HTTP_STATUS.RATE_LIMITED) return true;
|
|
356
|
+
if (ANTIGRAVITY_TRANSIENT_STATUSES.has(status)) return true;
|
|
357
|
+
return ANTIGRAVITY_TRANSIENT_ERROR_PATTERNS.some(pattern => pattern.test(message || ""));
|
|
358
|
+
}
|
|
359
|
+
|
|
360
|
+
async computeRetryDelay(response, attempt) {
|
|
361
|
+
let bodyText = "";
|
|
362
|
+
let errorJson = null;
|
|
363
|
+
let retryMs = this.parseRetryHeaders(response.headers);
|
|
364
|
+
|
|
365
|
+
try {
|
|
366
|
+
bodyText = await response.clone().text();
|
|
367
|
+
errorJson = bodyText ? JSON.parse(bodyText) : null;
|
|
368
|
+
} catch {
|
|
369
|
+
|
|
370
|
+
}
|
|
371
|
+
|
|
372
|
+
const errorMessage = this.extractErrorMessage(errorJson, bodyText);
|
|
373
|
+
|
|
374
|
+
if (!retryMs) {
|
|
375
|
+
retryMs = this.parseRetryFromErrorMessage(errorMessage);
|
|
376
|
+
}
|
|
377
|
+
if (retryMs) return retryMs <= MAX_RETRY_AFTER_MS ? retryMs : false;
|
|
378
|
+
|
|
379
|
+
if (!this.isTransientAntigravityError(response.status, errorMessage)) return false;
|
|
380
|
+
|
|
381
|
+
const cap = response.status === HTTP_STATUS.RATE_LIMITED
|
|
382
|
+
? MAX_RETRY_AFTER_MS
|
|
383
|
+
: ANTIGRAVITY_TRANSIENT_RETRY_MAX_MS;
|
|
384
|
+
return Math.min(1000 * (2 ** attempt), cap);
|
|
385
|
+
}
|
|
386
|
+
|
|
387
|
+
}
|
|
388
|
+
|
|
389
|
+
export default AntigravityExecutor;
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
import { DefaultExecutor } from "./default.js";
|
|
2
|
+
|
|
3
|
+
function pick(value, fallback) {
|
|
4
|
+
return value || fallback;
|
|
5
|
+
}
|
|
6
|
+
|
|
7
|
+
export class AzureExecutor extends DefaultExecutor {
|
|
8
|
+
constructor() {
|
|
9
|
+
super("azure");
|
|
10
|
+
}
|
|
11
|
+
|
|
12
|
+
buildUrl(model, _stream, _urlIndex = 0, credentials = null) {
|
|
13
|
+
const providerData = credentials?.providerSpecificData;
|
|
14
|
+
const azureEndpoint = pick(providerData?.azureEndpoint, pick(process.env.AZURE_ENDPOINT, "https://api.openai.com"));
|
|
15
|
+
const apiVersion = pick(providerData?.apiVersion, pick(process.env.AZURE_API_VERSION, "2024-10-01-preview"));
|
|
16
|
+
const deployment = pick(providerData?.deployment, pick(model, pick(process.env.AZURE_DEPLOYMENT, "gpt-4")));
|
|
17
|
+
const endpoint = azureEndpoint.replace(/\/$/, "");
|
|
18
|
+
return `${endpoint}/openai/deployments/${deployment}/chat/completions?api-version=${apiVersion}`;
|
|
19
|
+
}
|
|
20
|
+
|
|
21
|
+
buildHeaders(credentials, stream = true) {
|
|
22
|
+
const headers = {
|
|
23
|
+
"Content-Type": "application/json",
|
|
24
|
+
...this.config.headers
|
|
25
|
+
};
|
|
26
|
+
|
|
27
|
+
const apiKey = pick(credentials?.apiKey, pick(credentials?.accessToken, process.env.OPENAI_API_KEY));
|
|
28
|
+
|
|
29
|
+
if (apiKey) {
|
|
30
|
+
headers["api-key"] = apiKey;
|
|
31
|
+
}
|
|
32
|
+
|
|
33
|
+
const organization = pick(credentials?.providerSpecificData?.organization, process.env.AZURE_ORGANIZATION);
|
|
34
|
+
|
|
35
|
+
if (organization) {
|
|
36
|
+
headers["OpenAI-Organization"] = organization;
|
|
37
|
+
}
|
|
38
|
+
|
|
39
|
+
if (stream) {
|
|
40
|
+
headers["Accept"] = "text/event-stream";
|
|
41
|
+
}
|
|
42
|
+
|
|
43
|
+
return headers;
|
|
44
|
+
}
|
|
45
|
+
|
|
46
|
+
transformRequest(_model, body, _stream, _credentials) {
|
|
47
|
+
return body;
|
|
48
|
+
}
|
|
49
|
+
}
|
|
@@ -0,0 +1,225 @@
|
|
|
1
|
+
import { HTTP_STATUS, RETRY_CONFIG, DEFAULT_RETRY_CONFIG, resolveRetryEntry, FETCH_CONNECT_TIMEOUT_MS } from "../config/runtimeConfig.js";
|
|
2
|
+
import { shouldRefreshCredentials } from "../services/oauthCredentialManager.js";
|
|
3
|
+
import { proxyAwareFetch } from "../utils/proxyFetch.js";
|
|
4
|
+
import { dbg } from "../utils/debugLog.js";
|
|
5
|
+
import { ANTHROPIC_API_VERSION, OPENAI_COMPAT_BASE, ANTHROPIC_COMPAT_BASE } from "../providers/shared.js";
|
|
6
|
+
import { resolveOpenAICompatibleApiType } from "../services/provider.js";
|
|
7
|
+
import { env } from "@/lib/env";
|
|
8
|
+
|
|
9
|
+
export function waitForRetry(delayMs, signal) {
|
|
10
|
+
if (!delayMs || delayMs <= 0) {
|
|
11
|
+
if (signal?.aborted) return Promise.reject(signal.reason || new DOMException("The operation was aborted", "AbortError"));
|
|
12
|
+
return Promise.resolve();
|
|
13
|
+
}
|
|
14
|
+
return new Promise((resolve, reject) => {
|
|
15
|
+
let timer = null;
|
|
16
|
+
const onAbort = () => {
|
|
17
|
+
if (timer) clearTimeout(timer);
|
|
18
|
+
signal?.removeEventListener("abort", onAbort);
|
|
19
|
+
reject(signal.reason || new DOMException("The operation was aborted", "AbortError"));
|
|
20
|
+
};
|
|
21
|
+
timer = setTimeout(() => {
|
|
22
|
+
signal?.removeEventListener("abort", onAbort);
|
|
23
|
+
resolve();
|
|
24
|
+
}, delayMs);
|
|
25
|
+
if (signal) {
|
|
26
|
+
if (signal.aborted) onAbort();
|
|
27
|
+
else signal.addEventListener("abort", onAbort, { once: true });
|
|
28
|
+
}
|
|
29
|
+
});
|
|
30
|
+
}
|
|
31
|
+
|
|
32
|
+
async function cancelResponseBody(response) {
|
|
33
|
+
try {
|
|
34
|
+
if (response?.body?.cancel) await response.body.cancel();
|
|
35
|
+
} catch {
|
|
36
|
+
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
export class BaseExecutor {
|
|
41
|
+
constructor(provider, config) {
|
|
42
|
+
this.provider = provider;
|
|
43
|
+
this.config = config;
|
|
44
|
+
this.noAuth = config?.noAuth || false;
|
|
45
|
+
}
|
|
46
|
+
|
|
47
|
+
getProvider() {
|
|
48
|
+
return this.provider;
|
|
49
|
+
}
|
|
50
|
+
|
|
51
|
+
getBaseUrls() {
|
|
52
|
+
return this.config.baseUrls || (this.config.baseUrl ? [this.config.baseUrl] : []);
|
|
53
|
+
}
|
|
54
|
+
|
|
55
|
+
getFallbackCount() {
|
|
56
|
+
return this.getBaseUrls().length || 1;
|
|
57
|
+
}
|
|
58
|
+
|
|
59
|
+
buildUrl(model, stream, urlIndex = 0, credentials = null) {
|
|
60
|
+
if (this.provider?.startsWith?.("openai-compatible-")) {
|
|
61
|
+
const baseUrl = credentials?.providerSpecificData?.baseUrl || OPENAI_COMPAT_BASE;
|
|
62
|
+
const normalized = baseUrl.replace(/\/$/, "");
|
|
63
|
+
const path = resolveOpenAICompatibleApiType(this.provider, credentials) === "responses" ? "/responses" : "/chat/completions";
|
|
64
|
+
return `${normalized}${path}`;
|
|
65
|
+
}
|
|
66
|
+
if (this.provider?.startsWith?.("anthropic-compatible-")) {
|
|
67
|
+
const baseUrl = credentials?.providerSpecificData?.baseUrl || ANTHROPIC_COMPAT_BASE;
|
|
68
|
+
const normalized = baseUrl.replace(/\/$/, "");
|
|
69
|
+
return `${normalized}/messages`;
|
|
70
|
+
}
|
|
71
|
+
const baseUrls = this.getBaseUrls();
|
|
72
|
+
return baseUrls[urlIndex] || baseUrls[0] || this.config.baseUrl;
|
|
73
|
+
}
|
|
74
|
+
|
|
75
|
+
buildHeaders(credentials, stream = true) {
|
|
76
|
+
const headers = {
|
|
77
|
+
"Content-Type": "application/json",
|
|
78
|
+
...this.config.headers,
|
|
79
|
+
};
|
|
80
|
+
|
|
81
|
+
if (this.provider?.startsWith?.("anthropic-compatible-")) {
|
|
82
|
+
|
|
83
|
+
if (credentials.apiKey) {
|
|
84
|
+
headers["x-api-key"] = credentials.apiKey;
|
|
85
|
+
} else if (credentials.accessToken) {
|
|
86
|
+
headers["Authorization"] = `Bearer ${credentials.accessToken}`;
|
|
87
|
+
}
|
|
88
|
+
if (!headers["anthropic-version"]) {
|
|
89
|
+
headers["anthropic-version"] = ANTHROPIC_API_VERSION;
|
|
90
|
+
}
|
|
91
|
+
} else {
|
|
92
|
+
|
|
93
|
+
if (credentials.accessToken) {
|
|
94
|
+
headers["Authorization"] = `Bearer ${credentials.accessToken}`;
|
|
95
|
+
} else if (credentials.apiKey) {
|
|
96
|
+
headers["Authorization"] = `Bearer ${credentials.apiKey}`;
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
|
|
100
|
+
if (stream) {
|
|
101
|
+
headers["Accept"] = "text/event-stream";
|
|
102
|
+
}
|
|
103
|
+
|
|
104
|
+
return headers;
|
|
105
|
+
}
|
|
106
|
+
|
|
107
|
+
transformRequest(model, body, stream, credentials) {
|
|
108
|
+
return body;
|
|
109
|
+
}
|
|
110
|
+
|
|
111
|
+
shouldRetry(status, urlIndex) {
|
|
112
|
+
return status === HTTP_STATUS.RATE_LIMITED && urlIndex + 1 < this.getFallbackCount();
|
|
113
|
+
}
|
|
114
|
+
|
|
115
|
+
async refreshCredentials(credentials, log, proxyOptions = null) {
|
|
116
|
+
return null;
|
|
117
|
+
}
|
|
118
|
+
|
|
119
|
+
needsRefresh(credentials) {
|
|
120
|
+
return shouldRefreshCredentials(this.provider, credentials);
|
|
121
|
+
}
|
|
122
|
+
|
|
123
|
+
parseError(response, bodyText) {
|
|
124
|
+
return { status: response.status, message: bodyText || `HTTP ${response.status}` };
|
|
125
|
+
}
|
|
126
|
+
|
|
127
|
+
async execute({ model, body, stream, credentials, signal, log, proxyOptions = null, requestTag = "", attemptReason = "initial", maxTotalAttempts = env.upstreamMaxAttempts }) {
|
|
128
|
+
const fallbackCount = this.getFallbackCount();
|
|
129
|
+
const totalAttemptLimit = Math.max(1, Number(maxTotalAttempts) || env.upstreamMaxAttempts || 8);
|
|
130
|
+
let attempt = 0;
|
|
131
|
+
let lastError = null;
|
|
132
|
+
let lastStatus = 0;
|
|
133
|
+
const retryAttemptsByUrl = {};
|
|
134
|
+
|
|
135
|
+
const retryConfig = { ...DEFAULT_RETRY_CONFIG, ...this.config.retry };
|
|
136
|
+
|
|
137
|
+
const tryRetry = async (urlIndex, statusKey, reason, response = null) => {
|
|
138
|
+
const { attempts, delayMs } = resolveRetryEntry(retryConfig[statusKey]);
|
|
139
|
+
if (attempt >= totalAttemptLimit || attempts <= 0 || retryAttemptsByUrl[urlIndex] >= attempts) return false;
|
|
140
|
+
|
|
141
|
+
let waitMs = delayMs;
|
|
142
|
+
if (response && this.computeRetryDelay) {
|
|
143
|
+
const dynamic = await this.computeRetryDelay(response, retryAttemptsByUrl[urlIndex] + 1, delayMs);
|
|
144
|
+
if (dynamic === false) return false;
|
|
145
|
+
if (dynamic != null) waitMs = dynamic;
|
|
146
|
+
}
|
|
147
|
+
retryAttemptsByUrl[urlIndex]++;
|
|
148
|
+
log?.debug?.("RETRY", `${reason} retry ${retryAttemptsByUrl[urlIndex]}/${attempts} after ${waitMs / 1000}s`);
|
|
149
|
+
await waitForRetry(waitMs, signal);
|
|
150
|
+
return true;
|
|
151
|
+
};
|
|
152
|
+
|
|
153
|
+
for (let urlIndex = 0; urlIndex < fallbackCount; urlIndex++) {
|
|
154
|
+
const url = this.buildUrl(model, stream, urlIndex, credentials);
|
|
155
|
+
const transformedBody = this.transformRequest(model, body, stream, credentials);
|
|
156
|
+
const headers = this.buildHeaders(credentials, stream, url, model);
|
|
157
|
+
|
|
158
|
+
if (!retryAttemptsByUrl[urlIndex]) retryAttemptsByUrl[urlIndex] = 0;
|
|
159
|
+
|
|
160
|
+
const connectCtrl = new AbortController();
|
|
161
|
+
const timeoutMs = this.config?.timeoutMs || FETCH_CONNECT_TIMEOUT_MS;
|
|
162
|
+
const connectTimer = setTimeout(() => connectCtrl.abort(new Error("fetch connect timeout")), timeoutMs);
|
|
163
|
+
const mergedSignal = signal ? AbortSignal.any([signal, connectCtrl.signal]) : connectCtrl.signal;
|
|
164
|
+
|
|
165
|
+
try {
|
|
166
|
+
attempt++;
|
|
167
|
+
const bodyStr = JSON.stringify(transformedBody);
|
|
168
|
+
const fetchT0 = Date.now();
|
|
169
|
+
const attemptMeta = `attempt=${attempt} · reason=${attemptReason}`;
|
|
170
|
+
let upstreamHost = url;
|
|
171
|
+
try { upstreamHost = new URL(url).host || url; } catch { }
|
|
172
|
+
log?.line?.(requestTag, "FETCH", `${this.provider.toUpperCase()} → ${upstreamHost} · ${attemptMeta}`);
|
|
173
|
+
dbg("FETCH", `${this.provider.toUpperCase()} → ${url} | body=${bodyStr.length}B | connectTimeout=${timeoutMs}ms | ${attemptMeta}`);
|
|
174
|
+
const response = await proxyAwareFetch(url, {
|
|
175
|
+
method: "POST",
|
|
176
|
+
headers,
|
|
177
|
+
body: bodyStr,
|
|
178
|
+
signal: mergedSignal
|
|
179
|
+
}, proxyOptions);
|
|
180
|
+
clearTimeout(connectTimer);
|
|
181
|
+
const ct = response.headers?.get?.("content-type") || "";
|
|
182
|
+
const cl = response.headers?.get?.("content-length") || "?";
|
|
183
|
+
const responseLabel = response.status >= 400 ? "ERROR" : "RESPONSE";
|
|
184
|
+
log?.line?.(requestTag, responseLabel, `${this.provider.toUpperCase()} ${response.status} · ${Date.now() - fetchT0}ms · ${ct || "unknown"} · cl=${cl} · attempt=${attempt}`);
|
|
185
|
+
dbg("FETCH", `${this.provider.toUpperCase()} ← ${response.status} | ttft=${Date.now() - fetchT0}ms | ct=${ct} | cl=${cl} | attempt=${attempt}`);
|
|
186
|
+
|
|
187
|
+
if (await tryRetry(urlIndex, response.status, `status ${response.status}`, response)) {
|
|
188
|
+
await cancelResponseBody(response);
|
|
189
|
+
urlIndex--;
|
|
190
|
+
continue;
|
|
191
|
+
}
|
|
192
|
+
|
|
193
|
+
if (attempt < totalAttemptLimit && this.shouldRetry(response.status, urlIndex)) {
|
|
194
|
+
log?.debug?.("RETRY", `${response.status} on ${url}, trying fallback ${urlIndex + 1}`);
|
|
195
|
+
await cancelResponseBody(response);
|
|
196
|
+
lastStatus = response.status;
|
|
197
|
+
continue;
|
|
198
|
+
}
|
|
199
|
+
|
|
200
|
+
return { response, url, headers, transformedBody };
|
|
201
|
+
} catch (error) {
|
|
202
|
+
clearTimeout(connectTimer);
|
|
203
|
+
|
|
204
|
+
if (signal?.aborted) throw signal.reason || error;
|
|
205
|
+
lastError = error;
|
|
206
|
+
const isConnectTimeout = connectCtrl.signal.aborted && error.name === "AbortError";
|
|
207
|
+
dbg("FETCH", `${this.provider.toUpperCase()} ✖ ${error.name}: ${error.message}${isConnectTimeout ? " (connect timeout)" : ""}`);
|
|
208
|
+
|
|
209
|
+
if (error.name === "AbortError" && !isConnectTimeout) throw error;
|
|
210
|
+
|
|
211
|
+
if (await tryRetry(urlIndex, HTTP_STATUS.BAD_GATEWAY, `network "${error.message}"`)) { urlIndex--; continue; }
|
|
212
|
+
|
|
213
|
+
if (attempt < totalAttemptLimit && urlIndex + 1 < fallbackCount) {
|
|
214
|
+
log?.debug?.("RETRY", `Error on ${url}, trying fallback ${urlIndex + 1}`);
|
|
215
|
+
continue;
|
|
216
|
+
}
|
|
217
|
+
throw error;
|
|
218
|
+
}
|
|
219
|
+
}
|
|
220
|
+
|
|
221
|
+
throw lastError || new Error(`All ${fallbackCount} URLs failed with status ${lastStatus}`);
|
|
222
|
+
}
|
|
223
|
+
}
|
|
224
|
+
|
|
225
|
+
export default BaseExecutor;
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import { DefaultExecutor } from "./default.js";
|
|
2
|
+
export class CodeBuddyExecutor extends DefaultExecutor {
|
|
3
|
+
constructor() {
|
|
4
|
+
super("codebuddy-cn");
|
|
5
|
+
}
|
|
6
|
+
|
|
7
|
+
transformRequest(model, body, stream, credentials) {
|
|
8
|
+
const transformed = super.transformRequest(model, body, stream, credentials);
|
|
9
|
+
transformed.stream = true;
|
|
10
|
+
|
|
11
|
+
const eff = transformed.reasoning_effort;
|
|
12
|
+
if (eff === "none" || eff === "off") {
|
|
13
|
+
delete transformed.reasoning_effort;
|
|
14
|
+
} else if (eff) {
|
|
15
|
+
|
|
16
|
+
transformed.reasoning_summary = "auto";
|
|
17
|
+
}
|
|
18
|
+
|
|
19
|
+
return transformed;
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
|
|
23
|
+
export default CodeBuddyExecutor;
|