@ai-sdk/baseten 1.0.0-beta.6 → 1.0.0-beta.61

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.
package/CHANGELOG.md CHANGED
@@ -1,5 +1,494 @@
1
1
  # @ai-sdk/baseten
2
2
 
3
+ ## 1.0.0-beta.61
4
+
5
+ ### Patch Changes
6
+
7
+ - Updated dependencies [2625a04]
8
+ - @ai-sdk/openai-compatible@2.0.0-beta.59
9
+ - @ai-sdk/provider@3.0.0-beta.31
10
+ - @ai-sdk/provider-utils@4.0.0-beta.58
11
+
12
+ ## 1.0.0-beta.60
13
+
14
+ ### Patch Changes
15
+
16
+ - Updated dependencies [cbf52cd]
17
+ - @ai-sdk/openai-compatible@2.0.0-beta.58
18
+ - @ai-sdk/provider@3.0.0-beta.30
19
+ - @ai-sdk/provider-utils@4.0.0-beta.57
20
+
21
+ ## 1.0.0-beta.59
22
+
23
+ ### Patch Changes
24
+
25
+ - Updated dependencies [9549c9e]
26
+ - @ai-sdk/provider@3.0.0-beta.29
27
+ - @ai-sdk/openai-compatible@2.0.0-beta.57
28
+ - @ai-sdk/provider-utils@4.0.0-beta.56
29
+
30
+ ## 1.0.0-beta.58
31
+
32
+ ### Patch Changes
33
+
34
+ - Updated dependencies [50b70d6]
35
+ - @ai-sdk/provider-utils@4.0.0-beta.55
36
+ - @ai-sdk/openai-compatible@2.0.0-beta.56
37
+
38
+ ## 1.0.0-beta.57
39
+
40
+ ### Patch Changes
41
+
42
+ - Updated dependencies [9061dc0]
43
+ - @ai-sdk/openai-compatible@2.0.0-beta.55
44
+ - @ai-sdk/provider-utils@4.0.0-beta.54
45
+ - @ai-sdk/provider@3.0.0-beta.28
46
+
47
+ ## 1.0.0-beta.56
48
+
49
+ ### Patch Changes
50
+
51
+ - 366f50b: chore(provider): add deprecated textEmbeddingModel and textEmbedding aliases
52
+ - Updated dependencies [366f50b]
53
+ - @ai-sdk/openai-compatible@2.0.0-beta.54
54
+ - @ai-sdk/provider@3.0.0-beta.27
55
+ - @ai-sdk/provider-utils@4.0.0-beta.53
56
+
57
+ ## 1.0.0-beta.55
58
+
59
+ ### Patch Changes
60
+
61
+ - Updated dependencies [763d04a]
62
+ - @ai-sdk/provider-utils@4.0.0-beta.52
63
+ - @ai-sdk/openai-compatible@2.0.0-beta.53
64
+
65
+ ## 1.0.0-beta.54
66
+
67
+ ### Patch Changes
68
+
69
+ - Updated dependencies [c1efac4]
70
+ - @ai-sdk/provider-utils@4.0.0-beta.51
71
+ - @ai-sdk/openai-compatible@2.0.0-beta.52
72
+
73
+ ## 1.0.0-beta.53
74
+
75
+ ### Patch Changes
76
+
77
+ - Updated dependencies [32223c8]
78
+ - @ai-sdk/provider-utils@4.0.0-beta.50
79
+ - @ai-sdk/openai-compatible@2.0.0-beta.51
80
+
81
+ ## 1.0.0-beta.52
82
+
83
+ ### Patch Changes
84
+
85
+ - Updated dependencies [83e5744]
86
+ - @ai-sdk/provider-utils@4.0.0-beta.49
87
+ - @ai-sdk/openai-compatible@2.0.0-beta.50
88
+
89
+ ## 1.0.0-beta.51
90
+
91
+ ### Patch Changes
92
+
93
+ - Updated dependencies [960ec8f]
94
+ - @ai-sdk/provider-utils@4.0.0-beta.48
95
+ - @ai-sdk/openai-compatible@2.0.0-beta.49
96
+
97
+ ## 1.0.0-beta.50
98
+
99
+ ### Patch Changes
100
+
101
+ - Updated dependencies [e9e157f]
102
+ - @ai-sdk/provider-utils@4.0.0-beta.47
103
+ - @ai-sdk/openai-compatible@2.0.0-beta.48
104
+
105
+ ## 1.0.0-beta.49
106
+
107
+ ### Patch Changes
108
+
109
+ - Updated dependencies [81e29ab]
110
+ - @ai-sdk/provider-utils@4.0.0-beta.46
111
+ - @ai-sdk/openai-compatible@2.0.0-beta.47
112
+
113
+ ## 1.0.0-beta.48
114
+
115
+ ### Patch Changes
116
+
117
+ - Updated dependencies [3bd2689]
118
+ - @ai-sdk/openai-compatible@2.0.0-beta.46
119
+ - @ai-sdk/provider@3.0.0-beta.26
120
+ - @ai-sdk/provider-utils@4.0.0-beta.45
121
+
122
+ ## 1.0.0-beta.47
123
+
124
+ ### Patch Changes
125
+
126
+ - Updated dependencies [53f3368]
127
+ - @ai-sdk/provider@3.0.0-beta.25
128
+ - @ai-sdk/openai-compatible@2.0.0-beta.45
129
+ - @ai-sdk/provider-utils@4.0.0-beta.44
130
+
131
+ ## 1.0.0-beta.46
132
+
133
+ ### Patch Changes
134
+
135
+ - Updated dependencies [dce03c4]
136
+ - @ai-sdk/provider-utils@4.0.0-beta.43
137
+ - @ai-sdk/provider@3.0.0-beta.24
138
+ - @ai-sdk/openai-compatible@2.0.0-beta.44
139
+
140
+ ## 1.0.0-beta.45
141
+
142
+ ### Patch Changes
143
+
144
+ - Updated dependencies [3ed5519]
145
+ - @ai-sdk/provider-utils@4.0.0-beta.42
146
+ - @ai-sdk/openai-compatible@2.0.0-beta.43
147
+
148
+ ## 1.0.0-beta.44
149
+
150
+ ### Patch Changes
151
+
152
+ - Updated dependencies [1bd7d32]
153
+ - @ai-sdk/openai-compatible@2.0.0-beta.42
154
+ - @ai-sdk/provider-utils@4.0.0-beta.41
155
+ - @ai-sdk/provider@3.0.0-beta.23
156
+
157
+ ## 1.0.0-beta.43
158
+
159
+ ### Patch Changes
160
+
161
+ - Updated dependencies [544d4e8]
162
+ - @ai-sdk/openai-compatible@2.0.0-beta.41
163
+ - @ai-sdk/provider-utils@4.0.0-beta.40
164
+ - @ai-sdk/provider@3.0.0-beta.22
165
+
166
+ ## 1.0.0-beta.42
167
+
168
+ ### Patch Changes
169
+
170
+ - Updated dependencies [954c356]
171
+ - @ai-sdk/provider-utils@4.0.0-beta.39
172
+ - @ai-sdk/provider@3.0.0-beta.21
173
+ - @ai-sdk/openai-compatible@2.0.0-beta.40
174
+
175
+ ## 1.0.0-beta.41
176
+
177
+ ### Patch Changes
178
+
179
+ - Updated dependencies [03849b0]
180
+ - @ai-sdk/provider-utils@4.0.0-beta.38
181
+ - @ai-sdk/openai-compatible@2.0.0-beta.39
182
+
183
+ ## 1.0.0-beta.40
184
+
185
+ ### Patch Changes
186
+
187
+ - Updated dependencies [457318b]
188
+ - @ai-sdk/openai-compatible@2.0.0-beta.38
189
+ - @ai-sdk/provider@3.0.0-beta.20
190
+ - @ai-sdk/provider-utils@4.0.0-beta.37
191
+
192
+ ## 1.0.0-beta.39
193
+
194
+ ### Patch Changes
195
+
196
+ - 8d9e8ad: chore(provider): remove generics from EmbeddingModelV3
197
+
198
+ Before
199
+
200
+ ```ts
201
+ model.textEmbeddingModel('my-model-id');
202
+ ```
203
+
204
+ After
205
+
206
+ ```ts
207
+ model.embeddingModel('my-model-id');
208
+ ```
209
+
210
+ - Updated dependencies [8d9e8ad]
211
+ - @ai-sdk/openai-compatible@2.0.0-beta.37
212
+ - @ai-sdk/provider@3.0.0-beta.19
213
+ - @ai-sdk/provider-utils@4.0.0-beta.36
214
+
215
+ ## 1.0.0-beta.38
216
+
217
+ ### Patch Changes
218
+
219
+ - Updated dependencies [10d819b]
220
+ - @ai-sdk/provider@3.0.0-beta.18
221
+ - @ai-sdk/openai-compatible@2.0.0-beta.36
222
+ - @ai-sdk/provider-utils@4.0.0-beta.35
223
+
224
+ ## 1.0.0-beta.37
225
+
226
+ ### Patch Changes
227
+
228
+ - Updated dependencies [db913bd]
229
+ - @ai-sdk/provider@3.0.0-beta.17
230
+ - @ai-sdk/openai-compatible@2.0.0-beta.35
231
+ - @ai-sdk/provider-utils@4.0.0-beta.34
232
+
233
+ ## 1.0.0-beta.36
234
+
235
+ ### Patch Changes
236
+
237
+ - ab9af9c: add moonshotai/Kimi-K2-Thinking model ID for Baseten provider
238
+
239
+ ## 1.0.0-beta.35
240
+
241
+ ### Patch Changes
242
+
243
+ - Updated dependencies [b681d7d]
244
+ - @ai-sdk/provider@3.0.0-beta.16
245
+ - @ai-sdk/openai-compatible@2.0.0-beta.34
246
+ - @ai-sdk/provider-utils@4.0.0-beta.33
247
+
248
+ ## 1.0.0-beta.34
249
+
250
+ ### Patch Changes
251
+
252
+ - Updated dependencies [32d8dbb]
253
+ - @ai-sdk/provider-utils@4.0.0-beta.32
254
+ - @ai-sdk/openai-compatible@2.0.0-beta.33
255
+
256
+ ## 1.0.0-beta.33
257
+
258
+ ### Patch Changes
259
+
260
+ - Updated dependencies [bb36798]
261
+ - @ai-sdk/provider@3.0.0-beta.15
262
+ - @ai-sdk/openai-compatible@2.0.0-beta.32
263
+ - @ai-sdk/provider-utils@4.0.0-beta.31
264
+
265
+ ## 1.0.0-beta.32
266
+
267
+ ### Patch Changes
268
+
269
+ - Updated dependencies [4f16c37]
270
+ - @ai-sdk/provider-utils@4.0.0-beta.30
271
+ - @ai-sdk/openai-compatible@2.0.0-beta.31
272
+
273
+ ## 1.0.0-beta.31
274
+
275
+ ### Patch Changes
276
+
277
+ - Updated dependencies [af3780b]
278
+ - @ai-sdk/provider@3.0.0-beta.14
279
+ - @ai-sdk/openai-compatible@2.0.0-beta.30
280
+ - @ai-sdk/provider-utils@4.0.0-beta.29
281
+
282
+ ## 1.0.0-beta.30
283
+
284
+ ### Patch Changes
285
+
286
+ - Updated dependencies [016b111]
287
+ - @ai-sdk/provider-utils@4.0.0-beta.28
288
+ - @ai-sdk/openai-compatible@2.0.0-beta.29
289
+
290
+ ## 1.0.0-beta.29
291
+
292
+ ### Patch Changes
293
+
294
+ - Updated dependencies [37c58a0]
295
+ - @ai-sdk/provider@3.0.0-beta.13
296
+ - @ai-sdk/openai-compatible@2.0.0-beta.28
297
+ - @ai-sdk/provider-utils@4.0.0-beta.27
298
+
299
+ ## 1.0.0-beta.28
300
+
301
+ ### Patch Changes
302
+
303
+ - Updated dependencies [d1bdadb]
304
+ - @ai-sdk/provider@3.0.0-beta.12
305
+ - @ai-sdk/openai-compatible@2.0.0-beta.27
306
+ - @ai-sdk/provider-utils@4.0.0-beta.26
307
+
308
+ ## 1.0.0-beta.27
309
+
310
+ ### Patch Changes
311
+
312
+ - Updated dependencies [4c44a5b]
313
+ - @ai-sdk/provider@3.0.0-beta.11
314
+ - @ai-sdk/openai-compatible@2.0.0-beta.26
315
+ - @ai-sdk/provider-utils@4.0.0-beta.25
316
+
317
+ ## 1.0.0-beta.26
318
+
319
+ ### Patch Changes
320
+
321
+ - 0c3b58b: fix(provider): add specificationVersion to ProviderV3
322
+ - Updated dependencies [0c3b58b]
323
+ - @ai-sdk/openai-compatible@2.0.0-beta.25
324
+ - @ai-sdk/provider@3.0.0-beta.10
325
+ - @ai-sdk/provider-utils@4.0.0-beta.24
326
+
327
+ ## 1.0.0-beta.25
328
+
329
+ ### Patch Changes
330
+
331
+ - Updated dependencies [a755db5]
332
+ - @ai-sdk/provider@3.0.0-beta.9
333
+ - @ai-sdk/openai-compatible@2.0.0-beta.24
334
+ - @ai-sdk/provider-utils@4.0.0-beta.23
335
+
336
+ ## 1.0.0-beta.24
337
+
338
+ ### Patch Changes
339
+
340
+ - Updated dependencies [58920e0]
341
+ - @ai-sdk/provider-utils@4.0.0-beta.22
342
+ - @ai-sdk/openai-compatible@2.0.0-beta.23
343
+
344
+ ## 1.0.0-beta.23
345
+
346
+ ### Patch Changes
347
+
348
+ - Updated dependencies [293a6b7]
349
+ - @ai-sdk/provider-utils@4.0.0-beta.21
350
+ - @ai-sdk/openai-compatible@2.0.0-beta.22
351
+
352
+ ## 1.0.0-beta.22
353
+
354
+ ### Patch Changes
355
+
356
+ - Updated dependencies [fca786b]
357
+ - @ai-sdk/provider-utils@4.0.0-beta.20
358
+ - @ai-sdk/openai-compatible@2.0.0-beta.21
359
+
360
+ ## 1.0.0-beta.21
361
+
362
+ ### Patch Changes
363
+
364
+ - Updated dependencies [3794514]
365
+ - @ai-sdk/provider-utils@4.0.0-beta.19
366
+ - @ai-sdk/provider@3.0.0-beta.8
367
+ - @ai-sdk/openai-compatible@2.0.0-beta.20
368
+
369
+ ## 1.0.0-beta.20
370
+
371
+ ### Patch Changes
372
+
373
+ - Updated dependencies [81d4308]
374
+ - @ai-sdk/provider@3.0.0-beta.7
375
+ - @ai-sdk/openai-compatible@2.0.0-beta.19
376
+ - @ai-sdk/provider-utils@4.0.0-beta.18
377
+
378
+ ## 1.0.0-beta.19
379
+
380
+ ### Patch Changes
381
+
382
+ - Updated dependencies [703459a]
383
+ - @ai-sdk/provider-utils@4.0.0-beta.17
384
+ - @ai-sdk/openai-compatible@2.0.0-beta.18
385
+
386
+ ## 1.0.0-beta.18
387
+
388
+ ### Patch Changes
389
+
390
+ - Updated dependencies [b689220]
391
+ - @ai-sdk/openai-compatible@2.0.0-beta.17
392
+
393
+ ## 1.0.0-beta.17
394
+
395
+ ### Patch Changes
396
+
397
+ - Updated dependencies [6306603]
398
+ - @ai-sdk/provider-utils@4.0.0-beta.16
399
+ - @ai-sdk/openai-compatible@2.0.0-beta.16
400
+
401
+ ## 1.0.0-beta.16
402
+
403
+ ### Patch Changes
404
+
405
+ - Updated dependencies [f0b2157]
406
+ - @ai-sdk/provider-utils@4.0.0-beta.15
407
+ - @ai-sdk/openai-compatible@2.0.0-beta.15
408
+
409
+ ## 1.0.0-beta.15
410
+
411
+ ### Patch Changes
412
+
413
+ - Updated dependencies [3b1d015]
414
+ - @ai-sdk/provider-utils@4.0.0-beta.14
415
+ - @ai-sdk/openai-compatible@2.0.0-beta.14
416
+
417
+ ## 1.0.0-beta.14
418
+
419
+ ### Patch Changes
420
+
421
+ - Updated dependencies [d116b4b]
422
+ - @ai-sdk/provider-utils@4.0.0-beta.13
423
+ - @ai-sdk/openai-compatible@2.0.0-beta.13
424
+
425
+ ## 1.0.0-beta.13
426
+
427
+ ### Patch Changes
428
+
429
+ - Updated dependencies [7e32fea]
430
+ - @ai-sdk/provider-utils@4.0.0-beta.12
431
+ - @ai-sdk/openai-compatible@2.0.0-beta.12
432
+
433
+ ## 1.0.0-beta.12
434
+
435
+ ### Patch Changes
436
+
437
+ - 95f65c2: chore: use import \* from zod/v4
438
+ - Updated dependencies [95f65c2]
439
+ - Updated dependencies [95f65c2]
440
+ - @ai-sdk/openai-compatible@2.0.0-beta.11
441
+ - @ai-sdk/provider-utils@4.0.0-beta.11
442
+
443
+ ## 1.0.0-beta.11
444
+
445
+ ### Major Changes
446
+
447
+ - dee8b05: ai SDK 6 beta
448
+
449
+ ### Patch Changes
450
+
451
+ - Updated dependencies [dee8b05]
452
+ - @ai-sdk/openai-compatible@2.0.0-beta.10
453
+ - @ai-sdk/provider@3.0.0-beta.6
454
+ - @ai-sdk/provider-utils@4.0.0-beta.10
455
+
456
+ ## 1.0.0-beta.10
457
+
458
+ ### Patch Changes
459
+
460
+ - Updated dependencies [521c537]
461
+ - @ai-sdk/provider-utils@3.1.0-beta.9
462
+ - @ai-sdk/openai-compatible@1.1.0-beta.9
463
+
464
+ ## 1.0.0-beta.9
465
+
466
+ ### Patch Changes
467
+
468
+ - Updated dependencies [e06565c]
469
+ - @ai-sdk/provider-utils@3.1.0-beta.8
470
+ - @ai-sdk/openai-compatible@1.1.0-beta.8
471
+
472
+ ## 1.0.0-beta.8
473
+
474
+ ### Patch Changes
475
+
476
+ - Updated dependencies [046aa3b]
477
+ - Updated dependencies [e8109d3]
478
+ - @ai-sdk/provider@2.1.0-beta.5
479
+ - @ai-sdk/openai-compatible@1.1.0-beta.7
480
+ - @ai-sdk/provider-utils@3.1.0-beta.7
481
+
482
+ ## 1.0.0-beta.7
483
+
484
+ ### Patch Changes
485
+
486
+ - Updated dependencies [0adc679]
487
+ - Updated dependencies [2b0caef]
488
+ - @ai-sdk/openai-compatible@1.1.0-beta.6
489
+ - @ai-sdk/provider-utils@3.1.0-beta.6
490
+ - @ai-sdk/provider@2.1.0-beta.4
491
+
3
492
  ## 1.0.0-beta.6
