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.
Files changed (643) hide show
  1. package/.env.example +43 -0
  2. package/CHANGELOG.md +132 -0
  3. package/LICENSE +30 -0
  4. package/README.id.md +633 -0
  5. package/README.md +628 -0
  6. package/cli/cli.js +388 -0
  7. package/cli/package.json +4 -0
  8. package/cli/src/cli/commands/xaiVideo.js +280 -0
  9. package/cli/src/cli/tray/SwayAvatar.svg +20 -0
  10. package/cli/src/cli/tray/autostart.js +351 -0
  11. package/cli/src/cli/tray/icon.ico +0 -0
  12. package/cli/src/cli/tray/icon.png +0 -0
  13. package/cli/src/cli/tray/tray.js +258 -0
  14. package/cli/src/cli/tray/tray.ps1 +129 -0
  15. package/cli/src/cli/tray/trayHost.js +157 -0
  16. package/cli/src/cli/tray/trayWin.js +83 -0
  17. package/cli/src/cli/utils/display.js +123 -0
  18. package/cli/src/cli/utils/input.js +138 -0
  19. package/cli/src/runtime/discover.js +153 -0
  20. package/cli/src/runtime/lifecycle.js +333 -0
  21. package/cli/src/runtime/runMenu.js +250 -0
  22. package/cli/src/runtime/uiAdapters.js +12 -0
  23. package/cli/src/runtime/update.js +130 -0
  24. package/dashboard/dist/assets/CliTools-Ch3gKuNw.js +24 -0
  25. package/dashboard/dist/assets/Combo-BBCldjWi.js +1 -0
  26. package/dashboard/dist/assets/ConsoleLog-wywAYcPT.js +6 -0
  27. package/dashboard/dist/assets/Dashboard-DxzyF41b.js +1 -0
  28. package/dashboard/dist/assets/Header-C_9Su9BF.js +1 -0
  29. package/dashboard/dist/assets/Keys-D0ufgcsc.js +1 -0
  30. package/dashboard/dist/assets/Login-BeKpipS_.js +1 -0
  31. package/dashboard/dist/assets/ManageApiKey-CVXsmW2B.js +1 -0
  32. package/dashboard/dist/assets/OAuthCallback-D_UZYcF6.js +1 -0
  33. package/dashboard/dist/assets/Provider-BU7JfTqm.js +4 -0
  34. package/dashboard/dist/assets/ProviderModelAccordion-BAifkfVR.js +1 -0
  35. package/dashboard/dist/assets/Proxy-CU15VJ--.js +1 -0
  36. package/dashboard/dist/assets/QuotaMonitor-HzGK7bcc.js +1 -0
  37. package/dashboard/dist/assets/Settings-Dt4y2dC0.js +1 -0
  38. package/dashboard/dist/assets/Setup-DSDUwU7N.js +1 -0
  39. package/dashboard/dist/assets/StatusBadge-1nAOyHU2.js +1 -0
  40. package/dashboard/dist/assets/SwayChat-Bup7up6D.js +10 -0
  41. package/dashboard/dist/assets/Usage-eWnrPQ_v.js +4 -0
  42. package/dashboard/dist/assets/accordion-DOOPDDgt.js +1 -0
  43. package/dashboard/dist/assets/admin-extras-api-CCdu5TFM.js +1 -0
  44. package/dashboard/dist/assets/api-keys-api-ifAC2Ok3.js +1 -0
  45. package/dashboard/dist/assets/card-CC0Omjcv.js +1 -0
  46. package/dashboard/dist/assets/chatStudio-C0zPz_Z_.js +2 -0
  47. package/dashboard/dist/assets/checkbox-7SZBrmbN.js +1 -0
  48. package/dashboard/dist/assets/connections-api-_N54UD3S.js +1 -0
  49. package/dashboard/dist/assets/copy-button-qBvM6H9z.js +1 -0
  50. package/dashboard/dist/assets/currency-api-SV9CBf5e.js +1 -0
  51. package/dashboard/dist/assets/flip-button-BGgkjtl0.js +1 -0
  52. package/dashboard/dist/assets/frame-cUiti6Lq.js +1 -0
  53. package/dashboard/dist/assets/icons-vendor-CUx7SB5V.js +541 -0
  54. package/dashboard/dist/assets/index-B6j-9zHy.js +10 -0
  55. package/dashboard/dist/assets/index-C-_l0yYz.css +1 -0
  56. package/dashboard/dist/assets/live-mode-CHShLY4E.js +1 -0
  57. package/dashboard/dist/assets/markdown-vendor-BY8e1gQ8.js +29 -0
  58. package/dashboard/dist/assets/motion-vendor-g5fpGzmJ.js +9 -0
  59. package/dashboard/dist/assets/oauth-callback-CqYQzyd7.js +1 -0
  60. package/dashboard/dist/assets/query-vendor-DVevaSVS.js +1 -0
  61. package/dashboard/dist/assets/react-vendor-xnYrwzsL.js +61 -0
  62. package/dashboard/dist/assets/ripple-button-l8aMiVVG.js +1 -0
  63. package/dashboard/dist/assets/segmented-CqmEnQjZ.js +1 -0
  64. package/dashboard/dist/assets/select-D8pUtMXp.js +1 -0
  65. package/dashboard/dist/assets/skeleton-7uJDiIXe.js +1 -0
  66. package/dashboard/dist/assets/table-D0M60Tu1.js +1 -0
  67. package/dashboard/dist/assets/tabs-C6YkuulG.js +1 -0
  68. package/dashboard/dist/assets/tunnel-api-DDPsBaDV.js +1 -0
  69. package/dashboard/dist/assets/ui-vendor-BuM9f2NY.js +17 -0
  70. package/dashboard/dist/image2/SwayAvatar.svg +20 -0
  71. package/dashboard/dist/index.html +32 -0
  72. package/dashboard/dist/logo/avatar.svg +1 -0
  73. package/dashboard/dist/logo/sway.svg +20 -0
  74. package/dashboard/dist/logo/swaychat.svg +20 -0
  75. package/dashboard/dist/providers/SwayCustom.svg +19 -0
  76. package/dashboard/dist/providers/alicode.svg +12 -0
  77. package/dashboard/dist/providers/anthropic.svg +11 -0
  78. package/dashboard/dist/providers/antigravity.svg +3 -0
  79. package/dashboard/dist/providers/azure.svg +5 -0
  80. package/dashboard/dist/providers/baidu.svg +3 -0
  81. package/dashboard/dist/providers/blackbox.svg +5 -0
  82. package/dashboard/dist/providers/byteplus.svg +3 -0
  83. package/dashboard/dist/providers/cerebras.svg +4 -0
  84. package/dashboard/dist/providers/chutes.svg +4 -0
  85. package/dashboard/dist/providers/claude.svg +3 -0
  86. package/dashboard/dist/providers/cline.svg +11 -0
  87. package/dashboard/dist/providers/cloudflare-ai.svg +4 -0
  88. package/dashboard/dist/providers/codebuddy.svg +11 -0
  89. package/dashboard/dist/providers/codex.svg +10 -0
  90. package/dashboard/dist/providers/commandcode.svg +3 -0
  91. package/dashboard/dist/providers/cursor.svg +3 -0
  92. package/dashboard/dist/providers/deepseek.svg +10 -0
  93. package/dashboard/dist/providers/featherless.svg +3 -0
  94. package/dashboard/dist/providers/fireworks.svg +3 -0
  95. package/dashboard/dist/providers/gemini.svg +3 -0
  96. package/dashboard/dist/providers/geminicli.svg +11 -0
  97. package/dashboard/dist/providers/github.svg +3 -0
  98. package/dashboard/dist/providers/grok.svg +3 -0
  99. package/dashboard/dist/providers/groq.svg +3 -0
  100. package/dashboard/dist/providers/hermesagent.svg +1 -0
  101. package/dashboard/dist/providers/kilocode.svg +15 -0
  102. package/dashboard/dist/providers/kimchi.svg +3 -0
  103. package/dashboard/dist/providers/kimi.svg +4 -0
  104. package/dashboard/dist/providers/kiro.svg +10 -0
  105. package/dashboard/dist/providers/meta.svg +3 -0
  106. package/dashboard/dist/providers/mimo.svg +10 -0
  107. package/dashboard/dist/providers/minimax.svg +3 -0
  108. package/dashboard/dist/providers/mistral.svg +3 -0
  109. package/dashboard/dist/providers/morph.svg +3 -0
  110. package/dashboard/dist/providers/nebius.svg +4 -0
  111. package/dashboard/dist/providers/nvidia.svg +3 -0
  112. package/dashboard/dist/providers/ollama.svg +3 -0
  113. package/dashboard/dist/providers/omp.svg +5 -0
  114. package/dashboard/dist/providers/openai.svg +3 -0
  115. package/dashboard/dist/providers/opencode.svg +3 -0
  116. package/dashboard/dist/providers/openrouter.svg +3 -0
  117. package/dashboard/dist/providers/perplexity.svg +3 -0
  118. package/dashboard/dist/providers/poolside.svg +36 -0
  119. package/dashboard/dist/providers/qoder.svg +11 -0
  120. package/dashboard/dist/providers/sumopod.svg +3 -0
  121. package/dashboard/dist/providers/tencent.svg +5 -0
  122. package/dashboard/dist/providers/together.svg +12 -0
  123. package/dashboard/dist/providers/trae.svg +3 -0
  124. package/dashboard/dist/providers/venice.svg +4 -0
  125. package/dashboard/dist/providers/vercel.svg +3 -0
  126. package/dashboard/dist/providers/vertex.svg +17 -0
  127. package/dashboard/dist/providers/windsurf.svg +10 -0
  128. package/dashboard/dist/providers/zai.svg +3 -0
  129. package/dashboard/tsconfig.json +13 -0
  130. package/package.json +87 -0
  131. package/src/config/appConstants.js +153 -0
  132. package/src/config/defaultThinkingSignature.js +7 -0
  133. package/src/config/errorConfig.js +117 -0
  134. package/src/config/grokCli.js +9 -0
  135. package/src/config/kiroConstants.js +235 -0
  136. package/src/config/mediaConfig.js +20 -0
  137. package/src/config/ollamaModels.js +18 -0
  138. package/src/config/providerModels.js +109 -0
  139. package/src/config/providers.js +17 -0
  140. package/src/config/runtimeConfig.js +84 -0
  141. package/src/dashboardGuard.js +254 -0
  142. package/src/executors/antigravity.js +389 -0
  143. package/src/executors/azure.js +49 -0
  144. package/src/executors/base.js +225 -0
  145. package/src/executors/codebuddy-cn.js +23 -0
  146. package/src/executors/codebuddy-intl.js +60 -0
  147. package/src/executors/codex.js +436 -0
  148. package/src/executors/commandcode.js +83 -0
  149. package/src/executors/cursor.js +1075 -0
  150. package/src/executors/default.js +294 -0
  151. package/src/executors/gemini-cli.js +88 -0
  152. package/src/executors/github.js +393 -0
  153. package/src/executors/grok-cli.js +510 -0
  154. package/src/executors/grok-web.js +342 -0
  155. package/src/executors/index.js +86 -0
  156. package/src/executors/kimchi.js +116 -0
  157. package/src/executors/kiro.js +1155 -0
  158. package/src/executors/ollama-local.js +14 -0
  159. package/src/executors/opencode-go.js +47 -0
  160. package/src/executors/opencode.js +31 -0
  161. package/src/executors/qoder.js +433 -0
  162. package/src/executors/trae.js +318 -0
  163. package/src/executors/vertex.js +151 -0
  164. package/src/executors/windsurf.js +555 -0
  165. package/src/executors/xiaomi-tokenplan.js +16 -0
  166. package/src/handlers/chatCore/nonStreamingHandler.js +370 -0
  167. package/src/handlers/chatCore/requestDetail.js +130 -0
  168. package/src/handlers/chatCore/sseToJsonHandler.js +318 -0
  169. package/src/handlers/chatCore/streamingHandler.js +250 -0
  170. package/src/handlers/chatCore.js +652 -0
  171. package/src/lib/auth/dashboardSession.js +83 -0
  172. package/src/lib/auth/loginLimiter.js +99 -0
  173. package/src/lib/cache/ttlCache.js +153 -0
  174. package/src/lib/consoleLogBuffer.js +127 -0
  175. package/src/lib/dataDir.js +45 -0
  176. package/src/lib/db/adapters/betterSqliteAdapter.js +36 -0
  177. package/src/lib/db/adapters/bunSqliteAdapter.js +47 -0
  178. package/src/lib/db/adapters/nodeSqliteAdapter.js +65 -0
  179. package/src/lib/db/adapters/runtimeHelpers.js +23 -0
  180. package/src/lib/db/adapters/sqljsAdapter.js +103 -0
  181. package/src/lib/db/backup.js +75 -0
  182. package/src/lib/db/driver.js +110 -0
  183. package/src/lib/db/helpers/jsonCol.js +18 -0
  184. package/src/lib/db/helpers/kvStore.js +48 -0
  185. package/src/lib/db/helpers/metaStore.js +32 -0
  186. package/src/lib/db/index.js +473 -0
  187. package/src/lib/db/migrate.js +267 -0
  188. package/src/lib/db/migrations/001-initial.js +12 -0
  189. package/src/lib/db/migrations/002-prune-headroom-pxpipe-settings.js +32 -0
  190. package/src/lib/db/migrations/004-request-details-trace-id.js +12 -0
  191. package/src/lib/db/migrations/005-request-details-payload-capture.js +11 -0
  192. package/src/lib/db/migrations/007-headroom-readd-settings.js +35 -0
  193. package/src/lib/db/migrations/008-account-model-locks.js +22 -0
  194. package/src/lib/db/migrations/009-warp-pool-lifecycle.js +5 -0
  195. package/src/lib/db/migrations/010-api-key-file-permissions.js +7 -0
  196. package/src/lib/db/migrations/011-remove-upstream-user-agent-profile.js +41 -0
  197. package/src/lib/db/migrations/012-remove-headroom-settings.js +33 -0
  198. package/src/lib/db/migrations/013-default-fill-first-routing.js +20 -0
  199. package/src/lib/db/migrations/014-normalize-fill-first-routing.js +37 -0
  200. package/src/lib/db/migrations/015-remove-ip-ban-tables.js +8 -0
  201. package/src/lib/db/migrations/016-prune-legacy-codebuddy-intl-models.js +25 -0
  202. package/src/lib/db/migrations/index.js +20 -0
  203. package/src/lib/db/nineRouterMigration.js +785 -0
  204. package/src/lib/db/paths.js +23 -0
  205. package/src/lib/db/repos/accountModelLocksRepo.js +74 -0
  206. package/src/lib/db/repos/aliasRepo.js +108 -0
  207. package/src/lib/db/repos/apiKeysRepo.js +204 -0
  208. package/src/lib/db/repos/combosRepo.js +85 -0
  209. package/src/lib/db/repos/connectionsRepo.js +522 -0
  210. package/src/lib/db/repos/disabledModelsRepo.js +63 -0
  211. package/src/lib/db/repos/nodesRepo.js +127 -0
  212. package/src/lib/db/repos/pricingRepo.js +108 -0
  213. package/src/lib/db/repos/proxyPoolsRepo.js +104 -0
  214. package/src/lib/db/repos/requestDetailsRepo.js +345 -0
  215. package/src/lib/db/repos/settingsRepo.js +135 -0
  216. package/src/lib/db/repos/usageRepo.js +1335 -0
  217. package/src/lib/db/schema.js +179 -0
  218. package/src/lib/db/version.js +32 -0
  219. package/src/lib/db/writeBehind.js +174 -0
  220. package/src/lib/disabledModelsDb.js +3 -0
  221. package/src/lib/env.ts +175 -0
  222. package/src/lib/errors/publicError.js +18 -0
  223. package/src/lib/grokBuildConfig.js +243 -0
  224. package/src/lib/guardRequest.js +51 -0
  225. package/src/lib/localDb.js +21 -0
  226. package/src/lib/mcp/stdioSseBridge.js +174 -0
  227. package/src/lib/network/connectionProxy.js +200 -0
  228. package/src/lib/network/proxyTest.js +158 -0
  229. package/src/lib/network/proxyUrl.js +76 -0
  230. package/src/lib/network/relay.js +71 -0
  231. package/src/lib/oauth/constants/oauth.js +152 -0
  232. package/src/lib/oauth/constants/xai.js +43 -0
  233. package/src/lib/oauth/kiroExternalIdp.js +167 -0
  234. package/src/lib/oauth/providerHelpers.js +90 -0
  235. package/src/lib/oauth/providers/_shared.js +12 -0
  236. package/src/lib/oauth/providers/antigravity.js +126 -0
  237. package/src/lib/oauth/providers/claude.js +60 -0
  238. package/src/lib/oauth/providers/cline.js +62 -0
  239. package/src/lib/oauth/providers/clinepass.js +62 -0
  240. package/src/lib/oauth/providers/codebuddy-cn.js +75 -0
  241. package/src/lib/oauth/providers/codebuddy-intl.js +73 -0
  242. package/src/lib/oauth/providers/codex.js +72 -0
  243. package/src/lib/oauth/providers/cursor.js +18 -0
  244. package/src/lib/oauth/providers/gemini-cli.js +84 -0
  245. package/src/lib/oauth/providers/github.js +96 -0
  246. package/src/lib/oauth/providers/grok-cli.js +123 -0
  247. package/src/lib/oauth/providers/index.js +233 -0
  248. package/src/lib/oauth/providers/kilocode.js +60 -0
  249. package/src/lib/oauth/providers/kimchi.js +79 -0
  250. package/src/lib/oauth/providers/kimi.js +79 -0
  251. package/src/lib/oauth/providers/kiro.js +147 -0
  252. package/src/lib/oauth/providers/qoder.js +90 -0
  253. package/src/lib/oauth/providers/trae.js +244 -0
  254. package/src/lib/oauth/providers/windsurf.js +123 -0
  255. package/src/lib/oauth/providers/xai.js +95 -0
  256. package/src/lib/oauth/providers.js +16 -0
  257. package/src/lib/oauth/services/cursor.js +132 -0
  258. package/src/lib/oauth/services/kiro.js +351 -0
  259. package/src/lib/oauth/services/oauth.js +135 -0
  260. package/src/lib/oauth/services/qoder.js +151 -0
  261. package/src/lib/oauth/services/xai.js +204 -0
  262. package/src/lib/oauth/utils/ideDetect.js +55 -0
  263. package/src/lib/oauth/utils/pkce.js +25 -0
  264. package/src/lib/oauth/utils/server.js +732 -0
  265. package/src/lib/oauth/utils/ui.js +43 -0
  266. package/src/lib/providerNormalization.js +40 -0
  267. package/src/lib/qoder/constants.js +1 -0
  268. package/src/lib/requestDetailsDb.js +3 -0
  269. package/src/lib/resourceProfile.ts +124 -0
  270. package/src/lib/runtimeSecrets.ts +130 -0
  271. package/src/lib/security/payloadCapture.js +87 -0
  272. package/src/lib/tunnel/cloudflare/cloudflared.js +436 -0
  273. package/src/lib/tunnel/cloudflare/config.js +8 -0
  274. package/src/lib/tunnel/cloudflare/healthCheck.js +37 -0
  275. package/src/lib/tunnel/cloudflare/manager.js +149 -0
  276. package/src/lib/tunnel/cloudflare/pid.js +31 -0
  277. package/src/lib/tunnel/cloudflare/publicUrl.js +41 -0
  278. package/src/lib/tunnel/cloudflare/requirements.js +12 -0
  279. package/src/lib/tunnel/index.js +26 -0
  280. package/src/lib/tunnel/shared/dnsResolver.js +16 -0
  281. package/src/lib/tunnel/shared/internetCheck.js +30 -0
  282. package/src/lib/tunnel/shared/state.js +45 -0
  283. package/src/lib/tunnel/shared/watchdogConfig.js +6 -0
  284. package/src/lib/update/installer.js +52 -0
  285. package/src/lib/update/version.js +116 -0
  286. package/src/lib/usageDb.js +6 -0
  287. package/src/lib/usageRefresh.js +88 -0
  288. package/src/lifecycle/appLifecycle.ts +99 -0
  289. package/src/models.js +36 -0
  290. package/src/next/headers.js +12 -0
  291. package/src/next/server.js +74 -0
  292. package/src/observability/metrics.js +439 -0
  293. package/src/observability/tracing.js +107 -0
  294. package/src/open-sse-index.js +70 -0
  295. package/src/providers/capabilities.js +299 -0
  296. package/src/providers/display.ts +142 -0
  297. package/src/providers/index.js +27 -0
  298. package/src/providers/models/helpers.js +19 -0
  299. package/src/providers/models/namePatterns.js +27 -0
  300. package/src/providers/models/schema.js +32 -0
  301. package/src/providers/pricing.js +248 -0
  302. package/src/providers/registry/agentrouter.js +52 -0
  303. package/src/providers/registry/alicode-intl.js +32 -0
  304. package/src/providers/registry/alicode.js +33 -0
  305. package/src/providers/registry/alims-intl.js +30 -0
  306. package/src/providers/registry/anthropic.js +29 -0
  307. package/src/providers/registry/antigravity.js +139 -0
  308. package/src/providers/registry/azure.js +23 -0
  309. package/src/providers/registry/baidu.js +35 -0
  310. package/src/providers/registry/blackbox.js +41 -0
  311. package/src/providers/registry/byteplus.js +36 -0
  312. package/src/providers/registry/cerebras.js +33 -0
  313. package/src/providers/registry/chutes.js +26 -0
  314. package/src/providers/registry/claude.js +80 -0
  315. package/src/providers/registry/cline.js +53 -0
  316. package/src/providers/registry/clinepass.js +62 -0
  317. package/src/providers/registry/cloudflare-ai.js +84 -0
  318. package/src/providers/registry/codebuddy-cn.js +95 -0
  319. package/src/providers/registry/codebuddy-intl.js +91 -0
  320. package/src/providers/registry/codex.js +145 -0
  321. package/src/providers/registry/commandcode.js +45 -0
  322. package/src/providers/registry/cursor.js +61 -0
  323. package/src/providers/registry/deepseek.js +55 -0
  324. package/src/providers/registry/featherless.js +36 -0
  325. package/src/providers/registry/fireworks.js +34 -0
  326. package/src/providers/registry/gemini-cli.js +59 -0
  327. package/src/providers/registry/gemini.js +76 -0
  328. package/src/providers/registry/github.js +131 -0
  329. package/src/providers/registry/glm-cn.js +35 -0
  330. package/src/providers/registry/glm.js +58 -0
  331. package/src/providers/registry/grok-cli.js +87 -0
  332. package/src/providers/registry/grok-web.js +41 -0
  333. package/src/providers/registry/groq.js +38 -0
  334. package/src/providers/registry/index.js +77 -0
  335. package/src/providers/registry/kilo-gateway.js +34 -0
  336. package/src/providers/registry/kilocode.js +48 -0
  337. package/src/providers/registry/kimchi.js +49 -0
  338. package/src/providers/registry/kimi.js +82 -0
  339. package/src/providers/registry/kiro.js +132 -0
  340. package/src/providers/registry/meta.js +34 -0
  341. package/src/providers/registry/minimax-cn.js +90 -0
  342. package/src/providers/registry/minimax.js +90 -0
  343. package/src/providers/registry/mistral.js +36 -0
  344. package/src/providers/registry/morph.js +31 -0
  345. package/src/providers/registry/nebius.js +26 -0
  346. package/src/providers/registry/nvidia.js +55 -0
  347. package/src/providers/registry/ollama-local.js +18 -0
  348. package/src/providers/registry/ollama.js +38 -0
  349. package/src/providers/registry/openai.js +114 -0
  350. package/src/providers/registry/opencode-go.js +41 -0
  351. package/src/providers/registry/opencode.js +26 -0
  352. package/src/providers/registry/openrouter.js +36 -0
  353. package/src/providers/registry/perplexity.js +56 -0
  354. package/src/providers/registry/poolside.js +30 -0
  355. package/src/providers/registry/qoder.js +58 -0
  356. package/src/providers/registry/sumopod.js +84 -0
  357. package/src/providers/registry/tencent.js +29 -0
  358. package/src/providers/registry/together.js +38 -0
  359. package/src/providers/registry/trae.js +71 -0
  360. package/src/providers/registry/venice.js +69 -0
  361. package/src/providers/registry/vercel-ai-gateway.js +37 -0
  362. package/src/providers/registry/vertex-partner.js +31 -0
  363. package/src/providers/registry/vertex.js +33 -0
  364. package/src/providers/registry/windsurf.js +129 -0
  365. package/src/providers/registry/xai.js +46 -0
  366. package/src/providers/registry/xiaomi-mimo.js +66 -0
  367. package/src/providers/registry/xiaomi-tokenplan.js +58 -0
  368. package/src/providers/registryEntry.js +3 -0
  369. package/src/providers/schema.js +33 -0
  370. package/src/providers/shared.js +114 -0
  371. package/src/providers/thinkingLevels.js +49 -0
  372. package/src/routes/api/auth/login/route.js +82 -0
  373. package/src/routes/api/auth/logout/route.js +9 -0
  374. package/src/routes/api/auth/reset-password/route.js +12 -0
  375. package/src/routes/api/auth/status/route.js +33 -0
  376. package/src/routes/api/auth/verify-password/route.js +39 -0
  377. package/src/routes/api/chat/tools/curl/route.js +100 -0
  378. package/src/routes/api/chat/tools/router/route.js +215 -0
  379. package/src/routes/api/chat/tools/search/route.js +362 -0
  380. package/src/routes/api/chat/tools/shell/route.js +228 -0
  381. package/src/routes/api/cli-tools/all-statuses/route.js +37 -0
  382. package/src/routes/api/cli-tools/claude-settings/route.js +261 -0
  383. package/src/routes/api/cli-tools/codex-settings/route.js +220 -0
  384. package/src/routes/api/cli-tools/cowork-mcp-registry/route.js +80 -0
  385. package/src/routes/api/cli-tools/cowork-mcp-tools/route.js +96 -0
  386. package/src/routes/api/cli-tools/cowork-settings/route.js +380 -0
  387. package/src/routes/api/cli-tools/grok-build-settings/route.js +152 -0
  388. package/src/routes/api/cli-tools/hermes-settings/route.js +169 -0
  389. package/src/routes/api/cli-tools/kilo-settings/route.js +131 -0
  390. package/src/routes/api/cli-tools/model-mappings/route.js +43 -0
  391. package/src/routes/api/cli-tools/omp-settings/route.js +266 -0
  392. package/src/routes/api/cli-tools/opencode-settings/route.js +243 -0
  393. package/src/routes/api/combos/[id]/route.js +73 -0
  394. package/src/routes/api/combos/route.js +43 -0
  395. package/src/routes/api/health/live/route.js +7 -0
  396. package/src/routes/api/health/ready/route.js +19 -0
  397. package/src/routes/api/health/route.js +17 -0
  398. package/src/routes/api/inflight/stream/route.js +24 -0
  399. package/src/routes/api/init/route.js +3 -0
  400. package/src/routes/api/keys/[id]/route.js +71 -0
  401. package/src/routes/api/keys/route.js +41 -0
  402. package/src/routes/api/logs/route.js +19 -0
  403. package/src/routes/api/mcp/[plugin]/message/route.js +23 -0
  404. package/src/routes/api/mcp/[plugin]/sse/route.js +35 -0
  405. package/src/routes/api/models/availability/route.js +104 -0
  406. package/src/routes/api/models/custom/route.js +58 -0
  407. package/src/routes/api/models/disabled/route.js +47 -0
  408. package/src/routes/api/models/route.js +64 -0
  409. package/src/routes/api/models/test/ping.js +236 -0
  410. package/src/routes/api/models/test/route.js +14 -0
  411. package/src/routes/api/oauth/[provider]/[action]/route.js +491 -0
  412. package/src/routes/api/oauth/codex/bulk-import/route.js +105 -0
  413. package/src/routes/api/oauth/codex/import-token/route.js +82 -0
  414. package/src/routes/api/oauth/cursor/auto-import/route.js +250 -0
  415. package/src/routes/api/oauth/cursor/import/route.js +86 -0
  416. package/src/routes/api/oauth/kiro/api-key/route.js +60 -0
  417. package/src/routes/api/oauth/kiro/auto-import/route.js +120 -0
  418. package/src/routes/api/oauth/kiro/import/route.js +72 -0
  419. package/src/routes/api/oauth/kiro/import-cli-proxy/route.js +42 -0
  420. package/src/routes/api/oauth/kiro/social-authorize/route.js +40 -0
  421. package/src/routes/api/oauth/kiro/social-exchange/route.js +62 -0
  422. package/src/routes/api/pricing/route.js +96 -0
  423. package/src/routes/api/provider-nodes/[id]/route.js +106 -0
  424. package/src/routes/api/provider-nodes/route.js +110 -0
  425. package/src/routes/api/provider-nodes/validate/route.js +197 -0
  426. package/src/routes/api/providers/[id]/models/route.js +716 -0
  427. package/src/routes/api/providers/[id]/route.js +195 -0
  428. package/src/routes/api/providers/[id]/test/route.js +27 -0
  429. package/src/routes/api/providers/[id]/test/testUtils.js +912 -0
  430. package/src/routes/api/providers/[id]/test-models/route.js +211 -0
  431. package/src/routes/api/providers/catalog/route.js +58 -0
  432. package/src/routes/api/providers/client/route.js +133 -0
  433. package/src/routes/api/providers/counts/route.js +43 -0
  434. package/src/routes/api/providers/kilo/free-models/route.js +58 -0
  435. package/src/routes/api/providers/route.js +361 -0
  436. package/src/routes/api/providers/suggested-models/filters.js +24 -0
  437. package/src/routes/api/providers/suggested-models/route.js +34 -0
  438. package/src/routes/api/providers/test-batch/route.js +145 -0
  439. package/src/routes/api/providers/validate/route.js +633 -0
  440. package/src/routes/api/proxy-pools/[id]/route.js +129 -0
  441. package/src/routes/api/proxy-pools/[id]/test/route.js +64 -0
  442. package/src/routes/api/proxy-pools/cloudflare-deploy/route.js +183 -0
  443. package/src/routes/api/proxy-pools/route.js +85 -0
  444. package/src/routes/api/proxy-pools/vercel-deploy/route.js +181 -0
  445. package/src/routes/api/settings/currency/route.js +29 -0
  446. package/src/routes/api/settings/database/path/route.js +8 -0
  447. package/src/routes/api/settings/database/route.js +89 -0
  448. package/src/routes/api/settings/migrate-9router/route.js +81 -0
  449. package/src/routes/api/settings/require-login/route.js +14 -0
  450. package/src/routes/api/settings/route.js +186 -0
  451. package/src/routes/api/shutdown/route.js +25 -0
  452. package/src/routes/api/tags/route.js +17 -0
  453. package/src/routes/api/translator/console-logs/route.js +28 -0
  454. package/src/routes/api/translator/console-logs/stream/route.js +95 -0
  455. package/src/routes/api/tunnel/disable/route.js +18 -0
  456. package/src/routes/api/tunnel/enable/route.js +32 -0
  457. package/src/routes/api/tunnel/status/route.js +23 -0
  458. package/src/routes/api/update/route.js +42 -0
  459. package/src/routes/api/usage/[connectionId]/codex-reset-credits/route.js +169 -0
  460. package/src/routes/api/usage/[connectionId]/route.js +222 -0
  461. package/src/routes/api/usage/chart/route.js +27 -0
  462. package/src/routes/api/usage/history/route.js +94 -0
  463. package/src/routes/api/usage/logs/route.js +14 -0
  464. package/src/routes/api/usage/providers/route.js +36 -0
  465. package/src/routes/api/usage/request-details/route.js +61 -0
  466. package/src/routes/api/usage/request-logs/route.js +15 -0
  467. package/src/routes/api/usage/stats/route.js +23 -0
  468. package/src/routes/api/usage/stream/route.js +78 -0
  469. package/src/routes/api/v1/api/chat/route.js +36 -0
  470. package/src/routes/api/v1/chat/completions/route.js +28 -0
  471. package/src/routes/api/v1/images/edits/route.js +70 -0
  472. package/src/routes/api/v1/images/generations/route.js +50 -0
  473. package/src/routes/api/v1/images/imageValidation.js +45 -0
  474. package/src/routes/api/v1/messages/count_tokens/route.js +87 -0
  475. package/src/routes/api/v1/messages/route.js +26 -0
  476. package/src/routes/api/v1/models/route.js +440 -0
  477. package/src/routes/api/v1/responses/compact/route.js +34 -0
  478. package/src/routes/api/v1/responses/route.js +26 -0
  479. package/src/routes/api/v1/route.js +1 -0
  480. package/src/routes/api/version/route.js +11 -0
  481. package/src/routes/table.js +1141 -0
  482. package/src/rtk/applyFilter.js +13 -0
  483. package/src/rtk/autodetect.js +104 -0
  484. package/src/rtk/caveman.js +6 -0
  485. package/src/rtk/cavemanPrompts.js +107 -0
  486. package/src/rtk/constants.js +51 -0
  487. package/src/rtk/filters/buildOutput.js +120 -0
  488. package/src/rtk/filters/dedupLog.js +43 -0
  489. package/src/rtk/filters/find.js +47 -0
  490. package/src/rtk/filters/gitDiff.js +90 -0
  491. package/src/rtk/filters/gitLog.js +88 -0
  492. package/src/rtk/filters/gitStatus.js +101 -0
  493. package/src/rtk/filters/grep.js +45 -0
  494. package/src/rtk/filters/ls.js +73 -0
  495. package/src/rtk/filters/readNumbered.js +22 -0
  496. package/src/rtk/filters/searchList.js +49 -0
  497. package/src/rtk/filters/smartTruncate.js +13 -0
  498. package/src/rtk/filters/tree.js +28 -0
  499. package/src/rtk/index.js +143 -0
  500. package/src/rtk/ponytail.js +6 -0
  501. package/src/rtk/ponytailPrompt.js +49 -0
  502. package/src/rtk/systemInject.js +89 -0
  503. package/src/server.ts +474 -0
  504. package/src/services/accountFallback.js +241 -0
  505. package/src/services/accountPool.js +389 -0
  506. package/src/services/capacityAdapter.js +140 -0
  507. package/src/services/cliModelMapping.js +84 -0
  508. package/src/services/clinepassModels.js +60 -0
  509. package/src/services/codebuddyModels.js +250 -0
  510. package/src/services/combo.js +482 -0
  511. package/src/services/copilotModels.js +126 -0
  512. package/src/services/currencyRate.js +56 -0
  513. package/src/services/cursorModels.js +197 -0
  514. package/src/services/grokCliModels.js +130 -0
  515. package/src/services/kimchiModels.js +175 -0
  516. package/src/services/kiroModels.js +246 -0
  517. package/src/services/model.js +83 -0
  518. package/src/services/oauthCredentialManager.js +153 -0
  519. package/src/services/projectId.js +240 -0
  520. package/src/services/provider.js +145 -0
  521. package/src/services/qoderModels.js +274 -0
  522. package/src/services/tokenRefresh/dedup.js +70 -0
  523. package/src/services/tokenRefresh/providers.js +626 -0
  524. package/src/services/tokenRefresh.js +274 -0
  525. package/src/services/usage/claude.js +134 -0
  526. package/src/services/usage/codebuddy-cn.js +212 -0
  527. package/src/services/usage/codex.js +193 -0
  528. package/src/services/usage/deepseek.js +103 -0
  529. package/src/services/usage/github.js +94 -0
  530. package/src/services/usage/google.js +218 -0
  531. package/src/services/usage/grok-cli.js +368 -0
  532. package/src/services/usage/grokCliQuotaFrame.js +170 -0
  533. package/src/services/usage/kimi.js +185 -0
  534. package/src/services/usage/kiro.js +172 -0
  535. package/src/services/usage/minimax.js +215 -0
  536. package/src/services/usage/misc.js +264 -0
  537. package/src/services/usage/shared.js +59 -0
  538. package/src/services/usage.js +56 -0
  539. package/src/shared/clineAuth.js +37 -0
  540. package/src/shared/constants/config.js +80 -0
  541. package/src/shared/constants/coworkPlugins.js +69 -0
  542. package/src/shared/constants/models.js +41 -0
  543. package/src/shared/constants/providers.js +128 -0
  544. package/src/shared/constants/providersDisplay.js +1 -0
  545. package/src/shared/qoder/constants.js +50 -0
  546. package/src/shared/qoder/cosy.js +134 -0
  547. package/src/shared/qoder/encoding.js +36 -0
  548. package/src/shared/services/initializeApp.js +226 -0
  549. package/src/shared/services/quotaAutoPing.js +310 -0
  550. package/src/shared/utils/apiKey.js +65 -0
  551. package/src/shared/utils/clineAuth.js +1 -0
  552. package/src/shared/utils/index.js +3 -0
  553. package/src/shared/utils/machineId.js +60 -0
  554. package/src/shared/utils/ssrfGuard.js +280 -0
  555. package/src/sse/handlers/chat.js +426 -0
  556. package/src/sse/oauthLease.js +32 -0
  557. package/src/sse/recovery.js +235 -0
  558. package/src/sse/services/auth.js +489 -0
  559. package/src/sse/services/backgroundTokenRefresh.js +173 -0
  560. package/src/sse/services/model.js +129 -0
  561. package/src/sse/services/requestCancellation.js +4 -0
  562. package/src/sse/services/tokenRefresh.js +264 -0
  563. package/src/sse/utils/logger.js +142 -0
  564. package/src/staticServe.ts +95 -0
  565. package/src/transformer/streamToJsonConverter.js +87 -0
  566. package/src/translator/canonical.js +81 -0
  567. package/src/translator/concerns/chunk.js +9 -0
  568. package/src/translator/concerns/compatFallback.js +147 -0
  569. package/src/translator/concerns/finishReason.js +59 -0
  570. package/src/translator/concerns/image.js +105 -0
  571. package/src/translator/concerns/json.js +4 -0
  572. package/src/translator/concerns/kiroConversation.js +428 -0
  573. package/src/translator/concerns/message.js +5 -0
  574. package/src/translator/concerns/modality.js +134 -0
  575. package/src/translator/concerns/paramSupport.js +97 -0
  576. package/src/translator/concerns/prefetch.js +86 -0
  577. package/src/translator/concerns/reasoning.js +18 -0
  578. package/src/translator/concerns/thinking.js +40 -0
  579. package/src/translator/concerns/thinkingUnified.js +319 -0
  580. package/src/translator/concerns/toolCall.js +134 -0
  581. package/src/translator/concerns/usage.js +61 -0
  582. package/src/translator/detection.js +42 -0
  583. package/src/translator/formats/claude.js +323 -0
  584. package/src/translator/formats/gemini.js +366 -0
  585. package/src/translator/formats/maxTokens.js +19 -0
  586. package/src/translator/formats/openai.js +116 -0
  587. package/src/translator/formats/responsesApi.js +117 -0
  588. package/src/translator/formats.js +28 -0
  589. package/src/translator/index.js +237 -0
  590. package/src/translator/request/antigravity-to-openai.js +217 -0
  591. package/src/translator/request/claude-to-kiro.js +288 -0
  592. package/src/translator/request/claude-to-openai.js +199 -0
  593. package/src/translator/request/gemini-to-openai.js +142 -0
  594. package/src/translator/request/openai-responses.js +373 -0
  595. package/src/translator/request/openai-to-claude.js +260 -0
  596. package/src/translator/request/openai-to-commandcode.js +161 -0
  597. package/src/translator/request/openai-to-cursor.js +175 -0
  598. package/src/translator/request/openai-to-gemini.js +422 -0
  599. package/src/translator/request/openai-to-kiro.js +360 -0
  600. package/src/translator/request/openai-to-ollama.js +152 -0
  601. package/src/translator/request/openai-to-vertex.js +36 -0
  602. package/src/translator/response/claude-to-openai.js +185 -0
  603. package/src/translator/response/commandcode-to-openai.js +164 -0
  604. package/src/translator/response/cursor-to-openai.js +18 -0
  605. package/src/translator/response/gemini-to-openai.js +127 -0
  606. package/src/translator/response/kiro-to-claude.js +230 -0
  607. package/src/translator/response/kiro-to-openai.js +136 -0
  608. package/src/translator/response/ollama-to-openai.js +106 -0
  609. package/src/translator/response/openai-responses.js +534 -0
  610. package/src/translator/response/openai-to-antigravity.js +107 -0
  611. package/src/translator/response/openai-to-claude.js +231 -0
  612. package/src/translator/schema/blocks.js +40 -0
  613. package/src/translator/schema/defaults.js +3 -0
  614. package/src/translator/schema/finishReasons.js +22 -0
  615. package/src/translator/schema/index.js +7 -0
  616. package/src/translator/schema/roles.js +12 -0
  617. package/src/transport/requestBoundary.ts +223 -0
  618. package/src/transport/routerAdmission.ts +380 -0
  619. package/src/types/provider.ts +59 -0
  620. package/src/utils/bypassHandler.js +272 -0
  621. package/src/utils/claudeCloaking.js +10 -0
  622. package/src/utils/claudeSignature.js +35 -0
  623. package/src/utils/clientDetector.js +183 -0
  624. package/src/utils/cursorChecksum.js +104 -0
  625. package/src/utils/cursorProtobuf.js +801 -0
  626. package/src/utils/debugLog.js +12 -0
  627. package/src/utils/error.js +120 -0
  628. package/src/utils/inflight.js +127 -0
  629. package/src/utils/kiroSessionReplay.js +139 -0
  630. package/src/utils/ollamaTransform.js +83 -0
  631. package/src/utils/proxyFetch.js +308 -0
  632. package/src/utils/reasoningContentInjector.js +71 -0
  633. package/src/utils/requestLogger.js +272 -0
  634. package/src/utils/responsesStreamHelpers.js +46 -0
  635. package/src/utils/sessionManager.js +205 -0
  636. package/src/utils/sse.js +13 -0
  637. package/src/utils/sseConstants.js +21 -0
  638. package/src/utils/stream.js +511 -0
  639. package/src/utils/streamHandler.js +256 -0
  640. package/src/utils/streamHelpers.js +113 -0
  641. package/src/utils/toolDeduper.js +44 -0
  642. package/src/utils/usageTracking.js +329 -0
  643. 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;