@mcp-ts/sdk 1.4.0 → 1.5.1

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 (80) hide show
  1. package/README.md +20 -27
  2. package/dist/adapters/agui-adapter.d.mts +16 -0
  3. package/dist/adapters/agui-adapter.d.ts +16 -0
  4. package/dist/adapters/agui-adapter.js +185 -0
  5. package/dist/adapters/agui-adapter.js.map +1 -1
  6. package/dist/adapters/agui-adapter.mjs +185 -0
  7. package/dist/adapters/agui-adapter.mjs.map +1 -1
  8. package/dist/adapters/agui-middleware.d.mts +2 -0
  9. package/dist/adapters/agui-middleware.d.ts +2 -0
  10. package/dist/adapters/agui-middleware.js.map +1 -1
  11. package/dist/adapters/agui-middleware.mjs.map +1 -1
  12. package/dist/adapters/ai-adapter.d.mts +21 -0
  13. package/dist/adapters/ai-adapter.d.ts +21 -0
  14. package/dist/adapters/ai-adapter.js +175 -0
  15. package/dist/adapters/ai-adapter.js.map +1 -1
  16. package/dist/adapters/ai-adapter.mjs +175 -0
  17. package/dist/adapters/ai-adapter.mjs.map +1 -1
  18. package/dist/adapters/langchain-adapter.d.mts +16 -0
  19. package/dist/adapters/langchain-adapter.d.ts +16 -0
  20. package/dist/adapters/langchain-adapter.js +179 -0
  21. package/dist/adapters/langchain-adapter.js.map +1 -1
  22. package/dist/adapters/langchain-adapter.mjs +179 -0
  23. package/dist/adapters/langchain-adapter.mjs.map +1 -1
  24. package/dist/client/index.d.mts +2 -2
  25. package/dist/client/index.d.ts +2 -2
  26. package/dist/client/react.d.mts +94 -8
  27. package/dist/client/react.d.ts +94 -8
  28. package/dist/client/react.js +364 -26
  29. package/dist/client/react.js.map +1 -1
  30. package/dist/client/react.mjs +358 -27
  31. package/dist/client/react.mjs.map +1 -1
  32. package/dist/client/vue.d.mts +4 -4
  33. package/dist/client/vue.d.ts +4 -4
  34. package/dist/client/vue.js +11 -2
  35. package/dist/client/vue.js.map +1 -1
  36. package/dist/client/vue.mjs +11 -2
  37. package/dist/client/vue.mjs.map +1 -1
  38. package/dist/{index-CQr9q0bF.d.mts → index-DcYfpY3H.d.mts} +1 -1
  39. package/dist/{index-nE_7Io0I.d.ts → index-GfC_eNEv.d.ts} +1 -1
  40. package/dist/index.d.mts +4 -3
  41. package/dist/index.d.ts +4 -3
  42. package/dist/index.js +938 -12
  43. package/dist/index.js.map +1 -1
  44. package/dist/index.mjs +923 -13
  45. package/dist/index.mjs.map +1 -1
  46. package/dist/server/index.d.mts +2 -2
  47. package/dist/server/index.d.ts +2 -2
  48. package/dist/server/index.js +58 -12
  49. package/dist/server/index.js.map +1 -1
  50. package/dist/server/index.mjs +58 -12
  51. package/dist/server/index.mjs.map +1 -1
  52. package/dist/shared/index.d.mts +86 -4
  53. package/dist/shared/index.d.ts +86 -4
  54. package/dist/shared/index.js +874 -0
  55. package/dist/shared/index.js.map +1 -1
  56. package/dist/shared/index.mjs +865 -1
  57. package/dist/shared/index.mjs.map +1 -1
  58. package/dist/tool-router-Bo8qZbsD.d.ts +325 -0
  59. package/dist/tool-router-XnWVxPzv.d.mts +325 -0
  60. package/dist/{types-CW6lghof.d.mts → types-CfCoIsWI.d.mts} +27 -1
  61. package/dist/{types-CW6lghof.d.ts → types-CfCoIsWI.d.ts} +27 -1
  62. package/package.json +3 -2
  63. package/src/adapters/agui-adapter.ts +79 -0
  64. package/src/adapters/ai-adapter.ts +75 -0
  65. package/src/adapters/langchain-adapter.ts +74 -0
  66. package/src/client/react/index.ts +16 -0
  67. package/src/client/react/oauth-popup.tsx +446 -0
  68. package/src/client/react/use-mcp-apps.tsx +50 -32
  69. package/src/client/react/use-mcp.ts +36 -3
  70. package/src/client/vue/use-mcp.ts +38 -3
  71. package/src/server/handlers/sse-handler.ts +39 -0
  72. package/src/server/index.ts +2 -0
  73. package/src/server/mcp/oauth-client.ts +35 -15
  74. package/src/shared/index.ts +36 -0
  75. package/src/shared/meta-tools.ts +387 -0
  76. package/src/shared/schema-compressor.ts +124 -0
  77. package/src/shared/tool-index.ts +499 -0
  78. package/src/shared/tool-router.ts +469 -0
  79. package/src/shared/types.ts +30 -0
  80. package/supabase/migrations/20260421010000_add_session_cleanup_cron.sql +32 -0