4
493
 
5
494
  ### Patch Changes
package/dist/index.d.mts CHANGED
@@ -2,7 +2,7 @@ import { ProviderV3, LanguageModelV3, EmbeddingModelV3 } from '@ai-sdk/provider'
2
2
  import { FetchFunction } from '@ai-sdk/provider-utils';
3
3
  import { z } from 'zod/v4';
4
4
 
5
- type BasetenChatModelId = 'deepseek-ai/DeepSeek-R1-0528' | 'deepseek-ai/DeepSeek-V3-0324' | 'meta-llama/Llama-4-Maverick-17B-128E-Instruct' | 'meta-llama/Llama-4-Scout-17B-16E-Instruct' | 'moonshotai/Kimi-K2-Instruct' | 'Qwen/Qwen3-235B-A22B-Instruct-2507' | 'Qwen/Qwen3-Coder-480B-A35B-Instruct' | 'openai/gpt-oss-120b' | (string & {});
5
+ type BasetenChatModelId = 'deepseek-ai/DeepSeek-R1-0528' | 'deepseek-ai/DeepSeek-V3-0324' | 'deepseek-ai/DeepSeek-V3.1' | 'moonshotai/Kimi-K2-Instruct-0905' | 'moonshotai/Kimi-K2-Thinking' | 'Qwen/Qwen3-235B-A22B-Instruct-2507' | 'Qwen/Qwen3-Coder-480B-A35B-Instruct' | 'openai/gpt-oss-120b' | 'zai-org/GLM-4.6' | (string & {});
6
6
 
