@markusylisiurunen/tau 0.3.78 → 0.3.80

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 (309) hide show
  1. package/dist/code_mode/images.d.ts +3 -0
  2. package/dist/code_mode/images.js +37 -0
  3. package/dist/code_mode/images.js.map +1 -0
  4. package/dist/code_mode/output.d.ts +22 -0
  5. package/dist/code_mode/output.js +109 -0
  6. package/dist/code_mode/output.js.map +1 -0
  7. package/dist/code_mode/runtime.d.ts +2 -1
  8. package/dist/code_mode/runtime.js +92 -12
  9. package/dist/code_mode/runtime.js.map +1 -1
  10. package/dist/core/agent/agent_runtime.js +68 -76
  11. package/dist/core/agent/agent_runtime.js.map +1 -1
  12. package/dist/core/auth/auth_manager.js +3 -7
  13. package/dist/core/auth/auth_manager.js.map +1 -1
  14. package/dist/core/auth/auth_messages.js +1 -1
  15. package/dist/core/auth/auth_messages.js.map +1 -1
  16. package/dist/core/auth/auth_storage.js +71 -7
  17. package/dist/core/auth/auth_storage.js.map +1 -1
  18. package/dist/core/auth/cli.js +8 -7
  19. package/dist/core/auth/cli.js.map +1 -1
  20. package/dist/core/auth/credential_store.js +88 -186
  21. package/dist/core/auth/credential_store.js.map +1 -1
  22. package/dist/core/auth/index.js +1 -1
  23. package/dist/core/auth/index.js.map +1 -1
  24. package/dist/core/auth/providers/openai_codex.js +51 -229
  25. package/dist/core/auth/providers/openai_codex.js.map +1 -1
  26. package/dist/core/cli.js +1 -2
  27. package/dist/core/cli.js.map +1 -1
  28. package/dist/core/client_tools/command_client_tools.js +2 -2
  29. package/dist/core/commands/registry.js +16 -1
  30. package/dist/core/commands/registry.js.map +1 -1
  31. package/dist/core/config/content_loader.js +30 -131
  32. package/dist/core/config/content_loader.js.map +1 -1
  33. package/dist/core/config/index.js.map +1 -1
  34. package/dist/core/config/mcp.js +68 -0
  35. package/dist/core/config/mcp.js.map +1 -0
  36. package/dist/core/config/paths.js +0 -1
  37. package/dist/core/config/paths.js.map +1 -1
  38. package/dist/core/config/runtime.js +3 -2
  39. package/dist/core/config/runtime.js.map +1 -1
  40. package/dist/core/config/runtime_config_snapshot.js +23 -9
  41. package/dist/core/config/runtime_config_snapshot.js.map +1 -1
  42. package/dist/core/config/schema.js +39 -73
  43. package/dist/core/config/schema.js.map +1 -1
  44. package/dist/core/config/skills_loader.js +6 -2
  45. package/dist/core/config/skills_loader.js.map +1 -1
  46. package/dist/core/debug.js +1 -16
  47. package/dist/core/debug.js.map +1 -1
  48. package/dist/core/index.js +2 -2
  49. package/dist/core/index.js.map +1 -1
  50. package/dist/core/mcp/manager.js +245 -0
  51. package/dist/core/mcp/manager.js.map +1 -0
  52. package/dist/core/mcp/search.js +77 -0
  53. package/dist/core/mcp/search.js.map +1 -0
  54. package/dist/core/models/catalog.js +2 -239
  55. package/dist/core/models/catalog.js.map +1 -1
  56. package/dist/core/personas.js +2 -15
  57. package/dist/core/personas.js.map +1 -1
  58. package/dist/core/runtime/agent_model.js +0 -3
  59. package/dist/core/runtime/agent_model.js.map +1 -1
  60. package/dist/core/runtime/chat_runtime.js +19 -13
  61. package/dist/core/runtime/chat_runtime.js.map +1 -1
  62. package/dist/core/runtime/runtime_bootstrap.js +12 -59
  63. package/dist/core/runtime/runtime_bootstrap.js.map +1 -1
  64. package/dist/core/session/auto_compaction_archive.js +1 -1
  65. package/dist/core/session/auto_compaction_archive.js.map +1 -1
  66. package/dist/core/session/compaction.js +79 -506
  67. package/dist/core/session/compaction.js.map +1 -1
  68. package/dist/core/static/code_mode/mcp/documentation.md +113 -0
  69. package/dist/core/static/code_mode/sandbox_runner.mjs +58 -3
  70. package/dist/core/static/tau_docs/client-tools.md +32 -30
  71. package/dist/core/static/tau_docs/config-reference.md +60 -38
  72. package/dist/core/static/tau_docs/configuration.md +3 -16
  73. package/dist/core/static/tau_docs/credentials.md +20 -50
  74. package/dist/core/static/tau_docs/index.md +2 -1
  75. package/dist/core/static/tau_docs/manifest.json +1 -0
  76. package/dist/core/static/tau_docs/models.md +8 -159
  77. package/dist/core/static/tau_docs/node-sdk.md +4 -2
  78. package/dist/core/static/tau_docs/openrouter.md +165 -0
  79. package/dist/core/static/tau_docs/ownership-and-scope.md +5 -3
  80. package/dist/core/static/tau_docs/personas.md +11 -43
  81. package/dist/core/static/tau_docs/prompts-and-project-context.md +10 -44
  82. package/dist/core/static/tau_docs/remote-sessions.md +6 -7
  83. package/dist/core/static/tau_docs/security.md +17 -6
  84. package/dist/core/static/tau_docs/session-protocol-methods.md +8 -4
  85. package/dist/core/static/tau_docs/session-protocol.md +12 -10
  86. package/dist/core/static/tau_docs/sessions.md +8 -6
  87. package/dist/core/static/tau_docs/skills.md +8 -32
  88. package/dist/core/static/tau_docs/subagents.md +4 -3
  89. package/dist/core/static/tau_docs/telegram.md +2 -2
  90. package/dist/core/static/tau_docs/tools.md +22 -6
  91. package/dist/core/static/tau_docs/troubleshooting.md +11 -12
  92. package/dist/core/static/tau_docs/tui.md +10 -3
  93. package/dist/core/subagents/agent_supervisor.js +1 -2
  94. package/dist/core/subagents/agent_supervisor.js.map +1 -1
  95. package/dist/core/subagents/registry.js +2 -1
  96. package/dist/core/subagents/registry.js.map +1 -1
  97. package/dist/core/subagents/types.js +2 -1
  98. package/dist/core/subagents/types.js.map +1 -1
  99. package/dist/core/telegram/adapter.js +1 -1
  100. package/dist/core/telegram/adapter.js.map +1 -1
  101. package/dist/core/tool/cli.js +10 -2
  102. package/dist/core/tool/cli.js.map +1 -1
  103. package/dist/core/tool/openrouter.js +291 -0
  104. package/dist/core/tool/openrouter.js.map +1 -0
  105. package/dist/core/tool/openrouter_decisions.js +97 -0
  106. package/dist/core/tool/openrouter_decisions.js.map +1 -0
  107. package/dist/core/tool/openrouter_media.js +180 -0
  108. package/dist/core/tool/openrouter_media.js.map +1 -0
  109. package/dist/core/tools/catalog.js +14 -2
  110. package/dist/core/tools/catalog.js.map +1 -1
  111. package/dist/core/tools/code_mode.js +5 -2
  112. package/dist/core/tools/code_mode.js.map +1 -1
  113. package/dist/core/tools/code_mode_worker.js +33 -1
  114. package/dist/core/tools/code_mode_worker.js.map +1 -1
  115. package/dist/core/tools/history.js +1 -1
  116. package/dist/core/tools/history.js.map +1 -1
  117. package/dist/core/tools/mcp.js +167 -0
  118. package/dist/core/tools/mcp.js.map +1 -0
  119. package/dist/core/tools/nook.js +1 -1
  120. package/dist/core/tools/nook.js.map +1 -1
  121. package/dist/core/tools/spawn_agent.js +1 -0
  122. package/dist/core/tools/spawn_agent.js.map +1 -1
  123. package/dist/core/tools/tau_docs.js +1 -0
  124. package/dist/core/tools/tau_docs.js.map +1 -1
  125. package/dist/core/tools/tool_names.js +2 -0
  126. package/dist/core/tools/tool_names.js.map +1 -1
  127. package/dist/core/tools/view_image.js +1 -105
  128. package/dist/core/tools/view_image.js.map +1 -1
  129. package/dist/core/tools/web.js +1 -1
  130. package/dist/core/tools/web.js.map +1 -1
  131. package/dist/core/utils/bm25.js +74 -0
  132. package/dist/core/utils/bm25.js.map +1 -0
  133. package/dist/core/utils/model_image.js +106 -0
  134. package/dist/core/utils/model_image.js.map +1 -0
  135. package/dist/core/utils/model_stream.js +47 -34
  136. package/dist/core/utils/model_stream.js.map +1 -1
  137. package/dist/core/utils/user_metadata.js +2 -27
  138. package/dist/core/utils/user_metadata.js.map +1 -1
  139. package/dist/core/version.js +2 -2
  140. package/dist/core/version.js.map +1 -1
  141. package/dist/diff_tool/app/dist/assets/{ada-C5qYipkI.js → ada-X3Y2mlq2.js} +1 -1
  142. package/dist/diff_tool/app/dist/assets/{ahk-CA9gdOdF.js → ahk-DwADkxyd.js} +1 -1
  143. package/dist/diff_tool/app/dist/assets/{angular-html-CnO8qept.js → angular-html-BbLgbIhq.js} +1 -1
  144. package/dist/diff_tool/app/dist/assets/{angular-ts-DPXUio9Z.js → angular-ts-DXk17TRM.js} +1 -1
  145. package/dist/diff_tool/app/dist/assets/{apl-X8YsymsQ.js → apl-BsvksCXR.js} +1 -1
  146. package/dist/diff_tool/app/dist/assets/{astro-BQ9k8AQ0.js → astro-DH775E4V.js} +1 -1
  147. package/dist/diff_tool/app/dist/assets/beancount-C2IN4Vk8.js +1 -0
  148. package/dist/diff_tool/app/dist/assets/{blade-DZ1LYIzB.js → blade-Ck99Rm59.js} +1 -1
  149. package/dist/diff_tool/app/dist/assets/{c-VLTwNh7z.js → c-DJpHqu5J.js} +1 -1
  150. package/dist/diff_tool/app/dist/assets/{chapel-5s-sGN8D.js → chapel-Owff0Csy.js} +1 -1
  151. package/dist/diff_tool/app/dist/assets/{cobol-C9RzwrVu.js → cobol-Cv2gZuhO.js} +1 -1
  152. package/dist/diff_tool/app/dist/assets/{coffee-BHvjZ343.js → coffee-Bji0YCf_.js} +1 -1
  153. package/dist/diff_tool/app/dist/assets/coq-CGaSfhwR.js +1 -0
  154. package/dist/diff_tool/app/dist/assets/{cpp-CBTVXbBp.js → cpp-DTjA_yFF.js} +1 -1
  155. package/dist/diff_tool/app/dist/assets/{crystal-DQJua4y4.js → crystal-Dd00NOeb.js} +1 -1
  156. package/dist/diff_tool/app/dist/assets/csharp-dmMxLVLW.js +1 -0
  157. package/dist/diff_tool/app/dist/assets/css-D44TN1V6.js +1 -0
  158. package/dist/diff_tool/app/dist/assets/dart-CHZY9Xhz.js +1 -0
  159. package/dist/diff_tool/app/dist/assets/dotenv-CuH5my32.js +1 -0
  160. package/dist/diff_tool/app/dist/assets/{edge-BmdmXGH4.js → edge-CVmGSi2m.js} +1 -1
  161. package/dist/diff_tool/app/dist/assets/{elixir-oiOLs98L.js → elixir-B8Mq7no2.js} +1 -1
  162. package/dist/diff_tool/app/dist/assets/{elm-BgYuiU4E.js → elm-CXS0I3Z2.js} +1 -1
  163. package/dist/diff_tool/app/dist/assets/emacs-lisp-DR2vtWaj.js +1 -0
  164. package/dist/diff_tool/app/dist/assets/{erb-CvFrXpyY.js → erb-B9oDKaUo.js} +1 -1
  165. package/dist/diff_tool/app/dist/assets/{git-rebase-D3Zo1BLH.js → git-rebase-DT-YFebH.js} +1 -1
  166. package/dist/diff_tool/app/dist/assets/{glimmer-js-B4-8E3lJ.js → glimmer-js-s0DXlNjq.js} +1 -1
  167. package/dist/diff_tool/app/dist/assets/{glimmer-ts-CtKkD0t_.js → glimmer-ts-QgEB5S2z.js} +1 -1
  168. package/dist/diff_tool/app/dist/assets/{glsl-BAqforx_.js → glsl-NA0icwbY.js} +1 -1
  169. package/dist/diff_tool/app/dist/assets/gn-CAXYAgV-.js +1 -0
  170. package/dist/diff_tool/app/dist/assets/{graphql-DuE1RMyL.js → graphql-0_dOlfV0.js} +1 -1
  171. package/dist/diff_tool/app/dist/assets/{hack-CHBcXjz1.js → hack-D-mr7l1l.js} +1 -1
  172. package/dist/diff_tool/app/dist/assets/{haml-DAs2z3tD.js → haml-B0-oQmz7.js} +1 -1
  173. package/dist/diff_tool/app/dist/assets/handlebars-Yq9bXpWN.js +1 -0
  174. package/dist/diff_tool/app/dist/assets/{html-D5co0BLD.js → html-B8bMmdRK.js} +1 -1
  175. package/dist/diff_tool/app/dist/assets/{html-derivative-BZaWEmAT.js → html-derivative-BpxSlQb_.js} +1 -1
  176. package/dist/diff_tool/app/dist/assets/{http-Cii5MjJE.js → http-hS_53mxt.js} +1 -1
  177. package/dist/diff_tool/app/dist/assets/{hurl-Bur-XdJO.js → hurl-B2ST1kwl.js} +1 -1
  178. package/dist/diff_tool/app/dist/assets/index-CBT_sNIz.js +340 -0
  179. package/dist/diff_tool/app/dist/assets/{java-DN39wI4w.js → java-D4mj0lZJ.js} +1 -1
  180. package/dist/diff_tool/app/dist/assets/{javascript-VUReF1H4.js → javascript-wC7rrKiV.js} +1 -1
  181. package/dist/diff_tool/app/dist/assets/{jinja-BcrfV1JB.js → jinja-DbQ3o87F.js} +1 -1
  182. package/dist/diff_tool/app/dist/assets/{jison-Bez5Mtmy.js → jison-CrE9FdvP.js} +1 -1
  183. package/dist/diff_tool/app/dist/assets/{json-ChV8m0Le.js → json-xjicDyWN.js} +1 -1
  184. package/dist/diff_tool/app/dist/assets/{jsx-D0SXNtM0.js → jsx-m28lXojQ.js} +1 -1
  185. package/dist/diff_tool/app/dist/assets/{julia-DvhpcBwG.js → julia-19xkOmop.js} +1 -1
  186. package/dist/diff_tool/app/dist/assets/{just-BGwe8S6x.js → just-Cf7OorTr.js} +1 -1
  187. package/dist/diff_tool/app/dist/assets/latex-DDb7GQOy.js +1 -0
  188. package/dist/diff_tool/app/dist/assets/{liquid-UdnqDjKz.js → liquid-DHOuQ0rf.js} +1 -1
  189. package/dist/diff_tool/app/dist/assets/{lua-CjDHeHEr.js → lua-BGyIvonw.js} +1 -1
  190. package/dist/diff_tool/app/dist/assets/luau-VGWVLoqX.js +1 -0
  191. package/dist/diff_tool/app/dist/assets/marko-DB5OqI1C.js +1 -0
  192. package/dist/diff_tool/app/dist/assets/{mdc-BOLj6Oyq.js → mdc-DTyNMupO.js} +1 -1
  193. package/dist/diff_tool/app/dist/assets/monokai-BscAjiQZ.js +1 -0
  194. package/dist/diff_tool/app/dist/assets/moonbit-D9W784E_.js +1 -0
  195. package/dist/diff_tool/app/dist/assets/nextflow-CSgcYZcx.js +1 -0
  196. package/dist/diff_tool/app/dist/assets/{nextflow-groovy-DJMQeKeT.js → nextflow-groovy-D7IRt6Pp.js} +1 -1
  197. package/dist/diff_tool/app/dist/assets/{nginx-5r5xPc2T.js → nginx-DfF0fMYD.js} +1 -1
  198. package/dist/diff_tool/app/dist/assets/{nim-9exKQYMz.js → nim-BbwsXuZD.js} +1 -1
  199. package/dist/diff_tool/app/dist/assets/nushell--0fT3PI1.js +1 -0
  200. package/dist/diff_tool/app/dist/assets/one-dark-pro-C6YyR8OD.js +1 -0
  201. package/dist/diff_tool/app/dist/assets/{org-B2fpAlbt.js → org-BkOBBl1f.js} +1 -1
  202. package/dist/diff_tool/app/dist/assets/{perl-DV2fcwyY.js → perl-BTXTOl4K.js} +1 -1
  203. package/dist/diff_tool/app/dist/assets/php-DIMVQWiA.js +1 -0
  204. package/dist/diff_tool/app/dist/assets/{powershell-CCSX75yM.js → powershell-QtSnmXcu.js} +1 -1
  205. package/dist/diff_tool/app/dist/assets/{pug-DuKcQxFj.js → pug-B_nrb4dJ.js} +1 -1
  206. package/dist/diff_tool/app/dist/assets/{qml-C16b4SYx.js → qml-1PXi8G8Z.js} +1 -1
  207. package/dist/diff_tool/app/dist/assets/{r-B1ZldnJf.js → r-CJrAkc2Z.js} +1 -1
  208. package/dist/diff_tool/app/dist/assets/{razor-TYmslaAB.js → razor-DgPUftNd.js} +1 -1
  209. package/dist/diff_tool/app/dist/assets/red-CYlAq2b0.js +1 -0
  210. package/dist/diff_tool/app/dist/assets/{regexp-dWCffmDD.js → regexp-C2hS8XvL.js} +1 -1
  211. package/dist/diff_tool/app/dist/assets/{rst-DEkLr2YR.js → rst-CDouhKnc.js} +1 -1
  212. package/dist/diff_tool/app/dist/assets/{ruby-BJnJzr4u.js → ruby-CAjKW1uO.js} +1 -1
  213. package/dist/diff_tool/app/dist/assets/rust-DpiIf5JO.js +1 -0
  214. package/dist/diff_tool/app/dist/assets/{sas-DeRsDbUE.js → sas-qAHkHLc1.js} +1 -1
  215. package/dist/diff_tool/app/dist/assets/{scss-SGV_CwKm.js → scss-Bfx2Ydid.js} +1 -1
  216. package/dist/diff_tool/app/dist/assets/{shellscript-BI7khmmR.js → shellscript-Bu76R8y2.js} +1 -1
  217. package/dist/diff_tool/app/dist/assets/{shellsession-Nupw_Vvw.js → shellsession-BU8BL0oL.js} +1 -1
  218. package/dist/diff_tool/app/dist/assets/solarized-dark-b97czJOk.js +1 -0
  219. package/dist/diff_tool/app/dist/assets/solarized-light-DFPxpIYQ.js +1 -0
  220. package/dist/diff_tool/app/dist/assets/{soy-B96cBzxv.js → soy-FRl8ozP-.js} +1 -1
  221. package/dist/diff_tool/app/dist/assets/sql-BacQgsEs.js +1 -0
  222. package/dist/diff_tool/app/dist/assets/{stata-Dnq48cE6.js → stata-B9kkG-ET.js} +1 -1
  223. package/dist/diff_tool/app/dist/assets/{surrealql-D6lw4TiD.js → surrealql-qvETr7iY.js} +1 -1
  224. package/dist/diff_tool/app/dist/assets/{svelte-ChVoq9tB.js → svelte-B0z1MgHc.js} +1 -1
  225. package/dist/diff_tool/app/dist/assets/{templ-Dh1EXuHD.js → templ-DTnPshKb.js} +1 -1
  226. package/dist/diff_tool/app/dist/assets/{tex-KTYusNFe.js → tex-Cosab9_H.js} +1 -1
  227. package/dist/diff_tool/app/dist/assets/{ts-tags-DC2RU1_R.js → ts-tags-DNYvLFUH.js} +1 -1
  228. package/dist/diff_tool/app/dist/assets/{tsx-DtIM39Jy.js → tsx-Dstq6j1a.js} +1 -1
  229. package/dist/diff_tool/app/dist/assets/twig-BzqZdwaX.js +1 -0
  230. package/dist/diff_tool/app/dist/assets/{typescript-BGAE5iuw.js → typescript-YOQ_oE25.js} +1 -1
  231. package/dist/diff_tool/app/dist/assets/typespec-BGhR2SqG.js +1 -0
  232. package/dist/diff_tool/app/dist/assets/{typst-nWoGZ22y.js → typst-DTffwfpc.js} +1 -1
  233. package/dist/diff_tool/app/dist/assets/{vue-iyHf-l4z.js → vue-D4OMj_C5.js} +1 -1
  234. package/dist/diff_tool/app/dist/assets/{vue-html-prNdIjLu.js → vue-html-kT4WrHGK.js} +1 -1
  235. package/dist/diff_tool/app/dist/assets/{vue-vine-KSSlnVeg.js → vue-vine-CxeapeMO.js} +1 -1
  236. package/dist/diff_tool/app/dist/assets/{xml-DdCC83Xd.js → xml-DWVpuurg.js} +1 -1
  237. package/dist/diff_tool/app/dist/assets/{xsl-DSqZxsez.js → xsl-C0ciup5a.js} +1 -1
  238. package/dist/diff_tool/app/dist/assets/{yaml-CZcBUWvf.js → yaml-98Fvyfpc.js} +1 -1
  239. package/dist/diff_tool/app/dist/index.html +1 -1
  240. package/dist/execution/execution_environment.js.map +1 -1
  241. package/dist/execution/fly_sprite_execution_environment.js +5 -10
  242. package/dist/execution/fly_sprite_execution_environment.js.map +1 -1
  243. package/dist/execution/tool_backend_execution_environment.js +0 -2
  244. package/dist/execution/tool_backend_execution_environment.js.map +1 -1
  245. package/dist/host/client_tool_broker.js +1 -1
  246. package/dist/host/client_tool_broker.js.map +1 -1
  247. package/dist/host/execution_runtime.js +5 -4
  248. package/dist/host/execution_runtime.js.map +1 -1
  249. package/dist/host/hosted_ephemeral_agent_session.js +0 -2
  250. package/dist/host/hosted_ephemeral_agent_session.js.map +1 -1
  251. package/dist/host/local_session_host.js +26 -32
  252. package/dist/host/local_session_host.js.map +1 -1
  253. package/dist/host/session_host.js.map +1 -1
  254. package/dist/main.js +15 -29
  255. package/dist/main.js.map +1 -1
  256. package/dist/protocol/session_protocol.d.ts +14 -12
  257. package/dist/protocol/session_protocol.js +30 -7
  258. package/dist/protocol/session_protocol.js.map +1 -1
  259. package/dist/sdk/client_tool_command.d.ts +12 -11
  260. package/dist/sdk/client_tool_command.js +8 -8
  261. package/dist/sdk/client_tool_command.js.map +1 -1
  262. package/dist/sdk/code_mode.js +1 -1
  263. package/dist/sdk/code_mode.js.map +1 -1
  264. package/dist/sdk/local_client.js +4 -3
  265. package/dist/sdk/local_client.js.map +1 -1
  266. package/dist/sdk/session.js +1 -1
  267. package/dist/sdk/session.js.map +1 -1
  268. package/dist/sdk/types.d.ts +2 -1
  269. package/dist/store/session_snapshot_migrations.js +28 -1
  270. package/dist/store/session_snapshot_migrations.js.map +1 -1
  271. package/dist/tui/chat_view.js +3 -1
  272. package/dist/tui/chat_view.js.map +1 -1
  273. package/dist/tui/session_chat_app.js +2 -4
  274. package/dist/tui/session_chat_app.js.map +1 -1
  275. package/dist/tui/session_chat_controller.js +121 -6
  276. package/dist/tui/session_chat_controller.js.map +1 -1
  277. package/dist/tui/ui/custom_editor.js +49 -10
  278. package/dist/tui/ui/custom_editor.js.map +1 -1
  279. package/dist/tui/ui/footer.js +4 -2
  280. package/dist/tui/ui/footer.js.map +1 -1
  281. package/dist/tui/ui/slash_autocomplete.js +5 -1
  282. package/dist/tui/ui/slash_autocomplete.js.map +1 -1
  283. package/package.json +10 -9
  284. package/dist/diff_tool/app/dist/assets/beancount-D-usSTwE.js +0 -1
  285. package/dist/diff_tool/app/dist/assets/coq-DuBRPxYg.js +0 -1
  286. package/dist/diff_tool/app/dist/assets/csharp-oqKa8noW.js +0 -1
  287. package/dist/diff_tool/app/dist/assets/css-BQflXFb6.js +0 -1
  288. package/dist/diff_tool/app/dist/assets/dart-CnvKMtbv.js +0 -1
  289. package/dist/diff_tool/app/dist/assets/dotenv-_5a1GRtc.js +0 -1
  290. package/dist/diff_tool/app/dist/assets/emacs-lisp-c_oH4hRZ.js +0 -1
  291. package/dist/diff_tool/app/dist/assets/gn-ilITqXS6.js +0 -1
  292. package/dist/diff_tool/app/dist/assets/handlebars-CfAla6QK.js +0 -1
  293. package/dist/diff_tool/app/dist/assets/index-CZUi_I-p.js +0 -340
  294. package/dist/diff_tool/app/dist/assets/latex-pWSag83P.js +0 -1
  295. package/dist/diff_tool/app/dist/assets/luau-C2pBCSgt.js +0 -1
  296. package/dist/diff_tool/app/dist/assets/marko-yLX6iW68.js +0 -1
  297. package/dist/diff_tool/app/dist/assets/monokai-B_bV9PQ9.js +0 -1
  298. package/dist/diff_tool/app/dist/assets/moonbit-DgjMoLWr.js +0 -1
  299. package/dist/diff_tool/app/dist/assets/nextflow-BJtWHP5T.js +0 -1
  300. package/dist/diff_tool/app/dist/assets/nushell-DHYfO66s.js +0 -1
  301. package/dist/diff_tool/app/dist/assets/one-dark-pro-CLwyXe_n.js +0 -1
  302. package/dist/diff_tool/app/dist/assets/php-D-AthX_E.js +0 -1
  303. package/dist/diff_tool/app/dist/assets/red-DX1BQOs8.js +0 -1
  304. package/dist/diff_tool/app/dist/assets/rust-Cfkwpbl8.js +0 -1
  305. package/dist/diff_tool/app/dist/assets/solarized-dark-UdsmMDCL.js +0 -1
  306. package/dist/diff_tool/app/dist/assets/solarized-light-C1IUL_tW.js +0 -1
  307. package/dist/diff_tool/app/dist/assets/sql-CsIdKfaO.js +0 -1
  308. package/dist/diff_tool/app/dist/assets/twig-C6D_TA3n.js +0 -1
  309. package/dist/diff_tool/app/dist/assets/typespec-BRdr0IET.js +0 -1