@@ -1 +1 @@
1
- {"version":3,"sources":["../../src/adapters/agui-adapter.ts","../../src/adapters/agui-middleware.ts"],"names":[],"mappings":";;;;;;;;;AAmCA,IAAM,wBAAA,GAA2B;AAAA;AAAA,EAE7B,SAAA;AAAA,EAAW,KAAA;AAAA,EAAO,UAAA;AAAA,EAAY,OAAA;AAAA,EAAS,aAAA;AAAA;AAAA,EAEvC,SAAA;AAAA,EAAW,UAAA;AAAA,EAAY,YAAA;AAAA,EAAc,kBAAA;AAAA;AAAA,EAErC,YAAA;AAAA,EAAc,UAAA;AAAA,EAAY,WAAA;AAAA,EAAa,kBAAA;AAAA,EAAoB;AAC/D,CAAA;AASO,SAAS,YAAY,MAAA,EAA8D;AACtF,EAAA,IAAI,CAAC,MAAA,EAAQ;AACT,IAAA,OAAO,EAAE,IAAA,EAAM,QAAA,EAAU,UAAA,EAAY,EAAC,EAAE;AAAA,EAC5C;AAEA,EAAA,MAAM,OAAA,GAAU,EAAE,GAAG,MAAA,EAAO;AAG5B,EAAA,KAAA,MAAW,QAAQ,wBAAA,EAA0B;AACzC,IAAA,OAAO,QAAQ,IAAI,CAAA;AAAA,EACvB;AAGA,EAAA,IAAI,OAAA,CAAQ,UAAA,IAAc,OAAO,OAAA,CAAQ,eAAe,QAAA,EAAU;AAC9D,IAAA,MAAM,eAAoC,EAAC;AAC3C,IAAA,KAAA,MAAW,CAAC,KAAK,KAAK,CAAA,IAAK,OAAO,OAAA,CAAQ,OAAA,CAAQ,UAAU,CAAA,EAAG;AAC3D,MAAA,IAAI,OAAO,KAAA,KAAU,QAAA,IAAY,KAAA,KAAU,IAAA,EAAM;AAC7C,QAAA,YAAA,CAAa,GAAG,CAAA,GAAI,WAAA,CAAY,KAA4B,CAAA;AAAA,MAChE,CAAA,MAAO;AACH,QAAA,YAAA,CAAa,GAAG,CAAA,GAAI,KAAA;AAAA,MACxB;AAAA,IACJ;AACA,IAAA,OAAA,CAAQ,UAAA,GAAa,YAAA;AAAA,EACzB;AAGA,EAAA,IAAI,OAAA,CAAQ,KAAA,IAAS,OAAO,OAAA,CAAQ,UAAU,QAAA,EAAU;AACpD,IAAA,OAAA,CAAQ,KAAA,GAAQ,WAAA,CAAY,OAAA,CAAQ,KAAK,CAAA;AAAA,EAC7C;AAGA,EAAA,IAAI,OAAA,CAAQ,oBAAA,IAAwB,OAAO,OAAA,CAAQ,yBAAyB,QAAA,EAAU;AAClF,IAAA,OAAA,CAAQ,oBAAA,GAAuB,WAAA,CAAY,OAAA,CAAQ,oBAAoB,CAAA;AAAA,EAC3E;AAEA,EAAA,OAAO,OAAA;AACX;;;ACrCO,IAAM,aAAA,GAAN,cAA4B,UAAA,CAAW;AAAA,EAI1C,YAAY,MAAA,EAA6B;AACrC,IAAA,KAAA,EAAM;AAJV,IAAA,aAAA,CAAA,IAAA,EAAQ,OAAA,CAAA;AACR,IAAA,aAAA,CAAA,IAAA,EAAQ,aAAA,CAAA;AAIJ,IAAA,IAAA,CAAK,QAAQ,MAAA,CAAO,KAAA;AACpB,IAAA,IAAA,CAAK,WAAA,GAAc,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,CAAC,CAAA,MAAiB;AAAA,MAChD,MAAM,CAAA,CAAE,IAAA;AAAA,MACR,aAAa,CAAA,CAAE,WAAA;AAAA,MACf,UAAA,EAAY,WAAA,CAAY,CAAA,CAAE,UAAU;AAAA,KACxC,CAAE,CAAA;AAAA,EACN;AAAA,EAEQ,UAAU,QAAA,EAA2B;AACzC,IAAA,OAAO,KAAK,KAAA,CAAM,IAAA,CAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AAAA,EACnD;AAAA,EAEQ,UAAU,UAAA,EAAyC;AACvD,IAAA,IAAI,CAAC,UAAA,EAAY,IAAA,EAAK,SAAU,EAAC;AAEjC,IAAA,IAAI;AACA,MAAA,OAAO,IAAA,CAAK,MAAM,UAAU,CAAA;AAAA,IAChC,CAAA,CAAA,MAAQ;AAEJ,MAAA,MAAM,OAAA,GAAU,WAAW,IAAA,EAAK;AAChC,MAAA,IAAI,OAAA,CAAQ,QAAA,CAAS,IAAI,CAAA,EAAG;AACxB,QAAA,MAAM,WAAA,GAAc,QAAQ,KAAA,CAAM,CAAA,EAAG,QAAQ,OAAA,CAAQ,IAAI,IAAI,CAAC,CAAA;AAC9D,QAAA,IAAI;AACA,UAAA,OAAO,IAAA,CAAK,MAAM,WAAW,CAAA;AAAA,QACjC,CAAA,CAAA,MAAQ;AACJ,UAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,WAAW,CAAA;AAAA,QACtE;AAAA,MACJ;AACA,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,UAAU,CAAA;AACjE,MAAA,OAAO,EAAC;AAAA,IACZ;AAAA,EACJ;AAAA,EAEA,MAAc,WAAA,CAAY,QAAA,EAAkB,IAAA,EAA4C;AACpF,IAAA,MAAM,OAAO,IAAA,CAAK,KAAA,CAAM,KAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AACrD,IAAA,IAAI,CAAC,MAAM,OAAA,EAAS;AAChB,MAAA,OAAO,CAAA,YAAA,EAAe,IAAA,GAAO,gBAAA,GAAmB,WAAW,KAAK,QAAQ,CAAA,CAAA;AAAA,IAC5E;AAEA,IAAA,IAAI;AACA,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,gCAAA,EAAmC,QAAQ,CAAA,CAAA,EAAI,IAAI,CAAA;AAC/D,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA;AACtC,MAAA,MAAM,YAAY,OAAO,MAAA,KAAW,WAAW,MAAA,GAAS,IAAA,CAAK,UAAU,MAAM,CAAA;AAE7E,MAAA,OAAA,CAAQ,IAAI,CAAA,4BAAA,CAAA,EAAgC,SAAA,CAAU,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC,CAAA;AACnE,MAAA,OAAO,SAAA;AAAA,IACX,SAAS,KAAA,EAAY;AACjB,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,KAAK,CAAA;AAC5D,MAAA,OAAO,CAAA,OAAA,EAAU,KAAA,CAAM,OAAA,IAAW,MAAA,CAAO,KAAK,CAAC,CAAA,CAAA;AAAA,IACnD;AAAA,EACJ;AAAA,EAEQ,WAAW,MAAA,EAAwB;AACvC,IAAA,OAAO,GAAG,MAAM,CAAA,CAAA,EAAI,IAAA,CAAK,GAAA,EAAK,CAAA,CAAA,EAAI,IAAA,CAAK,MAAA,EAAO,CAAE,SAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAA,EAAG,CAAC,CAAC,CAAA,CAAA;AAAA,EAC5E;AAAA,EAEQ,UAAU,KAAA,EAA4B;AAC1C,IAAA,MAAM,QAAA,GAAW,KAAA;AACjB,IAAA,IAAI,CAAC,QAAA,CAAS,QAAA,WAAmB,QAAA,GAAW,IAAA,CAAK,WAAW,YAAY,CAAA;AACxE,IAAA,IAAI,CAAC,QAAA,CAAS,KAAA,WAAgB,KAAA,GAAQ,IAAA,CAAK,WAAW,SAAS,CAAA;AAAA,EACnE;AAAA;AAAA,EAGQ,mBAAA,CAAoB,OAAkB,KAAA,EAAuB;AACjE,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAG/D,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,kBAAA,EAAoB;AAC7C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,EAAE,KAAA,EAAO;AACT,QAAA,KAAA,CAAM,WAAA,GAAA,CAAe,KAAA,CAAM,WAAA,IAAe,EAAA,IAAM,CAAA,CAAE,KAAA;AAAA,MACtD;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,eAAA,EAAiB;AAC1C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,YAAA,EAAc;AAChC,QAAA,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,CAAA,CAAE,YAAY,CAAA;AAC9C,QAAA,IAAI,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAA,EAAG;AAChC,UAAA,eAAA,CAAgB,GAAA,CAAI,EAAE,UAAU,CAAA;AAAA,QACpC;AACA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iCAAA,EAAoC,CAAA,CAAE,YAAY,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,SAAA,EAAY,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAC,CAAA,CAAA,CAAG,CAAA;AAAA,MACpI;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,cAAA,EAAgB;AACzC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,KAAA,EAAO;AACzB,QAAA,MAAM,QAAA,GAAW,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,EAAA;AACzD,QAAA,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,QAAA,GAAW,EAAE,KAAK,CAAA;AAAA,MAC3D;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,aAAA,EAAe;AACxC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+BAAA,EAAkC,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,SAAS,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,CAAA,CAAG,CAAA;AAAA,IACtH;AAGA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,iBAAA,EAAmB;AAC5C,MAAA,MAAM,QAAA,GAAY,KAAA,CAAc,QAAA,IAAY,EAAC;AAC7C,MAAA,IAAI,QAAA,CAAS,SAAS,CAAA,EAAG;AACrB,QAAA,MAAM,OAAA,GAAU,QAAA,CAAS,QAAA,CAAS,MAAA,GAAS,CAAC,CAAA;AAE5C,QAAA,IAAI,OAAA,CAAQ,IAAA,KAAS,WAAA,IAAe,OAAA,CAAQ,OAAA,EAAS;AACjD,UAAA,KAAA,CAAM,cAAc,OAAA,CAAQ,OAAA;AAAA,QAChC;AAGA,QAAA,KAAA,IAAS,IAAI,QAAA,CAAS,MAAA,GAAS,CAAA,EAAG,CAAA,IAAK,GAAG,CAAA,EAAA,EAAK;AAC3C,UAAA,MAAM,GAAA,GAAM,SAAS,CAAC,CAAA;AACtB,UAAA,MAAM,KAAA,GAAQ,KAAA,CAAM,OAAA,CAAQ,GAAA,CAAI,SAAS,CAAA,GAAI,GAAA,CAAI,SAAA,GAC5C,KAAA,CAAM,QAAQ,GAAA,CAAI,UAAU,CAAA,GAAI,GAAA,CAAI,aAAa,EAAC;AAEvD,UAAA,IAAI,GAAA,CAAI,IAAA,KAAS,WAAA,IAAe,KAAA,CAAM,SAAS,CAAA,EAAG;AAC9C,YAAA,KAAA,MAAW,MAAM,KAAA,EAAO;AACpB,cAAA,IAAI,EAAA,CAAG,EAAA,IAAM,EAAA,CAAG,QAAA,EAAU,IAAA,IAAQ,CAAC,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAE,CAAA,EAAG;AACzD,gBAAA,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,SAAS,IAAI,CAAA;AACzC,gBAAA,kBAAA,CAAmB,IAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,QAAA,CAAS,aAAa,IAAI,CAAA;AAC3D,gBAAA,IAAI,IAAA,CAAK,SAAA,CAAU,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,EAAG;AAClC,kBAAA,eAAA,CAAgB,GAAA,CAAI,GAAG,EAAE,CAAA;AACzB,kBAAA,OAAA,CAAQ,GAAA,CAAI,iDAAiD,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,MAAA,EAAS,EAAA,CAAG,EAAE,CAAA,CAAA,CAAG,CAAA;AAAA,gBAClG;AAAA,cACJ;AAAA,YACJ;AACA,YAAA;AAAA,UACJ;AAAA,QACJ;AAAA,MACJ;AAAA,IACJ;AAAA,EACJ;AAAA;AAAA,EAGA,MAAc,aAAa,KAAA,EAAwC;AAC/D,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAC/D,IAAA,MAAM,UAAwB,EAAC;AAE/B,IAAA,MAAM,WAAW,CAAC,GAAG,eAAe,CAAA,CAAE,GAAA,CAAI,OAAO,UAAA,KAAe;AAC5D,MAAA,MAAM,QAAA,GAAW,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC7C,MAAA,IAAI,CAAC,QAAA,EAAU;AAEf,MAAA,MAAM,OAAO,IAAA,CAAK,SAAA,CAAU,mBAAmB,GAAA,CAAI,UAAU,KAAK,IAAI,CAAA;AACtE,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,wCAAA,EAA2C,QAAQ,CAAA,CAAE,CAAA;AAEjE,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,WAAA,CAAY,UAAU,IAAI,CAAA;AACpD,MAAA,OAAA,CAAQ,IAAA,CAAK;AAAA,QACT,UAAA;AAAA,QACA,QAAA;AAAA,QACA,MAAA;AAAA,QACA,SAAA,EAAW,IAAA,CAAK,UAAA,CAAW,YAAY;AAAA,OAC1C,CAAA;AACD,MAAA,eAAA,CAAgB,OAAO,UAAU,CAAA;AAAA,IACrC,CAAC,CAAA;AAED,IAAA,MAAM,OAAA,CAAQ,IAAI,QAAQ,CAAA;AAC1B,IAAA,OAAO,OAAA;AAAA,EACX;AAAA;AAAA,EAGQ,eAAA,CAAgB,UAAiC,OAAA,EAA6B;AAClF,IAAA,KAAA,MAAW,EAAE,UAAA,EAAY,QAAA,EAAU,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AAC/D,MAAA,QAAA,CAAS,IAAA,CAAK;AAAA,QACV,MAAM,SAAA,CAAU,gBAAA;AAAA,QAChB,UAAA;AAAA,QACA,SAAA;AAAA,QACA,OAAA,EAAS,MAAA;AAAA,QACT,IAAA,EAAM,MAAA;AAAA,QACN,SAAA,EAAW,KAAK,GAAA;AAAI,OAChB,CAAA;AACR,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+CAAA,EAAkD,QAAQ,CAAA,CAAE,CAAA;AAAA,IAC5E;AAAA,EACJ;AAAA,EAEA,GAAA,CAAI,OAAsB,IAAA,EAA4C;AAClE,IAAA,OAAO,IAAI,UAAA,CAAsB,CAAC,QAAA,KAAoC;AAClE,MAAA,MAAM,KAAA,GAAkB;AAAA,QACpB,kBAAA,sBAAwB,GAAA,EAAI;AAAA,QAC5B,aAAA,sBAAmB,GAAA,EAAI;AAAA,QACvB,eAAA,sBAAqB,GAAA,EAAI;AAAA,QACzB,WAAA,EAAa,EAAA;AAAA,QACb,KAAA,EAAO;AAAA,OACX;AAEA,MAAA,IAAA,CAAK,UAAU,KAAK,CAAA;AACpB,MAAA,MAAM,QAAA,GAAW,KAAA;AAEjB,MAAA,OAAA,CAAQ,IAAI,CAAA,+BAAA,CAAiC,CAAA;AAC7C,MAAA,OAAA,CAAQ,IAAI,CAAA,0BAAA,EAA6B,QAAA,CAAS,QAAQ,CAAA,SAAA,EAAY,QAAA,CAAS,KAAK,CAAA,CAAE,CAAA;AACtF,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,0BAAA,EAA6B,KAAA,CAAM,QAAA,EAAU,MAAA,IAAU,CAAC,CAAA,SAAA,EAAY,IAAA,CAAK,KAAA,EAAO,MAAA,IAAU,CAAC,CAAA,CAAE,CAAA;AAGzG,MAAA,IAAI,IAAA,CAAK,aAAa,MAAA,EAAQ;AAC1B,QAAA,KAAA,CAAM,KAAA,GAAQ,CAAC,GAAI,KAAA,CAAM,SAAS,EAAC,EAAI,GAAG,IAAA,CAAK,WAAW,CAAA;AAC1D,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,yBAAA,EAA4B,IAAA,CAAK,WAAA,CAAY,MAAM,CAAA,OAAA,CAAA,EAAW,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,CAAC,CAAA,KAAY,CAAA,CAAE,IAAI,CAAC,CAAA;AAAA,MACvH;AAEA,MAAA,MAAM,oBAAoB,YAAY;AAClC,QAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,QAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,KAAS,CAAA,EAAG;AAClC,UAAA,QAAA,CAAS,IAAA,CAAK;AAAA,YACV,MAAM,SAAA,CAAU,YAAA;AAAA,YAChB,UAAU,QAAA,CAAS,QAAA;AAAA,YACnB,OAAO,QAAA,CAAS,KAAA;AAAA,YAChB,SAAA,EAAW,KAAK,GAAA;AAAI,WAChB,CAAA;AACR,UAAA,QAAA,CAAS,QAAA,EAAS;AAClB,UAAA;AAAA,QACJ;AAEA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kCAAA,EAAqC,KAAA,CAAM,eAAA,CAAgB,IAAI,CAAA,cAAA,CAAgB,CAAA;AAG3F,QAAA,MAAM,YAAY,EAAC;AACnB,QAAA,KAAA,MAAW,UAAA,IAAc,MAAM,eAAA,EAAiB;AAC5C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC/C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,kBAAA,CAAmB,GAAA,CAAI,UAAU,CAAA,IAAK,IAAA;AACzD,UAAA,IAAI,IAAA,EAAM;AACN,YAAA,SAAA,CAAU,IAAA,CAAK;AAAA,cACX,EAAA,EAAI,UAAA;AAAA,cACJ,IAAA,EAAM,UAAA;AAAA,cACN,QAAA,EAAU,EAAE,IAAA,EAAM,SAAA,EAAW,IAAA;AAAK,aACrC,CAAA;AAAA,UACL;AAAA,QACJ;AAGA,QAAA,IAAI,SAAA,CAAU,MAAA,GAAS,CAAA,IAAK,KAAA,CAAM,WAAA,EAAa;AAC3C,UAAA,MAAM,YAAA,GAAe;AAAA,YACjB,EAAA,EAAI,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAAA,YAC7B,IAAA,EAAM,WAAA;AAAA,YACN,OAAA,EAAS,MAAM,WAAA,IAAe,IAAA;AAAA;AAAA,YAC9B,UAAA,EAAY,SAAA,CAAU,MAAA,GAAS,CAAA,GAAI,SAAA,GAAY;AAAA,WACnD;AACA,UAAA,KAAA,CAAM,QAAA,CAAS,KAAK,YAAmB,CAAA;AACvC,UAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iEAAA,EAAoE,KAAA,CAAM,WAAA,EAAa,KAAA,CAAM,CAAA,EAAG,EAAE,CAAC,CAAA,KAAA,EAAQ,SAAA,CAAU,MAAM,CAAA,OAAA,CAAS,CAAA;AAAA,QACpJ;AAGA,QAAA,MAAM,OAAA,GAAU,MAAM,IAAA,CAAK,YAAA,CAAa,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAK,eAAA,CAAgB,UAAU,OAAO,CAAA;AAGtC,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,6CAAA,EAAgD,OAAA,CAAQ,MAAM,CAAA,QAAA,CAAU,CAAA;AAGpF,QAAA,KAAA,MAAW,EAAE,UAAA,EAAY,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AACrD,UAAA,KAAA,CAAM,SAAS,IAAA,CAAK;AAAA,YAChB,EAAA,EAAI,SAAA;AAAA,YACJ,IAAA,EAAM,MAAA;AAAA,YACN,YAAA,EAAc,UAAA;AAAA,YACd,OAAA,EAAS;AAAA,WACL,CAAA;AAAA,QACZ;AAGA,QAAA,KAAA,CAAM,mBAAmB,KAAA,EAAM;AAC/B,QAAA,KAAA,CAAM,cAAc,KAAA,EAAM;AAC1B,QAAA,KAAA,CAAM,WAAA,GAAc,EAAA;AAEpB,QAAA,QAAA,CAAS,KAAA,GAAQ,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAC1C,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,mDAAA,EAAsD,KAAA,CAAM,QAAA,CAAS,MAAM,CAAA,CAAE,CAAA;AAGzF,QAAA,IAAA,CAAK,GAAA,CAAI,KAAK,CAAA,CAAE,SAAA,CAAU;AAAA,UACtB,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,YAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,YAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,SAAA,EAAW;AACpC,cAAA,OAAA,CAAQ,IAAI,CAAA,kDAAA,CAAoD,CAAA;AAChE,cAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,cAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,cAAA,QAAA,CAAS,QAAA,EAAS;AAClB,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,WAAA,EAAa;AACtC,cAAA,OAAA,CAAQ,IAAI,CAAA,uDAAA,CAAyD,CAAA;AACrE,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,cAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,GAAO,CAAA,EAAG;AAChC,gBAAA,iBAAA,EAAkB;AAAA,cACtB,CAAA,MAAO;AACH,gBAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,gBAAA,QAAA,CAAS,QAAA,EAAS;AAAA,cACtB;AACA,cAAA;AAAA,YACJ;AACA,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,UACvB,CAAA;AAAA,UACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,UACtB,CAAA;AAAA,UACA,UAAU,MAAM;AACZ,YAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,UAC5E;AAAA,SACH,CAAA;AAAA,MACL,CAAA;AAEA,MAAA,MAAM,YAAA,GAAe,IAAA,CAAK,GAAA,CAAI,KAAK,EAAE,SAAA,CAAU;AAAA,QAC3C,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,UAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,UAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,SAAA,EAAW;AACpC,YAAA,OAAA,CAAQ,IAAI,CAAA,kCAAA,CAAoC,CAAA;AAChD,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,YAAA,QAAA,CAAS,QAAA,EAAS;AAClB,YAAA;AAAA,UACJ;AAEA,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,YAAA,iBAAA,EAAkB;AAClB,YAAA;AAAA,UACJ;AACA,UAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,QACvB,CAAA;AAAA,QACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,UAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,UAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,QACtB,CAAA;AAAA,QACA,UAAU,MAAM;AACZ,UAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,QAC5E;AAAA,OACH,CAAA;AAED,MAAA,OAAO,MAAM,aAAa,WAAA,EAAY;AAAA,IAC1C,CAAC,CAAA;AAAA,EACL;AACJ;AAKO,SAAS,oBACZ,OAAA,EACF;AACE,EAAA,MAAM,UAAA,GAAa,IAAI,aAAA,CAAc,OAAO,CAAA;AAC5C,EAAA,OAAO,CAAC,OAAsB,IAAA,KAA+C;AACzE,IAAA,OAAO,UAAA,CAAW,GAAA,CAAI,KAAA,EAAO,IAAI,CAAA;AAAA,EACrC,CAAA;AACJ","file":"agui-middleware.mjs","sourcesContent":["/**\n * MCP Adapter for AG-UI Integration\n *\n * This adapter transforms MCP tools into formats compatible with AG-UI agents.\n * It provides tools with handlers for server-side execution and tool definitions\n * in JSON Schema format for passing to remote agents.\n *\n * @example\n * ```typescript\n * import { MultiSessionClient } from '@mcp-ts/sdk/server';\n * import { AguiAdapter } from '@mcp-ts/sdk/adapters/agui-adapter';\n * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';\n * import { HttpAgent } from '@ag-ui/client';\n *\n * // Create MCP client\n * const mcpClient = new MultiSessionClient('user_123');\n * await mcpClient.connect();\n *\n * // Create adapter and get tools\n * const adapter = new AguiAdapter(mcpClient);\n * const tools = await adapter.getTools();\n *\n * // Use with AG-UI middleware\n * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });\n * agent.use(createMcpMiddleware({ tools }));\n * ```\n */\n\nimport { MCPClient } from '../server/mcp/oauth-client.js';\nimport { MultiSessionClient } from '../server/mcp/multi-session-client.js';\n\n/**\n * Extended JSON Schema properties that Pydantic's strict validation rejects.\n * These are valid JSON Schema extensions but not part of the core spec.\n */\nconst PYDANTIC_FORBIDDEN_PROPS = [\n // JSON Schema meta-properties\n '$schema', '$id', '$comment', '$defs', 'definitions',\n // Extended properties used by some MCP servers (e.g., Apify)\n 'prefill', 'examples', 'enumTitles', 'enumDescriptions',\n // Other common extensions\n 'deprecated', 'readOnly', 'writeOnly', 'contentMediaType', 'contentEncoding',\n];\n\n/**\n * Cleans a JSON Schema by removing meta-properties that cause issues with\n * strict Pydantic validation (e.g., Google ADK, LangGraph).\n *\n * @param schema - The JSON Schema to clean\n * @returns Cleaned schema without forbidden properties\n */\nexport function cleanSchema(schema: Record<string, any> | undefined): Record<string, any> {\n if (!schema) {\n return { type: 'object', properties: {} };\n }\n\n const cleaned = { ...schema };\n\n // Remove all forbidden properties\n for (const prop of PYDANTIC_FORBIDDEN_PROPS) {\n delete cleaned[prop];\n }\n\n // Recursively clean nested properties\n if (cleaned.properties && typeof cleaned.properties === 'object') {\n const cleanedProps: Record<string, any> = {};\n for (const [key, value] of Object.entries(cleaned.properties)) {\n if (typeof value === 'object' && value !== null) {\n cleanedProps[key] = cleanSchema(value as Record<string, any>);\n } else {\n cleanedProps[key] = value;\n }\n }\n cleaned.properties = cleanedProps;\n }\n\n // Clean items if it's an array schema\n if (cleaned.items && typeof cleaned.items === 'object') {\n cleaned.items = cleanSchema(cleaned.items);\n }\n\n // Clean additionalProperties if it's an object schema\n if (cleaned.additionalProperties && typeof cleaned.additionalProperties === 'object') {\n cleaned.additionalProperties = cleanSchema(cleaned.additionalProperties);\n }\n\n return cleaned;\n}\n\n/**\n * Configuration options for AguiAdapter\n */\nexport interface AguiAdapterOptions {\n /**\n * Prefix for tool names to avoid collision with other tools.\n * @default serverId or 'mcp'\n */\n prefix?: string;\n}\n\n/**\n * AG-UI Tool with handler for server-side execution.\n */\nexport interface AguiTool {\n name: string;\n description: string;\n parameters?: Record<string, any>;\n _meta?: Record<string, any>; // Add _meta to AguiTool\n handler?: (args: any) => any | Promise<any>;\n}\n\n/**\n * Tool definition format for passing to remote agents (without handler).\n */\nexport interface AguiToolDefinition {\n name: string;\n description: string;\n parameters: Record<string, any>;\n _meta?: Record<string, any>; // Add _meta to AguiToolDefinition\n}\n\n/**\n * Adapter that transforms MCP tools into AG-UI compatible formats.\n */\nexport class AguiAdapter {\n constructor(\n private client: MCPClient | MultiSessionClient,\n private options: AguiAdapterOptions = {}\n ) { }\n\n /**\n * Get tools with handlers for MCP tool execution.\n */\n async getTools(): Promise<AguiTool[]> {\n if (this.isMultiSession()) {\n const clients = (this.client as MultiSessionClient).getClients();\n const allTools: AguiTool[] = [];\n for (const client of clients) {\n allTools.push(...await this.transformTools(client));\n }\n return allTools;\n }\n return this.transformTools(this.client as MCPClient);\n }\n\n /**\n * Get tool definitions in JSON Schema format for passing to remote agents.\n */\n async getToolDefinitions(): Promise<AguiToolDefinition[]> {\n if (this.isMultiSession()) {\n const clients = (this.client as MultiSessionClient).getClients();\n const allTools: AguiToolDefinition[] = [];\n for (const client of clients) {\n allTools.push(...await this.transformToolDefinitions(client));\n }\n return allTools;\n }\n return this.transformToolDefinitions(this.client as MCPClient);\n }\n\n /**\n * Get tools as a function (for dynamic loading).\n */\n getToolsFunction(): () => Promise<AguiTool[]> {\n return () => this.getTools();\n }\n\n private isMultiSession(): boolean {\n return typeof (this.client as any).getClients === 'function';\n }\n\n private async transformTools(client: MCPClient): Promise<AguiTool[]> {\n if (!client.isConnected()) return [];\n\n const result = await client.listTools();\n const serverId = (typeof (client as any).getServerId === 'function'\n ? (client as any).getServerId()\n : undefined) as string | undefined;\n const normalizedPrefix = this.options.prefix?.replace(/-/g, '') ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n const prefix = `tool_${normalizedPrefix}`;\n\n return result.tools.map(tool => {\n // Type assertion to access _meta if it exists on the tool object (it comes from MCP SDK)\n const mcpTool = tool as any;\n const mcpToolName = tool.name;\n return {\n name: `${prefix}_${tool.name}`,\n description: tool.description || `Execute ${tool.name}`,\n parameters: cleanSchema(tool.inputSchema),\n _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\n handler: async (args: any) => {\n // Call the actual MCP tool\n const callResult = await (client as any).callTool(mcpToolName, args ?? {});\n\n // Return the raw result object so middleware can inspect `_meta` (e.g. for UI triggers)\n return callResult;\n }\n }\n });\n }\n\n private async transformToolDefinitions(client: MCPClient): Promise<AguiToolDefinition[]> {\n if (!client.isConnected()) return [];\n\n const result = await client.listTools();\n const serverId = (typeof (client as any).getServerId === 'function'\n ? (client as any).getServerId()\n : undefined) as string | undefined;\n const normalizedPrefix = this.options.prefix?.replace(/-/g, '') ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n const prefix = `tool_${normalizedPrefix}`;\n\n return result.tools.map(tool => {\n const mcpTool = tool as any;\n return {\n name: `${prefix}_${tool.name}`,\n description: tool.description || `Execute ${tool.name}`,\n parameters: cleanSchema(tool.inputSchema),\n _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\n };\n });\n }\n}\n","/**\n * AG-UI Middleware for MCP Tool Execution\n *\n * This middleware intercepts tool calls from remote agents and executes\n * MCP tools server-side, returning results back to the agent.\n *\n * @requires @ag-ui/client - Peer dependency for AG-UI types\n * @requires rxjs - Uses RxJS Observables for event streaming\n */\n\nimport { Observable, Subscriber } from 'rxjs';\nimport {\n Middleware,\n EventType,\n type AbstractAgent,\n type RunAgentInput,\n type BaseEvent,\n type ToolCallEndEvent,\n type Tool,\n} from '@ag-ui/client';\nimport { type AguiTool, cleanSchema } from './agui-adapter.js';\n\n/** Tool execution result for continuation */\ninterface ToolResult {\n toolCallId: string;\n toolName: string;\n result: string;\n messageId: string;\n}\n\n/** State for tracking tool calls during a run */\ninterface RunState {\n toolCallArgsBuffer: Map<string, string>;\n toolCallNames: Map<string, string>;\n pendingMcpCalls: Set<string>;\n textContent?: string;\n error: boolean;\n}\n\n/**\n * Configuration for McpMiddleware\n */\nexport interface McpMiddlewareConfig {\n /** Pre-loaded tools with handlers (required) */\n tools: AguiTool[];\n}\n\n/**\n * AG-UI Middleware that executes MCP tools server-side.\n */\nexport class McpMiddleware extends Middleware {\n private tools: AguiTool[];\n private toolSchemas: Tool[];\n\n constructor(config: McpMiddlewareConfig) {\n super();\n this.tools = config.tools;\n this.toolSchemas = this.tools.map((t: AguiTool) => ({\n name: t.name,\n description: t.description,\n parameters: cleanSchema(t.parameters),\n }));\n }\n\n private isMcpTool(toolName: string): boolean {\n return this.tools.some(t => t.name === toolName);\n }\n\n private parseArgs(argsString: string): Record<string, any> {\n if (!argsString?.trim()) return {};\n\n try {\n return JSON.parse(argsString);\n } catch {\n // Handle duplicated JSON from streaming issues: {...}{...}\n const trimmed = argsString.trim();\n if (trimmed.includes('}{')) {\n const firstObject = trimmed.slice(0, trimmed.indexOf('}{') + 1);\n try {\n return JSON.parse(firstObject);\n } catch {\n console.error(`[McpMiddleware] Failed to parse JSON:`, firstObject);\n }\n }\n console.error(`[McpMiddleware] Failed to parse args:`, argsString);\n return {};\n }\n }\n\n private async executeTool(toolName: string, args: Record<string, any>): Promise<string> {\n const tool = this.tools.find(t => t.name === toolName);\n if (!tool?.handler) {\n return `Error: Tool ${tool ? 'has no handler' : 'not found'}: ${toolName}`;\n }\n\n try {\n console.log(`[McpMiddleware] Executing tool: ${toolName}`, args);\n const result = await tool.handler(args);\n const resultStr = typeof result === 'string' ? result : JSON.stringify(result);\n\n console.log(`[McpMiddleware] Tool result:`, resultStr.slice(0, 200));\n return resultStr;\n } catch (error: any) {\n console.error(`[McpMiddleware] Error executing tool:`, error);\n return `Error: ${error.message || String(error)}`;\n }\n }\n\n private generateId(prefix: string): string {\n return `${prefix}_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;\n }\n\n private ensureIds(input: RunAgentInput): void {\n const anyInput = input as any;\n if (!anyInput.threadId) anyInput.threadId = this.generateId('mcp_thread');\n if (!anyInput.runId) anyInput.runId = this.generateId('mcp_run');\n }\n\n /** Process tool call events and update state */\n private handleToolCallEvent(event: BaseEvent, state: RunState): void {\n const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\n\n // Accumulate text content for reconstruction\n if (event.type === EventType.TEXT_MESSAGE_CHUNK) {\n const e = event as any;\n if (e.delta) {\n state.textContent = (state.textContent || '') + e.delta;\n }\n }\n\n if (event.type === EventType.TOOL_CALL_START) {\n const e = event as any;\n if (e.toolCallId && e.toolCallName) {\n toolCallNames.set(e.toolCallId, e.toolCallName);\n if (this.isMcpTool(e.toolCallName)) {\n pendingMcpCalls.add(e.toolCallId);\n }\n console.log(`[McpMiddleware] TOOL_CALL_START: ${e.toolCallName} (id: ${e.toolCallId}, isMCP: ${this.isMcpTool(e.toolCallName)})`);\n }\n }\n\n if (event.type === EventType.TOOL_CALL_ARGS) {\n const e = event as any;\n if (e.toolCallId && e.delta) {\n const existing = toolCallArgsBuffer.get(e.toolCallId) || '';\n toolCallArgsBuffer.set(e.toolCallId, existing + e.delta);\n }\n }\n\n if (event.type === EventType.TOOL_CALL_END) {\n const e = event as ToolCallEndEvent;\n console.log(`[McpMiddleware] TOOL_CALL_END: ${toolCallNames.get(e.toolCallId) ?? 'unknown'} (id: ${e.toolCallId})`);\n }\n\n // Workaround: Extract parallel tool calls from MESSAGES_SNAPSHOT\n if (event.type === EventType.MESSAGES_SNAPSHOT) {\n const messages = (event as any).messages || [];\n if (messages.length > 0) {\n const lastMsg = messages[messages.length - 1];\n // Update text content from snapshot if available (often more reliable)\n if (lastMsg.role === 'assistant' && lastMsg.content) {\n state.textContent = lastMsg.content;\n }\n\n // Discover tools\n for (let i = messages.length - 1; i >= 0; i--) {\n const msg = messages[i];\n const tools = Array.isArray(msg.toolCalls) ? msg.toolCalls :\n (Array.isArray(msg.tool_calls) ? msg.tool_calls : []);\n\n if (msg.role === 'assistant' && tools.length > 0) {\n for (const tc of tools) {\n if (tc.id && tc.function?.name && !toolCallNames.has(tc.id)) {\n toolCallNames.set(tc.id, tc.function.name);\n toolCallArgsBuffer.set(tc.id, tc.function.arguments || '{}');\n if (this.isMcpTool(tc.function.name)) {\n pendingMcpCalls.add(tc.id);\n console.log(`[McpMiddleware] MESSAGES_SNAPSHOT: Discovered ${tc.function.name} (id: ${tc.id})`);\n }\n }\n }\n break;\n }\n }\n }\n }\n }\n\n /** Execute pending MCP tools and return results */\n private async executeTools(state: RunState): Promise<ToolResult[]> {\n const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\n const results: ToolResult[] = [];\n\n const promises = [...pendingMcpCalls].map(async (toolCallId) => {\n const toolName = toolCallNames.get(toolCallId);\n if (!toolName) return;\n\n const args = this.parseArgs(toolCallArgsBuffer.get(toolCallId) || '{}');\n console.log(`[McpMiddleware] Executing pending tool: ${toolName}`);\n\n const result = await this.executeTool(toolName, args);\n results.push({\n toolCallId,\n toolName,\n result,\n messageId: this.generateId('mcp_result'),\n });\n pendingMcpCalls.delete(toolCallId);\n });\n\n await Promise.all(promises);\n return results;\n }\n\n /** Emit tool results (without RUN_FINISHED - that's emitted when truly done) */\n private emitToolResults(observer: Subscriber<BaseEvent>, results: ToolResult[]): void {\n for (const { toolCallId, toolName, result, messageId } of results) {\n observer.next({\n type: EventType.TOOL_CALL_RESULT,\n toolCallId,\n messageId,\n content: result,\n role: 'tool',\n timestamp: Date.now(),\n } as any);\n console.log(`[McpMiddleware] Emitting TOOL_CALL_RESULT for: ${toolName}`);\n }\n }\n\n run(input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> {\n return new Observable<BaseEvent>((observer: Subscriber<BaseEvent>) => {\n const state: RunState = {\n toolCallArgsBuffer: new Map(),\n toolCallNames: new Map(),\n pendingMcpCalls: new Set(),\n textContent: '',\n error: false,\n };\n\n this.ensureIds(input);\n const anyInput = input as any;\n\n console.log(`[McpMiddleware] === NEW RUN ===`);\n console.log(`[McpMiddleware] threadId: ${anyInput.threadId}, runId: ${anyInput.runId}`);\n console.log(`[McpMiddleware] messages: ${input.messages?.length ?? 0}, tools: ${this.tools?.length ?? 0}`);\n\n // Inject MCP tools\n if (this.toolSchemas?.length) {\n input.tools = [...(input.tools || []), ...this.toolSchemas];\n console.log(`[McpMiddleware] Injected ${this.toolSchemas.length} tools:`, this.toolSchemas.map((t: Tool) => t.name));\n }\n\n const handleRunFinished = async () => {\n if (state.error) return; // Don't continue after error\n\n if (state.pendingMcpCalls.size === 0) {\n observer.next({\n type: EventType.RUN_FINISHED,\n threadId: anyInput.threadId,\n runId: anyInput.runId,\n timestamp: Date.now(),\n } as any);\n observer.complete();\n return;\n }\n\n console.log(`[McpMiddleware] RUN_FINISHED with ${state.pendingMcpCalls.size} pending calls`);\n\n // Reconstruct the Assistant Message that triggered these tools\n const toolCalls = [];\n for (const toolCallId of state.pendingMcpCalls) {\n const name = state.toolCallNames.get(toolCallId);\n const args = state.toolCallArgsBuffer.get(toolCallId) || '{}';\n if (name) {\n toolCalls.push({\n id: toolCallId,\n type: 'function',\n function: { name, arguments: args }\n });\n }\n }\n\n // Add the Assistant Message to history FIRST\n if (toolCalls.length > 0 || state.textContent) {\n const assistantMsg = {\n id: this.generateId('msg_ast'),\n role: 'assistant',\n content: state.textContent || null, // Ensure null if empty string for strict LLMs\n tool_calls: toolCalls.length > 0 ? toolCalls : undefined\n };\n input.messages.push(assistantMsg as any);\n console.log(`[McpMiddleware] Added assistant message to history before tools: ${state.textContent?.slice(0, 50)}... [${toolCalls.length} tools]`);\n }\n\n // Execute tools and emit results (no RUN_FINISHED yet - continuation follows)\n const results = await this.executeTools(state);\n this.emitToolResults(observer, results);\n\n // Prepare continuation\n console.log(`[McpMiddleware] Triggering continuation with ${results.length} results`);\n\n // Add tool result messages to history\n for (const { toolCallId, result, messageId } of results) {\n input.messages.push({\n id: messageId,\n role: 'tool',\n tool_call_id: toolCallId,\n content: result,\n } as any);\n }\n\n // Reset state for next turn\n state.toolCallArgsBuffer.clear();\n state.toolCallNames.clear();\n state.textContent = ''; // Clear text content for next turn\n\n anyInput.runId = this.generateId('mcp_run');\n console.log(`[McpMiddleware] === CONTINUATION RUN === messages: ${input.messages.length}`);\n\n // Subscribe to continuation\n next.run(input).subscribe({\n next: (event) => {\n if (state.error) return;\n\n this.handleToolCallEvent(event, state);\n\n if (event.type === EventType.RUN_ERROR) {\n console.log(`[McpMiddleware] RUN_ERROR received in continuation`);\n state.error = true;\n observer.next(event);\n observer.complete();\n return;\n }\n\n if (event.type === EventType.RUN_STARTED) {\n console.log(`[McpMiddleware] Filtering RUN_STARTED from continuation`);\n return;\n }\n\n if (event.type === EventType.RUN_FINISHED) {\n if (state.pendingMcpCalls.size > 0) {\n handleRunFinished();\n } else {\n observer.next(event);\n observer.complete();\n }\n return;\n }\n observer.next(event);\n },\n error: (err) => {\n state.error = true;\n observer.error(err);\n },\n complete: () => {\n if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\n },\n });\n };\n\n const subscription = next.run(input).subscribe({\n next: (event) => {\n if (state.error) return;\n\n this.handleToolCallEvent(event, state);\n\n if (event.type === EventType.RUN_ERROR) {\n console.log(`[McpMiddleware] RUN_ERROR received`);\n state.error = true;\n observer.next(event);\n observer.complete();\n return;\n }\n\n if (event.type === EventType.RUN_FINISHED) {\n handleRunFinished();\n return;\n }\n observer.next(event);\n },\n error: (err) => {\n state.error = true;\n observer.error(err);\n },\n complete: () => {\n if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\n },\n });\n\n return () => subscription.unsubscribe();\n });\n }\n}\n\n/**\n * Factory function to create MCP middleware.\n */\nexport function createMcpMiddleware(\n options: { tools: AguiTool[] }\n) {\n const middleware = new McpMiddleware(options);\n return (input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> => {\n return middleware.run(input, next);\n };\n}\n\n// Legacy exports\nexport { McpMiddleware as McpToolExecutorMiddleware };\nexport { createMcpMiddleware as createMcpToolMiddleware };\n\n// Re-exports\nexport { Middleware, EventType };\nexport type { RunAgentInput, BaseEvent, AbstractAgent, ToolCallEndEvent, Tool };"]}
1
+ {"version":3,"sources":["../../src/adapters/agui-adapter.ts","../../src/adapters/agui-middleware.ts"],"names":[],"mappings":";;;;;;;;;AAqCA,IAAM,wBAAA,GAA2B;AAAA;AAAA,EAE7B,SAAA;AAAA,EAAW,KAAA;AAAA,EAAO,UAAA;AAAA,EAAY,OAAA;AAAA,EAAS,aAAA;AAAA;AAAA,EAEvC,SAAA;AAAA,EAAW,UAAA;AAAA,EAAY,YAAA;AAAA,EAAc,kBAAA;AAAA;AAAA,EAErC,YAAA;AAAA,EAAc,UAAA;AAAA,EAAY,WAAA;AAAA,EAAa,kBAAA;AAAA,EAAoB;AAC/D,CAAA;AASO,SAAS,YAAY,MAAA,EAA8D;AACtF,EAAA,IAAI,CAAC,MAAA,EAAQ;AACT,IAAA,OAAO,EAAE,IAAA,EAAM,QAAA,EAAU,UAAA,EAAY,EAAC,EAAE;AAAA,EAC5C;AAEA,EAAA,MAAM,OAAA,GAAU,EAAE,GAAG,MAAA,EAAO;AAG5B,EAAA,KAAA,MAAW,QAAQ,wBAAA,EAA0B;AACzC,IAAA,OAAO,QAAQ,IAAI,CAAA;AAAA,EACvB;AAGA,EAAA,IAAI,OAAA,CAAQ,UAAA,IAAc,OAAO,OAAA,CAAQ,eAAe,QAAA,EAAU;AAC9D,IAAA,MAAM,eAAoC,EAAC;AAC3C,IAAA,KAAA,MAAW,CAAC,KAAK,KAAK,CAAA,IAAK,OAAO,OAAA,CAAQ,OAAA,CAAQ,UAAU,CAAA,EAAG;AAC3D,MAAA,IAAI,OAAO,KAAA,KAAU,QAAA,IAAY,KAAA,KAAU,IAAA,EAAM;AAC7C,QAAA,YAAA,CAAa,GAAG,CAAA,GAAI,WAAA,CAAY,KAA4B,CAAA;AAAA,MAChE,CAAA,MAAO;AACH,QAAA,YAAA,CAAa,GAAG,CAAA,GAAI,KAAA;AAAA,MACxB;AAAA,IACJ;AACA,IAAA,OAAA,CAAQ,UAAA,GAAa,YAAA;AAAA,EACzB;AAGA,EAAA,IAAI,OAAA,CAAQ,KAAA,IAAS,OAAO,OAAA,CAAQ,UAAU,QAAA,EAAU;AACpD,IAAA,OAAA,CAAQ,KAAA,GAAQ,WAAA,CAAY,OAAA,CAAQ,KAAK,CAAA;AAAA,EAC7C;AAGA,EAAA,IAAI,OAAA,CAAQ,oBAAA,IAAwB,OAAO,OAAA,CAAQ,yBAAyB,QAAA,EAAU;AAClF,IAAA,OAAA,CAAQ,oBAAA,GAAuB,WAAA,CAAY,OAAA,CAAQ,oBAAoB,CAAA;AAAA,EAC3E;AAEA,EAAA,OAAO,OAAA;AACX;;;ACvCO,IAAM,aAAA,GAAN,cAA4B,UAAA,CAAW;AAAA,EAI1C,YAAY,MAAA,EAA6B;AACrC,IAAA,KAAA,EAAM;AAJV,IAAA,aAAA,CAAA,IAAA,EAAQ,OAAA,CAAA;AACR,IAAA,aAAA,CAAA,IAAA,EAAQ,aAAA,CAAA;AAIJ,IAAA,IAAA,CAAK,QAAQ,MAAA,CAAO,KAAA;AACpB,IAAA,IAAA,CAAK,WAAA,GAAc,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,CAAC,CAAA,MAAiB;AAAA,MAChD,MAAM,CAAA,CAAE,IAAA;AAAA,MACR,aAAa,CAAA,CAAE,WAAA;AAAA,MACf,UAAA,EAAY,WAAA,CAAY,CAAA,CAAE,UAAU;AAAA,KACxC,CAAE,CAAA;AAAA,EACN;AAAA,EAEQ,UAAU,QAAA,EAA2B;AACzC,IAAA,OAAO,KAAK,KAAA,CAAM,IAAA,CAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AAAA,EACnD;AAAA,EAEQ,UAAU,UAAA,EAAyC;AACvD,IAAA,IAAI,CAAC,UAAA,EAAY,IAAA,EAAK,SAAU,EAAC;AAEjC,IAAA,IAAI;AACA,MAAA,OAAO,IAAA,CAAK,MAAM,UAAU,CAAA;AAAA,IAChC,CAAA,CAAA,MAAQ;AAEJ,MAAA,MAAM,OAAA,GAAU,WAAW,IAAA,EAAK;AAChC,MAAA,IAAI,OAAA,CAAQ,QAAA,CAAS,IAAI,CAAA,EAAG;AACxB,QAAA,MAAM,WAAA,GAAc,QAAQ,KAAA,CAAM,CAAA,EAAG,QAAQ,OAAA,CAAQ,IAAI,IAAI,CAAC,CAAA;AAC9D,QAAA,IAAI;AACA,UAAA,OAAO,IAAA,CAAK,MAAM,WAAW,CAAA;AAAA,QACjC,CAAA,CAAA,MAAQ;AACJ,UAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,WAAW,CAAA;AAAA,QACtE;AAAA,MACJ;AACA,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,UAAU,CAAA;AACjE,MAAA,OAAO,EAAC;AAAA,IACZ;AAAA,EACJ;AAAA,EAEA,MAAc,WAAA,CAAY,QAAA,EAAkB,IAAA,EAA4C;AACpF,IAAA,MAAM,OAAO,IAAA,CAAK,KAAA,CAAM,KAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AACrD,IAAA,IAAI,CAAC,MAAM,OAAA,EAAS;AAChB,MAAA,OAAO,CAAA,YAAA,EAAe,IAAA,GAAO,gBAAA,GAAmB,WAAW,KAAK,QAAQ,CAAA,CAAA;AAAA,IAC5E;AAEA,IAAA,IAAI;AACA,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,gCAAA,EAAmC,QAAQ,CAAA,CAAA,EAAI,IAAI,CAAA;AAC/D,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA;AACtC,MAAA,MAAM,YAAY,OAAO,MAAA,KAAW,WAAW,MAAA,GAAS,IAAA,CAAK,UAAU,MAAM,CAAA;AAE7E,MAAA,OAAA,CAAQ,IAAI,CAAA,4BAAA,CAAA,EAAgC,SAAA,CAAU,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC,CAAA;AACnE,MAAA,OAAO,SAAA;AAAA,IACX,SAAS,KAAA,EAAY;AACjB,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,KAAK,CAAA;AAC5D,MAAA,OAAO,CAAA,OAAA,EAAU,KAAA,CAAM,OAAA,IAAW,MAAA,CAAO,KAAK,CAAC,CAAA,CAAA;AAAA,IACnD;AAAA,EACJ;AAAA,EAEQ,WAAW,MAAA,EAAwB;AACvC,IAAA,OAAO,GAAG,MAAM,CAAA,CAAA,EAAI,IAAA,CAAK,GAAA,EAAK,CAAA,CAAA,EAAI,IAAA,CAAK,MAAA,EAAO,CAAE,SAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAA,EAAG,CAAC,CAAC,CAAA,CAAA;AAAA,EAC5E;AAAA,EAEQ,UAAU,KAAA,EAA4B;AAC1C,IAAA,MAAM,QAAA,GAAW,KAAA;AACjB,IAAA,IAAI,CAAC,QAAA,CAAS,QAAA,WAAmB,QAAA,GAAW,IAAA,CAAK,WAAW,YAAY,CAAA;AACxE,IAAA,IAAI,CAAC,QAAA,CAAS,KAAA,WAAgB,KAAA,GAAQ,IAAA,CAAK,WAAW,SAAS,CAAA;AAAA,EACnE;AAAA;AAAA,EAGQ,mBAAA,CAAoB,OAAkB,KAAA,EAAuB;AACjE,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAG/D,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,kBAAA,EAAoB;AAC7C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,EAAE,KAAA,EAAO;AACT,QAAA,KAAA,CAAM,WAAA,GAAA,CAAe,KAAA,CAAM,WAAA,IAAe,EAAA,IAAM,CAAA,CAAE,KAAA;AAAA,MACtD;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,eAAA,EAAiB;AAC1C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,YAAA,EAAc;AAChC,QAAA,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,CAAA,CAAE,YAAY,CAAA;AAC9C,QAAA,IAAI,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAA,EAAG;AAChC,UAAA,eAAA,CAAgB,GAAA,CAAI,EAAE,UAAU,CAAA;AAAA,QACpC;AACA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iCAAA,EAAoC,CAAA,CAAE,YAAY,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,SAAA,EAAY,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAC,CAAA,CAAA,CAAG,CAAA;AAAA,MACpI;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,cAAA,EAAgB;AACzC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,KAAA,EAAO;AACzB,QAAA,MAAM,QAAA,GAAW,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,EAAA;AACzD,QAAA,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,QAAA,GAAW,EAAE,KAAK,CAAA;AAAA,MAC3D;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,aAAA,EAAe;AACxC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+BAAA,EAAkC,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,SAAS,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,CAAA,CAAG,CAAA;AAAA,IACtH;AAGA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,iBAAA,EAAmB;AAC5C,MAAA,MAAM,QAAA,GAAY,KAAA,CAAc,QAAA,IAAY,EAAC;AAC7C,MAAA,IAAI,QAAA,CAAS,SAAS,CAAA,EAAG;AACrB,QAAA,MAAM,OAAA,GAAU,QAAA,CAAS,QAAA,CAAS,MAAA,GAAS,CAAC,CAAA;AAE5C,QAAA,IAAI,OAAA,CAAQ,IAAA,KAAS,WAAA,IAAe,OAAA,CAAQ,OAAA,EAAS;AACjD,UAAA,KAAA,CAAM,cAAc,OAAA,CAAQ,OAAA;AAAA,QAChC;AAGA,QAAA,KAAA,IAAS,IAAI,QAAA,CAAS,MAAA,GAAS,CAAA,EAAG,CAAA,IAAK,GAAG,CAAA,EAAA,EAAK;AAC3C,UAAA,MAAM,GAAA,GAAM,SAAS,CAAC,CAAA;AACtB,UAAA,MAAM,KAAA,GAAQ,KAAA,CAAM,OAAA,CAAQ,GAAA,CAAI,SAAS,CAAA,GAAI,GAAA,CAAI,SAAA,GAC5C,KAAA,CAAM,QAAQ,GAAA,CAAI,UAAU,CAAA,GAAI,GAAA,CAAI,aAAa,EAAC;AAEvD,UAAA,IAAI,GAAA,CAAI,IAAA,KAAS,WAAA,IAAe,KAAA,CAAM,SAAS,CAAA,EAAG;AAC9C,YAAA,KAAA,MAAW,MAAM,KAAA,EAAO;AACpB,cAAA,IAAI,EAAA,CAAG,EAAA,IAAM,EAAA,CAAG,QAAA,EAAU,IAAA,IAAQ,CAAC,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAE,CAAA,EAAG;AACzD,gBAAA,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,SAAS,IAAI,CAAA;AACzC,gBAAA,kBAAA,CAAmB,IAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,QAAA,CAAS,aAAa,IAAI,CAAA;AAC3D,gBAAA,IAAI,IAAA,CAAK,SAAA,CAAU,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,EAAG;AAClC,kBAAA,eAAA,CAAgB,GAAA,CAAI,GAAG,EAAE,CAAA;AACzB,kBAAA,OAAA,CAAQ,GAAA,CAAI,iDAAiD,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,MAAA,EAAS,EAAA,CAAG,EAAE,CAAA,CAAA,CAAG,CAAA;AAAA,gBAClG;AAAA,cACJ;AAAA,YACJ;AACA,YAAA;AAAA,UACJ;AAAA,QACJ;AAAA,MACJ;AAAA,IACJ;AAAA,EACJ;AAAA;AAAA,EAGA,MAAc,aAAa,KAAA,EAAwC;AAC/D,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAC/D,IAAA,MAAM,UAAwB,EAAC;AAE/B,IAAA,MAAM,WAAW,CAAC,GAAG,eAAe,CAAA,CAAE,GAAA,CAAI,OAAO,UAAA,KAAe;AAC5D,MAAA,MAAM,QAAA,GAAW,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC7C,MAAA,IAAI,CAAC,QAAA,EAAU;AAEf,MAAA,MAAM,OAAO,IAAA,CAAK,SAAA,CAAU,mBAAmB,GAAA,CAAI,UAAU,KAAK,IAAI,CAAA;AACtE,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,wCAAA,EAA2C,QAAQ,CAAA,CAAE,CAAA;AAEjE,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,WAAA,CAAY,UAAU,IAAI,CAAA;AACpD,MAAA,OAAA,CAAQ,IAAA,CAAK;AAAA,QACT,UAAA;AAAA,QACA,QAAA;AAAA,QACA,MAAA;AAAA,QACA,SAAA,EAAW,IAAA,CAAK,UAAA,CAAW,YAAY;AAAA,OAC1C,CAAA;AACD,MAAA,eAAA,CAAgB,OAAO,UAAU,CAAA;AAAA,IACrC,CAAC,CAAA;AAED,IAAA,MAAM,OAAA,CAAQ,IAAI,QAAQ,CAAA;AAC1B,IAAA,OAAO,OAAA;AAAA,EACX;AAAA;AAAA,EAGQ,eAAA,CAAgB,UAAiC,OAAA,EAA6B;AAClF,IAAA,KAAA,MAAW,EAAE,UAAA,EAAY,QAAA,EAAU,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AAC/D,MAAA,QAAA,CAAS,IAAA,CAAK;AAAA,QACV,MAAM,SAAA,CAAU,gBAAA;AAAA,QAChB,UAAA;AAAA,QACA,SAAA;AAAA,QACA,OAAA,EAAS,MAAA;AAAA,QACT,IAAA,EAAM,MAAA;AAAA,QACN,SAAA,EAAW,KAAK,GAAA;AAAI,OAChB,CAAA;AACR,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+CAAA,EAAkD,QAAQ,CAAA,CAAE,CAAA;AAAA,IAC5E;AAAA,EACJ;AAAA,EAEA,GAAA,CAAI,OAAsB,IAAA,EAA4C;AAClE,IAAA,OAAO,IAAI,UAAA,CAAsB,CAAC,QAAA,KAAoC;AAClE,MAAA,MAAM,KAAA,GAAkB;AAAA,QACpB,kBAAA,sBAAwB,GAAA,EAAI;AAAA,QAC5B,aAAA,sBAAmB,GAAA,EAAI;AAAA,QACvB,eAAA,sBAAqB,GAAA,EAAI;AAAA,QACzB,WAAA,EAAa,EAAA;AAAA,QACb,KAAA,EAAO;AAAA,OACX;AAEA,MAAA,IAAA,CAAK,UAAU,KAAK,CAAA;AACpB,MAAA,MAAM,QAAA,GAAW,KAAA;AAEjB,MAAA,OAAA,CAAQ,IAAI,CAAA,+BAAA,CAAiC,CAAA;AAC7C,MAAA,OAAA,CAAQ,IAAI,CAAA,0BAAA,EAA6B,QAAA,CAAS,QAAQ,CAAA,SAAA,EAAY,QAAA,CAAS,KAAK,CAAA,CAAE,CAAA;AACtF,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,0BAAA,EAA6B,KAAA,CAAM,QAAA,EAAU,MAAA,IAAU,CAAC,CAAA,SAAA,EAAY,IAAA,CAAK,KAAA,EAAO,MAAA,IAAU,CAAC,CAAA,CAAE,CAAA;AAGzG,MAAA,IAAI,IAAA,CAAK,aAAa,MAAA,EAAQ;AAC1B,QAAA,KAAA,CAAM,KAAA,GAAQ,CAAC,GAAI,KAAA,CAAM,SAAS,EAAC,EAAI,GAAG,IAAA,CAAK,WAAW,CAAA;AAC1D,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,yBAAA,EAA4B,IAAA,CAAK,WAAA,CAAY,MAAM,CAAA,OAAA,CAAA,EAAW,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,CAAC,CAAA,KAAY,CAAA,CAAE,IAAI,CAAC,CAAA;AAAA,MACvH;AAEA,MAAA,MAAM,oBAAoB,YAAY;AAClC,QAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,QAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,KAAS,CAAA,EAAG;AAClC,UAAA,QAAA,CAAS,IAAA,CAAK;AAAA,YACV,MAAM,SAAA,CAAU,YAAA;AAAA,YAChB,UAAU,QAAA,CAAS,QAAA;AAAA,YACnB,OAAO,QAAA,CAAS,KAAA;AAAA,YAChB,SAAA,EAAW,KAAK,GAAA;AAAI,WAChB,CAAA;AACR,UAAA,QAAA,CAAS,QAAA,EAAS;AAClB,UAAA;AAAA,QACJ;AAEA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kCAAA,EAAqC,KAAA,CAAM,eAAA,CAAgB,IAAI,CAAA,cAAA,CAAgB,CAAA;AAG3F,QAAA,MAAM,YAAY,EAAC;AACnB,QAAA,KAAA,MAAW,UAAA,IAAc,MAAM,eAAA,EAAiB;AAC5C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC/C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,kBAAA,CAAmB,GAAA,CAAI,UAAU,CAAA,IAAK,IAAA;AACzD,UAAA,IAAI,IAAA,EAAM;AACN,YAAA,SAAA,CAAU,IAAA,CAAK;AAAA,cACX,EAAA,EAAI,UAAA;AAAA,cACJ,IAAA,EAAM,UAAA;AAAA,cACN,QAAA,EAAU,EAAE,IAAA,EAAM,SAAA,EAAW,IAAA;AAAK,aACrC,CAAA;AAAA,UACL;AAAA,QACJ;AAGA,QAAA,IAAI,SAAA,CAAU,MAAA,GAAS,CAAA,IAAK,KAAA,CAAM,WAAA,EAAa;AAC3C,UAAA,MAAM,YAAA,GAAe;AAAA,YACjB,EAAA,EAAI,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAAA,YAC7B,IAAA,EAAM,WAAA;AAAA,YACN,OAAA,EAAS,MAAM,WAAA,IAAe,IAAA;AAAA;AAAA,YAC9B,UAAA,EAAY,SAAA,CAAU,MAAA,GAAS,CAAA,GAAI,SAAA,GAAY;AAAA,WACnD;AACA,UAAA,KAAA,CAAM,QAAA,CAAS,KAAK,YAAmB,CAAA;AACvC,UAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iEAAA,EAAoE,KAAA,CAAM,WAAA,EAAa,KAAA,CAAM,CAAA,EAAG,EAAE,CAAC,CAAA,KAAA,EAAQ,SAAA,CAAU,MAAM,CAAA,OAAA,CAAS,CAAA;AAAA,QACpJ;AAGA,QAAA,MAAM,OAAA,GAAU,MAAM,IAAA,CAAK,YAAA,CAAa,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAK,eAAA,CAAgB,UAAU,OAAO,CAAA;AAGtC,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,6CAAA,EAAgD,OAAA,CAAQ,MAAM,CAAA,QAAA,CAAU,CAAA;AAGpF,QAAA,KAAA,MAAW,EAAE,UAAA,EAAY,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AACrD,UAAA,KAAA,CAAM,SAAS,IAAA,CAAK;AAAA,YAChB,EAAA,EAAI,SAAA;AAAA,YACJ,IAAA,EAAM,MAAA;AAAA,YACN,YAAA,EAAc,UAAA;AAAA,YACd,OAAA,EAAS;AAAA,WACL,CAAA;AAAA,QACZ;AAGA,QAAA,KAAA,CAAM,mBAAmB,KAAA,EAAM;AAC/B,QAAA,KAAA,CAAM,cAAc,KAAA,EAAM;AAC1B,QAAA,KAAA,CAAM,WAAA,GAAc,EAAA;AAEpB,QAAA,QAAA,CAAS,KAAA,GAAQ,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAC1C,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,mDAAA,EAAsD,KAAA,CAAM,QAAA,CAAS,MAAM,CAAA,CAAE,CAAA;AAGzF,QAAA,IAAA,CAAK,GAAA,CAAI,KAAK,CAAA,CAAE,SAAA,CAAU;AAAA,UACtB,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,YAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,YAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,SAAA,EAAW;AACpC,cAAA,OAAA,CAAQ,IAAI,CAAA,kDAAA,CAAoD,CAAA;AAChE,cAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,cAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,cAAA,QAAA,CAAS,QAAA,EAAS;AAClB,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,WAAA,EAAa;AACtC,cAAA,OAAA,CAAQ,IAAI,CAAA,uDAAA,CAAyD,CAAA;AACrE,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,cAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,GAAO,CAAA,EAAG;AAChC,gBAAA,iBAAA,EAAkB;AAAA,cACtB,CAAA,MAAO;AACH,gBAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,gBAAA,QAAA,CAAS,QAAA,EAAS;AAAA,cACtB;AACA,cAAA;AAAA,YACJ;AACA,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,UACvB,CAAA;AAAA,UACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,UACtB,CAAA;AAAA,UACA,UAAU,MAAM;AACZ,YAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,UAC5E;AAAA,SACH,CAAA;AAAA,MACL,CAAA;AAEA,MAAA,MAAM,YAAA,GAAe,IAAA,CAAK,GAAA,CAAI,KAAK,EAAE,SAAA,CAAU;AAAA,QAC3C,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,UAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,UAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,SAAA,EAAW;AACpC,YAAA,OAAA,CAAQ,IAAI,CAAA,kCAAA,CAAoC,CAAA;AAChD,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,YAAA,QAAA,CAAS,QAAA,EAAS;AAClB,YAAA;AAAA,UACJ;AAEA,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,YAAA,iBAAA,EAAkB;AAClB,YAAA;AAAA,UACJ;AACA,UAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,QACvB,CAAA;AAAA,QACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,UAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,UAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,QACtB,CAAA;AAAA,QACA,UAAU,MAAM;AACZ,UAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,QAC5E;AAAA,OACH,CAAA;AAED,MAAA,OAAO,MAAM,aAAa,WAAA,EAAY;AAAA,IAC1C,CAAC,CAAA;AAAA,EACL;AACJ;AAKO,SAAS,oBACZ,OAAA,EACF;AACE,EAAA,MAAM,UAAA,GAAa,IAAI,aAAA,CAAc,OAAO,CAAA;AAC5C,EAAA,OAAO,CAAC,OAAsB,IAAA,KAA+C;AACzE,IAAA,OAAO,UAAA,CAAW,GAAA,CAAI,KAAA,EAAO,IAAI,CAAA;AAAA,EACrC,CAAA;AACJ","file":"agui-middleware.mjs","sourcesContent":["/**\n * MCP Adapter for AG-UI Integration\n *\n * This adapter transforms MCP tools into formats compatible with AG-UI agents.\n * It provides tools with handlers for server-side execution and tool definitions\n * in JSON Schema format for passing to remote agents.\n *\n * @example\n * ```typescript\n * import { MultiSessionClient } from '@mcp-ts/sdk/server';\n * import { AguiAdapter } from '@mcp-ts/sdk/adapters/agui-adapter';\n * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';\n * import { HttpAgent } from '@ag-ui/client';\n *\n * // Create MCP client\n * const mcpClient = new MultiSessionClient('user_123');\n * await mcpClient.connect();\n *\n * // Create adapter and get tools\n * const adapter = new AguiAdapter(mcpClient);\n * const tools = await adapter.getTools();\n *\n * // Use with AG-UI middleware\n * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });\n * agent.use(createMcpMiddleware({ tools }));\n * ```\n */\n\nimport { MCPClient } from '../server/mcp/oauth-client.js';\nimport { MultiSessionClient } from '../server/mcp/multi-session-client.js';\nimport { ToolRouter } from '../shared/tool-router.js';\nimport { executeMetaTool, isMetaTool } from '../shared/meta-tools.js';\n\n/**\n * Extended JSON Schema properties that Pydantic's strict validation rejects.\n * These are valid JSON Schema extensions but not part of the core spec.\n */\nconst PYDANTIC_FORBIDDEN_PROPS = [\n // JSON Schema meta-properties\n '$schema', '$id', '$comment', '$defs', 'definitions',\n // Extended properties used by some MCP servers (e.g., Apify)\n 'prefill', 'examples', 'enumTitles', 'enumDescriptions',\n // Other common extensions\n 'deprecated', 'readOnly', 'writeOnly', 'contentMediaType', 'contentEncoding',\n];\n\n/**\n * Cleans a JSON Schema by removing meta-properties that cause issues with\n * strict Pydantic validation (e.g., Google ADK, LangGraph).\n *\n * @param schema - The JSON Schema to clean\n * @returns Cleaned schema without forbidden properties\n */\nexport function cleanSchema(schema: Record<string, any> | undefined): Record<string, any> {\n if (!schema) {\n return { type: 'object', properties: {} };\n }\n\n const cleaned = { ...schema };\n\n // Remove all forbidden properties\n for (const prop of PYDANTIC_FORBIDDEN_PROPS) {\n delete cleaned[prop];\n }\n\n // Recursively clean nested properties\n if (cleaned.properties && typeof cleaned.properties === 'object') {\n const cleanedProps: Record<string, any> = {};\n for (const [key, value] of Object.entries(cleaned.properties)) {\n if (typeof value === 'object' && value !== null) {\n cleanedProps[key] = cleanSchema(value as Record<string, any>);\n } else {\n cleanedProps[key] = value;\n }\n }\n cleaned.properties = cleanedProps;\n }\n\n // Clean items if it's an array schema\n if (cleaned.items && typeof cleaned.items === 'object') {\n cleaned.items = cleanSchema(cleaned.items);\n }\n\n // Clean additionalProperties if it's an object schema\n if (cleaned.additionalProperties && typeof cleaned.additionalProperties === 'object') {\n cleaned.additionalProperties = cleanSchema(cleaned.additionalProperties);\n }\n\n return cleaned;\n}\n\n/**\n * Configuration options for AguiAdapter\n */\nexport interface AguiAdapterOptions {\n /**\n * Prefix for tool names to avoid collision with other tools.\n * @default serverId or 'mcp'\n */\n prefix?: string;\n\n /**\n * Optional ToolRouter for intelligent tool selection.\n */\n toolRouter?: ToolRouter;\n}\n\n/**\n * AG-UI Tool with handler for server-side execution.\n */\nexport interface AguiTool {\n name: string;\n description: string;\n parameters?: Record<string, any>;\n _meta?: Record<string, any>; // Add _meta to AguiTool\n handler?: (args: any) => any | Promise<any>;\n}\n\n/**\n * Tool definition format for passing to remote agents (without handler).\n */\nexport interface AguiToolDefinition {\n name: string;\n description: string;\n parameters: Record<string, any>;\n _meta?: Record<string, any>; // Add _meta to AguiToolDefinition\n}\n\n/**\n * Adapter that transforms MCP tools into AG-UI compatible formats.\n */\nexport class AguiAdapter {\n constructor(\n private client: MCPClient | MultiSessionClient,\n private options: AguiAdapterOptions = {}\n ) { }\n\n /**\n * Get tools with handlers for MCP tool execution.\n */\n async getTools(): Promise<AguiTool[]> {\n if (this.options.toolRouter) {\n return this.getToolsViaRouter(this.options.toolRouter);\n }\n\n if (this.isMultiSession()) {\n const clients = (this.client as MultiSessionClient).getClients();\n const allTools: AguiTool[] = [];\n for (const client of clients) {\n allTools.push(...await this.transformTools(client));\n }\n return allTools;\n }\n return this.transformTools(this.client as MCPClient);\n }\n\n /**\n * Get tool definitions in JSON Schema format for passing to remote agents.\n */\n async getToolDefinitions(): Promise<AguiToolDefinition[]> {\n if (this.options.toolRouter) {\n return this.getToolDefinitionsViaRouter(this.options.toolRouter);\n }\n\n if (this.isMultiSession()) {\n const clients = (this.client as MultiSessionClient).getClients();\n const allTools: AguiToolDefinition[] = [];\n for (const client of clients) {\n allTools.push(...await this.transformToolDefinitions(client));\n }\n return allTools;\n }\n return this.transformToolDefinitions(this.client as MCPClient);\n }\n\n /**\n * Get tools as a function (for dynamic loading).\n */\n getToolsFunction(): () => Promise<AguiTool[]> {\n return () => this.getTools();\n }\n\n private isMultiSession(): boolean {\n return typeof (this.client as any).getClients === 'function';\n }\n\n private async transformTools(client: MCPClient): Promise<AguiTool[]> {\n if (!client.isConnected()) return [];\n\n const result = await client.listTools();\n const serverId = (typeof (client as any).getServerId === 'function'\n ? (client as any).getServerId()\n : undefined) as string | undefined;\n const normalizedPrefix = this.options.prefix?.replace(/-/g, '') ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n const prefix = `tool_${normalizedPrefix}`;\n\n return result.tools.map(tool => {\n // Type assertion to access _meta if it exists on the tool object (it comes from MCP SDK)\n const mcpTool = tool as any;\n const mcpToolName = tool.name;\n return {\n name: `${prefix}_${tool.name}`,\n description: tool.description || `Execute ${tool.name}`,\n parameters: cleanSchema(tool.inputSchema),\n _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\n handler: async (args: any) => {\n // Call the actual MCP tool\n const callResult = await (client as any).callTool(mcpToolName, args ?? {});\n\n // Return the raw result object so middleware can inspect `_meta` (e.g. for UI triggers)\n return callResult;\n }\n }\n });\n }\n\n private async transformToolDefinitions(client: MCPClient): Promise<AguiToolDefinition[]> {\n if (!client.isConnected()) return [];\n\n const result = await client.listTools();\n const serverId = (typeof (client as any).getServerId === 'function'\n ? (client as any).getServerId()\n : undefined) as string | undefined;\n const normalizedPrefix = this.options.prefix?.replace(/-/g, '') ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n const prefix = `tool_${normalizedPrefix}`;\n\n return result.tools.map(tool => {\n const mcpTool = tool as any;\n return {\n name: `${prefix}_${tool.name}`,\n description: tool.description || `Execute ${tool.name}`,\n parameters: cleanSchema(tool.inputSchema),\n _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\n };\n });\n }\n\n /**\n * Build AG-UI tools from a ToolRouter's filtered output.\n *\n * In `search` strategy, only meta-tools are registered with the framework.\n * Real tool execution is proxied through `mcp_execute_tool` which uses\n * `router.callTool()` to route to the correct MCP client.\n */\n private async getToolsViaRouter(router: ToolRouter): Promise<AguiTool[]> {\n const filteredTools = await router.getFilteredTools();\n\n return filteredTools.map(tool => {\n const routedTool = tool as typeof tool & { sessionId?: string; serverName?: string };\n const namespace = routedTool.serverName ?? routedTool.sessionId;\n return {\n name: isMetaTool(tool.name)\n ? tool.name\n : this.getRouterToolKey(tool.name, routedTool.sessionId, routedTool.serverName),\n description: tool.description || `Execute ${tool.name}`,\n parameters: cleanSchema(tool.inputSchema),\n handler: async (args: any) => {\n if (isMetaTool(tool.name)) {\n const result = await executeMetaTool(\n tool.name,\n args,\n router,\n (name, toolArgs, namespace) => router.callTool(name, toolArgs, namespace)\n );\n if (result) {\n return result.content.map((c: any) => c.text ?? '').join('\\n');\n }\n return \"Failed to execute meta-tool\";\n }\n\n // For non-meta tools in 'all' or 'groups' strategy,\n // route directly to the correct MCP client\n return await router.callTool(tool.name, args, namespace);\n }\n };\n });\n }\n\n private async getToolDefinitionsViaRouter(router: ToolRouter): Promise<AguiToolDefinition[]> {\n const filteredTools = await router.getFilteredTools();\n return filteredTools.map(tool => {\n const routedTool = tool as typeof tool & { sessionId?: string; serverName?: string };\n return {\n name: isMetaTool(tool.name)\n ? tool.name\n : this.getRouterToolKey(tool.name, routedTool.sessionId, routedTool.serverName),\n description: tool.description || `Execute ${tool.name}`,\n parameters: cleanSchema(tool.inputSchema)\n };\n });\n }\n\n private getRouterToolKey(toolName: string, sessionId?: string, serverName?: string): string {\n const namespace = sessionId ?? serverName ?? 'mcp';\n const normalized = namespace\n .toLowerCase()\n .replace(/[^a-z0-9]+/g, '_')\n .replace(/^_+|_+$/g, '') || 'mcp';\n return `tool_${normalized}_${toolName}`;\n }\n}\n","/**\n * AG-UI Middleware for MCP Tool Execution\n *\n * This middleware intercepts tool calls from remote agents and executes\n * MCP tools server-side, returning results back to the agent.\n *\n * @requires @ag-ui/client - Peer dependency for AG-UI types\n * @requires rxjs - Uses RxJS Observables for event streaming\n */\n\nimport { Observable, Subscriber } from 'rxjs';\nimport {\n Middleware,\n EventType,\n type AbstractAgent,\n type RunAgentInput,\n type BaseEvent,\n type ToolCallEndEvent,\n type Tool,\n} from '@ag-ui/client';\nimport { type AguiTool, cleanSchema } from './agui-adapter.js';\n\n/** Tool execution result for continuation */\ninterface ToolResult {\n toolCallId: string;\n toolName: string;\n result: string;\n messageId: string;\n}\n\n/** State for tracking tool calls during a run */\ninterface RunState {\n toolCallArgsBuffer: Map<string, string>;\n toolCallNames: Map<string, string>;\n pendingMcpCalls: Set<string>;\n textContent?: string;\n error: boolean;\n}\n\n/**\n * Configuration for McpMiddleware\n */\nexport interface McpMiddlewareConfig {\n /** Pre-loaded tools with handlers (required) */\n tools: AguiTool[];\n}\n\n/**\n * AG-UI Middleware that executes MCP tools server-side.\n */\nexport class McpMiddleware extends Middleware {\n private tools: AguiTool[];\n private toolSchemas: Tool[];\n\n constructor(config: McpMiddlewareConfig) {\n super();\n this.tools = config.tools;\n this.toolSchemas = this.tools.map((t: AguiTool) => ({\n name: t.name,\n description: t.description,\n parameters: cleanSchema(t.parameters),\n }));\n }\n\n private isMcpTool(toolName: string): boolean {\n return this.tools.some(t => t.name === toolName);\n }\n\n private parseArgs(argsString: string): Record<string, any> {\n if (!argsString?.trim()) return {};\n\n try {\n return JSON.parse(argsString);\n } catch {\n // Handle duplicated JSON from streaming issues: {...}{...}\n const trimmed = argsString.trim();\n if (trimmed.includes('}{')) {\n const firstObject = trimmed.slice(0, trimmed.indexOf('}{') + 1);\n try {\n return JSON.parse(firstObject);\n } catch {\n console.error(`[McpMiddleware] Failed to parse JSON:`, firstObject);\n }\n }\n console.error(`[McpMiddleware] Failed to parse args:`, argsString);\n return {};\n }\n }\n\n private async executeTool(toolName: string, args: Record<string, any>): Promise<string> {\n const tool = this.tools.find(t => t.name === toolName);\n if (!tool?.handler) {\n return `Error: Tool ${tool ? 'has no handler' : 'not found'}: ${toolName}`;\n }\n\n try {\n console.log(`[McpMiddleware] Executing tool: ${toolName}`, args);\n const result = await tool.handler(args);\n const resultStr = typeof result === 'string' ? result : JSON.stringify(result);\n\n console.log(`[McpMiddleware] Tool result:`, resultStr.slice(0, 200));\n return resultStr;\n } catch (error: any) {\n console.error(`[McpMiddleware] Error executing tool:`, error);\n return `Error: ${error.message || String(error)}`;\n }\n }\n\n private generateId(prefix: string): string {\n return `${prefix}_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;\n }\n\n private ensureIds(input: RunAgentInput): void {\n const anyInput = input as any;\n if (!anyInput.threadId) anyInput.threadId = this.generateId('mcp_thread');\n if (!anyInput.runId) anyInput.runId = this.generateId('mcp_run');\n }\n\n /** Process tool call events and update state */\n private handleToolCallEvent(event: BaseEvent, state: RunState): void {\n const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\n\n // Accumulate text content for reconstruction\n if (event.type === EventType.TEXT_MESSAGE_CHUNK) {\n const e = event as any;\n if (e.delta) {\n state.textContent = (state.textContent || '') + e.delta;\n }\n }\n\n if (event.type === EventType.TOOL_CALL_START) {\n const e = event as any;\n if (e.toolCallId && e.toolCallName) {\n toolCallNames.set(e.toolCallId, e.toolCallName);\n if (this.isMcpTool(e.toolCallName)) {\n pendingMcpCalls.add(e.toolCallId);\n }\n console.log(`[McpMiddleware] TOOL_CALL_START: ${e.toolCallName} (id: ${e.toolCallId}, isMCP: ${this.isMcpTool(e.toolCallName)})`);\n }\n }\n\n if (event.type === EventType.TOOL_CALL_ARGS) {\n const e = event as any;\n if (e.toolCallId && e.delta) {\n const existing = toolCallArgsBuffer.get(e.toolCallId) || '';\n toolCallArgsBuffer.set(e.toolCallId, existing + e.delta);\n }\n }\n\n if (event.type === EventType.TOOL_CALL_END) {\n const e = event as ToolCallEndEvent;\n console.log(`[McpMiddleware] TOOL_CALL_END: ${toolCallNames.get(e.toolCallId) ?? 'unknown'} (id: ${e.toolCallId})`);\n }\n\n // Workaround: Extract parallel tool calls from MESSAGES_SNAPSHOT\n if (event.type === EventType.MESSAGES_SNAPSHOT) {\n const messages = (event as any).messages || [];\n if (messages.length > 0) {\n const lastMsg = messages[messages.length - 1];\n // Update text content from snapshot if available (often more reliable)\n if (lastMsg.role === 'assistant' && lastMsg.content) {\n state.textContent = lastMsg.content;\n }\n\n // Discover tools\n for (let i = messages.length - 1; i >= 0; i--) {\n const msg = messages[i];\n const tools = Array.isArray(msg.toolCalls) ? msg.toolCalls :\n (Array.isArray(msg.tool_calls) ? msg.tool_calls : []);\n\n if (msg.role === 'assistant' && tools.length > 0) {\n for (const tc of tools) {\n if (tc.id && tc.function?.name && !toolCallNames.has(tc.id)) {\n toolCallNames.set(tc.id, tc.function.name);\n toolCallArgsBuffer.set(tc.id, tc.function.arguments || '{}');\n if (this.isMcpTool(tc.function.name)) {\n pendingMcpCalls.add(tc.id);\n console.log(`[McpMiddleware] MESSAGES_SNAPSHOT: Discovered ${tc.function.name} (id: ${tc.id})`);\n }\n }\n }\n break;\n }\n }\n }\n }\n }\n\n /** Execute pending MCP tools and return results */\n private async executeTools(state: RunState): Promise<ToolResult[]> {\n const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\n const results: ToolResult[] = [];\n\n const promises = [...pendingMcpCalls].map(async (toolCallId) => {\n const toolName = toolCallNames.get(toolCallId);\n if (!toolName) return;\n\n const args = this.parseArgs(toolCallArgsBuffer.get(toolCallId) || '{}');\n console.log(`[McpMiddleware] Executing pending tool: ${toolName}`);\n\n const result = await this.executeTool(toolName, args);\n results.push({\n toolCallId,\n toolName,\n result,\n messageId: this.generateId('mcp_result'),\n });\n pendingMcpCalls.delete(toolCallId);\n });\n\n await Promise.all(promises);\n return results;\n }\n\n /** Emit tool results (without RUN_FINISHED - that's emitted when truly done) */\n private emitToolResults(observer: Subscriber<BaseEvent>, results: ToolResult[]): void {\n for (const { toolCallId, toolName, result, messageId } of results) {\n observer.next({\n type: EventType.TOOL_CALL_RESULT,\n toolCallId,\n messageId,\n content: result,\n role: 'tool',\n timestamp: Date.now(),\n } as any);\n console.log(`[McpMiddleware] Emitting TOOL_CALL_RESULT for: ${toolName}`);\n }\n }\n\n run(input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> {\n return new Observable<BaseEvent>((observer: Subscriber<BaseEvent>) => {\n const state: RunState = {\n toolCallArgsBuffer: new Map(),\n toolCallNames: new Map(),\n pendingMcpCalls: new Set(),\n textContent: '',\n error: false,\n };\n\n this.ensureIds(input);\n const anyInput = input as any;\n\n console.log(`[McpMiddleware] === NEW RUN ===`);\n console.log(`[McpMiddleware] threadId: ${anyInput.threadId}, runId: ${anyInput.runId}`);\n console.log(`[McpMiddleware] messages: ${input.messages?.length ?? 0}, tools: ${this.tools?.length ?? 0}`);\n\n // Inject MCP tools\n if (this.toolSchemas?.length) {\n input.tools = [...(input.tools || []), ...this.toolSchemas];\n console.log(`[McpMiddleware] Injected ${this.toolSchemas.length} tools:`, this.toolSchemas.map((t: Tool) => t.name));\n }\n\n const handleRunFinished = async () => {\n if (state.error) return; // Don't continue after error\n\n if (state.pendingMcpCalls.size === 0) {\n observer.next({\n type: EventType.RUN_FINISHED,\n threadId: anyInput.threadId,\n runId: anyInput.runId,\n timestamp: Date.now(),\n } as any);\n observer.complete();\n return;\n }\n\n console.log(`[McpMiddleware] RUN_FINISHED with ${state.pendingMcpCalls.size} pending calls`);\n\n // Reconstruct the Assistant Message that triggered these tools\n const toolCalls = [];\n for (const toolCallId of state.pendingMcpCalls) {\n const name = state.toolCallNames.get(toolCallId);\n const args = state.toolCallArgsBuffer.get(toolCallId) || '{}';\n if (name) {\n toolCalls.push({\n id: toolCallId,\n type: 'function',\n function: { name, arguments: args }\n });\n }\n }\n\n // Add the Assistant Message to history FIRST\n if (toolCalls.length > 0 || state.textContent) {\n const assistantMsg = {\n id: this.generateId('msg_ast'),\n role: 'assistant',\n content: state.textContent || null, // Ensure null if empty string for strict LLMs\n tool_calls: toolCalls.length > 0 ? toolCalls : undefined\n };\n input.messages.push(assistantMsg as any);\n console.log(`[McpMiddleware] Added assistant message to history before tools: ${state.textContent?.slice(0, 50)}... [${toolCalls.length} tools]`);\n }\n\n // Execute tools and emit results (no RUN_FINISHED yet - continuation follows)\n const results = await this.executeTools(state);\n this.emitToolResults(observer, results);\n\n // Prepare continuation\n console.log(`[McpMiddleware] Triggering continuation with ${results.length} results`);\n\n // Add tool result messages to history\n for (const { toolCallId, result, messageId } of results) {\n input.messages.push({\n id: messageId,\n role: 'tool',\n tool_call_id: toolCallId,\n content: result,\n } as any);\n }\n\n // Reset state for next turn\n state.toolCallArgsBuffer.clear();\n state.toolCallNames.clear();\n state.textContent = ''; // Clear text content for next turn\n\n anyInput.runId = this.generateId('mcp_run');\n console.log(`[McpMiddleware] === CONTINUATION RUN === messages: ${input.messages.length}`);\n\n // Subscribe to continuation\n next.run(input).subscribe({\n next: (event) => {\n if (state.error) return;\n\n this.handleToolCallEvent(event, state);\n\n if (event.type === EventType.RUN_ERROR) {\n console.log(`[McpMiddleware] RUN_ERROR received in continuation`);\n state.error = true;\n observer.next(event);\n observer.complete();\n return;\n }\n\n if (event.type === EventType.RUN_STARTED) {\n console.log(`[McpMiddleware] Filtering RUN_STARTED from continuation`);\n return;\n }\n\n if (event.type === EventType.RUN_FINISHED) {\n if (state.pendingMcpCalls.size > 0) {\n handleRunFinished();\n } else {\n observer.next(event);\n observer.complete();\n }\n return;\n }\n observer.next(event);\n },\n error: (err) => {\n state.error = true;\n observer.error(err);\n },\n complete: () => {\n if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\n },\n });\n };\n\n const subscription = next.run(input).subscribe({\n next: (event) => {\n if (state.error) return;\n\n this.handleToolCallEvent(event, state);\n\n if (event.type === EventType.RUN_ERROR) {\n console.log(`[McpMiddleware] RUN_ERROR received`);\n state.error = true;\n observer.next(event);\n observer.complete();\n return;\n }\n\n if (event.type === EventType.RUN_FINISHED) {\n handleRunFinished();\n return;\n }\n observer.next(event);\n },\n error: (err) => {\n state.error = true;\n observer.error(err);\n },\n complete: () => {\n if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\n },\n });\n\n return () => subscription.unsubscribe();\n });\n }\n}\n\n/**\n * Factory function to create MCP middleware.\n */\nexport function createMcpMiddleware(\n options: { tools: AguiTool[] }\n) {\n const middleware = new McpMiddleware(options);\n return (input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> => {\n return middleware.run(input, next);\n };\n}\n\n// Legacy exports\nexport { McpMiddleware as McpToolExecutorMiddleware };\nexport { createMcpMiddleware as createMcpToolMiddleware };\n\n// Re-exports\nexport { Middleware, EventType };\nexport type { RunAgentInput, BaseEvent, AbstractAgent, ToolCallEndEvent, Tool };"]}
@@ -1,9 +1,11 @@
1
1
  import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-CQsRbxYI.mjs';
