@tambo-ai/react 0.49.0 → 0.53.0

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 (141) hide show
  1. package/dist/hooks/__tests__/use-component-state.test.js +4 -1
  2. package/dist/hooks/__tests__/use-component-state.test.js.map +1 -1
  3. package/dist/hooks/__tests__/use-message-images.test.d.ts +2 -0
  4. package/dist/hooks/__tests__/use-message-images.test.d.ts.map +1 -0
  5. package/dist/hooks/__tests__/use-message-images.test.js +66 -0
  6. package/dist/hooks/__tests__/use-message-images.test.js.map +1 -0
  7. package/dist/hooks/__tests__/use-tambo-threads.test.js +1 -1
  8. package/dist/hooks/__tests__/use-tambo-threads.test.js.map +1 -1
  9. package/dist/hooks/use-component-state.js +3 -3
  10. package/dist/hooks/use-component-state.js.map +1 -1
  11. package/dist/hooks/use-message-images.d.ts +25 -0
  12. package/dist/hooks/use-message-images.d.ts.map +1 -0
  13. package/dist/hooks/use-message-images.js +66 -0
  14. package/dist/hooks/use-message-images.js.map +1 -0
  15. package/dist/hooks/use-suggestions.js +4 -2
  16. package/dist/hooks/use-suggestions.js.map +1 -1
  17. package/dist/index.d.ts +1 -0
  18. package/dist/index.d.ts.map +1 -1
  19. package/dist/index.js +3 -1
  20. package/dist/index.js.map +1 -1
  21. package/dist/mcp/__tests__/mcp-client.test.d.ts +2 -0
  22. package/dist/mcp/__tests__/mcp-client.test.d.ts.map +1 -0
  23. package/dist/mcp/__tests__/mcp-client.test.js +502 -0
  24. package/dist/mcp/__tests__/mcp-client.test.js.map +1 -0
  25. package/dist/mcp/mcp-client.d.ts +77 -3
  26. package/dist/mcp/mcp-client.d.ts.map +1 -1
  27. package/dist/mcp/mcp-client.js +184 -19
  28. package/dist/mcp/mcp-client.js.map +1 -1
  29. package/dist/mcp/tambo-mcp-provider.d.ts +1 -0
  30. package/dist/mcp/tambo-mcp-provider.d.ts.map +1 -1
  31. package/dist/mcp/tambo-mcp-provider.js +1 -0
  32. package/dist/mcp/tambo-mcp-provider.js.map +1 -1
  33. package/dist/model/tambo-interactable.d.ts +1 -1
  34. package/dist/model/tambo-interactable.d.ts.map +1 -1
  35. package/dist/model/tambo-interactable.js.map +1 -1
  36. package/dist/providers/__tests__/tambo-interactable-provider-partial-updates.test.d.ts +2 -0
  37. package/dist/providers/__tests__/tambo-interactable-provider-partial-updates.test.d.ts.map +1 -0
  38. package/dist/providers/__tests__/tambo-interactable-provider-partial-updates.test.js +677 -0
  39. package/dist/providers/__tests__/tambo-interactable-provider-partial-updates.test.js.map +1 -0
  40. package/dist/providers/__tests__/tambo-stubs.test.js +6 -2
  41. package/dist/providers/__tests__/tambo-stubs.test.js.map +1 -1
  42. package/dist/providers/__tests__/tambo-thread-provider.test.js +14 -14
  43. package/dist/providers/__tests__/tambo-thread-provider.test.js.map +1 -1
  44. package/dist/providers/tambo-interactable-provider.d.ts.map +1 -1
  45. package/dist/providers/tambo-interactable-provider.js +26 -24
  46. package/dist/providers/tambo-interactable-provider.js.map +1 -1
  47. package/dist/providers/tambo-provider.d.ts +1 -0
  48. package/dist/providers/tambo-provider.d.ts.map +1 -1
  49. package/dist/providers/tambo-provider.js +1 -0
  50. package/dist/providers/tambo-provider.js.map +1 -1
  51. package/dist/providers/tambo-thread-input-provider.d.ts +11 -0
  52. package/dist/providers/tambo-thread-input-provider.d.ts.map +1 -1
  53. package/dist/providers/tambo-thread-input-provider.js +63 -12
  54. package/dist/providers/tambo-thread-input-provider.js.map +1 -1
  55. package/dist/providers/tambo-thread-provider.d.ts +1 -0
  56. package/dist/providers/tambo-thread-provider.d.ts.map +1 -1
  57. package/dist/providers/tambo-thread-provider.js +9 -5
  58. package/dist/providers/tambo-thread-provider.js.map +1 -1
  59. package/dist/setupTests.d.ts +0 -1
  60. package/dist/setupTests.d.ts.map +1 -1
  61. package/dist/setupTests.js +0 -1
  62. package/dist/setupTests.js.map +1 -1
  63. package/dist/util/__tests__/message-builder.test.d.ts +2 -0
  64. package/dist/util/__tests__/message-builder.test.d.ts.map +1 -0
  65. package/dist/util/__tests__/message-builder.test.js +191 -0
  66. package/dist/util/__tests__/message-builder.test.js.map +1 -0
  67. package/dist/util/message-builder.d.ts +10 -0
  68. package/dist/util/message-builder.d.ts.map +1 -0
  69. package/dist/util/message-builder.js +31 -0
  70. package/dist/util/message-builder.js.map +1 -0
  71. package/esm/hooks/__tests__/use-component-state.test.js +4 -1
  72. package/esm/hooks/__tests__/use-component-state.test.js.map +1 -1
  73. package/esm/hooks/__tests__/use-message-images.test.d.ts +2 -0
  74. package/esm/hooks/__tests__/use-message-images.test.d.ts.map +1 -0
  75. package/esm/hooks/__tests__/use-message-images.test.js +64 -0
  76. package/esm/hooks/__tests__/use-message-images.test.js.map +1 -0
  77. package/esm/hooks/__tests__/use-tambo-threads.test.js +1 -1
  78. package/esm/hooks/__tests__/use-tambo-threads.test.js.map +1 -1
  79. package/esm/hooks/use-component-state.js +3 -3
  80. package/esm/hooks/use-component-state.js.map +1 -1
  81. package/esm/hooks/use-message-images.d.ts +25 -0
  82. package/esm/hooks/use-message-images.d.ts.map +1 -0
  83. package/esm/hooks/use-message-images.js +63 -0
  84. package/esm/hooks/use-message-images.js.map +1 -0
  85. package/esm/hooks/use-suggestions.js +4 -2
  86. package/esm/hooks/use-suggestions.js.map +1 -1
  87. package/esm/index.d.ts +1 -0
  88. package/esm/index.d.ts.map +1 -1
  89. package/esm/index.js +1 -0
  90. package/esm/index.js.map +1 -1
  91. package/esm/mcp/__tests__/mcp-client.test.d.ts +2 -0
  92. package/esm/mcp/__tests__/mcp-client.test.d.ts.map +1 -0
  93. package/esm/mcp/__tests__/mcp-client.test.js +500 -0
  94. package/esm/mcp/__tests__/mcp-client.test.js.map +1 -0
  95. package/esm/mcp/mcp-client.d.ts +77 -3
  96. package/esm/mcp/mcp-client.d.ts.map +1 -1
  97. package/esm/mcp/mcp-client.js +184 -19
  98. package/esm/mcp/mcp-client.js.map +1 -1
  99. package/esm/mcp/tambo-mcp-provider.d.ts +1 -0
  100. package/esm/mcp/tambo-mcp-provider.d.ts.map +1 -1
  101. package/esm/mcp/tambo-mcp-provider.js +1 -0
  102. package/esm/mcp/tambo-mcp-provider.js.map +1 -1
  103. package/esm/model/tambo-interactable.d.ts +1 -1
  104. package/esm/model/tambo-interactable.d.ts.map +1 -1
  105. package/esm/model/tambo-interactable.js.map +1 -1
  106. package/esm/providers/__tests__/tambo-interactable-provider-partial-updates.test.d.ts +2 -0
  107. package/esm/providers/__tests__/tambo-interactable-provider-partial-updates.test.d.ts.map +1 -0
  108. package/esm/providers/__tests__/tambo-interactable-provider-partial-updates.test.js +672 -0
  109. package/esm/providers/__tests__/tambo-interactable-provider-partial-updates.test.js.map +1 -0
  110. package/esm/providers/__tests__/tambo-stubs.test.js +6 -2
  111. package/esm/providers/__tests__/tambo-stubs.test.js.map +1 -1
  112. package/esm/providers/__tests__/tambo-thread-provider.test.js +14 -14
  113. package/esm/providers/__tests__/tambo-thread-provider.test.js.map +1 -1
  114. package/esm/providers/tambo-interactable-provider.d.ts.map +1 -1
  115. package/esm/providers/tambo-interactable-provider.js +26 -24
  116. package/esm/providers/tambo-interactable-provider.js.map +1 -1
  117. package/esm/providers/tambo-provider.d.ts +1 -0
  118. package/esm/providers/tambo-provider.d.ts.map +1 -1
  119. package/esm/providers/tambo-provider.js +1 -0
  120. package/esm/providers/tambo-provider.js.map +1 -1
  121. package/esm/providers/tambo-thread-input-provider.d.ts +11 -0
  122. package/esm/providers/tambo-thread-input-provider.d.ts.map +1 -1
  123. package/esm/providers/tambo-thread-input-provider.js +63 -12
  124. package/esm/providers/tambo-thread-input-provider.js.map +1 -1
  125. package/esm/providers/tambo-thread-provider.d.ts +1 -0
  126. package/esm/providers/tambo-thread-provider.d.ts.map +1 -1
  127. package/esm/providers/tambo-thread-provider.js +9 -5
  128. package/esm/providers/tambo-thread-provider.js.map +1 -1
  129. package/esm/setupTests.d.ts +0 -1
  130. package/esm/setupTests.d.ts.map +1 -1
  131. package/esm/setupTests.js +0 -1
  132. package/esm/setupTests.js.map +1 -1
  133. package/esm/util/__tests__/message-builder.test.d.ts +2 -0
  134. package/esm/util/__tests__/message-builder.test.d.ts.map +1 -0
  135. package/esm/util/__tests__/message-builder.test.js +189 -0
  136. package/esm/util/__tests__/message-builder.test.js.map +1 -0
  137. package/esm/util/message-builder.d.ts +10 -0
  138. package/esm/util/message-builder.d.ts.map +1 -0
  139. package/esm/util/message-builder.js +28 -0
  140. package/esm/util/message-builder.js.map +1 -0
  141. package/package.json +7 -7
