@neurocode-ai/core 1.18.8

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 (490) hide show
  1. package/bin/neurocode +199 -0
  2. package/bunfig.toml +2 -0
  3. package/drizzle.config.ts +10 -0
  4. package/package.json +131 -0
  5. package/schema.json +2071 -0
  6. package/script/fix-node-pty.ts +28 -0
  7. package/script/migration.ts +196 -0
  8. package/src/account/sql.ts +39 -0
  9. package/src/account.ts +101 -0
  10. package/src/agent.ts +111 -0
  11. package/src/aisdk.ts +235 -0
  12. package/src/background-job.ts +365 -0
  13. package/src/catalog.ts +301 -0
  14. package/src/command.ts +64 -0
  15. package/src/config/agent.ts +25 -0
  16. package/src/config/attachments.ts +15 -0
  17. package/src/config/command.ts +12 -0
  18. package/src/config/compaction.ts +15 -0
  19. package/src/config/experimental.ts +18 -0
  20. package/src/config/formatter.ts +12 -0
  21. package/src/config/lsp.ts +18 -0
  22. package/src/config/markdown.ts +36 -0
  23. package/src/config/mcp.ts +48 -0
  24. package/src/config/plugin/agent.ts +179 -0
  25. package/src/config/plugin/command.ts +83 -0
  26. package/src/config/plugin/external.ts +91 -0
  27. package/src/config/plugin/provider.ts +113 -0
  28. package/src/config/plugin/reference.ts +70 -0
  29. package/src/config/plugin/skill.ts +50 -0
  30. package/src/config/plugin.ts +13 -0
  31. package/src/config/provider.ts +71 -0
  32. package/src/config/reference.ts +22 -0
  33. package/src/config/tool-output.ts +9 -0
  34. package/src/config/watcher.ts +7 -0
  35. package/src/config.ts +227 -0
  36. package/src/control-plane/move-session.ts +148 -0
  37. package/src/control-plane/workspace.sql.ts +20 -0
  38. package/src/credential/sql.ts +14 -0
  39. package/src/credential.ts +138 -0
  40. package/src/cross-spawn-spawner.ts +507 -0
  41. package/src/data-migration.sql.ts +6 -0
  42. package/src/database/database.ts +57 -0
  43. package/src/database/migration/20260127222353_familiar_lady_ursula.ts +107 -0
  44. package/src/database/migration/20260211171708_add_project_commands.ts +11 -0
  45. package/src/database/migration/20260213144116_wakeful_the_professor.ts +23 -0
  46. package/src/database/migration/20260225215848_workspace.ts +19 -0
  47. package/src/database/migration/20260227213759_add_session_workspace_id.ts +12 -0
  48. package/src/database/migration/20260228203230_blue_harpoon.ts +30 -0
  49. package/src/database/migration/20260303231226_add_workspace_fields.ts +15 -0
  50. package/src/database/migration/20260309230000_move_org_to_state.ts +15 -0
  51. package/src/database/migration/20260312043431_session_message_cursor.ts +16 -0
  52. package/src/database/migration/20260323234822_events.ts +26 -0
  53. package/src/database/migration/20260410174513_workspace-name.ts +29 -0
  54. package/src/database/migration/20260413175956_chief_energizer.ts +24 -0
  55. package/src/database/migration/20260423070820_add_icon_url_override.ts +14 -0
  56. package/src/database/migration/20260427172553_slow_nightmare.ts +30 -0
  57. package/src/database/migration/20260428004200_add_session_path.ts +11 -0
  58. package/src/database/migration/20260501142318_next_venus.ts +12 -0
  59. package/src/database/migration/20260504145000_add_sync_owner.ts +11 -0
  60. package/src/database/migration/20260507164347_add_workspace_time.ts +11 -0
  61. package/src/database/migration/20260510033149_session_usage.ts +56 -0
  62. package/src/database/migration/20260511000411_data_migration_state.ts +16 -0
  63. package/src/database/migration/20260511173437_session-metadata.ts +16 -0
  64. package/src/database/migration/20260601010001_normalize_storage_paths.ts +22 -0
  65. package/src/database/migration/20260601202201_amazing_prowler.ts +11 -0
  66. package/src/database/migration/20260602002951_lowly_union_jack.ts +24 -0
  67. package/src/database/migration/20260602182828_add_project_directories.ts +20 -0
  68. package/src/database/migration/20260603001617_session_message_projection_indexes.ts +19 -0
  69. package/src/database/migration/20260603040000_session_message_projection_order.ts +19 -0
  70. package/src/database/migration/20260603141458_session_input_inbox.ts +25 -0
  71. package/src/database/migration/20260603160727_jittery_ezekiel_stane.ts +20 -0
  72. package/src/database/migration/20260604172448_event_sourced_session_input.ts +47 -0
  73. package/src/database/migration/20260605003541_add_session_context_snapshot.ts +21 -0
  74. package/src/database/migration/20260605042240_add_context_epoch_agent.ts +11 -0
  75. package/src/database/migration/20260611035744_credential.ts +25 -0
  76. package/src/database/migration/20260611192811_lush_chimera.ts +25 -0
  77. package/src/database/migration/20260612174303_project_dir_strategy.ts +29 -0
  78. package/src/database/migration/20260622142730_simplify_session_context_epoch.ts +13 -0
  79. package/src/database/migration/20260622170816_reset_v2_session_state.ts +17 -0
  80. package/src/database/migration/20260622202450_simplify_session_input.ts +17 -0
  81. package/src/database/migration.gen.ts +44 -0
  82. package/src/database/migration.ts +81 -0
  83. package/src/database/path.ts +91 -0
  84. package/src/database/schema.gen.ts +274 -0
  85. package/src/database/schema.sql.ts +10 -0
  86. package/src/database/sqlite.bun.ts +183 -0
  87. package/src/database/sqlite.node.ts +178 -0
  88. package/src/database/sqlite.ts +8 -0
  89. package/src/effect/app-node-builder.ts +23 -0
  90. package/src/effect/app-node-platform.ts +18 -0
  91. package/src/effect/app-node.ts +14 -0
  92. package/src/effect/dfdf +1 -0
  93. package/src/effect/keyed-mutex.ts +45 -0
  94. package/src/effect/layer-node.ts +333 -0
  95. package/src/effect/memo-map.ts +3 -0
  96. package/src/effect/runtime.ts +21 -0
  97. package/src/effect/service-use.ts +43 -0
  98. package/src/event/sql.ts +25 -0
  99. package/src/event.ts +638 -0
  100. package/src/file-mutation.ts +207 -0
  101. package/src/file.ts +6 -0
  102. package/src/filesystem/fff.bun.ts +140 -0
  103. package/src/filesystem/fff.node.ts +138 -0
  104. package/src/filesystem/ignore.ts +67 -0
  105. package/src/filesystem/protected.ts +53 -0
  106. package/src/filesystem/search.ts +239 -0
  107. package/src/filesystem/watcher.ts +140 -0
  108. package/src/filesystem.ts +118 -0
  109. package/src/flag/flag.ts +78 -0
  110. package/src/fs-util.ts +274 -0
  111. package/src/git.ts +987 -0
  112. package/src/github-copilot/README.md +5 -0
  113. package/src/github-copilot/chat/convert-to-openai-compatible-chat-messages.ts +170 -0
  114. package/src/github-copilot/chat/get-response-metadata.ts +15 -0
  115. package/src/github-copilot/chat/map-openai-compatible-finish-reason.ts +19 -0
  116. package/src/github-copilot/chat/openai-compatible-api-types.ts +64 -0
  117. package/src/github-copilot/chat/openai-compatible-chat-language-model.ts +815 -0
  118. package/src/github-copilot/chat/openai-compatible-chat-options.ts +28 -0
  119. package/src/github-copilot/chat/openai-compatible-metadata-extractor.ts +44 -0
  120. package/src/github-copilot/chat/openai-compatible-prepare-tools.ts +83 -0
  121. package/src/github-copilot/copilot-provider.ts +100 -0
  122. package/src/github-copilot/openai-compatible-error.ts +27 -0
  123. package/src/github-copilot/responses/convert-to-openai-responses-input.ts +335 -0
  124. package/src/github-copilot/responses/map-openai-responses-finish-reason.ts +22 -0
  125. package/src/github-copilot/responses/openai-config.ts +18 -0
  126. package/src/github-copilot/responses/openai-error.ts +22 -0
  127. package/src/github-copilot/responses/openai-responses-api-types.ts +214 -0
  128. package/src/github-copilot/responses/openai-responses-language-model.ts +1770 -0
  129. package/src/github-copilot/responses/openai-responses-prepare-tools.ts +173 -0
  130. package/src/github-copilot/responses/openai-responses-settings.ts +1 -0
  131. package/src/github-copilot/responses/tool/code-interpreter.ts +87 -0
  132. package/src/github-copilot/responses/tool/file-search.ts +127 -0
  133. package/src/github-copilot/responses/tool/image-generation.ts +114 -0
  134. package/src/github-copilot/responses/tool/local-shell.ts +64 -0
  135. package/src/github-copilot/responses/tool/web-search-preview.ts +103 -0
  136. package/src/github-copilot/responses/tool/web-search.ts +102 -0
  137. package/src/global.ts +87 -0
  138. package/src/id/id.ts +47 -0
  139. package/src/image/photon.ts +94 -0
  140. package/src/image.ts +81 -0
  141. package/src/installation/version.ts +8 -0
  142. package/src/instruction-context.ts +101 -0
  143. package/src/integration/connection.ts +12 -0
  144. package/src/integration.ts +520 -0
  145. package/src/location-mutation.ts +162 -0
  146. package/src/location-service-map.ts +18 -0
  147. package/src/location-services.ts +115 -0
  148. package/src/location.ts +39 -0
  149. package/src/markdown.d.ts +4 -0
  150. package/src/model.ts +41 -0
  151. package/src/models-dev.ts +260 -0
  152. package/src/npm-config.ts +40 -0
  153. package/src/npm.ts +269 -0
  154. package/src/oauth/page.ts +276 -0
  155. package/src/observability/logging.ts +71 -0
  156. package/src/observability/otlp.ts +79 -0
  157. package/src/observability/shared.ts +1 -0
  158. package/src/observability.ts +24 -0
  159. package/src/patch.ts +197 -0
  160. package/src/permission/saved.ts +79 -0
  161. package/src/permission/sql.ts +20 -0
  162. package/src/permission.ts +310 -0
  163. package/src/plugin/agent.ts +206 -0
  164. package/src/plugin/command/initialize.txt +65 -0
  165. package/src/plugin/command/review.txt +100 -0
  166. package/src/plugin/command.ts +25 -0
  167. package/src/plugin/host.ts +219 -0
  168. package/src/plugin/internal.ts +153 -0
  169. package/src/plugin/layer-map.example.ts +94 -0
  170. package/src/plugin/models-dev.ts +183 -0
  171. package/src/plugin/promise.ts +93 -0
  172. package/src/plugin/provider/alibaba.ts +15 -0
  173. package/src/plugin/provider/amazon-bedrock.ts +126 -0
  174. package/src/plugin/provider/anthropic.ts +27 -0
  175. package/src/plugin/provider/azure.ts +82 -0
  176. package/src/plugin/provider/cerebras.ts +26 -0
  177. package/src/plugin/provider/cloudflare-ai-gateway.ts +81 -0
  178. package/src/plugin/provider/cloudflare-workers-ai.ts +81 -0
  179. package/src/plugin/provider/cohere.ts +15 -0
  180. package/src/plugin/provider/deepinfra.ts +15 -0
  181. package/src/plugin/provider/dynamic.ts +31 -0
  182. package/src/plugin/provider/gateway.ts +15 -0
  183. package/src/plugin/provider/github-copilot.ts +52 -0
  184. package/src/plugin/provider/gitlab.ts +65 -0
  185. package/src/plugin/provider/google-vertex.ts +173 -0
  186. package/src/plugin/provider/google.ts +15 -0
  187. package/src/plugin/provider/groq.ts +15 -0
  188. package/src/plugin/provider/kilo.ts +21 -0
  189. package/src/plugin/provider/llmgateway.ts +26 -0
  190. package/src/plugin/provider/mistral.ts +15 -0
  191. package/src/plugin/provider/nvidia.ts +22 -0
  192. package/src/plugin/provider/openai-compatible.ts +17 -0
  193. package/src/plugin/provider/openai.ts +292 -0
  194. package/src/plugin/provider/opencode.ts +311 -0
  195. package/src/plugin/provider/openrouter.ts +36 -0
  196. package/src/plugin/provider/perplexity.ts +15 -0
  197. package/src/plugin/provider/sap-ai-core.ts +46 -0
  198. package/src/plugin/provider/snowflake-cortex.ts +89 -0
  199. package/src/plugin/provider/togetherai.ts +15 -0
  200. package/src/plugin/provider/venice.ts +15 -0
  201. package/src/plugin/provider/vercel.ts +27 -0
  202. package/src/plugin/provider/xai.ts +22 -0
  203. package/src/plugin/provider/zenmux.ts +21 -0
  204. package/src/plugin/provider.ts +71 -0
  205. package/src/plugin/skill/customize-neurocode.md +452 -0
  206. package/src/plugin/skill.ts +31 -0
  207. package/src/plugin/variant.ts +39 -0
  208. package/src/plugin.ts +167 -0
  209. package/src/policy.ts +49 -0
  210. package/src/process.ts +261 -0
  211. package/src/project/copy-strategies.ts +35 -0
  212. package/src/project/copy.ts +292 -0
  213. package/src/project/directories.ts +158 -0
  214. package/src/project/schema.ts +16 -0
  215. package/src/project/sql.ts +35 -0
  216. package/src/project.ts +136 -0
  217. package/src/provider.ts +25 -0
  218. package/src/pty/protocol.ts +37 -0
  219. package/src/pty/pty.bun.ts +26 -0
  220. package/src/pty/pty.node.ts +29 -0
  221. package/src/pty/pty.ts +25 -0
  222. package/src/pty/schema.ts +1 -0
  223. package/src/pty/ticket.ts +56 -0
  224. package/src/pty.ts +318 -0
  225. package/src/public-event-manifest.ts +7 -0
  226. package/src/question.ts +153 -0
  227. package/src/reference/guidance.ts +69 -0
  228. package/src/reference.ts +125 -0
  229. package/src/repository-cache.ts +259 -0
  230. package/src/repository.ts +214 -0
  231. package/src/ripgrep/binary.ts +132 -0
  232. package/src/ripgrep.ts +281 -0
  233. package/src/schema.ts +79 -0
  234. package/src/session/compaction.ts +241 -0
  235. package/src/session/context-epoch.ts +174 -0
  236. package/src/session/error.ts +24 -0
  237. package/src/session/event.ts +2 -0
  238. package/src/session/execution/local.ts +46 -0
  239. package/src/session/execution.ts +34 -0
  240. package/src/session/history.ts +101 -0
  241. package/src/session/info.ts +50 -0
  242. package/src/session/input.ts +288 -0
  243. package/src/session/message-updater.ts +397 -0
  244. package/src/session/message.ts +2 -0
  245. package/src/session/projector.ts +458 -0
  246. package/src/session/prompt.ts +1 -0
  247. package/src/session/revert.ts +121 -0
  248. package/src/session/run-coordinator.ts +104 -0
  249. package/src/session/runner/index.ts +28 -0
  250. package/src/session/runner/llm.ts +432 -0
  251. package/src/session/runner/max-steps.ts +16 -0
  252. package/src/session/runner/model.ts +218 -0
  253. package/src/session/runner/publish-llm-event.ts +423 -0
  254. package/src/session/runner/to-llm-message.ts +171 -0
  255. package/src/session/schema.ts +9 -0
  256. package/src/session/sql.ts +176 -0
  257. package/src/session/store.ts +63 -0
  258. package/src/session/todo.ts +78 -0
  259. package/src/session.ts +486 -0
  260. package/src/share/sql.ts +13 -0
  261. package/src/shell.ts +226 -0
  262. package/src/skill/discovery.ts +213 -0
  263. package/src/skill/guidance.ts +76 -0
  264. package/src/skill.ts +132 -0
  265. package/src/snapshot.ts +266 -0
  266. package/src/state.ts +128 -0
  267. package/src/system-context/builtins.ts +50 -0
  268. package/src/system-context/index.ts +320 -0
  269. package/src/system-context/registry.ts +49 -0
  270. package/src/tool/AGENTS.md +59 -0
  271. package/src/tool/application-tools.ts +57 -0
  272. package/src/tool/apply-patch.ts +219 -0
  273. package/src/tool/bash.ts +207 -0
  274. package/src/tool/builtins.ts +48 -0
  275. package/src/tool/edit.ts +223 -0
  276. package/src/tool/glob.ts +105 -0
  277. package/src/tool/grep.ts +137 -0
  278. package/src/tool/http-body.ts +30 -0
  279. package/src/tool/question.ts +94 -0
  280. package/src/tool/read-filesystem.ts +366 -0
  281. package/src/tool/read.ts +117 -0
  282. package/src/tool/registry.ts +147 -0
  283. package/src/tool/skill.ts +109 -0
  284. package/src/tool/todowrite.ts +62 -0
  285. package/src/tool/tool.ts +162 -0
  286. package/src/tool/tools.ts +13 -0
  287. package/src/tool/webfetch.ts +218 -0
  288. package/src/tool/websearch.ts +260 -0
  289. package/src/tool/write.ts +101 -0
  290. package/src/tool-output-store.ts +211 -0
  291. package/src/util/array.ts +10 -0
  292. package/src/util/binary.ts +41 -0
  293. package/src/util/effect-flock.ts +284 -0
  294. package/src/util/encode.ts +51 -0
  295. package/src/util/error.ts +70 -0
  296. package/src/util/flock.ts +358 -0
  297. package/src/util/glob.ts +34 -0
  298. package/src/util/hash.ts +11 -0
  299. package/src/util/identifier.ts +1 -0
  300. package/src/util/iife.ts +3 -0
  301. package/src/util/lazy.ts +11 -0
  302. package/src/util/module.ts +10 -0
  303. package/src/util/path.ts +37 -0
  304. package/src/util/retry.ts +42 -0
  305. package/src/util/slug.ts +74 -0
  306. package/src/util/token.ts +5 -0
  307. package/src/util/which.ts +14 -0
  308. package/src/util/wildcard.ts +14 -0
  309. package/src/v1/config/agent.ts +89 -0
  310. package/src/v1/config/attachment.ts +25 -0
  311. package/src/v1/config/command.ts +13 -0
  312. package/src/v1/config/config.ts +192 -0
  313. package/src/v1/config/console-state.ts +16 -0
  314. package/src/v1/config/error.ts +39 -0
  315. package/src/v1/config/formatter.ts +13 -0
  316. package/src/v1/config/layout.ts +6 -0
  317. package/src/v1/config/lsp.ts +80 -0
  318. package/src/v1/config/mcp.ts +63 -0
  319. package/src/v1/config/migrate.ts +254 -0
  320. package/src/v1/config/permission.ts +50 -0
  321. package/src/v1/config/plugin.ts +9 -0
  322. package/src/v1/config/provider-options.ts +227 -0
  323. package/src/v1/config/provider.ts +121 -0
  324. package/src/v1/config/server.ts +19 -0
  325. package/src/v1/config/skills.ts +13 -0
  326. package/src/v1/permission.ts +33 -0
  327. package/src/v1/session.ts +68 -0
  328. package/src/v2-schema.ts +3 -0
  329. package/src/workspace.ts +6 -0
  330. package/sst-env.d.ts +10 -0
  331. package/test/agent.test.ts +131 -0
  332. package/test/application-tools.test.ts +287 -0
  333. package/test/background-job.test.ts +106 -0
  334. package/test/catalog.test.ts +353 -0
  335. package/test/command.test.ts +56 -0
  336. package/test/config/agent.test.ts +349 -0
  337. package/test/config/command.test.ts +83 -0
  338. package/test/config/config.test.ts +798 -0
  339. package/test/config/fixtures/plugin/directory-plugin.ts +13 -0
  340. package/test/config/plugin.test.ts +248 -0
  341. package/test/config/provider-options.test.ts +224 -0
  342. package/test/config/provider.test.ts +269 -0
  343. package/test/config/skill.test.ts +80 -0
  344. package/test/credential.test.ts +36 -0
  345. package/test/database-migration.test.ts +644 -0
  346. package/test/effect/cross-spawn-spawner.test.ts +426 -0
  347. package/test/effect/keyed-mutex.test.ts +73 -0
  348. package/test/effect/layer-node/layer-node-types.test.ts +143 -0
  349. package/test/effect/layer-node/layer-node.test.ts +261 -0
  350. package/test/effect/layer-node/node-build.test.ts +125 -0
  351. package/test/effect/observability.test.ts +109 -0
  352. package/test/event.test.ts +1124 -0
  353. package/test/file-mutation.test.ts +368 -0
  354. package/test/filesystem/filesystem.test.ts +387 -0
  355. package/test/filesystem/ignore.test.ts +10 -0
  356. package/test/filesystem/search.test.ts +44 -0
  357. package/test/filesystem/watcher.test.ts +266 -0
  358. package/test/fixture/effect-flock-worker.ts +60 -0
  359. package/test/fixture/flock-worker.ts +72 -0
  360. package/test/fixture/git.ts +49 -0
  361. package/test/fixture/location.ts +26 -0
  362. package/test/fixture/tmpdir.ts +25 -0
  363. package/test/fixtures/recordings/session-runner/openai-chat-streams-text.json +27 -0
  364. package/test/git.test.ts +164 -0
  365. package/test/github-copilot/convert-to-copilot-messages.test.ts +523 -0
  366. package/test/github-copilot/copilot-chat-model.test.ts +592 -0
  367. package/test/github-copilot/openai-responses-language-model.test.ts +206 -0
  368. package/test/global.test.ts +16 -0
  369. package/test/instruction-context.test.ts +323 -0
  370. package/test/integration.test.ts +349 -0
  371. package/test/legacy-event-schema.test.ts +16 -0
  372. package/test/lib/effect.ts +53 -0
  373. package/test/lib/tool.ts +20 -0
  374. package/test/location-filesystem.test.ts +69 -0
  375. package/test/location-layer.test.ts +226 -0
  376. package/test/location-mutation.test.ts +178 -0
  377. package/test/location.test.ts +42 -0
  378. package/test/model.test.ts +23 -0
  379. package/test/models.test.ts +290 -0
  380. package/test/move-session.test.ts +235 -0
  381. package/test/npm-config.test.ts +51 -0
  382. package/test/npm.test.ts +84 -0
  383. package/test/oauth-page.test.ts +15 -0
  384. package/test/patch.test.ts +68 -0
  385. package/test/permission.test.ts +315 -0
  386. package/test/plugin/command.test.ts +47 -0
  387. package/test/plugin/fixture.ts +52 -0
  388. package/test/plugin/fixtures/config-effect-plugin.ts +15 -0
  389. package/test/plugin/fixtures/config-promise-plugin.ts +13 -0
  390. package/test/plugin/fixtures/invalid-plugin.ts +1 -0
  391. package/test/plugin/fixtures/models-dev.json +14 -0
  392. package/test/plugin/fixtures/provider-factory.ts +9 -0
  393. package/test/plugin/host.ts +303 -0
  394. package/test/plugin/models-dev.test.ts +170 -0
  395. package/test/plugin/promise.test.ts +67 -0
  396. package/test/plugin/provider-alibaba.test.ts +92 -0
  397. package/test/plugin/provider-amazon-bedrock.test.ts +622 -0
  398. package/test/plugin/provider-anthropic.test.ts +94 -0
  399. package/test/plugin/provider-azure-cognitive-services.test.ts +199 -0
  400. package/test/plugin/provider-azure.test.ts +293 -0
  401. package/test/plugin/provider-cerebras.test.ts +136 -0
  402. package/test/plugin/provider-cloudflare-ai-gateway.test.ts +426 -0
  403. package/test/plugin/provider-cloudflare-workers-ai.test.ts +279 -0
  404. package/test/plugin/provider-cohere.test.ts +123 -0
  405. package/test/plugin/provider-deepinfra.test.ts +159 -0
  406. package/test/plugin/provider-dynamic.test.ts +184 -0
  407. package/test/plugin/provider-gateway.test.ts +115 -0
  408. package/test/plugin/provider-github-copilot.test.ts +278 -0
  409. package/test/plugin/provider-gitlab.test.ts +308 -0
  410. package/test/plugin/provider-google-vertex-anthropic.test.ts +264 -0
  411. package/test/plugin/provider-google-vertex.test.ts +387 -0
  412. package/test/plugin/provider-google.test.ts +80 -0
  413. package/test/plugin/provider-groq.test.ts +121 -0
  414. package/test/plugin/provider-kilo.test.ts +99 -0
  415. package/test/plugin/provider-llmgateway.test.ts +80 -0
  416. package/test/plugin/provider-mistral.test.ts +129 -0
  417. package/test/plugin/provider-nvidia.test.ts +97 -0
  418. package/test/plugin/provider-openai-compatible.test.ts +107 -0
  419. package/test/plugin/provider-openai.test.ts +176 -0
  420. package/test/plugin/provider-opencode.test.ts +423 -0
  421. package/test/plugin/provider-openrouter.test.ts +113 -0
  422. package/test/plugin/provider-perplexity.test.ts +122 -0
  423. package/test/plugin/provider-sap-ai-core.test.ts +159 -0
  424. package/test/plugin/provider-snowflake-cortex.test.ts +261 -0
  425. package/test/plugin/provider-togetherai.test.ts +129 -0
  426. package/test/plugin/provider-venice.test.ts +115 -0
  427. package/test/plugin/provider-vercel.test.ts +83 -0
  428. package/test/plugin/provider-xai.test.ts +125 -0
  429. package/test/plugin/provider-zenmux.test.ts +110 -0
  430. package/test/plugin/skill.test.ts +25 -0
  431. package/test/plugin/variant.test.ts +67 -0
  432. package/test/plugin.test.ts +71 -0
  433. package/test/policy.test.ts +85 -0
  434. package/test/preload.ts +5 -0
  435. package/test/process/process.test.ts +368 -0
  436. package/test/project-copy.test.ts +400 -0
  437. package/test/project-directories.test.ts +61 -0
  438. package/test/project.test.ts +221 -0
  439. package/test/provider-mistral.test.ts +282 -0
  440. package/test/provider-xai-responses.test.ts +32 -0
  441. package/test/pty/info-schema.test.ts +32 -0
  442. package/test/pty/protocol.test.ts +27 -0
  443. package/test/pty/pty-session.test.ts +240 -0
  444. package/test/pty/ticket.test.ts +62 -0
  445. package/test/question.test.ts +114 -0
  446. package/test/reference-guidance.test.ts +76 -0
  447. package/test/reference.test.ts +78 -0
  448. package/test/repository-cache.test.ts +153 -0
  449. package/test/repository.test.ts +71 -0
  450. package/test/ripgrep.test.ts +65 -0
  451. package/test/session-compaction.test.ts +27 -0
  452. package/test/session-create.test.ts +425 -0
  453. package/test/session-history.test.ts +165 -0
  454. package/test/session-projector.test.ts +533 -0
  455. package/test/session-prompt.test.ts +584 -0
  456. package/test/session-run-coordinator.test.ts +418 -0
  457. package/test/session-runner-message.test.ts +501 -0
  458. package/test/session-runner-model.test.ts +347 -0
  459. package/test/session-runner-recorded.test.ts +193 -0
  460. package/test/session-runner-tool-events.test.ts +136 -0
  461. package/test/session-runner-tool-registry.test.ts +452 -0
  462. package/test/session-runner.test.ts +3365 -0
  463. package/test/session-todo.test.ts +94 -0
  464. package/test/session-tool-progress.test.ts +158 -0
  465. package/test/shared-schema.test.ts +206 -0
  466. package/test/shell.test.ts +108 -0
  467. package/test/skill/guidance.test.ts +144 -0
  468. package/test/skill-discovery.test.ts +165 -0
  469. package/test/skill.test.ts +125 -0
  470. package/test/snapshot.test.ts +178 -0
  471. package/test/state.test.ts +115 -0
  472. package/test/system-context/builtins.test.ts +129 -0
  473. package/test/system-context/index.test.ts +307 -0
  474. package/test/system-context/registry.test.ts +114 -0
  475. package/test/tool-apply-patch.test.ts +436 -0
  476. package/test/tool-bash.test.ts +437 -0
  477. package/test/tool-edit.test.ts +434 -0
  478. package/test/tool-output-store.test.ts +247 -0
  479. package/test/tool-question.test.ts +162 -0
  480. package/test/tool-read-filesystem.test.ts +118 -0
  481. package/test/tool-read.test.ts +674 -0
  482. package/test/tool-skill.test.ts +149 -0
  483. package/test/tool-todowrite.test.ts +125 -0
  484. package/test/tool-webfetch.test.ts +281 -0
  485. package/test/tool-websearch.test.ts +316 -0
  486. package/test/tool-write.test.ts +305 -0
  487. package/test/util/effect-flock.test.ts +392 -0
  488. package/test/util/flock.test.ts +428 -0
  489. package/test/util/which.test.ts +100 -0
  490. package/tsconfig.json +7 -0
