@tambo-ai/react 0.72.0 → 0.73.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (151) hide show
  1. package/dist/mcp/mcp-hooks.d.ts +4 -0
  2. package/dist/mcp/mcp-hooks.d.ts.map +1 -1
  3. package/dist/mcp/mcp-hooks.js +4 -0
  4. package/dist/mcp/mcp-hooks.js.map +1 -1
  5. package/dist/providers/tambo-provider.d.ts +3 -0
  6. package/dist/providers/tambo-provider.d.ts.map +1 -1
  7. package/dist/providers/tambo-provider.js +3 -0
  8. package/dist/providers/tambo-provider.js.map +1 -1
  9. package/dist/util/resource-content-resolver.d.ts.map +1 -1
  10. package/dist/util/resource-content-resolver.js +2 -0
  11. package/dist/util/resource-content-resolver.js.map +1 -1
  12. package/dist/v1/hooks/use-tambo-v1-messages.test.js +22 -9
  13. package/dist/v1/hooks/use-tambo-v1-messages.test.js.map +1 -1
  14. package/dist/v1/hooks/use-tambo-v1-send-message.d.ts +1 -0
  15. package/dist/v1/hooks/use-tambo-v1-send-message.d.ts.map +1 -1
  16. package/dist/v1/hooks/use-tambo-v1-send-message.js +9 -2
  17. package/dist/v1/hooks/use-tambo-v1-send-message.js.map +1 -1
  18. package/dist/v1/hooks/use-tambo-v1-send-message.test.js +22 -9
  19. package/dist/v1/hooks/use-tambo-v1-send-message.test.js.map +1 -1
  20. package/dist/v1/hooks/use-tambo-v1-suggestions.d.ts +91 -0
  21. package/dist/v1/hooks/use-tambo-v1-suggestions.d.ts.map +1 -0
  22. package/dist/v1/hooks/use-tambo-v1-suggestions.js +152 -0
  23. package/dist/v1/hooks/use-tambo-v1-suggestions.js.map +1 -0
  24. package/dist/v1/hooks/use-tambo-v1-suggestions.test.d.ts +2 -0
  25. package/dist/v1/hooks/use-tambo-v1-suggestions.test.d.ts.map +1 -0
  26. package/dist/v1/hooks/use-tambo-v1-suggestions.test.js +511 -0
  27. package/dist/v1/hooks/use-tambo-v1-suggestions.test.js.map +1 -0
  28. package/dist/v1/hooks/use-tambo-v1-thread-input.d.ts +6 -57
  29. package/dist/v1/hooks/use-tambo-v1-thread-input.d.ts.map +1 -1
  30. package/dist/v1/hooks/use-tambo-v1-thread-input.js +7 -67
  31. package/dist/v1/hooks/use-tambo-v1-thread-input.js.map +1 -1
  32. package/dist/v1/hooks/use-tambo-v1-thread-input.test.js +201 -72
  33. package/dist/v1/hooks/use-tambo-v1-thread-input.test.js.map +1 -1
  34. package/dist/v1/hooks/use-tambo-v1-thread-list.d.ts +6 -4
  35. package/dist/v1/hooks/use-tambo-v1-thread-list.d.ts.map +1 -1
  36. package/dist/v1/hooks/use-tambo-v1-thread-list.js +2 -2
  37. package/dist/v1/hooks/use-tambo-v1-thread-list.js.map +1 -1
  38. package/dist/v1/hooks/use-tambo-v1-thread-list.test.js +2 -2
  39. package/dist/v1/hooks/use-tambo-v1-thread-list.test.js.map +1 -1
  40. package/dist/v1/hooks/use-tambo-v1.test.js +16 -7
  41. package/dist/v1/hooks/use-tambo-v1.test.js.map +1 -1
  42. package/dist/v1/index.d.ts +22 -13
  43. package/dist/v1/index.d.ts.map +1 -1
  44. package/dist/v1/index.js +31 -39
  45. package/dist/v1/index.js.map +1 -1
  46. package/dist/v1/providers/tambo-v1-provider.d.ts +27 -9
  47. package/dist/v1/providers/tambo-v1-provider.d.ts.map +1 -1
  48. package/dist/v1/providers/tambo-v1-provider.js +22 -11
  49. package/dist/v1/providers/tambo-v1-provider.js.map +1 -1
  50. package/dist/v1/providers/tambo-v1-provider.test.js +27 -10
  51. package/dist/v1/providers/tambo-v1-provider.test.js.map +1 -1
  52. package/dist/v1/providers/tambo-v1-stream-context.d.ts +19 -10
  53. package/dist/v1/providers/tambo-v1-stream-context.d.ts.map +1 -1
  54. package/dist/v1/providers/tambo-v1-stream-context.js +43 -53
  55. package/dist/v1/providers/tambo-v1-stream-context.js.map +1 -1
  56. package/dist/v1/providers/tambo-v1-stream-context.test.js +94 -19
  57. package/dist/v1/providers/tambo-v1-stream-context.test.js.map +1 -1
  58. package/dist/v1/providers/tambo-v1-stub-provider.d.ts +74 -0
  59. package/dist/v1/providers/tambo-v1-stub-provider.d.ts.map +1 -0
  60. package/dist/v1/providers/tambo-v1-stub-provider.js +212 -0
  61. package/dist/v1/providers/tambo-v1-stub-provider.js.map +1 -0
  62. package/dist/v1/providers/tambo-v1-stub-provider.test.d.ts +2 -0
  63. package/dist/v1/providers/tambo-v1-stub-provider.test.d.ts.map +1 -0
  64. package/dist/v1/providers/tambo-v1-stub-provider.test.js +162 -0
  65. package/dist/v1/providers/tambo-v1-stub-provider.test.js.map +1 -0
  66. package/dist/v1/providers/tambo-v1-thread-input-provider.d.ts +105 -0
  67. package/dist/v1/providers/tambo-v1-thread-input-provider.d.ts.map +1 -0
  68. package/dist/v1/providers/tambo-v1-thread-input-provider.js +191 -0
  69. package/dist/v1/providers/tambo-v1-thread-input-provider.js.map +1 -0
  70. package/dist/v1/utils/component-renderer.d.ts +15 -67
  71. package/dist/v1/utils/component-renderer.d.ts.map +1 -1
  72. package/dist/v1/utils/component-renderer.js +3 -149
  73. package/dist/v1/utils/component-renderer.js.map +1 -1
  74. package/dist/v1/utils/component-renderer.test.js +15 -350
  75. package/dist/v1/utils/component-renderer.test.js.map +1 -1
  76. package/esm/mcp/mcp-hooks.d.ts +4 -0
  77. package/esm/mcp/mcp-hooks.d.ts.map +1 -1
  78. package/esm/mcp/mcp-hooks.js +4 -0
  79. package/esm/mcp/mcp-hooks.js.map +1 -1
  80. package/esm/providers/tambo-provider.d.ts +3 -0
  81. package/esm/providers/tambo-provider.d.ts.map +1 -1
  82. package/esm/providers/tambo-provider.js +3 -0
  83. package/esm/providers/tambo-provider.js.map +1 -1
  84. package/esm/util/resource-content-resolver.d.ts.map +1 -1
  85. package/esm/util/resource-content-resolver.js +2 -0
  86. package/esm/util/resource-content-resolver.js.map +1 -1
  87. package/esm/v1/hooks/use-tambo-v1-messages.test.js +22 -9
  88. package/esm/v1/hooks/use-tambo-v1-messages.test.js.map +1 -1
  89. package/esm/v1/hooks/use-tambo-v1-send-message.d.ts +1 -0
  90. package/esm/v1/hooks/use-tambo-v1-send-message.d.ts.map +1 -1
  91. package/esm/v1/hooks/use-tambo-v1-send-message.js +9 -2
  92. package/esm/v1/hooks/use-tambo-v1-send-message.js.map +1 -1
  93. package/esm/v1/hooks/use-tambo-v1-send-message.test.js +22 -9
  94. package/esm/v1/hooks/use-tambo-v1-send-message.test.js.map +1 -1
  95. package/esm/v1/hooks/use-tambo-v1-suggestions.d.ts +91 -0
  96. package/esm/v1/hooks/use-tambo-v1-suggestions.d.ts.map +1 -0
  97. package/esm/v1/hooks/use-tambo-v1-suggestions.js +149 -0
  98. package/esm/v1/hooks/use-tambo-v1-suggestions.js.map +1 -0
  99. package/esm/v1/hooks/use-tambo-v1-suggestions.test.d.ts +2 -0
  100. package/esm/v1/hooks/use-tambo-v1-suggestions.test.d.ts.map +1 -0
  101. package/esm/v1/hooks/use-tambo-v1-suggestions.test.js +506 -0
  102. package/esm/v1/hooks/use-tambo-v1-suggestions.test.js.map +1 -0
  103. package/esm/v1/hooks/use-tambo-v1-thread-input.d.ts +6 -57
  104. package/esm/v1/hooks/use-tambo-v1-thread-input.d.ts.map +1 -1
  105. package/esm/v1/hooks/use-tambo-v1-thread-input.js +5 -66
  106. package/esm/v1/hooks/use-tambo-v1-thread-input.js.map +1 -1
  107. package/esm/v1/hooks/use-tambo-v1-thread-input.test.js +199 -73
  108. package/esm/v1/hooks/use-tambo-v1-thread-input.test.js.map +1 -1
  109. package/esm/v1/hooks/use-tambo-v1-thread-list.d.ts +6 -4
  110. package/esm/v1/hooks/use-tambo-v1-thread-list.d.ts.map +1 -1
  111. package/esm/v1/hooks/use-tambo-v1-thread-list.js +2 -2
  112. package/esm/v1/hooks/use-tambo-v1-thread-list.js.map +1 -1
  113. package/esm/v1/hooks/use-tambo-v1-thread-list.test.js +2 -2
  114. package/esm/v1/hooks/use-tambo-v1-thread-list.test.js.map +1 -1
  115. package/esm/v1/hooks/use-tambo-v1.test.js +16 -7
  116. package/esm/v1/hooks/use-tambo-v1.test.js.map +1 -1
  117. package/esm/v1/index.d.ts +22 -13
  118. package/esm/v1/index.d.ts.map +1 -1
  119. package/esm/v1/index.js +23 -18
  120. package/esm/v1/index.js.map +1 -1
  121. package/esm/v1/providers/tambo-v1-provider.d.ts +27 -9
  122. package/esm/v1/providers/tambo-v1-provider.d.ts.map +1 -1
  123. package/esm/v1/providers/tambo-v1-provider.js +20 -10
  124. package/esm/v1/providers/tambo-v1-provider.js.map +1 -1
  125. package/esm/v1/providers/tambo-v1-provider.test.js +28 -11
  126. package/esm/v1/providers/tambo-v1-provider.test.js.map +1 -1
  127. package/esm/v1/providers/tambo-v1-stream-context.d.ts +19 -10
  128. package/esm/v1/providers/tambo-v1-stream-context.d.ts.map +1 -1
  129. package/esm/v1/providers/tambo-v1-stream-context.js +44 -54
  130. package/esm/v1/providers/tambo-v1-stream-context.js.map +1 -1
  131. package/esm/v1/providers/tambo-v1-stream-context.test.js +95 -20
  132. package/esm/v1/providers/tambo-v1-stream-context.test.js.map +1 -1
  133. package/esm/v1/providers/tambo-v1-stub-provider.d.ts +74 -0
  134. package/esm/v1/providers/tambo-v1-stub-provider.d.ts.map +1 -0
  135. package/esm/v1/providers/tambo-v1-stub-provider.js +176 -0
  136. package/esm/v1/providers/tambo-v1-stub-provider.js.map +1 -0
  137. package/esm/v1/providers/tambo-v1-stub-provider.test.d.ts +2 -0
  138. package/esm/v1/providers/tambo-v1-stub-provider.test.d.ts.map +1 -0
  139. package/esm/v1/providers/tambo-v1-stub-provider.test.js +157 -0
  140. package/esm/v1/providers/tambo-v1-stub-provider.test.js.map +1 -0
  141. package/esm/v1/providers/tambo-v1-thread-input-provider.d.ts +105 -0
  142. package/esm/v1/providers/tambo-v1-thread-input-provider.d.ts.map +1 -0
  143. package/esm/v1/providers/tambo-v1-thread-input-provider.js +153 -0
  144. package/esm/v1/providers/tambo-v1-thread-input-provider.js.map +1 -0
  145. package/esm/v1/utils/component-renderer.d.ts +15 -67
  146. package/esm/v1/utils/component-renderer.d.ts.map +1 -1
  147. package/esm/v1/utils/component-renderer.js +4 -146
  148. package/esm/v1/utils/component-renderer.js.map +1 -1
  149. package/esm/v1/utils/component-renderer.test.js +16 -351
  150. package/esm/v1/utils/component-renderer.test.js.map +1 -1
  151. package/package.json +2 -2