@@ -0,0 +1,245 @@
1
+ import { McpClient, McpSessionExpiredError, StdioTransport, StreamableHttpTransport, } from "@earendil-works/pi-mcp";
2
+ import { Value } from "typebox/value";
3
+ import { APP_VERSION } from "../version.js";
4
+ const DEFAULT_DISCOVERY_TIMEOUT_MS = 30_000;
5
+ const DEFAULT_EXECUTION_TIMEOUT_MS = 300_000;
6
+ const MAX_MESSAGE_BYTES = 16 * 1024 * 1024;
7
+ export class McpManager {
8
+ env;
9
+ servers;
10
+ connections = new Map();
11
+ startups = new Set();
12
+ cleanups = new Set();
13
+ closed = false;
14
+ closePromise;
15
+ constructor(servers = {}, env = process.env) {
16
+ this.env = env;
17
+ this.servers = structuredClone(servers);
18
+ }
19
+ get available() {
20
+ return Object.values(this.servers).some((server) => server.enabled !== false);
21
+ }
22
+ listServers() {
23
+ this.assertOpen();
24
+ return Object.entries(this.servers)
25
+ .filter(([, server]) => server.enabled !== false)
26
+ .map(([name]) => ({ name }));
27
+ }
28
+ async listTools(server, signal) {
29
+ signal.throwIfAborted();
30
+ const connection = this.getConnection(server);
31
+ const client = await waitForConnection(connection.ready, signal);
32
+ signal.throwIfAborted();
33
+ if (!connection.tools) {
34
+ const revision = connection.revision;
35
+ const tools = await this.request(client, () => client.listTools({ signal }));
36
+ if (revision === connection.revision)
37
+ connection.tools = tools;
38
+ return { ...(client.instructions ? { instructions: client.instructions } : {}), tools };
39
+ }
40
+ return {
41
+ ...(client.instructions ? { instructions: client.instructions } : {}),
42
+ tools: connection.tools,
43
+ };
44
+ }
45
+ async describeTool(server, name, signal) {
46
+ const { tools } = await this.listTools(server, signal);
47
+ const tool = tools.find((candidate) => candidate.name === name);
48
+ if (!tool)
49
+ throw new Error(`MCP server '${server}' has no tool '${name}'.`);
50
+ return tool;
51
+ }
52
+ async callTool(server, name, args, signal) {
53
+ const tool = await this.describeTool(server, name, signal);
54
+ if (!Value.Check(tool.inputSchema, args)) {
55
+ throw new Error(`Invalid arguments for MCP tool '${server}/${name}'; read its inputSchema.`);
56
+ }
57
+ signal.throwIfAborted();
58
+ const client = await waitForConnection(this.getConnection(server).ready, signal);
59
+ const result = await this.request(client, () => client.callTool(name, args, {
60
+ signal,
61
+ timeoutMs: this.servers[server].timeoutMs ?? DEFAULT_EXECUTION_TIMEOUT_MS,
62
+ }));
63
+ return {
64
+ content: result.content.map(stripContentMetadata),
65
+ ...(result.structuredContent === undefined
66
+ ? {}
67
+ : { structuredContent: result.structuredContent }),
68
+ isError: result.isError ?? false,
69
+ };
70
+ }
71
+ async listResources(server, signal) {
72
+ signal.throwIfAborted();
73
+ const client = await waitForConnection(this.getConnection(server).ready, signal);
74
+ const resources = await this.request(client, () => client.listResources({ signal }));
75
+ return resources.map(({ _meta, ...resource }) => resource);
76
+ }
77
+ async listResourceTemplates(server, signal) {
78
+ signal.throwIfAborted();
79
+ const client = await waitForConnection(this.getConnection(server).ready, signal);
80
+ const templates = await this.request(client, () => client.listResourceTemplates({ signal }));
81
+ return templates.map(({ _meta, ...template }) => template);
82
+ }
83
+ async readResource(server, uri, signal) {
84
+ signal.throwIfAborted();
85
+ const client = await waitForConnection(this.getConnection(server).ready, signal);
86
+ const result = await this.request(client, () => client.readResource(uri, {
87
+ signal,
88
+ timeoutMs: this.servers[server].timeoutMs ?? DEFAULT_EXECUTION_TIMEOUT_MS,
89
+ }));
90
+ return {
91
+ contents: result.contents.map(({ _meta, ...content }) => content),
92
+ };
93
+ }
94
+ close() {
95
+ this.closed = true;
96
+ this.closePromise ??= this.closeConnections();
97
+ return this.closePromise;
98
+ }
99
+ async closeConnections() {
100
+ const connections = [...this.connections.values()];
101
+ this.connections.clear();
102
+ const results = await Promise.allSettled(connections.map(async ({ client, ready }) => {
103
+ await this.closeClient(client);
104
+ await ready.catch(() => { });
105
+ }));
106
+ await Promise.allSettled(this.startups);
107
+ const cleanups = await Promise.allSettled(this.cleanups);
108
+ if ([...results, ...cleanups].some((result) => result.status === "rejected")) {
109
+ throw new Error("failed to close MCP connections");
110
+ }
111
+ }
112
+ async request(client, run) {
113
+ try {
114
+ return await run();
115
+ }
116
+ catch (error) {
117
+ if (error instanceof McpSessionExpiredError)
118
+ await this.closeClient(client);
119
+ throw error;
120
+ }
121
+ }
122
+ closeClient(client) {
123
+ const cleanup = client.close();
124
+ this.cleanups.add(cleanup);
125
+ void cleanup.then(() => this.cleanups.delete(cleanup), () => this.cleanups.delete(cleanup));
126
+ return cleanup;
127
+ }
128
+ assertOpen() {
129
+ if (this.closed)
130
+ throw new Error("MCP connections are closed.");
131
+ }
132
+ getConnection(name) {
133
+ this.assertOpen();
134
+ const server = Object.hasOwn(this.servers, name) ? this.servers[name] : undefined;
135
+ if (!server || server.enabled === false)
136
+ throw new Error(`MCP server '${name}' is unavailable.`);
137
+ const existing = this.connections.get(name);
138
+ if (existing && existing.client.connectionState !== "closed")
139
+ return existing;
140
+ const client = new McpClient({
141
+ name: "tau",
142
+ version: APP_VERSION,
143
+ requestTimeoutMs: server.discoveryTimeoutMs ?? DEFAULT_DISCOVERY_TIMEOUT_MS,
144
+ });
145
+ const transport = this.createTransport(name, server);
146
+ const connection = {
147
+ client,
148
+ ready: Promise.resolve(client),
149
+ revision: 0,
150
+ };
151
+ this.connections.set(name, connection);
152
+ client.onNotification("notifications/tools/list_changed", () => {
153
+ connection.tools = undefined;
154
+ connection.revision++;
155
+ });
156
+ const timer = setTimeout(() => {
157
+ void this.closeClient(client).catch(() => { });
158
+ }, server.discoveryTimeoutMs ?? DEFAULT_DISCOVERY_TIMEOUT_MS);
159
+ connection.ready = client
160
+ .connect(transport)
161
+ .then(() => {
162
+ this.assertOpen();
163
+ return client;
164
+ }, () => {
165
+ throw new Error(`failed to connect MCP server '${name}'; check its configuration and authentication`);
166
+ })
167
+ .finally(() => clearTimeout(timer));
168
+ this.startups.add(connection.ready);
169
+ // A caller can be cancelled while the shared startup continues to its request deadline.
170
+ void connection.ready.then(() => this.startups.delete(connection.ready), () => this.startups.delete(connection.ready));
171
+ return connection;
172
+ }
173
+ createTransport(name, server) {
174
+ const expand = (values = {}) => Object.fromEntries(Object.entries(values).map(([key, value]) => [
175
+ key,
176
+ value.replace(/\$\{([A-Za-z_][A-Za-z0-9_]*)\}/g, (_match, variable) => {
177
+ const resolved = this.env[variable];
178
+ if (resolved === undefined)
179
+ throw new Error(`MCP server '${name}' requires environment variable '${variable}'.`);
180
+ return resolved;
181
+ }),
182
+ ]));
183
+ if (server.type === "http") {
184
+ return new StreamableHttpTransport({
185
+ url: server.url,
186
+ headers: expand(server.headers),
187
+ fetch: boundedFetch(Math.max(server.timeoutMs ?? DEFAULT_EXECUTION_TIMEOUT_MS, server.discoveryTimeoutMs ?? DEFAULT_DISCOVERY_TIMEOUT_MS)),
188
+ maxMessageBytes: MAX_MESSAGE_BYTES,
189
+ });
190
+ }
191
+ return new StdioTransport({
192
+ command: server.command,
193
+ args: server.args,
194
+ cwd: server.cwd,
195
+ env: {
196
+ ...Object.fromEntries(Object.entries(this.env).filter((entry) => typeof entry[1] === "string")),
197
+ ...expand(server.env),
198
+ },
199
+ inheritEnv: false,
200
+ maxMessageBytes: MAX_MESSAGE_BYTES,
201
+ });
202
+ }
203
+ }
204
+ function stripContentMetadata(block) {
205
+ const { _meta, ...content } = block;
206
+ if (content.type !== "resource")
207
+ return content;
208
+ const { _meta: resourceMeta, ...resource } = content.resource;
209
+ return { ...content, resource };
210
+ }
211
+ function boundedFetch(timeoutMs) {
212
+ return async (input, init) => {
213
+ const signal = init?.method === "GET"
214
+ ? init.signal
215
+ : AbortSignal.any([...(init?.signal ? [init.signal] : []), AbortSignal.timeout(timeoutMs)]);
216
+ const response = await fetch(input, { ...init, signal });
217
+ if (!response.body ||
218
+ response.headers.get("content-type")?.split(";")[0]?.trim() === "text/event-stream") {
219
+ return response;
220
+ }
221
+ let bytes = 0;
222
+ const body = response.body.pipeThrough(new TransformStream({
223
+ transform(chunk, controller) {
224
+ bytes += chunk.byteLength;
225
+ if (bytes > MAX_MESSAGE_BYTES)
226
+ throw new Error("MCP HTTP response exceeded the 16 MiB limit.");
227
+ controller.enqueue(chunk);
228
+ },
229
+ }));
230
+ return new Response(body, {
231
+ status: response.status,
232
+ statusText: response.statusText,
233
+ headers: response.headers,
234
+ });
235
+ };
236
+ }
237
+ function waitForConnection(promise, signal) {
238
+ signal.throwIfAborted();
239
+ return new Promise((resolve, reject) => {
240
+ const onAbort = () => reject(signal.reason);
241
+ signal.addEventListener("abort", onAbort, { once: true });
242
+ promise.then(resolve, reject).finally(() => signal.removeEventListener("abort", onAbort));
243
+ });
244
+ }
245
+ //# sourceMappingURL=manager.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"manager.js","sourceRoot":"","sources":["../../../src/core/mcp/manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,SAAS,EAET,sBAAsB,EACtB,cAAc,EACd,uBAAuB,GAExB,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AAEtC,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,MAAM,4BAA4B,GAAG,MAAM,CAAC;AAC5C,MAAM,4BAA4B,GAAG,OAAO,CAAC;AAC7C,MAAM,iBAAiB,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;AAS3C,MAAM,OAAO,UAAU;IAUF,GAAG;IATL,OAAO,CAAmB;IAC1B,WAAW,GAAG,IAAI,GAAG,EAAsB,CAAC;IAC5C,QAAQ,GAAG,IAAI,GAAG,EAAsB,CAAC;IACzC,QAAQ,GAAG,IAAI,GAAG,EAAiB,CAAC;IAC7C,MAAM,GAAG,KAAK,CAAC;IACf,YAAY,CAAiB;IAErC,YACE,OAAO,GAAqB,EAAE,EACb,GAAG,GAAsB,OAAO,CAAC,GAAG;mBAApC,GAAG;QAEpB,IAAI,CAAC,OAAO,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IAC1C,CAAC;IAED,IAAI,SAAS;QACX,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,KAAK,KAAK,CAAC,CAAC;IAChF,CAAC;IAED,WAAW;QACT,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,OAAO,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;aAChC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,KAAK,KAAK,CAAC;aAChD,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,SAAS,CACb,MAAc,EACd,MAAmB;QAEnB,MAAM,CAAC,cAAc,EAAE,CAAC;QACxB,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,UAAU,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACjE,MAAM,CAAC,cAAc,EAAE,CAAC;QACxB,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACtB,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC;YACrC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;YAC7E,IAAI,QAAQ,KAAK,UAAU,CAAC,QAAQ;gBAAE,UAAU,CAAC,KAAK,GAAG,KAAK,CAAC;YAC/D,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC;QAC1F,CAAC;QACD,OAAO;YACL,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACrE,KAAK,EAAE,UAAU,CAAC,KAAK;SACxB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,IAAY,EAAE,MAAmB;QAClE,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACvD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QAChE,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,eAAe,MAAM,kBAAkB,IAAI,IAAI,CAAC,CAAC;QAC5E,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,IAAY,EAAE,IAA6B,EAAE,MAAmB;QAC7F,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QAC3D,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,WAAsB,EAAE,IAAI,CAAC,EAAE,CAAC;YACpD,MAAM,IAAI,KAAK,CAAC,mCAAmC,MAAM,IAAI,IAAI,0BAA0B,CAAC,CAAC;QAC/F,CAAC;QACD,MAAM,CAAC,cAAc,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACjF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,EAAE,CAC7C,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE;YAC1B,MAAM;YACN,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAE,CAAC,SAAS,IAAI,4BAA4B;SAC3E,CAAC,CACH,CAAC;QACF,OAAO;YACL,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;YACjD,GAAG,CAAC,MAAM,CAAC,iBAAiB,KAAK,SAAS;gBACxC,CAAC,CAAC,EAAE;gBACJ,CAAC,CAAC,EAAE,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,EAAE,CAAC;YACpD,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,KAAK;SACjC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAc,EAAE,MAAmB;QACrD,MAAM,CAAC,cAAc,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACjF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QACrF,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,QAAQ,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC;IAC7D,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,MAAc,EAAE,MAAmB;QAC7D,MAAM,CAAC,cAAc,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACjF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,qBAAqB,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QAC7F,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,QAAQ,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC;IAC7D,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,GAAW,EAAE,MAAmB;QACjE,MAAM,CAAC,cAAc,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACjF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,EAAE,CAC7C,MAAM,CAAC,YAAY,CAAC,GAAG,EAAE;YACvB,MAAM;YACN,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAE,CAAC,SAAS,IAAI,4BAA4B;SAC3E,CAAC,CACH,CAAC;QACF,OAAO;YACL,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,EAAE,EAAE,CAAC,OAAO,CAAC;SAClE,CAAC;IACJ,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,YAAY,KAAK,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC9C,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAEO,KAAK,CAAC,gBAAgB;QAC5B,MAAM,WAAW,GAAG,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;QACnD,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CACtC,WAAW,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE;YAC1C,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAC/B,MAAM,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC9B,CAAC,CAAC,CACH,CAAC;QACF,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzD,IAAI,CAAC,GAAG,OAAO,EAAE,GAAG,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,UAAU,CAAC,EAAE,CAAC;YAC7E,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,OAAO,CAAI,MAAiB,EAAE,GAAqB;QAC/D,IAAI,CAAC;YACH,OAAO,MAAM,GAAG,EAAE,CAAC;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,sBAAsB;gBAAE,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAC5E,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,MAAiB;QACnC,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC;QAC/B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3B,KAAK,OAAO,CAAC,IAAI,CACf,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,EACnC,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CACpC,CAAC;QACF,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,UAAU;QAChB,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;IAClE,CAAC;IAEO,aAAa,CAAC,IAAY;QAChC,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAClF,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,OAAO,KAAK,KAAK;YACrC,MAAM,IAAI,KAAK,CAAC,eAAe,IAAI,mBAAmB,CAAC,CAAC;QAC1D,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,CAAC,eAAe,KAAK,QAAQ;YAAE,OAAO,QAAQ,CAAC;QAC9E,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC;YAC3B,IAAI,EAAE,KAAK;YACX,OAAO,EAAE,WAAW;YACpB,gBAAgB,EAAE,MAAM,CAAC,kBAAkB,IAAI,4BAA4B;SAC5E,CAAC,CAAC;QACH,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACrD,MAAM,UAAU,GAAe;YAC7B,MAAM;YACN,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;YAC9B,QAAQ,EAAE,CAAC;SACZ,CAAC;QACF,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACvC,MAAM,CAAC,cAAc,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC7D,UAAU,CAAC,KAAK,GAAG,SAAS,CAAC;YAC7B,UAAU,CAAC,QAAQ,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC5B,KAAK,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAChD,CAAC,EAAE,MAAM,CAAC,kBAAkB,IAAI,4BAA4B,CAAC,CAAC;QAC9D,UAAU,CAAC,KAAK,GAAG,MAAM;aACtB,OAAO,CAAC,SAAS,CAAC;aAClB,IAAI,CACH,GAAG,EAAE;YACH,IAAI,CAAC,UAAU,EAAE,CAAC;YAClB,OAAO,MAAM,CAAC;QAChB,CAAC,EACD,GAAG,EAAE;YACH,MAAM,IAAI,KAAK,CACb,iCAAiC,IAAI,+CAA+C,CACrF,CAAC;QACJ,CAAC,CACF;aACA,OAAO,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACpC,wFAAwF;QACxF,KAAK,UAAU,CAAC,KAAK,CAAC,IAAI,CACxB,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,EAC5C,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAC7C,CAAC;QACF,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,eAAe,CAAC,IAAY,EAAE,MAAuB;QAC3D,MAAM,MAAM,GAAG,CAAC,MAAM,GAA2B,EAAE,EAAE,EAAE,CACrD,MAAM,CAAC,WAAW,CAChB,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;YAC3C,GAAG;YACH,KAAK,CAAC,OAAO,CAAC,iCAAiC,EAAE,CAAC,MAAM,EAAE,QAAgB,EAAE,EAAE;gBAC5E,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBACpC,IAAI,QAAQ,KAAK,SAAS;oBACxB,MAAM,IAAI,KAAK,CAAC,eAAe,IAAI,oCAAoC,QAAQ,IAAI,CAAC,CAAC;gBACvF,OAAO,QAAQ,CAAC;YAClB,CAAC,CAAC;SACH,CAAC,CACH,CAAC;QACJ,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC3B,OAAO,IAAI,uBAAuB,CAAC;gBACjC,GAAG,EAAE,MAAM,CAAC,GAAG;gBACf,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC;gBAC/B,KAAK,EAAE,YAAY,CACjB,IAAI,CAAC,GAAG,CACN,MAAM,CAAC,SAAS,IAAI,4BAA4B,EAChD,MAAM,CAAC,kBAAkB,IAAI,4BAA4B,CAC1D,CACF;gBACD,eAAe,EAAE,iBAAiB;aACnC,CAAC,CAAC;QACL,CAAC;QACD,OAAO,IAAI,cAAc,CAAC;YACxB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,GAAG,EAAE;gBACH,GAAG,MAAM,CAAC,WAAW,CACnB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,CAC7B,CAAC,KAAK,EAA6B,EAAE,CAAC,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,CACnE,CACF;gBACD,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC;aACtB;YACD,UAAU,EAAE,KAAK;YACjB,eAAe,EAAE,iBAAiB;SACnC,CAAC,CAAC;IACL,CAAC;CACF;AAED,SAAS,oBAAoB,CAAC,KAAmB;IAC/C,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,GAAG,KAAK,CAAC;IACpC,IAAI,OAAO,CAAC,IAAI,KAAK,UAAU;QAAE,OAAO,OAAO,CAAC;IAChD,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,GAAG,QAAQ,EAAE,GAAG,OAAO,CAAC,QAAQ,CAAC;IAC9D,OAAO,EAAE,GAAG,OAAO,EAAE,QAAQ,EAAE,CAAC;AAClC,CAAC;AAED,SAAS,YAAY,CAAC,SAAiB;IACrC,OAAO,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;QAC3B,MAAM,MAAM,GACV,IAAI,EAAE,MAAM,KAAK,KAAK;YACpB,CAAC,CAAC,IAAI,CAAC,MAAM;YACb,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,WAAW,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAChG,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,KAAK,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QACzD,IACE,CAAC,QAAQ,CAAC,IAAI;YACd,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,mBAAmB,EACnF,CAAC;YACD,OAAO,QAAQ,CAAC;QAClB,CAAC;QACD,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,WAAW,CACpC,IAAI,eAAe,CAAyB;YAC1C,SAAS,CAAC,KAAK,EAAE,UAAU;gBACzB,KAAK,IAAI,KAAK,CAAC,UAAU,CAAC;gBAC1B,IAAI,KAAK,GAAG,iBAAiB;oBAC3B,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;gBAClE,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;SACF,CAAC,CACH,CAAC;QACF,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE;YACxB,MAAM,EAAE,QAAQ,CAAC,MAAM;YACvB,UAAU,EAAE,QAAQ,CAAC,UAAU;YAC/B,OAAO,EAAE,QAAQ,CAAC,OAAO;SAC1B,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB,CAAI,OAAmB,EAAE,MAAmB;IACpE,MAAM,CAAC,cAAc,EAAE,CAAC;IACxB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC5C,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;IAC5F,CAAC,CAAC,CAAC;AACL,CAAC"}
@@ -0,0 +1,77 @@
1
+ import { rankBm25 } from "../utils/bm25.js";
2
+ function schemaText(schema, parts) {
3
+ if (typeof schema !== "object" || schema === null || Array.isArray(schema))
4
+ return;
5
+ const value = schema;
6
+ if (typeof value.description === "string")
7
+ parts.push(value.description);
8
+ if (typeof value.properties === "object" && value.properties !== null) {
9
+ for (const [name, property] of Object.entries(value.properties)) {
10
+ parts.push(name);
11
+ schemaText(property, parts);
12
+ }
13
+ }
14
+ schemaText(value.items, parts);
15
+ for (const key of ["anyOf", "oneOf", "allOf"]) {
16
+ const variants = value[key];
17
+ if (Array.isArray(variants))
18
+ for (const variant of variants)
19
+ schemaText(variant, parts);
20
+ }
21
+ }
22
+ export async function searchMcpTools(manager, query, options, signal) {
23
+ signal.throwIfAborted();
24
+ const servers = options.server === undefined
25
+ ? manager.listServers().map((server) => server.name)
26
+ : [options.server];
27
+ const catalogs = new Array(servers.length);
28
+ const failures = new Array(servers.length);
29
+ let next = 0;
30
+ await Promise.all(Array.from({ length: Math.min(4, servers.length) }, async () => {
31
+ while (next < servers.length) {
32
+ signal.throwIfAborted();
33
+ const index = next++;
34
+ const server = servers[index];
35
+ try {
36
+ catalogs[index] = await manager.listTools(server, signal);
37
+ }
38
+ catch {
39
+ signal.throwIfAborted();
40
+ failures[index] = { server, error: "failed to discover tools" };
41
+ }
42
+ }
43
+ }));
44
+ signal.throwIfAborted();
45
+ const tools = [];
46
+ const documents = catalogs.flatMap((catalog, index) => {
47
+ if (!catalog)
48
+ return [];
49
+ const server = servers[index];
50
+ return catalog.tools.map((tool) => {
51
+ const id = String(tools.length);
52
+ tools.push({
53
+ server,
54
+ name: tool.name,
55
+ ...(tool.description ? { description: tool.description } : {}),
56
+ });
57
+ // Repeated name tokens give exact tool-name matches more influence.
58
+ const parts = [
59
+ tool.name,
60
+ tool.name,
61
+ tool.description ?? "",
62
+ server,
63
+ catalog.instructions ?? "",
64
+ ];
65
+ schemaText(tool.inputSchema, parts);
66
+ return { id, text: parts.join(" ") };
67
+ });
68
+ });
69
+ return {
70
+ tools: rankBm25(query, documents, options.limit).map((match) => ({
71
+ ...tools[Number(match.id)],
72
+ score: match.score,
73
+ })),
74
+ errors: failures.filter((failure) => failure !== undefined),
75
+ };
76
+ }
77
+ //# sourceMappingURL=search.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"search.js","sourceRoot":"","sources":["../../../src/core/mcp/search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAG5C,SAAS,UAAU,CAAC,MAAe,EAAE,KAAe;IAClD,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO;IACnF,MAAM,KAAK,GAAG,MAAiC,CAAC;IAChD,IAAI,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ;QAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;IACzE,IAAI,OAAO,KAAK,CAAC,UAAU,KAAK,QAAQ,IAAI,KAAK,CAAC,UAAU,KAAK,IAAI,EAAE,CAAC;QACtE,KAAK,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;YAChE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjB,UAAU,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IACD,UAAU,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC/B,KAAK,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,CAAC;QAC9C,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;QAC5B,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;YAAE,KAAK,MAAM,OAAO,IAAI,QAAQ;gBAAE,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IAC1F,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,OAAmB,EACnB,KAAa,EACb,OAA2C,EAC3C,MAAmB;IAEnB,MAAM,CAAC,cAAc,EAAE,CAAC;IACxB,MAAM,OAAO,GACX,OAAO,CAAC,MAAM,KAAK,SAAS;QAC1B,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC;QACpD,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACvB,MAAM,QAAQ,GAAG,IAAI,KAAK,CACxB,OAAO,CAAC,MAAM,CACf,CAAC;IACF,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAgD,OAAO,CAAC,MAAM,CAAC,CAAC;IAC1F,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,IAAI,EAAE;QAC7D,OAAO,IAAI,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;YAC7B,MAAM,CAAC,cAAc,EAAE,CAAC;YACxB,MAAM,KAAK,GAAG,IAAI,EAAE,CAAC;YACrB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAE,CAAC;YAC/B,IAAI,CAAC;gBACH,QAAQ,CAAC,KAAK,CAAC,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAC5D,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,CAAC,cAAc,EAAE,CAAC;gBACxB,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,0BAA0B,EAAE,CAAC;YAClE,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CACH,CAAC;IACF,MAAM,CAAC,cAAc,EAAE,CAAC;IACxB,MAAM,KAAK,GAA6D,EAAE,CAAC;IAC3E,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;QACpD,IAAI,CAAC,OAAO;YAAE,OAAO,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAE,CAAC;QAC/B,OAAO,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YAChC,MAAM,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAChC,KAAK,CAAC,IAAI,CAAC;gBACT,MAAM;gBACN,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC/D,CAAC,CAAC;YACH,oEAAoE;YACpE,MAAM,KAAK,GAAG;gBACZ,IAAI,CAAC,IAAI;gBACT,IAAI,CAAC,IAAI;gBACT,IAAI,CAAC,WAAW,IAAI,EAAE;gBACtB,MAAM;gBACN,OAAO,CAAC,YAAY,IAAI,EAAE;aAC3B,CAAC;YACF,UAAU,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;YACpC,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IACH,OAAO;QACL,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YAC/D,GAAG,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAE;YAC3B,KAAK,EAAE,KAAK,CAAC,KAAK;SACnB,CAAC,CAAC;QACH,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,KAAK,SAAS,CAAC;KAC5D,CAAC;AACJ,CAAC"}
@@ -1,52 +1,4 @@
1
1
  import { getBuiltinModels, getBuiltinProviders } from "@earendil-works/pi-ai/providers/all";