2
2
  import { ToolSet } from 'ai';
3
+ import { d as ToolRouter } from '../tool-router-XnWVxPzv.mjs';
3
4
  import '../events-CK3N--3g.mjs';
4
5
  import '@modelcontextprotocol/sdk/types.js';
5
6
  import '@modelcontextprotocol/sdk/client/auth.js';
6
7
  import '@modelcontextprotocol/sdk/shared/auth.js';
8
+ import '../types-CfCoIsWI.mjs';
7
9
 
8
10
  interface AIAdapterOptions {
9
11
  /**
@@ -11,6 +13,16 @@ interface AIAdapterOptions {
11
13
  * Defaults to the client's serverId.
12
14
  */
13
15
  prefix?: string;
16
+ /**
17
+ * Optional ToolRouter for intelligent tool selection.
18
+ *
19
+ * When provided with `strategy: 'search'`, the adapter exposes only
20
+ * meta-tools (search_tools, get_tool_schema) instead of all tool schemas,
21
+ * reducing context window usage by 80–95%.
22
+ *
23
+ * When not provided, all tools are returned as before (backward-compatible).
24
+ */
25
+ toolRouter?: ToolRouter;
14
26
  }
15
27
  /**
16
28
  * Adapter to use MCP tools with the Vercel AI SDK.
@@ -29,6 +41,15 @@ declare class AIAdapter {
29
41
  * Fetches tools from the client(s) and converts them to AI SDK tools.
30
42
  */
