@tambo-ai/react 0.58.1 → 0.60.0

package/esm/mcp/__tests__/mcp-hooks.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-hooks.test.js","sourceRoot":"","sources":["../../../src/mcp/__tests__/mcp-hooks.test.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,KAAK,EAAE,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AACzC,OAAO,EAAE,kBAAkB,EAAE,MAAM,uCAAuC,CAAC;AAC3E,OAAO,EAAE,qBAAqB,EAAE,MAAM,0CAA0C,CAAC;AACjF,OAAO,EAAE,qBAAqB,EAAE,MAAM,yCAAyC,CAAC;AAChF,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAC7E,OAAO,EAAE,qBAAqB,EAAwB,MAAM,cAAc,CAAC;AAE3E,gDAAgD;AAChD,IAAI,UAAU,GAAwB,IAAI,CAAC,EAAE,EAAE,CAAC;AAChD,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;IAChC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,GAAG,IAAW,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,EAAE;IAC9D,YAAY,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE;CAC3C,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,mDAAmD,EAAE,GAAG,EAAE;IACjE,IAAI,WAAwB,CAAC;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QACvB,WAAW,GAAG,IAAI,WAAW,CAAC;YAC5B,cAAc,EAAE;gBACd,OAAO,EAAE;oBACP,KAAK,EAAE,KAAK,EAAE,4BAA4B;iBAC3C;aACF;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,WAAW,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,0CAA0C;QAC1C,MAAM,cAAc,GAAG;YACrB,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE;gBAC/C,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE;aAChD;SACF,CAAC;QACF,MAAM,cAAc,GAAG;YACrB,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE;gBAC/C,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE;aAChD;SACF,CAAC;QAEF,MAAM,WAAW,GAAG;YAClB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC;YACxD,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QACF,MAAM,WAAW,GAAG;YAClB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC;YACxD,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QAEF,MAAM,OAAO,GAAG;YACd,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QACF,MAAM,OAAO,GAAG;YACd,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QAEF,uEAAuE;QACvE,UAAU,CAAC,kBAAkB,CAAC,KAAK,EAAE,GAAW,EAAE,EAAE;YAClD,IAAI,GAAG,KAAK,0BAA0B;gBAAE,OAAO,OAAO,CAAC;YACvD,IAAI,GAAG,KAAK,0BAA0B;gBAAE,OAAO,OAAO,CAAC;YACvD,MAAM,IAAI,KAAK,CAAC,mBAAmB,GAAG,EAAE,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,IAAI,eAAe,GAAsB,EAAE,CAAC;QAC5C,MAAM,OAAO,GAAa,GAAG,EAAE;YAC7B,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,qBAAqB,EAAE,CAAC;YAClD,SAAS,CAAC,GAAG,EAAE;gBACb,IAAI,OAAO,EAAE,CAAC;oBACZ,eAAe,GAAG,OAAO,CAAC;gBAC5B,CAAC;YACH,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;YACd,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,CACJ,oBAAC,kBAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;gBACL,MAAM,EAAE,EAAE,OAAO,EAAE,sBAAsB,EAAS;gBAClD,WAAW;gBACX,eAAe,EAAE,KAAK;aACvB;YAED,oBAAC,qBAAqB;gBACpB,oBAAC,qBAAqB;oBACpB,oBAAC,gBAAgB,IACf,UAAU,EAAE;4BACV;gCACE,GAAG,EAAE,0BAA0B;gCAC/B,SAAS,EAAE,YAAY,CAAC,GAAG;6BAC5B;4BACD;gCACE,GAAG,EAAE,0BAA0B;gCAC/B,SAAS,EAAE,YAAY,CAAC,GAAG;6BAC5B;yBACF;wBAED,oBAAC,OAAO,OAAG,CACM,CACG,CACF,CACI,CAC/B,CAAC;QAEF,oCAAoC;QACpC,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,iCAAiC;QACjC,MAAM,WAAW,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9D,MAAM,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAC3C,MAAM,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAC3C,MAAM,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAC3C,MAAM,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAE3C,iDAAiD;QACjD,MAAM,QAAQ,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QAC5E,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QAE9D,MAAM,QAAQ,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QAC5E,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,0CAA0C;QAC1C,MAAM,cAAc,GAAG;YACrB,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE;gBAC/C,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE;aAChD;SACF,CAAC;QACF,MAAM,cAAc,GAAG;YACrB,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE;gBAC/C,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE;aAChD;SACF,CAAC;QAEF,MAAM,WAAW,GAAG;YAClB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC;YACxD,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QACF,MAAM,WAAW,GAAG;YAClB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC;YACxD,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QAEF,MAAM,OAAO,GAAG;YACd,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QACF,MAAM,OAAO,GAAG;YACd,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QAEF,UAAU,CAAC,kBAAkB,CAAC,KAAK,EAAE,GAAW,EAAE,EAAE;YAClD,IAAI,GAAG,KAAK,0BAA0B;gBAAE,OAAO,OAAO,CAAC;YACvD,IAAI,GAAG,KAAK,0BAA0B;gBAAE,OAAO,OAAO,CAAC;YACvD,MAAM,IAAI,KAAK,CAAC,mBAAmB,GAAG,EAAE,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,IAAI,eAAe,GAAsB,EAAE,CAAC;QAC5C,MAAM,OAAO,GAAa,GAAG,EAAE;YAC7B,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,qBAAqB,EAAE,CAAC;YAClD,SAAS,CAAC,GAAG,EAAE;gBACb,IAAI,OAAO,EAAE,CAAC;oBACZ,eAAe,GAAG,OAAO,CAAC;gBAC5B,CAAC;YACH,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;YACd,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CACzB,oBAAC,kBAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;gBACL,MAAM,EAAE,EAAE,OAAO,EAAE,sBAAsB,EAAS;gBAClD,WAAW;gBACX,eAAe,EAAE,KAAK;aACvB;YAED,oBAAC,qBAAqB;gBACpB,oBAAC,qBAAqB;oBACpB,oBAAC,gBAAgB,IACf,UAAU,EAAE;4BACV;gCACE,GAAG,EAAE,0BAA0B;gCAC/B,SAAS,EAAE,YAAY,CAAC,GAAG;6BAC5B;4BACD;gCACE,GAAG,EAAE,0BAA0B;gCAC/B,SAAS,EAAE,YAAY,CAAC,GAAG;6BAC5B;yBACF;wBAED,oBAAC,OAAO,OAAG,CACM,CACG,CACF,CACI,CAC/B,CAAC;QAEF,wCAAwC;QACxC,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,kBAAkB,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACrE,MAAM,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAClD,MAAM,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAClD,MAAM,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAClD,MAAM,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAElD,kBAAkB;QAClB,QAAQ,CACN,oBAAC,kBAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;gBACL,MAAM,EAAE,EAAE,OAAO,EAAE,sBAAsB,EAAS;gBAClD,WAAW;gBACX,eAAe,EAAE,KAAK;aACvB;YAED,oBAAC,qBAAqB;gBACpB,oBAAC,qBAAqB;oBACpB,oBAAC,gBAAgB,IACf,UAAU,EAAE;4BACV;gCACE,GAAG,EAAE,0BAA0B;gCAC/B,SAAS,EAAE,YAAY,CAAC,GAAG;6BAC5B;yBACF;wBAED,oBAAC,OAAO,OAAG,CACM,CACG,CACF,CACI,CAC/B,CAAC;QAEF,qEAAqE;QACrE,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,kBAAkB,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACrE,MAAM,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAClD,MAAM,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAClD,MAAM,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACtD,MAAM,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAEtD,sCAAsC;QACtC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,cAAc,GAAG;YACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;SACzD,CAAC;QACF,MAAM,cAAc,GAAG;YACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;SACzD,CAAC;QAEF,MAAM,WAAW,GAAG;YAClB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC;YACxD,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QACF,MAAM,WAAW,GAAG;YAClB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC;YACxD,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QAEF,MAAM,OAAO,GAAG;YACd,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QACF,MAAM,OAAO,GAAG;YACd,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QAEF,UAAU,CAAC,kBAAkB,CAAC,KAAK,EAAE,GAAW,EAAE,EAAE;YAClD,IAAI,GAAG,KAAK,0BAA0B;gBAAE,OAAO,OAAO,CAAC;YACvD,IAAI,GAAG,KAAK,0BAA0B;gBAAE,OAAO,OAAO,CAAC;YACvD,MAAM,IAAI,KAAK,CAAC,mBAAmB,GAAG,EAAE,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,IAAI,eAAe,GAAsB,EAAE,CAAC;QAC5C,IAAI,kBAAkB,GAAa,EAAE,CAAC;QACtC,MAAM,OAAO,GAAa,GAAG,EAAE;YAC7B,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,qBAAqB,EAAE,CAAC;YAClD,MAAM,OAAO,GAAG,kBAAkB,EAAE,CAAC;YACrC,SAAS,CAAC,GAAG,EAAE;gBACb,IAAI,OAAO,EAAE,CAAC;oBACZ,eAAe,GAAG,OAAO,CAAC;gBAC5B,CAAC;gBACD,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACjD,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YACvB,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,CACJ,oBAAC,kBAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;gBACL,MAAM,EAAE,EAAE,OAAO,EAAE,sBAAsB,EAAS;gBAClD,WAAW;gBACX,eAAe,EAAE,KAAK;aACvB;YAED,oBAAC,qBAAqB;gBACpB,oBAAC,qBAAqB;oBACpB,oBAAC,gBAAgB,IACf,UAAU,EAAE;4BACV;gCACE,GAAG,EAAE,0BAA0B;gCAC/B,SAAS,EAAE,YAAY,CAAC,GAAG;6BAC5B;4BACD;gCACE,GAAG,EAAE,0BAA0B;gCAC/B,SAAS,EAAE,YAAY,CAAC,GAAG;6BAC5B;yBACF;wBAED,oBAAC,OAAO,OAAG,CACM,CACG,CACF,CACI,CAC/B,CAAC;QAEF,gCAAgC;QAChC,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,qDAAqD;QACrD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACzD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEzD,8DAA8D;QAC9D,MAAM,SAAS,GAAG,WAAW;aAC1B,aAAa,EAAE;aACf,MAAM,EAAE;aACR,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAElC,+DAA+D;QAC/D,MAAM,eAAe,GAAG,SAAS,CAAC,MAAM,CACtC,CAAC,GAAG,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,KAAK,aAAa,CACxD,CAAC;QACF,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACvC,sDAAsD;QACtD,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAE/D,6EAA6E;QAC7E,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACvE,uFAAuF;QACvF,kBAAkB,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YACjC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QACH,oBAAoB;QACpB,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,cAAc,GAAG;YACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;SACzD,CAAC;QAEF,MAAM,WAAW,GAAG;YAClB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC;YACxD,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QAEF,MAAM,OAAO,GAAG;YACd,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QAEF,gCAAgC;QAChC,UAAU,CAAC,kBAAkB,CAAC,KAAK,EAAE,GAAW,EAAE,EAAE;YAClD,IAAI,GAAG,KAAK,0BAA0B;gBAAE,OAAO,OAAO,CAAC;YACvD,IAAI,GAAG,KAAK,0BAA0B,EAAE,CAAC;gBACvC,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;YACvC,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,mBAAmB,GAAG,EAAE,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,IAAI,eAAe,GAAsB,EAAE,CAAC;QAC5C,IAAI,eAAe,GAAG,CAAC,CAAC;QACxB,MAAM,OAAO,GAAa,GAAG,EAAE;YAC7B,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,qBAAqB,EAAE,CAAC;YAClD,MAAM,OAAO,GAAG,kBAAkB,EAAE,CAAC;YACrC,SAAS,CAAC,GAAG,EAAE;gBACb,IAAI,OAAO,EAAE,CAAC;oBACZ,eAAe,GAAG,OAAO,CAAC;gBAC5B,CAAC;gBACD,eAAe,GAAG,OAAO,CAAC,MAAM,CAAC;YACnC,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YACvB,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,CACJ,oBAAC,kBAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;gBACL,MAAM,EAAE,EAAE,OAAO,EAAE,sBAAsB,EAAS;gBAClD,WAAW;gBACX,eAAe,EAAE,KAAK;aACvB;YAED,oBAAC,qBAAqB;gBACpB,oBAAC,qBAAqB;oBACpB,oBAAC,gBAAgB,IACf,UAAU,EAAE;4BACV;gCACE,GAAG,EAAE,0BAA0B;gCAC/B,SAAS,EAAE,YAAY,CAAC,GAAG;6BAC5B;4BACD;gCACE,GAAG,EAAE,0BAA0B;gCAC/B,SAAS,EAAE,YAAY,CAAC,GAAG;6BAC5B;yBACF;wBAED,oBAAC,OAAO,OAAG,CACM,CACG,CACF,CACI,CAC/B,CAAC;QAEF,yCAAyC;QACzC,8CAA8C;QAC9C,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACvC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,qCAAqC;QACxE,CAAC,CAAC,CAAC;QAEH,6CAA6C;QAC7C,MAAM,WAAW,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9D,MAAM,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC1C,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,cAAc,GAAG;YACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;SACzD,CAAC;QACF,MAAM,cAAc,GAAG;YACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;SACzD,CAAC;QAEF,MAAM,WAAW,GAAG;YAClB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC;YACxD,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QACF,MAAM,WAAW,GAAG;YAClB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC;YACxD,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QAEF,MAAM,OAAO,GAAG;YACd,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QACF,MAAM,OAAO,GAAG;YACd,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;SACjB,CAAC;QAEF,UAAU,CAAC,kBAAkB,CAAC,KAAK,EAAE,GAAW,EAAE,EAAE;YAClD,IAAI,GAAG,KAAK,0BAA0B;gBAAE,OAAO,OAAO,CAAC;YACvD,IAAI,GAAG,KAAK,0BAA0B;gBAAE,OAAO,OAAO,CAAC;YACvD,MAAM,IAAI,KAAK,CAAC,mBAAmB,GAAG,EAAE,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,IAAI,eAAe,GAAsB,EAAE,CAAC;QAC5C,MAAM,OAAO,GAAa,GAAG,EAAE;YAC7B,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,qBAAqB,EAAE,CAAC;YAClD,SAAS,CAAC,GAAG,EAAE;gBACb,IAAI,OAAO,EAAE,CAAC;oBACZ,eAAe,GAAG,OAAO,CAAC;gBAC5B,CAAC;YACH,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;YACd,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;QAEF,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CACzB,oBAAC,kBAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;gBACL,MAAM,EAAE,EAAE,OAAO,EAAE,sBAAsB,EAAS;gBAClD,WAAW;gBACX,eAAe,EAAE,KAAK;aACvB;YAED,oBAAC,qBAAqB;gBACpB,oBAAC,qBAAqB;oBACpB,oBAAC,gBAAgB,IACf,UAAU,EAAE;4BACV;gCACE,GAAG,EAAE,0BAA0B;gCAC/B,SAAS,EAAE,YAAY,CAAC,GAAG;6BAC5B;yBACF;wBAED,oBAAC,OAAO,OAAG,CACM,CACG,CACF,CACI,CAC/B,CAAC;QAEF,yCAAyC;QACzC,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAExE,eAAe;QACf,QAAQ,CACN,oBAAC,kBAAkB,CAAC,QAAQ,IAC1B,KAAK,EAAE;gBACL,MAAM,EAAE,EAAE,OAAO,EAAE,sBAAsB,EAAS;gBAClD,WAAW;gBACX,eAAe,EAAE,KAAK;aACvB;YAED,oBAAC,qBAAqB;gBACpB,oBAAC,qBAAqB;oBACpB,oBAAC,gBAAgB,IACf,UAAU,EAAE;4BACV;gCACE,GAAG,EAAE,0BAA0B;gCAC/B,SAAS,EAAE,YAAY,CAAC,GAAG;6BAC5B;4BACD;gCACE,GAAG,EAAE,0BAA0B;gCAC/B,SAAS,EAAE,YAAY,CAAC,GAAG;6BAC5B;yBACF;wBAED,oBAAC,OAAO,OAAG,CACM,CACG,CACF,CACI,CAC/B,CAAC;QAEF,wCAAwC;QACxC,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9D,MAAM,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC1C,MAAM,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { render, waitFor } from \"@testing-library/react\";\nimport { QueryClient } from \"@tanstack/react-query\";\nimport React, { useEffect } from \"react\";\nimport { TamboClientContext } from \"../../providers/tambo-client-provider\";\nimport { TamboMcpTokenProvider } from \"../../providers/tambo-mcp-token-provider\";\nimport { TamboRegistryProvider } from \"../../providers/tambo-registry-provider\";\nimport { MCPTransport } from \"../mcp-client\";\nimport { TamboMcpProvider, useTamboMcpServers } from \"../tambo-mcp-provider\";\nimport { useTamboMcpPromptList, type ListPromptEntry } from \"../mcp-hooks\";\n\n// Mock the MCP client to avoid ES module issues\nlet createImpl: jest.Mock<any, any> = jest.fn();\njest.mock(\"../mcp-client\", () => ({\n  MCPClient: { create: (...args: any[]) => createImpl(...args) },\n  MCPTransport: { SSE: \"sse\", HTTP: \"http\" },\n}));\n\ndescribe(\"useTamboMcpPromptList - individual server caching\", () => {\n  let queryClient: QueryClient;\n\n  beforeEach(() => {\n    createImpl = jest.fn();\n    queryClient = new QueryClient({\n      defaultOptions: {\n        queries: {\n          retry: false, // Disable retries for tests\n        },\n      },\n    });\n  });\n\n  afterEach(() => {\n    queryClient.clear();\n  });\n\n  it(\"should fetch and combine prompts from multiple servers\", async () => {\n    // Mock two servers with different prompts\n    const serverAPrompts = {\n      prompts: [\n        { name: \"prompt-a1\", description: \"Prompt A1\" },\n        { name: \"prompt-a2\", description: \"Prompt A2\" },\n      ],\n    };\n    const serverBPrompts = {\n      prompts: [\n        { name: \"prompt-b1\", description: \"Prompt B1\" },\n        { name: \"prompt-b2\", description: \"Prompt B2\" },\n      ],\n    };\n\n    const mockClientA = {\n      listTools: jest.fn().mockResolvedValue([]),\n      listPrompts: jest.fn().mockResolvedValue(serverAPrompts),\n      close: jest.fn(),\n    };\n    const mockClientB = {\n      listTools: jest.fn().mockResolvedValue([]),\n      listPrompts: jest.fn().mockResolvedValue(serverBPrompts),\n      close: jest.fn(),\n    };\n\n    const clientA = {\n      client: mockClientA,\n      listTools: jest.fn().mockResolvedValue([]),\n      close: jest.fn(),\n    };\n    const clientB = {\n      client: mockClientB,\n      listTools: jest.fn().mockResolvedValue([]),\n      close: jest.fn(),\n    };\n\n    // Mock MCPClient.create to return different clients for different URLs\n    createImpl.mockImplementation(async (url: string) => {\n      if (url === \"https://server-a.example\") return clientA;\n      if (url === \"https://server-b.example\") return clientB;\n      throw new Error(`Unexpected URL: ${url}`);\n    });\n\n    let capturedPrompts: ListPromptEntry[] = [];\n    const Capture: React.FC = () => {\n      const { data: prompts } = useTamboMcpPromptList();\n      useEffect(() => {\n        if (prompts) {\n          capturedPrompts = prompts;\n        }\n      }, [prompts]);\n      return null;\n    };\n\n    render(\n      <TamboClientContext.Provider\n        value={{\n          client: { baseURL: \"https://api.tambo.co\" } as any,\n          queryClient,\n          isUpdatingToken: false,\n        }}\n      >\n        <TamboRegistryProvider>\n          <TamboMcpTokenProvider>\n            <TamboMcpProvider\n              mcpServers={[\n                {\n                  url: \"https://server-a.example\",\n                  transport: MCPTransport.SSE,\n                },\n                {\n                  url: \"https://server-b.example\",\n                  transport: MCPTransport.SSE,\n                },\n              ]}\n            >\n              <Capture />\n            </TamboMcpProvider>\n          </TamboMcpTokenProvider>\n        </TamboRegistryProvider>\n      </TamboClientContext.Provider>,\n    );\n\n    // Wait for all prompts to be loaded\n    await waitFor(() => {\n      expect(capturedPrompts.length).toBe(4);\n    });\n\n    // Verify all prompts are present\n    const promptNames = capturedPrompts.map((p) => p.prompt.name);\n    expect(promptNames).toContain(\"prompt-a1\");\n    expect(promptNames).toContain(\"prompt-a2\");\n    expect(promptNames).toContain(\"prompt-b1\");\n    expect(promptNames).toContain(\"prompt-b2\");\n\n    // Verify each prompt has the correct server info\n    const promptA1 = capturedPrompts.find((p) => p.prompt.name === \"prompt-a1\");\n    expect(promptA1?.server.url).toBe(\"https://server-a.example\");\n\n    const promptB1 = capturedPrompts.find((p) => p.prompt.name === \"prompt-b1\");\n    expect(promptB1?.server.url).toBe(\"https://server-b.example\");\n  });\n\n  it(\"should remove prompts when a server is removed\", async () => {\n    // Mock two servers with different prompts\n    const serverAPrompts = {\n      prompts: [\n        { name: \"prompt-a1\", description: \"Prompt A1\" },\n        { name: \"prompt-a2\", description: \"Prompt A2\" },\n      ],\n    };\n    const serverBPrompts = {\n      prompts: [\n        { name: \"prompt-b1\", description: \"Prompt B1\" },\n        { name: \"prompt-b2\", description: \"Prompt B2\" },\n      ],\n    };\n\n    const mockClientA = {\n      listTools: jest.fn().mockResolvedValue([]),\n      listPrompts: jest.fn().mockResolvedValue(serverAPrompts),\n      close: jest.fn(),\n    };\n    const mockClientB = {\n      listTools: jest.fn().mockResolvedValue([]),\n      listPrompts: jest.fn().mockResolvedValue(serverBPrompts),\n      close: jest.fn(),\n    };\n\n    const clientA = {\n      client: mockClientA,\n      listTools: jest.fn().mockResolvedValue([]),\n      close: jest.fn(),\n    };\n    const clientB = {\n      client: mockClientB,\n      listTools: jest.fn().mockResolvedValue([]),\n      close: jest.fn(),\n    };\n\n    createImpl.mockImplementation(async (url: string) => {\n      if (url === \"https://server-a.example\") return clientA;\n      if (url === \"https://server-b.example\") return clientB;\n      throw new Error(`Unexpected URL: ${url}`);\n    });\n\n    let capturedPrompts: ListPromptEntry[] = [];\n    const Capture: React.FC = () => {\n      const { data: prompts } = useTamboMcpPromptList();\n      useEffect(() => {\n        if (prompts) {\n          capturedPrompts = prompts;\n        }\n      }, [prompts]);\n      return null;\n    };\n\n    const { rerender } = render(\n      <TamboClientContext.Provider\n        value={{\n          client: { baseURL: \"https://api.tambo.co\" } as any,\n          queryClient,\n          isUpdatingToken: false,\n        }}\n      >\n        <TamboRegistryProvider>\n          <TamboMcpTokenProvider>\n            <TamboMcpProvider\n              mcpServers={[\n                {\n                  url: \"https://server-a.example\",\n                  transport: MCPTransport.SSE,\n                },\n                {\n                  url: \"https://server-b.example\",\n                  transport: MCPTransport.SSE,\n                },\n              ]}\n            >\n              <Capture />\n            </TamboMcpProvider>\n          </TamboMcpTokenProvider>\n        </TamboRegistryProvider>\n      </TamboClientContext.Provider>,\n    );\n\n    // Wait for initial prompts to be loaded\n    await waitFor(() => {\n      expect(capturedPrompts.length).toBe(4);\n    });\n\n    const initialPromptNames = capturedPrompts.map((p) => p.prompt.name);\n    expect(initialPromptNames).toContain(\"prompt-a1\");\n    expect(initialPromptNames).toContain(\"prompt-a2\");\n    expect(initialPromptNames).toContain(\"prompt-b1\");\n    expect(initialPromptNames).toContain(\"prompt-b2\");\n\n    // Remove server B\n    rerender(\n      <TamboClientContext.Provider\n        value={{\n          client: { baseURL: \"https://api.tambo.co\" } as any,\n          queryClient,\n          isUpdatingToken: false,\n        }}\n      >\n        <TamboRegistryProvider>\n          <TamboMcpTokenProvider>\n            <TamboMcpProvider\n              mcpServers={[\n                {\n                  url: \"https://server-a.example\",\n                  transport: MCPTransport.SSE,\n                },\n              ]}\n            >\n              <Capture />\n            </TamboMcpProvider>\n          </TamboMcpTokenProvider>\n        </TamboRegistryProvider>\n      </TamboClientContext.Provider>,\n    );\n\n    // Wait for prompts to be updated (server B prompts should disappear)\n    await waitFor(() => {\n      expect(capturedPrompts.length).toBe(2);\n    });\n\n    const updatedPromptNames = capturedPrompts.map((p) => p.prompt.name);\n    expect(updatedPromptNames).toContain(\"prompt-a1\");\n    expect(updatedPromptNames).toContain(\"prompt-a2\");\n    expect(updatedPromptNames).not.toContain(\"prompt-b1\");\n    expect(updatedPromptNames).not.toContain(\"prompt-b2\");\n\n    // Verify server B's client was closed\n    expect(clientB.close).toHaveBeenCalled();\n  });\n\n  it(\"should cache prompts individually per server using server key\", async () => {\n    const serverAPrompts = {\n      prompts: [{ name: \"prompt-a\", description: \"Prompt A\" }],\n    };\n    const serverBPrompts = {\n      prompts: [{ name: \"prompt-b\", description: \"Prompt B\" }],\n    };\n\n    const mockClientA = {\n      listTools: jest.fn().mockResolvedValue([]),\n      listPrompts: jest.fn().mockResolvedValue(serverAPrompts),\n      close: jest.fn(),\n    };\n    const mockClientB = {\n      listTools: jest.fn().mockResolvedValue([]),\n      listPrompts: jest.fn().mockResolvedValue(serverBPrompts),\n      close: jest.fn(),\n    };\n\n    const clientA = {\n      client: mockClientA,\n      listTools: jest.fn().mockResolvedValue([]),\n      close: jest.fn(),\n    };\n    const clientB = {\n      client: mockClientB,\n      listTools: jest.fn().mockResolvedValue([]),\n      close: jest.fn(),\n    };\n\n    createImpl.mockImplementation(async (url: string) => {\n      if (url === \"https://server-a.example\") return clientA;\n      if (url === \"https://server-b.example\") return clientB;\n      throw new Error(`Unexpected URL: ${url}`);\n    });\n\n    let capturedPrompts: ListPromptEntry[] = [];\n    let capturedServerKeys: string[] = [];\n    const Capture: React.FC = () => {\n      const { data: prompts } = useTamboMcpPromptList();\n      const servers = useTamboMcpServers();\n      useEffect(() => {\n        if (prompts) {\n          capturedPrompts = prompts;\n        }\n        capturedServerKeys = servers.map((s) => s.key);\n      }, [prompts, servers]);\n      return null;\n    };\n\n    render(\n      <TamboClientContext.Provider\n        value={{\n          client: { baseURL: \"https://api.tambo.co\" } as any,\n          queryClient,\n          isUpdatingToken: false,\n        }}\n      >\n        <TamboRegistryProvider>\n          <TamboMcpTokenProvider>\n            <TamboMcpProvider\n              mcpServers={[\n                {\n                  url: \"https://server-a.example\",\n                  transport: MCPTransport.SSE,\n                },\n                {\n                  url: \"https://server-b.example\",\n                  transport: MCPTransport.SSE,\n                },\n              ]}\n            >\n              <Capture />\n            </TamboMcpProvider>\n          </TamboMcpTokenProvider>\n        </TamboRegistryProvider>\n      </TamboClientContext.Provider>,\n    );\n\n    // Wait for prompts to be loaded\n    await waitFor(() => {\n      expect(capturedPrompts.length).toBe(2);\n    });\n\n    // Verify that listPrompts was called once per server\n    expect(mockClientA.listPrompts).toHaveBeenCalledTimes(1);\n    expect(mockClientB.listPrompts).toHaveBeenCalledTimes(1);\n\n    // Verify the query cache has separate entries for each server\n    const cacheKeys = queryClient\n      .getQueryCache()\n      .getAll()\n      .map((query) => query.queryKey);\n\n    // Should have separate cache entries for each server's prompts\n    const promptCacheKeys = cacheKeys.filter(\n      (key) => Array.isArray(key) && key[0] === \"mcp-prompts\",\n    );\n    expect(promptCacheKeys.length).toBe(2);\n    // Ensure provider keys are populated before asserting\n    await waitFor(() => expect(capturedServerKeys.length).toBe(2));\n\n    // Verify each connected server key is present exactly once in the cache keys\n    const promptKeySet = new Set(promptCacheKeys.map((k) => String(k[1])));\n    // The server keys are captured from the provider to avoid relying on string substrings\n    capturedServerKeys.forEach((key) => {\n      expect(promptKeySet.has(key)).toBe(true);\n    });\n    // And no duplicates\n    expect(promptKeySet.size).toBe(capturedServerKeys.length);\n  });\n\n  it(\"should handle server errors gracefully without affecting other servers\", async () => {\n    const serverAPrompts = {\n      prompts: [{ name: \"prompt-a\", description: \"Prompt A\" }],\n    };\n\n    const mockClientA = {\n      listTools: jest.fn().mockResolvedValue([]),\n      listPrompts: jest.fn().mockResolvedValue(serverAPrompts),\n      close: jest.fn(),\n    };\n\n    const clientA = {\n      client: mockClientA,\n      listTools: jest.fn().mockResolvedValue([]),\n      close: jest.fn(),\n    };\n\n    // Server B will fail to connect\n    createImpl.mockImplementation(async (url: string) => {\n      if (url === \"https://server-a.example\") return clientA;\n      if (url === \"https://server-b.example\") {\n        throw new Error(\"Connection failed\");\n      }\n      throw new Error(`Unexpected URL: ${url}`);\n    });\n\n    let capturedPrompts: ListPromptEntry[] = [];\n    let mcpServersCount = 0;\n    const Capture: React.FC = () => {\n      const { data: prompts } = useTamboMcpPromptList();\n      const servers = useTamboMcpServers();\n      useEffect(() => {\n        if (prompts) {\n          capturedPrompts = prompts;\n        }\n        mcpServersCount = servers.length;\n      }, [prompts, servers]);\n      return null;\n    };\n\n    render(\n      <TamboClientContext.Provider\n        value={{\n          client: { baseURL: \"https://api.tambo.co\" } as any,\n          queryClient,\n          isUpdatingToken: false,\n        }}\n      >\n        <TamboRegistryProvider>\n          <TamboMcpTokenProvider>\n            <TamboMcpProvider\n              mcpServers={[\n                {\n                  url: \"https://server-a.example\",\n                  transport: MCPTransport.SSE,\n                },\n                {\n                  url: \"https://server-b.example\",\n                  transport: MCPTransport.SSE,\n                },\n              ]}\n            >\n              <Capture />\n            </TamboMcpProvider>\n          </TamboMcpTokenProvider>\n        </TamboRegistryProvider>\n      </TamboClientContext.Provider>,\n    );\n\n    // Wait for server A prompts to be loaded\n    // Server B should fail but not block server A\n    await waitFor(() => {\n      expect(capturedPrompts.length).toBe(1);\n      expect(mcpServersCount).toBe(2); // Both servers should be in the list\n    });\n\n    // Verify only server A's prompts are present\n    const promptNames = capturedPrompts.map((p) => p.prompt.name);\n    expect(promptNames).toContain(\"prompt-a\");\n    expect(promptNames).not.toContain(\"prompt-b\");\n  });\n\n  it(\"should add prompts when a new server is added\", async () => {\n    const serverAPrompts = {\n      prompts: [{ name: \"prompt-a\", description: \"Prompt A\" }],\n    };\n    const serverBPrompts = {\n      prompts: [{ name: \"prompt-b\", description: \"Prompt B\" }],\n    };\n\n    const mockClientA = {\n      listTools: jest.fn().mockResolvedValue([]),\n      listPrompts: jest.fn().mockResolvedValue(serverAPrompts),\n      close: jest.fn(),\n    };\n    const mockClientB = {\n      listTools: jest.fn().mockResolvedValue([]),\n      listPrompts: jest.fn().mockResolvedValue(serverBPrompts),\n      close: jest.fn(),\n    };\n\n    const clientA = {\n      client: mockClientA,\n      listTools: jest.fn().mockResolvedValue([]),\n      close: jest.fn(),\n    };\n    const clientB = {\n      client: mockClientB,\n      listTools: jest.fn().mockResolvedValue([]),\n      close: jest.fn(),\n    };\n\n    createImpl.mockImplementation(async (url: string) => {\n      if (url === \"https://server-a.example\") return clientA;\n      if (url === \"https://server-b.example\") return clientB;\n      throw new Error(`Unexpected URL: ${url}`);\n    });\n\n    let capturedPrompts: ListPromptEntry[] = [];\n    const Capture: React.FC = () => {\n      const { data: prompts } = useTamboMcpPromptList();\n      useEffect(() => {\n        if (prompts) {\n          capturedPrompts = prompts;\n        }\n      }, [prompts]);\n      return null;\n    };\n\n    const { rerender } = render(\n      <TamboClientContext.Provider\n        value={{\n          client: { baseURL: \"https://api.tambo.co\" } as any,\n          queryClient,\n          isUpdatingToken: false,\n        }}\n      >\n        <TamboRegistryProvider>\n          <TamboMcpTokenProvider>\n            <TamboMcpProvider\n              mcpServers={[\n                {\n                  url: \"https://server-a.example\",\n                  transport: MCPTransport.SSE,\n                },\n              ]}\n            >\n              <Capture />\n            </TamboMcpProvider>\n          </TamboMcpTokenProvider>\n        </TamboRegistryProvider>\n      </TamboClientContext.Provider>,\n    );\n\n    // Wait for initial prompts from server A\n    await waitFor(() => {\n      expect(capturedPrompts.length).toBe(1);\n    });\n\n    expect(capturedPrompts.map((p) => p.prompt.name)).toContain(\"prompt-a\");\n\n    // Add server B\n    rerender(\n      <TamboClientContext.Provider\n        value={{\n          client: { baseURL: \"https://api.tambo.co\" } as any,\n          queryClient,\n          isUpdatingToken: false,\n        }}\n      >\n        <TamboRegistryProvider>\n          <TamboMcpTokenProvider>\n            <TamboMcpProvider\n              mcpServers={[\n                {\n                  url: \"https://server-a.example\",\n                  transport: MCPTransport.SSE,\n                },\n                {\n                  url: \"https://server-b.example\",\n                  transport: MCPTransport.SSE,\n                },\n              ]}\n            >\n              <Capture />\n            </TamboMcpProvider>\n          </TamboMcpTokenProvider>\n        </TamboRegistryProvider>\n      </TamboClientContext.Provider>,\n    );\n\n    // Wait for server B prompts to be added\n    await waitFor(() => {\n      expect(capturedPrompts.length).toBe(2);\n    });\n\n    const promptNames = capturedPrompts.map((p) => p.prompt.name);\n    expect(promptNames).toContain(\"prompt-a\");\n    expect(promptNames).toContain(\"prompt-b\");\n  });\n});\n"]}

