1
1
import { LDContext } from '@launchdarkly/js-server-sdk-common' ;
2
2
3
3
import { LDAIAgentDefaults } from '../src/api/agents' ;
4
- import {
5
- LDAIDefaults ,
6
- VercelAISDKConfig ,
7
- VercelAISDKMapOptions ,
8
- VercelAISDKProvider ,
9
- } from '../src/api/config' ;
4
+ import { LDAIDefaults } from '../src/api/config' ;
10
5
import { LDAIClientImpl } from '../src/LDAIClientImpl' ;
11
6
import { LDClientMin } from '../src/LDClientMin' ;
12
7
@@ -17,7 +12,7 @@ const mockLdClient: jest.Mocked<LDClientMin> = {
17
12
18
13
const testContext : LDContext = { kind : 'user' , key : 'test-user' } ;
19
14
20
- it ( 'returns config with interpolated messagess ' , async ( ) => {
15
+ it ( 'returns config with interpolated messages ' , async ( ) => {
21
16
const client = new LDAIClientImpl ( mockLdClient ) ;
22
17
const key = 'test-flag' ;
23
18
const defaultValue : LDAIDefaults = {
@@ -156,21 +151,6 @@ it('returns single agent config with interpolated instructions', async () => {
156
151
model : { name : 'test' , parameters : { name : 'test-model' } } ,
157
152
instructions : 'You are a helpful assistant.' ,
158
153
enabled : true ,
159
- toVercelAISDK : < TMod > (
160
- provider : VercelAISDKProvider < TMod > | Record < string , VercelAISDKProvider < TMod > > ,
161
- options ?: VercelAISDKMapOptions ,
162
- ) : VercelAISDKConfig < TMod > => {
163
- const modelProvider = typeof provider === 'function' ? provider : provider . test ;
164
- return {
165
- model : modelProvider ( 'test-model' ) ,
166
- messages : [ ] ,
167
- ...( options ?. nonInterpolatedMessages
168
- ? {
169
- messages : options . nonInterpolatedMessages ,
170
- }
171
- : { } ) ,
172
- } ;
173
- } ,
174
154
} ;
175
155
176
156
const mockVariation = {
@@ -205,7 +185,6 @@ it('returns single agent config with interpolated instructions', async () => {
205
185
instructions : 'You are a helpful assistant. Your name is John and your score is 42' ,
206
186
tracker : expect . any ( Object ) ,
207
187
enabled : true ,
208
- toVercelAISDK : expect . any ( Function ) ,
209
188
} ) ;
210
189
211
190
// Verify tracking was called
@@ -224,21 +203,6 @@ it('includes context in variables for agent instructions interpolation', async (
224
203
model : { name : 'test' , parameters : { name : 'test-model' } } ,
225
204
instructions : 'You are a helpful assistant.' ,
226
205
enabled : true ,
227
- toVercelAISDK : < TMod > (
228
- provider : VercelAISDKProvider < TMod > | Record < string , VercelAISDKProvider < TMod > > ,
229
- options ?: VercelAISDKMapOptions ,
230
- ) : VercelAISDKConfig < TMod > => {
231
- const modelProvider = typeof provider === 'function' ? provider : provider . test ;
232
- return {
233
- model : modelProvider ( 'test-model' ) ,
234
- messages : [ ] ,
235
- ...( options ?. nonInterpolatedMessages
236
- ? {
237
- messages : options . nonInterpolatedMessages ,
238
- }
239
- : { } ) ,
240
- } ;
241
- } ,
242
206
} ;
243
207
244
208
const mockVariation = {
@@ -260,21 +224,6 @@ it('handles missing metadata in agent variation', async () => {
260
224
model : { name : 'test' , parameters : { name : 'test-model' } } ,
261
225
instructions : 'You are a helpful assistant.' ,
262
226
enabled : true ,
263
- toVercelAISDK : < TMod > (
264
- provider : VercelAISDKProvider < TMod > | Record < string , VercelAISDKProvider < TMod > > ,
265
- options ?: VercelAISDKMapOptions ,
266
- ) : VercelAISDKConfig < TMod > => {
267
- const modelProvider = typeof provider === 'function' ? provider : provider . test ;
268
- return {
269
- model : modelProvider ( 'test-model' ) ,
270
- messages : [ ] ,
271
- ...( options ?. nonInterpolatedMessages
272
- ? {
273
- messages : options . nonInterpolatedMessages ,
274
- }
275
- : { } ) ,
276
- } ;
277
- } ,
278
227
} ;
279
228
280
229
const mockVariation = {
@@ -291,7 +240,6 @@ it('handles missing metadata in agent variation', async () => {
291
240
instructions : 'Hello.' ,
292
241
tracker : expect . any ( Object ) ,
293
242
enabled : false ,
294
- toVercelAISDK : expect . any ( Function ) ,
295
243
} ) ;
296
244
} ) ;
297
245
@@ -303,22 +251,6 @@ it('passes the default value to the underlying client for single agent', async (
303
251
provider : { name : 'default-provider' } ,
304
252
instructions : 'Default instructions' ,
305
253
enabled : true ,
306
- toVercelAISDK : < TMod > (
307
- provider : VercelAISDKProvider < TMod > | Record < string , VercelAISDKProvider < TMod > > ,
308
- options ?: VercelAISDKMapOptions ,
309
- ) : VercelAISDKConfig < TMod > => {
310
- const modelProvider =
311
- typeof provider === 'function' ? provider : provider [ 'default-provider' ] ;
312
- return {
313
- model : modelProvider ( 'default-model' ) ,
314
- messages : [ ] ,
315
- ...( options ?. nonInterpolatedMessages
316
- ? {
317
- messages : options . nonInterpolatedMessages ,
318
- }
319
- : { } ) ,
320
- } ;
321
- } ,
322
254
} ;
323
255
324
256
mockLdClient . variation . mockResolvedValue ( defaultValue ) ;
@@ -331,7 +263,6 @@ it('passes the default value to the underlying client for single agent', async (
331
263
provider : defaultValue . provider ,
332
264
tracker : expect . any ( Object ) ,
333
265
enabled : false ,
334
- toVercelAISDK : expect . any ( Function ) ,
335
266
} ) ;
336
267
337
268
expect ( mockLdClient . variation ) . toHaveBeenCalledWith ( key , testContext , defaultValue ) ;
@@ -347,21 +278,6 @@ it('returns multiple agents config with interpolated instructions', async () =>
347
278
model : { name : 'test' , parameters : { name : 'test-model' } } ,
348
279
instructions : 'You are a research assistant.' ,
349
280
enabled : true ,
350
- toVercelAISDK : < TMod > (
351
- provider : VercelAISDKProvider < TMod > | Record < string , VercelAISDKProvider < TMod > > ,
352
- options ?: VercelAISDKMapOptions ,
353
- ) : VercelAISDKConfig < TMod > => {
354
- const modelProvider = typeof provider === 'function' ? provider : provider . test ;
355
- return {
356
- model : modelProvider ( 'test-model' ) ,
357
- messages : [ ] ,
358
- ...( options ?. nonInterpolatedMessages
359
- ? {
360
- messages : options . nonInterpolatedMessages ,
361
- }
362
- : { } ) ,
363
- } ;
364
- } ,
365
281
} ,
366
282
variables : { topic : 'climate change' } ,
367
283
} ,
@@ -371,21 +287,6 @@ it('returns multiple agents config with interpolated instructions', async () =>
371
287
model : { name : 'test' , parameters : { name : 'test-model' } } ,
372
288
instructions : 'You are a writing assistant.' ,
373
289
enabled : true ,
374
- toVercelAISDK : < TMod > (
375
- provider : VercelAISDKProvider < TMod > | Record < string , VercelAISDKProvider < TMod > > ,
376
- options ?: VercelAISDKMapOptions ,
377
- ) : VercelAISDKConfig < TMod > => {
378
- const modelProvider = typeof provider === 'function' ? provider : provider . test ;
379
- return {
380
- model : modelProvider ( 'test-model' ) ,
381
- messages : [ ] ,
382
- ...( options ?. nonInterpolatedMessages
383
- ? {
384
- messages : options . nonInterpolatedMessages ,
385
- }
386
- : { } ) ,
387
- } ;
388
- } ,
389
290
} ,
390
291
variables : { style : 'academic' } ,
391
292
} ,
@@ -428,7 +329,6 @@ it('returns multiple agents config with interpolated instructions', async () =>
428
329
instructions : 'You are a research assistant specializing in climate change.' ,
429
330
tracker : expect . any ( Object ) ,
430
331
enabled : true ,
431
- toVercelAISDK : expect . any ( Function ) ,
432
332
} ,
433
333
'writing-agent' : {
434
334
model : {
@@ -439,7 +339,6 @@ it('returns multiple agents config with interpolated instructions', async () =>
439
339
instructions : 'You are a writing assistant with academic style.' ,
440
340
tracker : expect . any ( Object ) ,
441
341
enabled : true ,
442
- toVercelAISDK : expect . any ( Function ) ,
443
342
} ,
444
343
} ) ;
445
344
0 commit comments