31
43
  getTools(): Promise<ToolSet>;
44
+ /**
45
+ * Build a ToolSet from a ToolRouter's filtered output.
46
+ *
47
+ * In `search` strategy, only meta-tools are registered with the framework.
48
+ * Real tool execution is proxied through `mcp_execute_tool` which uses
49
+ * `router.callTool()` to route to the correct MCP client.
50
+ */
51
+ private getToolsViaRouter;
52
+ private getRouterToolKey;
32
53
  /**
33
54
  * Convenience static method to fetch tools in a single line.
34
55
  */
@@ -1,9 +1,11 @@
1
1
  import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-CHE8QpVE.js';
2
2
  import { ToolSet } from 'ai';
3
+ import { d as ToolRouter } from '../tool-router-Bo8qZbsD.js';
3
4
  import '../events-CK3N--3g.js';
4
5
  import '@modelcontextprotocol/sdk/types.js';
5
6
  import '@modelcontextprotocol/sdk/client/auth.js';
6
7
  import '@modelcontextprotocol/sdk/shared/auth.js';
8
+ import '../types-CfCoIsWI.js';
7
9
 
8
10
  interface AIAdapterOptions {
9
11
  /**
@@ -11,6 +13,16 @@ interface AIAdapterOptions {
11
13
  * Defaults to the client's serverId.
12
14
  */
13
15
  prefix?: string;
16
+ /**
17
+ * Optional ToolRouter for intelligent tool selection.
18
+ *
19
+ * When provided with `strategy: 'search'`, the adapter exposes only
20
+ * meta-tools (search_tools, get_tool_schema) instead of all tool schemas,
21
+ * reducing context window usage by 80–95%.
22
+ *
23
+ * When not provided, all tools are returned as before (backward-compatible).
24
+ */
25
+ toolRouter?: ToolRouter;
14
26
  }