2
- import { z } from "zod";
3
- const CostTierSchema = z
4
- .object({
5
- inputTokensAbove: z.number().int().nonnegative(),
6
- input: z.number().nonnegative(),
7
- output: z.number().nonnegative(),
8
- cacheRead: z.number().nonnegative(),
9
- cacheWrite: z.number().nonnegative(),
10
- })
11
- .strip();
12
- const CostSchema = z
13
- .object({
14
- input: z.number().nonnegative().optional(),
15
- output: z.number().nonnegative().optional(),
16
- cacheRead: z.number().nonnegative().optional(),
17
- cacheWrite: z.number().nonnegative().optional(),
18
- tiers: z.array(CostTierSchema).optional(),
19
- })
20
- .optional();
21
- const ModelSchema = z
22
- .object({
23
- id: z.string().trim().min(1),
24
- name: z.string().trim().min(1).optional(),
25
- api: z.string().trim().min(1).optional(),
26
- baseUrl: z.string().trim().min(1).optional(),
27
- headers: z.record(z.string(), z.string()).optional(),
28
- reasoning: z.boolean().optional(),
29
- input: z.array(z.enum(["text", "image"])).optional(),
30
- cost: CostSchema,
31
- contextWindow: z.number().int().positive().optional(),
32
- maxTokens: z.number().int().positive().optional(),
33
- compat: z.unknown().optional(),
34
- })
35
- .strip();
36
- const ProviderSchema = z
37
- .object({
38
- api: z.string().trim().min(1).optional(),
39
- baseUrl: z.string().trim().min(1).optional(),
40
- headers: z.record(z.string(), z.string()).optional(),
41
- compat: z.unknown().optional(),
42
- models: z.array(ModelSchema).optional(),
43
- })
44
- .strip();
45
- const ModelsFileSchema = z
46
- .object({
47
- providers: z.record(z.string(), ProviderSchema),
48
- })
49
- .strip();
50
2
  let catalogState;
