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,380 @@
|
|
|
1
|
+
import { readFileSync } from "node:fs";
|
|
2
|
+
import { env } from "@/lib/env";
|
|
3
|
+
import {
|
|
4
|
+
routerActiveRequests,
|
|
5
|
+
routerQueuedRequests,
|
|
6
|
+
routerConcurrencyLimit,
|
|
7
|
+
routerBodyBytesInUse,
|
|
8
|
+
routerAdmissionRejectsTotal,
|
|
9
|
+
} from "@/observability/metrics.js";
|
|
10
|
+
|
|
11
|
+
const MIB = 1024 * 1024;
|
|
12
|
+
|
|
13
|
+
export type AdmissionReason = "queue_full" | "timeout" | "aborted" | "stopping" | "body_budget";
|
|
14
|
+
|
|
15
|
+
export interface AdmissionLease {
|
|
16
|
+
release(): void;
|
|
17
|
+
releaseBody(): void;
|
|
18
|
+
}
|
|
19
|
+
|
|
20
|
+
export interface AdmissionResult {
|
|
21
|
+
ok: boolean;
|
|
22
|
+
reason?: AdmissionReason;
|
|
23
|
+
lease?: AdmissionLease;
|
|
24
|
+
}
|
|
25
|
+
|
|
26
|
+
export interface RouterAdmissionOptions {
|
|
27
|
+
hardMaxConcurrent?: number;
|
|
28
|
+
initialConcurrent?: number;
|
|
29
|
+
minConcurrent?: number;
|
|
30
|
+
maxQueue?: number;
|
|
31
|
+
queueWaitMs?: number;
|
|
32
|
+
bodyBudgetBytes?: number;
|
|
33
|
+
memoryHighWaterBytes?: number;
|
|
34
|
+
memoryCriticalWaterBytes?: number;
|
|
35
|
+
sampleIntervalMs?: number;
|
|
36
|
+
autoSample?: boolean;
|
|
37
|
+
}
|
|
38
|
+
|
|
39
|
+
interface Waiter {
|
|
40
|
+
bodyBytes: number;
|
|
41
|
+
workUnits: number;
|
|
42
|
+
resolve: (result: AdmissionResult) => void;
|
|
43
|
+
timer: ReturnType<typeof setTimeout> | null;
|
|
44
|
+
abortHandler: (() => void) | null;
|
|
45
|
+
signal: AbortSignal | null;
|
|
46
|
+
settled: boolean;
|
|
47
|
+
}
|
|
48
|
+
|
|
49
|
+
function positiveInt(value: unknown, fallback: number, max = 1_000_000): number {
|
|
50
|
+
const parsed = Number(value);
|
|
51
|
+
if (!Number.isFinite(parsed) || parsed < 1) return fallback;
|
|
52
|
+
return Math.min(Math.floor(parsed), max);
|
|
53
|
+
}
|
|
54
|
+
|
|
55
|
+
function clampBodyBytes(value: number, budget: number): number {
|
|
56
|
+
if (!Number.isFinite(value) || value <= 0) return 0;
|
|
57
|
+
return Math.min(Math.floor(value), budget);
|
|
58
|
+
}
|
|
59
|
+
|
|
60
|
+
function readContainerMemoryLimit(): number | null {
|
|
61
|
+
for (const path of ["/sys/fs/cgroup/memory.max", "/sys/fs/cgroup/memory/memory.limit_in_bytes"]) {
|
|
62
|
+
try {
|
|
63
|
+
const raw = readFileSync(path, "utf8").trim();
|
|
64
|
+
if (!raw || raw === "max") continue;
|
|
65
|
+
const value = Number(raw);
|
|
66
|
+
|
|
67
|
+
if (Number.isFinite(value) && value > 0 && value < 1024 * 1024 * 1024 * 1024) {
|
|
68
|
+
return value;
|
|
69
|
+
}
|
|
70
|
+
} catch {
|
|
71
|
+
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
return null;
|
|
75
|
+
}
|
|
76
|
+
|
|
77
|
+
export class RouterAdmission {
|
|
78
|
+
readonly hardMaxConcurrent: number;
|
|
79
|
+
readonly minConcurrent: number;
|
|
80
|
+
readonly maxQueue: number;
|
|
81
|
+
readonly queueWaitMs: number;
|
|
82
|
+
readonly bodyBudgetBytes: number;
|
|
83
|
+
readonly sampleIntervalMs: number;
|
|
84
|
+
|
|
85
|
+
private targetConcurrent: number;
|
|
86
|
+
private active = 0;
|
|
87
|
+
private activeWorkUnits = 0;
|
|
88
|
+
private bodyBytesInUse = 0;
|
|
89
|
+
private readonly waiters: Waiter[] = [];
|
|
90
|
+
private stopping = false;
|
|
91
|
+
private stableSamples = 0;
|
|
92
|
+
private lastEventLoopLagMs = 0;
|
|
93
|
+
private readonly baselineRss: number;
|
|
94
|
+
private readonly memoryLimitBytes: number | null;
|
|
95
|
+
private readonly memoryHighWaterBytes: number;
|
|
96
|
+
private readonly memoryCriticalWaterBytes: number;
|
|
97
|
+
private sampleTimer: ReturnType<typeof setTimeout> | null = null;
|
|
98
|
+
|
|
99
|
+
constructor(options: RouterAdmissionOptions = {}) {
|
|
100
|
+
this.hardMaxConcurrent = positiveInt(options.hardMaxConcurrent, env.routerMaxConcurrent, 100_000);
|
|
101
|
+
const initial = positiveInt(options.initialConcurrent, env.routerInitialConcurrent, 100_000);
|
|
102
|
+
this.minConcurrent = Math.min(
|
|
103
|
+
this.hardMaxConcurrent,
|
|
104
|
+
positiveInt(options.minConcurrent, env.routerMinConcurrent, 100_000),
|
|
105
|
+
);
|
|
106
|
+
this.targetConcurrent = Math.max(this.minConcurrent, Math.min(this.hardMaxConcurrent, initial));
|
|
107
|
+
this.maxQueue = positiveInt(options.maxQueue, env.routerMaxQueue, 1_000_000);
|
|
108
|
+
this.queueWaitMs = positiveInt(options.queueWaitMs, env.routerQueueWaitMs, 10 * 60 * 1000);
|
|
109
|
+
this.bodyBudgetBytes = positiveInt(options.bodyBudgetBytes, env.routerBodyBudgetBytes, 16 * 1024 * MIB);
|
|
110
|
+
this.sampleIntervalMs = positiveInt(options.sampleIntervalMs, 2000, 60_000);
|
|
111
|
+
|
|
112
|
+
this.baselineRss = this.readRss();
|
|
113
|
+
this.memoryLimitBytes = readContainerMemoryLimit();
|
|
114
|
+
const automaticHigh = this.memoryLimitBytes
|
|
115
|
+
? Math.floor(this.memoryLimitBytes * 0.70)
|
|
116
|
+
: this.baselineRss + (512 * MIB);
|
|
117
|
+
const automaticCritical = this.memoryLimitBytes
|
|
118
|
+
? Math.max(automaticHigh + 1, Math.floor(this.memoryLimitBytes * 0.85))
|
|
119
|
+
: this.baselineRss + (768 * MIB);
|
|
120
|
+
this.memoryHighWaterBytes = options.memoryHighWaterBytes
|
|
121
|
+
|| env.routerMemoryHighWaterBytes
|
|
122
|
+
|| automaticHigh;
|
|
123
|
+
this.memoryCriticalWaterBytes = options.memoryCriticalWaterBytes
|
|
124
|
+
|| env.routerMemoryCriticalWaterBytes
|
|
125
|
+
|| Math.max(automaticCritical, this.memoryHighWaterBytes + (128 * MIB));
|
|
126
|
+
|
|
127
|
+
this.updateMetrics();
|
|
128
|
+
if (options.autoSample !== false) this.armSampler();
|
|
129
|
+
}
|
|
130
|
+
|
|
131
|
+
private readRss(): number {
|
|
132
|
+
try {
|
|
133
|
+
return Number(process.memoryUsage?.().rss) || 0;
|
|
134
|
+
} catch {
|
|
135
|
+
return 0;
|
|
136
|
+
}
|
|
137
|
+
}
|
|
138
|
+
|
|
139
|
+
private updateMetrics(): void {
|
|
140
|
+
try {
|
|
141
|
+
routerActiveRequests.set({}, this.active);
|
|
142
|
+
routerQueuedRequests.set({}, this.waiters.length);
|
|
143
|
+
routerConcurrencyLimit.set({}, this.targetConcurrent);
|
|
144
|
+
routerBodyBytesInUse.set({}, this.bodyBytesInUse);
|
|
145
|
+
} catch {
|
|
146
|
+
|
|
147
|
+
}
|
|
148
|
+
}
|
|
149
|
+
|
|
150
|
+
private countReject(reason: AdmissionReason): void {
|
|
151
|
+
try { routerAdmissionRejectsTotal.inc({ reason }); } catch {}
|
|
152
|
+
}
|
|
153
|
+
|
|
154
|
+
private armSampler(): void {
|
|
155
|
+
if (this.sampleTimer) return;
|
|
156
|
+
const expectedAt = Date.now() + this.sampleIntervalMs;
|
|
157
|
+
this.sampleTimer = setTimeout(() => {
|
|
158
|
+
this.sampleTimer = null;
|
|
159
|
+
this.lastEventLoopLagMs = Math.max(0, Date.now() - expectedAt);
|
|
160
|
+
this.sample();
|
|
161
|
+
this.armSampler();
|
|
162
|
+
}, this.sampleIntervalMs);
|
|
163
|
+
this.sampleTimer?.unref?.();
|
|
164
|
+
}
|
|
165
|
+
|
|
166
|
+
sample(): void {
|
|
167
|
+
const rss = this.readRss();
|
|
168
|
+
const critical = rss >= this.memoryCriticalWaterBytes
|
|
169
|
+
|| this.bodyBytesInUse >= this.bodyBudgetBytes * 0.95
|
|
170
|
+
|| this.lastEventLoopLagMs >= 750;
|
|
171
|
+
const pressured = rss >= this.memoryHighWaterBytes
|
|
172
|
+
|| this.bodyBytesInUse >= this.bodyBudgetBytes * 0.80
|
|
173
|
+
|| this.lastEventLoopLagMs >= 250;
|
|
174
|
+
|
|
175
|
+
if (critical) {
|
|
176
|
+
this.targetConcurrent = Math.max(this.minConcurrent, Math.floor(this.targetConcurrent * 0.5));
|
|
177
|
+
this.stableSamples = 0;
|
|
178
|
+
} else if (pressured) {
|
|
179
|
+
this.targetConcurrent = Math.max(this.minConcurrent, Math.floor(this.targetConcurrent * 0.75));
|
|
180
|
+
this.stableSamples = 0;
|
|
181
|
+
} else if (this.waiters.length > 0 || this.activeWorkUnits >= this.targetConcurrent * 0.75) {
|
|
182
|
+
this.stableSamples += 1;
|
|
183
|
+
if (this.stableSamples >= 2) {
|
|
184
|
+
this.targetConcurrent = Math.min(
|
|
185
|
+
this.hardMaxConcurrent,
|
|
186
|
+
Math.max(this.targetConcurrent + 1, Math.ceil(this.targetConcurrent * 1.15)),
|
|
187
|
+
);
|
|
188
|
+
this.stableSamples = 0;
|
|
189
|
+
}
|
|
190
|
+
} else {
|
|
191
|
+
this.stableSamples = 0;
|
|
192
|
+
}
|
|
193
|
+
this.updateMetrics();
|
|
194
|
+
this.drain();
|
|
195
|
+
}
|
|
196
|
+
|
|
197
|
+
private canClaim(bodyBytes: number, workUnits: number): boolean {
|
|
198
|
+
return !this.stopping
|
|
199
|
+
&& this.activeWorkUnits + workUnits <= this.targetConcurrent
|
|
200
|
+
&& this.bodyBytesInUse + bodyBytes <= this.bodyBudgetBytes;
|
|
201
|
+
}
|
|
202
|
+
|
|
203
|
+
private claim(bodyBytes: number, workUnits: number): AdmissionLease {
|
|
204
|
+
this.active += 1;
|
|
205
|
+
this.activeWorkUnits += workUnits;
|
|
206
|
+
this.bodyBytesInUse += bodyBytes;
|
|
207
|
+
this.updateMetrics();
|
|
208
|
+
let released = false;
|
|
209
|
+
let bodyReleased = bodyBytes === 0;
|
|
210
|
+
|
|
211
|
+
return {
|
|
212
|
+
release: () => {
|
|
213
|
+
if (released) return;
|
|
214
|
+
released = true;
|
|
215
|
+
this.active = Math.max(0, this.active - 1);
|
|
216
|
+
this.activeWorkUnits = Math.max(0, this.activeWorkUnits - workUnits);
|
|
217
|
+
this.updateMetrics();
|
|
218
|
+
this.drain();
|
|
219
|
+
},
|
|
220
|
+
releaseBody: () => {
|
|
221
|
+
if (bodyReleased) return;
|
|
222
|
+
bodyReleased = true;
|
|
223
|
+
this.bodyBytesInUse = Math.max(0, this.bodyBytesInUse - bodyBytes);
|
|
224
|
+
this.updateMetrics();
|
|
225
|
+
this.drain();
|
|
226
|
+
},
|
|
227
|
+
};
|
|
228
|
+
}
|
|
229
|
+
|
|
230
|
+
private removeWaiter(waiter: Waiter): void {
|
|
231
|
+
const index = this.waiters.indexOf(waiter);
|
|
232
|
+
if (index >= 0) this.waiters.splice(index, 1);
|
|
233
|
+
if (waiter.timer) clearTimeout(waiter.timer);
|
|
234
|
+
waiter.timer = null;
|
|
235
|
+
if (waiter.signal && waiter.abortHandler) {
|
|
236
|
+
waiter.signal.removeEventListener("abort", waiter.abortHandler);
|
|
237
|
+
}
|
|
238
|
+
this.updateMetrics();
|
|
239
|
+
}
|
|
240
|
+
|
|
241
|
+
private settleWaiter(waiter: Waiter, result: AdmissionResult): void {
|
|
242
|
+
if (waiter.settled) return;
|
|
243
|
+
waiter.settled = true;
|
|
244
|
+
this.removeWaiter(waiter);
|
|
245
|
+
waiter.abortHandler = null;
|
|
246
|
+
waiter.resolve(result);
|
|
247
|
+
}
|
|
248
|
+
|
|
249
|
+
private drain(): void {
|
|
250
|
+
if (this.stopping) return;
|
|
251
|
+
while (this.activeWorkUnits < this.targetConcurrent && this.waiters.length > 0) {
|
|
252
|
+
|
|
253
|
+
const index = this.waiters.findIndex((waiter) => (
|
|
254
|
+
this.activeWorkUnits + waiter.workUnits <= this.targetConcurrent
|
|
255
|
+
&& this.bodyBytesInUse + waiter.bodyBytes <= this.bodyBudgetBytes
|
|
256
|
+
));
|
|
257
|
+
if (index < 0) return;
|
|
258
|
+
const waiter = this.waiters[index];
|
|
259
|
+
if (!waiter) return;
|
|
260
|
+
this.settleWaiter(waiter, { ok: true, lease: this.claim(waiter.bodyBytes, waiter.workUnits) });
|
|
261
|
+
}
|
|
262
|
+
}
|
|
263
|
+
|
|
264
|
+
acquire(bodyBytes = 0, signal: AbortSignal | null = null, workUnits = 1): Promise<AdmissionResult> {
|
|
265
|
+
const requestedBodyBytes = clampBodyBytes(bodyBytes, this.bodyBudgetBytes);
|
|
266
|
+
const requestedWorkUnits = positiveInt(workUnits, 1, this.hardMaxConcurrent);
|
|
267
|
+
if (bodyBytes > this.bodyBudgetBytes) {
|
|
268
|
+
this.countReject("body_budget");
|
|
269
|
+
return Promise.resolve({ ok: false, reason: "body_budget" });
|
|
270
|
+
}
|
|
271
|
+
if (this.stopping) {
|
|
272
|
+
this.countReject("stopping");
|
|
273
|
+
return Promise.resolve({ ok: false, reason: "stopping" });
|
|
274
|
+
}
|
|
275
|
+
if (signal?.aborted) {
|
|
276
|
+
this.countReject("aborted");
|
|
277
|
+
return Promise.resolve({ ok: false, reason: "aborted" });
|
|
278
|
+
}
|
|
279
|
+
if (this.canClaim(requestedBodyBytes, requestedWorkUnits)) {
|
|
280
|
+
return Promise.resolve({ ok: true, lease: this.claim(requestedBodyBytes, requestedWorkUnits) });
|
|
281
|
+
}
|
|
282
|
+
if (this.waiters.length >= this.maxQueue) {
|
|
283
|
+
this.countReject("queue_full");
|
|
284
|
+
return Promise.resolve({ ok: false, reason: "queue_full" });
|
|
285
|
+
}
|
|
286
|
+
|
|
287
|
+
return new Promise((resolve) => {
|
|
288
|
+
const waiter: Waiter = {
|
|
289
|
+
bodyBytes: requestedBodyBytes,
|
|
290
|
+
workUnits: requestedWorkUnits,
|
|
291
|
+
resolve,
|
|
292
|
+
timer: null,
|
|
293
|
+
abortHandler: null,
|
|
294
|
+
signal,
|
|
295
|
+
settled: false,
|
|
296
|
+
};
|
|
297
|
+
waiter.timer = setTimeout(() => this.settleWaiter(waiter, { ok: false, reason: "timeout" }), this.queueWaitMs);
|
|
298
|
+
waiter.abortHandler = () => this.settleWaiter(waiter, { ok: false, reason: "aborted" });
|
|
299
|
+
signal?.addEventListener("abort", waiter.abortHandler, { once: true });
|
|
300
|
+
this.waiters.push(waiter);
|
|
301
|
+
this.updateMetrics();
|
|
302
|
+
});
|
|
303
|
+
}
|
|
304
|
+
|
|
305
|
+
stopAccepting(): void {
|
|
306
|
+
this.stopping = true;
|
|
307
|
+
for (const waiter of [...this.waiters]) {
|
|
308
|
+
this.settleWaiter(waiter, { ok: false, reason: "stopping" });
|
|
309
|
+
}
|
|
310
|
+
if (this.sampleTimer) clearTimeout(this.sampleTimer);
|
|
311
|
+
this.sampleTimer = null;
|
|
312
|
+
this.updateMetrics();
|
|
313
|
+
}
|
|
314
|
+
|
|
315
|
+
snapshot() {
|
|
316
|
+
return {
|
|
317
|
+
active: this.active,
|
|
318
|
+
activeWorkUnits: this.activeWorkUnits,
|
|
319
|
+
queued: this.waiters.length,
|
|
320
|
+
bodyBytesInUse: this.bodyBytesInUse,
|
|
321
|
+
bodyBudgetBytes: this.bodyBudgetBytes,
|
|
322
|
+
currentLimit: this.targetConcurrent,
|
|
323
|
+
workUnitLimit: this.targetConcurrent,
|
|
324
|
+
hardMaxConcurrent: this.hardMaxConcurrent,
|
|
325
|
+
minConcurrent: this.minConcurrent,
|
|
326
|
+
rssBytes: this.readRss(),
|
|
327
|
+
baselineRssBytes: this.baselineRss,
|
|
328
|
+
memoryLimitBytes: this.memoryLimitBytes,
|
|
329
|
+
memoryHighWaterBytes: this.memoryHighWaterBytes,
|
|
330
|
+
memoryCriticalWaterBytes: this.memoryCriticalWaterBytes,
|
|
331
|
+
eventLoopLagMs: this.lastEventLoopLagMs,
|
|
332
|
+
stopping: this.stopping,
|
|
333
|
+
};
|
|
334
|
+
}
|
|
335
|
+
}
|
|
336
|
+
|
|
337
|
+
export const routerAdmission = new RouterAdmission();
|
|
338
|
+
|
|
339
|
+
export function wrapResponseWithAdmission(response: Response, release: () => void): Response {
|
|
340
|
+
let released = false;
|
|
341
|
+
const releaseOnce = () => {
|
|
342
|
+
if (released) return;
|
|
343
|
+
released = true;
|
|
344
|
+
release();
|
|
345
|
+
};
|
|
346
|
+
|
|
347
|
+
if (!response?.body) {
|
|
348
|
+
releaseOnce();
|
|
349
|
+
return response;
|
|
350
|
+
}
|
|
351
|
+
|
|
352
|
+
const reader = response.body.getReader();
|
|
353
|
+
const body = new ReadableStream({
|
|
354
|
+
async pull(controller) {
|
|
355
|
+
try {
|
|
356
|
+
const { done, value } = await reader.read();
|
|
357
|
+
if (done) {
|
|
358
|
+
releaseOnce();
|
|
359
|
+
controller.close();
|
|
360
|
+
return;
|
|
361
|
+
}
|
|
362
|
+
controller.enqueue(value);
|
|
363
|
+
} catch (error) {
|
|
364
|
+
releaseOnce();
|
|
365
|
+
try { await reader.cancel(error); } catch {}
|
|
366
|
+
try { controller.error(error); } catch {}
|
|
367
|
+
}
|
|
368
|
+
},
|
|
369
|
+
cancel(reason) {
|
|
370
|
+
releaseOnce();
|
|
371
|
+
return reader.cancel(reason);
|
|
372
|
+
},
|
|
373
|
+
});
|
|
374
|
+
|
|
375
|
+
return new Response(body, {
|
|
376
|
+
status: response.status,
|
|
377
|
+
statusText: response.statusText,
|
|
378
|
+
headers: new Headers(response.headers),
|
|
379
|
+
});
|
|
380
|
+
}
|
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
export type ProviderStatus = "online" | "offline" | "degraded";
|
|
2
|
+
|
|
3
|
+
export type ProviderType =
|
|
4
|
+
| "alicode" | "alicode-intl" | "alims-intl" | "anthropic" | "antigravity"
|
|
5
|
+
| "azure" | "baidu" | "blackbox" | "byteplus" | "cerebras"
|
|
6
|
+
| "chutes" | "claude" | "cline" | "clinepass" | "cloudflare-ai"
|
|
7
|
+
| "codebuddy-cn" | "codebuddy-intl" | "codex" | "commandcode" | "cursor"
|
|
8
|
+
| "deepseek" | "featherless" | "fireworks" | "gemini" | "gemini-cli"
|
|
9
|
+
| "github" | "glm" | "glm-cn" | "grok-cli" | "grok-web"
|
|
10
|
+
| "groq" | "kilo-gateway" | "kilocode" | "kimchi" | "kimi"
|
|
11
|
+
| "kiro" | "minimax" | "minimax-cn" | "mistral"
|
|
12
|
+
| "morph" | "nebius" | "nvidia" | "ollama" | "ollama-local"
|
|
13
|
+
| "openai" | "opencode" | "opencode-go" | "openrouter" | "perplexity"
|
|
14
|
+
| "poolside" | "qoder" | "tencent"
|
|
15
|
+
| "together" | "trae" | "venice" | "vercel-ai-gateway"
|
|
16
|
+
| "sumopod" | "vertex" | "vertex-partner" | "windsurf" | "xai" | "xiaomi-mimo"
|
|
17
|
+
| "xiaomi-tokenplan";
|
|
18
|
+
|
|
19
|
+
export type ProviderCategory = "apikey" | "oauth" | "webCookie" | "none";
|
|
20
|
+
export type AuthType = "apikey" | "oauth" | "cookie" | "none";
|
|
21
|
+
|
|
22
|
+
export type ProviderFormat =
|
|
23
|
+
| "openai" | "claude" | "gemini" | "ollama" | "vertex"
|
|
24
|
+
| "cursor" | "commandcode" | "antigravity" | "kiro"
|
|
25
|
+
| "grok-web" | "gemini-cli"
|
|
26
|
+
| "custom";
|
|
27
|
+
|
|
28
|
+
export type ProviderService = "llm" | "search" | "embeddings" | "image" | "audio";
|
|
29
|
+
|
|
30
|
+
export interface OAuthConfig {
|
|
31
|
+
clientId?: string;
|
|
32
|
+
clientSecret?: string;
|
|
33
|
+
authUrl?: string;
|
|
34
|
+
tokenUrl?: string;
|
|
35
|
+
scopes?: string[];
|
|
36
|
+
[key: string]: unknown;
|
|
37
|
+
}
|
|
38
|
+
|
|
39
|
+
export interface Provider {
|
|
40
|
+
id: string;
|
|
41
|
+
name: string;
|
|
42
|
+
type: ProviderType;
|
|
43
|
+
category: ProviderCategory;
|
|
44
|
+
authType: AuthType;
|
|
45
|
+
authModes: AuthType[];
|
|
46
|
+
format: ProviderFormat;
|
|
47
|
+
service: ProviderService;
|
|
48
|
+
baseUrl: string;
|
|
49
|
+
apiKey?: string;
|
|
50
|
+
oauthConfig?: OAuthConfig;
|
|
51
|
+
models: string[];
|
|
52
|
+
status: ProviderStatus;
|
|
53
|
+
priority: number;
|
|
54
|
+
weight: number;
|
|
55
|
+
maxTokens: number;
|
|
56
|
+
rateLimit: number;
|
|
57
|
+
createdAt: string;
|
|
58
|
+
updatedAt: string;
|
|
59
|
+
}
|
|
@@ -0,0 +1,272 @@
|
|
|
1
|
+
import { detectFormat } from "../services/provider.js";
|
|
2
|
+
import { translateResponse, initState } from "../translator/index.js";
|
|
3
|
+
import { FORMATS } from "../translator/formats.js";
|
|
4
|
+
import { SKIP_PATTERNS } from "../config/runtimeConfig.js";
|
|
5
|
+
import { formatSSE } from "./stream.js";
|
|
6
|
+
|
|
7
|
+
export function handleBypassRequest(body, model, userAgent = "", ccFilterNaming = false) {
|
|
8
|
+
if (!userAgent.includes("claude-cli")) return null;
|
|
9
|
+
if (!body.messages?.length) return null;
|
|
10
|
+
|
|
11
|
+
const messages = body.messages;
|
|
12
|
+
const getText = (content) => {
|
|
13
|
+
if (typeof content === "string") return content;
|
|
14
|
+
if (Array.isArray(content)) {
|
|
15
|
+
return content.filter(c => c.type === "text").map(c => c.text).join(" ");
|
|
16
|
+
}
|
|
17
|
+
return "";
|
|
18
|
+
};
|
|
19
|
+
|
|
20
|
+
let shouldBypass = false;
|
|
21
|
+
let namingBypass = false;
|
|
22
|
+
|
|
23
|
+
const lastMsg = messages[messages.length - 1];
|
|
24
|
+
if (lastMsg?.role === "assistant" && lastMsg.content?.[0]?.text === "{") {
|
|
25
|
+
shouldBypass = true;
|
|
26
|
+
}
|
|
27
|
+
|
|
28
|
+
if (!shouldBypass) {
|
|
29
|
+
const firstText = getText(messages[0]?.content);
|
|
30
|
+
if (firstText === "Warmup") {
|
|
31
|
+
shouldBypass = true;
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
|
|
35
|
+
if (!shouldBypass && messages.length === 1 && messages[0]?.role === "user") {
|
|
36
|
+
const firstText = getText(messages[0]?.content);
|
|
37
|
+
if (firstText === "count") {
|
|
38
|
+
shouldBypass = true;
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
|
|
42
|
+
if (!shouldBypass && SKIP_PATTERNS?.length) {
|
|
43
|
+
const userMessages = messages.filter(m => m.role === "user");
|
|
44
|
+
const userText = userMessages.map(m => getText(m.content)).join(" ");
|
|
45
|
+
if (SKIP_PATTERNS.some(p => userText.includes(p))) {
|
|
46
|
+
shouldBypass = true;
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
|
|
50
|
+
if (!shouldBypass && ccFilterNaming) {
|
|
51
|
+
const systemMsg = messages.find(m => m.role === "system");
|
|
52
|
+
const systemFromMessages = getText(systemMsg?.content);
|
|
53
|
+
const systemFromBody = Array.isArray(body.system)
|
|
54
|
+
? body.system.filter(s => s.type === "text").map(s => s.text).join(" ")
|
|
55
|
+
: (typeof body.system === "string" ? body.system : "");
|
|
56
|
+
const systemText = systemFromMessages || systemFromBody;
|
|
57
|
+
if (systemText.includes("isNewTopic")) {
|
|
58
|
+
shouldBypass = true;
|
|
59
|
+
namingBypass = true;
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
|
|
63
|
+
if (!shouldBypass) return null;
|
|
64
|
+
|
|
65
|
+
const sourceFormat = detectFormat(body);
|
|
66
|
+
const stream = body.stream !== false;
|
|
67
|
+
|
|
68
|
+
if (namingBypass) {
|
|
69
|
+
const userMsg = messages.find(m => m.role === "user");
|
|
70
|
+
const userText = getText(userMsg?.content);
|
|
71
|
+
const title = userText.trim().split(/\s+/).slice(0, 3).join(" ");
|
|
72
|
+
const namingText = JSON.stringify({ isNewTopic: true, title });
|
|
73
|
+
return stream
|
|
74
|
+
? createStreamingResponse(sourceFormat, model, namingText)
|
|
75
|
+
: createNonStreamingResponse(sourceFormat, model, namingText);
|
|
76
|
+
}
|
|
77
|
+
|
|
78
|
+
return stream
|
|
79
|
+
? createStreamingResponse(sourceFormat, model)
|
|
80
|
+
: createNonStreamingResponse(sourceFormat, model);
|
|
81
|
+
}
|
|
82
|
+
|
|
83
|
+
const DEFAULT_BYPASS_TEXT = "CLI Command Execution: Clear Terminal";
|
|
84
|
+
|
|
85
|
+
function createOpenAIResponse(model, text = DEFAULT_BYPASS_TEXT) {
|
|
86
|
+
const id = `chatcmpl-${Date.now()}`;
|
|
87
|
+
const created = Math.floor(Date.now() / 1000);
|
|
88
|
+
|
|
89
|
+
return {
|
|
90
|
+
id,
|
|
91
|
+
object: "chat.completion",
|
|
92
|
+
created,
|
|
93
|
+
model,
|
|
94
|
+
choices: [{
|
|
95
|
+
index: 0,
|
|
96
|
+
message: {
|
|
97
|
+
role: "assistant",
|
|
98
|
+
content: text
|
|
99
|
+
},
|
|
100
|
+
finish_reason: "stop"
|
|
101
|
+
}],
|
|
102
|
+
usage: {
|
|
103
|
+
prompt_tokens: 1,
|
|
104
|
+
completion_tokens: 1,
|
|
105
|
+
total_tokens: 2
|
|
106
|
+
}
|
|
107
|
+
};
|
|
108
|
+
}
|
|
109
|
+
|
|
110
|
+
function createNonStreamingResponse(sourceFormat, model, text) {
|
|
111
|
+
const openaiResponse = createOpenAIResponse(model, text);
|
|
112
|
+
|
|
113
|
+
if (sourceFormat === FORMATS.OPENAI) {
|
|
114
|
+
return {
|
|
115
|
+
success: true,
|
|
116
|
+
response: new Response(JSON.stringify(openaiResponse), {
|
|
117
|
+
headers: {
|
|
118
|
+
"Content-Type": "application/json",
|
|
119
|
+
"Access-Control-Allow-Origin": "*"
|
|
120
|
+
}
|
|
121
|
+
})
|
|
122
|
+
};
|
|
123
|
+
}
|
|
124
|
+
|
|
125
|
+
const state = initState(sourceFormat);
|
|
126
|
+
state.model = model;
|
|
127
|
+
|
|
128
|
+
const openaiChunks = createOpenAIStreamingChunks(openaiResponse);
|
|
129
|
+
const allTranslated = [];
|
|
130
|
+
|
|
131
|
+
for (const chunk of openaiChunks) {
|
|
132
|
+
const translated = translateResponse(FORMATS.OPENAI, sourceFormat, chunk, state);
|
|
133
|
+
if (translated?.length > 0) {
|
|
134
|
+
allTranslated.push(...translated);
|
|
135
|
+
}
|
|
136
|
+
}
|
|
137
|
+
|
|
138
|
+
const flushed = translateResponse(FORMATS.OPENAI, sourceFormat, null, state);
|
|
139
|
+
if (flushed?.length > 0) {
|
|
140
|
+
allTranslated.push(...flushed);
|
|
141
|
+
}
|
|
142
|
+
|
|
143
|
+
const finalResponse = mergeChunksToResponse(allTranslated, sourceFormat);
|
|
144
|
+
|
|
145
|
+
return {
|
|
146
|
+
success: true,
|
|
147
|
+
response: new Response(JSON.stringify(finalResponse), {
|
|
148
|
+
headers: {
|
|
149
|
+
"Content-Type": "application/json",
|
|
150
|
+
"Access-Control-Allow-Origin": "*"
|
|
151
|
+
}
|
|
152
|
+
})
|
|
153
|
+
};
|
|
154
|
+
}
|
|
155
|
+
|
|
156
|
+
function createStreamingResponse(sourceFormat, model, text) {
|
|
157
|
+
const openaiResponse = createOpenAIResponse(model, text);
|
|
158
|
+
const state = initState(sourceFormat);
|
|
159
|
+
state.model = model;
|
|
160
|
+
|
|
161
|
+
const openaiChunks = createOpenAIStreamingChunks(openaiResponse);
|
|
162
|
+
|
|
163
|
+
const translatedChunks = [];
|
|
164
|
+
|
|
165
|
+
for (const chunk of openaiChunks) {
|
|
166
|
+
const translated = translateResponse(FORMATS.OPENAI, sourceFormat, chunk, state);
|
|
167
|
+
if (translated?.length > 0) {
|
|
168
|
+
for (const item of translated) {
|
|
169
|
+
translatedChunks.push(formatSSE(item, sourceFormat));
|
|
170
|
+
}
|
|
171
|
+
}
|
|
172
|
+
}
|
|
173
|
+
|
|
174
|
+
const flushed = translateResponse(FORMATS.OPENAI, sourceFormat, null, state);
|
|
175
|
+
if (flushed?.length > 0) {
|
|
176
|
+
for (const item of flushed) {
|
|
177
|
+
translatedChunks.push(formatSSE(item, sourceFormat));
|
|
178
|
+
}
|
|
179
|
+
}
|
|
180
|
+
|
|
181
|
+
translatedChunks.push("data: [DONE]\n\n");
|
|
182
|
+
|
|
183
|
+
return {
|
|
184
|
+
success: true,
|
|
185
|
+
response: new Response(translatedChunks.join(""), {
|
|
186
|
+
headers: {
|
|
187
|
+
"Content-Type": "text/event-stream",
|
|
188
|
+
"Cache-Control": "no-cache",
|
|
189
|
+
"Connection": "keep-alive",
|
|
190
|
+
"Access-Control-Allow-Origin": "*"
|
|
191
|
+
}
|
|
192
|
+
})
|
|
193
|
+
};
|
|
194
|
+
}
|
|
195
|
+
|
|
196
|
+
function mergeChunksToResponse(chunks, sourceFormat) {
|
|
197
|
+
if (!chunks || chunks.length === 0) {
|
|
198
|
+
return createOpenAIResponse("unknown");
|
|
199
|
+
}
|
|
200
|
+
|
|
201
|
+
let finalChunk = chunks[chunks.length - 1];
|
|
202
|
+
|
|
203
|
+
if (sourceFormat === FORMATS.CLAUDE) {
|
|
204
|
+
const messageStop = chunks.find(c => c.type === "message_stop");
|
|
205
|
+
if (messageStop) {
|
|
206
|
+
|
|
207
|
+
const contentDelta = chunks.find(c => c.type === "content_block_delta");
|
|
208
|
+
const messageDelta = chunks.find(c => c.type === "message_delta");
|
|
209
|
+
const messageStart = chunks.find(c => c.type === "message_start");
|
|
210
|
+
|
|
211
|
+
if (messageStart?.message) {
|
|
212
|
+
finalChunk = messageStart.message;
|
|
213
|
+
|
|
214
|
+
const startUsage = messageStart.message.usage;
|
|
215
|
+
const deltaUsage = messageDelta?.usage;
|
|
216
|
+
if (startUsage || deltaUsage) {
|
|
217
|
+
finalChunk.usage = {
|
|
218
|
+
...(startUsage || {}),
|
|
219
|
+
...(deltaUsage || {}),
|
|
220
|
+
...(startUsage?.cache_read_input_tokens !== undefined
|
|
221
|
+
? { cache_read_input_tokens: startUsage.cache_read_input_tokens }
|
|
222
|
+
: {}),
|
|
223
|
+
...(startUsage?.cache_creation_input_tokens !== undefined
|
|
224
|
+
? { cache_creation_input_tokens: startUsage.cache_creation_input_tokens }
|
|
225
|
+
: {}),
|
|
226
|
+
...(startUsage?.input_tokens !== undefined
|
|
227
|
+
? { input_tokens: startUsage.input_tokens }
|
|
228
|
+
: {})
|
|
229
|
+
};
|
|
230
|
+
}
|
|
231
|
+
}
|
|
232
|
+
}
|
|
233
|
+
}
|
|
234
|
+
|
|
235
|
+
return finalChunk;
|
|
236
|
+
}
|
|
237
|
+
|
|
238
|
+
function createOpenAIStreamingChunks(completeResponse) {
|
|
239
|
+
const { id, created, model, choices } = completeResponse;
|
|
240
|
+
const content = choices[0].message.content;
|
|
241
|
+
|
|
242
|
+
return [
|
|
243
|
+
|
|
244
|
+
{
|
|
245
|
+
id,
|
|
246
|
+
object: "chat.completion.chunk",
|
|
247
|
+
created,
|
|
248
|
+
model,
|
|
249
|
+
choices: [{
|
|
250
|
+
index: 0,
|
|
251
|
+
delta: {
|
|
252
|
+
role: "assistant",
|
|
253
|
+
content
|
|
254
|
+
},
|
|
255
|
+
finish_reason: null
|
|
256
|
+
}]
|
|
257
|
+
},
|
|
258
|
+
|
|
259
|
+
{
|
|
260
|
+
id,
|
|
261
|
+
object: "chat.completion.chunk",
|
|
262
|
+
created,
|
|
263
|
+
model,
|
|
264
|
+
choices: [{
|
|
265
|
+
index: 0,
|
|
266
|
+
delta: {},
|
|
267
|
+
finish_reason: "stop"
|
|
268
|
+
}],
|
|
269
|
+
usage: completeResponse.usage
|
|
270
|
+
}
|
|
271
|
+
];
|
|
272
|
+
}
|