15
27
  /**
16
28
  * Adapter to use MCP tools with the Vercel AI SDK.
@@ -29,6 +41,15 @@ declare class AIAdapter {
29
41
  * Fetches tools from the client(s) and converts them to AI SDK tools.
30
42
  */
31
43
  getTools(): Promise<ToolSet>;
44
+ /**
45
+ * Build a ToolSet from a ToolRouter's filtered output.
46
+ *
47
+ * In `search` strategy, only meta-tools are registered with the framework.
48
+ * Real tool execution is proxied through `mcp_execute_tool` which uses
49
+ * `router.callTool()` to route to the correct MCP client.
50
+ */
51
+ private getToolsViaRouter;
52
+ private getRouterToolKey;
32
53
  /**
33
54
  * Convenience static method to fetch tools in a single line.
34
55
  */
@@ -4,6 +4,135 @@ var __defProp = Object.defineProperty;
4
4
  var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
5
5
  var __publicField = (obj, key, value) => __defNormalProp(obj, key + "" , value);
6
6
 
7
+ // src/shared/meta-tools.ts
8
+ async function executeMetaTool(toolName, args, router, callToolFn) {
9
+ const resolveToolSchema = (name, namespace) => {
10
+ try {
11
+ return { tool: router.getToolSchema(name, namespace) };
12
+ } catch (err) {
13
+ const errorMessage = err instanceof Error ? err.message : String(err);
14
+ return {
15
+ error: {
16
+ content: [{ type: "text", text: errorMessage }],
17
+ isError: true
18
+ }
19
+ };
20
+ }
21
+ };
22
+ switch (toolName) {
23
+ case "mcp_search_tool_bm25": {
24
+ const query = String(args.query ?? "");
25
+ const limit = Math.min(Number(args.limit) || 5, 20);
26
+ const results = await router.searchTools(query, limit);
27
+ const text = results.length === 0 ? "No tools found matching your query. Try different keywords." : results.map(
28
+ (t, i) => `${i + 1}. **${t.name}** (server: ${t.serverName})
29
+ ${t.description}
30
+ Estimated tokens: ${t.estimatedTokens}`
31
+ ).join("\n");
32
+ return {
33
+ content: [{ type: "text", text }],
34
+ isError: false
35
+ };
36
+ }
37
+ case "mcp_search_tool_regex": {
38
+ const pattern = String(args.query ?? "");
39
+ const limit = Math.min(Number(args.limit) || 5, 20);
40
+ const results = await router.searchToolsRegex(pattern, limit);
41
+ const text = results.length === 0 ? "No tools matched your regex pattern. Try a broader pattern." : results.map(
42
+ (t, i) => `${i + 1}. **${t.name}** (server: ${t.serverName})
43
+ ${t.description}
44
+ Estimated tokens: ${t.estimatedTokens}`
45
+ ).join("\n");
46
+ return {
47
+ content: [{ type: "text", text }],
48
+ isError: false
49
+ };
50
+ }
51
+ case "mcp_get_tool_schema": {
52
+ const name = String(args.toolName ?? "");
53
+ const namespace = String(args.serverName ?? "") || void 0;
54
+ const { tool, error } = resolveToolSchema(name, namespace);
55
+ if (error) {
56
+ return error;
57
+ }
58
+ if (!tool) {
59
+ return {
60
+ content: [
61
+ {
62
+ type: "text",
63
+ text: `Tool "${name}" not found. Use mcp_search_tool_bm25 to find available tools first.`
64
+ }
65
+ ],
66
+ isError: true
67
+ };
68
+ }
69
+ const schema = {
70
+ name: tool.name,
71
+ description: tool.description,
72
+ inputSchema: tool.inputSchema
73
+ };
74
+ return {
75
+ content: [{ type: "text", text: JSON.stringify(schema, null, 2) }],
76
+ isError: false
77
+ };
78
+ }
79
+ case "mcp_execute_tool": {
80
+ const targetToolName = String(args.toolName ?? "");
81
+ const namespace = String(args.serverName ?? "") || void 0;
82
+ const toolArgs = args.args ?? {};
83
+ if (!targetToolName) {
84
+ return {
85
+ content: [{ type: "text", text: 'Missing required parameter "toolName". Specify which tool to execute.' }],
86
+ isError: true
87
+ };
88
+ }
89
+ const { tool, error } = resolveToolSchema(targetToolName, namespace);
90
+ if (error) {
91
+ return error;
92
+ }
93
+ if (!tool) {
94
+ return {
95
+ content: [
96
+ {
97
+ type: "text",
98
+ text: `Tool "${targetToolName}" not found. Use mcp_search_tool_bm25 to discover available tools first.`
99
+ }
100
+ ],
101
+ isError: true
102
+ };
103
+ }
104
+ if (!callToolFn) {
105
+ return {
106
+ content: [{ type: "text", text: "Tool execution is not available. No callToolFn was configured." }],
107
+ isError: true
108
+ };
109
+ }
110
+ try {
111
+ const result = await callToolFn(targetToolName, toolArgs, namespace);
112
+ if (result && typeof result === "object" && "content" in result) {
113
+ return result;
114
+ }
115
+ const text = typeof result === "string" ? result : JSON.stringify(result, null, 2);
116
+ return {
117
+ content: [{ type: "text", text }],
118
+ isError: false
119
+ };
120
+ } catch (err) {
121
+ const errorMessage = err instanceof Error ? err.message : String(err);
122
+ return {
123
+ content: [{ type: "text", text: `Tool execution failed: ${errorMessage}` }],
124
+ isError: true
125
+ };
126
+ }
127
+ }
128
+ default:
129
+ return null;
130
+ }
131
+ }
132
+ function isMetaTool(toolName) {
133
+ return toolName === "mcp_search_tool_bm25" || toolName === "mcp_search_tool_regex" || toolName === "mcp_get_tool_schema" || toolName === "mcp_execute_tool";
134
+ }
135
+
7
136
  // src/adapters/ai-adapter.ts
