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

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,485 @@
1
1
  # @ai-sdk/baseten
2
2
 
3
+ ## 1.0.0-beta.60
4
+
5
+ ### Patch Changes
6
+
7
+ - Updated dependencies [cbf52cd]
8
+ - @ai-sdk/openai-compatible@2.0.0-beta.58
9
+ - @ai-sdk/provider@3.0.0-beta.30
10
+ - @ai-sdk/provider-utils@4.0.0-beta.57
11
+
12
+ ## 1.0.0-beta.59
13
+
14
+ ### Patch Changes
15
+
16
+ - Updated dependencies [9549c9e]
17
+ - @ai-sdk/provider@3.0.0-beta.29
18
+ - @ai-sdk/openai-compatible@2.0.0-beta.57
19
+ - @ai-sdk/provider-utils@4.0.0-beta.56
20
+
21
+ ## 1.0.0-beta.58
22
+
23
+ ### Patch Changes
24
+
25
+ - Updated dependencies [50b70d6]
26
+ - @ai-sdk/provider-utils@4.0.0-beta.55
27
+ - @ai-sdk/openai-compatible@2.0.0-beta.56
28
+
29
+ ## 1.0.0-beta.57
30
+
31
+ ### Patch Changes
32
+
33
+ - Updated dependencies [9061dc0]
34
+ - @ai-sdk/openai-compatible@2.0.0-beta.55
35
+ - @ai-sdk/provider-utils@4.0.0-beta.54
36
+ - @ai-sdk/provider@3.0.0-beta.28
37
+
38
+ ## 1.0.0-beta.56
39
+
40
+ ### Patch Changes
41
+
42
+ - 366f50b: chore(provider): add deprecated textEmbeddingModel and textEmbedding aliases
43
+ - Updated dependencies [366f50b]
44
+ - @ai-sdk/openai-compatible@2.0.0-beta.54
45
+ - @ai-sdk/provider@3.0.0-beta.27
46
+ - @ai-sdk/provider-utils@4.0.0-beta.53
47
+
48
+ ## 1.0.0-beta.55
49
+
50
+ ### Patch Changes
51
+
52
+ - Updated dependencies [763d04a]
53
+ - @ai-sdk/provider-utils@4.0.0-beta.52
54
+ - @ai-sdk/openai-compatible@2.0.0-beta.53
55
+
56
+ ## 1.0.0-beta.54
57
+
58
+ ### Patch Changes
59
+
60
+ - Updated dependencies [c1efac4]
61
+ - @ai-sdk/provider-utils@4.0.0-beta.51
62
+ - @ai-sdk/openai-compatible@2.0.0-beta.52
63
+
64
+ ## 1.0.0-beta.53
65
+
66
+ ### Patch Changes
67
+
68
+ - Updated dependencies [32223c8]
69
+ - @ai-sdk/provider-utils@4.0.0-beta.50
70
+ - @ai-sdk/openai-compatible@2.0.0-beta.51
71
+
72
+ ## 1.0.0-beta.52
73
+
74
+ ### Patch Changes
75
+
76
+ - Updated dependencies [83e5744]
77
+ - @ai-sdk/provider-utils@4.0.0-beta.49
78
+ - @ai-sdk/openai-compatible@2.0.0-beta.50
79
+
80
+ ## 1.0.0-beta.51
81
+
82
+ ### Patch Changes
83
+
84
+ - Updated dependencies [960ec8f]
85
+ - @ai-sdk/provider-utils@4.0.0-beta.48
86
+ - @ai-sdk/openai-compatible@2.0.0-beta.49
87
+
88
+ ## 1.0.0-beta.50
89
+
90
+ ### Patch Changes
91
+
92
+ - Updated dependencies [e9e157f]
93
+ - @ai-sdk/provider-utils@4.0.0-beta.47
94
+ - @ai-sdk/openai-compatible@2.0.0-beta.48
95
+
96
+ ## 1.0.0-beta.49
97
+
98
+ ### Patch Changes
99
+
100
+ - Updated dependencies [81e29ab]
101
+ - @ai-sdk/provider-utils@4.0.0-beta.46
102
+ - @ai-sdk/openai-compatible@2.0.0-beta.47
103
+
104
+ ## 1.0.0-beta.48
105
+
106
+ ### Patch Changes
107
+
108
+ - Updated dependencies [3bd2689]
109
+ - @ai-sdk/openai-compatible@2.0.0-beta.46
110
+ - @ai-sdk/provider@3.0.0-beta.26
111
+ - @ai-sdk/provider-utils@4.0.0-beta.45
112
+
113
+ ## 1.0.0-beta.47
114
+
115
+ ### Patch Changes
116
+
117
+ - Updated dependencies [53f3368]
118
+ - @ai-sdk/provider@3.0.0-beta.25
119
+ - @ai-sdk/openai-compatible@2.0.0-beta.45
120
+ - @ai-sdk/provider-utils@4.0.0-beta.44
121
+
122
+ ## 1.0.0-beta.46
123
+
124
+ ### Patch Changes
125
+
126
+ - Updated dependencies [dce03c4]
127
+ - @ai-sdk/provider-utils@4.0.0-beta.43
128
+ - @ai-sdk/provider@3.0.0-beta.24
129
+ - @ai-sdk/openai-compatible@2.0.0-beta.44
130
+
131
+ ## 1.0.0-beta.45
132
+
133
+ ### Patch Changes
134
+
135
+ - Updated dependencies [3ed5519]
136
+ - @ai-sdk/provider-utils@4.0.0-beta.42
137
+ - @ai-sdk/openai-compatible@2.0.0-beta.43
138
+
139
+ ## 1.0.0-beta.44
140
+
141
+ ### Patch Changes
142
+
143
+ - Updated dependencies [1bd7d32]
144
+ - @ai-sdk/openai-compatible@2.0.0-beta.42
145
+ - @ai-sdk/provider-utils@4.0.0-beta.41
146
+ - @ai-sdk/provider@3.0.0-beta.23
147
+
148
+ ## 1.0.0-beta.43
149
+
150
+ ### Patch Changes
151
+
152
+ - Updated dependencies [544d4e8]
153
+ - @ai-sdk/openai-compatible@2.0.0-beta.41
154
+ - @ai-sdk/provider-utils@4.0.0-beta.40
155
+ - @ai-sdk/provider@3.0.0-beta.22
156
+
157
+ ## 1.0.0-beta.42
158
+
159
+ ### Patch Changes
160
+
161
+ - Updated dependencies [954c356]
162
+ - @ai-sdk/provider-utils@4.0.0-beta.39
163
+ - @ai-sdk/provider@3.0.0-beta.21
164
+ - @ai-sdk/openai-compatible@2.0.0-beta.40
165
+
166
+ ## 1.0.0-beta.41
167
+
168
+ ### Patch Changes
169
+
170
+ - Updated dependencies [03849b0]
171
+ - @ai-sdk/provider-utils@4.0.0-beta.38
172
+ - @ai-sdk/openai-compatible@2.0.0-beta.39
173
+
174
+ ## 1.0.0-beta.40
175
+
176
+ ### Patch Changes
177
+
178
+ - Updated dependencies [457318b]
179
+ - @ai-sdk/openai-compatible@2.0.0-beta.38
180
+ - @ai-sdk/provider@3.0.0-beta.20
181
+ - @ai-sdk/provider-utils@4.0.0-beta.37
182
+
183
+ ## 1.0.0-beta.39
184
+
185
+ ### Patch Changes
186
+
187
+ - 8d9e8ad: chore(provider): remove generics from EmbeddingModelV3
188
+
189
+ Before
190
+
191
+ ```ts
192
+ model.textEmbeddingModel('my-model-id');
193
+ ```
194
+
195
+ After
196
+
197
+ ```ts
198
+ model.embeddingModel('my-model-id');
199
+ ```
200
+
201
+ - Updated dependencies [8d9e8ad]
202
+ - @ai-sdk/openai-compatible@2.0.0-beta.37
203
+ - @ai-sdk/provider@3.0.0-beta.19
204
+ - @ai-sdk/provider-utils@4.0.0-beta.36
205
+
206
+ ## 1.0.0-beta.38
207
+
208
+ ### Patch Changes
209
+
210
+ - Updated dependencies [10d819b]
211
+ - @ai-sdk/provider@3.0.0-beta.18
212
+ - @ai-sdk/openai-compatible@2.0.0-beta.36
213
+ - @ai-sdk/provider-utils@4.0.0-beta.35
214
+
215
+ ## 1.0.0-beta.37
216
+
217
+ ### Patch Changes
218
+
219
+ - Updated dependencies [db913bd]
220
+ - @ai-sdk/provider@3.0.0-beta.17
221
+ - @ai-sdk/openai-compatible@2.0.0-beta.35
222
+ - @ai-sdk/provider-utils@4.0.0-beta.34
223
+
224
+ ## 1.0.0-beta.36
225
+
226
+ ### Patch Changes
227
+
228
+ - ab9af9c: add moonshotai/Kimi-K2-Thinking model ID for Baseten provider
229
+
230
+ ## 1.0.0-beta.35
231
+
232
+ ### Patch Changes
233
+
234
+ - Updated dependencies [b681d7d]
235
+ - @ai-sdk/provider@3.0.0-beta.16
236
+ - @ai-sdk/openai-compatible@2.0.0-beta.34
237
+ - @ai-sdk/provider-utils@4.0.0-beta.33
238
+
239
+ ## 1.0.0-beta.34
240
+
241
+ ### Patch Changes
242
+
243
+ - Updated dependencies [32d8dbb]
244
+ - @ai-sdk/provider-utils@4.0.0-beta.32
245
+ - @ai-sdk/openai-compatible@2.0.0-beta.33
246
+
247
+ ## 1.0.0-beta.33
248
+
249
+ ### Patch Changes
250
+
251
+ - Updated dependencies [bb36798]
252
+ - @ai-sdk/provider@3.0.0-beta.15
253
+ - @ai-sdk/openai-compatible@2.0.0-beta.32
254
+ - @ai-sdk/provider-utils@4.0.0-beta.31
255
+
256
+ ## 1.0.0-beta.32
257
+
258
+ ### Patch Changes
259
+
260
+ - Updated dependencies [4f16c37]
261
+ - @ai-sdk/provider-utils@4.0.0-beta.30
262
+ - @ai-sdk/openai-compatible@2.0.0-beta.31
263
+
264
+ ## 1.0.0-beta.31
265
+
266
+ ### Patch Changes
267
+
268
+ - Updated dependencies [af3780b]
269
+ - @ai-sdk/provider@3.0.0-beta.14
270
+ - @ai-sdk/openai-compatible@2.0.0-beta.30
271
+ - @ai-sdk/provider-utils@4.0.0-beta.29
272
+
273
+ ## 1.0.0-beta.30
274
+
275
+ ### Patch Changes
276
+
277
+ - Updated dependencies [016b111]
278
+ - @ai-sdk/provider-utils@4.0.0-beta.28
279
+ - @ai-sdk/openai-compatible@2.0.0-beta.29
280
+
281
+ ## 1.0.0-beta.29
282
+
283
+ ### Patch Changes
284
+
285
+ - Updated dependencies [37c58a0]
286
+ - @ai-sdk/provider@3.0.0-beta.13
287
+ - @ai-sdk/openai-compatible@2.0.0-beta.28
288
+ - @ai-sdk/provider-utils@4.0.0-beta.27
289
+
290
+ ## 1.0.0-beta.28
291
+
292
+ ### Patch Changes
293
+
294
+ - Updated dependencies [d1bdadb]
295
+ - @ai-sdk/provider@3.0.0-beta.12
296
+ - @ai-sdk/openai-compatible@2.0.0-beta.27
297
+ - @ai-sdk/provider-utils@4.0.0-beta.26
298
+
299
+ ## 1.0.0-beta.27
300
+
301
+ ### Patch Changes
302
+
303
+ - Updated dependencies [4c44a5b]
304
+ - @ai-sdk/provider@3.0.0-beta.11
305
+ - @ai-sdk/openai-compatible@2.0.0-beta.26
306
+ - @ai-sdk/provider-utils@4.0.0-beta.25
307
+
308
+ ## 1.0.0-beta.26
309
+
310
+ ### Patch Changes
311
+
312
+ - 0c3b58b: fix(provider): add specificationVersion to ProviderV3
313
+ - Updated dependencies [0c3b58b]
314
+ - @ai-sdk/openai-compatible@2.0.0-beta.25
315
+ - @ai-sdk/provider@3.0.0-beta.10
316
+ - @ai-sdk/provider-utils@4.0.0-beta.24
317
+
318
+ ## 1.0.0-beta.25
319
+
320
+ ### Patch Changes
321
+
322
+ - Updated dependencies [a755db5]
323
+ - @ai-sdk/provider@3.0.0-beta.9
324
+ - @ai-sdk/openai-compatible@2.0.0-beta.24
325
+ - @ai-sdk/provider-utils@4.0.0-beta.23
326
+
327
+ ## 1.0.0-beta.24
328
+
329
+ ### Patch Changes
330
+
331
+ - Updated dependencies [58920e0]
332
+ - @ai-sdk/provider-utils@4.0.0-beta.22
333
+ - @ai-sdk/openai-compatible@2.0.0-beta.23
334
+
335
+ ## 1.0.0-beta.23
336
+
337
+ ### Patch Changes
338
+
339
+ - Updated dependencies [293a6b7]
340
+ - @ai-sdk/provider-utils@4.0.0-beta.21
341
+ - @ai-sdk/openai-compatible@2.0.0-beta.22
342
+
343
+ ## 1.0.0-beta.22
344
+
345
+ ### Patch Changes
346
+
347
+ - Updated dependencies [fca786b]
348
+ - @ai-sdk/provider-utils@4.0.0-beta.20
349
+ - @ai-sdk/openai-compatible@2.0.0-beta.21
350
+
351
+ ## 1.0.0-beta.21
352
+
353
+ ### Patch Changes
354
+
355
+ - Updated dependencies [3794514]
356
+ - @ai-sdk/provider-utils@4.0.0-beta.19
357
+ - @ai-sdk/provider@3.0.0-beta.8
358
+ - @ai-sdk/openai-compatible@2.0.0-beta.20
359
+
360
+ ## 1.0.0-beta.20
361
+
362
+ ### Patch Changes
363
+
364
+ - Updated dependencies [81d4308]
365
+ - @ai-sdk/provider@3.0.0-beta.7
366
+ - @ai-sdk/openai-compatible@2.0.0-beta.19
367
+ - @ai-sdk/provider-utils@4.0.0-beta.18
368
+
369
+ ## 1.0.0-beta.19
370
+
371
+ ### Patch Changes
372
+
373
+ - Updated dependencies [703459a]
374
+ - @ai-sdk/provider-utils@4.0.0-beta.17
375
+ - @ai-sdk/openai-compatible@2.0.0-beta.18
376
+
377
+ ## 1.0.0-beta.18
378
+
379
+ ### Patch Changes
380
+
381
+ - Updated dependencies [b689220]
382
+ - @ai-sdk/openai-compatible@2.0.0-beta.17
383
+
384
+ ## 1.0.0-beta.17
385
+
386
+ ### Patch Changes
387
+
388
+ - Updated dependencies [6306603]
389
+ - @ai-sdk/provider-utils@4.0.0-beta.16
390
+ - @ai-sdk/openai-compatible@2.0.0-beta.16
391
+
392
+ ## 1.0.0-beta.16
393
+
394
+ ### Patch Changes
395
+
396
+ - Updated dependencies [f0b2157]
397
+ - @ai-sdk/provider-utils@4.0.0-beta.15
398
+ - @ai-sdk/openai-compatible@2.0.0-beta.15
399
+
400
+ ## 1.0.0-beta.15
401
+
402
+ ### Patch Changes
403
+
404
+ - Updated dependencies [3b1d015]
405
+ - @ai-sdk/provider-utils@4.0.0-beta.14
406
+ - @ai-sdk/openai-compatible@2.0.0-beta.14
407
+
408
+ ## 1.0.0-beta.14
409
+
410
+ ### Patch Changes
411
+
412
+ - Updated dependencies [d116b4b]
413
+ - @ai-sdk/provider-utils@4.0.0-beta.13
414
+ - @ai-sdk/openai-compatible@2.0.0-beta.13
415
+
416
+ ## 1.0.0-beta.13
417
+
418
+ ### Patch Changes
419
+
420
+ - Updated dependencies [7e32fea]
421
+ - @ai-sdk/provider-utils@4.0.0-beta.12
422
+ - @ai-sdk/openai-compatible@2.0.0-beta.12
423
+
424
+ ## 1.0.0-beta.12
425
+
426
+ ### Patch Changes
427
+
428
+ - 95f65c2: chore: use import \* from zod/v4
429
+ - Updated dependencies [95f65c2]
430
+ - Updated dependencies [95f65c2]
431
+ - @ai-sdk/openai-compatible@2.0.0-beta.11
432
+ - @ai-sdk/provider-utils@4.0.0-beta.11
433
+
434
+ ## 1.0.0-beta.11
435
+
436
+ ### Major Changes
437
+
438
+ - dee8b05: ai SDK 6 beta
439
+
440
+ ### Patch Changes
441
+
442
+ - Updated dependencies [dee8b05]
443
+ - @ai-sdk/openai-compatible@2.0.0-beta.10
444
+ - @ai-sdk/provider@3.0.0-beta.6
445
+ - @ai-sdk/provider-utils@4.0.0-beta.10
446
+
447
+ ## 1.0.0-beta.10
448
+
449
+ ### Patch Changes
450
+
451
+ - Updated dependencies [521c537]
452
+ - @ai-sdk/provider-utils@3.1.0-beta.9
453
+ - @ai-sdk/openai-compatible@1.1.0-beta.9
454
+
455
+ ## 1.0.0-beta.9
456
+
457
+ ### Patch Changes
458
+
459
+ - Updated dependencies [e06565c]
460
+ - @ai-sdk/provider-utils@3.1.0-beta.8
461
+ - @ai-sdk/openai-compatible@1.1.0-beta.8
462
+
463
+ ## 1.0.0-beta.8
464
+
465
+ ### Patch Changes
466
+
467
+ - Updated dependencies [046aa3b]
468
+ - Updated dependencies [e8109d3]
469
+ - @ai-sdk/provider@2.1.0-beta.5
470
+ - @ai-sdk/openai-compatible@1.1.0-beta.7
471
+ - @ai-sdk/provider-utils@3.1.0-beta.7
472
+
473
+ ## 1.0.0-beta.7
474
+
475
+ ### Patch Changes
476
+
477
+ - Updated dependencies [0adc679]
478
+ - Updated dependencies [2b0caef]
479
+ - @ai-sdk/openai-compatible@1.1.0-beta.6
480
+ - @ai-sdk/provider-utils@3.1.0-beta.6
481
+ - @ai-sdk/provider@2.1.0-beta.4
482
+
3
483
  ## 1.0.0-beta.6
4
484
 
5
485
  ### 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.60" : "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.60" : "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.60",
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.58",
25
+ "@ai-sdk/provider": "3.0.0-beta.30",
26
+ "@ai-sdk/provider-utils": "4.0.0-beta.57"
26
27
  },
27
28
  "devDependencies": {
28
29
  "@types/node": "20.17.24",