@@ -0,0 +1,511 @@
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ const react_1 = require("@testing-library/react");
7
+ const react_2 = __importDefault(require("react"));
8
+ const react_query_1 = require("@tanstack/react-query");
9
+ const use_tambo_v1_suggestions_1 = require("./use-tambo-v1-suggestions");
10
+ const use_tambo_v1_thread_input_1 = require("./use-tambo-v1-thread-input");
11
+ const use_tambo_v1_1 = require("./use-tambo-v1");
12
+ const tambo_client_provider_1 = require("../../providers/tambo-client-provider");
13
+ const tambo_registry_provider_1 = require("../../providers/tambo-registry-provider");
14
+ // Mock dependencies
15
+ jest.mock("./use-tambo-v1-thread-input", () => ({
16
+ useTamboV1ThreadInput: jest.fn(),
17
+ }));
18
+ jest.mock("./use-tambo-v1", () => ({
19
+ useTamboV1: jest.fn(),
20
+ }));
21
+ jest.mock("../../providers/tambo-client-provider", () => ({
22
+ useTamboClient: jest.fn(),
23
+ useTamboQueryClient: jest.fn(() => new react_query_1.QueryClient()),
24
+ }));
25
+ jest.mock("../../providers/tambo-registry-provider", () => ({
26
+ useTamboRegistry: jest.fn(),
27
+ }));
28
+ describe("useTamboV1Suggestions", () => {
29
+ let queryClient;
30
+ const mockSetValue = jest.fn();
31
+ const mockSubmit = jest.fn();
32
+ const mockGenerateSuggestions = jest.fn();
33
+ function createWrapper() {
34
+ return function Wrapper({ children }) {
35
+ return (react_2.default.createElement(react_query_1.QueryClientProvider, { client: queryClient }, children));
36
+ };
37
+ }
38
+ beforeEach(() => {
39
+ jest.clearAllMocks();
40
+ queryClient = new react_query_1.QueryClient({
41
+ defaultOptions: {
42
+ queries: { retry: false },
43
+ mutations: { retry: false },
44
+ },
45
+ });
46
+ // Default mock for thread input
47
+ jest.mocked(use_tambo_v1_thread_input_1.useTamboV1ThreadInput).mockReturnValue({
48
+ value: "",
49
+ setValue: mockSetValue,
50
+ submit: mockSubmit,
51
+ threadId: "thread_123",
52
+ setThreadId: jest.fn(),
53
+ images: [],
54
+ addImage: jest.fn(),
55
+ addImages: jest.fn(),
56
+ removeImage: jest.fn(),
57
+ clearImages: jest.fn(),
58
+ isPending: false,
59
+ isError: false,
60
+ error: undefined,
61
+ });
62
+ // Default mock for useTamboV1
63
+ jest.mocked(use_tambo_v1_1.useTamboV1).mockReturnValue({
64
+ messages: [],
65
+ thread: undefined,
66
+ isIdle: true,
67
+ isStreaming: false,
68
+ startNewThread: jest.fn(),
69
+ switchThread: jest.fn(),
70
+ initThread: jest.fn(),
71
+ streamingState: {
72
+ status: "idle",
73
+ },
74
+ });
75
+ // Default mock for registry
76
+ jest.mocked(tambo_registry_provider_1.useTamboRegistry).mockReturnValue({
77
+ componentList: {},
78
+ toolRegistry: {},
79
+ componentToolAssociations: {},
80
+ mcpServerInfos: [],
81
+ resources: [],
82
+ resourceSource: null,
83
+ onCallUnregisteredTool: undefined,
84
+ registerComponent: jest.fn(),
85
+ unregisterComponent: jest.fn(),
86
+ registerTool: jest.fn(),
87
+ unregisterTool: jest.fn(),
88
+ registerMcpServer: jest.fn(),
89
+ unregisterMcpServer: jest.fn(),
90
+ registerResource: jest.fn(),
91
+ unregisterResource: jest.fn(),
92
+ setResourceSource: jest.fn(),
93
+ });
94
+ // Default mock for client
95
+ mockGenerateSuggestions.mockResolvedValue([
96
+ {
97
+ id: "suggestion_1",
98
+ suggestion: "What's the weather?",
99
+ detailedSuggestion: "What's the weather like today?",
100
+ },
101
+ {
102
+ id: "suggestion_2",
103
+ suggestion: "Tell me a joke",
104
+ detailedSuggestion: "Can you tell me a funny joke?",
105
+ },
106
+ ]);
107
+ jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue({
108
+ beta: {
109
+ threads: {
110
+ suggestions: {
111
+ generate: mockGenerateSuggestions,
112
+ },
113
+ },
114
+ },
115
+ });
116
+ });
117
+ describe("Initial State", () => {
118
+ it("returns empty suggestions when no messages", () => {
119
+ const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_suggestions_1.useTamboV1Suggestions)(), {
120
+ wrapper: createWrapper(),
121
+ });
122
+ expect(result.current.suggestions).toEqual([]);
123
+ expect(result.current.selectedSuggestionId).toBeNull();
124
+ });
125
+ it("returns empty suggestions when latest message is from user", () => {
126
+ jest.mocked(use_tambo_v1_1.useTamboV1).mockReturnValue({
127
+ messages: [
128
+ {
129
+ id: "msg_1",
130
+ role: "user",
131
+ content: [],
132
+ createdAt: "2024-01-01T00:00:00Z",
133
+ },
134
+ ],
135
+ thread: undefined,
136
+ isIdle: true,
137
+ isStreaming: false,
138
+ startNewThread: jest.fn(),
139
+ switchThread: jest.fn(),
140
+ initThread: jest.fn(),
141
+ streamingState: {
142
+ status: "idle",
143
+ },
144
+ });
145
+ const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_suggestions_1.useTamboV1Suggestions)(), {
146
+ wrapper: createWrapper(),
147
+ });
148
+ expect(result.current.suggestions).toEqual([]);
149
+ });
150
+ });
151
+ describe("Suggestion Generation", () => {
152
+ it("generates suggestions when latest message is from assistant and thread is idle", async () => {
153
+ jest.mocked(use_tambo_v1_1.useTamboV1).mockReturnValue({
154
+ messages: [
155
+ {
156
+ id: "msg_1",
157
+ role: "assistant",
158
+ content: [],
159
+ createdAt: "2024-01-01T00:00:00Z",
160
+ },
161
+ ],
162
+ thread: undefined,
163
+ isIdle: true,
164
+ isStreaming: false,
165
+ startNewThread: jest.fn(),
166
+ switchThread: jest.fn(),
167
+ initThread: jest.fn(),
168
+ streamingState: {
169
+ status: "idle",
170
+ },
171
+ });
172
+ const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_suggestions_1.useTamboV1Suggestions)(), {
173
+ wrapper: createWrapper(),
174
+ });
175
+ await (0, react_1.waitFor)(() => {
176
+ expect(result.current.suggestions).toHaveLength(2);
177
+ });
178
+ expect(mockGenerateSuggestions).toHaveBeenCalledWith("msg_1", expect.objectContaining({
179
+ id: "thread_123",
180
+ maxSuggestions: 3,
181
+ }));
182
+ });
183
+ it("does not generate suggestions when thread is streaming", () => {
184
+ jest.mocked(use_tambo_v1_1.useTamboV1).mockReturnValue({
185
+ messages: [
186
+ {
187
+ id: "msg_1",
188
+ role: "assistant",
189
+ content: [],
190
+ createdAt: "2024-01-01T00:00:00Z",
191
+ },
192
+ ],
193
+ thread: undefined,
194
+ isIdle: false,
195
+ isStreaming: true,
196
+ startNewThread: jest.fn(),
197
+ switchThread: jest.fn(),
198
+ initThread: jest.fn(),
199
+ streamingState: {
200
+ status: "streaming",
201
+ runId: "run_1",
202
+ },
203
+ });
204
+ (0, react_1.renderHook)(() => (0, use_tambo_v1_suggestions_1.useTamboV1Suggestions)(), {
205
+ wrapper: createWrapper(),
206
+ });
207
+ expect(mockGenerateSuggestions).not.toHaveBeenCalled();
208
+ });
209
+ it("does not generate suggestions when no threadId", () => {
210
+ jest.mocked(use_tambo_v1_thread_input_1.useTamboV1ThreadInput).mockReturnValue({
211
+ value: "",
212
+ setValue: mockSetValue,
213
+ submit: mockSubmit,
214
+ threadId: undefined,
215
+ setThreadId: jest.fn(),
216
+ images: [],
217
+ addImage: jest.fn(),
218
+ addImages: jest.fn(),
219
+ removeImage: jest.fn(),
220
+ clearImages: jest.fn(),
221
+ isPending: false,
222
+ isError: false,
223
+ error: undefined,
224
+ });
225
+ jest.mocked(use_tambo_v1_1.useTamboV1).mockReturnValue({
226
+ messages: [
227
+ {
228
+ id: "msg_1",
229
+ role: "assistant",
230
+ content: [],
231
+ createdAt: "2024-01-01T00:00:00Z",
232
+ },
233
+ ],
234
+ thread: undefined,
235
+ isIdle: true,
236
+ isStreaming: false,
237
+ startNewThread: jest.fn(),
238
+ switchThread: jest.fn(),
239
+ initThread: jest.fn(),
240
+ streamingState: {
241
+ status: "idle",
242
+ },
243
+ });
244
+ (0, react_1.renderHook)(() => (0, use_tambo_v1_suggestions_1.useTamboV1Suggestions)(), {
245
+ wrapper: createWrapper(),
246
+ });
247
+ expect(mockGenerateSuggestions).not.toHaveBeenCalled();
248
+ });
249
+ it("uses custom maxSuggestions option", async () => {
250
+ jest.mocked(use_tambo_v1_1.useTamboV1).mockReturnValue({
251
+ messages: [
252
+ {
253
+ id: "msg_1",
254
+ role: "assistant",
255
+ content: [],
256
+ createdAt: "2024-01-01T00:00:00Z",
257
+ },
258
+ ],
259
+ thread: undefined,
260
+ isIdle: true,
261
+ isStreaming: false,
262
+ startNewThread: jest.fn(),
263
+ switchThread: jest.fn(),
264
+ initThread: jest.fn(),
265
+ streamingState: {
266
+ status: "idle",
267
+ },
268
+ });
269
+ const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_suggestions_1.useTamboV1Suggestions)({ maxSuggestions: 5 }), { wrapper: createWrapper() });
270
+ await (0, react_1.waitFor)(() => {
271
+ expect(result.current.suggestions).toHaveLength(2);
272
+ });
273
+ expect(mockGenerateSuggestions).toHaveBeenCalledWith("msg_1", expect.objectContaining({
274
+ maxSuggestions: 5,
275
+ }));
276
+ });
277
+ });
278
+ describe("Accepting Suggestions", () => {
279
+ it("updates shared input value when accepting without submit", async () => {
280
+ jest.mocked(use_tambo_v1_1.useTamboV1).mockReturnValue({
281
+ messages: [
282
+ {
283
+ id: "msg_1",
284
+ role: "assistant",
285
+ content: [],
286
+ createdAt: "2024-01-01T00:00:00Z",
287
+ },
288
+ ],
289
+ thread: undefined,
290
+ isIdle: true,
291
+ isStreaming: false,
292
+ startNewThread: jest.fn(),
293
+ switchThread: jest.fn(),
294
+ initThread: jest.fn(),
295
+ streamingState: {
296
+ status: "idle",
297
+ },
298
+ });
299
+ const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_suggestions_1.useTamboV1Suggestions)(), {
300
+ wrapper: createWrapper(),
301
+ });
302
+ await (0, react_1.waitFor)(() => {
303
+ expect(result.current.suggestions).toHaveLength(2);
304
+ });
305
+ const suggestion = result.current.suggestions[0];
306
+ await (0, react_1.act)(async () => {
307
+ await result.current.accept({ suggestion });
308
+ });
309
+ expect(mockSetValue).toHaveBeenCalledWith("What's the weather like today?");
310
+ expect(mockSubmit).not.toHaveBeenCalled();
311
+ expect(result.current.selectedSuggestionId).toBe("suggestion_1");
312
+ });
313
+ it("submits when accepting with shouldSubmit=true", async () => {
314
+ mockSubmit.mockResolvedValue({ threadId: "thread_123" });
315
+ jest.mocked(use_tambo_v1_1.useTamboV1).mockReturnValue({
316
+ messages: [
317
+ {
318
+ id: "msg_1",
319
+ role: "assistant",
320
+ content: [],
321
+ createdAt: "2024-01-01T00:00:00Z",
322
+ },
323
+ ],
324
+ thread: undefined,
325
+ isIdle: true,
326
+ isStreaming: false,
327
+ startNewThread: jest.fn(),
328
+ switchThread: jest.fn(),
329
+ initThread: jest.fn(),
330
+ streamingState: {
331
+ status: "idle",
332
+ },
333
+ });
334
+ const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_suggestions_1.useTamboV1Suggestions)(), {
335
+ wrapper: createWrapper(),
336
+ });
337
+ await (0, react_1.waitFor)(() => {
338
+ expect(result.current.suggestions).toHaveLength(2);
339
+ });
340
+ const suggestion = result.current.suggestions[0];
341
+ await (0, react_1.act)(async () => {
342
+ await result.current.accept({ suggestion, shouldSubmit: true });
343
+ });
344
+ expect(mockSetValue).toHaveBeenCalledWith("What's the weather like today?");
345
+ expect(mockSubmit).toHaveBeenCalled();
346
+ });
347
+ it("throws error when suggestion has no detailed content", async () => {
348
+ jest.mocked(use_tambo_v1_1.useTamboV1).mockReturnValue({
349
+ messages: [
350
+ {
351
+ id: "msg_1",
352
+ role: "assistant",
353
+ content: [],
354
+ createdAt: "2024-01-01T00:00:00Z",
355
+ },
356
+ ],
357
+ thread: undefined,
358
+ isIdle: true,
359
+ isStreaming: false,
360
+ startNewThread: jest.fn(),
361
+ switchThread: jest.fn(),
362
+ initThread: jest.fn(),
363
+ streamingState: {
364
+ status: "idle",
365
+ },
366
+ });
367
+ const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_suggestions_1.useTamboV1Suggestions)(), {
368
+ wrapper: createWrapper(),
369
+ });
370
+ const emptySuggestion = {
371
+ id: "empty_suggestion",
372
+ suggestion: "Empty",
373
+ detailedSuggestion: "",
374
+ };
375
+ await expect(result.current.accept({ suggestion: emptySuggestion })).rejects.toThrow("Suggestion has no detailed content");
376
+ });
377
+ it("throws error when detailedSuggestion is only whitespace", async () => {
378
+ jest.mocked(use_tambo_v1_1.useTamboV1).mockReturnValue({
379
+ messages: [
380
+ {
381
+ id: "msg_1",
382
+ role: "assistant",
383
+ content: [],
384
+ createdAt: "2024-01-01T00:00:00Z",
385
+ },
386
+ ],
387
+ thread: undefined,
388
+ isIdle: true,
389
+ isStreaming: false,
390
+ startNewThread: jest.fn(),
391
+ switchThread: jest.fn(),
392
+ initThread: jest.fn(),
393
+ streamingState: {
394
+ status: "idle",
395
+ },
396
+ });
397
+ const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_suggestions_1.useTamboV1Suggestions)(), {
398
+ wrapper: createWrapper(),
399
+ });
400
+ const whitespaceSuggestion = {
401
+ id: "whitespace_suggestion",
402
+ suggestion: "Whitespace",
403
+ detailedSuggestion: " ",
404
+ };
405
+ await expect(result.current.accept({ suggestion: whitespaceSuggestion })).rejects.toThrow("Suggestion has no detailed content");
406
+ });
407
+ });
408
+ describe("State Management", () => {
409
+ it("resets selectedSuggestionId when message changes", async () => {
410
+ const mockUseTamboV1 = jest.mocked(use_tambo_v1_1.useTamboV1);
411
+ mockUseTamboV1.mockReturnValue({
412
+ messages: [
413
+ {
414
+ id: "msg_1",
415
+ role: "assistant",
416
+ content: [],
417
+ createdAt: "2024-01-01T00:00:00Z",
418
+ },
419
+ ],
420
+ thread: undefined,
421
+ isIdle: true,
422
+ isStreaming: false,
423
+ startNewThread: jest.fn(),
424
+ switchThread: jest.fn(),
425
+ initThread: jest.fn(),
426
+ streamingState: {
427
+ status: "idle",
428
+ },
429
+ });
430
+ const { result, rerender } = (0, react_1.renderHook)(() => (0, use_tambo_v1_suggestions_1.useTamboV1Suggestions)(), {
431
+ wrapper: createWrapper(),
432
+ });
433
+ await (0, react_1.waitFor)(() => {
434
+ expect(result.current.suggestions).toHaveLength(2);
435
+ });
436
+ // Accept a suggestion
437
+ await (0, react_1.act)(async () => {
438
+ await result.current.accept({
439
+ suggestion: result.current.suggestions[0],
440
+ });
441
+ });
442
+ expect(result.current.selectedSuggestionId).toBe("suggestion_1");
443
+ // Change the latest message
444
+ mockUseTamboV1.mockReturnValue({
445
+ messages: [
446
+ {
447
+ id: "msg_1",
448
+ role: "assistant",
449
+ content: [],
450
+ createdAt: "2024-01-01T00:00:00Z",
451
+ },
452
+ {
453
+ id: "msg_2",
454
+ role: "assistant",
455
+ content: [],
456
+ createdAt: "2024-01-01T00:00:01Z",
457
+ },
458
+ ],
459
+ thread: undefined,
460
+ isIdle: true,
461
+ isStreaming: false,
462
+ startNewThread: jest.fn(),
463
+ switchThread: jest.fn(),
464
+ initThread: jest.fn(),
465
+ streamingState: {
466
+ status: "idle",
467
+ },
468
+ });
469
+ rerender();
470
+ await (0, react_1.waitFor)(() => {
471
+ expect(result.current.selectedSuggestionId).toBeNull();
472
+ });
473
+ });
474
+ it("exposes mutation and query results", async () => {
475
+ jest.mocked(use_tambo_v1_1.useTamboV1).mockReturnValue({
476
+ messages: [
477
+ {
478
+ id: "msg_1",
479
+ role: "assistant",
480
+ content: [],
481
+ createdAt: "2024-01-01T00:00:00Z",
482
+ },
483
+ ],
484
+ thread: undefined,
485
+ isIdle: true,
486
+ isStreaming: false,
487
+ startNewThread: jest.fn(),
488
+ switchThread: jest.fn(),
489
+ initThread: jest.fn(),
490
+ streamingState: {
491
+ status: "idle",
492
+ },
493
+ });
494
+ const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_suggestions_1.useTamboV1Suggestions)(), {
495
+ wrapper: createWrapper(),
496
+ });
497
+ expect(result.current.acceptResult).toBeDefined();
498
+ expect(result.current.generateResult).toBeDefined();
499
+ expect(result.current.suggestionsResult).toBeDefined();
500
+ });
501
+ it("exposes combined mutation results (isPending, isError, error)", () => {
502
+ const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_suggestions_1.useTamboV1Suggestions)(), {
503
+ wrapper: createWrapper(),
504
+ });
505
+ expect(result.current.isPending).toBe(false);
506
+ expect(result.current.isError).toBe(false);
507
+ expect(result.current.error).toBeNull();
508
+ });
509
+ });
510
+ });
511
+ //# sourceMappingURL=use-tambo-v1-suggestions.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"use-tambo-v1-suggestions.test.js","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-suggestions.test.tsx"],"names":[],"mappings":";;;;;AAAA,kDAAkE;AAClE,kDAA0B;AAC1B,uDAAyE;AACzE,yEAAmE;AACnE,2EAAoE;AACpE,iDAA4C;AAC5C,iFAAuE;AACvE,qFAA2E;AAE3E,oBAAoB;AACpB,IAAI,CAAC,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9C,qBAAqB,EAAE,IAAI,CAAC,EAAE,EAAE;CACjC,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,GAAG,EAAE,CAAC,CAAC;IACjC,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,mBAAmB,EAAE,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,yBAAW,EAAE,CAAC;CACtD,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,yCAAyC,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;CAC5B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,IAAI,WAAwB,CAAC;IAC7B,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IAC/B,MAAM,UAAU,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IAC7B,MAAM,uBAAuB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IAE1C,SAAS,aAAa;QACpB,OAAO,SAAS,OAAO,CAAC,EAAE,QAAQ,EAAiC;YACjE,OAAO,CACL,8BAAC,iCAAmB,IAAC,MAAM,EAAE,WAAW,IACrC,QAAQ,CACW,CACvB,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,WAAW,GAAG,IAAI,yBAAW,CAAC;YAC5B,cAAc,EAAE;gBACd,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;gBACzB,SAAS,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;aAC5B;SACF,CAAC,CAAC;QAEH,gCAAgC;QAChC,IAAI,CAAC,MAAM,CAAC,iDAAqB,CAAC,CAAC,eAAe,CAAC;YACjD,KAAK,EAAE,EAAE;YACT,QAAQ,EAAE,YAAY;YACtB,MAAM,EAAE,UAAU;YAClB,QAAQ,EAAE,YAAY;YACtB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;YACtB,MAAM,EAAE,EAAE;YACV,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;YACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;YACpB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;YACtB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;YACtB,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,SAAS;SACV,CAAC,CAAC;QAEV,8BAA8B;QAC9B,IAAI,CAAC,MAAM,CAAC,yBAAU,CAAC,CAAC,eAAe,CAAC;YACtC,QAAQ,EAAE,EAAE;YACZ,MAAM,EAAE,SAAS;YACjB,MAAM,EAAE,IAAI;YACZ,WAAW,EAAE,KAAK;YAClB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;YACzB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;YACrB,cAAc,EAAE;gBACd,MAAM,EAAE,MAAM;aACf;SACK,CAAC,CAAC;QAEV,4BAA4B;QAC5B,IAAI,CAAC,MAAM,CAAC,0CAAgB,CAAC,CAAC,eAAe,CAAC;YAC5C,aAAa,EAAE,EAAE;YACjB,YAAY,EAAE,EAAE;YAChB,yBAAyB,EAAE,EAAE;YAC7B,cAAc,EAAE,EAAE;YAClB,SAAS,EAAE,EAAE;YACb,cAAc,EAAE,IAAI;YACpB,sBAAsB,EAAE,SAAS;YACjC,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC5B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC9B,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;YACzB,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC5B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC9B,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC3B,kBAAkB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC7B,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;SACtB,CAAC,CAAC;QAEV,0BAA0B;QAC1B,uBAAuB,CAAC,iBAAiB,CAAC;YACxC;gBACE,EAAE,EAAE,cAAc;gBAClB,UAAU,EAAE,qBAAqB;gBACjC,kBAAkB,EAAE,gCAAgC;aACrD;YACD;gBACE,EAAE,EAAE,cAAc;gBAClB,UAAU,EAAE,gBAAgB;gBAC5B,kBAAkB,EAAE,+BAA+B;aACpD;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CAAC;YAC1C,IAAI,EAAE;gBACJ,OAAO,EAAE;oBACP,WAAW,EAAE;wBACX,QAAQ,EAAE,uBAAuB;qBAClC;iBACF;aACF;SACK,CAAC,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,gDAAqB,GAAE,EAAE;gBAC3D,OAAO,EAAE,aAAa,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,QAAQ,EAAE,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,IAAI,CAAC,MAAM,CAAC,yBAAU,CAAC,CAAC,eAAe,CAAC;gBACtC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,OAAO;wBACX,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,sBAAsB;qBAClC;iBACF;gBACD,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,KAAK;gBAClB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;gBACzB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,MAAM;iBACf;aACK,CAAC,CAAC;YAEV,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,gDAAqB,GAAE,EAAE;gBAC3D,OAAO,EAAE,aAAa,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;YAC9F,IAAI,CAAC,MAAM,CAAC,yBAAU,CAAC,CAAC,eAAe,CAAC;gBACtC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,OAAO;wBACX,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,sBAAsB;qBAClC;iBACF;gBACD,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,KAAK;gBAClB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;gBACzB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,MAAM;iBACf;aACK,CAAC,CAAC;YAEV,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,gDAAqB,GAAE,EAAE;gBAC3D,OAAO,EAAE,aAAa,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;gBACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,uBAAuB,CAAC,CAAC,oBAAoB,CAClD,OAAO,EACP,MAAM,CAAC,gBAAgB,CAAC;gBACtB,EAAE,EAAE,YAAY;gBAChB,cAAc,EAAE,CAAC;aAClB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,IAAI,CAAC,MAAM,CAAC,yBAAU,CAAC,CAAC,eAAe,CAAC;gBACtC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,OAAO;wBACX,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,sBAAsB;qBAClC;iBACF;gBACD,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,KAAK;gBACb,WAAW,EAAE,IAAI;gBACjB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;gBACzB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,WAAW;oBACnB,KAAK,EAAE,OAAO;iBACf;aACK,CAAC,CAAC;YAEV,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,gDAAqB,GAAE,EAAE;gBACxC,OAAO,EAAE,aAAa,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,CAAC,uBAAuB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,IAAI,CAAC,MAAM,CAAC,iDAAqB,CAAC,CAAC,eAAe,CAAC;gBACjD,KAAK,EAAE,EAAE;gBACT,QAAQ,EAAE,YAAY;gBACtB,MAAM,EAAE,UAAU;gBAClB,QAAQ,EAAE,SAAS;gBACnB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;gBACtB,MAAM,EAAE,EAAE;gBACV,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;gBACnB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;gBACtB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;gBACtB,SAAS,EAAE,KAAK;gBAChB,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,SAAS;aACV,CAAC,CAAC;YAEV,IAAI,CAAC,MAAM,CAAC,yBAAU,CAAC,CAAC,eAAe,CAAC;gBACtC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,OAAO;wBACX,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,sBAAsB;qBAClC;iBACF;gBACD,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,KAAK;gBAClB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;gBACzB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,MAAM;iBACf;aACK,CAAC,CAAC;YAEV,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,gDAAqB,GAAE,EAAE;gBACxC,OAAO,EAAE,aAAa,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,CAAC,uBAAuB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,IAAI,CAAC,MAAM,CAAC,yBAAU,CAAC,CAAC,eAAe,CAAC;gBACtC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,OAAO;wBACX,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,sBAAsB;qBAClC;iBACF;gBACD,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,KAAK;gBAClB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;gBACzB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,MAAM;iBACf;aACK,CAAC,CAAC;YAEV,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAC3B,GAAG,EAAE,CAAC,IAAA,gDAAqB,EAAC,EAAE,cAAc,EAAE,CAAC,EAAE,CAAC,EAClD,EAAE,OAAO,EAAE,aAAa,EAAE,EAAE,CAC7B,CAAC;YAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;gBACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,uBAAuB,CAAC,CAAC,oBAAoB,CAClD,OAAO,EACP,MAAM,CAAC,gBAAgB,CAAC;gBACtB,cAAc,EAAE,CAAC;aAClB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,IAAI,CAAC,MAAM,CAAC,yBAAU,CAAC,CAAC,eAAe,CAAC;gBACtC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,OAAO;wBACX,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,sBAAsB;qBAClC;iBACF;gBACD,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,KAAK;gBAClB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;gBACzB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,MAAM;iBACf;aACK,CAAC,CAAC;YAEV,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,gDAAqB,GAAE,EAAE;gBAC3D,OAAO,EAAE,aAAa,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;gBACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YAEjD,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;gBACnB,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,gCAAgC,CACjC,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,UAAU,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;YAEzD,IAAI,CAAC,MAAM,CAAC,yBAAU,CAAC,CAAC,eAAe,CAAC;gBACtC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,OAAO;wBACX,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,sBAAsB;qBAClC;iBACF;gBACD,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,KAAK;gBAClB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;gBACzB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,MAAM;iBACf;aACK,CAAC,CAAC;YAEV,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,gDAAqB,GAAE,EAAE;gBAC3D,OAAO,EAAE,aAAa,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;gBACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YAEjD,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;gBACnB,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;YAClE,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,gCAAgC,CACjC,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,IAAI,CAAC,MAAM,CAAC,yBAAU,CAAC,CAAC,eAAe,CAAC;gBACtC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,OAAO;wBACX,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,sBAAsB;qBAClC;iBACF;gBACD,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,KAAK;gBAClB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;gBACzB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,MAAM;iBACf;aACK,CAAC,CAAC;YAEV,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,gDAAqB,GAAE,EAAE;gBAC3D,OAAO,EAAE,aAAa,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,eAAe,GAAG;gBACtB,EAAE,EAAE,kBAAkB;gBACtB,UAAU,EAAE,OAAO;gBACnB,kBAAkB,EAAE,EAAE;aACvB,CAAC;YAEF,MAAM,MAAM,CACV,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,eAAsB,EAAE,CAAC,CAC9D,CAAC,OAAO,CAAC,OAAO,CAAC,oCAAoC,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,IAAI,CAAC,MAAM,CAAC,yBAAU,CAAC,CAAC,eAAe,CAAC;gBACtC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,OAAO;wBACX,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,sBAAsB;qBAClC;iBACF;gBACD,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,KAAK;gBAClB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;gBACzB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,MAAM;iBACf;aACK,CAAC,CAAC;YAEV,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,gDAAqB,GAAE,EAAE;gBAC3D,OAAO,EAAE,aAAa,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,oBAAoB,GAAG;gBAC3B,EAAE,EAAE,uBAAuB;gBAC3B,UAAU,EAAE,YAAY;gBACxB,kBAAkB,EAAE,KAAK;aAC1B,CAAC;YAEF,MAAM,MAAM,CACV,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,oBAA2B,EAAE,CAAC,CACnE,CAAC,OAAO,CAAC,OAAO,CAAC,oCAAoC,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,yBAAU,CAAC,CAAC;YAE/C,cAAc,CAAC,eAAe,CAAC;gBAC7B,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,OAAO;wBACX,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,sBAAsB;qBAClC;iBACF;gBACD,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,KAAK;gBAClB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;gBACzB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,MAAM;iBACf;aACK,CAAC,CAAC;YAEV,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,gDAAqB,GAAE,EAAE;gBACrE,OAAO,EAAE,aAAa,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;gBACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YAEH,sBAAsB;YACtB,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;gBACnB,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;oBAC1B,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;iBAC1C,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAEjE,4BAA4B;YAC5B,cAAc,CAAC,eAAe,CAAC;gBAC7B,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,OAAO;wBACX,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,sBAAsB;qBAClC;oBACD;wBACE,EAAE,EAAE,OAAO;wBACX,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,sBAAsB;qBAClC;iBACF;gBACD,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,KAAK;gBAClB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;gBACzB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,MAAM;iBACf;aACK,CAAC,CAAC;YAEV,QAAQ,EAAE,CAAC;YAEX,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;gBACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,QAAQ,EAAE,CAAC;YACzD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,IAAI,CAAC,MAAM,CAAC,yBAAU,CAAC,CAAC,eAAe,CAAC;gBACtC,QAAQ,EAAE;oBACR;wBACE,EAAE,EAAE,OAAO;wBACX,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,EAAE;wBACX,SAAS,EAAE,sBAAsB;qBAClC;iBACF;gBACD,MAAM,EAAE,SAAS;gBACjB,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,KAAK;gBAClB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;gBACzB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;gBACrB,cAAc,EAAE;oBACd,MAAM,EAAE,MAAM;iBACf;aACK,CAAC,CAAC;YAEV,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,gDAAqB,GAAE,EAAE;gBAC3D,OAAO,EAAE,aAAa,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,WAAW,EAAE,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,WAAW,EAAE,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,WAAW,EAAE,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,gDAAqB,GAAE,EAAE;gBAC3D,OAAO,EAAE,aAAa,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC3C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { renderHook, act, waitFor } from \"@testing-library/react\";\nimport React from \"react\";\nimport { QueryClient, QueryClientProvider } from \"@tanstack/react-query\";\nimport { useTamboV1Suggestions } from \"./use-tambo-v1-suggestions\";\nimport { useTamboV1ThreadInput } from \"./use-tambo-v1-thread-input\";\nimport { useTamboV1 } from \"./use-tambo-v1\";\nimport { useTamboClient } from \"../../providers/tambo-client-provider\";\nimport { useTamboRegistry } from \"../../providers/tambo-registry-provider\";\n\n// Mock dependencies\njest.mock(\"./use-tambo-v1-thread-input\", () => ({\n useTamboV1ThreadInput: jest.fn(),\n}));\n\njest.mock(\"./use-tambo-v1\", () => ({\n useTamboV1: jest.fn(),\n}));\n\njest.mock(\"../../providers/tambo-client-provider\", () => ({\n useTamboClient: jest.fn(),\n useTamboQueryClient: jest.fn(() => new QueryClient()),\n}));\n\njest.mock(\"../../providers/tambo-registry-provider\", () => ({\n useTamboRegistry: jest.fn(),\n}));\n\ndescribe(\"useTamboV1Suggestions\", () => {\n let queryClient: QueryClient;\n const mockSetValue = jest.fn();\n const mockSubmit = jest.fn();\n const mockGenerateSuggestions = jest.fn();\n\n function createWrapper() {\n return function Wrapper({ children }: { children: React.ReactNode }) {\n return (\n <QueryClientProvider client={queryClient}>\n {children}\n </QueryClientProvider>\n );\n };\n }\n\n beforeEach(() => {\n jest.clearAllMocks();\n queryClient = new QueryClient({\n defaultOptions: {\n queries: { retry: false },\n mutations: { retry: false },\n },\n });\n\n // Default mock for thread input\n jest.mocked(useTamboV1ThreadInput).mockReturnValue({\n value: \"\",\n setValue: mockSetValue,\n submit: mockSubmit,\n threadId: \"thread_123\",\n setThreadId: jest.fn(),\n images: [],\n addImage: jest.fn(),\n addImages: jest.fn(),\n removeImage: jest.fn(),\n clearImages: jest.fn(),\n isPending: false,\n isError: false,\n error: undefined,\n } as any);\n\n // Default mock for useTamboV1\n jest.mocked(useTamboV1).mockReturnValue({\n messages: [],\n thread: undefined,\n isIdle: true,\n isStreaming: false,\n startNewThread: jest.fn(),\n switchThread: jest.fn(),\n initThread: jest.fn(),\n streamingState: {\n status: \"idle\",\n },\n } as any);\n\n // Default mock for registry\n jest.mocked(useTamboRegistry).mockReturnValue({\n componentList: {},\n toolRegistry: {},\n componentToolAssociations: {},\n mcpServerInfos: [],\n resources: [],\n resourceSource: null,\n onCallUnregisteredTool: undefined,\n registerComponent: jest.fn(),\n unregisterComponent: jest.fn(),\n registerTool: jest.fn(),\n unregisterTool: jest.fn(),\n registerMcpServer: jest.fn(),\n unregisterMcpServer: jest.fn(),\n registerResource: jest.fn(),\n unregisterResource: jest.fn(),\n setResourceSource: jest.fn(),\n } as any);\n\n // Default mock for client\n mockGenerateSuggestions.mockResolvedValue([\n {\n id: \"suggestion_1\",\n suggestion: \"What's the weather?\",\n detailedSuggestion: \"What's the weather like today?\",\n },\n {\n id: \"suggestion_2\",\n suggestion: \"Tell me a joke\",\n detailedSuggestion: \"Can you tell me a funny joke?\",\n },\n ]);\n\n jest.mocked(useTamboClient).mockReturnValue({\n beta: {\n threads: {\n suggestions: {\n generate: mockGenerateSuggestions,\n },\n },\n },\n } as any);\n });\n\n describe(\"Initial State\", () => {\n it(\"returns empty suggestions when no messages\", () => {\n const { result } = renderHook(() => useTamboV1Suggestions(), {\n wrapper: createWrapper(),\n });\n\n expect(result.current.suggestions).toEqual([]);\n expect(result.current.selectedSuggestionId).toBeNull();\n });\n\n it(\"returns empty suggestions when latest message is from user\", () => {\n jest.mocked(useTamboV1).mockReturnValue({\n messages: [\n {\n id: \"msg_1\",\n role: \"user\",\n content: [],\n createdAt: \"2024-01-01T00:00:00Z\",\n },\n ],\n thread: undefined,\n isIdle: true,\n isStreaming: false,\n startNewThread: jest.fn(),\n switchThread: jest.fn(),\n initThread: jest.fn(),\n streamingState: {\n status: \"idle\",\n },\n } as any);\n\n const { result } = renderHook(() => useTamboV1Suggestions(), {\n wrapper: createWrapper(),\n });\n\n expect(result.current.suggestions).toEqual([]);\n });\n });\n\n describe(\"Suggestion Generation\", () => {\n it(\"generates suggestions when latest message is from assistant and thread is idle\", async () => {\n jest.mocked(useTamboV1).mockReturnValue({\n messages: [\n {\n id: \"msg_1\",\n role: \"assistant\",\n content: [],\n createdAt: \"2024-01-01T00:00:00Z\",\n },\n ],\n thread: undefined,\n isIdle: true,\n isStreaming: false,\n startNewThread: jest.fn(),\n switchThread: jest.fn(),\n initThread: jest.fn(),\n streamingState: {\n status: \"idle\",\n },\n } as any);\n\n const { result } = renderHook(() => useTamboV1Suggestions(), {\n wrapper: createWrapper(),\n });\n\n await waitFor(() => {\n expect(result.current.suggestions).toHaveLength(2);\n });\n\n expect(mockGenerateSuggestions).toHaveBeenCalledWith(\n \"msg_1\",\n expect.objectContaining({\n id: \"thread_123\",\n maxSuggestions: 3,\n }),\n );\n });\n\n it(\"does not generate suggestions when thread is streaming\", () => {\n jest.mocked(useTamboV1).mockReturnValue({\n messages: [\n {\n id: \"msg_1\",\n role: \"assistant\",\n content: [],\n createdAt: \"2024-01-01T00:00:00Z\",\n },\n ],\n thread: undefined,\n isIdle: false,\n isStreaming: true,\n startNewThread: jest.fn(),\n switchThread: jest.fn(),\n initThread: jest.fn(),\n streamingState: {\n status: \"streaming\",\n runId: \"run_1\",\n },\n } as any);\n\n renderHook(() => useTamboV1Suggestions(), {\n wrapper: createWrapper(),\n });\n\n expect(mockGenerateSuggestions).not.toHaveBeenCalled();\n });\n\n it(\"does not generate suggestions when no threadId\", () => {\n jest.mocked(useTamboV1ThreadInput).mockReturnValue({\n value: \"\",\n setValue: mockSetValue,\n submit: mockSubmit,\n threadId: undefined,\n setThreadId: jest.fn(),\n images: [],\n addImage: jest.fn(),\n addImages: jest.fn(),\n removeImage: jest.fn(),\n clearImages: jest.fn(),\n isPending: false,\n isError: false,\n error: undefined,\n } as any);\n\n jest.mocked(useTamboV1).mockReturnValue({\n messages: [\n {\n id: \"msg_1\",\n role: \"assistant\",\n content: [],\n createdAt: \"2024-01-01T00:00:00Z\",\n },\n ],\n thread: undefined,\n isIdle: true,\n isStreaming: false,\n startNewThread: jest.fn(),\n switchThread: jest.fn(),\n initThread: jest.fn(),\n streamingState: {\n status: \"idle\",\n },\n } as any);\n\n renderHook(() => useTamboV1Suggestions(), {\n wrapper: createWrapper(),\n });\n\n expect(mockGenerateSuggestions).not.toHaveBeenCalled();\n });\n\n it(\"uses custom maxSuggestions option\", async () => {\n jest.mocked(useTamboV1).mockReturnValue({\n messages: [\n {\n id: \"msg_1\",\n role: \"assistant\",\n content: [],\n createdAt: \"2024-01-01T00:00:00Z\",\n },\n ],\n thread: undefined,\n isIdle: true,\n isStreaming: false,\n startNewThread: jest.fn(),\n switchThread: jest.fn(),\n initThread: jest.fn(),\n streamingState: {\n status: \"idle\",\n },\n } as any);\n\n const { result } = renderHook(\n () => useTamboV1Suggestions({ maxSuggestions: 5 }),\n { wrapper: createWrapper() },\n );\n\n await waitFor(() => {\n expect(result.current.suggestions).toHaveLength(2);\n });\n\n expect(mockGenerateSuggestions).toHaveBeenCalledWith(\n \"msg_1\",\n expect.objectContaining({\n maxSuggestions: 5,\n }),\n );\n });\n });\n\n describe(\"Accepting Suggestions\", () => {\n it(\"updates shared input value when accepting without submit\", async () => {\n jest.mocked(useTamboV1).mockReturnValue({\n messages: [\n {\n id: \"msg_1\",\n role: \"assistant\",\n content: [],\n createdAt: \"2024-01-01T00:00:00Z\",\n },\n ],\n thread: undefined,\n isIdle: true,\n isStreaming: false,\n startNewThread: jest.fn(),\n switchThread: jest.fn(),\n initThread: jest.fn(),\n streamingState: {\n status: \"idle\",\n },\n } as any);\n\n const { result } = renderHook(() => useTamboV1Suggestions(), {\n wrapper: createWrapper(),\n });\n\n await waitFor(() => {\n expect(result.current.suggestions).toHaveLength(2);\n });\n\n const suggestion = result.current.suggestions[0];\n\n await act(async () => {\n await result.current.accept({ suggestion });\n });\n\n expect(mockSetValue).toHaveBeenCalledWith(\n \"What's the weather like today?\",\n );\n expect(mockSubmit).not.toHaveBeenCalled();\n expect(result.current.selectedSuggestionId).toBe(\"suggestion_1\");\n });\n\n it(\"submits when accepting with shouldSubmit=true\", async () => {\n mockSubmit.mockResolvedValue({ threadId: \"thread_123\" });\n\n jest.mocked(useTamboV1).mockReturnValue({\n messages: [\n {\n id: \"msg_1\",\n role: \"assistant\",\n content: [],\n createdAt: \"2024-01-01T00:00:00Z\",\n },\n ],\n thread: undefined,\n isIdle: true,\n isStreaming: false,\n startNewThread: jest.fn(),\n switchThread: jest.fn(),\n initThread: jest.fn(),\n streamingState: {\n status: \"idle\",\n },\n } as any);\n\n const { result } = renderHook(() => useTamboV1Suggestions(), {\n wrapper: createWrapper(),\n });\n\n await waitFor(() => {\n expect(result.current.suggestions).toHaveLength(2);\n });\n\n const suggestion = result.current.suggestions[0];\n\n await act(async () => {\n await result.current.accept({ suggestion, shouldSubmit: true });\n });\n\n expect(mockSetValue).toHaveBeenCalledWith(\n \"What's the weather like today?\",\n );\n expect(mockSubmit).toHaveBeenCalled();\n });\n\n it(\"throws error when suggestion has no detailed content\", async () => {\n jest.mocked(useTamboV1).mockReturnValue({\n messages: [\n {\n id: \"msg_1\",\n role: \"assistant\",\n content: [],\n createdAt: \"2024-01-01T00:00:00Z\",\n },\n ],\n thread: undefined,\n isIdle: true,\n isStreaming: false,\n startNewThread: jest.fn(),\n switchThread: jest.fn(),\n initThread: jest.fn(),\n streamingState: {\n status: \"idle\",\n },\n } as any);\n\n const { result } = renderHook(() => useTamboV1Suggestions(), {\n wrapper: createWrapper(),\n });\n\n const emptySuggestion = {\n id: \"empty_suggestion\",\n suggestion: \"Empty\",\n detailedSuggestion: \"\",\n };\n\n await expect(\n result.current.accept({ suggestion: emptySuggestion as any }),\n ).rejects.toThrow(\"Suggestion has no detailed content\");\n });\n\n it(\"throws error when detailedSuggestion is only whitespace\", async () => {\n jest.mocked(useTamboV1).mockReturnValue({\n messages: [\n {\n id: \"msg_1\",\n role: \"assistant\",\n content: [],\n createdAt: \"2024-01-01T00:00:00Z\",\n },\n ],\n thread: undefined,\n isIdle: true,\n isStreaming: false,\n startNewThread: jest.fn(),\n switchThread: jest.fn(),\n initThread: jest.fn(),\n streamingState: {\n status: \"idle\",\n },\n } as any);\n\n const { result } = renderHook(() => useTamboV1Suggestions(), {\n wrapper: createWrapper(),\n });\n\n const whitespaceSuggestion = {\n id: \"whitespace_suggestion\",\n suggestion: \"Whitespace\",\n detailedSuggestion: \" \",\n };\n\n await expect(\n result.current.accept({ suggestion: whitespaceSuggestion as any }),\n ).rejects.toThrow(\"Suggestion has no detailed content\");\n });\n });\n\n describe(\"State Management\", () => {\n it(\"resets selectedSuggestionId when message changes\", async () => {\n const mockUseTamboV1 = jest.mocked(useTamboV1);\n\n mockUseTamboV1.mockReturnValue({\n messages: [\n {\n id: \"msg_1\",\n role: \"assistant\",\n content: [],\n createdAt: \"2024-01-01T00:00:00Z\",\n },\n ],\n thread: undefined,\n isIdle: true,\n isStreaming: false,\n startNewThread: jest.fn(),\n switchThread: jest.fn(),\n initThread: jest.fn(),\n streamingState: {\n status: \"idle\",\n },\n } as any);\n\n const { result, rerender } = renderHook(() => useTamboV1Suggestions(), {\n wrapper: createWrapper(),\n });\n\n await waitFor(() => {\n expect(result.current.suggestions).toHaveLength(2);\n });\n\n // Accept a suggestion\n await act(async () => {\n await result.current.accept({\n suggestion: result.current.suggestions[0],\n });\n });\n\n expect(result.current.selectedSuggestionId).toBe(\"suggestion_1\");\n\n // Change the latest message\n mockUseTamboV1.mockReturnValue({\n messages: [\n {\n id: \"msg_1\",\n role: \"assistant\",\n content: [],\n createdAt: \"2024-01-01T00:00:00Z\",\n },\n {\n id: \"msg_2\",\n role: \"assistant\",\n content: [],\n createdAt: \"2024-01-01T00:00:01Z\",\n },\n ],\n thread: undefined,\n isIdle: true,\n isStreaming: false,\n startNewThread: jest.fn(),\n switchThread: jest.fn(),\n initThread: jest.fn(),\n streamingState: {\n status: \"idle\",\n },\n } as any);\n\n rerender();\n\n await waitFor(() => {\n expect(result.current.selectedSuggestionId).toBeNull();\n });\n });\n\n it(\"exposes mutation and query results\", async () => {\n jest.mocked(useTamboV1).mockReturnValue({\n messages: [\n {\n id: \"msg_1\",\n role: \"assistant\",\n content: [],\n createdAt: \"2024-01-01T00:00:00Z\",\n },\n ],\n thread: undefined,\n isIdle: true,\n isStreaming: false,\n startNewThread: jest.fn(),\n switchThread: jest.fn(),\n initThread: jest.fn(),\n streamingState: {\n status: \"idle\",\n },\n } as any);\n\n const { result } = renderHook(() => useTamboV1Suggestions(), {\n wrapper: createWrapper(),\n });\n\n expect(result.current.acceptResult).toBeDefined();\n expect(result.current.generateResult).toBeDefined();\n expect(result.current.suggestionsResult).toBeDefined();\n });\n\n it(\"exposes combined mutation results (isPending, isError, error)\", () => {\n const { result } = renderHook(() => useTamboV1Suggestions(), {\n wrapper: createWrapper(),\n });\n\n expect(result.current.isPending).toBe(false);\n expect(result.current.isError).toBe(false);\n expect(result.current.error).toBeNull();\n });\n });\n});\n"]}
@@ -1,62 +1,11 @@
1
1
  /**
2
- * Options for submitting a message
3
- */
4
- export interface SubmitOptions {
5
- /**
6
- * Enable debug logging for the stream
7
- */
8
- debug?: boolean;
9
- }
10
- /**
11
- * Return type for useTamboV1ThreadInput hook
12
- */
13
- export interface UseTamboV1ThreadInputReturn {
14
- /** Current value of the input field */
15
- value: string;
16
- /** Function to update the input value */
17
- setValue: React.Dispatch<React.SetStateAction<string>>;
18
- /** Submit the current input. Clears input on success. */
19
- submit: (options?: SubmitOptions) => Promise<{
20
- threadId: string | undefined;
21
- }>;
22
- isPending: boolean;
23
- isError: boolean;
24
- error: Error | null;
25
- isSuccess: boolean;
26
- reset: () => void;
27
- }
28
- /**
29
- * Hook to manage thread input state and message submission.
30
- *
31
- * Provides a similar API to the beta SDK's useTamboThreadInput,
32
- * managing input value state and providing a submit function.
33
- * @param threadId - Optional thread ID to send messages to. If not provided, creates new thread
34
- * @returns Thread input state and submit function
35
- * @example
36
- * ```tsx
37
- * function ChatInput({ threadId }: { threadId?: string }) {
38
- * const { value, setValue, submit, isPending } = useTamboV1ThreadInput(threadId);
2
+ * useTamboV1ThreadInput - Thread Input Hook for v1 API
39
3
  *
40
- * const handleSubmit = async (e: React.FormEvent) => {
41
- * e.preventDefault();
42
- * const result = await submit();
43
- * console.log('Sent to thread:', result.threadId);
44
- * };
4
+ * Re-exports the shared thread input hook from the provider.
5
+ * This hook uses shared context, enabling features like suggestions
6
+ * to update the input field directly.
45
7
  *
46
- * return (
47
- * <form onSubmit={handleSubmit}>
48
- * <input
49
- * value={value}
50
- * onChange={(e) => setValue(e.target.value)}
51
- * disabled={isPending}
52
- * />
53
- * <button type="submit" disabled={isPending || !value.trim()}>
54
- * Send
55
- * </button>
56
- * </form>
57
- * );
58
- * }
59
- * ```
8
+ * For direct thread control without shared state, use useTamboV1SendMessage instead.
60
9
  */