8
137
  var AIAdapter = class _AIAdapter {
9
138
  constructor(client, options = {}) {
@@ -54,6 +183,9 @@ var AIAdapter = class _AIAdapter {
54
183
  */
55
184
  async getTools() {
56
185
  await this.ensureJsonSchema();
186
+ if (this.options.toolRouter) {
187
+ return this.getToolsViaRouter(this.options.toolRouter);
188
+ }
57
189
  const isMultiSession = typeof this.client.getClients === "function";
58
190
  const clients = isMultiSession ? this.client.getClients() : [this.client];
59
191
  const results = await Promise.all(
@@ -69,6 +201,49 @@ var AIAdapter = class _AIAdapter {
69
201
  );
70
202
  return results.reduce((acc, tools) => ({ ...acc, ...tools }), {});
71
203
  }
204
+ /**
205
+ * Build a ToolSet from a ToolRouter's filtered output.
206
+ *
207
+ * In `search` strategy, only meta-tools are registered with the framework.
208
+ * Real tool execution is proxied through `mcp_execute_tool` which uses
209
+ * `router.callTool()` to route to the correct MCP client.
210
+ */
211
+ async getToolsViaRouter(router) {
212
+ const filteredTools = await router.getFilteredTools();
213
+ return Object.fromEntries(
214
+ filteredTools.map((tool) => {
215
+ const routedTool = tool;
216
+ const namespace = routedTool.serverName ?? routedTool.sessionId;
217
+ const toolKey = isMetaTool(tool.name) ? tool.name : this.getRouterToolKey(tool.name, routedTool.sessionId, routedTool.serverName);
218
+ return [
219
+ toolKey,
220
+ {
221
+ description: tool.description,
222
+ inputSchema: this.jsonSchema(tool.inputSchema),
223
+ execute: async (args) => {
224
+ if (isMetaTool(tool.name)) {
225
+ const result = await executeMetaTool(
226
+ tool.name,
227
+ args,
228
+ router,
229
+ (name, toolArgs, targetNamespace) => router.callTool(name, toolArgs, targetNamespace)
230
+ );
231
+ if (result) {
232
+ return result;
233
+ }
234
+ }
235
+ return await router.callTool(tool.name, args, namespace);
236
+ }
237
+ }
238
+ ];
239
+ })
240
+ );
241
+ }
242
+ getRouterToolKey(toolName, sessionId, serverName) {
243
+ const namespace = sessionId ?? serverName ?? "mcp";
244
+ const normalized = namespace.toLowerCase().replace(/[^a-z0-9]+/g, "_").replace(/^_+|_+$/g, "") || "mcp";
245
+ return `tool_${normalized}_${toolName}`;
246
+ }
72
247
  /**
73
248
  * Convenience static method to fetch tools in a single line.
74
249
  */
@@ -1 +1 @@
1
- {"version":3,"sources":["../../src/adapters/ai-adapter.ts"],"names":[],"mappings":";;;;;;;AAgBO,IAAM,SAAA,GAAN,MAAM,UAAA,CAAU;AAAA,EAGnB,WAAA,CACY,MAAA,EACA,OAAA,GAA4B,EAAC,EACvC;AAFU,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA;AACA,IAAA,IAAA,CAAA,OAAA,GAAA,OAAA;AAJZ,IAAA,aAAA,CAAA,IAAA,EAAQ,YAAA,CAAA;AAAA,EAKJ;AAAA;AAAA;AAAA;AAAA,EAOJ,MAAc,gBAAA,GAAmB;AAC7B,IAAA,IAAI,CAAC,KAAK,UAAA,EAAY;AAClB,MAAA,MAAM,EAAE,UAAA,EAAW,GAAI,MAAM,OAAO,IAAI,CAAA;AACxC,MAAA,IAAA,CAAK,UAAA,GAAa,UAAA;AAAA,IACtB;AAAA,EACJ;AAAA,EAEA,MAAc,eAAe,MAAA,EAAqC;AAE9D,IAAA,MAAM,cAAc,OAAO,MAAA,CAAO,gBAAgB,UAAA,GAC5C,MAAA,CAAO,aAAY,GACnB,KAAA;AAEN,IAAA,IAAI,CAAC,WAAA,EAAa;AACd,MAAA,OAAO,EAAC;AAAA,IACZ;AAEA,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,SAAA,EAAU;AAGtC,IAAA,OAAO,MAAA,CAAO,WAAA;AAAA,MACV,MAAA,CAAO,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,KAAS;AAEvB,QAAA,MAAM,WAAW,OAAO,MAAA,CAAO,gBAAgB,UAAA,GACzC,MAAA,CAAO,aAAY,GACnB,MAAA;AACN,QAAA,MAAM,MAAA,GAAS,IAAA,CAAK,OAAA,CAAQ,MAAA,IAAU,QAAA,EAAU,OAAA,CAAQ,IAAA,EAAM,EAAE,CAAA,CAAE,SAAA,CAAU,CAAA,EAAG,CAAC,CAAA,IAAK,KAAA;AACrF,QAAA,OAAO;AAAA,UACH,CAAA,KAAA,EAAQ,MAAM,CAAA,CAAA,EAAI,IAAA,CAAK,IAAI,CAAA,CAAA;AAAA,UAC3B;AAAA,YACI,aAAa,IAAA,CAAK,WAAA;AAAA,YAClB,WAAA,EAAa,IAAA,CAAK,UAAA,CAAY,IAAA,CAAK,WAA0B,CAAA;AAAA,YAC7D,OAAA,EAAS,OAAO,IAAA,KAAc;AAC1B,cAAA,IAAI;AACA,gBAAA,MAAM,WAAW,MAAM,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,MAAM,IAAI,CAAA;AACtD,gBAAA,OAAO,QAAA;AAAA,cACX,SAAS,KAAA,EAAO;AACZ,gBAAA,MAAM,eAAe,KAAA,YAAiB,KAAA,GAAQ,KAAA,CAAM,OAAA,GAAU,OAAO,KAAK,CAAA;AAC1E,gBAAA,MAAM,IAAI,KAAA,CAAM,CAAA,uBAAA,EAA0B,YAAY,CAAA,CAAE,CAAA;AAAA,cAC5D;AAAA,YACJ;AAAA;AACJ,SACJ;AAAA,MACJ,CAAC;AAAA,KACL;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,QAAA,GAA6B;AAC/B,IAAA,MAAM,KAAK,gBAAA,EAAiB;AAI5B,IAAA,MAAM,cAAA,GAAiB,OAAQ,IAAA,CAAK,MAAA,CAAe,UAAA,KAAe,UAAA;AAClE,IAAA,MAAM,OAAA,GAAU,iBACT,IAAA,CAAK,MAAA,CAA8B,YAAW,GAC/C,CAAC,KAAK,MAAmB,CAAA;AAE/B,IAAA,MAAM,OAAA,GAAU,MAAM,OAAA,CAAQ,GAAA;AAAA,MAC1B,OAAA,CAAQ,GAAA,CAAI,OAAO,MAAA,KAAW;AAC1B,QAAA,IAAI;AACA,UAAA,OAAO,MAAM,IAAA,CAAK,cAAA,CAAe,MAAM,CAAA;AAAA,QAC3C,SAAS,KAAA,EAAO;AAGZ,UAAA,MAAM,QAAA,GAAW,OAAO,MAAA,CAAO,WAAA,KAAgB,aACzC,MAAA,CAAO,WAAA,MAAiB,SAAA,GACxB,SAAA;AACN,UAAA,OAAA,CAAQ,KAAA,CAAM,CAAA,uCAAA,EAA0C,QAAQ,CAAA,CAAA,CAAA,EAAK,KAAK,CAAA;AAC1E,UAAA,OAAO,EAAC;AAAA,QACZ;AAAA,MACJ,CAAC;AAAA,KACL;AAEA,IAAA,OAAO,OAAA,CAAQ,MAAA,CAAO,CAAC,GAAA,EAAK,KAAA,MAAW,EAAE,GAAG,GAAA,EAAK,GAAG,KAAA,EAAM,CAAA,EAAI,EAAE,CAAA;AAAA,EACpE;AAAA;AAAA;AAAA;AAAA,EAKA,aAAa,QAAA,CAAS,MAAA,EAAwC,OAAA,GAA4B,EAAC,EAAqB;AAC5G,IAAA,OAAO,IAAI,UAAA,CAAU,MAAA,EAAQ,OAAO,EAAE,QAAA,EAAS;AAAA,EACnD;AACJ","file":"ai-adapter.js","sourcesContent":["import { MCPClient } from '../server/mcp/oauth-client';\nimport { MultiSessionClient } from '../server/mcp/multi-session-client';\nimport type { JSONSchema7 } from 'json-schema';\nimport type { ToolSet } from 'ai';\n\nexport interface AIAdapterOptions {\n /** \n * Prefix for tool names to avoid collision with other tools.\n * Defaults to the client's serverId.\n */\n prefix?: string;\n}\n\n/**\n * Adapter to use MCP tools with the Vercel AI SDK.\n */\nexport class AIAdapter {\n private jsonSchema: typeof import('ai').jsonSchema | undefined;\n\n constructor(\n private client: MCPClient | MultiSessionClient,\n private options: AIAdapterOptions = {}\n ) { }\n\n\n\n /**\n * Lazy-loads the jsonSchema function from the AI SDK.\n */\n private async ensureJsonSchema() {\n if (!this.jsonSchema) {\n const { jsonSchema } = await import('ai');\n this.jsonSchema = jsonSchema;\n }\n }\n\n private async transformTools(client: MCPClient): Promise<ToolSet> {\n // Safe check for isConnected method (duck typing for bundler compatibility)\n const isConnected = typeof client.isConnected === 'function'\n ? client.isConnected()\n : false;\n\n if (!isConnected) {\n return {};\n }\n\n const result = await client.listTools();\n\n // @ts-ignore: ToolSet type inference can be tricky with dynamic imports\n return Object.fromEntries(\n result.tools.map((tool) => {\n // Safe access to getServerId\n const serverId = typeof client.getServerId === 'function'\n ? client.getServerId()\n : undefined;\n const prefix = this.options.prefix ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n return [\n `tool_${prefix}_${tool.name}`,\n {\n description: tool.description,\n inputSchema: this.jsonSchema!(tool.inputSchema as JSONSchema7),\n execute: async (args: any) => {\n try {\n const response = await client.callTool(tool.name, args);\n return response;\n } catch (error) {\n const errorMessage = error instanceof Error ? error.message : String(error);\n throw new Error(`Tool execution failed: ${errorMessage}`);\n }\n }\n }\n ];\n })\n );\n }\n\n /**\n * Fetches tools from the client(s) and converts them to AI SDK tools.\n */\n async getTools(): Promise<ToolSet> {\n await this.ensureJsonSchema();\n\n // Use duck typing instead of instanceof to handle module bundling issues\n // MultiSessionClient has getClients(), MCPClient does not\n const isMultiSession = typeof (this.client as any).getClients === 'function';\n const clients = isMultiSession\n ? (this.client as MultiSessionClient).getClients()\n : [this.client as MCPClient];\n\n const results = await Promise.all(\n clients.map(async (client) => {\n try {\n return await this.transformTools(client);\n } catch (error) {\n // For multi-client, we log and continue.\n // This is safer than throwing.\n const serverId = typeof client.getServerId === 'function'\n ? client.getServerId() ?? 'unknown'\n : 'unknown';\n console.error(`[AIAdapter] Failed to fetch tools from ${serverId}:`, error);\n return {};\n }\n })\n );\n\n return results.reduce((acc, tools) => ({ ...acc, ...tools }), {});\n }\n\n /**\n * Convenience static method to fetch tools in a single line.\n */\n static async getTools(client: MCPClient | MultiSessionClient, options: AIAdapterOptions = {}): Promise<ToolSet> {\n return new AIAdapter(client, options).getTools();\n }\n}\n"]}
1
+ {"version":3,"sources":["../../src/shared/meta-tools.ts","../../src/adapters/ai-adapter.ts"],"names":[],"mappings":";;;;;;;AAuLA,eAAsB,eAAA,CACpB,QAAA,EACA,IAAA,EACA,MAAA,EACA,UAAA,EACgC;AAChC,EAAA,MAAM,iBAAA,GAAoB,CAAC,IAAA,EAAc,SAAA,KAAgE;AACvG,IAAA,IAAI;AACF,MAAA,OAAO,EAAE,IAAA,EAAM,MAAA,CAAO,aAAA,CAAc,IAAA,EAAM,SAAS,CAAA,EAAE;AAAA,IACvD,SAAS,GAAA,EAAK;AACZ,MAAA,MAAM,eAAe,GAAA,YAAe,KAAA,GAAQ,GAAA,CAAI,OAAA,GAAU,OAAO,GAAG,CAAA;AACpE,MAAA,OAAO;AAAA,QACL,KAAA,EAAO;AAAA,UACL,SAAS,CAAC,EAAE,MAAM,MAAA,EAAQ,IAAA,EAAM,cAAc,CAAA;AAAA,UAC9C,OAAA,EAAS;AAAA;AACX,OACF;AAAA,IACF;AAAA,EACF,CAAA;AAEA,EAAA,QAAQ,QAAA;AAAU,IAChB,KAAK,sBAAA,EAAwB;AAC3B,MAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,IAAA,CAAK,KAAA,IAAS,EAAE,CAAA;AACrC,MAAA,MAAM,KAAA,GAAQ,KAAK,GAAA,CAAI,MAAA,CAAO,KAAK,KAAK,CAAA,IAAK,GAAG,EAAE,CAAA;AAElD,MAAA,MAAM,OAAA,GAAU,MAAM,MAAA,CAAO,WAAA,CAAY,OAAO,KAAK,CAAA;AAErD,MAAA,MAAM,IAAA,GAAO,OAAA,CAAQ,MAAA,KAAW,CAAA,GAC5B,gEACA,OAAA,CACG,GAAA;AAAA,QACC,CAAC,CAAA,EAAG,CAAA,KACF,CAAA,EAAG,CAAA,GAAI,CAAC,CAAA,IAAA,EAAO,CAAA,CAAE,IAAI,CAAA,YAAA,EAAe,CAAA,CAAE,UAAU,CAAA;AAAA,GAAA,EAC1C,EAAE,WAAW;AAAA,qBAAA,EACK,EAAE,eAAe,CAAA;AAAA,OAC7C,CACC,KAAK,IAAI,CAAA;AAEhB,MAAA,OAAO;AAAA,QACL,SAAS,CAAC,EAAE,IAAA,EAAM,MAAA,EAAQ,MAAM,CAAA;AAAA,QAChC,OAAA,EAAS;AAAA,OACX;AAAA,IACF;AAAA,IAEA,KAAK,uBAAA,EAAyB;AAC5B,MAAA,MAAM,OAAA,GAAU,MAAA,CAAO,IAAA,CAAK,KAAA,IAAS,EAAE,CAAA;AACvC,MAAA,MAAM,KAAA,GAAQ,KAAK,GAAA,CAAI,MAAA,CAAO,KAAK,KAAK,CAAA,IAAK,GAAG,EAAE,CAAA;AAElD,MAAA,MAAM,OAAA,GAAU,MAAM,MAAA,CAAO,gBAAA,CAAiB,SAAS,KAAK,CAAA;AAE5D,MAAA,MAAM,IAAA,GAAO,OAAA,CAAQ,MAAA,KAAW,CAAA,GAC5B,gEACA,OAAA,CACG,GAAA;AAAA,QACC,CAAC,CAAA,EAAG,CAAA,KACF,CAAA,EAAG,CAAA,GAAI,CAAC,CAAA,IAAA,EAAO,CAAA,CAAE,IAAI,CAAA,YAAA,EAAe,CAAA,CAAE,UAAU,CAAA;AAAA,GAAA,EAC1C,EAAE,WAAW;AAAA,qBAAA,EACK,EAAE,eAAe,CAAA;AAAA,OAC7C,CACC,KAAK,IAAI,CAAA;AAEhB,MAAA,OAAO;AAAA,QACL,SAAS,CAAC,EAAE,IAAA,EAAM,MAAA,EAAQ,MAAM,CAAA;AAAA,QAChC,OAAA,EAAS;AAAA,OACX;AAAA,IACF;AAAA,IAEA,KAAK,qBAAA,EAAuB;AAC1B,MAAA,MAAM,IAAA,GAAO,MAAA,CAAO,IAAA,CAAK,QAAA,IAAY,EAAE,CAAA;AACvC,MAAA,MAAM,SAAA,GAAY,MAAA,CAAO,IAAA,CAAK,UAAA,IAAc,EAAE,CAAA,IAAK,MAAA;AACnD,MAAA,MAAM,EAAE,IAAA,EAAM,KAAA,EAAM,GAAI,iBAAA,CAAkB,MAAM,SAAS,CAAA;AAEzD,MAAA,IAAI,KAAA,EAAO;AACT,QAAA,OAAO,KAAA;AAAA,MACT;AAEA,MAAA,IAAI,CAAC,IAAA,EAAM;AACT,QAAA,OAAO;AAAA,UACL,OAAA,EAAS;AAAA,YACP;AAAA,cACE,IAAA,EAAM,MAAA;AAAA,cACN,IAAA,EAAM,SAAS,IAAI,CAAA,oEAAA;AAAA;AACrB,WACF;AAAA,UACA,OAAA,EAAS;AAAA,SACX;AAAA,MACF;AAEA,MAAA,MAAM,MAAA,GAAS;AAAA,QACb,MAAM,IAAA,CAAK,IAAA;AAAA,QACX,aAAa,IAAA,CAAK,WAAA;AAAA,QAClB,aAAa,IAAA,CAAK;AAAA,OACpB;AAEA,MAAA,OAAO;AAAA,QACL,OAAA,EAAS,CAAC,EAAE,IAAA,EAAM,MAAA,EAAQ,IAAA,EAAM,IAAA,CAAK,SAAA,CAAU,MAAA,EAAQ,IAAA,EAAM,CAAC,CAAA,EAAG,CAAA;AAAA,QACjE,OAAA,EAAS;AAAA,OACX;AAAA,IACF;AAAA,IAEA,KAAK,kBAAA,EAAoB;AACvB,MAAA,MAAM,cAAA,GAAiB,MAAA,CAAO,IAAA,CAAK,QAAA,IAAY,EAAE,CAAA;AACjD,MAAA,MAAM,SAAA,GAAY,MAAA,CAAO,IAAA,CAAK,UAAA,IAAc,EAAE,CAAA,IAAK,MAAA;AACnD,MAAA,MAAM,QAAA,GAAY,IAAA,CAAK,IAAA,IAAoC,EAAC;AAE5D,MAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,QAAA,OAAO;AAAA,UACL,SAAS,CAAC,EAAE,MAAM,MAAA,EAAQ,IAAA,EAAM,yEAAyE,CAAA;AAAA,UACzG,OAAA,EAAS;AAAA,SACX;AAAA,MACF;AAGA,MAAA,MAAM,EAAE,IAAA,EAAM,KAAA,EAAM,GAAI,iBAAA,CAAkB,gBAAgB,SAAS,CAAA;AACnE,MAAA,IAAI,KAAA,EAAO;AACT,QAAA,OAAO,KAAA;AAAA,MACT;AAEA,MAAA,IAAI,CAAC,IAAA,EAAM;AACT,QAAA,OAAO;AAAA,UACL,OAAA,EAAS;AAAA,YACP;AAAA,cACE,IAAA,EAAM,MAAA;AAAA,cACN,IAAA,EAAM,SAAS,cAAc,CAAA,wEAAA;AAAA;AAC/B,WACF;AAAA,UACA,OAAA,EAAS;AAAA,SACX;AAAA,MACF;AAEA,MAAA,IAAI,CAAC,UAAA,EAAY;AACf,QAAA,OAAO;AAAA,UACL,SAAS,CAAC,EAAE,MAAM,MAAA,EAAQ,IAAA,EAAM,kEAAkE,CAAA;AAAA,UAClG,OAAA,EAAS;AAAA,SACX;AAAA,MACF;AAEA,MAAA,IAAI;AACF,QAAA,MAAM,MAAA,GAAS,MAAM,UAAA,CAAW,cAAA,EAAgB,UAAU,SAAS,CAAA;AAGnE,QAAA,IAAI,MAAA,IAAU,OAAO,MAAA,KAAW,QAAA,IAAY,aAAa,MAAA,EAAQ;AAE/D,UAAA,OAAO,MAAA;AAAA,QACT;AAEA,QAAA,MAAM,IAAA,GAAO,OAAO,MAAA,KAAW,QAAA,GAAW,SAAS,IAAA,CAAK,SAAA,CAAU,MAAA,EAAQ,IAAA,EAAM,CAAC,CAAA;AACjF,QAAA,OAAO;AAAA,UACL,SAAS,CAAC,EAAE,IAAA,EAAM,MAAA,EAAQ,MAAM,CAAA;AAAA,UAChC,OAAA,EAAS;AAAA,SACX;AAAA,MACF,SAAS,GAAA,EAAK;AACZ,QAAA,MAAM,eAAe,GAAA,YAAe,KAAA,GAAQ,GAAA,CAAI,OAAA,GAAU,OAAO,GAAG,CAAA;AACpE,QAAA,OAAO;AAAA,UACL,OAAA,EAAS,CAAC,EAAE,IAAA,EAAM,QAAQ,IAAA,EAAM,CAAA,uBAAA,EAA0B,YAAY,CAAA,CAAA,EAAI,CAAA;AAAA,UAC1E,OAAA,EAAS;AAAA,SACX;AAAA,MACF;AAAA,IACF;AAAA,IAEA;AACE,MAAA,OAAO,IAAA;AAAA;AAEb;AAGO,SAAS,WAAW,QAAA,EAA2B;AACpD,EAAA,OACE,aAAa,sBAAA,IACb,QAAA,KAAa,uBAAA,IACb,QAAA,KAAa,yBACb,QAAA,KAAa,kBAAA;AAEjB;;;ACvUO,IAAM,SAAA,GAAN,MAAM,UAAA,CAAU;AAAA,EAGnB,WAAA,CACY,MAAA,EACA,OAAA,GAA4B,EAAC,EACvC;AAFU,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA;AACA,IAAA,IAAA,CAAA,OAAA,GAAA,OAAA;AAJZ,IAAA,aAAA,CAAA,IAAA,EAAQ,YAAA,CAAA;AAAA,EAKJ;AAAA;AAAA;AAAA;AAAA,EAOJ,MAAc,gBAAA,GAAmB;AAC7B,IAAA,IAAI,CAAC,KAAK,UAAA,EAAY;AAClB,MAAA,MAAM,EAAE,UAAA,EAAW,GAAI,MAAM,OAAO,IAAI,CAAA;AACxC,MAAA,IAAA,CAAK,UAAA,GAAa,UAAA;AAAA,IACtB;AAAA,EACJ;AAAA,EAEA,MAAc,eAAe,MAAA,EAAqC;AAE9D,IAAA,MAAM,cAAc,OAAO,MAAA,CAAO,gBAAgB,UAAA,GAC5C,MAAA,CAAO,aAAY,GACnB,KAAA;AAEN,IAAA,IAAI,CAAC,WAAA,EAAa;AACd,MAAA,OAAO,EAAC;AAAA,IACZ;AAEA,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,SAAA,EAAU;AAGtC,IAAA,OAAO,MAAA,CAAO,WAAA;AAAA,MACV,MAAA,CAAO,KAAA,CAAM,GAAA,CAAI,CAAC,IAAA,KAAS;AAEvB,QAAA,MAAM,WAAW,OAAO,MAAA,CAAO,gBAAgB,UAAA,GACzC,MAAA,CAAO,aAAY,GACnB,MAAA;AACN,QAAA,MAAM,MAAA,GAAS,IAAA,CAAK,OAAA,CAAQ,MAAA,IAAU,QAAA,EAAU,OAAA,CAAQ,IAAA,EAAM,EAAE,CAAA,CAAE,SAAA,CAAU,CAAA,EAAG,CAAC,CAAA,IAAK,KAAA;AACrF,QAAA,OAAO;AAAA,UACH,CAAA,KAAA,EAAQ,MAAM,CAAA,CAAA,EAAI,IAAA,CAAK,IAAI,CAAA,CAAA;AAAA,UAC3B;AAAA,YACI,aAAa,IAAA,CAAK,WAAA;AAAA,YAClB,WAAA,EAAa,IAAA,CAAK,UAAA,CAAY,IAAA,CAAK,WAA0B,CAAA;AAAA,YAC7D,OAAA,EAAS,OAAO,IAAA,KAAc;AAC1B,cAAA,IAAI;AACA,gBAAA,MAAM,WAAW,MAAM,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,MAAM,IAAI,CAAA;AACtD,gBAAA,OAAO,QAAA;AAAA,cACX,SAAS,KAAA,EAAO;AACZ,gBAAA,MAAM,eAAe,KAAA,YAAiB,KAAA,GAAQ,KAAA,CAAM,OAAA,GAAU,OAAO,KAAK,CAAA;AAC1E,gBAAA,MAAM,IAAI,KAAA,CAAM,CAAA,uBAAA,EAA0B,YAAY,CAAA,CAAE,CAAA;AAAA,cAC5D;AAAA,YACJ;AAAA;AACJ,SACJ;AAAA,MACJ,CAAC;AAAA,KACL;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,QAAA,GAA6B;AAC/B,IAAA,MAAM,KAAK,gBAAA,EAAiB;AAG5B,IAAA,IAAI,IAAA,CAAK,QAAQ,UAAA,EAAY;AACzB,MAAA,OAAO,IAAA,CAAK,iBAAA,CAAkB,IAAA,CAAK,OAAA,CAAQ,UAAU,CAAA;AAAA,IACzD;AAIA,IAAA,MAAM,cAAA,GAAiB,OAAQ,IAAA,CAAK,MAAA,CAAe,UAAA,KAAe,UAAA;AAClE,IAAA,MAAM,OAAA,GAAU,iBACT,IAAA,CAAK,MAAA,CAA8B,YAAW,GAC/C,CAAC,KAAK,MAAmB,CAAA;AAE/B,IAAA,MAAM,OAAA,GAAU,MAAM,OAAA,CAAQ,GAAA;AAAA,MAC1B,OAAA,CAAQ,GAAA,CAAI,OAAO,MAAA,KAAW;AAC1B,QAAA,IAAI;AACA,UAAA,OAAO,MAAM,IAAA,CAAK,cAAA,CAAe,MAAM,CAAA;AAAA,QAC3C,SAAS,KAAA,EAAO;AAGZ,UAAA,MAAM,QAAA,GAAW,OAAO,MAAA,CAAO,WAAA,KAAgB,aACzC,MAAA,CAAO,WAAA,MAAiB,SAAA,GACxB,SAAA;AACN,UAAA,OAAA,CAAQ,KAAA,CAAM,CAAA,uCAAA,EAA0C,QAAQ,CAAA,CAAA,CAAA,EAAK,KAAK,CAAA;AAC1E,UAAA,OAAO,EAAC;AAAA,QACZ;AAAA,MACJ,CAAC;AAAA,KACL;AAEA,IAAA,OAAO,OAAA,CAAQ,MAAA,CAAO,CAAC,GAAA,EAAK,KAAA,MAAW,EAAE,GAAG,GAAA,EAAK,GAAG,KAAA,EAAM,CAAA,EAAI,EAAE,CAAA;AAAA,EACpE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,kBAAkB,MAAA,EAAsC;AAClE,IAAA,MAAM,aAAA,GAAgB,MAAM,MAAA,CAAO,gBAAA,EAAiB;AAGpD,IAAA,OAAO,MAAA,CAAO,WAAA;AAAA,MACV,aAAA,CAAc,GAAA,CAAI,CAAC,IAAA,KAAS;AACxB,QAAA,MAAM,UAAA,GAAa,IAAA;AACnB,QAAA,MAAM,SAAA,GAAY,UAAA,CAAW,UAAA,IAAc,UAAA,CAAW,SAAA;AACtD,QAAA,MAAM,OAAA,GAAU,UAAA,CAAW,IAAA,CAAK,IAAI,IAC9B,IAAA,CAAK,IAAA,GACL,IAAA,CAAK,gBAAA,CAAiB,IAAA,CAAK,IAAA,EAAM,UAAA,CAAW,SAAA,EAAW,WAAW,UAAU,CAAA;AAElF,QAAA,OAAO;AAAA,UACH,OAAA;AAAA,UACA;AAAA,YACI,aAAa,IAAA,CAAK,WAAA;AAAA,YAClB,WAAA,EAAa,IAAA,CAAK,UAAA,CAAY,IAAA,CAAK,WAA0B,CAAA;AAAA,YAC7D,OAAA,EAAS,OAAO,IAAA,KAAc;AAE1B,cAAA,IAAI,UAAA,CAAW,IAAA,CAAK,IAAI,CAAA,EAAG;AACvB,gBAAA,MAAM,SAAS,MAAM,eAAA;AAAA,kBACjB,IAAA,CAAK,IAAA;AAAA,kBACL,IAAA;AAAA,kBACA,MAAA;AAAA,kBACA,CAAC,MAAM,QAAA,EAAU,eAAA,KAAoB,OAAO,QAAA,CAAS,IAAA,EAAM,UAAU,eAAe;AAAA,iBACxF;AACA,gBAAA,IAAI,MAAA,EAAQ;AACV,kBAAA,OAAO,MAAA;AAAA,gBACT;AAAA,cACJ;AAIA,cAAA,OAAO,MAAM,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,IAAA,EAAM,MAAM,SAAS,CAAA;AAAA,YAC3D;AAAA;AACJ,SACJ;AAAA,MACJ,CAAC;AAAA,KACL;AAAA,EACJ;AAAA,EAEQ,gBAAA,CAAiB,QAAA,EAAkB,SAAA,EAAoB,UAAA,EAA6B;AACxF,IAAA,MAAM,SAAA,GAAY,aAAa,UAAA,IAAc,KAAA;AAC7C,IAAA,MAAM,UAAA,GAAa,SAAA,CACd,WAAA,EAAY,CACZ,OAAA,CAAQ,aAAA,EAAe,GAAG,CAAA,CAC1B,OAAA,CAAQ,UAAA,EAAY,EAAE,CAAA,IAAK,KAAA;AAChC,IAAA,OAAO,CAAA,KAAA,EAAQ,UAAU,CAAA,CAAA,EAAI,QAAQ,CAAA,CAAA;AAAA,EACzC;AAAA;AAAA;AAAA;AAAA,EAKA,aAAa,QAAA,CAAS,MAAA,EAAwC,OAAA,GAA4B,EAAC,EAAqB;AAC5G,IAAA,OAAO,IAAI,UAAA,CAAU,MAAA,EAAQ,OAAO,EAAE,QAAA,EAAS;AAAA,EACnD;AACJ","file":"ai-adapter.js","sourcesContent":["/**\n * Meta-tools — Injectable tool definitions that let the LLM discover and\n * load MCP tools on-demand, following Anthropic's Tool Search pattern.\n *\n * Instead of injecting 50+ full tool schemas into the context window, you\n * inject just these 4 meta-tools. The LLM calls them to find and load\n * only the tools it actually needs.\n *\n * Meta-tools:\n * • `mcp_search_tool_bm25` — BM25 natural language search\n * • `mcp_search_tool_regex` — Regex pattern search\n * • `mcp_get_tool_schema` — Get full inputSchema for a discovered tool\n * • `mcp_execute_tool` — Execute a discovered tool\n *\n * @packageDocumentation\n */\n\nimport type { Tool, CallToolResult } from '@modelcontextprotocol/sdk/types.js';\nimport type { ToolRouter } from './tool-router.js';\n\n// ---------------------------------------------------------------------------\n// Tool Definitions\n// ---------------------------------------------------------------------------\n\n/**\n * Creates the `mcp_search_tool_bm25` tool definition.\n *\n * This tool lets the LLM search the full catalog of available MCP tools\n * using a BM25 natural-language query. Returns tool names and descriptions\n * without the full inputSchema to save context space.\n */\nexport function createSearchToolDefinition(): Tool {\n return {\n name: 'mcp_search_tool_bm25',\n description:\n 'Search the catalog of available tools using BM25 natural language ranking. ' +\n 'Returns tool names, descriptions, and server info. ' +\n 'Use this FIRST to find relevant tools before calling them. ' +\n 'Example queries: \"database query\", \"send email\", \"github pull request\".',\n inputSchema: {\n type: 'object' as const,\n properties: {\n query: {\n type: 'string',\n description: 'Natural language description of the capability you need.',\n },\n limit: {\n type: 'number',\n description: 'Maximum number of results to return (default: 5, max: 20).',\n },\n },\n required: ['query'],\n },\n };\n}\n\n/**\n * Creates the `mcp_search_tool_regex` tool definition.\n * \n * Matches Anthropic's tool_search_tool_regex exactly (takes a 'query' regex pattern).\n */\nexport function createRegexSearchToolDefinition(): Tool {\n return {\n name: 'mcp_search_tool_regex',\n description:\n 'Search the catalog of available tools using a Python-style regex pattern. ' +\n 'Matches against tool names, descriptions, and parameter descriptions. ' +\n 'Example patterns: \"^github_\", \"weather\", \"(?i)slack\".',\n inputSchema: {\n type: 'object' as const,\n properties: {\n query: {\n type: 'string',\n description: 'Regex pattern to search for (e.g., \"^get_.*_data\", \"database\").',\n },\n limit: {\n type: 'number',\n description: 'Maximum number of results to return (default: 5, max: 20).',\n },\n },\n required: ['query'],\n },\n };\n}\n\n/**\n * Creates the `mcp_get_tool_schema` tool definition.\n *\n * After discovering tools via `mcp_search_tool_bm25` or\n * `mcp_search_tool_regex`, the LLM calls this to load the full\n * inputSchema for a specific tool so it can construct the correct\n * arguments.\n */\nexport function createGetSchemaToolDefinition(): Tool {\n return {\n name: 'mcp_get_tool_schema',\n description:\n 'Get the full input schema (parameters) for a specific tool. ' +\n 'Call this after mcp_search_tool_bm25 to get the parameter details ' +\n 'needed to call a tool correctly.',\n inputSchema: {\n type: 'object' as const,\n properties: {\n toolName: {\n type: 'string',\n description: 'The exact tool name returned by mcp_search_tool_bm25.',\n },\n serverName: {\n type: 'string',\n description:\n 'Optional: The server name provided in mcp_search_tool_bm25. Required if multiple tools have the same name.',\n },\n },\n required: ['toolName'],\n },\n };\n}\n\n/**\n * Creates the `mcp_execute_tool` tool definition.\n *\n * This is the execution meta-tool — the LLM calls this to execute any\n * tool discovered via `mcp_search_tool_bm25` or `mcp_search_tool_regex`.\n * The LLM should first call `mcp_get_tool_schema` to know the correct\n * arguments.\n *\n * Instead of registering every real tool with the framework, we proxy\n * all execution through a single meta-tool.\n */\nexport function createExecuteToolDefinition(): Tool {\n return {\n name: 'mcp_execute_tool',\n description:\n 'Execute a tool that was discovered via mcp_search_tool_bm25. ' +\n 'You MUST call mcp_get_tool_schema first to know the correct parameters. ' +\n 'Pass the exact tool name and its arguments.',\n inputSchema: {\n type: 'object' as const,\n properties: {\n toolName: {\n type: 'string',\n description: 'The exact tool name from mcp_search_tool_bm25 results.',\n },\n serverName: {\n type: 'string',\n description:\n 'Optional: The server name provided in mcp_search_tool_bm25. Required if multiple tools have the same name.',\n },\n args: {\n type: 'object',\n description:\n \"Arguments matching the tool's inputSchema. Omit or pass {} if the tool takes no parameters.\",\n additionalProperties: true,\n },\n },\n required: ['toolName'],\n },\n };\n}\n\n// ---------------------------------------------------------------------------\n// Meta-tool Executors\n// ---------------------------------------------------------------------------\n\n/**\n * Callback for executing a real MCP tool via the correct client.\n * Provided by adapters that wire up client routing.\n */\nexport type CallToolFn = (\n toolName: string,\n args: Record<string, unknown>,\n namespace?: string\n) => Promise<any>;\n\n/**\n * Execute a meta-tool call and return the result in MCP CallToolResult format.\n *\n * @param toolName - One of the meta-tool names (mcp_search_tool_bm25, mcp_search_tool_regex, etc.)\n * @param args - The arguments from the LLM's tool call\n * @param router - The ToolRouter to query\n * @param callToolFn - Optional callback for executing real tools (required for mcp_execute_tool)\n * @returns MCP-compatible CallToolResult, or null if this isn't a meta-tool\n */\nexport async function executeMetaTool(\n toolName: string,\n args: Record<string, unknown>,\n router: ToolRouter,\n callToolFn?: CallToolFn\n): Promise<CallToolResult | null> {\n const resolveToolSchema = (name: string, namespace?: string): { tool?: Tool; error?: CallToolResult } => {\n try {\n return { tool: router.getToolSchema(name, namespace) };\n } catch (err) {\n const errorMessage = err instanceof Error ? err.message : String(err);\n return {\n error: {\n content: [{ type: 'text', text: errorMessage }],\n isError: true,\n },\n };\n }\n };\n\n switch (toolName) {\n case 'mcp_search_tool_bm25': {\n const query = String(args.query ?? '');\n const limit = Math.min(Number(args.limit) || 5, 20);\n\n const results = await router.searchTools(query, limit);\n\n const text = results.length === 0\n ? 'No tools found matching your query. Try different keywords.'\n : results\n .map(\n (t, i) =>\n `${i + 1}. **${t.name}** (server: ${t.serverName})\\n` +\n ` ${t.description}\\n` +\n ` Estimated tokens: ${t.estimatedTokens}`\n )\n .join('\\n');\n\n return {\n content: [{ type: 'text', text }],\n isError: false,\n };\n }\n\n case 'mcp_search_tool_regex': {\n const pattern = String(args.query ?? '');\n const limit = Math.min(Number(args.limit) || 5, 20);\n\n const results = await router.searchToolsRegex(pattern, limit);\n\n const text = results.length === 0\n ? 'No tools matched your regex pattern. Try a broader pattern.'\n : results\n .map(\n (t, i) =>\n `${i + 1}. **${t.name}** (server: ${t.serverName})\\n` +\n ` ${t.description}\\n` +\n ` Estimated tokens: ${t.estimatedTokens}`\n )\n .join('\\n');\n\n return {\n content: [{ type: 'text', text }],\n isError: false,\n };\n }\n\n case 'mcp_get_tool_schema': {\n const name = String(args.toolName ?? '');\n const namespace = String(args.serverName ?? '') || undefined;\n const { tool, error } = resolveToolSchema(name, namespace);\n\n if (error) {\n return error;\n }\n\n if (!tool) {\n return {\n content: [\n {\n type: 'text',\n text: `Tool \"${name}\" not found. Use mcp_search_tool_bm25 to find available tools first.`,\n },\n ],\n isError: true,\n };\n }\n\n const schema = {\n name: tool.name,\n description: tool.description,\n inputSchema: tool.inputSchema,\n };\n\n return {\n content: [{ type: 'text', text: JSON.stringify(schema, null, 2) }],\n isError: false,\n };\n }\n\n case 'mcp_execute_tool': {\n const targetToolName = String(args.toolName ?? '');\n const namespace = String(args.serverName ?? '') || undefined;\n const toolArgs = (args.args as Record<string, unknown>) ?? {};\n\n if (!targetToolName) {\n return {\n content: [{ type: 'text', text: 'Missing required parameter \"toolName\". Specify which tool to execute.' }],\n isError: true,\n };\n }\n\n // Verify the tool exists in our index\n const { tool, error } = resolveToolSchema(targetToolName, namespace);\n if (error) {\n return error;\n }\n\n if (!tool) {\n return {\n content: [\n {\n type: 'text',\n text: `Tool \"${targetToolName}\" not found. Use mcp_search_tool_bm25 to discover available tools first.`,\n },\n ],\n isError: true,\n };\n }\n\n if (!callToolFn) {\n return {\n content: [{ type: 'text', text: 'Tool execution is not available. No callToolFn was configured.' }],\n isError: true,\n };\n }\n\n try {\n const result = await callToolFn(targetToolName, toolArgs, namespace);\n\n // Normalize result to text\n if (result && typeof result === 'object' && 'content' in result) {\n // Already MCP CallToolResult format\n return result as CallToolResult;\n }\n\n const text = typeof result === 'string' ? result : JSON.stringify(result, null, 2);\n return {\n content: [{ type: 'text', text }],\n isError: false,\n };\n } catch (err) {\n const errorMessage = err instanceof Error ? err.message : String(err);\n return {\n content: [{ type: 'text', text: `Tool execution failed: ${errorMessage}` }],\n isError: true,\n };\n }\n }\n\n default:\n return null;\n }\n}\n\n/** Check if a tool name is one of the meta-tools. */\nexport function isMetaTool(toolName: string): boolean {\n return (\n toolName === 'mcp_search_tool_bm25' ||\n toolName === 'mcp_search_tool_regex' ||\n toolName === 'mcp_get_tool_schema' ||\n toolName === 'mcp_execute_tool'\n );\n}\n\n/**\n * Unwraps a meta-tool proxy call (like mcp_execute_tool) to find the real target tool name and arguments.\n * Also automatically strips routing prefixes like tool_{serverId}_.\n * \n * Useful for frontend components that need to determine the actual tool being executed by an AI agent.\n */\nexport function resolveMetaToolProxy(\n toolName: string,\n args: Record<string, unknown> | null | undefined\n): { toolName: string; args: Record<string, unknown> } {\n // Unwrap mcp_execute_tool proxy arguments\n if (toolName === 'mcp_execute_tool') {\n const innerName = args?.toolName;\n const innerArgs = args?.args;\n return {\n toolName: typeof innerName === 'string' && innerName ? innerName : toolName,\n args: innerArgs && typeof innerArgs === 'object' && !Array.isArray(innerArgs)\n ? (innerArgs as Record<string, unknown>)\n : {},\n };\n }\n\n // Strip tool_<serverId>_ prefix used by AIAdapter\n const match = toolName.match(/(?:tool_[^_]+_)?(.+)$/);\n const resolvedName = match?.[1] ?? toolName;\n\n return { toolName: resolvedName, args: args ?? {} };\n}\n\n","import { MCPClient } from '../server/mcp/oauth-client';\nimport { MultiSessionClient } from '../server/mcp/multi-session-client';\nimport type { JSONSchema7 } from 'json-schema';\nimport type { ToolSet } from 'ai';\nimport { ToolRouter } from '../shared/tool-router.js';\nimport { executeMetaTool, isMetaTool } from '../shared/meta-tools.js';\n\nexport interface AIAdapterOptions {\n /** \n * Prefix for tool names to avoid collision with other tools.\n * Defaults to the client's serverId.\n */\n prefix?: string;\n\n /**\n * Optional ToolRouter for intelligent tool selection.\n *\n * When provided with `strategy: 'search'`, the adapter exposes only\n * meta-tools (search_tools, get_tool_schema) instead of all tool schemas,\n * reducing context window usage by 80–95%.\n *\n * When not provided, all tools are returned as before (backward-compatible).\n */\n toolRouter?: ToolRouter;\n}\n\n/**\n * Adapter to use MCP tools with the Vercel AI SDK.\n */\nexport class AIAdapter {\n private jsonSchema: typeof import('ai').jsonSchema | undefined;\n\n constructor(\n private client: MCPClient | MultiSessionClient,\n private options: AIAdapterOptions = {}\n ) { }\n\n\n\n /**\n * Lazy-loads the jsonSchema function from the AI SDK.\n */\n private async ensureJsonSchema() {\n if (!this.jsonSchema) {\n const { jsonSchema } = await import('ai');\n this.jsonSchema = jsonSchema;\n }\n }\n\n private async transformTools(client: MCPClient): Promise<ToolSet> {\n // Safe check for isConnected method (duck typing for bundler compatibility)\n const isConnected = typeof client.isConnected === 'function'\n ? client.isConnected()\n : false;\n\n if (!isConnected) {\n return {};\n }\n\n const result = await client.listTools();\n\n // @ts-ignore: ToolSet type inference can be tricky with dynamic imports\n return Object.fromEntries(\n result.tools.map((tool) => {\n // Safe access to getServerId\n const serverId = typeof client.getServerId === 'function'\n ? client.getServerId()\n : undefined;\n const prefix = this.options.prefix ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n return [\n `tool_${prefix}_${tool.name}`,\n {\n description: tool.description,\n inputSchema: this.jsonSchema!(tool.inputSchema as JSONSchema7),\n execute: async (args: any) => {\n try {\n const response = await client.callTool(tool.name, args);\n return response;\n } catch (error) {\n const errorMessage = error instanceof Error ? error.message : String(error);\n throw new Error(`Tool execution failed: ${errorMessage}`);\n }\n }\n }\n ];\n })\n );\n }\n\n /**\n * Fetches tools from the client(s) and converts them to AI SDK tools.\n */\n async getTools(): Promise<ToolSet> {\n await this.ensureJsonSchema();\n\n // If a ToolRouter is provided, use its filtered output\n if (this.options.toolRouter) {\n return this.getToolsViaRouter(this.options.toolRouter);\n }\n\n // Use duck typing instead of instanceof to handle module bundling issues\n // MultiSessionClient has getClients(), MCPClient does not\n const isMultiSession = typeof (this.client as any).getClients === 'function';\n const clients = isMultiSession\n ? (this.client as MultiSessionClient).getClients()\n : [this.client as MCPClient];\n\n const results = await Promise.all(\n clients.map(async (client) => {\n try {\n return await this.transformTools(client);\n } catch (error) {\n // For multi-client, we log and continue.\n // This is safer than throwing.\n const serverId = typeof client.getServerId === 'function'\n ? client.getServerId() ?? 'unknown'\n : 'unknown';\n console.error(`[AIAdapter] Failed to fetch tools from ${serverId}:`, error);\n return {};\n }\n })\n );\n\n return results.reduce((acc, tools) => ({ ...acc, ...tools }), {});\n }\n\n /**\n * Build a ToolSet from a ToolRouter's filtered output.\n *\n * In `search` strategy, only meta-tools are registered with the framework.\n * Real tool execution is proxied through `mcp_execute_tool` which uses\n * `router.callTool()` to route to the correct MCP client.\n */\n private async getToolsViaRouter(router: ToolRouter): Promise<ToolSet> {\n const filteredTools = await router.getFilteredTools();\n\n // @ts-ignore: ToolSet type inference can be tricky with dynamic imports\n return Object.fromEntries(\n filteredTools.map((tool) => {\n const routedTool = tool as typeof tool & { sessionId?: string; serverName?: string };\n const namespace = routedTool.serverName ?? routedTool.sessionId;\n const toolKey = isMetaTool(tool.name)\n ? tool.name\n : this.getRouterToolKey(tool.name, routedTool.sessionId, routedTool.serverName);\n\n return [\n toolKey,\n {\n description: tool.description,\n inputSchema: this.jsonSchema!(tool.inputSchema as JSONSchema7),\n execute: async (args: any) => {\n // Handle meta-tool calls via the router\n if (isMetaTool(tool.name)) {\n const result = await executeMetaTool(\n tool.name,\n args,\n router,\n (name, toolArgs, targetNamespace) => router.callTool(name, toolArgs, targetNamespace)\n );\n if (result) {\n return result;\n }\n }\n\n // For non-meta tools in 'all' or 'groups' strategy,\n // route directly to the correct MCP client\n return await router.callTool(tool.name, args, namespace);\n },\n },\n ];\n })\n );\n }\n\n private getRouterToolKey(toolName: string, sessionId?: string, serverName?: string): string {\n const namespace = sessionId ?? serverName ?? 'mcp';\n const normalized = namespace\n .toLowerCase()\n .replace(/[^a-z0-9]+/g, '_')\n .replace(/^_+|_+$/g, '') || 'mcp';\n return `tool_${normalized}_${toolName}`;\n }\n\n /**\n * Convenience static method to fetch tools in a single line.\n */\n static async getTools(client: MCPClient | MultiSessionClient, options: AIAdapterOptions = {}): Promise<ToolSet> {\n return new AIAdapter(client, options).getTools();\n }\n}\n"]}