@@ -0,0 +1,500 @@
1
+ import { MCPClient, MCPTransport } from "../mcp-client";
2
+ // Mock the MCP SDK modules
3
+ jest.mock("@modelcontextprotocol/sdk/client/index.js", () => ({
4
+ Client: jest.fn().mockImplementation(() => ({
5
+ connect: jest.fn(),
6
+ close: jest.fn(),
7
+ listTools: jest.fn(),
8
+ callTool: jest.fn(),
9
+ onclose: null,
10
+ })),
11
+ }));
12
+ jest.mock("@modelcontextprotocol/sdk/client/sse.js", () => ({
13
+ SSEClientTransport: jest.fn().mockImplementation(() => ({
14
+ // SSE transport doesn't have sessionId
15
+ })),
16
+ }));
17
+ jest.mock("@modelcontextprotocol/sdk/client/streamableHttp.js", () => ({
18
+ StreamableHTTPClientTransport: jest
19
+ .fn()
20
+ .mockImplementation((url, options) => ({
21
+ sessionId: options?.sessionId,
22
+ })),
23
+ }));
24
+ import { Client } from "@modelcontextprotocol/sdk/client/index.js";
25
+ import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
26
+ import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
27
+ // Type the mocked modules
28
+ const MockedClient = Client;
29
+ const MockedSSEClientTransport = SSEClientTransport;
30
+ const MockedStreamableHTTPClientTransport = StreamableHTTPClientTransport;
31
+ describe("MCPClient", () => {
32
+ let mockClientInstance;
33
+ let mockTransportInstance;
34
+ beforeEach(() => {
35
+ jest.clearAllMocks();
36
+ // Create mock instances
37
+ mockClientInstance = {
38
+ connect: jest.fn().mockResolvedValue(undefined),
39
+ close: jest.fn().mockResolvedValue(undefined),
40
+ listTools: jest.fn(),
41
+ callTool: jest.fn(),
42
+ onclose: null,
43
+ };
44
+ mockTransportInstance = {
45
+ sessionId: "test-session-id",
46
+ };
47
+ // Setup mocks
48
+ MockedClient.mockImplementation(() => mockClientInstance);
49
+ MockedStreamableHTTPClientTransport.mockImplementation(() => mockTransportInstance);
50
+ MockedSSEClientTransport.mockImplementation(() => ({}));
51
+ });
52
+ describe("create", () => {
53
+ it("should create and connect an MCPClient with HTTP transport by default", async () => {
54
+ const endpoint = "https://api.example.com/mcp";
55
+ const headers = { Authorization: "Bearer token" };
56
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP, headers);
57
+ expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(new URL(endpoint), { sessionId: undefined, requestInit: { headers } });
58
+ expect(MockedClient).toHaveBeenCalledWith({
59
+ name: "tambo-mcp-client",
60
+ version: "1.0.0",
61
+ });
62
+ expect(mockClientInstance.connect).toHaveBeenCalledWith(mockTransportInstance);
63
+ expect(client).toBeInstanceOf(MCPClient);
64
+ });
65
+ it("should create and connect an MCPClient with SSE transport", async () => {
66
+ const endpoint = "https://api.example.com/mcp";
67
+ const client = await MCPClient.create(endpoint, MCPTransport.SSE);
68
+ expect(MockedSSEClientTransport).toHaveBeenCalledWith(new URL(endpoint), {
69
+ requestInit: { headers: {} },
70
+ });
71
+ expect(mockClientInstance.connect).toHaveBeenCalledWith({});
72
+ expect(client).toBeInstanceOf(MCPClient);
73
+ });
74
+ it("should create client with default headers when none provided", async () => {
75
+ const endpoint = "https://api.example.com/mcp";
76
+ await MCPClient.create(endpoint);
77
+ expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(new URL(endpoint), { sessionId: undefined, requestInit: { headers: {} } });
78
+ });
79
+ });
80
+ describe("reconnect", () => {
81
+ it("should create new transport and client instances and call connect when reconnect() is called (default behavior)", async () => {
82
+ const endpoint = "https://api.example.com/mcp";
83
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP);
84
+ // Clear previous calls to focus on reconnect behavior
85
+ jest.clearAllMocks();
86
+ // Create new mock instances to verify new instances are created
87
+ const newMockClientInstance = {
88
+ connect: jest.fn().mockResolvedValue(undefined),
89
+ close: jest.fn().mockResolvedValue(undefined),
90
+ listTools: jest.fn(),
91
+ callTool: jest.fn(),
92
+ onclose: null,
93
+ };
94
+ const newMockTransportInstance = {
95
+ sessionId: "new-session-id",
96
+ };
97
+ // Mock the constructors to return new instances
98
+ MockedClient.mockImplementation(() => newMockClientInstance);
99
+ MockedStreamableHTTPClientTransport.mockImplementation(() => newMockTransportInstance);
100
+ await client.reconnect(); // Uses default parameters
101
+ // Verify old client was closed
102
+ expect(mockClientInstance.close).toHaveBeenCalled();
103
+ // Verify new transport was created with preserved session ID (default behavior)
104
+ expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(new URL(endpoint), { sessionId: "test-session-id", requestInit: { headers: {} } });
105
+ // Verify new client was created
106
+ expect(MockedClient).toHaveBeenCalledWith({
107
+ name: "tambo-mcp-client",
108
+ version: "1.0.0",
109
+ });
110
+ // Verify new client's connect was called with new transport
111
+ expect(newMockClientInstance.connect).toHaveBeenCalledWith(newMockTransportInstance);
112
+ });
113
+ it("should reconnect without session ID for SSE transport", async () => {
114
+ const endpoint = "https://api.example.com/mcp";
115
+ const client = await MCPClient.create(endpoint, MCPTransport.SSE);
116
+ // Clear previous calls
117
+ jest.clearAllMocks();
118
+ await client.reconnect();
119
+ expect(mockClientInstance.close).toHaveBeenCalled();
120
+ expect(MockedSSEClientTransport).toHaveBeenCalledWith(new URL(endpoint), {
121
+ requestInit: { headers: {} },
122
+ });
123
+ expect(mockClientInstance.connect).toHaveBeenCalledWith({});
124
+ });
125
+ it("should handle close errors when reportErrorOnClose is true", async () => {
126
+ const endpoint = "https://api.example.com/mcp";
127
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP);
128
+ const consoleSpy = jest.spyOn(console, "error").mockImplementation();
129
+ // Make close throw an error
130
+ mockClientInstance.close.mockRejectedValue(new Error("Close failed"));
131
+ await client.reconnect(false, true);
132
+ expect(consoleSpy).toHaveBeenCalledWith("Error closing Tambo MCP Client:", expect.any(Error));
133
+ consoleSpy.mockRestore();
134
+ });
135
+ it("should not log close errors when reportErrorOnClose is false", async () => {
136
+ const endpoint = "https://api.example.com/mcp";
137
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP);
138
+ const consoleSpy = jest.spyOn(console, "error").mockImplementation();
139
+ // Make close throw an error
140
+ mockClientInstance.close.mockRejectedValue(new Error("Close failed"));
141
+ await client.reconnect(false, false);
142
+ expect(consoleSpy).not.toHaveBeenCalled();
143
+ consoleSpy.mockRestore();
144
+ });
145
+ it("should create new session when newSession is true", async () => {
146
+ const endpoint = "https://api.example.com/mcp";
147
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP);
148
+ // Clear previous calls to focus on reconnect behavior
149
+ jest.clearAllMocks();
150
+ // Create new mock instances to verify new instances are created
151
+ const newMockClientInstance = {
152
+ connect: jest.fn().mockResolvedValue(undefined),
153
+ close: jest.fn().mockResolvedValue(undefined),
154
+ listTools: jest.fn(),
155
+ callTool: jest.fn(),
156
+ onclose: null,
157
+ };
158
+ const newMockTransportInstance = {
159
+ sessionId: "new-session-id",
160
+ };
161
+ // Mock the constructors to return new instances
162
+ MockedClient.mockImplementation(() => newMockClientInstance);
163
+ MockedStreamableHTTPClientTransport.mockImplementation(() => newMockTransportInstance);
164
+ await client.reconnect(true, true); // newSession = true, reportErrorOnClose = true
165
+ // Verify old client was closed
166
+ expect(mockClientInstance.close).toHaveBeenCalled();
167
+ // Verify new transport was created with undefined session ID (new session)
168
+ expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(new URL(endpoint), { sessionId: undefined, requestInit: { headers: {} } });
169
+ // Verify new client was created
170
+ expect(MockedClient).toHaveBeenCalledWith({
171
+ name: "tambo-mcp-client",
172
+ version: "1.0.0",
173
+ });
174
+ // Verify new client's connect was called with new transport
175
+ expect(newMockClientInstance.connect).toHaveBeenCalledWith(newMockTransportInstance);
176
+ });
177
+ it("should reuse existing session when newSession is false (default)", async () => {
178
+ const endpoint = "https://api.example.com/mcp";
179
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP);
180
+ // Clear previous calls to focus on reconnect behavior
181
+ jest.clearAllMocks();
182
+ // Create new mock instances to verify new instances are created
183
+ const newMockClientInstance = {
184
+ connect: jest.fn().mockResolvedValue(undefined),
185
+ close: jest.fn().mockResolvedValue(undefined),
186
+ listTools: jest.fn(),
187
+ callTool: jest.fn(),
188
+ onclose: null,
189
+ };
190
+ const newMockTransportInstance = {
191
+ sessionId: "reused-session-id",
192
+ };
193
+ // Mock the constructors to return new instances
194
+ MockedClient.mockImplementation(() => newMockClientInstance);
195
+ MockedStreamableHTTPClientTransport.mockImplementation(() => newMockTransportInstance);
196
+ await client.reconnect(false, true); // newSession = false, reportErrorOnClose = true
197
+ // Verify old client was closed
198
+ expect(mockClientInstance.close).toHaveBeenCalled();
199
+ // Verify new transport was created with preserved session ID
200
+ expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(new URL(endpoint), { sessionId: "test-session-id", requestInit: { headers: {} } });
201
+ // Verify new client was created
202
+ expect(MockedClient).toHaveBeenCalledWith({
203
+ name: "tambo-mcp-client",
204
+ version: "1.0.0",
205
+ });
206
+ // Verify new client's connect was called with new transport
207
+ expect(newMockClientInstance.connect).toHaveBeenCalledWith(newMockTransportInstance);
208
+ });
209
+ });
210
+ describe("onclose", () => {
211
+ it("should reconnect MCPClient when client is closed by external means (no backoff on manual preemption)", async () => {
212
+ const endpoint = "https://api.example.com/mcp";
213
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP);
214
+ jest.useFakeTimers();
215
+ const consoleSpy = jest.spyOn(console, "warn").mockImplementation();
216
+ // Create new mock instances to verify reconnection creates new instances
217
+ const newMockClientInstance = {
218
+ connect: jest.fn().mockResolvedValue(undefined),
219
+ close: jest.fn().mockResolvedValue(undefined),
220
+ listTools: jest.fn(),
221
+ callTool: jest.fn(),
222
+ onclose: null,
223
+ };
224
+ const newMockTransportInstance = {
225
+ sessionId: "reconnected-session-id",
226
+ };
227
+ // Mock the constructors to return new instances for reconnection
228
+ MockedClient.mockImplementation(() => newMockClientInstance);
229
+ MockedStreamableHTTPClientTransport.mockImplementation(() => newMockTransportInstance);
230
+ // Reset counts after initial creation
231
+ jest.clearAllMocks();
232
+ // Trigger automatic onclose (schedules a delayed reconnect)
233
+ client.onclose();
234
+ // Manual reconnect should preempt the scheduled automatic attempt
235
+ const reconnectPromise = client.reconnect();
236
+ // No timers should be pending after manual preemption
237
+ await reconnectPromise;
238
+ // Verify warning message is logged
239
+ expect(consoleSpy).toHaveBeenCalled();
240
+ // Verify old client was closed
241
+ expect(mockClientInstance.close).toHaveBeenCalled();
242
+ // Verify new transport was created with preserved session ID
243
+ expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(new URL(endpoint), { sessionId: "test-session-id", requestInit: { headers: {} } });
244
+ // Verify new client was created
245
+ expect(MockedClient).toHaveBeenCalledWith({
246
+ name: "tambo-mcp-client",
247
+ version: "1.0.0",
248
+ });
249
+ // Verify new client's connect was called with new transport
250
+ expect(newMockClientInstance.connect).toHaveBeenCalledWith(newMockTransportInstance);
251
+ // Ensure only a single reconnect attempt occurred
252
+ expect(MockedClient).toHaveBeenCalledTimes(1);
253
+ expect(newMockClientInstance.connect).toHaveBeenCalledTimes(1);
254
+ consoleSpy.mockRestore();
255
+ jest.useRealTimers();
256
+ });
257
+ });
258
+ describe("reconnect re-entrancy and single-flight", () => {
259
+ it("prevents re-entrant onclose during deliberate close and coalesces concurrent calls", async () => {
260
+ const endpoint = "https://api.example.com/mcp";
261
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP);
262
+ // Simulate an implementation where closing the client would call its own onclose handler
263
+ const closeImpl = jest.fn(async () => {
264
+ if (typeof mockClientInstance.onclose === "function") {
265
+ // would cause recursion if not detached
266
+ mockClientInstance.onclose();
267
+ }
268
+ return;
269
+ });
270
+ mockClientInstance.close = closeImpl;
271
+ // Prepare new instances for the reconnect
272
+ const newMockClientInstance = {
273
+ connect: jest.fn().mockResolvedValue(undefined),
274
+ close: jest.fn().mockResolvedValue(undefined),
275
+ listTools: jest.fn(),
276
+ callTool: jest.fn(),
277
+ onclose: null,
278
+ };
279
+ MockedClient.mockImplementation(() => newMockClientInstance);
280
+ // Reset counts after initial creation
281
+ jest.clearAllMocks();
282
+ // Trigger auto onclose and manual reconnect nearly simultaneously
283
+ client.onclose();
284
+ await client.reconnect();
285
+ // Should have detached onclose before calling close, avoiding recursion
286
+ expect(closeImpl).toHaveBeenCalledTimes(1);
287
+ expect(mockClientInstance.onclose).toBeUndefined();
288
+ // Single-flight: only one new client/connect
289
+ expect(MockedClient).toHaveBeenCalledTimes(1);
290
+ expect(newMockClientInstance.connect).toHaveBeenCalledTimes(1);
291
+ });
292
+ });
293
+ describe("backoff + jitter (automatic reconnect)", () => {
294
+ it("applies jitter and resets to initial delay after a successful reconnect (manual preempt)", async () => {
295
+ jest.useFakeTimers();
296
+ const base = MCPClient.BACKOFF_INITIAL_MS;
297
+ const ratio = MCPClient.BACKOFF_JITTER_RATIO;
298
+ const min = Math.round(base * (1 - ratio));
299
+ const max = Math.round(base * (1 + ratio));
300
+ const setTimeoutSpy = jest.spyOn(global, "setTimeout");
301
+ const randSpy = jest.spyOn(Math, "random").mockReturnValue(0.0); // extreme low jitter
302
+ const endpoint = "https://api.example.com/mcp";
303
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP);
304
+ // Prepare one attempt that will succeed to avoid rescheduling
305
+ MockedClient.mockImplementation(() => ({
306
+ connect: jest.fn().mockResolvedValue(undefined),
307
+ close: jest.fn().mockResolvedValue(undefined),
308
+ listTools: jest.fn(),
309
+ callTool: jest.fn(),
310
+ onclose: null,
311
+ }));
312
+ // Trigger once to capture the delay
313
+ client.onclose();
314
+ expect(setTimeoutSpy).toHaveBeenCalled();
315
+ const numericDelays = setTimeoutSpy.mock.calls
316
+ .map((c) => c[1])
317
+ .filter((v) => typeof v === "number");
318
+ expect(numericDelays.length).toBeGreaterThan(0);
319
+ const d = numericDelays[0];
320
+ expect(d).toBeGreaterThanOrEqual(min);
321
+ expect(d).toBeLessThanOrEqual(max);
322
+ // Manual reconnect succeeds and should reset backoff attempts
323
+ await client.reconnect();
324
+ // Trigger onclose again and ensure we start from initial range again
325
+ client.onclose();
326
+ const numericDelays2 = setTimeoutSpy.mock.calls
327
+ .map((c) => c[1])
328
+ .filter((v) => typeof v === "number");
329
+ expect(numericDelays2.length).toBeGreaterThanOrEqual(2);
330
+ const dAgain = numericDelays2[1];
331
+ expect(dAgain).toBeGreaterThanOrEqual(min);
332
+ expect(dAgain).toBeLessThanOrEqual(max);
333
+ jest.useRealTimers();
334
+ setTimeoutSpy.mockRestore();
335
+ randSpy.mockRestore();
336
+ });
337
+ });
338
+ describe("listTools", () => {
339
+ it("should list all tools with pagination", async () => {
340
+ const endpoint = "https://api.example.com/mcp";
341
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP);
342
+ const mockTools = [
343
+ {
344
+ name: "tool1",
345
+ description: "First tool",
346
+ inputSchema: {
347
+ type: "object",
348
+ properties: { arg1: { type: "string" } },
349
+ },
350
+ },
351
+ {
352
+ name: "tool2",
353
+ description: "Second tool",
354
+ inputSchema: {
355
+ type: "object",
356
+ properties: { arg2: { type: "number" } },
357
+ },
358
+ },
359
+ ];
360
+ const mockResponse1 = {
361
+ tools: [mockTools[0]],
362
+ nextCursor: "cursor1",
363
+ };
364
+ const mockResponse2 = {
365
+ tools: [mockTools[1]],
366
+ nextCursor: undefined,
367
+ };
368
+ mockClientInstance.listTools
369
+ .mockResolvedValueOnce(mockResponse1)
370
+ .mockResolvedValueOnce(mockResponse2);
371
+ const result = await client.listTools();
372
+ expect(mockClientInstance.listTools).toHaveBeenCalledTimes(2);
373
+ expect(mockClientInstance.listTools).toHaveBeenNthCalledWith(1, { cursor: undefined }, {});
374
+ expect(mockClientInstance.listTools).toHaveBeenNthCalledWith(2, { cursor: "cursor1" }, {});
375
+ expect(result).toEqual([
376
+ {
377
+ name: "tool1",
378
+ description: "First tool",
379
+ inputSchema: {
380
+ type: "object",
381
+ properties: { arg1: { type: "string" } },
382
+ },
383
+ },
384
+ {
385
+ name: "tool2",
386
+ description: "Second tool",
387
+ inputSchema: {
388
+ type: "object",
389
+ properties: { arg2: { type: "number" } },
390
+ },
391
+ },
392
+ ]);
393
+ });
394
+ it("should handle single page of tools", async () => {
395
+ const endpoint = "https://api.example.com/mcp";
396
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP);
397
+ const mockTools = [
398
+ {
399
+ name: "tool1",
400
+ description: "Only tool",
401
+ inputSchema: {
402
+ type: "object",
403
+ properties: { arg1: { type: "string" } },
404
+ },
405
+ },
406
+ ];
407
+ const mockResponse = {
408
+ tools: mockTools,
409
+ nextCursor: undefined,
410
+ };
411
+ mockClientInstance.listTools.mockResolvedValue(mockResponse);
412
+ const result = await client.listTools();
413
+ expect(mockClientInstance.listTools).toHaveBeenCalledTimes(1);
414
+ expect(result).toEqual([
415
+ {
416
+ name: "tool1",
417
+ description: "Only tool",
418
+ inputSchema: {
419
+ type: "object",
420
+ properties: { arg1: { type: "string" } },
421
+ },
422
+ },
423
+ ]);
424
+ });
425
+ it("should throw error for invalid input schema", async () => {
426
+ const endpoint = "https://api.example.com/mcp";
427
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP);
428
+ const mockTools = [
429
+ {
430
+ name: "invalid-tool",
431
+ description: "Tool with invalid schema",
432
+ inputSchema: { type: "string" }, // Invalid - should be object
433
+ },
434
+ ];
435
+ const mockResponse = {
436
+ tools: mockTools,
437
+ nextCursor: undefined,
438
+ };
439
+ mockClientInstance.listTools.mockResolvedValue(mockResponse);
440
+ await expect(client.listTools()).rejects.toThrow("Input schema for tool invalid-tool is not an object");
441
+ });
442
+ });
443
+ describe("callTool", () => {
444
+ it("should call a tool with arguments", async () => {
445
+ const endpoint = "https://api.example.com/mcp";
446
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP);
447
+ const mockResult = { success: true, data: "test result" };
448
+ mockClientInstance.callTool.mockResolvedValue(mockResult);
449
+ const result = await client.callTool("testTool", {
450
+ arg1: "value1",
451
+ arg2: 42,
452
+ });
453
+ expect(mockClientInstance.callTool).toHaveBeenCalledWith({
454
+ name: "testTool",
455
+ arguments: { arg1: "value1", arg2: 42 },
456
+ });
457
+ expect(result).toBe(mockResult);
458
+ });
459
+ it("should handle tool call errors", async () => {
460
+ const endpoint = "https://api.example.com/mcp";
461
+ const client = await MCPClient.create(endpoint, MCPTransport.HTTP);
462
+ const error = new Error("Tool call failed");
463
+ mockClientInstance.callTool.mockRejectedValue(error);
464
+ await expect(client.callTool("testTool", {})).rejects.toThrow("Tool call failed");
465
+ });
466
+ });
467
+ describe("transport initialization", () => {
468
+ it("should initialize HTTP transport with session ID", async () => {
469
+ const endpoint = "https://api.example.com/mcp";
470
+ const headers = { Authorization: "Bearer token" };
471
+ await MCPClient.create(endpoint, MCPTransport.HTTP, headers);
472
+ expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(new URL(endpoint), { sessionId: undefined, requestInit: { headers } });
473
+ });
474
+ it("should initialize SSE transport without session ID", async () => {
475
+ const endpoint = "https://api.example.com/mcp";
476
+ const headers = { Authorization: "Bearer token" };
477
+ await MCPClient.create(endpoint, MCPTransport.SSE, headers);
478
+ expect(MockedSSEClientTransport).toHaveBeenCalledWith(new URL(endpoint), {
479
+ requestInit: { headers },
480
+ });
481
+ });
482
+ });
483
+ describe("client initialization", () => {
484
+ it("should initialize client with correct name and version", async () => {
485
+ const endpoint = "https://api.example.com/mcp";
486
+ await MCPClient.create(endpoint);
487
+ expect(MockedClient).toHaveBeenCalledWith({
488
+ name: "tambo-mcp-client",
489
+ version: "1.0.0",
490
+ });
491
+ });
492
+ it("should set onclose handler", async () => {
493
+ const endpoint = "https://api.example.com/mcp";
494
+ const _client = await MCPClient.create(endpoint);
495
+ expect(mockClientInstance.onclose).toBeDefined();
496
+ expect(typeof mockClientInstance.onclose).toBe("function");
497
+ });
498
+ });
499
+ });
500
+ //# sourceMappingURL=mcp-client.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"mcp-client.test.js","sourceRoot":"","sources":["../../../src/mcp/__tests__/mcp-client.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAExD,2BAA2B;AAC3B,IAAI,CAAC,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5D,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QAC1C,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;QAClB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;QAChB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;QACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;QACnB,OAAO,EAAE,IAAI;KACd,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,yCAAyC,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,kBAAkB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;IACtD,uCAAuC;KACxC,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,oDAAoD,EAAE,GAAG,EAAE,CAAC,CAAC;IACrE,6BAA6B,EAAE,IAAI;SAChC,EAAE,EAAE;SACJ,kBAAkB,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;QACrC,SAAS,EAAE,OAAO,EAAE,SAAS;KAC9B,CAAC,CAAC;CACN,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yCAAyC,CAAC;AAC7E,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AAEnG,0BAA0B;AAC1B,MAAM,YAAY,GAAG,MAAyC,CAAC;AAC/D,MAAM,wBAAwB,GAAG,kBAEhC,CAAC;AACF,MAAM,mCAAmC,GACvC,6BAEC,CAAC;AAEJ,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,kBAAuB,CAAC;IAC5B,IAAI,qBAA0B,CAAC;IAE/B,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,wBAAwB;QACxB,kBAAkB,GAAG;YACnB,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;YACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;YACnB,OAAO,EAAE,IAAI;SACd,CAAC;QAEF,qBAAqB,GAAG;YACtB,SAAS,EAAE,iBAAiB;SAC7B,CAAC;QAEF,cAAc;QACd,YAAY,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,CAAC;QAC1D,mCAAmC,CAAC,kBAAkB,CACpD,GAAG,EAAE,CAAC,qBAAqB,CAC5B,CAAC;QACF,wBAAwB,CAAC,kBAAkB,CACzC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAuB,CACjC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;YACrF,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,OAAO,GAAG,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC;YAElD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CACnC,QAAQ,EACR,YAAY,CAAC,IAAI,EACjB,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,CACnD,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC;gBACxC,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,CAAC,CAAC;YACH,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACrD,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAE/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC;YAElE,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,EAAE;gBACvE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;aAC7B,CAAC,CAAC;YACH,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAE/C,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAEjC,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CACvD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,iHAAiH,EAAE,KAAK,IAAI,EAAE;YAC/H,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YAEnE,sDAAsD;YACtD,IAAI,CAAC,aAAa,EAAE,CAAC;YAErB,gEAAgE;YAChE,MAAM,qBAAqB,GAAG;gBAC5B,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,IAAI;aACd,CAAC;YAEF,MAAM,wBAAwB,GAAG;gBAC/B,SAAS,EAAE,gBAAgB;aAC5B,CAAC;YAEF,gDAAgD;YAChD,YAAY,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,qBAA4B,CAAC,CAAC;YACpE,mCAAmC,CAAC,kBAAkB,CACpD,GAAG,EAAE,CAAC,wBAA+B,CACtC,CAAC;YAEF,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,0BAA0B;YAEpD,+BAA+B;YAC/B,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAEpD,gFAAgF;YAChF,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,iBAAiB,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CAC/D,CAAC;YAEF,gCAAgC;YAChC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC;gBACxC,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,CAAC,CAAC;YAEH,4DAA4D;YAC5D,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACxD,wBAAwB,CACzB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC;YAElE,uBAAuB;YACvB,IAAI,CAAC,aAAa,EAAE,CAAC;YAErB,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAEzB,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACpD,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,EAAE;gBACvE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;aAC7B,CAAC,CAAC;YACH,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YACnE,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,kBAAkB,EAAE,CAAC;YAErE,4BAA4B;YAC5B,kBAAkB,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;YAEtE,MAAM,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAEpC,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,iCAAiC,EACjC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAClB,CAAC;YACF,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YACnE,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,kBAAkB,EAAE,CAAC;YAErE,4BAA4B;YAC5B,kBAAkB,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;YAEtE,MAAM,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YAErC,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAC1C,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YAEnE,sDAAsD;YACtD,IAAI,CAAC,aAAa,EAAE,CAAC;YAErB,gEAAgE;YAChE,MAAM,qBAAqB,GAAG;gBAC5B,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,IAAI;aACd,CAAC;YAEF,MAAM,wBAAwB,GAAG;gBAC/B,SAAS,EAAE,gBAAgB;aAC5B,CAAC;YAEF,gDAAgD;YAChD,YAAY,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,qBAA4B,CAAC,CAAC;YACpE,mCAAmC,CAAC,kBAAkB,CACpD,GAAG,EAAE,CAAC,wBAA+B,CACtC,CAAC;YAEF,MAAM,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,+CAA+C;YAEnF,+BAA+B;YAC/B,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAEpD,2EAA2E;YAC3E,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CACvD,CAAC;YAEF,gCAAgC;YAChC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC;gBACxC,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,CAAC,CAAC;YAEH,4DAA4D;YAC5D,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACxD,wBAAwB,CACzB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YAEnE,sDAAsD;YACtD,IAAI,CAAC,aAAa,EAAE,CAAC;YAErB,gEAAgE;YAChE,MAAM,qBAAqB,GAAG;gBAC5B,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,IAAI;aACd,CAAC;YAEF,MAAM,wBAAwB,GAAG;gBAC/B,SAAS,EAAE,mBAAmB;aAC/B,CAAC;YAEF,gDAAgD;YAChD,YAAY,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,qBAA4B,CAAC,CAAC;YACpE,mCAAmC,CAAC,kBAAkB,CACpD,GAAG,EAAE,CAAC,wBAA+B,CACtC,CAAC;YAEF,MAAM,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,gDAAgD;YAErF,+BAA+B;YAC/B,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAEpD,6DAA6D;YAC7D,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,iBAAiB,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CAC/D,CAAC;YAEF,gCAAgC;YAChC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC;gBACxC,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,CAAC,CAAC;YAEH,4DAA4D;YAC5D,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACxD,wBAAwB,CACzB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,sGAAsG,EAAE,KAAK,IAAI,EAAE;YACpH,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YACnE,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,kBAAkB,EAAE,CAAC;YAEpE,yEAAyE;YACzE,MAAM,qBAAqB,GAAG;gBAC5B,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,IAAI;aACd,CAAC;YAEF,MAAM,wBAAwB,GAAG;gBAC/B,SAAS,EAAE,wBAAwB;aACpC,CAAC;YAEF,iEAAiE;YACjE,YAAY,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,qBAA4B,CAAC,CAAC;YACpE,mCAAmC,CAAC,kBAAkB,CACpD,GAAG,EAAE,CAAC,wBAA+B,CACtC,CAAC;YAEF,sCAAsC;YACtC,IAAI,CAAC,aAAa,EAAE,CAAC;YAErB,4DAA4D;YAC3D,MAAc,CAAC,OAAO,EAAE,CAAC;YAC1B,kEAAkE;YAClE,MAAM,gBAAgB,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;YAC5C,sDAAsD;YACtD,MAAM,gBAAgB,CAAC;YAEvB,mCAAmC;YACnC,MAAM,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAEtC,+BAA+B;YAC/B,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAEpD,6DAA6D;YAC7D,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,iBAAiB,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CAC/D,CAAC;YAEF,gCAAgC;YAChC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC;gBACxC,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,CAAC,CAAC;YAEH,4DAA4D;YAC5D,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACxD,wBAAwB,CACzB,CAAC;YAEF,kDAAkD;YAClD,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAE/D,UAAU,CAAC,WAAW,EAAE,CAAC;YACzB,IAAI,CAAC,aAAa,EAAE,CAAC;QACvB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACvD,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;YAClG,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YAEnE,yFAAyF;YACzF,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;gBACnC,IAAI,OAAO,kBAAkB,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;oBACrD,wCAAwC;oBACvC,kBAAkB,CAAC,OAAiC,EAAE,CAAC;gBAC1D,CAAC;gBACD,OAAO;YACT,CAAC,CAAC,CAAC;YACH,kBAAkB,CAAC,KAAK,GAAG,SAAS,CAAC;YAErC,0CAA0C;YAC1C,MAAM,qBAAqB,GAAG;gBAC5B,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,IAAI;aACd,CAAC;YACF,YAAY,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,qBAA4B,CAAC,CAAC;YAEpE,sCAAsC;YACtC,IAAI,CAAC,aAAa,EAAE,CAAC;YAErB,kEAAkE;YACjE,MAAc,CAAC,OAAO,EAAE,CAAC;YAC1B,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAEzB,wEAAwE;YACxE,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC3C,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;YAEnD,6CAA6C;YAC7C,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wCAAwC,EAAE,GAAG,EAAE;QACtD,EAAE,CAAC,0FAA0F,EAAE,KAAK,IAAI,EAAE;YACxG,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,MAAM,IAAI,GAAG,SAAS,CAAC,kBAAkB,CAAC;YAC1C,MAAM,KAAK,GAAG,SAAS,CAAC,oBAAoB,CAAC;YAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC;YAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC;YAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YACvD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,qBAAqB;YAEtF,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YAEnE,8DAA8D;YAC9D,YAAY,CAAC,kBAAkB,CAC7B,GAAG,EAAE,CACH,CAAC;gBACC,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC/C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,IAAI;aACd,CAAQ,CACZ,CAAC;YAEF,oCAAoC;YACnC,MAAc,CAAC,OAAO,EAAE,CAAC;YAC1B,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACzC,MAAM,aAAa,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK;iBAC3C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;iBAChB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAa,CAAC;YACpD,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAE,CAAC;YAC5B,MAAM,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC;YACtC,MAAM,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;YAEnC,8DAA8D;YAC9D,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAEzB,qEAAqE;YACpE,MAAc,CAAC,OAAO,EAAE,CAAC;YAC1B,MAAM,cAAc,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK;iBAC5C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;iBAChB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAa,CAAC;YACpD,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;YACxD,MAAM,MAAM,GAAG,cAAc,CAAC,CAAC,CAAE,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;YAExC,IAAI,CAAC,aAAa,EAAE,CAAC;YACrB,aAAa,CAAC,WAAW,EAAE,CAAC;YAC5B,OAAO,CAAC,WAAW,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YAEnE,MAAM,SAAS,GAAG;gBAChB;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,YAAY;oBACzB,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;gBACD;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,aAAa;oBAC1B,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;aACF,CAAC;YAEF,MAAM,aAAa,GAAG;gBACpB,KAAK,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBACrB,UAAU,EAAE,SAAS;aACtB,CAAC;YAEF,MAAM,aAAa,GAAG;gBACpB,KAAK,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBACrB,UAAU,EAAE,SAAS;aACtB,CAAC;YAEF,kBAAkB,CAAC,SAAS;iBACzB,qBAAqB,CAAC,aAAa,CAAC;iBACpC,qBAAqB,CAAC,aAAa,CAAC,CAAC;YAExC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAExC,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9D,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,uBAAuB,CAC1D,CAAC,EACD,EAAE,MAAM,EAAE,SAAS,EAAE,EACrB,EAAE,CACH,CAAC;YACF,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,uBAAuB,CAC1D,CAAC,EACD,EAAE,MAAM,EAAE,SAAS,EAAE,EACrB,EAAE,CACH,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,YAAY;oBACzB,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;gBACD;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,aAAa;oBAC1B,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YAEnE,MAAM,SAAS,GAAG;gBAChB;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,WAAW;oBACxB,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;aACF,CAAC;YAEF,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,SAAS;gBAChB,UAAU,EAAE,SAAS;aACtB,CAAC;YAEF,kBAAkB,CAAC,SAAS,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAE7D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAExC,MAAM,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB;oBACE,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,WAAW;oBACxB,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;qBACzC;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YAEnE,MAAM,SAAS,GAAG;gBAChB;oBACE,IAAI,EAAE,cAAc;oBACpB,WAAW,EAAE,0BAA0B;oBACvC,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,6BAA6B;iBAC/D;aACF,CAAC;YAEF,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,SAAS;gBAChB,UAAU,EAAE,SAAS;aACtB,CAAC;YAEF,kBAAkB,CAAC,SAAS,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAE7D,MAAM,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAC9C,qDAAqD,CACtD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YAEnE,MAAM,UAAU,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;YAC1D,kBAAkB,CAAC,QAAQ,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE;gBAC/C,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,EAAE;aACT,CAAC,CAAC;YAEH,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC;gBACvD,IAAI,EAAE,UAAU;gBAChB,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE;aACxC,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YAEnE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;YAC5C,kBAAkB,CAAC,QAAQ,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAErD,MAAM,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3D,kBAAkB,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,OAAO,GAAG,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC;YAElD,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YAE7D,MAAM,CAAC,mCAAmC,CAAC,CAAC,oBAAoB,CAC9D,IAAI,GAAG,CAAC,QAAQ,CAAC,EACjB,EAAE,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,EAAE,CACnD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,OAAO,GAAG,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC;YAElD,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YAE5D,MAAM,CAAC,wBAAwB,CAAC,CAAC,oBAAoB,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,EAAE;gBACvE,WAAW,EAAE,EAAE,OAAO,EAAE;aACzB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAE/C,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAEjC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC;gBACxC,IAAI,EAAE,kBAAkB;gBACxB,OAAO,EAAE,OAAO;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,QAAQ,GAAG,6BAA6B,CAAC;YAC/C,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAEjD,MAAM,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;YACjD,MAAM,CAAC,OAAO,kBAAkB,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { MCPClient, MCPTransport } from \"../mcp-client\";\n\n// Mock the MCP SDK modules\njest.mock(\"@modelcontextprotocol/sdk/client/index.js\", () => ({\n Client: jest.fn().mockImplementation(() => ({\n connect: jest.fn(),\n close: jest.fn(),\n listTools: jest.fn(),\n callTool: jest.fn(),\n onclose: null,\n })),\n}));\n\njest.mock(\"@modelcontextprotocol/sdk/client/sse.js\", () => ({\n SSEClientTransport: jest.fn().mockImplementation(() => ({\n // SSE transport doesn't have sessionId\n })),\n}));\n\njest.mock(\"@modelcontextprotocol/sdk/client/streamableHttp.js\", () => ({\n StreamableHTTPClientTransport: jest\n .fn()\n .mockImplementation((url, options) => ({\n sessionId: options?.sessionId,\n })),\n}));\n\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\n\n// Type the mocked modules\nconst MockedClient = Client as jest.MockedClass<typeof Client>;\nconst MockedSSEClientTransport = SSEClientTransport as jest.MockedClass<\n typeof SSEClientTransport\n>;\nconst MockedStreamableHTTPClientTransport =\n StreamableHTTPClientTransport as jest.MockedClass<\n typeof StreamableHTTPClientTransport\n >;\n\ndescribe(\"MCPClient\", () => {\n let mockClientInstance: any;\n let mockTransportInstance: any;\n\n beforeEach(() => {\n jest.clearAllMocks();\n\n // Create mock instances\n mockClientInstance = {\n connect: jest.fn().mockResolvedValue(undefined),\n close: jest.fn().mockResolvedValue(undefined),\n listTools: jest.fn(),\n callTool: jest.fn(),\n onclose: null,\n };\n\n mockTransportInstance = {\n sessionId: \"test-session-id\",\n };\n\n // Setup mocks\n MockedClient.mockImplementation(() => mockClientInstance);\n MockedStreamableHTTPClientTransport.mockImplementation(\n () => mockTransportInstance,\n );\n MockedSSEClientTransport.mockImplementation(\n () => ({}) as SSEClientTransport,\n );\n });\n\n describe(\"create\", () => {\n it(\"should create and connect an MCPClient with HTTP transport by default\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const headers = { Authorization: \"Bearer token\" };\n\n const client = await MCPClient.create(\n endpoint,\n MCPTransport.HTTP,\n headers,\n );\n\n expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n new URL(endpoint),\n { sessionId: undefined, requestInit: { headers } },\n );\n expect(MockedClient).toHaveBeenCalledWith({\n name: \"tambo-mcp-client\",\n version: \"1.0.0\",\n });\n expect(mockClientInstance.connect).toHaveBeenCalledWith(\n mockTransportInstance,\n );\n expect(client).toBeInstanceOf(MCPClient);\n });\n\n it(\"should create and connect an MCPClient with SSE transport\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n\n const client = await MCPClient.create(endpoint, MCPTransport.SSE);\n\n expect(MockedSSEClientTransport).toHaveBeenCalledWith(new URL(endpoint), {\n requestInit: { headers: {} },\n });\n expect(mockClientInstance.connect).toHaveBeenCalledWith({});\n expect(client).toBeInstanceOf(MCPClient);\n });\n\n it(\"should create client with default headers when none provided\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n\n await MCPClient.create(endpoint);\n\n expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n new URL(endpoint),\n { sessionId: undefined, requestInit: { headers: {} } },\n );\n });\n });\n\n describe(\"reconnect\", () => {\n it(\"should create new transport and client instances and call connect when reconnect() is called (default behavior)\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.HTTP);\n\n // Clear previous calls to focus on reconnect behavior\n jest.clearAllMocks();\n\n // Create new mock instances to verify new instances are created\n const newMockClientInstance = {\n connect: jest.fn().mockResolvedValue(undefined),\n close: jest.fn().mockResolvedValue(undefined),\n listTools: jest.fn(),\n callTool: jest.fn(),\n onclose: null,\n };\n\n const newMockTransportInstance = {\n sessionId: \"new-session-id\",\n };\n\n // Mock the constructors to return new instances\n MockedClient.mockImplementation(() => newMockClientInstance as any);\n MockedStreamableHTTPClientTransport.mockImplementation(\n () => newMockTransportInstance as any,\n );\n\n await client.reconnect(); // Uses default parameters\n\n // Verify old client was closed\n expect(mockClientInstance.close).toHaveBeenCalled();\n\n // Verify new transport was created with preserved session ID (default behavior)\n expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n new URL(endpoint),\n { sessionId: \"test-session-id\", requestInit: { headers: {} } },\n );\n\n // Verify new client was created\n expect(MockedClient).toHaveBeenCalledWith({\n name: \"tambo-mcp-client\",\n version: \"1.0.0\",\n });\n\n // Verify new client's connect was called with new transport\n expect(newMockClientInstance.connect).toHaveBeenCalledWith(\n newMockTransportInstance,\n );\n });\n\n it(\"should reconnect without session ID for SSE transport\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.SSE);\n\n // Clear previous calls\n jest.clearAllMocks();\n\n await client.reconnect();\n\n expect(mockClientInstance.close).toHaveBeenCalled();\n expect(MockedSSEClientTransport).toHaveBeenCalledWith(new URL(endpoint), {\n requestInit: { headers: {} },\n });\n expect(mockClientInstance.connect).toHaveBeenCalledWith({});\n });\n\n it(\"should handle close errors when reportErrorOnClose is true\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.HTTP);\n const consoleSpy = jest.spyOn(console, \"error\").mockImplementation();\n\n // Make close throw an error\n mockClientInstance.close.mockRejectedValue(new Error(\"Close failed\"));\n\n await client.reconnect(false, true);\n\n expect(consoleSpy).toHaveBeenCalledWith(\n \"Error closing Tambo MCP Client:\",\n expect.any(Error),\n );\n consoleSpy.mockRestore();\n });\n\n it(\"should not log close errors when reportErrorOnClose is false\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.HTTP);\n const consoleSpy = jest.spyOn(console, \"error\").mockImplementation();\n\n // Make close throw an error\n mockClientInstance.close.mockRejectedValue(new Error(\"Close failed\"));\n\n await client.reconnect(false, false);\n\n expect(consoleSpy).not.toHaveBeenCalled();\n consoleSpy.mockRestore();\n });\n\n it(\"should create new session when newSession is true\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.HTTP);\n\n // Clear previous calls to focus on reconnect behavior\n jest.clearAllMocks();\n\n // Create new mock instances to verify new instances are created\n const newMockClientInstance = {\n connect: jest.fn().mockResolvedValue(undefined),\n close: jest.fn().mockResolvedValue(undefined),\n listTools: jest.fn(),\n callTool: jest.fn(),\n onclose: null,\n };\n\n const newMockTransportInstance = {\n sessionId: \"new-session-id\",\n };\n\n // Mock the constructors to return new instances\n MockedClient.mockImplementation(() => newMockClientInstance as any);\n MockedStreamableHTTPClientTransport.mockImplementation(\n () => newMockTransportInstance as any,\n );\n\n await client.reconnect(true, true); // newSession = true, reportErrorOnClose = true\n\n // Verify old client was closed\n expect(mockClientInstance.close).toHaveBeenCalled();\n\n // Verify new transport was created with undefined session ID (new session)\n expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n new URL(endpoint),\n { sessionId: undefined, requestInit: { headers: {} } },\n );\n\n // Verify new client was created\n expect(MockedClient).toHaveBeenCalledWith({\n name: \"tambo-mcp-client\",\n version: \"1.0.0\",\n });\n\n // Verify new client's connect was called with new transport\n expect(newMockClientInstance.connect).toHaveBeenCalledWith(\n newMockTransportInstance,\n );\n });\n\n it(\"should reuse existing session when newSession is false (default)\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.HTTP);\n\n // Clear previous calls to focus on reconnect behavior\n jest.clearAllMocks();\n\n // Create new mock instances to verify new instances are created\n const newMockClientInstance = {\n connect: jest.fn().mockResolvedValue(undefined),\n close: jest.fn().mockResolvedValue(undefined),\n listTools: jest.fn(),\n callTool: jest.fn(),\n onclose: null,\n };\n\n const newMockTransportInstance = {\n sessionId: \"reused-session-id\",\n };\n\n // Mock the constructors to return new instances\n MockedClient.mockImplementation(() => newMockClientInstance as any);\n MockedStreamableHTTPClientTransport.mockImplementation(\n () => newMockTransportInstance as any,\n );\n\n await client.reconnect(false, true); // newSession = false, reportErrorOnClose = true\n\n // Verify old client was closed\n expect(mockClientInstance.close).toHaveBeenCalled();\n\n // Verify new transport was created with preserved session ID\n expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n new URL(endpoint),\n { sessionId: \"test-session-id\", requestInit: { headers: {} } },\n );\n\n // Verify new client was created\n expect(MockedClient).toHaveBeenCalledWith({\n name: \"tambo-mcp-client\",\n version: \"1.0.0\",\n });\n\n // Verify new client's connect was called with new transport\n expect(newMockClientInstance.connect).toHaveBeenCalledWith(\n newMockTransportInstance,\n );\n });\n });\n\n describe(\"onclose\", () => {\n it(\"should reconnect MCPClient when client is closed by external means (no backoff on manual preemption)\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.HTTP);\n jest.useFakeTimers();\n const consoleSpy = jest.spyOn(console, \"warn\").mockImplementation();\n\n // Create new mock instances to verify reconnection creates new instances\n const newMockClientInstance = {\n connect: jest.fn().mockResolvedValue(undefined),\n close: jest.fn().mockResolvedValue(undefined),\n listTools: jest.fn(),\n callTool: jest.fn(),\n onclose: null,\n };\n\n const newMockTransportInstance = {\n sessionId: \"reconnected-session-id\",\n };\n\n // Mock the constructors to return new instances for reconnection\n MockedClient.mockImplementation(() => newMockClientInstance as any);\n MockedStreamableHTTPClientTransport.mockImplementation(\n () => newMockTransportInstance as any,\n );\n\n // Reset counts after initial creation\n jest.clearAllMocks();\n\n // Trigger automatic onclose (schedules a delayed reconnect)\n (client as any).onclose();\n // Manual reconnect should preempt the scheduled automatic attempt\n const reconnectPromise = client.reconnect();\n // No timers should be pending after manual preemption\n await reconnectPromise;\n\n // Verify warning message is logged\n expect(consoleSpy).toHaveBeenCalled();\n\n // Verify old client was closed\n expect(mockClientInstance.close).toHaveBeenCalled();\n\n // Verify new transport was created with preserved session ID\n expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n new URL(endpoint),\n { sessionId: \"test-session-id\", requestInit: { headers: {} } },\n );\n\n // Verify new client was created\n expect(MockedClient).toHaveBeenCalledWith({\n name: \"tambo-mcp-client\",\n version: \"1.0.0\",\n });\n\n // Verify new client's connect was called with new transport\n expect(newMockClientInstance.connect).toHaveBeenCalledWith(\n newMockTransportInstance,\n );\n\n // Ensure only a single reconnect attempt occurred\n expect(MockedClient).toHaveBeenCalledTimes(1);\n expect(newMockClientInstance.connect).toHaveBeenCalledTimes(1);\n\n consoleSpy.mockRestore();\n jest.useRealTimers();\n });\n });\n\n describe(\"reconnect re-entrancy and single-flight\", () => {\n it(\"prevents re-entrant onclose during deliberate close and coalesces concurrent calls\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.HTTP);\n\n // Simulate an implementation where closing the client would call its own onclose handler\n const closeImpl = jest.fn(async () => {\n if (typeof mockClientInstance.onclose === \"function\") {\n // would cause recursion if not detached\n (mockClientInstance.onclose as unknown as () => void)();\n }\n return;\n });\n mockClientInstance.close = closeImpl;\n\n // Prepare new instances for the reconnect\n const newMockClientInstance = {\n connect: jest.fn().mockResolvedValue(undefined),\n close: jest.fn().mockResolvedValue(undefined),\n listTools: jest.fn(),\n callTool: jest.fn(),\n onclose: null,\n };\n MockedClient.mockImplementation(() => newMockClientInstance as any);\n\n // Reset counts after initial creation\n jest.clearAllMocks();\n\n // Trigger auto onclose and manual reconnect nearly simultaneously\n (client as any).onclose();\n await client.reconnect();\n\n // Should have detached onclose before calling close, avoiding recursion\n expect(closeImpl).toHaveBeenCalledTimes(1);\n expect(mockClientInstance.onclose).toBeUndefined();\n\n // Single-flight: only one new client/connect\n expect(MockedClient).toHaveBeenCalledTimes(1);\n expect(newMockClientInstance.connect).toHaveBeenCalledTimes(1);\n });\n });\n\n describe(\"backoff + jitter (automatic reconnect)\", () => {\n it(\"applies jitter and resets to initial delay after a successful reconnect (manual preempt)\", async () => {\n jest.useFakeTimers();\n const base = MCPClient.BACKOFF_INITIAL_MS;\n const ratio = MCPClient.BACKOFF_JITTER_RATIO;\n const min = Math.round(base * (1 - ratio));\n const max = Math.round(base * (1 + ratio));\n const setTimeoutSpy = jest.spyOn(global, \"setTimeout\");\n const randSpy = jest.spyOn(Math, \"random\").mockReturnValue(0.0); // extreme low jitter\n\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.HTTP);\n\n // Prepare one attempt that will succeed to avoid rescheduling\n MockedClient.mockImplementation(\n () =>\n ({\n connect: jest.fn().mockResolvedValue(undefined),\n close: jest.fn().mockResolvedValue(undefined),\n listTools: jest.fn(),\n callTool: jest.fn(),\n onclose: null,\n }) as any,\n );\n\n // Trigger once to capture the delay\n (client as any).onclose();\n expect(setTimeoutSpy).toHaveBeenCalled();\n const numericDelays = setTimeoutSpy.mock.calls\n .map((c) => c[1])\n .filter((v) => typeof v === \"number\") as number[];\n expect(numericDelays.length).toBeGreaterThan(0);\n const d = numericDelays[0]!;\n expect(d).toBeGreaterThanOrEqual(min);\n expect(d).toBeLessThanOrEqual(max);\n\n // Manual reconnect succeeds and should reset backoff attempts\n await client.reconnect();\n\n // Trigger onclose again and ensure we start from initial range again\n (client as any).onclose();\n const numericDelays2 = setTimeoutSpy.mock.calls\n .map((c) => c[1])\n .filter((v) => typeof v === \"number\") as number[];\n expect(numericDelays2.length).toBeGreaterThanOrEqual(2);\n const dAgain = numericDelays2[1]!;\n expect(dAgain).toBeGreaterThanOrEqual(min);\n expect(dAgain).toBeLessThanOrEqual(max);\n\n jest.useRealTimers();\n setTimeoutSpy.mockRestore();\n randSpy.mockRestore();\n });\n });\n\n describe(\"listTools\", () => {\n it(\"should list all tools with pagination\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.HTTP);\n\n const mockTools = [\n {\n name: \"tool1\",\n description: \"First tool\",\n inputSchema: {\n type: \"object\",\n properties: { arg1: { type: \"string\" } },\n },\n },\n {\n name: \"tool2\",\n description: \"Second tool\",\n inputSchema: {\n type: \"object\",\n properties: { arg2: { type: \"number\" } },\n },\n },\n ];\n\n const mockResponse1 = {\n tools: [mockTools[0]],\n nextCursor: \"cursor1\",\n };\n\n const mockResponse2 = {\n tools: [mockTools[1]],\n nextCursor: undefined,\n };\n\n mockClientInstance.listTools\n .mockResolvedValueOnce(mockResponse1)\n .mockResolvedValueOnce(mockResponse2);\n\n const result = await client.listTools();\n\n expect(mockClientInstance.listTools).toHaveBeenCalledTimes(2);\n expect(mockClientInstance.listTools).toHaveBeenNthCalledWith(\n 1,\n { cursor: undefined },\n {},\n );\n expect(mockClientInstance.listTools).toHaveBeenNthCalledWith(\n 2,\n { cursor: \"cursor1\" },\n {},\n );\n expect(result).toEqual([\n {\n name: \"tool1\",\n description: \"First tool\",\n inputSchema: {\n type: \"object\",\n properties: { arg1: { type: \"string\" } },\n },\n },\n {\n name: \"tool2\",\n description: \"Second tool\",\n inputSchema: {\n type: \"object\",\n properties: { arg2: { type: \"number\" } },\n },\n },\n ]);\n });\n\n it(\"should handle single page of tools\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.HTTP);\n\n const mockTools = [\n {\n name: \"tool1\",\n description: \"Only tool\",\n inputSchema: {\n type: \"object\",\n properties: { arg1: { type: \"string\" } },\n },\n },\n ];\n\n const mockResponse = {\n tools: mockTools,\n nextCursor: undefined,\n };\n\n mockClientInstance.listTools.mockResolvedValue(mockResponse);\n\n const result = await client.listTools();\n\n expect(mockClientInstance.listTools).toHaveBeenCalledTimes(1);\n expect(result).toEqual([\n {\n name: \"tool1\",\n description: \"Only tool\",\n inputSchema: {\n type: \"object\",\n properties: { arg1: { type: \"string\" } },\n },\n },\n ]);\n });\n\n it(\"should throw error for invalid input schema\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.HTTP);\n\n const mockTools = [\n {\n name: \"invalid-tool\",\n description: \"Tool with invalid schema\",\n inputSchema: { type: \"string\" }, // Invalid - should be object\n },\n ];\n\n const mockResponse = {\n tools: mockTools,\n nextCursor: undefined,\n };\n\n mockClientInstance.listTools.mockResolvedValue(mockResponse);\n\n await expect(client.listTools()).rejects.toThrow(\n \"Input schema for tool invalid-tool is not an object\",\n );\n });\n });\n\n describe(\"callTool\", () => {\n it(\"should call a tool with arguments\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.HTTP);\n\n const mockResult = { success: true, data: \"test result\" };\n mockClientInstance.callTool.mockResolvedValue(mockResult);\n\n const result = await client.callTool(\"testTool\", {\n arg1: \"value1\",\n arg2: 42,\n });\n\n expect(mockClientInstance.callTool).toHaveBeenCalledWith({\n name: \"testTool\",\n arguments: { arg1: \"value1\", arg2: 42 },\n });\n expect(result).toBe(mockResult);\n });\n\n it(\"should handle tool call errors\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const client = await MCPClient.create(endpoint, MCPTransport.HTTP);\n\n const error = new Error(\"Tool call failed\");\n mockClientInstance.callTool.mockRejectedValue(error);\n\n await expect(client.callTool(\"testTool\", {})).rejects.toThrow(\n \"Tool call failed\",\n );\n });\n });\n\n describe(\"transport initialization\", () => {\n it(\"should initialize HTTP transport with session ID\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const headers = { Authorization: \"Bearer token\" };\n\n await MCPClient.create(endpoint, MCPTransport.HTTP, headers);\n\n expect(MockedStreamableHTTPClientTransport).toHaveBeenCalledWith(\n new URL(endpoint),\n { sessionId: undefined, requestInit: { headers } },\n );\n });\n\n it(\"should initialize SSE transport without session ID\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const headers = { Authorization: \"Bearer token\" };\n\n await MCPClient.create(endpoint, MCPTransport.SSE, headers);\n\n expect(MockedSSEClientTransport).toHaveBeenCalledWith(new URL(endpoint), {\n requestInit: { headers },\n });\n });\n });\n\n describe(\"client initialization\", () => {\n it(\"should initialize client with correct name and version\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n\n await MCPClient.create(endpoint);\n\n expect(MockedClient).toHaveBeenCalledWith({\n name: \"tambo-mcp-client\",\n version: \"1.0.0\",\n });\n });\n\n it(\"should set onclose handler\", async () => {\n const endpoint = \"https://api.example.com/mcp\";\n const _client = await MCPClient.create(endpoint);\n\n expect(mockClientInstance.onclose).toBeDefined();\n expect(typeof mockClientInstance.onclose).toBe(\"function\");\n });\n });\n});\n"]}