7
7
  type BasetenEmbeddingModelId = string & {};
8
8
 
@@ -49,9 +49,13 @@ interface BasetenProvider extends ProviderV3 {
49
49
  */
50
50
  languageModel(modelId?: BasetenChatModelId): LanguageModelV3;
51
51
  /**
52
- Creates a text embedding model for text generation.
52
+ Creates a embedding model for text generation.
53
53
  */
54
- textEmbeddingModel(modelId?: BasetenEmbeddingModelId): EmbeddingModelV3<string>;
54
+ embeddingModel(modelId?: BasetenEmbeddingModelId): EmbeddingModelV3;
55
+ /**
56
+ * @deprecated Use `embeddingModel` instead.
57
+ */
58
+ textEmbeddingModel(modelId?: BasetenEmbeddingModelId): EmbeddingModelV3;
55
59
  }
56
60
  declare function createBaseten(options?: BasetenProviderSettings): BasetenProvider;
57
61
  declare const baseten: BasetenProvider;
package/dist/index.d.ts CHANGED
@@ -2,7 +2,7 @@ import { ProviderV3, LanguageModelV3, EmbeddingModelV3 } from '@ai-sdk/provider'
2
2
  import { FetchFunction } from '@ai-sdk/provider-utils';
3
3
  import { z } from 'zod/v4';
4
4
 
5
- type BasetenChatModelId = 'deepseek-ai/DeepSeek-R1-0528' | 'deepseek-ai/DeepSeek-V3-0324' | 'meta-llama/Llama-4-Maverick-17B-128E-Instruct' | 'meta-llama/Llama-4-Scout-17B-16E-Instruct' | 'moonshotai/Kimi-K2-Instruct' | 'Qwen/Qwen3-235B-A22B-Instruct-2507' | 'Qwen/Qwen3-Coder-480B-A35B-Instruct' | 'openai/gpt-oss-120b' | (string & {});
5
+ type BasetenChatModelId = 'deepseek-ai/DeepSeek-R1-0528' | 'deepseek-ai/DeepSeek-V3-0324' | 'deepseek-ai/DeepSeek-V3.1' | 'moonshotai/Kimi-K2-Instruct-0905' | 'moonshotai/Kimi-K2-Thinking' | 'Qwen/Qwen3-235B-A22B-Instruct-2507' | 'Qwen/Qwen3-Coder-480B-A35B-Instruct' | 'openai/gpt-oss-120b' | 'zai-org/GLM-4.6' | (string & {});
6
6
 
7
7
  type BasetenEmbeddingModelId = string & {};
8
8
 
@@ -49,9 +49,13 @@ interface BasetenProvider extends ProviderV3 {
49
49
  */
50
50
  languageModel(modelId?: BasetenChatModelId): LanguageModelV3;
51
51
  /**
52
- Creates a text embedding model for text generation.
52
+ Creates a embedding model for text generation.
53
53
  */
54
- textEmbeddingModel(modelId?: BasetenEmbeddingModelId): EmbeddingModelV3<string>;
54
+ embeddingModel(modelId?: BasetenEmbeddingModelId): EmbeddingModelV3;
55
+ /**
56
+ * @deprecated Use `embeddingModel` instead.
57
+ */
58
+ textEmbeddingModel(modelId?: BasetenEmbeddingModelId): EmbeddingModelV3;
55
59
  }
56
60
  declare function createBaseten(options?: BasetenProviderSettings): BasetenProvider;
57
61
  declare const baseten: BasetenProvider;
package/dist/index.js CHANGED
@@ -34,7 +34,7 @@ var import_v4 = require("zod/v4");
34
34
  var import_performance_client = require("@basetenlabs/performance-client");
35
35
 
36
36
  // src/version.ts
37
- var VERSION = true ? "1.0.0-beta.6" : "0.0.0-test";
37
+ var VERSION = true ? "1.0.0-beta.61" : "0.0.0-test";
38
38
 
39
39
  // src/baseten-provider.ts
40
40
  var basetenErrorSchema = import_v4.z.object({
@@ -90,7 +90,7 @@ function createBaseten(options = {}) {
90
90
  errorStructure: basetenErrorStructure
91
91
  });
92
92
  };