@@ -0,0 +1,25 @@
1
+ export * as CommandPlugin from "./command"
2
+
3
+ import { define } from "./internal"
4
+ import { Effect } from "effect"
5
+ import { Location } from "../location"
6
+ import PROMPT_INITIALIZE from "./command/initialize.txt"
7
+ import PROMPT_REVIEW from "./command/review.txt"
8
+
9
+ export const Plugin = define({
10
+ id: "command",
11
+ effect: Effect.fn(function* (ctx) {
12
+ const location = yield* Location.Service
13
+ yield* ctx.command.transform((draft) => {
14
+ draft.update("init", (command) => {
15
+ command.template = PROMPT_INITIALIZE.replace("${path}", location.project.directory)
16
+ command.description = "guided AGENTS.md setup"
17
+ })
18
+ draft.update("review", (command) => {
19
+ command.template = PROMPT_REVIEW.replace("${path}", location.project.directory)
20
+ command.description = "review changes [commit|branch|pr], defaults to uncommitted"
21
+ command.subtask = true
22
+ })
23
+ })
24
+ }),
25
+ })
@@ -0,0 +1,219 @@
1
+ export * as PluginHost from "./host"
2
+
3
+ import type { PluginContext as Interface } from "@neurocode-ai/plugin/v2/effect"
4
+ import { Effect, Schema } from "effect"
5
+ import { AgentV2 } from "../agent"
6
+ import { AISDK } from "../aisdk"
7
+ import { Catalog } from "../catalog"
8
+ import { CommandV2 } from "../command"
9
+ import { Credential } from "../credential"
10
+ import { Integration } from "../integration"
11
+ import { ModelV2 } from "../model"
12
+ import { PluginV2 } from "../plugin"
13
+ import { ProviderV2 } from "../provider"
14
+ import { Reference } from "../reference"
15
+ import type { DeepMutable } from "../schema"
16
+ import { SkillV2 } from "../skill"
17
+
18
+ const mutable = <T>(value: T) => value as DeepMutable<T>
19
+
20
+ export const make = Effect.fn("PluginHost.make")(function* (plugin: PluginV2.Interface) {
21
+ const agents = yield* AgentV2.Service
22
+ const aisdk = yield* AISDK.Service
23
+ const catalog = yield* Catalog.Service
24
+ const commands = yield* CommandV2.Service
25
+ const integration = yield* Integration.Service
26
+ const reference = yield* Reference.Service
27
+ const skill = yield* SkillV2.Service
28
+
29
+ return {
30
+ options: {},
31
+ agent: {
32
+ reload: agents.reload,
33
+ transform: (callback) =>
34
+ agents.transform((draft) =>
35
+ callback({
36
+ list: () => mutable(draft.list()),
37
+ get: (id) => mutable(draft.get(AgentV2.ID.make(id))),
38
+ default: (id) => draft.default(id === undefined ? undefined : AgentV2.ID.make(id)),
39
+ update: (id, update) => draft.update(AgentV2.ID.make(id), update),
40
+ remove: (id) => draft.remove(AgentV2.ID.make(id)),
41
+ }),
42
+ ),
43
+ },
44
+ aisdk: {
45
+ sdk: (callback) =>
46
+ aisdk.hook.sdk((event) => {
47
+ const output = {
48
+ model: mutable(event.model),
49
+ package: event.package,
50
+ options: event.options,
51
+ sdk: event.sdk,
52
+ }
53
+ const result = callback(output)
54
+ return Effect.suspend(() => (Effect.isEffect(result) ? result : Effect.void)).pipe(
55
+ Effect.tap(() => Effect.sync(() => (event.sdk = output.sdk))),
56
+ )
57
+ }),
58
+ language: (callback) =>
59
+ aisdk.hook.language((event) => {
60
+ const output = {
61
+ model: mutable(event.model),
62
+ sdk: event.sdk,
63
+ options: event.options,
64
+ language: event.language,
65
+ }
66
+ const result = callback(output)
67
+ return Effect.suspend(() => (Effect.isEffect(result) ? result : Effect.void)).pipe(
68
+ Effect.tap(() => Effect.sync(() => (event.language = output.language))),
69
+ )
70
+ }),
71
+ },
72
+ catalog: {
73
+ reload: catalog.reload,
74
+ transform: (callback) =>
75
+ catalog.transform((draft) =>
76
+ callback({
77
+ provider: {
78
+ list: () => mutable(draft.provider.list()),
79
+ get: (id) => mutable(draft.provider.get(ProviderV2.ID.make(id))),
80
+ update: (id, update) => draft.provider.update(ProviderV2.ID.make(id), update),
81
+ remove: (id) => draft.provider.remove(ProviderV2.ID.make(id)),
82
+ },
83
+ model: {
84
+ get: (providerID, modelID) =>
85
+ mutable(draft.model.get(ProviderV2.ID.make(providerID), ModelV2.ID.make(modelID))),
86
+ update: (providerID, modelID, update) =>
87
+ draft.model.update(ProviderV2.ID.make(providerID), ModelV2.ID.make(modelID), update),
88
+ remove: (providerID, modelID) =>
89
+ draft.model.remove(ProviderV2.ID.make(providerID), ModelV2.ID.make(modelID)),
90
+ default: {
91
+ get: draft.model.default.get,
92
+ set: (providerID, modelID) =>
93
+ draft.model.default.set(ProviderV2.ID.make(providerID), ModelV2.ID.make(modelID)),
94
+ },
95
+ },
96
+ }),
97
+ ),
98
+ },
99
+ command: {
100
+ reload: commands.reload,
101
+ transform: commands.transform,
102
+ },
103
+ integration: {
104
+ reload: integration.reload,
105
+ connection: {
106
+ active: (id) => integration.connection.active(Integration.ID.make(id)),
107
+ resolve: (connection) =>
108
+ integration.connection.resolve(
109
+ connection.type === "credential" ? { ...connection, id: Credential.ID.make(connection.id) } : connection,
110
+ ),
111
+ },
112
+ transform: (callback) =>
113
+ integration.transform((draft) =>
114
+ callback({
115
+ list: () => mutable(draft.list()),
116
+ get: (id) => mutable(draft.get(Integration.ID.make(id))),
117
+ update: (id, update) => draft.update(Integration.ID.make(id), update),
118
+ remove: (id) => draft.remove(Integration.ID.make(id)),
119
+ method: {
120
+ list: (id) => mutable(draft.method.list(Integration.ID.make(id))),
121
+ update: (input) => {
122
+ if ("authorize" in input) {
123
+ const methodID = Integration.MethodID.make(input.method.id)
124
+ const refresh = input.refresh
125
+ draft.method.update({
126
+ integrationID: Integration.ID.make(input.integrationID),
127
+ method: { ...input.method, id: methodID },
128
+ authorize: (inputs) =>
129
+ input.authorize(inputs).pipe(
130
+ Effect.map((authorization) => {
131
+ if (authorization.mode === "auto") {
132
+ return {
133
+ ...authorization,
134
+ callback: authorization.callback.pipe(
135
+ Effect.map((credential) =>
136
+ Credential.OAuth.make({
137
+ ...credential,
138
+ methodID: Integration.MethodID.make(credential.methodID),
139
+ }),
140
+ ),
141
+ ),
142
+ }
143
+ }
144
+ return {
145
+ ...authorization,
146
+ callback: (code: string) =>
147
+ authorization.callback(code).pipe(
148
+ Effect.map((credential) =>
149
+ Credential.OAuth.make({
150
+ ...credential,
151
+ methodID: Integration.MethodID.make(credential.methodID),
152
+ }),
153
+ ),
154
+ ),
155
+ }
156
+ }),
157
+ ),
158
+ ...(refresh
159
+ ? {
160
+ refresh: (value: Credential.OAuth) =>
161
+ refresh(value).pipe(
162
+ Effect.map((next) =>
163
+ Credential.OAuth.make({
164
+ ...next,
165
+ methodID: Integration.MethodID.make(next.methodID),
166
+ }),
167
+ ),
168
+ ),
169
+ }
170
+ : {}),
171
+ ...(input.label ? { label: input.label } : {}),
172
+ })
173
+ return
174
+ }
175
+ if (input.method.type === "env") {
176
+ draft.method.update({
177
+ integrationID: Integration.ID.make(input.integrationID),
178
+ method: { type: "env", names: input.method.names },
179
+ })
180
+ return
181
+ }
182
+ draft.method.update({
183
+ integrationID: Integration.ID.make(input.integrationID),
184
+ method: { type: "key", label: input.method.label },
185
+ })
186
+ },
187
+ remove: (id, method) =>
188
+ draft.method.remove(Integration.ID.make(id), Schema.decodeUnknownSync(Integration.Method)(method)),
189
+ },
190
+ }),
191
+ ),
192
+ },
193
+ plugin: {
194
+ add: (input) => plugin.add(PluginV2.ID.make(input.id), input.effect),
195
+ remove: (id) => plugin.remove(PluginV2.ID.make(id)),
196
+ },
197
+ reference: {
198
+ reload: reference.reload,
199
+ transform: (callback) =>
200
+ reference.transform((draft) =>
201
+ callback({
202
+ add: (name, source) => draft.add(name, Schema.decodeUnknownSync(Reference.Source)(source)),
203
+ remove: draft.remove,
204
+ list: draft.list,
205
+ }),
206
+ ),
207
+ },
208
+ skill: {
209
+ reload: skill.reload,
210
+ transform: (callback) =>
211
+ skill.transform((draft) =>
212
+ callback({
213
+ source: (source) => draft.source(Schema.decodeUnknownSync(SkillV2.Source)(source)),
214
+ list: draft.list,
215
+ }),
216
+ ),
217
+ },
218
+ } satisfies Interface
219
+ })
@@ -0,0 +1,153 @@
1
+ export * as PluginInternal from "./internal"
2
+
3
+ import { makeLocationNode } from "../effect/app-node"
4
+ import { httpClient } from "../effect/app-node-platform"
5
+ import type { PluginContext } from "@neurocode-ai/plugin/v2/effect"
6
+ import { Effect, Layer, Scope } from "effect"
7
+ import { AgentV2 } from "../agent"
8
+ import { Catalog } from "../catalog"
9
+ import { CommandV2 } from "../command"
10
+ import { Config } from "../config"
11
+ import { ConfigAgentPlugin } from "../config/plugin/agent"
12
+ import { ConfigCommandPlugin } from "../config/plugin/command"
13
+ import { ConfigExternalPlugin } from "../config/plugin/external"
14
+ import { ConfigProviderPlugin } from "../config/plugin/provider"
15
+ import { ConfigReferencePlugin } from "../config/plugin/reference"
16
+ import { ConfigSkillPlugin } from "../config/plugin/skill"
17
+ import { EventV2 } from "../event"
18
+ import { FileSystem } from "../filesystem"
19
+ import { FSUtil } from "../fs-util"
20
+ import { Global } from "../global"
21
+ import { Integration } from "../integration"
22
+ import { Location } from "../location"
23
+ import { ModelsDev } from "../models-dev"
24
+ import { Npm } from "../npm"
25
+ import { PluginV2 } from "../plugin"
26
+ import { Reference } from "../reference"
27
+ import { SkillV2 } from "../skill"
28
+ import { State } from "../state"
29
+ import { FetchHttpClient, HttpClient } from "effect/unstable/http"
30
+ import { AgentPlugin } from "./agent"
31
+ import { CommandPlugin } from "./command"
32
+ import { ModelsDevPlugin } from "./models-dev"
33
+ import { ProviderPlugins } from "./provider"
34
+ import { SkillPlugin } from "./skill"
35
+ import { VariantPlugin } from "./variant"
36
+
37
+ export type Requirements =
38
+ | AgentV2.Service
39
+ | Catalog.Service
40
+ | CommandV2.Service
41
+ | Config.Service
42
+ | EventV2.Service
43
+ | FileSystem.Service
44
+ | FSUtil.Service
45
+ | Global.Service
46
+ | HttpClient.HttpClient
47
+ | Integration.Service
48
+ | Location.Service
49
+ | ModelsDev.Service
50
+ | Npm.Service
51
+ | Reference.Service
52
+ | SkillV2.Service
53
+
54
+ export interface Plugin<R = never> {
55
+ readonly id: string
56
+ readonly effect: (context: PluginContext) => Effect.Effect<void, never, R | Scope.Scope>
57
+ }
58
+
59
+ export function define<R>(plugin: Plugin<R>) {
60
+ return plugin
61
+ }
62
+
63
+ const layer = Layer.effectDiscard(
64
+ Effect.gen(function* () {
65
+ const catalog = yield* Catalog.Service
66
+ const commands = yield* CommandV2.Service
67
+ const plugin = yield* PluginV2.Service
68
+ const integration = yield* Integration.Service
69
+ const agents = yield* AgentV2.Service
70
+ const config = yield* Config.Service
71
+ const location = yield* Location.Service
72
+ const modelsDev = yield* ModelsDev.Service
73
+ const npm = yield* Npm.Service
74
+ const events = yield* EventV2.Service
75
+ const fs = yield* FSUtil.Service
76
+ const filesystem = yield* FileSystem.Service
77
+ const global = yield* Global.Service
78
+ const http = yield* HttpClient.HttpClient
79
+ const skill = yield* SkillV2.Service
80
+ const reference = yield* Reference.Service
81
+ const add = <R>(input: Plugin<R>) => {
82
+ const loaded = {
83
+ id: input.id,
84
+ effect: (context: PluginContext) =>
85
+ input
86
+ .effect(context)
87
+ .pipe(
88
+ Effect.provideService(Catalog.Service, catalog),
89
+ Effect.provideService(CommandV2.Service, commands),
90
+ Effect.provideService(Integration.Service, integration),
91
+ Effect.provideService(AgentV2.Service, agents),
92
+ Effect.provideService(Config.Service, config),
93
+ Effect.provideService(Location.Service, location),
94
+ Effect.provideService(ModelsDev.Service, modelsDev),
95
+ Effect.provideService(Npm.Service, npm),
96
+ Effect.provideService(EventV2.Service, events),
97
+ Effect.provideService(FSUtil.Service, fs),
98
+ Effect.provideService(FileSystem.Service, filesystem),
99
+ Effect.provideService(Global.Service, global),
100
+ Effect.provideService(HttpClient.HttpClient, http),
101
+ Effect.provideService(SkillV2.Service, skill),
102
+ Effect.provideService(Reference.Service, reference),
103
+ ),
104
+ }
105
+ return plugin.add(PluginV2.ID.make(loaded.id), loaded.effect)
106
+ }
107
+
108
+ yield* State.batch(
109
+ Effect.gen(function* () {
110
+ yield* add(ConfigReferencePlugin.Plugin)
111
+ yield* add(AgentPlugin.Plugin)
112
+ yield* add(CommandPlugin.Plugin)
113
+ yield* add(SkillPlugin.Plugin)
114
+ yield* add(ModelsDevPlugin)
115
+ yield* add(ConfigAgentPlugin.Plugin)
116
+ yield* add(ConfigCommandPlugin.Plugin)
117
+ yield* add(ConfigSkillPlugin.Plugin)
118
+ for (const item of ProviderPlugins) yield* add(item)
119
+ yield* add(ConfigExternalPlugin.Plugin)
120
+ yield* add(ConfigProviderPlugin.Plugin)
121
+ yield* add(VariantPlugin.Plugin)
122
+ }),
123
+ ).pipe(Effect.withSpan("PluginInternal.boot"), Effect.forkScoped({ startImmediately: true }))
124
+ }),
125
+ )
126
+
127
+ export const locationLayer = layer.pipe(
128
+ Layer.provideMerge(Config.locationLayer),
129
+ Layer.provideMerge(FetchHttpClient.layer),
130
+ )
131
+
132
+ export const node = makeLocationNode({
133
+ name: "plugin-internal",
134
+ layer,
135
+ deps: [
136
+ Catalog.node,
137
+ CommandV2.node,
138
+ PluginV2.node,
139
+ Integration.node,
140
+ AgentV2.node,
141
+ Config.node,
142
+ Location.node,
143
+ ModelsDev.node,
144
+ Npm.node,
145
+ EventV2.node,
146
+ FSUtil.node,
147
+ FileSystem.node,
148
+ Global.node,
149
+ httpClient,
150
+ SkillV2.node,
151
+ Reference.node,
152
+ ],
153
+ })
@@ -0,0 +1,94 @@
1
+ export * as LayerMapExample from "./layer-map.example"
2
+
3
+ import { Context, Effect, Layer, LayerMap } from "effect"
4
+ import { Npm } from "../npm"
5
+
6
+ /**
7
+ * Tutorial: split global services from context-specific services.
8
+ *
9
+ * Use this pattern when part of the app should be constructed once at the app edge,
10
+ * while another part should be cached per request/project/workspace key.
11
+ *
12
+ * In this example:
13
+ * - Npm.Service is the global service. It is not keyed by request context and should
14
+ * be provided once by the application runtime.
15
+ * - ConfigService is context-specific. It is built from a RequestContext key and is
16
+ * cached by LayerMap for that key.
17
+ * - ConfigServiceMap.layer owns the cache. Provide it once globally, then each
18
+ * request can provide ConfigServiceMap.get(context) to select the right instance.
19
+ *
20
+ * Lifetime model:
21
+ * - ConfigServiceMap.layer has the app/global lifetime and depends on Npm.Service.
22
+ * - ConfigServiceMap.get(context) has the request/context lifetime and provides
23
+ * ConfigService for exactly that context key.
24
+ * - The cached ConfigService entry stays alive while something is using it. Once idle,
25
+ * it remains cached for idleTimeToLive, then its scope is finalized.
26
+ * - invalidate(context) removes the cache entry for future lookups. Active users keep
27
+ * running on the old instance; the next lookup can create a fresh instance.
28
+ *
29
+ * Key model:
30
+ * - Keys can be strings, structs, classes, arrays, etc.
31
+ * - Prefer primitive or immutable keys. Effect uses Hash / Equal semantics for cache
32
+ * lookup, so mutating an object after it has been used as a key is a bug.
33
+ */
34
+
35
+ export type RequestContext = {
36
+ readonly directory: string
37
+ readonly workspace: string
38
+ }
39
+
40
+ export class RequestContextRef extends Context.Service<RequestContextRef, RequestContext>()(
41
+ "@neurocode/example/RequestContextRef",
42
+ ) {}
43
+
44
+ export interface ConfigServiceShape {
45
+ readonly directory: string
46
+ readonly workspace: string
47
+ readonly nextUse: () => Effect.Effect<number>
48
+ readonly which: Npm.Interface["which"]
49
+ }
50
+
51
+ export class ConfigService extends Context.Service<ConfigService, ConfigServiceShape>()(
52
+ "@neurocode/example/ConfigService",
53
+ ) {}
54
+
55
+ const configServiceLayer = Layer.effect(
56
+ ConfigService,
57
+ Effect.gen(function* () {
58
+ const context = yield* RequestContextRef
59
+ const npm = yield* Npm.Service
60
+
61
+ let useCount = 0
62
+
63
+ return ConfigService.of({
64
+ directory: context.directory,
65
+ workspace: context.workspace,
66
+ nextUse: () => Effect.succeed(++useCount),
67
+ which: npm.which,
68
+ })
69
+ }),
70
+ )
71
+
72
+ export class ConfigServiceMap extends LayerMap.Service<ConfigServiceMap>()("@neurocode/example/ConfigServiceMap", {
73
+ lookup: (context: RequestContext) =>
74
+ configServiceLayer.pipe(Layer.provide(Layer.succeed(RequestContextRef, RequestContextRef.of(context)))),
75
+ idleTimeToLive: "5 minutes",
76
+ }) {}
77
+
78
+ export const appLayer = ConfigServiceMap.layer
79
+
80
+ export const readConfig = Effect.fn("LayerMapExample.readConfig")(function* () {
81
+ const config = yield* ConfigService
82
+
83
+ return {
84
+ directory: config.directory,
85
+ workspace: config.workspace,
86
+ useCount: yield* config.nextUse(),
87
+ }
88
+ })
89
+
90
+ export const handleRequest = Effect.fn("LayerMapExample.handleRequest")(function* (context: RequestContext) {
91
+ return yield* readConfig().pipe(Effect.provide(ConfigServiceMap.get(context)))
92
+ })
93
+
94
+ export const invalidateContext = (context: RequestContext) => ConfigServiceMap.invalidate(context)
@@ -0,0 +1,183 @@
1
+ import { define } from "./internal"
2
+ import type { ModelV2Info } from "@neurocode-ai/sdk/v2/types"
3
+ import { Effect, Stream } from "effect"
4
+ import { EventV2 } from "../event"
5
+ import { ModelsDev } from "../models-dev"
6
+ import { ProviderV2 } from "../provider"
7
+
8
+ function released(date: string) {
9
+ const time = Date.parse(date)
10
+ return Number.isFinite(time) ? time : 0
11
+ }
12
+
13
+ function cost(input: ModelsDev.Model["cost"]): ModelV2Info["cost"] {
14
+ const base = {
15
+ input: input?.input ?? 0,
16
+ output: input?.output ?? 0,
17
+ cache: {
18
+ read: input?.cache_read ?? 0,
19
+ write: input?.cache_write ?? 0,
20
+ },
21
+ }
22
+ return [
23
+ base,
24
+ ...(input?.tiers?.map((item) => ({
25
+ tier: item.tier,
26
+ input: item.input,
27
+ output: item.output,
28
+ cache: {
29
+ read: item.cache_read ?? 0,
30
+ write: item.cache_write ?? 0,
31
+ },
32
+ })) ?? []),
33
+ ...(input?.context_over_200k
34
+ ? [
35
+ {
36
+ tier: {
37
+ type: "context" as const,
38
+ size: 200_000,
39
+ },
40
+ input: input.context_over_200k.input,
41
+ output: input.context_over_200k.output,
42
+ cache: {
43
+ read: input.context_over_200k.cache_read ?? 0,
44
+ write: input.context_over_200k.cache_write ?? 0,
45
+ },
46
+ },
47
+ ]
48
+ : []),
49
+ ]
50
+ }
51
+
52
+ function mergeCost(base: ModelV2Info["cost"], override: ModelsDev.Model["cost"] | undefined) {
53
+ if (!override) return base
54
+ const next = cost(override)
55
+ const [baseDefault, ...baseTiers] = base
56
+ const [nextDefault, ...nextTiers] = next
57
+ const tierKey = (item: ModelV2Info["cost"][number]) => `${item.tier?.type ?? "base"}:${item.tier?.size ?? 0}`
58
+ const merge = (left: ModelV2Info["cost"][number], right: ModelV2Info["cost"][number]) => ({
59
+ ...left,
60
+ ...right,
61
+ tier: right.tier ?? left.tier,
62
+ cache: { ...left.cache, ...right.cache },
63
+ })
64
+ const tiers = new Map(baseTiers.map((item) => [tierKey(item), item]))
65
+ for (const item of nextTiers) {
66
+ const current = tiers.get(tierKey(item))
67
+ tiers.set(tierKey(item), current ? merge(current, item) : item)
68
+ }
69
+ return [merge(baseDefault ?? { input: 0, output: 0, cache: { read: 0, write: 0 } }, nextDefault), ...tiers.values()]
70
+ }
71
+
72
+ function modeName(model: ModelsDev.Model, mode: string) {
73
+ return `${model.name} ${mode.charAt(0).toUpperCase()}${mode.slice(1)}`
74
+ }
75
+
76
+ function applyModel(
77
+ draft: ModelV2Info,
78
+ model: ModelsDev.Model,
79
+ input: {
80
+ readonly name?: string
81
+ readonly cost?: ModelV2Info["cost"]
82
+ readonly request?: NonNullable<NonNullable<ModelsDev.Model["experimental"]>["modes"]>[string]["provider"]
83
+ } = {},
84
+ ) {
85
+ draft.name = input.name ?? model.name
86
+ draft.family = model.family
87
+ draft.api = model.provider?.npm
88
+ ? {
89
+ id: model.id,
90
+ type: "aisdk",
91
+ package: model.provider.npm,
92
+ url: model.provider.api,
93
+ }
94
+ : {
95
+ id: model.id,
96
+ type: "native",
97
+ url: model.provider?.api,
98
+ settings: {},
99
+ }
100
+ draft.capabilities = {
101
+ tools: model.tool_call,
102
+ input: [...(model.modalities?.input ?? [])],
103
+ output: [...(model.modalities?.output ?? [])],
104
+ }
105
+ draft.variants = []
106
+ draft.time.released = released(model.release_date)
107
+ draft.cost = input.cost ?? cost(model.cost)
108
+ draft.status = model.status ?? "active"
109
+ draft.enabled = true
110
+ draft.limit = {
111
+ context: model.limit.context,
112
+ input: model.limit.input,
113
+ output: model.limit.output,
114
+ }
115
+ Object.assign(draft.request.headers, input.request?.headers ?? {})
116
+ Object.assign(draft.request.body, input.request?.body ?? {})
117
+ }
118
+
119
+ export const ModelsDevPlugin = define({
120
+ id: "models-dev",
121
+ effect: Effect.fn(function* (ctx) {
122
+ const modelsDev = yield* ModelsDev.Service
123
+ const events = yield* EventV2.Service
124
+ yield* ctx.integration.transform(
125
+ Effect.fn(function* (integrations) {
126
+ const data = yield* modelsDev.get()
127
+ for (const item of Object.values(data)) {
128
+ if (item.env.length === 0) continue
129
+ const integrationID = item.id
130
+ integrations.update(integrationID, (integration) => (integration.name = item.name))
131
+ integrations.method.update({
132
+ integrationID,
133
+ method: { type: "key" },
134
+ })
135
+ integrations.method.update({
136
+ integrationID,
137
+ method: { type: "env", names: [...item.env] },
138
+ })
139
+ }
140
+ }),
141
+ )
142
+ yield* ctx.catalog.transform(
143
+ Effect.fn(function* (catalog) {
144
+ const data = yield* modelsDev.get()
145
+ for (const item of Object.values(data)) {
146
+ const providerID = ProviderV2.ID.make(item.id)
147
+ catalog.provider.update(providerID, (provider) => {
148
+ provider.name = item.name
149
+ provider.api = item.npm
150
+ ? {
151
+ type: "aisdk",
152
+ package: item.npm,
153
+ url: item.api,
154
+ }
155
+ : {
156
+ type: "native",
157
+ url: item.api,
158
+ settings: {},
159
+ }
160
+ })
161
+
162
+ for (const model of Object.values(item.models)) {
163
+ const baseCost = cost(model.cost)
164
+ catalog.model.update(providerID, model.id, (draft) => applyModel(draft, model, { cost: baseCost }))
165
+ for (const [mode, options] of Object.entries(model.experimental?.modes ?? {})) {
166
+ catalog.model.update(providerID, `${model.id}-${mode}`, (draft) =>
167
+ applyModel(draft, model, {
168
+ name: modeName(model, mode),
169
+ cost: mergeCost(baseCost, options.cost),
170
+ request: options.provider,
171
+ }),
172
+ )
173
+ }
174
+ }
175
+ }
176
+ }),
177
+ )
178
+ yield* events.subscribe(ModelsDev.Event.Refreshed).pipe(
179
+ Stream.runForEach(() => ctx.integration.reload().pipe(Effect.andThen(ctx.catalog.reload()))),
180
+ Effect.forkScoped({ startImmediately: true }),
181
+ )
182
+ }),
183
+ })