package/esm/mcp/__tests__/tambo-mcp-provider.test.js

@@ -1,8 +1,19 @@
 import { render, waitFor } from "@testing-library/react";
 import React, { useEffect } from "react";
+import { useTamboClient } from "../../providers/tambo-client-provider";
+import { TamboMcpTokenProvider } from "../../providers/tambo-mcp-token-provider";
 import { useTamboRegistry } from "../../providers/tambo-registry-provider";
 import { MCPClient, MCPTransport } from "../mcp-client";
 import { extractErrorMessage, TamboMcpProvider, useTamboMcpServers, } from "../tambo-mcp-provider";
+// Helper to create mock RequestHandlerExtra for testing
+function createMockExtra() {
+    return {
+        signal: new AbortController().signal,
+        requestId: "test-request-id",
+        sendNotification: (async () => { }),
+        sendRequest: (async () => ({ _meta: {} })),
+    };
+}
 // Mock the MCP client to avoid ES module issues
 jest.mock("../mcp-client", () => ({
     MCPClient: jest.fn(),
@@ -15,6 +26,10 @@ jest.mock("../mcp-client", () => ({
 jest.mock("../../providers/tambo-registry-provider", () => ({
     useTamboRegistry: jest.fn(),
 }));
+// Mock the client provider to avoid dependency issues
+jest.mock("../../providers/tambo-client-provider", () => ({
+    useTamboClient: jest.fn(),
+}));
 describe("extractErrorMessage", () => {
     describe("Array content handling", () => {
         it("should extract text from array content with multiple text items", () => {
@@ -120,10 +135,15 @@ describe("TamboMcpProvider server list changes", () => {
         useTamboRegistry.mockReturnValue({
             registerTool: jest.fn(),
         });
-        // Ensure MCPClient.create exists and returns a fake client with listTools
-        MCPClient.create = jest
-            .fn()
-            .mockResolvedValue({ listTools: jest.fn().mockResolvedValue([]) });
+        // Mock client with baseURL
+        useTamboClient.mockReturnValue({
+            baseURL: "https://api.tambo.co",
+        });
+        // Ensure MCPClient.create exists and returns a fake client with listTools and close
+        MCPClient.create = jest.fn().mockResolvedValue({
+            listTools: jest.fn().mockResolvedValue([]),
+            close: jest.fn(),
+        });
     });
     const Capture = ({ onUpdate, }) => {
         const servers = useTamboMcpServers();
@@ -134,15 +154,17 @@ describe("TamboMcpProvider server list changes", () => {
     };
     it("adds a new server when the list grows", async () => {
         let latest = [];
-        const { rerender, getByTestId } = render(React.createElement(TamboMcpProvider, { mcpServers: ["https://a.example"] },
-            React.createElement(Capture, { onUpdate: (s) => (latest = s) })));
+        const { rerender, getByTestId } = render(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: ["https://a.example"] },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
         // Wait for initial connection
         await waitFor(() => {
             expect(latest.length).toBe(1);
         });
         // Add new server
-        rerender(React.createElement(TamboMcpProvider, { mcpServers: ["https://a.example", "https://b.example"] },
-            React.createElement(Capture, { onUpdate: (s) => (latest = s) })));
+        rerender(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: ["https://a.example", "https://b.example"] },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
         await waitFor(() => {
             expect(latest.length).toBe(2);
             const urls = getByTestId("urls").textContent || "";
@@ -152,14 +174,16 @@ describe("TamboMcpProvider server list changes", () => {
     });
     it("removes a server when the list shrinks", async () => {
         let latest = [];
-        const { rerender, getByTestId } = render(React.createElement(TamboMcpProvider, { mcpServers: ["https://a.example", "https://b.example"] },
-            React.createElement(Capture, { onUpdate: (s) => (latest = s) })));
+        const { rerender, getByTestId } = render(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: ["https://a.example", "https://b.example"] },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
         await waitFor(() => {
             expect(latest.length).toBe(2);
         });
         // Remove one server
-        rerender(React.createElement(TamboMcpProvider, { mcpServers: ["https://a.example"] },
-            React.createElement(Capture, { onUpdate: (s) => (latest = s) })));
+        rerender(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: ["https://a.example"] },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
         await waitFor(() => {
             expect(latest.length).toBe(1);
             const urls = getByTestId("urls").textContent || "";
@@ -173,8 +197,9 @@ describe("TamboMcpProvider server list changes", () => {
             { url: "https://a.example", transport: MCPTransport.SSE },
             { url: "https://b.example", transport: MCPTransport.SSE },
         ];
-        const { rerender } = render(React.createElement(TamboMcpProvider, { mcpServers: initial },
-            React.createElement(Capture, { onUpdate: (s) => (latest = s) })));
+        const { rerender } = render(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: initial },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
         await waitFor(() => {
             expect(latest.length).toBe(2);
         });
@@ -183,13 +208,333 @@ describe("TamboMcpProvider server list changes", () => {
             { url: "https://a.example", transport: MCPTransport.SSE },
             { url: "https://b.example", transport: MCPTransport.SSE },
         ];
-        rerender(React.createElement(TamboMcpProvider, { mcpServers: same },
-            React.createElement(Capture, { onUpdate: (s) => (latest = s) })));
+        rerender(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: same },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
         await waitFor(() => {
             expect(latest.length).toBe(2);
             const urls = latest.map((s) => s.url).sort();
             expect(urls).toEqual(["https://a.example", "https://b.example"].sort());
         });
     });
+    it("reuses client when same server is passed with new array instance", async () => {
+        const createSpy = jest.fn().mockResolvedValue({
+            listTools: jest.fn().mockResolvedValue([]),
+            close: jest.fn(),
+        });
+        MCPClient.create = createSpy;
+        let latest = [];
+        const initial = [{ url: "https://a.example", transport: MCPTransport.SSE }];
+        const { rerender } = render(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: initial },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
+        await waitFor(() => {
+            expect(latest.length).toBe(1);
+        });
+        // Verify client was created once
+        expect(createSpy).toHaveBeenCalledTimes(1);
+        const firstClient = latest[0].client;
+        // Pass a new array with same server
+        const same = [{ url: "https://a.example", transport: MCPTransport.SSE }];
+        rerender(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: same },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
+        await waitFor(() => {
+            expect(latest.length).toBe(1);
+        });
+        // Client should NOT have been created again
+        expect(createSpy).toHaveBeenCalledTimes(1);
+        // Should be the exact same client instance
+        expect(latest[0].client).toBe(firstClient);
+    });
+    it("calls close() on removed server clients", async () => {
+        const closeSpy = jest.fn();
+        const createSpy = jest.fn().mockResolvedValue({
+            listTools: jest.fn().mockResolvedValue([]),
+            close: closeSpy,
+        });
+        MCPClient.create = createSpy;
+        let latest = [];
+        const { rerender } = render(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: ["https://a.example", "https://b.example"] },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
+        await waitFor(() => {
+            expect(latest.length).toBe(2);
+        });
+        expect(closeSpy).not.toHaveBeenCalled();
+        // Remove one server
+        rerender(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: ["https://a.example"] },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
+        await waitFor(() => {
+            expect(latest.length).toBe(1);
+        });
+        // close() should have been called once for the removed server
+        expect(closeSpy).toHaveBeenCalledTimes(1);
+    });
+    it("calls close() on all clients when provider unmounts", async () => {
+        const closeSpy = jest.fn();
+        const createSpy = jest.fn().mockResolvedValue({
+            listTools: jest.fn().mockResolvedValue([]),
+            close: closeSpy,
+        });
+        MCPClient.create = createSpy;
+        let latest = [];
+        const { unmount } = render(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: ["https://a.example", "https://b.example"] },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
+        await waitFor(() => {
+            expect(latest.length).toBe(2);
+        });
+        expect(closeSpy).not.toHaveBeenCalled();
+        // Unmount the provider
+        unmount();
+        // close() should have been called for both clients
+        expect(closeSpy).toHaveBeenCalledTimes(2);
+    });
+    it("creates new client when customHeaders change", async () => {
+        const closeSpy = jest.fn();
+        let clientIdCounter = 0;
+        const createSpy = jest.fn().mockImplementation(async () => {
+            const id = ++clientIdCounter;
+            return {
+                id, // Add an ID so we can track which client is which
+                listTools: jest.fn().mockResolvedValue([]),
+                close: closeSpy,
+            };
+        });
+        MCPClient.create = createSpy;
+        let latest = [];
+        const { rerender } = render(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: [
+                    {
+                        url: "https://a.example",
+                        transport: MCPTransport.SSE,
+                        customHeaders: { Authorization: "Bearer token1" },
+                    },
+                ] },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
+        await waitFor(() => {
+            expect(latest.length).toBe(1);
+            expect(createSpy).toHaveBeenCalledTimes(1);
+        });
+        const firstClientId = latest[0].client?.id;
+        expect(firstClientId).toBe(1);
+        // Change the customHeaders
+        rerender(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: [
+                    {
+                        url: "https://a.example",
+                        transport: MCPTransport.SSE,
+                        customHeaders: { Authorization: "Bearer token2" },
+                    },
+                ] },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
+        // Wait for old client to be closed and new client to be created
+        await waitFor(() => {
+            expect(closeSpy).toHaveBeenCalledTimes(1);
+            expect(createSpy).toHaveBeenCalledTimes(2);
+            expect(latest.length).toBe(1);
+            const newClientId = latest[0].client?.id;
+            expect(newClientId).toBe(2); // Should be the new client
+        }, { timeout: 3000 });
+    });
+    // Note: Token changes for the internal Tambo server are covered by the
+    // "creates new client when customHeaders change" test above, since token
+    // changes result in different Authorization headers, which trigger client recreation.
+});
+describe("TamboMcpProvider handler support", () => {
+    let mockClient;
+    let createSpy;
+    beforeEach(() => {
+        // Mock registry so tool registration is a no-op
+        useTamboRegistry.mockReturnValue({
+            registerTool: jest.fn(),
+        });
+        // Mock client with baseURL
+        useTamboClient.mockReturnValue({
+            baseURL: "https://api.tambo.co",
+        });
+        // Create a mock client with update methods
+        mockClient = {
+            listTools: jest.fn().mockResolvedValue([]),
+            updateElicitationHandler: jest.fn(),
+            updateSamplingHandler: jest.fn(),
+            close: jest.fn(),
+        };
+        // Mock MCPClient.create to return our mock client
+        createSpy = jest.fn().mockResolvedValue(mockClient);
+        MCPClient.create = createSpy;
+    });
+    const Capture = ({ onUpdate, }) => {
+        const servers = useTamboMcpServers();
+        useEffect(() => {
+            onUpdate(servers);
+        }, [servers, onUpdate]);
+        return null;
+    };
+    it("should pass provider-level elicitation handler to MCPClient.create", async () => {
+        const mockElicitationHandler = jest.fn().mockResolvedValue({
+            content: [{ type: "text", text: "test" }],
+        });
+        const handlers = {
+            elicitation: mockElicitationHandler,
+        };
+        let latest = [];
+        render(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: ["https://test.example"], handlers: handlers },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
+        await waitFor(() => {
+            expect(latest.length).toBe(1);
+        });
+        // Verify MCPClient.create was called
+        expect(createSpy).toHaveBeenCalledWith("https://test.example", MCPTransport.SSE, undefined, undefined, undefined, expect.objectContaining({
+            elicitation: expect.any(Function),
+        }));
+        // Get the actual handler that was passed
+        const passedHandlers = createSpy.mock.calls[0][5];
+        expect(passedHandlers.elicitation).toBeDefined();
+        // Call the wrapped handler and verify it receives serverInfo
+        const mockRequest = {
+            method: "sampling/createMessage",
+            params: {},
+        };
+        const mockExtra = createMockExtra();
+        await passedHandlers.elicitation(mockRequest, mockExtra);
+        expect(mockElicitationHandler).toHaveBeenCalledWith(mockRequest, mockExtra, expect.objectContaining({
+            url: "https://test.example",
+            transport: MCPTransport.SSE,
+        }));
+    });
+    it("should pass provider-level sampling handler to MCPClient.create", async () => {
+        const mockSamplingHandler = jest.fn().mockResolvedValue({
+            model: "test-model",
+            stopReason: "endTurn",
+            role: "assistant",
+            content: { type: "text", text: "response" },
+        });
+        const handlers = {
+            sampling: mockSamplingHandler,
+        };
+        let latest = [];
+        render(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: ["https://test.example"], handlers: handlers },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
+        await waitFor(() => {
+            expect(latest.length).toBe(1);
+        });
+        // Verify MCPClient.create was called with sampling handler
+        expect(createSpy).toHaveBeenCalledWith("https://test.example", MCPTransport.SSE, undefined, undefined, undefined, expect.objectContaining({
+            sampling: expect.any(Function),
+        }));
+        // Get the actual handler and verify it receives serverInfo
+        const passedHandlers = createSpy.mock.calls[0][5];
+        const mockRequest = {
+            method: "sampling/createMessage",
+            params: {
+                messages: [],
+                modelPreferences: {},
+            },
+        };
+        const mockExtra = createMockExtra();
+        await passedHandlers.sampling(mockRequest, mockExtra);
+        expect(mockSamplingHandler).toHaveBeenCalledWith(mockRequest, mockExtra, expect.objectContaining({
+            url: "https://test.example",
+            transport: MCPTransport.SSE,
+        }));
+    });
+    it("should allow per-server handlers to override provider-level handlers", async () => {
+        const providerElicitationHandler = jest.fn().mockResolvedValue({
+            content: [{ type: "text", text: "provider" }],
+        });
+        const serverElicitationHandler = jest.fn().mockResolvedValue({
+            content: [{ type: "text", text: "server" }],
+        });
+        const handlers = {
+            elicitation: providerElicitationHandler,
+        };
+        let latest = [];
+        render(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: [
+                    {
+                        url: "https://test.example",
+                        handlers: {
+                            elicitation: serverElicitationHandler,
+                        },
+                    },
+                ], handlers: handlers },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
+        await waitFor(() => {
+            expect(latest.length).toBe(1);
+        });
+        // Verify the per-server handler was used, not the provider handler
+        const passedHandlers = createSpy.mock.calls[0][5];
+        expect(passedHandlers.elicitation).toBe(serverElicitationHandler);
+    });
+    it("should pass different serverInfo to handlers for different servers", async () => {
+        const mockElicitationHandler = jest.fn().mockResolvedValue({
+            content: [{ type: "text", text: "test" }],
+        });
+        const handlers = {
+            elicitation: mockElicitationHandler,
+        };
+        let latest = [];
+        render(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: [
+                    { url: "https://server-a.example", name: "Server A" },
+                    { url: "https://server-b.example", name: "Server B" },
+                ], handlers: handlers },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
+        await waitFor(() => {
+            expect(latest.length).toBe(2);
+        });
+        // Verify both servers got handlers
+        expect(createSpy).toHaveBeenCalledTimes(2);
+        // Get handlers for both servers
+        const serverAHandlers = createSpy.mock.calls[0][5];
+        const serverBHandlers = createSpy.mock.calls[1][5];
+        const mockRequest = {
+            method: "sampling/createMessage",
+            params: {},
+        };
+        const mockExtra = createMockExtra();
+        // Call handler for server A
+        await serverAHandlers.elicitation(mockRequest, mockExtra);
+        expect(mockElicitationHandler).toHaveBeenCalledWith(mockRequest, mockExtra, expect.objectContaining({
+            url: "https://server-a.example",
+            name: "Server A",
+        }));
+        mockElicitationHandler.mockClear();
+        // Call handler for server B
+        await serverBHandlers.elicitation(mockRequest, mockExtra);
+        expect(mockElicitationHandler).toHaveBeenCalledWith(mockRequest, mockExtra, expect.objectContaining({
+            url: "https://server-b.example",
+            name: "Server B",
+        }));
+    });
+    it("should update handlers when provider handlers change", async () => {
+        const initialHandler = jest.fn().mockResolvedValue({
+            content: [{ type: "text", text: "initial" }],
+        });
+        const updatedHandler = jest.fn().mockResolvedValue({
+            content: [{ type: "text", text: "updated" }],
+        });
+        let latest = [];
+        const { rerender } = render(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: ["https://test.example"], handlers: { elicitation: initialHandler } },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
+        await waitFor(() => {
+            expect(latest.length).toBe(1);
+        });
+        // Update the handlers
+        rerender(React.createElement(TamboMcpTokenProvider, null,
+            React.createElement(TamboMcpProvider, { mcpServers: ["https://test.example"], handlers: { elicitation: updatedHandler } },
+                React.createElement(Capture, { onUpdate: (s) => (latest = s) }))));
+        await waitFor(() => {
+            expect(mockClient.updateElicitationHandler).toHaveBeenCalled();
+        });
+        // Verify the handler was updated
+        expect(mockClient.updateElicitationHandler).toHaveBeenCalledWith(expect.any(Function));
+    });
 });
 //# sourceMappingURL=tambo-mcp-provider.test.js.map