93
- const createTextEmbeddingModel = (modelId) => {
93
+ const createEmbeddingModel = (modelId) => {
94
94
  const customURL = options.modelURL;
95
95
  if (!customURL) {
96
96
  throw new Error(
@@ -128,7 +128,8 @@ function createBaseten(options = {}) {
128
128
  return {
129
129
  embeddings,
130
130
  usage: response.usage ? { tokens: response.usage.total_tokens } : void 0,
131
- response: { headers: {}, body: response }
131
+ response: { headers: {}, body: response },
132
+ warnings: []
132
133
  };
133
134
  };
134
135
  return model;
@@ -139,12 +140,14 @@ function createBaseten(options = {}) {
139
140
  }
140
141
  };
141
142
  const provider = (modelId) => createChatModel(modelId);
143
+ provider.specificationVersion = "v3";
142
144
  provider.chatModel = createChatModel;
143
145
  provider.languageModel = createChatModel;
144
146
  provider.imageModel = (modelId) => {
145
147
  throw new import_provider.NoSuchModelError({ modelId, modelType: "imageModel" });
146
148
  };
147
- provider.textEmbeddingModel = createTextEmbeddingModel;
149
+ provider.embeddingModel = createEmbeddingModel;
150
+ provider.textEmbeddingModel = createEmbeddingModel;
148
151
  return provider;
149
152
  }
150
153
  var baseten = createBaseten();
package/dist/index.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"sources":["../src/index.ts","../src/baseten-provider.ts","../src/version.ts"],"sourcesContent":["export type { BasetenChatModelId } from './baseten-chat-options';\nexport { baseten, createBaseten } from './baseten-provider';\nexport type {\n BasetenProvider,\n BasetenProviderSettings,\n BasetenErrorData,\n} from './baseten-provider';\nexport { VERSION } from './version';\n","import {\n OpenAICompatibleChatLanguageModel,\n OpenAICompatibleEmbeddingModel,\n ProviderErrorStructure,\n} from '@ai-sdk/openai-compatible';\nimport {\n EmbeddingModelV3,\n LanguageModelV3,\n NoSuchModelError,\n ProviderV3,\n} from '@ai-sdk/provider';\nimport {\n FetchFunction,\n loadApiKey,\n withoutTrailingSlash,\n withUserAgentSuffix,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport { BasetenChatModelId } from './baseten-chat-options';\nimport { BasetenEmbeddingModelId } from './baseten-embedding-options';\nimport { PerformanceClient } from '@basetenlabs/performance-client';\nimport { VERSION } from './version';\n\nexport type BasetenErrorData = z.infer<typeof basetenErrorSchema>;\n\nconst basetenErrorSchema = z.object({\n error: z.string(),\n});\n\nconst basetenErrorStructure: ProviderErrorStructure<BasetenErrorData> = {\n errorSchema: basetenErrorSchema,\n errorToMessage: data => data.error,\n};\n\nexport interface BasetenProviderSettings {\n /**\n * Baseten API key. Default value is taken from the `BASETEN_API_KEY`\n * environment variable.\n */\n apiKey?: string;\n\n /**\n * Base URL for the Model APIs. Default: 'https://inference.baseten.co/v1'\n */\n baseURL?: string;\n\n /**\n * Model URL for custom models (chat or embeddings).\n * If not supplied, the default Model APIs will be used.\n */\n modelURL?: string;\n /**\n * Custom headers to include in the requests.\n */\n headers?: Record<string, string>;\n\n /**\n * Custom fetch implementation. You can use it as a middleware to intercept requests,\n * or to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n}\n\nexport interface BasetenProvider extends ProviderV3 {\n /**\nCreates a chat model for text generation. \n*/\n (modelId?: BasetenChatModelId): LanguageModelV3;\n\n /**\nCreates a chat model for text generation. \n*/\n chatModel(modelId?: BasetenChatModelId): LanguageModelV3;\n\n /**\nCreates a language model for text generation. Alias for chatModel.\n*/\n languageModel(modelId?: BasetenChatModelId): LanguageModelV3;\n\n /**\nCreates a text embedding model for text generation.\n*/\n textEmbeddingModel(\n modelId?: BasetenEmbeddingModelId,\n ): EmbeddingModelV3<string>;\n}\n\n// by default, we use the Model APIs\nconst defaultBaseURL = 'https://inference.baseten.co/v1';\n\nexport function createBaseten(\n options: BasetenProviderSettings = {},\n): BasetenProvider {\n const baseURL = withoutTrailingSlash(options.baseURL ?? defaultBaseURL);\n const getHeaders = () =>\n withUserAgentSuffix(\n {\n Authorization: `Bearer ${loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'BASETEN_API_KEY',\n description: 'Baseten API key',\n })}`,\n ...options.headers,\n },\n `ai-sdk/baseten/${VERSION}`,\n );\n\n interface CommonModelConfig {\n provider: string;\n url: ({ path }: { path: string }) => string;\n headers: () => Record<string, string>;\n fetch?: FetchFunction;\n }\n\n const getCommonModelConfig = (\n modelType: string,\n customURL?: string,\n ): CommonModelConfig => ({\n provider: `baseten.${modelType}`,\n url: ({ path }) => {\n // For embeddings with /sync URLs (but not /sync/v1), we need to add /v1\n if (\n modelType === 'embedding' &&\n customURL?.includes('/sync') &&\n !customURL?.includes('/sync/v1')\n ) {\n return `${customURL}/v1${path}`;\n }\n return `${customURL || baseURL}${path}`;\n },\n headers: getHeaders,\n fetch: options.fetch,\n });\n\n const createChatModel = (modelId?: BasetenChatModelId) => {\n // Use modelURL if provided, otherwise use default Model APIs\n const customURL = options.modelURL;\n\n if (customURL) {\n // Check if this is a /sync/v1 endpoint (OpenAI-compatible) or /predict endpoint (custom)\n const isOpenAICompatible = customURL.includes('/sync/v1');\n\n if (isOpenAICompatible) {\n // For /sync/v1 endpoints, use standard OpenAI-compatible format\n return new OpenAICompatibleChatLanguageModel(modelId ?? 'placeholder', {\n ...getCommonModelConfig('chat', customURL),\n errorStructure: basetenErrorStructure,\n });\n } else if (customURL.includes('/predict')) {\n throw new Error(\n 'Not supported. You must use a /sync/v1 endpoint for chat models.',\n );\n }\n }\n\n // Use default OpenAI-compatible format for Model APIs\n return new OpenAICompatibleChatLanguageModel(modelId ?? 'chat', {\n ...getCommonModelConfig('chat'),\n errorStructure: basetenErrorStructure,\n });\n };\n\n const createTextEmbeddingModel = (modelId?: BasetenEmbeddingModelId) => {\n // Use modelURL if provided\n const customURL = options.modelURL;\n if (!customURL) {\n throw new Error(\n 'No model URL provided for embeddings. Please set modelURL option for embeddings.',\n );\n }\n\n // Check if this is a /sync or /sync/v1 endpoint (OpenAI-compatible)\n // We support both /sync and /sync/v1, stripping /v1 before passing to Performance Client, as Performance Client adds /v1 itself\n const isOpenAICompatible = customURL.includes('/sync');\n\n if (isOpenAICompatible) {\n // Create the model using OpenAICompatibleEmbeddingModel and override doEmbed\n const model = new OpenAICompatibleEmbeddingModel(\n modelId ?? 'embeddings',\n {\n ...getCommonModelConfig('embedding', customURL),\n errorStructure: basetenErrorStructure,\n },\n );\n\n // Strip /v1 from URL if present before passing to Performance Client to avoid double /v1\n const performanceClientURL = customURL.replace('/sync/v1', '/sync');\n\n // Initialize the B10 Performance Client once for reuse\n const performanceClient = new PerformanceClient(\n performanceClientURL,\n loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'BASETEN_API_KEY',\n description: 'Baseten API key',\n }),\n );\n\n // Override the doEmbed method to use the pre-created Performance Client\n model.doEmbed = async params => {\n if (!params.values || !Array.isArray(params.values)) {\n throw new Error('params.values must be an array of strings');\n }\n\n // Performance Client handles batching internally, so we don't need to limit in 128 here\n const response = await performanceClient.embed(\n params.values,\n modelId ?? 'embeddings', // model_id is for Model APIs, we don't use it here for dedicated\n );\n // Transform the response to match the expected format\n const embeddings = response.data.map((item: any) => item.embedding);\n\n return {\n embeddings: embeddings,\n usage: response.usage\n ? { tokens: response.usage.total_tokens }\n : undefined,\n response: { headers: {}, body: response },\n };\n };\n\n return model;\n } else {\n throw new Error(\n 'Not supported. You must use a /sync or /sync/v1 endpoint for embeddings.',\n );\n }\n };\n\n const provider = (modelId?: BasetenChatModelId) => createChatModel(modelId);\n provider.chatModel = createChatModel;\n provider.languageModel = createChatModel;\n provider.imageModel = (modelId: string) => {\n throw new NoSuchModelError({ modelId, modelType: 'imageModel' });\n };\n provider.textEmbeddingModel = createTextEmbeddingModel;\n return provider;\n}\n\nexport const baseten = createBaseten();\n","// Version string of this package injected at build time.\ndeclare const __PACKAGE_VERSION__: string | undefined;\nexport const VERSION: string =\n typeof __PACKAGE_VERSION__ !== 'undefined'\n ? __PACKAGE_VERSION__\n : '0.0.0-test';\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,+BAIO;AACP,sBAKO;AACP,4BAKO;AACP,gBAAkB;AAGlB,gCAAkC;;;AClB3B,IAAM,UACX,OACI,iBACA;;;ADoBN,IAAM,qBAAqB,YAAE,OAAO;AAAA,EAClC,OAAO,YAAE,OAAO;AAClB,CAAC;AAED,IAAM,wBAAkE;AAAA,EACtE,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AAwDA,IAAM,iBAAiB;AAEhB,SAAS,cACd,UAAmC,CAAC,GACnB;AA5FnB;AA6FE,QAAM,cAAU,6CAAqB,aAAQ,YAAR,YAAmB,cAAc;AACtE,QAAM,aAAa,UACjB;AAAA,IACE;AAAA,MACE,eAAe,cAAU,kCAAW;AAAA,QAClC,QAAQ,QAAQ;AAAA,QAChB,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC,CAAC;AAAA,MACF,GAAG,QAAQ;AAAA,IACb;AAAA,IACA,kBAAkB,OAAO;AAAA,EAC3B;AASF,QAAM,uBAAuB,CAC3B,WACA,eACuB;AAAA,IACvB,UAAU,WAAW,SAAS;AAAA,IAC9B,KAAK,CAAC,EAAE,KAAK,MAAM;AAEjB,UACE,cAAc,gBACd,uCAAW,SAAS,aACpB,EAAC,uCAAW,SAAS,cACrB;AACA,eAAO,GAAG,SAAS,MAAM,IAAI;AAAA,MAC/B;AACA,aAAO,GAAG,aAAa,OAAO,GAAG,IAAI;AAAA,IACvC;AAAA,IACA,SAAS;AAAA,IACT,OAAO,QAAQ;AAAA,EACjB;AAEA,QAAM,kBAAkB,CAAC,YAAiC;AAExD,UAAM,YAAY,QAAQ;AAE1B,QAAI,WAAW;AAEb,YAAM,qBAAqB,UAAU,SAAS,UAAU;AAExD,UAAI,oBAAoB;AAEtB,eAAO,IAAI,2DAAkC,4BAAW,eAAe;AAAA,UACrE,GAAG,qBAAqB,QAAQ,SAAS;AAAA,UACzC,gBAAgB;AAAA,QAClB,CAAC;AAAA,MACH,WAAW,UAAU,SAAS,UAAU,GAAG;AACzC,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAGA,WAAO,IAAI,2DAAkC,4BAAW,QAAQ;AAAA,MAC9D,GAAG,qBAAqB,MAAM;AAAA,MAC9B,gBAAgB;AAAA,IAClB,CAAC;AAAA,EACH;AAEA,QAAM,2BAA2B,CAAC,YAAsC;AAEtE,UAAM,YAAY,QAAQ;AAC1B,QAAI,CAAC,WAAW;AACd,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAIA,UAAM,qBAAqB,UAAU,SAAS,OAAO;AAErD,QAAI,oBAAoB;AAEtB,YAAM,QAAQ,IAAI;AAAA,QAChB,4BAAW;AAAA,QACX;AAAA,UACE,GAAG,qBAAqB,aAAa,SAAS;AAAA,UAC9C,gBAAgB;AAAA,QAClB;AAAA,MACF;AAGA,YAAM,uBAAuB,UAAU,QAAQ,YAAY,OAAO;AAGlE,YAAM,oBAAoB,IAAI;AAAA,QAC5B;AAAA,YACA,kCAAW;AAAA,UACT,QAAQ,QAAQ;AAAA,UAChB,yBAAyB;AAAA,UACzB,aAAa;AAAA,QACf,CAAC;AAAA,MACH;AAGA,YAAM,UAAU,OAAM,WAAU;AAC9B,YAAI,CAAC,OAAO,UAAU,CAAC,MAAM,QAAQ,OAAO,MAAM,GAAG;AACnD,gBAAM,IAAI,MAAM,2CAA2C;AAAA,QAC7D;AAGA,cAAM,WAAW,MAAM,kBAAkB;AAAA,UACvC,OAAO;AAAA,UACP,4BAAW;AAAA;AAAA,QACb;AAEA,cAAM,aAAa,SAAS,KAAK,IAAI,CAAC,SAAc,KAAK,SAAS;AAElE,eAAO;AAAA,UACL;AAAA,UACA,OAAO,SAAS,QACZ,EAAE,QAAQ,SAAS,MAAM,aAAa,IACtC;AAAA,UACJ,UAAU,EAAE,SAAS,CAAC,GAAG,MAAM,SAAS;AAAA,QAC1C;AAAA,MACF;AAEA,aAAO;AAAA,IACT,OAAO;AACL,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,QAAM,WAAW,CAAC,YAAiC,gBAAgB,OAAO;AAC1E,WAAS,YAAY;AACrB,WAAS,gBAAgB;AACzB,WAAS,aAAa,CAAC,YAAoB;AACzC,UAAM,IAAI,iCAAiB,EAAE,SAAS,WAAW,aAAa,CAAC;AAAA,EACjE;AACA,WAAS,qBAAqB;AAC9B,SAAO;AACT;AAEO,IAAM,UAAU,cAAc;","names":[]}
1
+ {"version":3,"sources":["../src/index.ts","../src/baseten-provider.ts","../src/version.ts"],"sourcesContent":["export type { BasetenChatModelId } from './baseten-chat-options';\nexport { baseten, createBaseten } from './baseten-provider';\nexport type {\n BasetenProvider,\n BasetenProviderSettings,\n BasetenErrorData,\n} from './baseten-provider';\nexport { VERSION } from './version';\n","import {\n OpenAICompatibleChatLanguageModel,\n OpenAICompatibleEmbeddingModel,\n ProviderErrorStructure,\n} from '@ai-sdk/openai-compatible';\nimport {\n EmbeddingModelV3,\n LanguageModelV3,\n NoSuchModelError,\n ProviderV3,\n} from '@ai-sdk/provider';\nimport {\n FetchFunction,\n loadApiKey,\n withoutTrailingSlash,\n withUserAgentSuffix,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport { BasetenChatModelId } from './baseten-chat-options';\nimport { BasetenEmbeddingModelId } from './baseten-embedding-options';\nimport { PerformanceClient } from '@basetenlabs/performance-client';\nimport { VERSION } from './version';\n\nexport type BasetenErrorData = z.infer<typeof basetenErrorSchema>;\n\nconst basetenErrorSchema = z.object({\n error: z.string(),\n});\n\nconst basetenErrorStructure: ProviderErrorStructure<BasetenErrorData> = {\n errorSchema: basetenErrorSchema,\n errorToMessage: data => data.error,\n};\n\nexport interface BasetenProviderSettings {\n /**\n * Baseten API key. Default value is taken from the `BASETEN_API_KEY`\n * environment variable.\n */\n apiKey?: string;\n\n /**\n * Base URL for the Model APIs. Default: 'https://inference.baseten.co/v1'\n */\n baseURL?: string;\n\n /**\n * Model URL for custom models (chat or embeddings).\n * If not supplied, the default Model APIs will be used.\n */\n modelURL?: string;\n /**\n * Custom headers to include in the requests.\n */\n headers?: Record<string, string>;\n\n /**\n * Custom fetch implementation. You can use it as a middleware to intercept requests,\n * or to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n}\n\nexport interface BasetenProvider extends ProviderV3 {\n /**\nCreates a chat model for text generation.\n*/\n (modelId?: BasetenChatModelId): LanguageModelV3;\n\n /**\nCreates a chat model for text generation.\n*/\n chatModel(modelId?: BasetenChatModelId): LanguageModelV3;\n\n /**\nCreates a language model for text generation. Alias for chatModel.\n*/\n languageModel(modelId?: BasetenChatModelId): LanguageModelV3;\n\n /**\nCreates a embedding model for text generation.\n*/\n embeddingModel(modelId?: BasetenEmbeddingModelId): EmbeddingModelV3;\n\n /**\n * @deprecated Use `embeddingModel` instead.\n */\n textEmbeddingModel(modelId?: BasetenEmbeddingModelId): EmbeddingModelV3;\n}\n\n// by default, we use the Model APIs\nconst defaultBaseURL = 'https://inference.baseten.co/v1';\n\nexport function createBaseten(\n options: BasetenProviderSettings = {},\n): BasetenProvider {\n const baseURL = withoutTrailingSlash(options.baseURL ?? defaultBaseURL);\n const getHeaders = () =>\n withUserAgentSuffix(\n {\n Authorization: `Bearer ${loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'BASETEN_API_KEY',\n description: 'Baseten API key',\n })}`,\n ...options.headers,\n },\n `ai-sdk/baseten/${VERSION}`,\n );\n\n interface CommonModelConfig {\n provider: string;\n url: ({ path }: { path: string }) => string;\n headers: () => Record<string, string>;\n fetch?: FetchFunction;\n }\n\n const getCommonModelConfig = (\n modelType: string,\n customURL?: string,\n ): CommonModelConfig => ({\n provider: `baseten.${modelType}`,\n url: ({ path }) => {\n // For embeddings with /sync URLs (but not /sync/v1), we need to add /v1\n if (\n modelType === 'embedding' &&\n customURL?.includes('/sync') &&\n !customURL?.includes('/sync/v1')\n ) {\n return `${customURL}/v1${path}`;\n }\n return `${customURL || baseURL}${path}`;\n },\n headers: getHeaders,\n fetch: options.fetch,\n });\n\n const createChatModel = (modelId?: BasetenChatModelId) => {\n // Use modelURL if provided, otherwise use default Model APIs\n const customURL = options.modelURL;\n\n if (customURL) {\n // Check if this is a /sync/v1 endpoint (OpenAI-compatible) or /predict endpoint (custom)\n const isOpenAICompatible = customURL.includes('/sync/v1');\n\n if (isOpenAICompatible) {\n // For /sync/v1 endpoints, use standard OpenAI-compatible format\n return new OpenAICompatibleChatLanguageModel(modelId ?? 'placeholder', {\n ...getCommonModelConfig('chat', customURL),\n errorStructure: basetenErrorStructure,\n });\n } else if (customURL.includes('/predict')) {\n throw new Error(\n 'Not supported. You must use a /sync/v1 endpoint for chat models.',\n );\n }\n }\n\n // Use default OpenAI-compatible format for Model APIs\n return new OpenAICompatibleChatLanguageModel(modelId ?? 'chat', {\n ...getCommonModelConfig('chat'),\n errorStructure: basetenErrorStructure,\n });\n };\n\n const createEmbeddingModel = (modelId?: BasetenEmbeddingModelId) => {\n // Use modelURL if provided\n const customURL = options.modelURL;\n if (!customURL) {\n throw new Error(\n 'No model URL provided for embeddings. Please set modelURL option for embeddings.',\n );\n }\n\n // Check if this is a /sync or /sync/v1 endpoint (OpenAI-compatible)\n // We support both /sync and /sync/v1, stripping /v1 before passing to Performance Client, as Performance Client adds /v1 itself\n const isOpenAICompatible = customURL.includes('/sync');\n\n if (isOpenAICompatible) {\n // Create the model using OpenAICompatibleEmbeddingModel and override doEmbed\n const model = new OpenAICompatibleEmbeddingModel(\n modelId ?? 'embeddings',\n {\n ...getCommonModelConfig('embedding', customURL),\n errorStructure: basetenErrorStructure,\n },\n );\n\n // Strip /v1 from URL if present before passing to Performance Client to avoid double /v1\n const performanceClientURL = customURL.replace('/sync/v1', '/sync');\n\n // Initialize the B10 Performance Client once for reuse\n const performanceClient = new PerformanceClient(\n performanceClientURL,\n loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'BASETEN_API_KEY',\n description: 'Baseten API key',\n }),\n );\n\n // Override the doEmbed method to use the pre-created Performance Client\n model.doEmbed = async params => {\n if (!params.values || !Array.isArray(params.values)) {\n throw new Error('params.values must be an array of strings');\n }\n\n // Performance Client handles batching internally, so we don't need to limit in 128 here\n const response = await performanceClient.embed(\n params.values,\n modelId ?? 'embeddings', // model_id is for Model APIs, we don't use it here for dedicated\n );\n // Transform the response to match the expected format\n const embeddings = response.data.map((item: any) => item.embedding);\n\n return {\n embeddings,\n usage: response.usage\n ? { tokens: response.usage.total_tokens }\n : undefined,\n response: { headers: {}, body: response },\n warnings: [],\n };\n };\n\n return model;\n } else {\n throw new Error(\n 'Not supported. You must use a /sync or /sync/v1 endpoint for embeddings.',\n );\n }\n };\n\n const provider = (modelId?: BasetenChatModelId) => createChatModel(modelId);\n\n provider.specificationVersion = 'v3' as const;\n provider.chatModel = createChatModel;\n provider.languageModel = createChatModel;\n provider.imageModel = (modelId: string) => {\n throw new NoSuchModelError({ modelId, modelType: 'imageModel' });\n };\n provider.embeddingModel = createEmbeddingModel;\n provider.textEmbeddingModel = createEmbeddingModel;\n return provider;\n}\n\nexport const baseten = createBaseten();\n","// Version string of this package injected at build time.\ndeclare const __PACKAGE_VERSION__: string | undefined;\nexport const VERSION: string =\n typeof __PACKAGE_VERSION__ !== 'undefined'\n ? __PACKAGE_VERSION__\n : '0.0.0-test';\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,+BAIO;AACP,sBAKO;AACP,4BAKO;AACP,gBAAkB;AAGlB,gCAAkC;;;AClB3B,IAAM,UACX,OACI,kBACA;;;ADoBN,IAAM,qBAAqB,YAAE,OAAO;AAAA,EAClC,OAAO,YAAE,OAAO;AAClB,CAAC;AAED,IAAM,wBAAkE;AAAA,EACtE,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AA2DA,IAAM,iBAAiB;AAEhB,SAAS,cACd,UAAmC,CAAC,GACnB;AA/FnB;AAgGE,QAAM,cAAU,6CAAqB,aAAQ,YAAR,YAAmB,cAAc;AACtE,QAAM,aAAa,UACjB;AAAA,IACE;AAAA,MACE,eAAe,cAAU,kCAAW;AAAA,QAClC,QAAQ,QAAQ;AAAA,QAChB,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC,CAAC;AAAA,MACF,GAAG,QAAQ;AAAA,IACb;AAAA,IACA,kBAAkB,OAAO;AAAA,EAC3B;AASF,QAAM,uBAAuB,CAC3B,WACA,eACuB;AAAA,IACvB,UAAU,WAAW,SAAS;AAAA,IAC9B,KAAK,CAAC,EAAE,KAAK,MAAM;AAEjB,UACE,cAAc,gBACd,uCAAW,SAAS,aACpB,EAAC,uCAAW,SAAS,cACrB;AACA,eAAO,GAAG,SAAS,MAAM,IAAI;AAAA,MAC/B;AACA,aAAO,GAAG,aAAa,OAAO,GAAG,IAAI;AAAA,IACvC;AAAA,IACA,SAAS;AAAA,IACT,OAAO,QAAQ;AAAA,EACjB;AAEA,QAAM,kBAAkB,CAAC,YAAiC;AAExD,UAAM,YAAY,QAAQ;AAE1B,QAAI,WAAW;AAEb,YAAM,qBAAqB,UAAU,SAAS,UAAU;AAExD,UAAI,oBAAoB;AAEtB,eAAO,IAAI,2DAAkC,4BAAW,eAAe;AAAA,UACrE,GAAG,qBAAqB,QAAQ,SAAS;AAAA,UACzC,gBAAgB;AAAA,QAClB,CAAC;AAAA,MACH,WAAW,UAAU,SAAS,UAAU,GAAG;AACzC,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAGA,WAAO,IAAI,2DAAkC,4BAAW,QAAQ;AAAA,MAC9D,GAAG,qBAAqB,MAAM;AAAA,MAC9B,gBAAgB;AAAA,IAClB,CAAC;AAAA,EACH;AAEA,QAAM,uBAAuB,CAAC,YAAsC;AAElE,UAAM,YAAY,QAAQ;AAC1B,QAAI,CAAC,WAAW;AACd,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAIA,UAAM,qBAAqB,UAAU,SAAS,OAAO;AAErD,QAAI,oBAAoB;AAEtB,YAAM,QAAQ,IAAI;AAAA,QAChB,4BAAW;AAAA,QACX;AAAA,UACE,GAAG,qBAAqB,aAAa,SAAS;AAAA,UAC9C,gBAAgB;AAAA,QAClB;AAAA,MACF;AAGA,YAAM,uBAAuB,UAAU,QAAQ,YAAY,OAAO;AAGlE,YAAM,oBAAoB,IAAI;AAAA,QAC5B;AAAA,YACA,kCAAW;AAAA,UACT,QAAQ,QAAQ;AAAA,UAChB,yBAAyB;AAAA,UACzB,aAAa;AAAA,QACf,CAAC;AAAA,MACH;AAGA,YAAM,UAAU,OAAM,WAAU;AAC9B,YAAI,CAAC,OAAO,UAAU,CAAC,MAAM,QAAQ,OAAO,MAAM,GAAG;AACnD,gBAAM,IAAI,MAAM,2CAA2C;AAAA,QAC7D;AAGA,cAAM,WAAW,MAAM,kBAAkB;AAAA,UACvC,OAAO;AAAA,UACP,4BAAW;AAAA;AAAA,QACb;AAEA,cAAM,aAAa,SAAS,KAAK,IAAI,CAAC,SAAc,KAAK,SAAS;AAElE,eAAO;AAAA,UACL;AAAA,UACA,OAAO,SAAS,QACZ,EAAE,QAAQ,SAAS,MAAM,aAAa,IACtC;AAAA,UACJ,UAAU,EAAE,SAAS,CAAC,GAAG,MAAM,SAAS;AAAA,UACxC,UAAU,CAAC;AAAA,QACb;AAAA,MACF;AAEA,aAAO;AAAA,IACT,OAAO;AACL,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,QAAM,WAAW,CAAC,YAAiC,gBAAgB,OAAO;AAE1E,WAAS,uBAAuB;AAChC,WAAS,YAAY;AACrB,WAAS,gBAAgB;AACzB,WAAS,aAAa,CAAC,YAAoB;AACzC,UAAM,IAAI,iCAAiB,EAAE,SAAS,WAAW,aAAa,CAAC;AAAA,EACjE;AACA,WAAS,iBAAiB;AAC1B,WAAS,qBAAqB;AAC9B,SAAO;AACT;AAEO,IAAM,UAAU,cAAc;","names":[]}
package/dist/index.mjs CHANGED
@@ -15,7 +15,7 @@ import { z } from "zod/v4";
15
15
  import { PerformanceClient } from "@basetenlabs/performance-client";
16
16
 
17
17
  // src/version.ts
18
- var VERSION = true ? "1.0.0-beta.6" : "0.0.0-test";
18
+ var VERSION = true ? "1.0.0-beta.61" : "0.0.0-test";
19
19
 
20
20
  // src/baseten-provider.ts
21
21
  var basetenErrorSchema = z.object({
@@ -71,7 +71,7 @@ function createBaseten(options = {}) {
71
71
  errorStructure: basetenErrorStructure
72
72
  });
73
73
  };
74
- const createTextEmbeddingModel = (modelId) => {
74
+ const createEmbeddingModel = (modelId) => {
75
75
  const customURL = options.modelURL;
76
76
  if (!customURL) {
77
77
  throw new Error(
@@ -109,7 +109,8 @@ function createBaseten(options = {}) {
109
109
  return {
110
110
  embeddings,
111
111
  usage: response.usage ? { tokens: response.usage.total_tokens } : void 0,
112
- response: { headers: {}, body: response }
112
+ response: { headers: {}, body: response },
113
+ warnings: []
113
114
  };
114
115
  };
115
116
  return model;
@@ -120,12 +121,14 @@ function createBaseten(options = {}) {
120
121
  }
121
122
  };
122
123
  const provider = (modelId) => createChatModel(modelId);
124
+ provider.specificationVersion = "v3";
123
125
  provider.chatModel = createChatModel;
124
126
  provider.languageModel = createChatModel;
125
127
  provider.imageModel = (modelId) => {
126
128
  throw new NoSuchModelError({ modelId, modelType: "imageModel" });
127
129
  };
128
- provider.textEmbeddingModel = createTextEmbeddingModel;
130
+ provider.embeddingModel = createEmbeddingModel;
131
+ provider.textEmbeddingModel = createEmbeddingModel;
129
132
  return provider;
130
133
  }
131
134
  var baseten = createBaseten();
@@ -1 +1 @@
1
- {"version":3,"sources":["../src/baseten-provider.ts","../src/version.ts"],"sourcesContent":["import {\n OpenAICompatibleChatLanguageModel,\n OpenAICompatibleEmbeddingModel,\n ProviderErrorStructure,\n} from '@ai-sdk/openai-compatible';\nimport {\n EmbeddingModelV3,\n LanguageModelV3,\n NoSuchModelError,\n ProviderV3,\n} from '@ai-sdk/provider';\nimport {\n FetchFunction,\n loadApiKey,\n withoutTrailingSlash,\n withUserAgentSuffix,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport { BasetenChatModelId } from './baseten-chat-options';\nimport { BasetenEmbeddingModelId } from './baseten-embedding-options';\nimport { PerformanceClient } from '@basetenlabs/performance-client';\nimport { VERSION } from './version';\n\nexport type BasetenErrorData = z.infer<typeof basetenErrorSchema>;\n\nconst basetenErrorSchema = z.object({\n error: z.string(),\n});\n\nconst basetenErrorStructure: ProviderErrorStructure<BasetenErrorData> = {\n errorSchema: basetenErrorSchema,\n errorToMessage: data => data.error,\n};\n\nexport interface BasetenProviderSettings {\n /**\n * Baseten API key. Default value is taken from the `BASETEN_API_KEY`\n * environment variable.\n */\n apiKey?: string;\n\n /**\n * Base URL for the Model APIs. Default: 'https://inference.baseten.co/v1'\n */\n baseURL?: string;\n\n /**\n * Model URL for custom models (chat or embeddings).\n * If not supplied, the default Model APIs will be used.\n */\n modelURL?: string;\n /**\n * Custom headers to include in the requests.\n */\n headers?: Record<string, string>;\n\n /**\n * Custom fetch implementation. You can use it as a middleware to intercept requests,\n * or to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n}\n\nexport interface BasetenProvider extends ProviderV3 {\n /**\nCreates a chat model for text generation. \n*/\n (modelId?: BasetenChatModelId): LanguageModelV3;\n\n /**\nCreates a chat model for text generation. \n*/\n chatModel(modelId?: BasetenChatModelId): LanguageModelV3;\n\n /**\nCreates a language model for text generation. Alias for chatModel.\n*/\n languageModel(modelId?: BasetenChatModelId): LanguageModelV3;\n\n /**\nCreates a text embedding model for text generation.\n*/\n textEmbeddingModel(\n modelId?: BasetenEmbeddingModelId,\n ): EmbeddingModelV3<string>;\n}\n\n// by default, we use the Model APIs\nconst defaultBaseURL = 'https://inference.baseten.co/v1';\n\nexport function createBaseten(\n options: BasetenProviderSettings = {},\n): BasetenProvider {\n const baseURL = withoutTrailingSlash(options.baseURL ?? defaultBaseURL);\n const getHeaders = () =>\n withUserAgentSuffix(\n {\n Authorization: `Bearer ${loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'BASETEN_API_KEY',\n description: 'Baseten API key',\n })}`,\n ...options.headers,\n },\n `ai-sdk/baseten/${VERSION}`,\n );\n\n interface CommonModelConfig {\n provider: string;\n url: ({ path }: { path: string }) => string;\n headers: () => Record<string, string>;\n fetch?: FetchFunction;\n }\n\n const getCommonModelConfig = (\n modelType: string,\n customURL?: string,\n ): CommonModelConfig => ({\n provider: `baseten.${modelType}`,\n url: ({ path }) => {\n // For embeddings with /sync URLs (but not /sync/v1), we need to add /v1\n if (\n modelType === 'embedding' &&\n customURL?.includes('/sync') &&\n !customURL?.includes('/sync/v1')\n ) {\n return `${customURL}/v1${path}`;\n }\n return `${customURL || baseURL}${path}`;\n },\n headers: getHeaders,\n fetch: options.fetch,\n });\n\n const createChatModel = (modelId?: BasetenChatModelId) => {\n // Use modelURL if provided, otherwise use default Model APIs\n const customURL = options.modelURL;\n\n if (customURL) {\n // Check if this is a /sync/v1 endpoint (OpenAI-compatible) or /predict endpoint (custom)\n const isOpenAICompatible = customURL.includes('/sync/v1');\n\n if (isOpenAICompatible) {\n // For /sync/v1 endpoints, use standard OpenAI-compatible format\n return new OpenAICompatibleChatLanguageModel(modelId ?? 'placeholder', {\n ...getCommonModelConfig('chat', customURL),\n errorStructure: basetenErrorStructure,\n });\n } else if (customURL.includes('/predict')) {\n throw new Error(\n 'Not supported. You must use a /sync/v1 endpoint for chat models.',\n );\n }\n }\n\n // Use default OpenAI-compatible format for Model APIs\n return new OpenAICompatibleChatLanguageModel(modelId ?? 'chat', {\n ...getCommonModelConfig('chat'),\n errorStructure: basetenErrorStructure,\n });\n };\n\n const createTextEmbeddingModel = (modelId?: BasetenEmbeddingModelId) => {\n // Use modelURL if provided\n const customURL = options.modelURL;\n if (!customURL) {\n throw new Error(\n 'No model URL provided for embeddings. Please set modelURL option for embeddings.',\n );\n }\n\n // Check if this is a /sync or /sync/v1 endpoint (OpenAI-compatible)\n // We support both /sync and /sync/v1, stripping /v1 before passing to Performance Client, as Performance Client adds /v1 itself\n const isOpenAICompatible = customURL.includes('/sync');\n\n if (isOpenAICompatible) {\n // Create the model using OpenAICompatibleEmbeddingModel and override doEmbed\n const model = new OpenAICompatibleEmbeddingModel(\n modelId ?? 'embeddings',\n {\n ...getCommonModelConfig('embedding', customURL),\n errorStructure: basetenErrorStructure,\n },\n );\n\n // Strip /v1 from URL if present before passing to Performance Client to avoid double /v1\n const performanceClientURL = customURL.replace('/sync/v1', '/sync');\n\n // Initialize the B10 Performance Client once for reuse\n const performanceClient = new PerformanceClient(\n performanceClientURL,\n loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'BASETEN_API_KEY',\n description: 'Baseten API key',\n }),\n );\n\n // Override the doEmbed method to use the pre-created Performance Client\n model.doEmbed = async params => {\n if (!params.values || !Array.isArray(params.values)) {\n throw new Error('params.values must be an array of strings');\n }\n\n // Performance Client handles batching internally, so we don't need to limit in 128 here\n const response = await performanceClient.embed(\n params.values,\n modelId ?? 'embeddings', // model_id is for Model APIs, we don't use it here for dedicated\n );\n // Transform the response to match the expected format\n const embeddings = response.data.map((item: any) => item.embedding);\n\n return {\n embeddings: embeddings,\n usage: response.usage\n ? { tokens: response.usage.total_tokens }\n : undefined,\n response: { headers: {}, body: response },\n };\n };\n\n return model;\n } else {\n throw new Error(\n 'Not supported. You must use a /sync or /sync/v1 endpoint for embeddings.',\n );\n }\n };\n\n const provider = (modelId?: BasetenChatModelId) => createChatModel(modelId);\n provider.chatModel = createChatModel;\n provider.languageModel = createChatModel;\n provider.imageModel = (modelId: string) => {\n throw new NoSuchModelError({ modelId, modelType: 'imageModel' });\n };\n provider.textEmbeddingModel = createTextEmbeddingModel;\n return provider;\n}\n\nexport const baseten = createBaseten();\n","// Version string of this package injected at build time.\ndeclare const __PACKAGE_VERSION__: string | undefined;\nexport const VERSION: string =\n typeof __PACKAGE_VERSION__ !== 'undefined'\n ? __PACKAGE_VERSION__\n : '0.0.0-test';\n"],"mappings":";AAAA;AAAA,EACE;AAAA,EACA;AAAA,OAEK;AACP;AAAA,EAGE;AAAA,OAEK;AACP;AAAA,EAEE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,SAAS;AAGlB,SAAS,yBAAyB;;;AClB3B,IAAM,UACX,OACI,iBACA;;;ADoBN,IAAM,qBAAqB,EAAE,OAAO;AAAA,EAClC,OAAO,EAAE,OAAO;AAClB,CAAC;AAED,IAAM,wBAAkE;AAAA,EACtE,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AAwDA,IAAM,iBAAiB;AAEhB,SAAS,cACd,UAAmC,CAAC,GACnB;AA5FnB;AA6FE,QAAM,UAAU,sBAAqB,aAAQ,YAAR,YAAmB,cAAc;AACtE,QAAM,aAAa,MACjB;AAAA,IACE;AAAA,MACE,eAAe,UAAU,WAAW;AAAA,QAClC,QAAQ,QAAQ;AAAA,QAChB,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC,CAAC;AAAA,MACF,GAAG,QAAQ;AAAA,IACb;AAAA,IACA,kBAAkB,OAAO;AAAA,EAC3B;AASF,QAAM,uBAAuB,CAC3B,WACA,eACuB;AAAA,IACvB,UAAU,WAAW,SAAS;AAAA,IAC9B,KAAK,CAAC,EAAE,KAAK,MAAM;AAEjB,UACE,cAAc,gBACd,uCAAW,SAAS,aACpB,EAAC,uCAAW,SAAS,cACrB;AACA,eAAO,GAAG,SAAS,MAAM,IAAI;AAAA,MAC/B;AACA,aAAO,GAAG,aAAa,OAAO,GAAG,IAAI;AAAA,IACvC;AAAA,IACA,SAAS;AAAA,IACT,OAAO,QAAQ;AAAA,EACjB;AAEA,QAAM,kBAAkB,CAAC,YAAiC;AAExD,UAAM,YAAY,QAAQ;AAE1B,QAAI,WAAW;AAEb,YAAM,qBAAqB,UAAU,SAAS,UAAU;AAExD,UAAI,oBAAoB;AAEtB,eAAO,IAAI,kCAAkC,4BAAW,eAAe;AAAA,UACrE,GAAG,qBAAqB,QAAQ,SAAS;AAAA,UACzC,gBAAgB;AAAA,QAClB,CAAC;AAAA,MACH,WAAW,UAAU,SAAS,UAAU,GAAG;AACzC,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAGA,WAAO,IAAI,kCAAkC,4BAAW,QAAQ;AAAA,MAC9D,GAAG,qBAAqB,MAAM;AAAA,MAC9B,gBAAgB;AAAA,IAClB,CAAC;AAAA,EACH;AAEA,QAAM,2BAA2B,CAAC,YAAsC;AAEtE,UAAM,YAAY,QAAQ;AAC1B,QAAI,CAAC,WAAW;AACd,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAIA,UAAM,qBAAqB,UAAU,SAAS,OAAO;AAErD,QAAI,oBAAoB;AAEtB,YAAM,QAAQ,IAAI;AAAA,QAChB,4BAAW;AAAA,QACX;AAAA,UACE,GAAG,qBAAqB,aAAa,SAAS;AAAA,UAC9C,gBAAgB;AAAA,QAClB;AAAA,MACF;AAGA,YAAM,uBAAuB,UAAU,QAAQ,YAAY,OAAO;AAGlE,YAAM,oBAAoB,IAAI;AAAA,QAC5B;AAAA,QACA,WAAW;AAAA,UACT,QAAQ,QAAQ;AAAA,UAChB,yBAAyB;AAAA,UACzB,aAAa;AAAA,QACf,CAAC;AAAA,MACH;AAGA,YAAM,UAAU,OAAM,WAAU;AAC9B,YAAI,CAAC,OAAO,UAAU,CAAC,MAAM,QAAQ,OAAO,MAAM,GAAG;AACnD,gBAAM,IAAI,MAAM,2CAA2C;AAAA,QAC7D;AAGA,cAAM,WAAW,MAAM,kBAAkB;AAAA,UACvC,OAAO;AAAA,UACP,4BAAW;AAAA;AAAA,QACb;AAEA,cAAM,aAAa,SAAS,KAAK,IAAI,CAAC,SAAc,KAAK,SAAS;AAElE,eAAO;AAAA,UACL;AAAA,UACA,OAAO,SAAS,QACZ,EAAE,QAAQ,SAAS,MAAM,aAAa,IACtC;AAAA,UACJ,UAAU,EAAE,SAAS,CAAC,GAAG,MAAM,SAAS;AAAA,QAC1C;AAAA,MACF;AAEA,aAAO;AAAA,IACT,OAAO;AACL,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,QAAM,WAAW,CAAC,YAAiC,gBAAgB,OAAO;AAC1E,WAAS,YAAY;AACrB,WAAS,gBAAgB;AACzB,WAAS,aAAa,CAAC,YAAoB;AACzC,UAAM,IAAI,iBAAiB,EAAE,SAAS,WAAW,aAAa,CAAC;AAAA,EACjE;AACA,WAAS,qBAAqB;AAC9B,SAAO;AACT;AAEO,IAAM,UAAU,cAAc;","names":[]}
1
+ {"version":3,"sources":["../src/baseten-provider.ts","../src/version.ts"],"sourcesContent":["import {\n OpenAICompatibleChatLanguageModel,\n OpenAICompatibleEmbeddingModel,\n ProviderErrorStructure,\n} from '@ai-sdk/openai-compatible';\nimport {\n EmbeddingModelV3,\n LanguageModelV3,\n NoSuchModelError,\n ProviderV3,\n} from '@ai-sdk/provider';\nimport {\n FetchFunction,\n loadApiKey,\n withoutTrailingSlash,\n withUserAgentSuffix,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport { BasetenChatModelId } from './baseten-chat-options';\nimport { BasetenEmbeddingModelId } from './baseten-embedding-options';\nimport { PerformanceClient } from '@basetenlabs/performance-client';\nimport { VERSION } from './version';\n\nexport type BasetenErrorData = z.infer<typeof basetenErrorSchema>;\n\nconst basetenErrorSchema = z.object({\n error: z.string(),\n});\n\nconst basetenErrorStructure: ProviderErrorStructure<BasetenErrorData> = {\n errorSchema: basetenErrorSchema,\n errorToMessage: data => data.error,\n};\n\nexport interface BasetenProviderSettings {\n /**\n * Baseten API key. Default value is taken from the `BASETEN_API_KEY`\n * environment variable.\n */\n apiKey?: string;\n\n /**\n * Base URL for the Model APIs. Default: 'https://inference.baseten.co/v1'\n */\n baseURL?: string;\n\n /**\n * Model URL for custom models (chat or embeddings).\n * If not supplied, the default Model APIs will be used.\n */\n modelURL?: string;\n /**\n * Custom headers to include in the requests.\n */\n headers?: Record<string, string>;\n\n /**\n * Custom fetch implementation. You can use it as a middleware to intercept requests,\n * or to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n}\n\nexport interface BasetenProvider extends ProviderV3 {\n /**\nCreates a chat model for text generation.\n*/\n (modelId?: BasetenChatModelId): LanguageModelV3;\n\n /**\nCreates a chat model for text generation.\n*/\n chatModel(modelId?: BasetenChatModelId): LanguageModelV3;\n\n /**\nCreates a language model for text generation. Alias for chatModel.\n*/\n languageModel(modelId?: BasetenChatModelId): LanguageModelV3;\n\n /**\nCreates a embedding model for text generation.\n*/\n embeddingModel(modelId?: BasetenEmbeddingModelId): EmbeddingModelV3;\n\n /**\n * @deprecated Use `embeddingModel` instead.\n */\n textEmbeddingModel(modelId?: BasetenEmbeddingModelId): EmbeddingModelV3;\n}\n\n// by default, we use the Model APIs\nconst defaultBaseURL = 'https://inference.baseten.co/v1';\n\nexport function createBaseten(\n options: BasetenProviderSettings = {},\n): BasetenProvider {\n const baseURL = withoutTrailingSlash(options.baseURL ?? defaultBaseURL);\n const getHeaders = () =>\n withUserAgentSuffix(\n {\n Authorization: `Bearer ${loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'BASETEN_API_KEY',\n description: 'Baseten API key',\n })}`,\n ...options.headers,\n },\n `ai-sdk/baseten/${VERSION}`,\n );\n\n interface CommonModelConfig {\n provider: string;\n url: ({ path }: { path: string }) => string;\n headers: () => Record<string, string>;\n fetch?: FetchFunction;\n }\n\n const getCommonModelConfig = (\n modelType: string,\n customURL?: string,\n ): CommonModelConfig => ({\n provider: `baseten.${modelType}`,\n url: ({ path }) => {\n // For embeddings with /sync URLs (but not /sync/v1), we need to add /v1\n if (\n modelType === 'embedding' &&\n customURL?.includes('/sync') &&\n !customURL?.includes('/sync/v1')\n ) {\n return `${customURL}/v1${path}`;\n }\n return `${customURL || baseURL}${path}`;\n },\n headers: getHeaders,\n fetch: options.fetch,\n });\n\n const createChatModel = (modelId?: BasetenChatModelId) => {\n // Use modelURL if provided, otherwise use default Model APIs\n const customURL = options.modelURL;\n\n if (customURL) {\n // Check if this is a /sync/v1 endpoint (OpenAI-compatible) or /predict endpoint (custom)\n const isOpenAICompatible = customURL.includes('/sync/v1');\n\n if (isOpenAICompatible) {\n // For /sync/v1 endpoints, use standard OpenAI-compatible format\n return new OpenAICompatibleChatLanguageModel(modelId ?? 'placeholder', {\n ...getCommonModelConfig('chat', customURL),\n errorStructure: basetenErrorStructure,\n });\n } else if (customURL.includes('/predict')) {\n throw new Error(\n 'Not supported. You must use a /sync/v1 endpoint for chat models.',\n );\n }\n }\n\n // Use default OpenAI-compatible format for Model APIs\n return new OpenAICompatibleChatLanguageModel(modelId ?? 'chat', {\n ...getCommonModelConfig('chat'),\n errorStructure: basetenErrorStructure,\n });\n };\n\n const createEmbeddingModel = (modelId?: BasetenEmbeddingModelId) => {\n // Use modelURL if provided\n const customURL = options.modelURL;\n if (!customURL) {\n throw new Error(\n 'No model URL provided for embeddings. Please set modelURL option for embeddings.',\n );\n }\n\n // Check if this is a /sync or /sync/v1 endpoint (OpenAI-compatible)\n // We support both /sync and /sync/v1, stripping /v1 before passing to Performance Client, as Performance Client adds /v1 itself\n const isOpenAICompatible = customURL.includes('/sync');\n\n if (isOpenAICompatible) {\n // Create the model using OpenAICompatibleEmbeddingModel and override doEmbed\n const model = new OpenAICompatibleEmbeddingModel(\n modelId ?? 'embeddings',\n {\n ...getCommonModelConfig('embedding', customURL),\n errorStructure: basetenErrorStructure,\n },\n );\n\n // Strip /v1 from URL if present before passing to Performance Client to avoid double /v1\n const performanceClientURL = customURL.replace('/sync/v1', '/sync');\n\n // Initialize the B10 Performance Client once for reuse\n const performanceClient = new PerformanceClient(\n performanceClientURL,\n loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'BASETEN_API_KEY',\n description: 'Baseten API key',\n }),\n );\n\n // Override the doEmbed method to use the pre-created Performance Client\n model.doEmbed = async params => {\n if (!params.values || !Array.isArray(params.values)) {\n throw new Error('params.values must be an array of strings');\n }\n\n // Performance Client handles batching internally, so we don't need to limit in 128 here\n const response = await performanceClient.embed(\n params.values,\n modelId ?? 'embeddings', // model_id is for Model APIs, we don't use it here for dedicated\n );\n // Transform the response to match the expected format\n const embeddings = response.data.map((item: any) => item.embedding);\n\n return {\n embeddings,\n usage: response.usage\n ? { tokens: response.usage.total_tokens }\n : undefined,\n response: { headers: {}, body: response },\n warnings: [],\n };\n };\n\n return model;\n } else {\n throw new Error(\n 'Not supported. You must use a /sync or /sync/v1 endpoint for embeddings.',\n );\n }\n };\n\n const provider = (modelId?: BasetenChatModelId) => createChatModel(modelId);\n\n provider.specificationVersion = 'v3' as const;\n provider.chatModel = createChatModel;\n provider.languageModel = createChatModel;\n provider.imageModel = (modelId: string) => {\n throw new NoSuchModelError({ modelId, modelType: 'imageModel' });\n };\n provider.embeddingModel = createEmbeddingModel;\n provider.textEmbeddingModel = createEmbeddingModel;\n return provider;\n}\n\nexport const baseten = createBaseten();\n","// Version string of this package injected at build time.\ndeclare const __PACKAGE_VERSION__: string | undefined;\nexport const VERSION: string =\n typeof __PACKAGE_VERSION__ !== 'undefined'\n ? __PACKAGE_VERSION__\n : '0.0.0-test';\n"],"mappings":";AAAA;AAAA,EACE;AAAA,EACA;AAAA,OAEK;AACP;AAAA,EAGE;AAAA,OAEK;AACP;AAAA,EAEE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,SAAS;AAGlB,SAAS,yBAAyB;;;AClB3B,IAAM,UACX,OACI,kBACA;;;ADoBN,IAAM,qBAAqB,EAAE,OAAO;AAAA,EAClC,OAAO,EAAE,OAAO;AAClB,CAAC;AAED,IAAM,wBAAkE;AAAA,EACtE,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AA2DA,IAAM,iBAAiB;AAEhB,SAAS,cACd,UAAmC,CAAC,GACnB;AA/FnB;AAgGE,QAAM,UAAU,sBAAqB,aAAQ,YAAR,YAAmB,cAAc;AACtE,QAAM,aAAa,MACjB;AAAA,IACE;AAAA,MACE,eAAe,UAAU,WAAW;AAAA,QAClC,QAAQ,QAAQ;AAAA,QAChB,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC,CAAC;AAAA,MACF,GAAG,QAAQ;AAAA,IACb;AAAA,IACA,kBAAkB,OAAO;AAAA,EAC3B;AASF,QAAM,uBAAuB,CAC3B,WACA,eACuB;AAAA,IACvB,UAAU,WAAW,SAAS;AAAA,IAC9B,KAAK,CAAC,EAAE,KAAK,MAAM;AAEjB,UACE,cAAc,gBACd,uCAAW,SAAS,aACpB,EAAC,uCAAW,SAAS,cACrB;AACA,eAAO,GAAG,SAAS,MAAM,IAAI;AAAA,MAC/B;AACA,aAAO,GAAG,aAAa,OAAO,GAAG,IAAI;AAAA,IACvC;AAAA,IACA,SAAS;AAAA,IACT,OAAO,QAAQ;AAAA,EACjB;AAEA,QAAM,kBAAkB,CAAC,YAAiC;AAExD,UAAM,YAAY,QAAQ;AAE1B,QAAI,WAAW;AAEb,YAAM,qBAAqB,UAAU,SAAS,UAAU;AAExD,UAAI,oBAAoB;AAEtB,eAAO,IAAI,kCAAkC,4BAAW,eAAe;AAAA,UACrE,GAAG,qBAAqB,QAAQ,SAAS;AAAA,UACzC,gBAAgB;AAAA,QAClB,CAAC;AAAA,MACH,WAAW,UAAU,SAAS,UAAU,GAAG;AACzC,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAGA,WAAO,IAAI,kCAAkC,4BAAW,QAAQ;AAAA,MAC9D,GAAG,qBAAqB,MAAM;AAAA,MAC9B,gBAAgB;AAAA,IAClB,CAAC;AAAA,EACH;AAEA,QAAM,uBAAuB,CAAC,YAAsC;AAElE,UAAM,YAAY,QAAQ;AAC1B,QAAI,CAAC,WAAW;AACd,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAIA,UAAM,qBAAqB,UAAU,SAAS,OAAO;AAErD,QAAI,oBAAoB;AAEtB,YAAM,QAAQ,IAAI;AAAA,QAChB,4BAAW;AAAA,QACX;AAAA,UACE,GAAG,qBAAqB,aAAa,SAAS;AAAA,UAC9C,gBAAgB;AAAA,QAClB;AAAA,MACF;AAGA,YAAM,uBAAuB,UAAU,QAAQ,YAAY,OAAO;AAGlE,YAAM,oBAAoB,IAAI;AAAA,QAC5B;AAAA,QACA,WAAW;AAAA,UACT,QAAQ,QAAQ;AAAA,UAChB,yBAAyB;AAAA,UACzB,aAAa;AAAA,QACf,CAAC;AAAA,MACH;AAGA,YAAM,UAAU,OAAM,WAAU;AAC9B,YAAI,CAAC,OAAO,UAAU,CAAC,MAAM,QAAQ,OAAO,MAAM,GAAG;AACnD,gBAAM,IAAI,MAAM,2CAA2C;AAAA,QAC7D;AAGA,cAAM,WAAW,MAAM,kBAAkB;AAAA,UACvC,OAAO;AAAA,UACP,4BAAW;AAAA;AAAA,QACb;AAEA,cAAM,aAAa,SAAS,KAAK,IAAI,CAAC,SAAc,KAAK,SAAS;AAElE,eAAO;AAAA,UACL;AAAA,UACA,OAAO,SAAS,QACZ,EAAE,QAAQ,SAAS,MAAM,aAAa,IACtC;AAAA,UACJ,UAAU,EAAE,SAAS,CAAC,GAAG,MAAM,SAAS;AAAA,UACxC,UAAU,CAAC;AAAA,QACb;AAAA,MACF;AAEA,aAAO;AAAA,IACT,OAAO;AACL,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,QAAM,WAAW,CAAC,YAAiC,gBAAgB,OAAO;AAE1E,WAAS,uBAAuB;AAChC,WAAS,YAAY;AACrB,WAAS,gBAAgB;AACzB,WAAS,aAAa,CAAC,YAAoB;AACzC,UAAM,IAAI,iBAAiB,EAAE,SAAS,WAAW,aAAa,CAAC;AAAA,EACjE;AACA,WAAS,iBAAiB;AAC1B,WAAS,qBAAqB;AAC9B,SAAO;AACT;AAEO,IAAM,UAAU,cAAc;","names":[]}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@ai-sdk/baseten",
3
- "version": "1.0.0-beta.6",
3
+ "version": "1.0.0-beta.61",
4
4
  "license": "Apache-2.0",
5
5
  "sideEffects": false,
6
6
  "main": "./dist/index.js",
@@ -8,7 +8,8 @@
8
8
  "types": "./dist/index.d.ts",
9
9
  "files": [
10
10
  "dist/**/*",
11
- "CHANGELOG.md"
11
+ "CHANGELOG.md",
12
+ "README.md"
12
13
  ],
13
14
  "exports": {
14
15
  "./package.json": "./package.json",
@@ -20,9 +21,9 @@
20
21
  },
21
22
  "dependencies": {
22
23
  "@basetenlabs/performance-client": "^0.0.10",
23
- "@ai-sdk/openai-compatible": "1.1.0-beta.5",
24
- "@ai-sdk/provider": "2.1.0-beta.3",
25
- "@ai-sdk/provider-utils": "3.1.0-beta.5"
24
+ "@ai-sdk/openai-compatible": "2.0.0-beta.59",
25
+ "@ai-sdk/provider": "3.0.0-beta.31",
26
+ "@ai-sdk/provider-utils": "4.0.0-beta.58"
26
27
  },
27
28
  "devDependencies": {
28
29
  "@types/node": "20.17.24",