61
- export declare function useTamboV1ThreadInput(threadId?: string): UseTamboV1ThreadInputReturn;
10
+ export { useTamboV1ThreadInput, type TamboV1ThreadInputContextProps, type SubmitOptions, } from "../providers/tambo-v1-thread-input-provider";
62
11
  //# sourceMappingURL=use-tambo-v1-thread-input.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"use-tambo-v1-thread-input.d.ts","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-thread-input.ts"],"names":[],"mappings":"AAYA;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B;;OAEG;IACH,KAAK,CAAC,EAAE,OAAO,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,2BAA2B;IAC1C,uCAAuC;IACvC,KAAK,EAAE,MAAM,CAAC;IACd,yCAAyC;IACzC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;IACvD,yDAAyD;IACzD,MAAM,EAAE,CACN,OAAO,CAAC,EAAE,aAAa,KACpB,OAAO,CAAC;QAAE,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAA;KAAE,CAAC,CAAC;IAC/C,SAAS,EAAE,OAAO,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,EAAE,KAAK,GAAG,IAAI,CAAC;IACpB,SAAS,EAAE,OAAO,CAAC;IACnB,KAAK,EAAE,MAAM,IAAI,CAAC;CACnB;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,wBAAgB,qBAAqB,CACnC,QAAQ,CAAC,EAAE,MAAM,GAChB,2BAA2B,CAqC7B"}
1
+ {"version":3,"file":"use-tambo-v1-thread-input.d.ts","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-thread-input.ts"],"names":[],"mappings":"AAEA;;;;;;;;GAQG;AAEH,OAAO,EACL,qBAAqB,EACrB,KAAK,8BAA8B,EACnC,KAAK,aAAa,GACnB,MAAM,6CAA6C,CAAC"}