@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.
- package/dist/code_mode/images.d.ts +3 -0
- package/dist/code_mode/images.js +37 -0
- package/dist/code_mode/images.js.map +1 -0
- package/dist/code_mode/output.d.ts +22 -0
- package/dist/code_mode/output.js +109 -0
- package/dist/code_mode/output.js.map +1 -0
- package/dist/code_mode/runtime.d.ts +2 -1
- package/dist/code_mode/runtime.js +92 -12
- package/dist/code_mode/runtime.js.map +1 -1
- package/dist/core/agent/agent_runtime.js +68 -76
- package/dist/core/agent/agent_runtime.js.map +1 -1
- package/dist/core/auth/auth_manager.js +3 -7
- package/dist/core/auth/auth_manager.js.map +1 -1
- package/dist/core/auth/auth_messages.js +1 -1
- package/dist/core/auth/auth_messages.js.map +1 -1
- package/dist/core/auth/auth_storage.js +71 -7
- package/dist/core/auth/auth_storage.js.map +1 -1
- package/dist/core/auth/cli.js +8 -7
- package/dist/core/auth/cli.js.map +1 -1
- package/dist/core/auth/credential_store.js +88 -186
- package/dist/core/auth/credential_store.js.map +1 -1
- package/dist/core/auth/index.js +1 -1
- package/dist/core/auth/index.js.map +1 -1
- package/dist/core/auth/providers/openai_codex.js +51 -229
- package/dist/core/auth/providers/openai_codex.js.map +1 -1
- package/dist/core/cli.js +1 -2
- package/dist/core/cli.js.map +1 -1
- package/dist/core/client_tools/command_client_tools.js +2 -2
- package/dist/core/commands/registry.js +16 -1
- package/dist/core/commands/registry.js.map +1 -1
- package/dist/core/config/content_loader.js +30 -131
- package/dist/core/config/content_loader.js.map +1 -1
- package/dist/core/config/index.js.map +1 -1
- package/dist/core/config/mcp.js +68 -0
- package/dist/core/config/mcp.js.map +1 -0
- package/dist/core/config/paths.js +0 -1
- package/dist/core/config/paths.js.map +1 -1
- package/dist/core/config/runtime.js +3 -2
- package/dist/core/config/runtime.js.map +1 -1
- package/dist/core/config/runtime_config_snapshot.js +23 -9
- package/dist/core/config/runtime_config_snapshot.js.map +1 -1
- package/dist/core/config/schema.js +39 -73
- package/dist/core/config/schema.js.map +1 -1
- package/dist/core/config/skills_loader.js +6 -2
- package/dist/core/config/skills_loader.js.map +1 -1
- package/dist/core/debug.js +1 -16
- package/dist/core/debug.js.map +1 -1
- package/dist/core/index.js +2 -2
- package/dist/core/index.js.map +1 -1
- package/dist/core/mcp/manager.js +245 -0
- package/dist/core/mcp/manager.js.map +1 -0
- package/dist/core/mcp/search.js +77 -0
- package/dist/core/mcp/search.js.map +1 -0
- package/dist/core/models/catalog.js +2 -239
- package/dist/core/models/catalog.js.map +1 -1
- package/dist/core/personas.js +2 -15
- package/dist/core/personas.js.map +1 -1
- package/dist/core/runtime/agent_model.js +0 -3
- package/dist/core/runtime/agent_model.js.map +1 -1
- package/dist/core/runtime/chat_runtime.js +19 -13
- package/dist/core/runtime/chat_runtime.js.map +1 -1
- package/dist/core/runtime/runtime_bootstrap.js +12 -59
- package/dist/core/runtime/runtime_bootstrap.js.map +1 -1
- package/dist/core/session/auto_compaction_archive.js +1 -1
- package/dist/core/session/auto_compaction_archive.js.map +1 -1
- package/dist/core/session/compaction.js +79 -506
- package/dist/core/session/compaction.js.map +1 -1
- package/dist/core/static/code_mode/mcp/documentation.md +113 -0
- package/dist/core/static/code_mode/sandbox_runner.mjs +58 -3
- package/dist/core/static/tau_docs/client-tools.md +32 -30
- package/dist/core/static/tau_docs/config-reference.md +60 -38
- package/dist/core/static/tau_docs/configuration.md +3 -16
- package/dist/core/static/tau_docs/credentials.md +20 -50
- package/dist/core/static/tau_docs/index.md +2 -1
- package/dist/core/static/tau_docs/manifest.json +1 -0
- package/dist/core/static/tau_docs/models.md +8 -159
- package/dist/core/static/tau_docs/node-sdk.md +4 -2
- package/dist/core/static/tau_docs/openrouter.md +165 -0
- package/dist/core/static/tau_docs/ownership-and-scope.md +5 -3
- package/dist/core/static/tau_docs/personas.md +11 -43
- package/dist/core/static/tau_docs/prompts-and-project-context.md +10 -44
- package/dist/core/static/tau_docs/remote-sessions.md +6 -7
- package/dist/core/static/tau_docs/security.md +17 -6
- package/dist/core/static/tau_docs/session-protocol-methods.md +8 -4
- package/dist/core/static/tau_docs/session-protocol.md +12 -10
- package/dist/core/static/tau_docs/sessions.md +8 -6
- package/dist/core/static/tau_docs/skills.md +8 -32
- package/dist/core/static/tau_docs/subagents.md +4 -3
- package/dist/core/static/tau_docs/telegram.md +2 -2
- package/dist/core/static/tau_docs/tools.md +22 -6
- package/dist/core/static/tau_docs/troubleshooting.md +11 -12
- package/dist/core/static/tau_docs/tui.md +10 -3
- package/dist/core/subagents/agent_supervisor.js +1 -2
- package/dist/core/subagents/agent_supervisor.js.map +1 -1
- package/dist/core/subagents/registry.js +2 -1
- package/dist/core/subagents/registry.js.map +1 -1
- package/dist/core/subagents/types.js +2 -1
- package/dist/core/subagents/types.js.map +1 -1
- package/dist/core/telegram/adapter.js +1 -1
- package/dist/core/telegram/adapter.js.map +1 -1
- package/dist/core/tool/cli.js +10 -2
- package/dist/core/tool/cli.js.map +1 -1
- package/dist/core/tool/openrouter.js +291 -0
- package/dist/core/tool/openrouter.js.map +1 -0
- package/dist/core/tool/openrouter_decisions.js +97 -0
- package/dist/core/tool/openrouter_decisions.js.map +1 -0
- package/dist/core/tool/openrouter_media.js +180 -0
- package/dist/core/tool/openrouter_media.js.map +1 -0
- package/dist/core/tools/catalog.js +14 -2
- package/dist/core/tools/catalog.js.map +1 -1
- package/dist/core/tools/code_mode.js +5 -2
- package/dist/core/tools/code_mode.js.map +1 -1
- package/dist/core/tools/code_mode_worker.js +33 -1
- package/dist/core/tools/code_mode_worker.js.map +1 -1
- package/dist/core/tools/history.js +1 -1
- package/dist/core/tools/history.js.map +1 -1
- package/dist/core/tools/mcp.js +167 -0
- package/dist/core/tools/mcp.js.map +1 -0
- package/dist/core/tools/nook.js +1 -1
- package/dist/core/tools/nook.js.map +1 -1
- package/dist/core/tools/spawn_agent.js +1 -0
- package/dist/core/tools/spawn_agent.js.map +1 -1
- package/dist/core/tools/tau_docs.js +1 -0
- package/dist/core/tools/tau_docs.js.map +1 -1
- package/dist/core/tools/tool_names.js +2 -0
- package/dist/core/tools/tool_names.js.map +1 -1
- package/dist/core/tools/view_image.js +1 -105
- package/dist/core/tools/view_image.js.map +1 -1
- package/dist/core/tools/web.js +1 -1
- package/dist/core/tools/web.js.map +1 -1
- package/dist/core/utils/bm25.js +74 -0
- package/dist/core/utils/bm25.js.map +1 -0
- package/dist/core/utils/model_image.js +106 -0
- package/dist/core/utils/model_image.js.map +1 -0
- package/dist/core/utils/model_stream.js +47 -34
- package/dist/core/utils/model_stream.js.map +1 -1
- package/dist/core/utils/user_metadata.js +2 -27
- package/dist/core/utils/user_metadata.js.map +1 -1
- package/dist/core/version.js +2 -2
- package/dist/core/version.js.map +1 -1
- package/dist/diff_tool/app/dist/assets/{ada-C5qYipkI.js → ada-X3Y2mlq2.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{ahk-CA9gdOdF.js → ahk-DwADkxyd.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{angular-html-CnO8qept.js → angular-html-BbLgbIhq.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{angular-ts-DPXUio9Z.js → angular-ts-DXk17TRM.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{apl-X8YsymsQ.js → apl-BsvksCXR.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{astro-BQ9k8AQ0.js → astro-DH775E4V.js} +1 -1
- package/dist/diff_tool/app/dist/assets/beancount-C2IN4Vk8.js +1 -0
- package/dist/diff_tool/app/dist/assets/{blade-DZ1LYIzB.js → blade-Ck99Rm59.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{c-VLTwNh7z.js → c-DJpHqu5J.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{chapel-5s-sGN8D.js → chapel-Owff0Csy.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{cobol-C9RzwrVu.js → cobol-Cv2gZuhO.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{coffee-BHvjZ343.js → coffee-Bji0YCf_.js} +1 -1
- package/dist/diff_tool/app/dist/assets/coq-CGaSfhwR.js +1 -0
- package/dist/diff_tool/app/dist/assets/{cpp-CBTVXbBp.js → cpp-DTjA_yFF.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{crystal-DQJua4y4.js → crystal-Dd00NOeb.js} +1 -1
- package/dist/diff_tool/app/dist/assets/csharp-dmMxLVLW.js +1 -0
- package/dist/diff_tool/app/dist/assets/css-D44TN1V6.js +1 -0
- package/dist/diff_tool/app/dist/assets/dart-CHZY9Xhz.js +1 -0
- package/dist/diff_tool/app/dist/assets/dotenv-CuH5my32.js +1 -0
- package/dist/diff_tool/app/dist/assets/{edge-BmdmXGH4.js → edge-CVmGSi2m.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{elixir-oiOLs98L.js → elixir-B8Mq7no2.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{elm-BgYuiU4E.js → elm-CXS0I3Z2.js} +1 -1
- package/dist/diff_tool/app/dist/assets/emacs-lisp-DR2vtWaj.js +1 -0
- package/dist/diff_tool/app/dist/assets/{erb-CvFrXpyY.js → erb-B9oDKaUo.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{git-rebase-D3Zo1BLH.js → git-rebase-DT-YFebH.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{glimmer-js-B4-8E3lJ.js → glimmer-js-s0DXlNjq.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{glimmer-ts-CtKkD0t_.js → glimmer-ts-QgEB5S2z.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{glsl-BAqforx_.js → glsl-NA0icwbY.js} +1 -1
- package/dist/diff_tool/app/dist/assets/gn-CAXYAgV-.js +1 -0
- package/dist/diff_tool/app/dist/assets/{graphql-DuE1RMyL.js → graphql-0_dOlfV0.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{hack-CHBcXjz1.js → hack-D-mr7l1l.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{haml-DAs2z3tD.js → haml-B0-oQmz7.js} +1 -1
- package/dist/diff_tool/app/dist/assets/handlebars-Yq9bXpWN.js +1 -0
- package/dist/diff_tool/app/dist/assets/{html-D5co0BLD.js → html-B8bMmdRK.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{html-derivative-BZaWEmAT.js → html-derivative-BpxSlQb_.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{http-Cii5MjJE.js → http-hS_53mxt.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{hurl-Bur-XdJO.js → hurl-B2ST1kwl.js} +1 -1
- package/dist/diff_tool/app/dist/assets/index-CBT_sNIz.js +340 -0
- package/dist/diff_tool/app/dist/assets/{java-DN39wI4w.js → java-D4mj0lZJ.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{javascript-VUReF1H4.js → javascript-wC7rrKiV.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{jinja-BcrfV1JB.js → jinja-DbQ3o87F.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{jison-Bez5Mtmy.js → jison-CrE9FdvP.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{json-ChV8m0Le.js → json-xjicDyWN.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{jsx-D0SXNtM0.js → jsx-m28lXojQ.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{julia-DvhpcBwG.js → julia-19xkOmop.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{just-BGwe8S6x.js → just-Cf7OorTr.js} +1 -1
- package/dist/diff_tool/app/dist/assets/latex-DDb7GQOy.js +1 -0
- package/dist/diff_tool/app/dist/assets/{liquid-UdnqDjKz.js → liquid-DHOuQ0rf.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{lua-CjDHeHEr.js → lua-BGyIvonw.js} +1 -1
- package/dist/diff_tool/app/dist/assets/luau-VGWVLoqX.js +1 -0
- package/dist/diff_tool/app/dist/assets/marko-DB5OqI1C.js +1 -0
- package/dist/diff_tool/app/dist/assets/{mdc-BOLj6Oyq.js → mdc-DTyNMupO.js} +1 -1
- package/dist/diff_tool/app/dist/assets/monokai-BscAjiQZ.js +1 -0
- package/dist/diff_tool/app/dist/assets/moonbit-D9W784E_.js +1 -0
- package/dist/diff_tool/app/dist/assets/nextflow-CSgcYZcx.js +1 -0
- package/dist/diff_tool/app/dist/assets/{nextflow-groovy-DJMQeKeT.js → nextflow-groovy-D7IRt6Pp.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{nginx-5r5xPc2T.js → nginx-DfF0fMYD.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{nim-9exKQYMz.js → nim-BbwsXuZD.js} +1 -1
- package/dist/diff_tool/app/dist/assets/nushell--0fT3PI1.js +1 -0
- package/dist/diff_tool/app/dist/assets/one-dark-pro-C6YyR8OD.js +1 -0
- package/dist/diff_tool/app/dist/assets/{org-B2fpAlbt.js → org-BkOBBl1f.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{perl-DV2fcwyY.js → perl-BTXTOl4K.js} +1 -1
- package/dist/diff_tool/app/dist/assets/php-DIMVQWiA.js +1 -0
- package/dist/diff_tool/app/dist/assets/{powershell-CCSX75yM.js → powershell-QtSnmXcu.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{pug-DuKcQxFj.js → pug-B_nrb4dJ.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{qml-C16b4SYx.js → qml-1PXi8G8Z.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{r-B1ZldnJf.js → r-CJrAkc2Z.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{razor-TYmslaAB.js → razor-DgPUftNd.js} +1 -1
- package/dist/diff_tool/app/dist/assets/red-CYlAq2b0.js +1 -0
- package/dist/diff_tool/app/dist/assets/{regexp-dWCffmDD.js → regexp-C2hS8XvL.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{rst-DEkLr2YR.js → rst-CDouhKnc.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{ruby-BJnJzr4u.js → ruby-CAjKW1uO.js} +1 -1
- package/dist/diff_tool/app/dist/assets/rust-DpiIf5JO.js +1 -0
- package/dist/diff_tool/app/dist/assets/{sas-DeRsDbUE.js → sas-qAHkHLc1.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{scss-SGV_CwKm.js → scss-Bfx2Ydid.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{shellscript-BI7khmmR.js → shellscript-Bu76R8y2.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{shellsession-Nupw_Vvw.js → shellsession-BU8BL0oL.js} +1 -1
- package/dist/diff_tool/app/dist/assets/solarized-dark-b97czJOk.js +1 -0
- package/dist/diff_tool/app/dist/assets/solarized-light-DFPxpIYQ.js +1 -0
- package/dist/diff_tool/app/dist/assets/{soy-B96cBzxv.js → soy-FRl8ozP-.js} +1 -1
- package/dist/diff_tool/app/dist/assets/sql-BacQgsEs.js +1 -0
- package/dist/diff_tool/app/dist/assets/{stata-Dnq48cE6.js → stata-B9kkG-ET.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{surrealql-D6lw4TiD.js → surrealql-qvETr7iY.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{svelte-ChVoq9tB.js → svelte-B0z1MgHc.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{templ-Dh1EXuHD.js → templ-DTnPshKb.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{tex-KTYusNFe.js → tex-Cosab9_H.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{ts-tags-DC2RU1_R.js → ts-tags-DNYvLFUH.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{tsx-DtIM39Jy.js → tsx-Dstq6j1a.js} +1 -1
- package/dist/diff_tool/app/dist/assets/twig-BzqZdwaX.js +1 -0
- package/dist/diff_tool/app/dist/assets/{typescript-BGAE5iuw.js → typescript-YOQ_oE25.js} +1 -1
- package/dist/diff_tool/app/dist/assets/typespec-BGhR2SqG.js +1 -0
- package/dist/diff_tool/app/dist/assets/{typst-nWoGZ22y.js → typst-DTffwfpc.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{vue-iyHf-l4z.js → vue-D4OMj_C5.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{vue-html-prNdIjLu.js → vue-html-kT4WrHGK.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{vue-vine-KSSlnVeg.js → vue-vine-CxeapeMO.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{xml-DdCC83Xd.js → xml-DWVpuurg.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{xsl-DSqZxsez.js → xsl-C0ciup5a.js} +1 -1
- package/dist/diff_tool/app/dist/assets/{yaml-CZcBUWvf.js → yaml-98Fvyfpc.js} +1 -1
- package/dist/diff_tool/app/dist/index.html +1 -1
- package/dist/execution/execution_environment.js.map +1 -1
- package/dist/execution/fly_sprite_execution_environment.js +5 -10
- package/dist/execution/fly_sprite_execution_environment.js.map +1 -1
- package/dist/execution/tool_backend_execution_environment.js +0 -2
- package/dist/execution/tool_backend_execution_environment.js.map +1 -1
- package/dist/host/client_tool_broker.js +1 -1
- package/dist/host/client_tool_broker.js.map +1 -1
- package/dist/host/execution_runtime.js +5 -4
- package/dist/host/execution_runtime.js.map +1 -1
- package/dist/host/hosted_ephemeral_agent_session.js +0 -2
- package/dist/host/hosted_ephemeral_agent_session.js.map +1 -1
- package/dist/host/local_session_host.js +26 -32
- package/dist/host/local_session_host.js.map +1 -1
- package/dist/host/session_host.js.map +1 -1
- package/dist/main.js +15 -29
- package/dist/main.js.map +1 -1
- package/dist/protocol/session_protocol.d.ts +14 -12
- package/dist/protocol/session_protocol.js +30 -7
- package/dist/protocol/session_protocol.js.map +1 -1
- package/dist/sdk/client_tool_command.d.ts +12 -11
- package/dist/sdk/client_tool_command.js +8 -8
- package/dist/sdk/client_tool_command.js.map +1 -1
- package/dist/sdk/code_mode.js +1 -1
- package/dist/sdk/code_mode.js.map +1 -1
- package/dist/sdk/local_client.js +4 -3
- package/dist/sdk/local_client.js.map +1 -1
- package/dist/sdk/session.js +1 -1
- package/dist/sdk/session.js.map +1 -1
- package/dist/sdk/types.d.ts +2 -1
- package/dist/store/session_snapshot_migrations.js +28 -1
- package/dist/store/session_snapshot_migrations.js.map +1 -1
- package/dist/tui/chat_view.js +3 -1
- package/dist/tui/chat_view.js.map +1 -1
- package/dist/tui/session_chat_app.js +2 -4
- package/dist/tui/session_chat_app.js.map +1 -1
- package/dist/tui/session_chat_controller.js +121 -6
- package/dist/tui/session_chat_controller.js.map +1 -1
- package/dist/tui/ui/custom_editor.js +49 -10
- package/dist/tui/ui/custom_editor.js.map +1 -1
- package/dist/tui/ui/footer.js +4 -2
- package/dist/tui/ui/footer.js.map +1 -1
- package/dist/tui/ui/slash_autocomplete.js +5 -1
- package/dist/tui/ui/slash_autocomplete.js.map +1 -1
- package/package.json +10 -9
- package/dist/diff_tool/app/dist/assets/beancount-D-usSTwE.js +0 -1
- package/dist/diff_tool/app/dist/assets/coq-DuBRPxYg.js +0 -1
- package/dist/diff_tool/app/dist/assets/csharp-oqKa8noW.js +0 -1
- package/dist/diff_tool/app/dist/assets/css-BQflXFb6.js +0 -1
- package/dist/diff_tool/app/dist/assets/dart-CnvKMtbv.js +0 -1
- package/dist/diff_tool/app/dist/assets/dotenv-_5a1GRtc.js +0 -1
- package/dist/diff_tool/app/dist/assets/emacs-lisp-c_oH4hRZ.js +0 -1
- package/dist/diff_tool/app/dist/assets/gn-ilITqXS6.js +0 -1
- package/dist/diff_tool/app/dist/assets/handlebars-CfAla6QK.js +0 -1
- package/dist/diff_tool/app/dist/assets/index-CZUi_I-p.js +0 -340
- package/dist/diff_tool/app/dist/assets/latex-pWSag83P.js +0 -1
- package/dist/diff_tool/app/dist/assets/luau-C2pBCSgt.js +0 -1
- package/dist/diff_tool/app/dist/assets/marko-yLX6iW68.js +0 -1
- package/dist/diff_tool/app/dist/assets/monokai-B_bV9PQ9.js +0 -1
- package/dist/diff_tool/app/dist/assets/moonbit-DgjMoLWr.js +0 -1
- package/dist/diff_tool/app/dist/assets/nextflow-BJtWHP5T.js +0 -1
- package/dist/diff_tool/app/dist/assets/nushell-DHYfO66s.js +0 -1
- package/dist/diff_tool/app/dist/assets/one-dark-pro-CLwyXe_n.js +0 -1
- package/dist/diff_tool/app/dist/assets/php-D-AthX_E.js +0 -1
- package/dist/diff_tool/app/dist/assets/red-DX1BQOs8.js +0 -1
- package/dist/diff_tool/app/dist/assets/rust-Cfkwpbl8.js +0 -1
- package/dist/diff_tool/app/dist/assets/solarized-dark-UdsmMDCL.js +0 -1
- package/dist/diff_tool/app/dist/assets/solarized-light-C1IUL_tW.js +0 -1
- package/dist/diff_tool/app/dist/assets/sql-CsIdKfaO.js +0 -1
- package/dist/diff_tool/app/dist/assets/twig-C6D_TA3n.js +0 -1
- 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
|
|
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
|
-
|
|
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() {
|