51
3
  function ensureProviderModels(providers, provider) {
52
4
  const existing = providers.get(provider);
@@ -121,104 +73,6 @@ function cloneModel(model) {
121
73
  : {}),
122
74
  };
123
75
  }
124
- function isObject(value) {
125
- return typeof value === "object" && value !== null && !Array.isArray(value);
126
- }
127
- function mergeCompat(base, patch) {
128
- if (patch === undefined) {
129
- return base;
130
- }
131
- if (isObject(base) && isObject(patch)) {
132
- return {
133
- ...base,
134
- ...patch,
135
- };
136
- }
137
- return patch;
138
- }
139
- function applyModelPatch(model, patch) {
140
- let next = cloneModel(model);
141
- if (patch.api !== undefined) {
142
- next = { ...next, api: patch.api };
143
- }
144
- if (patch.baseUrl !== undefined) {
145
- next = { ...next, baseUrl: patch.baseUrl };
146
- }
147
- if (patch.headers !== undefined) {
148
- next = {
149
- ...next,
150
- headers: {
151
- ...(next.headers ?? {}),
152
- ...patch.headers,
153
- },
154
- };
155
- }
156
- if (patch.compat !== undefined) {
157
- next = {
158
- ...next,
159
- compat: mergeCompat(next.compat, patch.compat),
160
- };
161
- }
162
- if (patch.name !== undefined) {
163
- next = { ...next, name: patch.name };
164
- }
165
- if (patch.reasoning !== undefined) {
166
- next = { ...next, reasoning: patch.reasoning };
167
- }
168
- if (patch.input !== undefined) {
169
- next = { ...next, input: [...patch.input] };
170
- }
171
- if (patch.cost !== undefined) {
172
- next = {
173
- ...next,
174
- cost: {
175
- ...next.cost,
176
- ...patch.cost,
177
- },
178
- };
179
- }
180
- if (patch.contextWindow !== undefined) {
181
- next = { ...next, contextWindow: patch.contextWindow };
182
- }
183
- if (patch.maxTokens !== undefined) {
184
- next = { ...next, maxTokens: patch.maxTokens };
185
- }
186
- return next;
187
- }
188
- function mergeProviderPatch(base, overlay) {
189
- const mergedHeaders = {
190
- ...(base?.headers ?? {}),
191
- ...(overlay.headers ?? {}),
192
- };
193
- return {
194
- ...(base ?? {}),
195
- ...overlay,
196
- ...(Object.keys(mergedHeaders).length > 0 ? { headers: mergedHeaders } : {}),
197
- compat: mergeCompat(base?.compat, overlay.compat),
198
- };
199
- }
200
- function createProviderPatch(provider) {
201
- return {
202
- ...(provider.api !== undefined ? { api: provider.api } : {}),
203
- ...(provider.baseUrl !== undefined ? { baseUrl: provider.baseUrl } : {}),
204
- ...(provider.headers !== undefined ? { headers: provider.headers } : {}),
205
- ...(provider.compat !== undefined ? { compat: provider.compat } : {}),
206
- };
207
- }
208
- function createModelPatch(model) {
209
- return {
210
- ...(model.api !== undefined ? { api: model.api } : {}),
211
- ...(model.baseUrl !== undefined ? { baseUrl: model.baseUrl } : {}),
212
- ...(model.headers !== undefined ? { headers: model.headers } : {}),
213
- ...(model.compat !== undefined ? { compat: model.compat } : {}),
214
- ...(model.name !== undefined ? { name: model.name } : {}),
215
- ...(model.reasoning !== undefined ? { reasoning: model.reasoning } : {}),
216
- ...(model.input !== undefined ? { input: model.input } : {}),
217
- ...(model.cost !== undefined ? { cost: model.cost } : {}),
218
- ...(model.contextWindow !== undefined ? { contextWindow: model.contextWindow } : {}),
219
- ...(model.maxTokens !== undefined ? { maxTokens: model.maxTokens } : {}),
220
- };
221
- }
222
76
  function createSyntheticModel(provider, modelId, providerTemplates) {
223
77
  const template = providerTemplates.get(provider);
224
78
  if (!template) {
@@ -230,44 +84,7 @@ function createSyntheticModel(provider, modelId, providerTemplates) {
230
84
  name: modelId,
231
85
  };
232
86
  }
233
- function formatZodError(error) {
234
- return error.issues
235
- .map((issue) => {
236
- const path = issue.path.length > 0 ? `${issue.path.join(".")}: ` : "";
237
- return `${path}${issue.message}`;
238
- })
239
- .join("; ");
240
- }
241
- function loadModelsFile(path, deps) {
242
- let rawText;
243
- try {
244
- rawText = deps.fs.readFile(path);
245
- }
246
- catch (error) {
247
- return {
248
- error: `${path}: failed to read models file: ${error instanceof Error ? error.message : String(error)}`,
249
- };
250
- }
251
- let parsedJson;
252
- try {
253
- parsedJson = JSON.parse(rawText);
254
- }
255
- catch (error) {
256
- return {
257
- error: `${path}: failed to parse json: ${error instanceof Error ? error.message : String(error)}`,
258
- };
259
- }
260
- const parsed = ModelsFileSchema.safeParse(parsedJson);
261
- if (!parsed.success) {
262
- return {
263
- error: `${path}: invalid models schema: ${formatZodError(parsed.error)}`,
264
- };
265
- }
266
- return { data: parsed.data };
267
- }
268
- export function loadModelResolver(options) {
269
- const deps = options.deps;
270
- const levels = options.levels;
87
+ export function loadModelResolver(options = {}) {
271
88
  const state = options.remoteCatalog
272
89
  ? createCatalogState(options.remoteCatalog)
273
90
  : getCatalogState();
@@ -284,55 +101,6 @@ export function loadModelResolver(options) {
284
101
  modelsByKey.set(modelKey(provider, model.id), cloneModel(model));
285
102
  }
286
103
  }
287
- const providerPatches = new Map();
288
- const modelPatches = new Map();
289
- const errors = [];
290
- for (const level of levels) {
291
- if (!deps.fs.exists(level.modelsPath)) {
292
- continue;
293
- }
294
- const loaded = loadModelsFile(level.modelsPath, deps);
295
- if (loaded.error) {
296
- errors.push(loaded.error);
297
- continue;
298
- }
299
- if (!loaded.data) {
300
- continue;
301
- }
302
- for (const [providerNameRaw, providerConfig] of Object.entries(loaded.data.providers)) {
303
- const providerName = providerNameRaw.trim().toLowerCase();
304
- if (!providerName || !knownProviders.has(providerName)) {
305
- errors.push(`${level.modelsPath}: unknown provider '${providerNameRaw}'.`);
306
- continue;
307
- }
308
- const providerPatch = createProviderPatch(providerConfig);
309
- if (Object.keys(providerPatch).length > 0) {
310
- providerPatches.set(providerName, mergeProviderPatch(providerPatches.get(providerName), providerPatch));
311
- }
312
- for (const modelConfig of providerConfig.models ?? []) {
313
- const key = modelKey(providerName, modelConfig.id);
314
- if (!modelsByKey.has(key)) {
315
- const synthetic = createSyntheticModel(providerName, modelConfig.id, providerTemplates);
316
- if (!synthetic) {
317
- errors.push(`${level.modelsPath}: provider '${providerName}' does not have bundled defaults to derive model '${modelConfig.id}'.`);
318
- continue;
319
- }
320
- modelsByKey.set(key, synthetic);
321
- }
322
- const patches = modelPatches.get(key) ?? [];
323
- patches.push(createModelPatch(modelConfig));
324
- modelPatches.set(key, patches);
325
- }
326
- }
327
- }
328
- for (const [key, baseModel] of modelsByKey.entries()) {
329
- const providerPatch = providerPatches.get(baseModel.provider);
330
- let resolved = providerPatch ? applyModelPatch(baseModel, providerPatch) : baseModel;
331
- for (const modelPatch of modelPatches.get(key) ?? []) {
332
- resolved = applyModelPatch(resolved, modelPatch);
333
- }
334
- modelsByKey.set(key, resolved);
335
- }
336
104
  const resolveConfiguredModel = (providerRaw, modelIdRaw) => {
337
105
  const provider = providerRaw.trim().toLowerCase();
338
106
  const modelId = modelIdRaw.trim();
@@ -359,16 +127,11 @@ export function loadModelResolver(options) {
359
127
  if (!synthetic) {
360
128
  return undefined;
361
129
  }
362
- const providerPatch = providerPatches.get(provider);
363
- if (!providerPatch) {
364
- return synthetic;
365
- }
366
- return applyModelPatch(synthetic, providerPatch);
130
+ return synthetic;
367
131
  };
368
132
  return {
369
133
  resolveModel,
370
134
  resolveConfiguredModel,
371
- errors,
372
135
  };
373
136
  }
374
137
  export function listProviders() {