@ax-llm/ax-ai-sdk-provider 11.0.17 → 11.0.19

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/index.cjs CHANGED
@@ -30,10 +30,14 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
30
30
  // index.ts
31
31
  var index_exports = {};
32
32
  __export(index_exports, {
33
+ AxAIProvider: () => AxAIProvider,
33
34
  AxAgentProvider: () => AxAgentProvider
34
35
  });
35
36
  module.exports = __toCommonJS(index_exports);
36
37
 
38
+ // provider.ts
39
+ var import_web = require("stream/web");
40
+
37
41
  // ../../node_modules/nanoid/index.js
38
42
  var import_crypto = __toESM(require("crypto"), 1);
39
43
  var POOL_SIZE_MULTIPLIER = 128;
@@ -136,6 +140,325 @@ var AxAgentProvider = class {
136
140
  return fn;
137
141
  }
138
142
  };
143
+ var AxAIProvider = class {
144
+ specificationVersion = "v1";
145
+ defaultObjectGenerationMode = "json";
146
+ ai;
147
+ config;
148
+ modelId;
149
+ constructor(ai, config) {
150
+ this.ai = ai;
151
+ this.config = config;
152
+ this.modelId = this.ai.getModelInfo().name;
153
+ }
154
+ get provider() {
155
+ return this.ai.getName();
156
+ }
157
+ async doGenerate(options) {
158
+ const { req, warnings } = createChatRequest(options);
159
+ const res = await this.ai.chat(req);
160
+ const choice = res.results.at(0);
161
+ if (!choice) {
162
+ throw new Error("No choice returned");
163
+ }
164
+ return {
165
+ text: choice.content ?? void 0,
166
+ toolCalls: choice.functionCalls?.map((tc) => ({
167
+ toolCallType: "function",
168
+ toolCallId: tc.id,
169
+ toolName: tc.function.name,
170
+ args: typeof tc.function.params === "string" ? tc.function.params : JSON.stringify(tc.function.params)
171
+ })),
172
+ finishReason: mapAxFinishReason(choice.finishReason),
173
+ usage: {
174
+ promptTokens: res.modelUsage?.promptTokens ?? 0,
175
+ completionTokens: res.modelUsage?.completionTokens ?? 0
176
+ },
177
+ rawCall: { rawPrompt: "", rawSettings: req.modelConfig ?? {} },
178
+ warnings
179
+ };
180
+ }
181
+ async doStream(options) {
182
+ const { req, warnings } = createChatRequest(options);
183
+ const res = await this.ai.chat(req, {
184
+ stream: true
185
+ });
186
+ return {
187
+ stream: res.pipeThrough(new AxToSDKTransformer()),
188
+ rawCall: { rawPrompt: "", rawSettings: req.modelConfig ?? {} },
189
+ warnings
190
+ };
191
+ }
192
+ };
193
+ function prepareToolsAndToolChoice(mode) {
194
+ if (!mode.tools || mode.tools.length === 0) {
195
+ return {};
196
+ }
197
+ const tools = mode.tools;
198
+ const functions = tools.map((f) => ({
199
+ name: f.name,
200
+ description: "description" in f ? f.description ?? "" : "",
201
+ parameters: f.parameters
202
+ }));
203
+ const toolChoice = mode.toolChoice;
204
+ if (!toolChoice) {
205
+ return { functions };
206
+ }
207
+ const type = toolChoice.type;
208
+ switch (type) {
209
+ case "auto":
210
+ return { functions, functionCall: "auto" };
211
+ case "none":
212
+ return { functions, functionCall: "none" };
213
+ case "required":
214
+ return { functions, functionCall: "required" };
215
+ case "tool":
216
+ return {
217
+ functions,
218
+ functionCall: {
219
+ type: "function",
220
+ function: { name: toolChoice.toolName }
221
+ }
222
+ };
223
+ default: {
224
+ const _exhaustiveCheck = type;
225
+ throw new Error(`Unsupported tool choice type: ${_exhaustiveCheck}`);
226
+ }
227
+ }
228
+ }
229
+ function convertToAxChatPrompt(prompt) {
230
+ const messages = [];
231
+ for (const { role, content } of prompt) {
232
+ switch (role) {
233
+ case "system": {
234
+ messages.push({ role: "system", content });
235
+ break;
236
+ }
237
+ case "user": {
238
+ messages.push({
239
+ role: "user",
240
+ content: content.map((part) => {
241
+ switch (part.type) {
242
+ case "text": {
243
+ return { type: "text", text: part.text };
244
+ }
245
+ case "image": {
246
+ if (!part.mimeType) {
247
+ throw new Error("Image part must have a mimeType");
248
+ }
249
+ if (!ArrayBuffer.isView(part.image)) {
250
+ throw new Error("Image part must have an ArrayBuffer");
251
+ }
252
+ const image = Buffer.from(part.image).toString("base64");
253
+ return {
254
+ type: "image",
255
+ mimeType: part.mimeType,
256
+ image
257
+ };
258
+ }
259
+ default:
260
+ throw new Error(`Unsupported part: ${part}`);
261
+ }
262
+ })
263
+ });
264
+ break;
265
+ }
266
+ case "assistant": {
267
+ let text = "";
268
+ const toolCalls = [];
269
+ for (const part of content) {
270
+ switch (part.type) {
271
+ case "text": {
272
+ text += part.text;
273
+ break;
274
+ }
275
+ case "tool-call": {
276
+ toolCalls.push({
277
+ id: part.toolCallId,
278
+ type: "function",
279
+ function: {
280
+ name: part.toolName,
281
+ params: part.args
282
+ }
283
+ });
284
+ break;
285
+ }
286
+ default: {
287
+ const _exhaustiveCheck = part;
288
+ throw new Error(`Unsupported part: ${_exhaustiveCheck}`);
289
+ }
290
+ }
291
+ }
292
+ const functionCalls = toolCalls.length === 0 ? void 0 : toolCalls;
293
+ if (functionCalls || text.length > 0) {
294
+ messages.push({
295
+ role: "assistant",
296
+ content: text,
297
+ functionCalls
298
+ });
299
+ }
300
+ break;
301
+ }
302
+ case "tool": {
303
+ for (const part of content) {
304
+ messages.push({
305
+ role: "function",
306
+ functionId: part.toolCallId,
307
+ result: JSON.stringify(part.result, null, 2)
308
+ });
309
+ }
310
+ break;
311
+ }
312
+ default: {
313
+ const _exhaustiveCheck = role;
314
+ throw new Error(`Unsupported role: ${_exhaustiveCheck}`);
315
+ }
316
+ }
317
+ }
318
+ return messages;
319
+ }
320
+ function mapAxFinishReason(finishReason) {
321
+ switch (finishReason) {
322
+ case "stop":
323
+ return "stop";
324
+ case "length":
325
+ return "length";
326
+ case "function_call":
327
+ return "tool-calls";
328
+ default:
329
+ return "other";
330
+ }
331
+ }
332
+ function createChatRequest({
333
+ mode,
334
+ prompt,
335
+ maxTokens,
336
+ temperature,
337
+ topP,
338
+ frequencyPenalty,
339
+ presencePenalty
340
+ //seed,
341
+ }) {
342
+ const req = {
343
+ chatPrompt: convertToAxChatPrompt(prompt),
344
+ ...frequencyPenalty != null ? { frequencyPenalty } : {},
345
+ ...presencePenalty != null ? { presencePenalty } : {},
346
+ ...maxTokens != null ? { maxTokens } : {},
347
+ ...temperature != null ? { temperature } : {},
348
+ ...topP != null ? { topP } : {}
349
+ };
350
+ const warnings = [];
351
+ switch (mode.type) {
352
+ case "regular": {
353
+ return {
354
+ req: { ...req, ...prepareToolsAndToolChoice(mode) },
355
+ warnings
356
+ };
357
+ }
358
+ case "object-json": {
359
+ return {
360
+ req,
361
+ warnings
362
+ };
363
+ }
364
+ case "object-tool": {
365
+ const tool = {
366
+ type: "function",
367
+ function: {
368
+ name: mode.tool.name,
369
+ params: mode.tool.parameters
370
+ }
371
+ };
372
+ return {
373
+ req: { ...req, ...tool },
374
+ warnings
375
+ };
376
+ }
377
+ default: {
378
+ throw new Error(`Unsupported type`);
379
+ }
380
+ }
381
+ }
382
+ var AxToSDKTransformer = class extends import_web.TransformStream {
383
+ usage = {
384
+ promptTokens: 0,
385
+ completionTokens: 0
386
+ };
387
+ finishReason = "other";
388
+ functionCalls = [];
389
+ constructor() {
390
+ const transformer = {
391
+ transform: (chunk, controller) => {
392
+ const choice = chunk.results.at(0);
393
+ if (!choice) {
394
+ const val = {
395
+ type: "finish",
396
+ finishReason: this.finishReason,
397
+ usage: this.usage
398
+ };
399
+ controller.enqueue(val);
400
+ return;
401
+ }
402
+ if (chunk.modelUsage) {
403
+ this.usage = {
404
+ promptTokens: this.usage.promptTokens + chunk.modelUsage.promptTokens,
405
+ completionTokens: this.usage.completionTokens + chunk.modelUsage?.completionTokens
406
+ };
407
+ }
408
+ if (choice.functionCalls) {
409
+ for (const fc of choice.functionCalls) {
410
+ const index = this.functionCalls.findIndex(
411
+ (f) => f.toolCallId === fc.id
412
+ );
413
+ if (index === -1) {
414
+ this.functionCalls.push({
415
+ toolCallType: "function",
416
+ toolCallId: fc.id,
417
+ toolName: fc.function.name,
418
+ args: typeof fc.function.params === "string" ? fc.function.params : JSON.stringify(fc.function.params)
419
+ });
420
+ } else {
421
+ const obj = this.functionCalls[index];
422
+ if (!obj) {
423
+ continue;
424
+ }
425
+ if (typeof fc.function.params === "string") {
426
+ obj.args = (obj.args ?? "") + fc.function.params;
427
+ } else {
428
+ obj.args = JSON.stringify(fc.function.params);
429
+ }
430
+ }
431
+ this.finishReason = "tool-calls";
432
+ }
433
+ }
434
+ if (choice.content && choice.content.length > 0) {
435
+ controller.enqueue({
436
+ type: "text-delta",
437
+ textDelta: choice.content ?? ""
438
+ });
439
+ this.finishReason = mapAxFinishReason(choice.finishReason);
440
+ }
441
+ },
442
+ flush: (controller) => {
443
+ for (const fc of this.functionCalls) {
444
+ const tc = {
445
+ type: "tool-call",
446
+ ...fc
447
+ };
448
+ controller.enqueue(tc);
449
+ }
450
+ const val = {
451
+ type: "finish",
452
+ finishReason: this.finishReason,
453
+ usage: this.usage
454
+ };
455
+ controller.enqueue(val);
456
+ controller.terminate();
457
+ }
458
+ };
459
+ super(transformer);
460
+ }
461
+ };
139
462
  function convertToZodSchema(jsonSchema) {
140
463
  const { type, properties, required, items } = jsonSchema;
141
464
  switch (type) {
@@ -169,6 +492,7 @@ function convertToZodSchema(jsonSchema) {
169
492
  }
170
493
  // Annotate the CommonJS export names for ESM import in node:
171
494
  0 && (module.exports = {
495
+ AxAIProvider,
172
496
  AxAgentProvider
173
497
  });
174
498
  //# sourceMappingURL=index.cjs.map
package/index.cjs.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"sources":["../index.ts","../../../node_modules/nanoid/index.js","../provider.ts"],"sourcesContent":["import { AxAgentProvider } from './provider.js'\nexport { AxAgentProvider }\n","import crypto from 'crypto'\n\nimport { urlAlphabet } from './url-alphabet/index.js'\n\n// It is best to make fewer, larger requests to the crypto module to\n// avoid system call overhead. So, random numbers are generated in a\n// pool. The pool is a Buffer that is larger than the initial random\n// request size by this multiplier. The pool is enlarged if subsequent\n// requests exceed the maximum buffer size.\nconst POOL_SIZE_MULTIPLIER = 128\nlet pool, poolOffset\n\nlet fillPool = bytes => {\n if (!pool || pool.length < bytes) {\n pool = Buffer.allocUnsafe(bytes * POOL_SIZE_MULTIPLIER)\n crypto.randomFillSync(pool)\n poolOffset = 0\n } else if (poolOffset + bytes > pool.length) {\n crypto.randomFillSync(pool)\n poolOffset = 0\n }\n poolOffset += bytes\n}\n\nlet random = bytes => {\n // `|=` convert `bytes` to number to prevent `valueOf` abusing and pool pollution\n fillPool((bytes |= 0))\n return pool.subarray(poolOffset - bytes, poolOffset)\n}\n\nlet customRandom = (alphabet, defaultSize, getRandom) => {\n // First, a bitmask is necessary to generate the ID. The bitmask makes bytes\n // values closer to the alphabet size. The bitmask calculates the closest\n // `2^31 - 1` number, which exceeds the alphabet size.\n // For example, the bitmask for the alphabet size 30 is 31 (00011111).\n let mask = (2 << (31 - Math.clz32((alphabet.length - 1) | 1))) - 1\n // Though, the bitmask solution is not perfect since the bytes exceeding\n // the alphabet size are refused. Therefore, to reliably generate the ID,\n // the random bytes redundancy has to be satisfied.\n\n // Note: every hardware random generator call is performance expensive,\n // because the system call for entropy collection takes a lot of time.\n // So, to avoid additional system calls, extra bytes are requested in advance.\n\n // Next, a step determines how many random bytes to generate.\n // The number of random bytes gets decided upon the ID size, mask,\n // alphabet size, and magic number 1.6 (using 1.6 peaks at performance\n // according to benchmarks).\n let step = Math.ceil((1.6 * mask * defaultSize) / alphabet.length)\n\n return (size = defaultSize) => {\n let id = ''\n while (true) {\n let bytes = getRandom(step)\n // A compact alternative for `for (let i = 0; i < step; i++)`.\n let i = step\n while (i--) {\n // Adding `|| ''` refuses a random byte that exceeds the alphabet size.\n id += alphabet[bytes[i] & mask] || ''\n if (id.length === size) return id\n }\n }\n }\n}\n\nlet customAlphabet = (alphabet, size = 21) =>\n customRandom(alphabet, size, random)\n\nlet nanoid = (size = 21) => {\n // `|=` convert `size` to number to prevent `valueOf` abusing and pool pollution\n fillPool((size |= 0))\n let id = ''\n // We are reading directly from the random pool to avoid creating new array\n for (let i = poolOffset - size; i < poolOffset; i++) {\n // It is incorrect to use bytes exceeding the alphabet size.\n // The following mask reduces the random byte in the 0-255 value\n // range to the 0-63 value range. Therefore, adding hacks, such\n // as empty string fallback or magic numbers, is unneccessary because\n // the bitmask trims bytes down to the alphabet size.\n id += urlAlphabet[pool[i] & 63]\n }\n return id\n}\n\nexport { nanoid, customAlphabet, customRandom, urlAlphabet, random }\n","// cspell:ignore Streamable\n\nimport {\n type ReadableStream,\n TransformStream,\n TransformStreamDefaultController,\n} from 'stream/web'\n\nimport {\n type LanguageModelV1,\n type LanguageModelV1CallWarning,\n type LanguageModelV1FinishReason,\n type LanguageModelV1FunctionTool,\n type LanguageModelV1FunctionToolCall,\n type LanguageModelV1Prompt,\n type LanguageModelV1StreamPart,\n} from '@ai-sdk/provider'\nimport type {\n AxAgentic,\n AxAIService,\n AxChatRequest,\n AxChatResponse,\n AxChatResponseResult,\n AxFunction,\n AxFunctionJSONSchema,\n AxGenIn,\n AxGenOut,\n} from '@ax-llm/ax/index.js'\nimport type { CoreMessage } from 'ai'\nimport { customAlphabet } from 'nanoid'\nimport type { ReactNode } from 'react'\nimport { z } from 'zod'\n\ntype Writeable<T> = { -readonly [P in keyof T]: T[P] }\ntype AxChatRequestChatPrompt = Writeable<AxChatRequest['chatPrompt'][0]>\n\ntype AxConfig = {\n fetch?: typeof fetch\n}\n\ntype Streamable = ReactNode | Promise<ReactNode>\ntype Renderer<T> = (\n args: T\n) =>\n | Streamable\n | Generator<Streamable, Streamable, void>\n | AsyncGenerator<Streamable, Streamable, void>\n\nconst nanoid = customAlphabet(\n '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz',\n 7\n)\n\nexport class AxAgentProvider<IN extends AxGenIn, OUT extends AxGenOut> {\n private readonly config?: AxConfig\n private readonly funcInfo: AxFunction\n private generateFunction: Renderer<OUT>\n private updateState: (msgs: readonly CoreMessage[]) => void\n\n constructor({\n agent,\n updateState,\n generate,\n config,\n }: Readonly<{\n agent: AxAgentic\n updateState: (msgs: readonly CoreMessage[]) => void\n generate: Renderer<OUT>\n config?: Readonly<AxConfig>\n }>) {\n this.funcInfo = agent.getFunction()\n this.generateFunction = generate\n this.updateState = updateState\n this.config = config\n }\n\n get description() {\n return this.funcInfo.description\n }\n\n get parameters(): z.ZodTypeAny {\n const schema = this.funcInfo.parameters ?? {\n type: 'object',\n properties: {},\n }\n\n return convertToZodSchema(schema)\n }\n\n get generate(): Renderer<IN> {\n const fn = async (input: IN) => {\n const res = (await this.funcInfo.func(input)) as OUT\n const toolCallId = nanoid()\n\n this.updateState([\n {\n role: 'assistant',\n content: [\n {\n type: 'tool-call',\n toolName: this.funcInfo.name,\n toolCallId,\n args: input,\n },\n ],\n },\n {\n role: 'tool',\n content: [\n {\n type: 'tool-result',\n toolName: this.funcInfo.name,\n toolCallId,\n result: res,\n },\n ],\n },\n ])\n\n return this.generateFunction(res)\n }\n return fn as Renderer<IN>\n }\n}\n\nexport class AxAIProvider implements LanguageModelV1 {\n readonly specificationVersion = 'v1'\n readonly defaultObjectGenerationMode = 'json'\n\n private readonly ai: AxAIService\n private readonly config?: AxConfig\n\n public modelId: string\n\n constructor(ai: AxAIService, config?: Readonly<AxConfig>) {\n this.ai = ai\n this.config = config\n this.modelId = this.ai.getModelInfo().name\n }\n\n get provider(): string {\n return this.ai.getName()\n }\n\n async doGenerate(\n options: Readonly<Parameters<LanguageModelV1['doGenerate']>[0]>\n ): Promise<Awaited<ReturnType<LanguageModelV1['doGenerate']>>> {\n const { req, warnings } = createChatRequest(options)\n const res = (await this.ai.chat(req)) as AxChatResponse\n const choice = res.results.at(0)\n\n if (!choice) {\n throw new Error('No choice returned')\n }\n\n return {\n text: choice.content ?? undefined,\n toolCalls: choice.functionCalls?.map((tc) => ({\n toolCallType: 'function',\n toolCallId: tc.id,\n toolName: tc.function.name,\n args:\n typeof tc.function.params === 'string'\n ? tc.function.params\n : JSON.stringify(tc.function.params),\n })),\n finishReason: mapAxFinishReason(choice.finishReason),\n usage: {\n promptTokens: res.modelUsage?.promptTokens ?? 0,\n completionTokens: res.modelUsage?.completionTokens ?? 0,\n },\n rawCall: { rawPrompt: '', rawSettings: req.modelConfig ?? {} },\n warnings,\n }\n }\n\n async doStream(\n options: Readonly<Parameters<LanguageModelV1['doStream']>[0]>\n ): Promise<Awaited<ReturnType<LanguageModelV1['doStream']>>> {\n const { req, warnings } = createChatRequest(options)\n\n const res = (await this.ai.chat(req, {\n stream: true,\n })) as ReadableStream<AxChatResponse>\n\n return {\n stream: res.pipeThrough(new AxToSDKTransformer()),\n rawCall: { rawPrompt: '', rawSettings: req.modelConfig ?? {} },\n warnings,\n }\n }\n}\n\nfunction prepareToolsAndToolChoice(\n mode: Readonly<\n Parameters<LanguageModelV1['doGenerate']>[0]['mode'] & { type: 'regular' }\n >\n): Pick<AxChatRequest, 'functions' | 'functionCall'> {\n // when the tools array is empty, change it to undefined to prevent errors:\n if (!mode.tools || mode.tools.length === 0) {\n return {}\n }\n\n const tools = mode.tools as Array<LanguageModelV1FunctionTool>\n const functions = tools.map((f) => ({\n name: f.name,\n description: 'description' in f ? (f.description ?? '') : '',\n parameters: f.parameters as AxFunctionJSONSchema,\n }))\n\n const toolChoice = mode.toolChoice\n if (!toolChoice) {\n return { functions }\n }\n\n const type = toolChoice.type\n\n switch (type) {\n case 'auto':\n return { functions, functionCall: 'auto' }\n case 'none':\n return { functions, functionCall: 'none' }\n case 'required':\n return { functions, functionCall: 'required' }\n case 'tool':\n return {\n functions,\n functionCall: {\n type: 'function',\n function: { name: toolChoice.toolName },\n },\n }\n default: {\n const _exhaustiveCheck: never = type\n throw new Error(`Unsupported tool choice type: ${_exhaustiveCheck}`)\n }\n }\n}\n\nfunction convertToAxChatPrompt(\n prompt: Readonly<LanguageModelV1Prompt>\n): AxChatRequest['chatPrompt'] {\n const messages: AxChatRequest['chatPrompt'] = []\n\n for (const { role, content } of prompt) {\n switch (role) {\n case 'system': {\n messages.push({ role: 'system', content })\n break\n }\n\n case 'user': {\n messages.push({\n role: 'user',\n content: content.map((part) => {\n switch (part.type) {\n case 'text': {\n return { type: 'text', text: part.text }\n }\n case 'image': {\n if (!part.mimeType) {\n throw new Error('Image part must have a mimeType')\n }\n if (!ArrayBuffer.isView(part.image)) {\n throw new Error('Image part must have an ArrayBuffer')\n }\n const image = Buffer.from(part.image).toString('base64')\n return {\n type: 'image',\n mimeType: part.mimeType,\n image,\n }\n }\n default:\n throw new Error(`Unsupported part: ${part}`)\n // case 'audio': {\n // if (!part.data) {\n // throw new Error('Audio part must have a audio');\n // }\n // if (!ArrayBuffer.isView(part.data)) {\n // throw new Error('Audio part must have an ArrayBuffer');\n // }\n // const data = Buffer.from(part.data).toString('base64');\n // return {\n // type: 'audio',\n // format: 'wav',\n // data\n // };\n // }\n }\n }),\n })\n break\n }\n\n case 'assistant': {\n let text = ''\n const toolCalls: Extract<\n AxChatRequestChatPrompt,\n { role: 'assistant' }\n >['functionCalls'] = []\n\n for (const part of content) {\n switch (part.type) {\n case 'text': {\n text += part.text\n break\n }\n case 'tool-call': {\n toolCalls.push({\n id: part.toolCallId,\n type: 'function',\n function: {\n name: part.toolName,\n params: part.args as Record<string, unknown>,\n },\n })\n break\n }\n\n default: {\n const _exhaustiveCheck: never = part\n throw new Error(`Unsupported part: ${_exhaustiveCheck}`)\n }\n }\n }\n\n const functionCalls = toolCalls.length === 0 ? undefined : toolCalls\n\n if (functionCalls || text.length > 0) {\n messages.push({\n role: 'assistant',\n content: text,\n functionCalls,\n })\n }\n\n break\n }\n case 'tool': {\n for (const part of content) {\n messages.push({\n role: 'function' as const,\n functionId: part.toolCallId,\n result: JSON.stringify(part.result, null, 2),\n })\n }\n break\n }\n default: {\n const _exhaustiveCheck: never = role\n throw new Error(`Unsupported role: ${_exhaustiveCheck}`)\n }\n }\n }\n\n return messages\n}\n\nfunction mapAxFinishReason(\n finishReason: AxChatResponseResult['finishReason']\n): LanguageModelV1FinishReason {\n switch (finishReason) {\n case 'stop':\n return 'stop'\n case 'length':\n return 'length'\n case 'function_call':\n return 'tool-calls'\n default:\n return 'other'\n }\n}\n\nfunction createChatRequest({\n mode,\n prompt,\n maxTokens,\n temperature,\n topP,\n frequencyPenalty,\n presencePenalty,\n //seed,\n}: Readonly<Parameters<LanguageModelV1['doGenerate']>[0]>): {\n req: AxChatRequest\n warnings: LanguageModelV1CallWarning[]\n} {\n const req: AxChatRequest = {\n chatPrompt: convertToAxChatPrompt(prompt),\n ...(frequencyPenalty != null ? { frequencyPenalty } : {}),\n ...(presencePenalty != null ? { presencePenalty } : {}),\n ...(maxTokens != null ? { maxTokens } : {}),\n ...(temperature != null ? { temperature } : {}),\n ...(topP != null ? { topP } : {}),\n }\n\n const warnings: LanguageModelV1CallWarning[] = []\n\n switch (mode.type) {\n case 'regular': {\n return {\n req: { ...req, ...prepareToolsAndToolChoice(mode) },\n warnings,\n }\n }\n\n case 'object-json': {\n return {\n req,\n warnings,\n }\n }\n\n case 'object-tool': {\n const tool = {\n type: 'function',\n function: {\n name: mode.tool.name,\n params: mode.tool.parameters,\n },\n }\n return {\n req: { ...req, ...tool },\n warnings,\n }\n }\n\n default: {\n throw new Error(`Unsupported type`)\n }\n }\n}\n\nclass AxToSDKTransformer extends TransformStream<\n AxChatResponse,\n LanguageModelV1StreamPart\n> {\n private usage: Extract<\n LanguageModelV1StreamPart,\n { type: 'finish' }\n >['usage'] = {\n promptTokens: 0,\n completionTokens: 0,\n }\n\n private finishReason: Extract<\n LanguageModelV1StreamPart,\n { type: 'finish' }\n >['finishReason'] = 'other'\n\n private functionCalls: LanguageModelV1FunctionToolCall[] = []\n\n constructor() {\n const transformer = {\n transform: (\n chunk: Readonly<AxChatResponse>,\n controller: TransformStreamDefaultController<LanguageModelV1StreamPart>\n ) => {\n const choice = chunk.results.at(0)\n if (!choice) {\n const val = {\n type: 'finish' as const,\n finishReason: this.finishReason,\n usage: this.usage,\n }\n controller.enqueue(val)\n return\n }\n\n if (chunk.modelUsage) {\n this.usage = {\n promptTokens:\n this.usage.promptTokens + chunk.modelUsage.promptTokens,\n completionTokens:\n this.usage.completionTokens + chunk.modelUsage?.completionTokens,\n }\n }\n\n if (choice.functionCalls) {\n for (const fc of choice.functionCalls) {\n const index = this.functionCalls.findIndex(\n (f) => f.toolCallId === fc.id\n )\n if (index === -1) {\n this.functionCalls.push({\n toolCallType: 'function' as const,\n toolCallId: fc.id,\n toolName: fc.function.name,\n args:\n typeof fc.function.params === 'string'\n ? fc.function.params\n : JSON.stringify(fc.function.params),\n })\n } else {\n const obj = this.functionCalls[index]\n if (!obj) {\n continue\n }\n if (typeof fc.function.params === 'string') {\n obj.args = (obj.args ?? '') + fc.function.params\n } else {\n obj.args = JSON.stringify(fc.function.params)\n }\n }\n this.finishReason = 'tool-calls'\n }\n }\n\n if (choice.content && choice.content.length > 0) {\n controller.enqueue({\n type: 'text-delta',\n textDelta: choice.content ?? '',\n })\n this.finishReason = mapAxFinishReason(choice.finishReason)\n }\n },\n flush: (\n controller: TransformStreamDefaultController<LanguageModelV1StreamPart>\n ) => {\n for (const fc of this.functionCalls) {\n const tc = {\n type: 'tool-call' as const,\n ...fc,\n }\n controller.enqueue(tc)\n }\n\n const val = {\n type: 'finish' as const,\n finishReason: this.finishReason,\n usage: this.usage,\n }\n controller.enqueue(val)\n controller.terminate()\n },\n }\n\n super(transformer)\n }\n}\n\ntype AnyZod =\n | z.AnyZodObject\n | z.ZodString\n | z.ZodNumber\n | z.ZodBoolean\n | z.ZodArray<AnyZod>\n | z.ZodOptional<AnyZod>\n\nfunction convertToZodSchema(\n jsonSchema: Readonly<AxFunctionJSONSchema>\n): AnyZod {\n const { type, properties, required, items } = jsonSchema\n\n switch (type) {\n case 'string':\n return z.string()\n case 'number':\n return z.number()\n case 'boolean':\n return z.boolean()\n case 'array':\n if (!items) {\n throw new Error(\"Array type must have 'items' property.\")\n }\n return z.array(convertToZodSchema(items))\n case 'object': {\n if (!properties) {\n throw new Error(\"Object type must have 'properties' property.\")\n }\n const shape: Record<string, AnyZod> = {}\n\n for (const [key, value] of Object.entries(properties)) {\n const schema = convertToZodSchema(value)\n let val = required?.includes(key) ? schema : schema.optional()\n val = value.description ? val.describe(value.description) : val\n shape[key] = val\n }\n return z.object(shape)\n }\n default:\n throw new Error(`Unsupported type: ${type}`)\n }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,oBAAmB;AASnB,IAAM,uBAAuB;AAC7B,IAAI;AAAJ,IAAU;AAEV,IAAI,WAAW,WAAS;AACtB,MAAI,CAAC,QAAQ,KAAK,SAAS,OAAO;AAChC,WAAO,OAAO,YAAY,QAAQ,oBAAoB;AACtD,kBAAAA,QAAO,eAAe,IAAI;AAC1B,iBAAa;AAAA,EACf,WAAW,aAAa,QAAQ,KAAK,QAAQ;AAC3C,kBAAAA,QAAO,eAAe,IAAI;AAC1B,iBAAa;AAAA,EACf;AACA,gBAAc;AAChB;AAEA,IAAI,SAAS,WAAS;AAEpB,WAAU,SAAS,CAAE;AACrB,SAAO,KAAK,SAAS,aAAa,OAAO,UAAU;AACrD;AAEA,IAAI,eAAe,CAAC,UAAU,aAAa,cAAc;AAKvD,MAAI,QAAQ,KAAM,KAAK,KAAK,MAAO,SAAS,SAAS,IAAK,CAAC,KAAM;AAajE,MAAI,OAAO,KAAK,KAAM,MAAM,OAAO,cAAe,SAAS,MAAM;AAEjE,SAAO,CAAC,OAAO,gBAAgB;AAC7B,QAAI,KAAK;AACT,WAAO,MAAM;AACX,UAAI,QAAQ,UAAU,IAAI;AAE1B,UAAI,IAAI;AACR,aAAO,KAAK;AAEV,cAAM,SAAS,MAAM,CAAC,IAAI,IAAI,KAAK;AACnC,YAAI,GAAG,WAAW,KAAM,QAAO;AAAA,MACjC;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAI,iBAAiB,CAAC,UAAU,OAAO,OACrC,aAAa,UAAU,MAAM,MAAM;;;ACnCrC,iBAAkB;AAiBlB,IAAM,SAAS;AAAA,EACb;AAAA,EACA;AACF;AAEO,IAAM,kBAAN,MAAgE;AAAA,EACpD;AAAA,EACA;AAAA,EACT;AAAA,EACA;AAAA,EAER,YAAY;AAAA,IACV;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAKI;AACF,SAAK,WAAW,MAAM,YAAY;AAClC,SAAK,mBAAmB;AACxB,SAAK,cAAc;AACnB,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,IAAI,cAAc;AAChB,WAAO,KAAK,SAAS;AAAA,EACvB;AAAA,EAEA,IAAI,aAA2B;AAC7B,UAAM,SAAS,KAAK,SAAS,cAAc;AAAA,MACzC,MAAM;AAAA,MACN,YAAY,CAAC;AAAA,IACf;AAEA,WAAO,mBAAmB,MAAM;AAAA,EAClC;AAAA,EAEA,IAAI,WAAyB;AAC3B,UAAM,KAAK,OAAO,UAAc;AAC9B,YAAM,MAAO,MAAM,KAAK,SAAS,KAAK,KAAK;AAC3C,YAAM,aAAa,OAAO;AAE1B,WAAK,YAAY;AAAA,QACf;AAAA,UACE,MAAM;AAAA,UACN,SAAS;AAAA,YACP;AAAA,cACE,MAAM;AAAA,cACN,UAAU,KAAK,SAAS;AAAA,cACxB;AAAA,cACA,MAAM;AAAA,YACR;AAAA,UACF;AAAA,QACF;AAAA,QACA;AAAA,UACE,MAAM;AAAA,UACN,SAAS;AAAA,YACP;AAAA,cACE,MAAM;AAAA,cACN,UAAU,KAAK,SAAS;AAAA,cACxB;AAAA,cACA,QAAQ;AAAA,YACV;AAAA,UACF;AAAA,QACF;AAAA,MACF,CAAC;AAED,aAAO,KAAK,iBAAiB,GAAG;AAAA,IAClC;AACA,WAAO;AAAA,EACT;AACF;AA0aA,SAAS,mBACP,YACQ;AACR,QAAM,EAAE,MAAM,YAAY,UAAU,MAAM,IAAI;AAE9C,UAAQ,MAAM;AAAA,IACZ,KAAK;AACH,aAAO,aAAE,OAAO;AAAA,IAClB,KAAK;AACH,aAAO,aAAE,OAAO;AAAA,IAClB,KAAK;AACH,aAAO,aAAE,QAAQ;AAAA,IACnB,KAAK;AACH,UAAI,CAAC,OAAO;AACV,cAAM,IAAI,MAAM,wCAAwC;AAAA,MAC1D;AACA,aAAO,aAAE,MAAM,mBAAmB,KAAK,CAAC;AAAA,IAC1C,KAAK,UAAU;AACb,UAAI,CAAC,YAAY;AACf,cAAM,IAAI,MAAM,8CAA8C;AAAA,MAChE;AACA,YAAM,QAAgC,CAAC;AAEvC,iBAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,UAAU,GAAG;AACrD,cAAM,SAAS,mBAAmB,KAAK;AACvC,YAAI,MAAM,UAAU,SAAS,GAAG,IAAI,SAAS,OAAO,SAAS;AAC7D,cAAM,MAAM,cAAc,IAAI,SAAS,MAAM,WAAW,IAAI;AAC5D,cAAM,GAAG,IAAI;AAAA,MACf;AACA,aAAO,aAAE,OAAO,KAAK;AAAA,IACvB;AAAA,IACA;AACE,YAAM,IAAI,MAAM,qBAAqB,IAAI,EAAE;AAAA,EAC/C;AACF;","names":["crypto"]}
1
+ {"version":3,"sources":["../index.ts","../provider.ts","../../../node_modules/nanoid/index.js"],"sourcesContent":["import { AxAgentProvider } from './provider.js'\nimport { AxAIProvider } from './provider.js'\n\nexport { AxAgentProvider, AxAIProvider }\n","// cspell:ignore Streamable\n\nimport {\n type ReadableStream,\n TransformStream,\n TransformStreamDefaultController,\n} from 'stream/web'\n\nimport {\n type LanguageModelV1,\n type LanguageModelV1CallWarning,\n type LanguageModelV1FinishReason,\n type LanguageModelV1FunctionTool,\n type LanguageModelV1FunctionToolCall,\n type LanguageModelV1Prompt,\n type LanguageModelV1StreamPart,\n} from '@ai-sdk/provider'\nimport type {\n AxAgentic,\n AxAIService,\n AxChatRequest,\n AxChatResponse,\n AxChatResponseResult,\n AxFunction,\n AxFunctionJSONSchema,\n AxGenIn,\n AxGenOut,\n} from '@ax-llm/ax/index.js'\nimport type { CoreMessage } from 'ai'\nimport { customAlphabet } from 'nanoid'\nimport type { ReactNode } from 'react'\nimport { z } from 'zod'\n\ntype Writeable<T> = { -readonly [P in keyof T]: T[P] }\ntype AxChatRequestChatPrompt = Writeable<AxChatRequest['chatPrompt'][0]>\n\ntype AxConfig = {\n fetch?: typeof fetch\n}\n\ntype Streamable = ReactNode | Promise<ReactNode>\ntype Renderer<T> = (\n args: T\n) =>\n | Streamable\n | Generator<Streamable, Streamable, void>\n | AsyncGenerator<Streamable, Streamable, void>\n\nconst nanoid = customAlphabet(\n '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz',\n 7\n)\n\nexport class AxAgentProvider<IN extends AxGenIn, OUT extends AxGenOut> {\n private readonly config?: AxConfig\n private readonly funcInfo: AxFunction\n private generateFunction: Renderer<OUT>\n private updateState: (msgs: readonly CoreMessage[]) => void\n\n constructor({\n agent,\n updateState,\n generate,\n config,\n }: Readonly<{\n agent: AxAgentic\n updateState: (msgs: readonly CoreMessage[]) => void\n generate: Renderer<OUT>\n config?: Readonly<AxConfig>\n }>) {\n this.funcInfo = agent.getFunction()\n this.generateFunction = generate\n this.updateState = updateState\n this.config = config\n }\n\n get description() {\n return this.funcInfo.description\n }\n\n get parameters(): z.ZodTypeAny {\n const schema = this.funcInfo.parameters ?? {\n type: 'object',\n properties: {},\n }\n\n return convertToZodSchema(schema)\n }\n\n get generate(): Renderer<IN> {\n const fn = async (input: IN) => {\n const res = (await this.funcInfo.func(input)) as OUT\n const toolCallId = nanoid()\n\n this.updateState([\n {\n role: 'assistant',\n content: [\n {\n type: 'tool-call',\n toolName: this.funcInfo.name,\n toolCallId,\n args: input,\n },\n ],\n },\n {\n role: 'tool',\n content: [\n {\n type: 'tool-result',\n toolName: this.funcInfo.name,\n toolCallId,\n result: res,\n },\n ],\n },\n ])\n\n return this.generateFunction(res)\n }\n return fn as Renderer<IN>\n }\n}\n\nexport class AxAIProvider implements LanguageModelV1 {\n readonly specificationVersion = 'v1'\n readonly defaultObjectGenerationMode = 'json'\n\n private readonly ai: AxAIService\n private readonly config?: AxConfig\n\n public modelId: string\n\n constructor(ai: AxAIService, config?: Readonly<AxConfig>) {\n this.ai = ai\n this.config = config\n this.modelId = this.ai.getModelInfo().name\n }\n\n get provider(): string {\n return this.ai.getName()\n }\n\n async doGenerate(\n options: Readonly<Parameters<LanguageModelV1['doGenerate']>[0]>\n ): Promise<Awaited<ReturnType<LanguageModelV1['doGenerate']>>> {\n const { req, warnings } = createChatRequest(options)\n const res = (await this.ai.chat(req)) as AxChatResponse\n const choice = res.results.at(0)\n\n if (!choice) {\n throw new Error('No choice returned')\n }\n\n return {\n text: choice.content ?? undefined,\n toolCalls: choice.functionCalls?.map((tc) => ({\n toolCallType: 'function',\n toolCallId: tc.id,\n toolName: tc.function.name,\n args:\n typeof tc.function.params === 'string'\n ? tc.function.params\n : JSON.stringify(tc.function.params),\n })),\n finishReason: mapAxFinishReason(choice.finishReason),\n usage: {\n promptTokens: res.modelUsage?.promptTokens ?? 0,\n completionTokens: res.modelUsage?.completionTokens ?? 0,\n },\n rawCall: { rawPrompt: '', rawSettings: req.modelConfig ?? {} },\n warnings,\n }\n }\n\n async doStream(\n options: Readonly<Parameters<LanguageModelV1['doStream']>[0]>\n ): Promise<Awaited<ReturnType<LanguageModelV1['doStream']>>> {\n const { req, warnings } = createChatRequest(options)\n\n const res = (await this.ai.chat(req, {\n stream: true,\n })) as ReadableStream<AxChatResponse>\n\n return {\n stream: res.pipeThrough(new AxToSDKTransformer()),\n rawCall: { rawPrompt: '', rawSettings: req.modelConfig ?? {} },\n warnings,\n }\n }\n}\n\nfunction prepareToolsAndToolChoice(\n mode: Readonly<\n Parameters<LanguageModelV1['doGenerate']>[0]['mode'] & { type: 'regular' }\n >\n): Pick<AxChatRequest, 'functions' | 'functionCall'> {\n // when the tools array is empty, change it to undefined to prevent errors:\n if (!mode.tools || mode.tools.length === 0) {\n return {}\n }\n\n const tools = mode.tools as Array<LanguageModelV1FunctionTool>\n const functions = tools.map((f) => ({\n name: f.name,\n description: 'description' in f ? (f.description ?? '') : '',\n parameters: f.parameters as AxFunctionJSONSchema,\n }))\n\n const toolChoice = mode.toolChoice\n if (!toolChoice) {\n return { functions }\n }\n\n const type = toolChoice.type\n\n switch (type) {\n case 'auto':\n return { functions, functionCall: 'auto' }\n case 'none':\n return { functions, functionCall: 'none' }\n case 'required':\n return { functions, functionCall: 'required' }\n case 'tool':\n return {\n functions,\n functionCall: {\n type: 'function',\n function: { name: toolChoice.toolName },\n },\n }\n default: {\n const _exhaustiveCheck: never = type\n throw new Error(`Unsupported tool choice type: ${_exhaustiveCheck}`)\n }\n }\n}\n\nfunction convertToAxChatPrompt(\n prompt: Readonly<LanguageModelV1Prompt>\n): AxChatRequest['chatPrompt'] {\n const messages: AxChatRequest['chatPrompt'] = []\n\n for (const { role, content } of prompt) {\n switch (role) {\n case 'system': {\n messages.push({ role: 'system', content })\n break\n }\n\n case 'user': {\n messages.push({\n role: 'user',\n content: content.map((part) => {\n switch (part.type) {\n case 'text': {\n return { type: 'text', text: part.text }\n }\n case 'image': {\n if (!part.mimeType) {\n throw new Error('Image part must have a mimeType')\n }\n if (!ArrayBuffer.isView(part.image)) {\n throw new Error('Image part must have an ArrayBuffer')\n }\n const image = Buffer.from(part.image).toString('base64')\n return {\n type: 'image',\n mimeType: part.mimeType,\n image,\n }\n }\n default:\n throw new Error(`Unsupported part: ${part}`)\n // case 'audio': {\n // if (!part.data) {\n // throw new Error('Audio part must have a audio');\n // }\n // if (!ArrayBuffer.isView(part.data)) {\n // throw new Error('Audio part must have an ArrayBuffer');\n // }\n // const data = Buffer.from(part.data).toString('base64');\n // return {\n // type: 'audio',\n // format: 'wav',\n // data\n // };\n // }\n }\n }),\n })\n break\n }\n\n case 'assistant': {\n let text = ''\n const toolCalls: Extract<\n AxChatRequestChatPrompt,\n { role: 'assistant' }\n >['functionCalls'] = []\n\n for (const part of content) {\n switch (part.type) {\n case 'text': {\n text += part.text\n break\n }\n case 'tool-call': {\n toolCalls.push({\n id: part.toolCallId,\n type: 'function',\n function: {\n name: part.toolName,\n params: part.args as Record<string, unknown>,\n },\n })\n break\n }\n\n default: {\n const _exhaustiveCheck: never = part\n throw new Error(`Unsupported part: ${_exhaustiveCheck}`)\n }\n }\n }\n\n const functionCalls = toolCalls.length === 0 ? undefined : toolCalls\n\n if (functionCalls || text.length > 0) {\n messages.push({\n role: 'assistant',\n content: text,\n functionCalls,\n })\n }\n\n break\n }\n case 'tool': {\n for (const part of content) {\n messages.push({\n role: 'function' as const,\n functionId: part.toolCallId,\n result: JSON.stringify(part.result, null, 2),\n })\n }\n break\n }\n default: {\n const _exhaustiveCheck: never = role\n throw new Error(`Unsupported role: ${_exhaustiveCheck}`)\n }\n }\n }\n\n return messages\n}\n\nfunction mapAxFinishReason(\n finishReason: AxChatResponseResult['finishReason']\n): LanguageModelV1FinishReason {\n switch (finishReason) {\n case 'stop':\n return 'stop'\n case 'length':\n return 'length'\n case 'function_call':\n return 'tool-calls'\n default:\n return 'other'\n }\n}\n\nfunction createChatRequest({\n mode,\n prompt,\n maxTokens,\n temperature,\n topP,\n frequencyPenalty,\n presencePenalty,\n //seed,\n}: Readonly<Parameters<LanguageModelV1['doGenerate']>[0]>): {\n req: AxChatRequest\n warnings: LanguageModelV1CallWarning[]\n} {\n const req: AxChatRequest = {\n chatPrompt: convertToAxChatPrompt(prompt),\n ...(frequencyPenalty != null ? { frequencyPenalty } : {}),\n ...(presencePenalty != null ? { presencePenalty } : {}),\n ...(maxTokens != null ? { maxTokens } : {}),\n ...(temperature != null ? { temperature } : {}),\n ...(topP != null ? { topP } : {}),\n }\n\n const warnings: LanguageModelV1CallWarning[] = []\n\n switch (mode.type) {\n case 'regular': {\n return {\n req: { ...req, ...prepareToolsAndToolChoice(mode) },\n warnings,\n }\n }\n\n case 'object-json': {\n return {\n req,\n warnings,\n }\n }\n\n case 'object-tool': {\n const tool = {\n type: 'function',\n function: {\n name: mode.tool.name,\n params: mode.tool.parameters,\n },\n }\n return {\n req: { ...req, ...tool },\n warnings,\n }\n }\n\n default: {\n throw new Error(`Unsupported type`)\n }\n }\n}\n\nclass AxToSDKTransformer extends TransformStream<\n AxChatResponse,\n LanguageModelV1StreamPart\n> {\n private usage: Extract<\n LanguageModelV1StreamPart,\n { type: 'finish' }\n >['usage'] = {\n promptTokens: 0,\n completionTokens: 0,\n }\n\n private finishReason: Extract<\n LanguageModelV1StreamPart,\n { type: 'finish' }\n >['finishReason'] = 'other'\n\n private functionCalls: LanguageModelV1FunctionToolCall[] = []\n\n constructor() {\n const transformer = {\n transform: (\n chunk: Readonly<AxChatResponse>,\n controller: TransformStreamDefaultController<LanguageModelV1StreamPart>\n ) => {\n const choice = chunk.results.at(0)\n if (!choice) {\n const val = {\n type: 'finish' as const,\n finishReason: this.finishReason,\n usage: this.usage,\n }\n controller.enqueue(val)\n return\n }\n\n if (chunk.modelUsage) {\n this.usage = {\n promptTokens:\n this.usage.promptTokens + chunk.modelUsage.promptTokens,\n completionTokens:\n this.usage.completionTokens + chunk.modelUsage?.completionTokens,\n }\n }\n\n if (choice.functionCalls) {\n for (const fc of choice.functionCalls) {\n const index = this.functionCalls.findIndex(\n (f) => f.toolCallId === fc.id\n )\n if (index === -1) {\n this.functionCalls.push({\n toolCallType: 'function' as const,\n toolCallId: fc.id,\n toolName: fc.function.name,\n args:\n typeof fc.function.params === 'string'\n ? fc.function.params\n : JSON.stringify(fc.function.params),\n })\n } else {\n const obj = this.functionCalls[index]\n if (!obj) {\n continue\n }\n if (typeof fc.function.params === 'string') {\n obj.args = (obj.args ?? '') + fc.function.params\n } else {\n obj.args = JSON.stringify(fc.function.params)\n }\n }\n this.finishReason = 'tool-calls'\n }\n }\n\n if (choice.content && choice.content.length > 0) {\n controller.enqueue({\n type: 'text-delta',\n textDelta: choice.content ?? '',\n })\n this.finishReason = mapAxFinishReason(choice.finishReason)\n }\n },\n flush: (\n controller: TransformStreamDefaultController<LanguageModelV1StreamPart>\n ) => {\n for (const fc of this.functionCalls) {\n const tc = {\n type: 'tool-call' as const,\n ...fc,\n }\n controller.enqueue(tc)\n }\n\n const val = {\n type: 'finish' as const,\n finishReason: this.finishReason,\n usage: this.usage,\n }\n controller.enqueue(val)\n controller.terminate()\n },\n }\n\n super(transformer)\n }\n}\n\ntype AnyZod =\n | z.AnyZodObject\n | z.ZodString\n | z.ZodNumber\n | z.ZodBoolean\n | z.ZodArray<AnyZod>\n | z.ZodOptional<AnyZod>\n\nfunction convertToZodSchema(\n jsonSchema: Readonly<AxFunctionJSONSchema>\n): AnyZod {\n const { type, properties, required, items } = jsonSchema\n\n switch (type) {\n case 'string':\n return z.string()\n case 'number':\n return z.number()\n case 'boolean':\n return z.boolean()\n case 'array':\n if (!items) {\n throw new Error(\"Array type must have 'items' property.\")\n }\n return z.array(convertToZodSchema(items))\n case 'object': {\n if (!properties) {\n throw new Error(\"Object type must have 'properties' property.\")\n }\n const shape: Record<string, AnyZod> = {}\n\n for (const [key, value] of Object.entries(properties)) {\n const schema = convertToZodSchema(value)\n let val = required?.includes(key) ? schema : schema.optional()\n val = value.description ? val.describe(value.description) : val\n shape[key] = val\n }\n return z.object(shape)\n }\n default:\n throw new Error(`Unsupported type: ${type}`)\n }\n}\n","import crypto from 'crypto'\n\nimport { urlAlphabet } from './url-alphabet/index.js'\n\n// It is best to make fewer, larger requests to the crypto module to\n// avoid system call overhead. So, random numbers are generated in a\n// pool. The pool is a Buffer that is larger than the initial random\n// request size by this multiplier. The pool is enlarged if subsequent\n// requests exceed the maximum buffer size.\nconst POOL_SIZE_MULTIPLIER = 128\nlet pool, poolOffset\n\nlet fillPool = bytes => {\n if (!pool || pool.length < bytes) {\n pool = Buffer.allocUnsafe(bytes * POOL_SIZE_MULTIPLIER)\n crypto.randomFillSync(pool)\n poolOffset = 0\n } else if (poolOffset + bytes > pool.length) {\n crypto.randomFillSync(pool)\n poolOffset = 0\n }\n poolOffset += bytes\n}\n\nlet random = bytes => {\n // `|=` convert `bytes` to number to prevent `valueOf` abusing and pool pollution\n fillPool((bytes |= 0))\n return pool.subarray(poolOffset - bytes, poolOffset)\n}\n\nlet customRandom = (alphabet, defaultSize, getRandom) => {\n // First, a bitmask is necessary to generate the ID. The bitmask makes bytes\n // values closer to the alphabet size. The bitmask calculates the closest\n // `2^31 - 1` number, which exceeds the alphabet size.\n // For example, the bitmask for the alphabet size 30 is 31 (00011111).\n let mask = (2 << (31 - Math.clz32((alphabet.length - 1) | 1))) - 1\n // Though, the bitmask solution is not perfect since the bytes exceeding\n // the alphabet size are refused. Therefore, to reliably generate the ID,\n // the random bytes redundancy has to be satisfied.\n\n // Note: every hardware random generator call is performance expensive,\n // because the system call for entropy collection takes a lot of time.\n // So, to avoid additional system calls, extra bytes are requested in advance.\n\n // Next, a step determines how many random bytes to generate.\n // The number of random bytes gets decided upon the ID size, mask,\n // alphabet size, and magic number 1.6 (using 1.6 peaks at performance\n // according to benchmarks).\n let step = Math.ceil((1.6 * mask * defaultSize) / alphabet.length)\n\n return (size = defaultSize) => {\n let id = ''\n while (true) {\n let bytes = getRandom(step)\n // A compact alternative for `for (let i = 0; i < step; i++)`.\n let i = step\n while (i--) {\n // Adding `|| ''` refuses a random byte that exceeds the alphabet size.\n id += alphabet[bytes[i] & mask] || ''\n if (id.length === size) return id\n }\n }\n }\n}\n\nlet customAlphabet = (alphabet, size = 21) =>\n customRandom(alphabet, size, random)\n\nlet nanoid = (size = 21) => {\n // `|=` convert `size` to number to prevent `valueOf` abusing and pool pollution\n fillPool((size |= 0))\n let id = ''\n // We are reading directly from the random pool to avoid creating new array\n for (let i = poolOffset - size; i < poolOffset; i++) {\n // It is incorrect to use bytes exceeding the alphabet size.\n // The following mask reduces the random byte in the 0-255 value\n // range to the 0-63 value range. Therefore, adding hacks, such\n // as empty string fallback or magic numbers, is unneccessary because\n // the bitmask trims bytes down to the alphabet size.\n id += urlAlphabet[pool[i] & 63]\n }\n return id\n}\n\nexport { nanoid, customAlphabet, customRandom, urlAlphabet, random }\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACEA,iBAIO;;;ACNP,oBAAmB;AASnB,IAAM,uBAAuB;AAC7B,IAAI;AAAJ,IAAU;AAEV,IAAI,WAAW,WAAS;AACtB,MAAI,CAAC,QAAQ,KAAK,SAAS,OAAO;AAChC,WAAO,OAAO,YAAY,QAAQ,oBAAoB;AACtD,kBAAAA,QAAO,eAAe,IAAI;AAC1B,iBAAa;AAAA,EACf,WAAW,aAAa,QAAQ,KAAK,QAAQ;AAC3C,kBAAAA,QAAO,eAAe,IAAI;AAC1B,iBAAa;AAAA,EACf;AACA,gBAAc;AAChB;AAEA,IAAI,SAAS,WAAS;AAEpB,WAAU,SAAS,CAAE;AACrB,SAAO,KAAK,SAAS,aAAa,OAAO,UAAU;AACrD;AAEA,IAAI,eAAe,CAAC,UAAU,aAAa,cAAc;AAKvD,MAAI,QAAQ,KAAM,KAAK,KAAK,MAAO,SAAS,SAAS,IAAK,CAAC,KAAM;AAajE,MAAI,OAAO,KAAK,KAAM,MAAM,OAAO,cAAe,SAAS,MAAM;AAEjE,SAAO,CAAC,OAAO,gBAAgB;AAC7B,QAAI,KAAK;AACT,WAAO,MAAM;AACX,UAAI,QAAQ,UAAU,IAAI;AAE1B,UAAI,IAAI;AACR,aAAO,KAAK;AAEV,cAAM,SAAS,MAAM,CAAC,IAAI,IAAI,KAAK;AACnC,YAAI,GAAG,WAAW,KAAM,QAAO;AAAA,MACjC;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAI,iBAAiB,CAAC,UAAU,OAAO,OACrC,aAAa,UAAU,MAAM,MAAM;;;ADnCrC,iBAAkB;AAiBlB,IAAM,SAAS;AAAA,EACb;AAAA,EACA;AACF;AAEO,IAAM,kBAAN,MAAgE;AAAA,EACpD;AAAA,EACA;AAAA,EACT;AAAA,EACA;AAAA,EAER,YAAY;AAAA,IACV;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAKI;AACF,SAAK,WAAW,MAAM,YAAY;AAClC,SAAK,mBAAmB;AACxB,SAAK,cAAc;AACnB,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,IAAI,cAAc;AAChB,WAAO,KAAK,SAAS;AAAA,EACvB;AAAA,EAEA,IAAI,aAA2B;AAC7B,UAAM,SAAS,KAAK,SAAS,cAAc;AAAA,MACzC,MAAM;AAAA,MACN,YAAY,CAAC;AAAA,IACf;AAEA,WAAO,mBAAmB,MAAM;AAAA,EAClC;AAAA,EAEA,IAAI,WAAyB;AAC3B,UAAM,KAAK,OAAO,UAAc;AAC9B,YAAM,MAAO,MAAM,KAAK,SAAS,KAAK,KAAK;AAC3C,YAAM,aAAa,OAAO;AAE1B,WAAK,YAAY;AAAA,QACf;AAAA,UACE,MAAM;AAAA,UACN,SAAS;AAAA,YACP;AAAA,cACE,MAAM;AAAA,cACN,UAAU,KAAK,SAAS;AAAA,cACxB;AAAA,cACA,MAAM;AAAA,YACR;AAAA,UACF;AAAA,QACF;AAAA,QACA;AAAA,UACE,MAAM;AAAA,UACN,SAAS;AAAA,YACP;AAAA,cACE,MAAM;AAAA,cACN,UAAU,KAAK,SAAS;AAAA,cACxB;AAAA,cACA,QAAQ;AAAA,YACV;AAAA,UACF;AAAA,QACF;AAAA,MACF,CAAC;AAED,aAAO,KAAK,iBAAiB,GAAG;AAAA,IAClC;AACA,WAAO;AAAA,EACT;AACF;AAEO,IAAM,eAAN,MAA8C;AAAA,EAC1C,uBAAuB;AAAA,EACvB,8BAA8B;AAAA,EAEtB;AAAA,EACA;AAAA,EAEV;AAAA,EAEP,YAAY,IAAiB,QAA6B;AACxD,SAAK,KAAK;AACV,SAAK,SAAS;AACd,SAAK,UAAU,KAAK,GAAG,aAAa,EAAE;AAAA,EACxC;AAAA,EAEA,IAAI,WAAmB;AACrB,WAAO,KAAK,GAAG,QAAQ;AAAA,EACzB;AAAA,EAEA,MAAM,WACJ,SAC6D;AAC7D,UAAM,EAAE,KAAK,SAAS,IAAI,kBAAkB,OAAO;AACnD,UAAM,MAAO,MAAM,KAAK,GAAG,KAAK,GAAG;AACnC,UAAM,SAAS,IAAI,QAAQ,GAAG,CAAC;AAE/B,QAAI,CAAC,QAAQ;AACX,YAAM,IAAI,MAAM,oBAAoB;AAAA,IACtC;AAEA,WAAO;AAAA,MACL,MAAM,OAAO,WAAW;AAAA,MACxB,WAAW,OAAO,eAAe,IAAI,CAAC,QAAQ;AAAA,QAC5C,cAAc;AAAA,QACd,YAAY,GAAG;AAAA,QACf,UAAU,GAAG,SAAS;AAAA,QACtB,MACE,OAAO,GAAG,SAAS,WAAW,WAC1B,GAAG,SAAS,SACZ,KAAK,UAAU,GAAG,SAAS,MAAM;AAAA,MACzC,EAAE;AAAA,MACF,cAAc,kBAAkB,OAAO,YAAY;AAAA,MACnD,OAAO;AAAA,QACL,cAAc,IAAI,YAAY,gBAAgB;AAAA,QAC9C,kBAAkB,IAAI,YAAY,oBAAoB;AAAA,MACxD;AAAA,MACA,SAAS,EAAE,WAAW,IAAI,aAAa,IAAI,eAAe,CAAC,EAAE;AAAA,MAC7D;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,SACJ,SAC2D;AAC3D,UAAM,EAAE,KAAK,SAAS,IAAI,kBAAkB,OAAO;AAEnD,UAAM,MAAO,MAAM,KAAK,GAAG,KAAK,KAAK;AAAA,MACnC,QAAQ;AAAA,IACV,CAAC;AAED,WAAO;AAAA,MACL,QAAQ,IAAI,YAAY,IAAI,mBAAmB,CAAC;AAAA,MAChD,SAAS,EAAE,WAAW,IAAI,aAAa,IAAI,eAAe,CAAC,EAAE;AAAA,MAC7D;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,0BACP,MAGmD;AAEnD,MAAI,CAAC,KAAK,SAAS,KAAK,MAAM,WAAW,GAAG;AAC1C,WAAO,CAAC;AAAA,EACV;AAEA,QAAM,QAAQ,KAAK;AACnB,QAAM,YAAY,MAAM,IAAI,CAAC,OAAO;AAAA,IAClC,MAAM,EAAE;AAAA,IACR,aAAa,iBAAiB,IAAK,EAAE,eAAe,KAAM;AAAA,IAC1D,YAAY,EAAE;AAAA,EAChB,EAAE;AAEF,QAAM,aAAa,KAAK;AACxB,MAAI,CAAC,YAAY;AACf,WAAO,EAAE,UAAU;AAAA,EACrB;AAEA,QAAM,OAAO,WAAW;AAExB,UAAQ,MAAM;AAAA,IACZ,KAAK;AACH,aAAO,EAAE,WAAW,cAAc,OAAO;AAAA,IAC3C,KAAK;AACH,aAAO,EAAE,WAAW,cAAc,OAAO;AAAA,IAC3C,KAAK;AACH,aAAO,EAAE,WAAW,cAAc,WAAW;AAAA,IAC/C,KAAK;AACH,aAAO;AAAA,QACL;AAAA,QACA,cAAc;AAAA,UACZ,MAAM;AAAA,UACN,UAAU,EAAE,MAAM,WAAW,SAAS;AAAA,QACxC;AAAA,MACF;AAAA,IACF,SAAS;AACP,YAAM,mBAA0B;AAChC,YAAM,IAAI,MAAM,iCAAiC,gBAAgB,EAAE;AAAA,IACrE;AAAA,EACF;AACF;AAEA,SAAS,sBACP,QAC6B;AAC7B,QAAM,WAAwC,CAAC;AAE/C,aAAW,EAAE,MAAM,QAAQ,KAAK,QAAQ;AACtC,YAAQ,MAAM;AAAA,MACZ,KAAK,UAAU;AACb,iBAAS,KAAK,EAAE,MAAM,UAAU,QAAQ,CAAC;AACzC;AAAA,MACF;AAAA,MAEA,KAAK,QAAQ;AACX,iBAAS,KAAK;AAAA,UACZ,MAAM;AAAA,UACN,SAAS,QAAQ,IAAI,CAAC,SAAS;AAC7B,oBAAQ,KAAK,MAAM;AAAA,cACjB,KAAK,QAAQ;AACX,uBAAO,EAAE,MAAM,QAAQ,MAAM,KAAK,KAAK;AAAA,cACzC;AAAA,cACA,KAAK,SAAS;AACZ,oBAAI,CAAC,KAAK,UAAU;AAClB,wBAAM,IAAI,MAAM,iCAAiC;AAAA,gBACnD;AACA,oBAAI,CAAC,YAAY,OAAO,KAAK,KAAK,GAAG;AACnC,wBAAM,IAAI,MAAM,qCAAqC;AAAA,gBACvD;AACA,sBAAM,QAAQ,OAAO,KAAK,KAAK,KAAK,EAAE,SAAS,QAAQ;AACvD,uBAAO;AAAA,kBACL,MAAM;AAAA,kBACN,UAAU,KAAK;AAAA,kBACf;AAAA,gBACF;AAAA,cACF;AAAA,cACA;AACE,sBAAM,IAAI,MAAM,qBAAqB,IAAI,EAAE;AAAA,YAe/C;AAAA,UACF,CAAC;AAAA,QACH,CAAC;AACD;AAAA,MACF;AAAA,MAEA,KAAK,aAAa;AAChB,YAAI,OAAO;AACX,cAAM,YAGe,CAAC;AAEtB,mBAAW,QAAQ,SAAS;AAC1B,kBAAQ,KAAK,MAAM;AAAA,YACjB,KAAK,QAAQ;AACX,sBAAQ,KAAK;AACb;AAAA,YACF;AAAA,YACA,KAAK,aAAa;AAChB,wBAAU,KAAK;AAAA,gBACb,IAAI,KAAK;AAAA,gBACT,MAAM;AAAA,gBACN,UAAU;AAAA,kBACR,MAAM,KAAK;AAAA,kBACX,QAAQ,KAAK;AAAA,gBACf;AAAA,cACF,CAAC;AACD;AAAA,YACF;AAAA,YAEA,SAAS;AACP,oBAAM,mBAA0B;AAChC,oBAAM,IAAI,MAAM,qBAAqB,gBAAgB,EAAE;AAAA,YACzD;AAAA,UACF;AAAA,QACF;AAEA,cAAM,gBAAgB,UAAU,WAAW,IAAI,SAAY;AAE3D,YAAI,iBAAiB,KAAK,SAAS,GAAG;AACpC,mBAAS,KAAK;AAAA,YACZ,MAAM;AAAA,YACN,SAAS;AAAA,YACT;AAAA,UACF,CAAC;AAAA,QACH;AAEA;AAAA,MACF;AAAA,MACA,KAAK,QAAQ;AACX,mBAAW,QAAQ,SAAS;AAC1B,mBAAS,KAAK;AAAA,YACZ,MAAM;AAAA,YACN,YAAY,KAAK;AAAA,YACjB,QAAQ,KAAK,UAAU,KAAK,QAAQ,MAAM,CAAC;AAAA,UAC7C,CAAC;AAAA,QACH;AACA;AAAA,MACF;AAAA,MACA,SAAS;AACP,cAAM,mBAA0B;AAChC,cAAM,IAAI,MAAM,qBAAqB,gBAAgB,EAAE;AAAA,MACzD;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AACT;AAEA,SAAS,kBACP,cAC6B;AAC7B,UAAQ,cAAc;AAAA,IACpB,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AACH,aAAO;AAAA,IACT;AACE,aAAO;AAAA,EACX;AACF;AAEA,SAAS,kBAAkB;AAAA,EACzB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAEF,GAGE;AACA,QAAM,MAAqB;AAAA,IACzB,YAAY,sBAAsB,MAAM;AAAA,IACxC,GAAI,oBAAoB,OAAO,EAAE,iBAAiB,IAAI,CAAC;AAAA,IACvD,GAAI,mBAAmB,OAAO,EAAE,gBAAgB,IAAI,CAAC;AAAA,IACrD,GAAI,aAAa,OAAO,EAAE,UAAU,IAAI,CAAC;AAAA,IACzC,GAAI,eAAe,OAAO,EAAE,YAAY,IAAI,CAAC;AAAA,IAC7C,GAAI,QAAQ,OAAO,EAAE,KAAK,IAAI,CAAC;AAAA,EACjC;AAEA,QAAM,WAAyC,CAAC;AAEhD,UAAQ,KAAK,MAAM;AAAA,IACjB,KAAK,WAAW;AACd,aAAO;AAAA,QACL,KAAK,EAAE,GAAG,KAAK,GAAG,0BAA0B,IAAI,EAAE;AAAA,QAClD;AAAA,MACF;AAAA,IACF;AAAA,IAEA,KAAK,eAAe;AAClB,aAAO;AAAA,QACL;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAAA,IAEA,KAAK,eAAe;AAClB,YAAM,OAAO;AAAA,QACX,MAAM;AAAA,QACN,UAAU;AAAA,UACR,MAAM,KAAK,KAAK;AAAA,UAChB,QAAQ,KAAK,KAAK;AAAA,QACpB;AAAA,MACF;AACA,aAAO;AAAA,QACL,KAAK,EAAE,GAAG,KAAK,GAAG,KAAK;AAAA,QACvB;AAAA,MACF;AAAA,IACF;AAAA,IAEA,SAAS;AACP,YAAM,IAAI,MAAM,kBAAkB;AAAA,IACpC;AAAA,EACF;AACF;AAEA,IAAM,qBAAN,cAAiC,2BAG/B;AAAA,EACQ,QAGK;AAAA,IACX,cAAc;AAAA,IACd,kBAAkB;AAAA,EACpB;AAAA,EAEQ,eAGY;AAAA,EAEZ,gBAAmD,CAAC;AAAA,EAE5D,cAAc;AACZ,UAAM,cAAc;AAAA,MAClB,WAAW,CACT,OACA,eACG;AACH,cAAM,SAAS,MAAM,QAAQ,GAAG,CAAC;AACjC,YAAI,CAAC,QAAQ;AACX,gBAAM,MAAM;AAAA,YACV,MAAM;AAAA,YACN,cAAc,KAAK;AAAA,YACnB,OAAO,KAAK;AAAA,UACd;AACA,qBAAW,QAAQ,GAAG;AACtB;AAAA,QACF;AAEA,YAAI,MAAM,YAAY;AACpB,eAAK,QAAQ;AAAA,YACX,cACE,KAAK,MAAM,eAAe,MAAM,WAAW;AAAA,YAC7C,kBACE,KAAK,MAAM,mBAAmB,MAAM,YAAY;AAAA,UACpD;AAAA,QACF;AAEA,YAAI,OAAO,eAAe;AACxB,qBAAW,MAAM,OAAO,eAAe;AACrC,kBAAM,QAAQ,KAAK,cAAc;AAAA,cAC/B,CAAC,MAAM,EAAE,eAAe,GAAG;AAAA,YAC7B;AACA,gBAAI,UAAU,IAAI;AAChB,mBAAK,cAAc,KAAK;AAAA,gBACtB,cAAc;AAAA,gBACd,YAAY,GAAG;AAAA,gBACf,UAAU,GAAG,SAAS;AAAA,gBACtB,MACE,OAAO,GAAG,SAAS,WAAW,WAC1B,GAAG,SAAS,SACZ,KAAK,UAAU,GAAG,SAAS,MAAM;AAAA,cACzC,CAAC;AAAA,YACH,OAAO;AACL,oBAAM,MAAM,KAAK,cAAc,KAAK;AACpC,kBAAI,CAAC,KAAK;AACR;AAAA,cACF;AACA,kBAAI,OAAO,GAAG,SAAS,WAAW,UAAU;AAC1C,oBAAI,QAAQ,IAAI,QAAQ,MAAM,GAAG,SAAS;AAAA,cAC5C,OAAO;AACL,oBAAI,OAAO,KAAK,UAAU,GAAG,SAAS,MAAM;AAAA,cAC9C;AAAA,YACF;AACA,iBAAK,eAAe;AAAA,UACtB;AAAA,QACF;AAEA,YAAI,OAAO,WAAW,OAAO,QAAQ,SAAS,GAAG;AAC/C,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN,WAAW,OAAO,WAAW;AAAA,UAC/B,CAAC;AACD,eAAK,eAAe,kBAAkB,OAAO,YAAY;AAAA,QAC3D;AAAA,MACF;AAAA,MACA,OAAO,CACL,eACG;AACH,mBAAW,MAAM,KAAK,eAAe;AACnC,gBAAM,KAAK;AAAA,YACT,MAAM;AAAA,YACN,GAAG;AAAA,UACL;AACA,qBAAW,QAAQ,EAAE;AAAA,QACvB;AAEA,cAAM,MAAM;AAAA,UACV,MAAM;AAAA,UACN,cAAc,KAAK;AAAA,UACnB,OAAO,KAAK;AAAA,QACd;AACA,mBAAW,QAAQ,GAAG;AACtB,mBAAW,UAAU;AAAA,MACvB;AAAA,IACF;AAEA,UAAM,WAAW;AAAA,EACnB;AACF;AAUA,SAAS,mBACP,YACQ;AACR,QAAM,EAAE,MAAM,YAAY,UAAU,MAAM,IAAI;AAE9C,UAAQ,MAAM;AAAA,IACZ,KAAK;AACH,aAAO,aAAE,OAAO;AAAA,IAClB,KAAK;AACH,aAAO,aAAE,OAAO;AAAA,IAClB,KAAK;AACH,aAAO,aAAE,QAAQ;AAAA,IACnB,KAAK;AACH,UAAI,CAAC,OAAO;AACV,cAAM,IAAI,MAAM,wCAAwC;AAAA,MAC1D;AACA,aAAO,aAAE,MAAM,mBAAmB,KAAK,CAAC;AAAA,IAC1C,KAAK,UAAU;AACb,UAAI,CAAC,YAAY;AACf,cAAM,IAAI,MAAM,8CAA8C;AAAA,MAChE;AACA,YAAM,QAAgC,CAAC;AAEvC,iBAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,UAAU,GAAG;AACrD,cAAM,SAAS,mBAAmB,KAAK;AACvC,YAAI,MAAM,UAAU,SAAS,GAAG,IAAI,SAAS,OAAO,SAAS;AAC7D,cAAM,MAAM,cAAc,IAAI,SAAS,MAAM,WAAW,IAAI;AAC5D,cAAM,GAAG,IAAI;AAAA,MACf;AACA,aAAO,aAAE,OAAO,KAAK;AAAA,IACvB;AAAA,IACA;AACE,YAAM,IAAI,MAAM,qBAAqB,IAAI,EAAE;AAAA,EAC/C;AACF;","names":["crypto"]}
package/index.d.cts CHANGED
@@ -1,4 +1,5 @@
1
- import { AxGenIn, AxGenOut, AxAgentic } from '@ax-llm/ax/index.js';
1
+ import { LanguageModelV1 } from '@ai-sdk/provider';
2
+ import { AxGenIn, AxGenOut, AxAgentic, AxAIService } from '@ax-llm/ax/index.js';
2
3
  import { CoreMessage } from 'ai';
3
4
  import { ReactNode } from 'react';
4
5
  import { z } from 'zod';
@@ -23,5 +24,16 @@ declare class AxAgentProvider<IN extends AxGenIn, OUT extends AxGenOut> {
23
24
  get parameters(): z.ZodTypeAny;
24
25
  get generate(): Renderer<IN>;
25
26
  }
27
+ declare class AxAIProvider implements LanguageModelV1 {
28
+ readonly specificationVersion = "v1";
29
+ readonly defaultObjectGenerationMode = "json";
30
+ private readonly ai;
31
+ private readonly config?;
32
+ modelId: string;
33
+ constructor(ai: AxAIService, config?: Readonly<AxConfig>);
34
+ get provider(): string;
35
+ doGenerate(options: Readonly<Parameters<LanguageModelV1['doGenerate']>[0]>): Promise<Awaited<ReturnType<LanguageModelV1['doGenerate']>>>;
36
+ doStream(options: Readonly<Parameters<LanguageModelV1['doStream']>[0]>): Promise<Awaited<ReturnType<LanguageModelV1['doStream']>>>;
37
+ }
26
38
 
27
- export { AxAgentProvider };
39
+ export { AxAIProvider, AxAgentProvider };
package/index.d.ts CHANGED
@@ -1,4 +1,5 @@
1
- import { AxGenIn, AxGenOut, AxAgentic } from '@ax-llm/ax/index.js';
1
+ import { LanguageModelV1 } from '@ai-sdk/provider';
2
+ import { AxGenIn, AxGenOut, AxAgentic, AxAIService } from '@ax-llm/ax/index.js';
2
3
  import { CoreMessage } from 'ai';
3
4
  import { ReactNode } from 'react';
4
5
  import { z } from 'zod';
@@ -23,5 +24,16 @@ declare class AxAgentProvider<IN extends AxGenIn, OUT extends AxGenOut> {
23
24
  get parameters(): z.ZodTypeAny;
24
25
  get generate(): Renderer<IN>;
25
26
  }
27
+ declare class AxAIProvider implements LanguageModelV1 {
28
+ readonly specificationVersion = "v1";
29
+ readonly defaultObjectGenerationMode = "json";
30
+ private readonly ai;
31
+ private readonly config?;
32
+ modelId: string;
33
+ constructor(ai: AxAIService, config?: Readonly<AxConfig>);
34
+ get provider(): string;
35
+ doGenerate(options: Readonly<Parameters<LanguageModelV1['doGenerate']>[0]>): Promise<Awaited<ReturnType<LanguageModelV1['doGenerate']>>>;
36
+ doStream(options: Readonly<Parameters<LanguageModelV1['doStream']>[0]>): Promise<Awaited<ReturnType<LanguageModelV1['doStream']>>>;
37
+ }
26
38
 
27
- export { AxAgentProvider };
39
+ export { AxAIProvider, AxAgentProvider };
package/index.js CHANGED
@@ -1,3 +1,8 @@
1
+ // provider.ts
2
+ import {
3
+ TransformStream
4
+ } from "stream/web";
5
+
1
6
  // ../../node_modules/nanoid/index.js
2
7
  import crypto from "crypto";
3
8
  var POOL_SIZE_MULTIPLIER = 128;
@@ -100,6 +105,325 @@ var AxAgentProvider = class {
100
105
  return fn;
101
106
  }
102
107
  };
108
+ var AxAIProvider = class {
109
+ specificationVersion = "v1";
110
+ defaultObjectGenerationMode = "json";
111
+ ai;
112
+ config;
113
+ modelId;
114
+ constructor(ai, config) {
115
+ this.ai = ai;
116
+ this.config = config;
117
+ this.modelId = this.ai.getModelInfo().name;
118
+ }
119
+ get provider() {
120
+ return this.ai.getName();
121
+ }
122
+ async doGenerate(options) {
123
+ const { req, warnings } = createChatRequest(options);
124
+ const res = await this.ai.chat(req);
125
+ const choice = res.results.at(0);
126
+ if (!choice) {
127
+ throw new Error("No choice returned");
128
+ }
129
+ return {
130
+ text: choice.content ?? void 0,
131
+ toolCalls: choice.functionCalls?.map((tc) => ({
132
+ toolCallType: "function",
133
+ toolCallId: tc.id,
134
+ toolName: tc.function.name,
135
+ args: typeof tc.function.params === "string" ? tc.function.params : JSON.stringify(tc.function.params)
136
+ })),
137
+ finishReason: mapAxFinishReason(choice.finishReason),
138
+ usage: {
139
+ promptTokens: res.modelUsage?.promptTokens ?? 0,
140
+ completionTokens: res.modelUsage?.completionTokens ?? 0
141
+ },
142
+ rawCall: { rawPrompt: "", rawSettings: req.modelConfig ?? {} },
143
+ warnings
144
+ };
145
+ }
146
+ async doStream(options) {
147
+ const { req, warnings } = createChatRequest(options);
148
+ const res = await this.ai.chat(req, {
149
+ stream: true
150
+ });
151
+ return {
152
+ stream: res.pipeThrough(new AxToSDKTransformer()),
153
+ rawCall: { rawPrompt: "", rawSettings: req.modelConfig ?? {} },
154
+ warnings
155
+ };
156
+ }
157
+ };
158
+ function prepareToolsAndToolChoice(mode) {
159
+ if (!mode.tools || mode.tools.length === 0) {
160
+ return {};
161
+ }
162
+ const tools = mode.tools;
163
+ const functions = tools.map((f) => ({
164
+ name: f.name,
165
+ description: "description" in f ? f.description ?? "" : "",
166
+ parameters: f.parameters
167
+ }));
168
+ const toolChoice = mode.toolChoice;
169
+ if (!toolChoice) {
170
+ return { functions };
171
+ }
172
+ const type = toolChoice.type;
173
+ switch (type) {
174
+ case "auto":
175
+ return { functions, functionCall: "auto" };
176
+ case "none":
177
+ return { functions, functionCall: "none" };
178
+ case "required":
179
+ return { functions, functionCall: "required" };
180
+ case "tool":
181
+ return {
182
+ functions,
183
+ functionCall: {
184
+ type: "function",
185
+ function: { name: toolChoice.toolName }
186
+ }
187
+ };
188
+ default: {
189
+ const _exhaustiveCheck = type;
190
+ throw new Error(`Unsupported tool choice type: ${_exhaustiveCheck}`);
191
+ }
192
+ }
193
+ }
194
+ function convertToAxChatPrompt(prompt) {
195
+ const messages = [];
196
+ for (const { role, content } of prompt) {
197
+ switch (role) {
198
+ case "system": {
199
+ messages.push({ role: "system", content });
200
+ break;
201
+ }
202
+ case "user": {
203
+ messages.push({
204
+ role: "user",
205
+ content: content.map((part) => {
206
+ switch (part.type) {
207
+ case "text": {
208
+ return { type: "text", text: part.text };
209
+ }
210
+ case "image": {
211
+ if (!part.mimeType) {
212
+ throw new Error("Image part must have a mimeType");
213
+ }
214
+ if (!ArrayBuffer.isView(part.image)) {
215
+ throw new Error("Image part must have an ArrayBuffer");
216
+ }
217
+ const image = Buffer.from(part.image).toString("base64");
218
+ return {
219
+ type: "image",
220
+ mimeType: part.mimeType,
221
+ image
222
+ };
223
+ }
224
+ default:
225
+ throw new Error(`Unsupported part: ${part}`);
226
+ }
227
+ })
228
+ });
229
+ break;
230
+ }
231
+ case "assistant": {
232
+ let text = "";
233
+ const toolCalls = [];
234
+ for (const part of content) {
235
+ switch (part.type) {
236
+ case "text": {
237
+ text += part.text;
238
+ break;
239
+ }
240
+ case "tool-call": {
241
+ toolCalls.push({
242
+ id: part.toolCallId,
243
+ type: "function",
244
+ function: {
245
+ name: part.toolName,
246
+ params: part.args
247
+ }
248
+ });
249
+ break;
250
+ }
251
+ default: {
252
+ const _exhaustiveCheck = part;
253
+ throw new Error(`Unsupported part: ${_exhaustiveCheck}`);
254
+ }
255
+ }
256
+ }
257
+ const functionCalls = toolCalls.length === 0 ? void 0 : toolCalls;
258
+ if (functionCalls || text.length > 0) {
259
+ messages.push({
260
+ role: "assistant",
261
+ content: text,
262
+ functionCalls
263
+ });
264
+ }
265
+ break;
266
+ }
267
+ case "tool": {
268
+ for (const part of content) {
269
+ messages.push({
270
+ role: "function",
271
+ functionId: part.toolCallId,
272
+ result: JSON.stringify(part.result, null, 2)
273
+ });
274
+ }
275
+ break;
276
+ }
277
+ default: {
278
+ const _exhaustiveCheck = role;
279
+ throw new Error(`Unsupported role: ${_exhaustiveCheck}`);
280
+ }
281
+ }
282
+ }
283
+ return messages;
284
+ }
285
+ function mapAxFinishReason(finishReason) {
286
+ switch (finishReason) {
287
+ case "stop":
288
+ return "stop";
289
+ case "length":
290
+ return "length";
291
+ case "function_call":
292
+ return "tool-calls";
293
+ default:
294
+ return "other";
295
+ }
296
+ }
297
+ function createChatRequest({
298
+ mode,
299
+ prompt,
300
+ maxTokens,
301
+ temperature,
302
+ topP,
303
+ frequencyPenalty,
304
+ presencePenalty
305
+ //seed,
306
+ }) {
307
+ const req = {
308
+ chatPrompt: convertToAxChatPrompt(prompt),
309
+ ...frequencyPenalty != null ? { frequencyPenalty } : {},
310
+ ...presencePenalty != null ? { presencePenalty } : {},
311
+ ...maxTokens != null ? { maxTokens } : {},
312
+ ...temperature != null ? { temperature } : {},
313
+ ...topP != null ? { topP } : {}
314
+ };
315
+ const warnings = [];
316
+ switch (mode.type) {
317
+ case "regular": {
318
+ return {
319
+ req: { ...req, ...prepareToolsAndToolChoice(mode) },
320
+ warnings
321
+ };
322
+ }
323
+ case "object-json": {
324
+ return {
325
+ req,
326
+ warnings
327
+ };
328
+ }
329
+ case "object-tool": {
330
+ const tool = {
331
+ type: "function",
332
+ function: {
333
+ name: mode.tool.name,
334
+ params: mode.tool.parameters
335
+ }
336
+ };
337
+ return {
338
+ req: { ...req, ...tool },
339
+ warnings
340
+ };
341
+ }
342
+ default: {
343
+ throw new Error(`Unsupported type`);
344
+ }
345
+ }
346
+ }
347
+ var AxToSDKTransformer = class extends TransformStream {
348
+ usage = {
349
+ promptTokens: 0,
350
+ completionTokens: 0
351
+ };
352
+ finishReason = "other";
353
+ functionCalls = [];
354
+ constructor() {
355
+ const transformer = {
356
+ transform: (chunk, controller) => {
357
+ const choice = chunk.results.at(0);
358
+ if (!choice) {
359
+ const val = {
360
+ type: "finish",
361
+ finishReason: this.finishReason,
362
+ usage: this.usage
363
+ };
364
+ controller.enqueue(val);
365
+ return;
366
+ }
367
+ if (chunk.modelUsage) {
368
+ this.usage = {
369
+ promptTokens: this.usage.promptTokens + chunk.modelUsage.promptTokens,
370
+ completionTokens: this.usage.completionTokens + chunk.modelUsage?.completionTokens
371
+ };
372
+ }
373
+ if (choice.functionCalls) {
374
+ for (const fc of choice.functionCalls) {
375
+ const index = this.functionCalls.findIndex(
376
+ (f) => f.toolCallId === fc.id
377
+ );
378
+ if (index === -1) {
379
+ this.functionCalls.push({
380
+ toolCallType: "function",
381
+ toolCallId: fc.id,
382
+ toolName: fc.function.name,
383
+ args: typeof fc.function.params === "string" ? fc.function.params : JSON.stringify(fc.function.params)
384
+ });
385
+ } else {
386
+ const obj = this.functionCalls[index];
387
+ if (!obj) {
388
+ continue;
389
+ }
390
+ if (typeof fc.function.params === "string") {
391
+ obj.args = (obj.args ?? "") + fc.function.params;
392
+ } else {
393
+ obj.args = JSON.stringify(fc.function.params);
394
+ }
395
+ }
396
+ this.finishReason = "tool-calls";
397
+ }
398
+ }
399
+ if (choice.content && choice.content.length > 0) {
400
+ controller.enqueue({
401
+ type: "text-delta",
402
+ textDelta: choice.content ?? ""
403
+ });
404
+ this.finishReason = mapAxFinishReason(choice.finishReason);
405
+ }
406
+ },
407
+ flush: (controller) => {
408
+ for (const fc of this.functionCalls) {
409
+ const tc = {
410
+ type: "tool-call",
411
+ ...fc
412
+ };
413
+ controller.enqueue(tc);
414
+ }
415
+ const val = {
416
+ type: "finish",
417
+ finishReason: this.finishReason,
418
+ usage: this.usage
419
+ };
420
+ controller.enqueue(val);
421
+ controller.terminate();
422
+ }
423
+ };
424
+ super(transformer);
425
+ }
426
+ };
103
427
  function convertToZodSchema(jsonSchema) {
104
428
  const { type, properties, required, items } = jsonSchema;
105
429
  switch (type) {
@@ -132,6 +456,7 @@ function convertToZodSchema(jsonSchema) {
132
456
  }
133
457
  }
134
458
  export {
459
+ AxAIProvider,
135
460
  AxAgentProvider
136
461
  };
137
462
  //# sourceMappingURL=index.js.map
package/index.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"sources":["../../../node_modules/nanoid/index.js","../provider.ts"],"sourcesContent":["import crypto from 'crypto'\n\nimport { urlAlphabet } from './url-alphabet/index.js'\n\n// It is best to make fewer, larger requests to the crypto module to\n// avoid system call overhead. So, random numbers are generated in a\n// pool. The pool is a Buffer that is larger than the initial random\n// request size by this multiplier. The pool is enlarged if subsequent\n// requests exceed the maximum buffer size.\nconst POOL_SIZE_MULTIPLIER = 128\nlet pool, poolOffset\n\nlet fillPool = bytes => {\n if (!pool || pool.length < bytes) {\n pool = Buffer.allocUnsafe(bytes * POOL_SIZE_MULTIPLIER)\n crypto.randomFillSync(pool)\n poolOffset = 0\n } else if (poolOffset + bytes > pool.length) {\n crypto.randomFillSync(pool)\n poolOffset = 0\n }\n poolOffset += bytes\n}\n\nlet random = bytes => {\n // `|=` convert `bytes` to number to prevent `valueOf` abusing and pool pollution\n fillPool((bytes |= 0))\n return pool.subarray(poolOffset - bytes, poolOffset)\n}\n\nlet customRandom = (alphabet, defaultSize, getRandom) => {\n // First, a bitmask is necessary to generate the ID. The bitmask makes bytes\n // values closer to the alphabet size. The bitmask calculates the closest\n // `2^31 - 1` number, which exceeds the alphabet size.\n // For example, the bitmask for the alphabet size 30 is 31 (00011111).\n let mask = (2 << (31 - Math.clz32((alphabet.length - 1) | 1))) - 1\n // Though, the bitmask solution is not perfect since the bytes exceeding\n // the alphabet size are refused. Therefore, to reliably generate the ID,\n // the random bytes redundancy has to be satisfied.\n\n // Note: every hardware random generator call is performance expensive,\n // because the system call for entropy collection takes a lot of time.\n // So, to avoid additional system calls, extra bytes are requested in advance.\n\n // Next, a step determines how many random bytes to generate.\n // The number of random bytes gets decided upon the ID size, mask,\n // alphabet size, and magic number 1.6 (using 1.6 peaks at performance\n // according to benchmarks).\n let step = Math.ceil((1.6 * mask * defaultSize) / alphabet.length)\n\n return (size = defaultSize) => {\n let id = ''\n while (true) {\n let bytes = getRandom(step)\n // A compact alternative for `for (let i = 0; i < step; i++)`.\n let i = step\n while (i--) {\n // Adding `|| ''` refuses a random byte that exceeds the alphabet size.\n id += alphabet[bytes[i] & mask] || ''\n if (id.length === size) return id\n }\n }\n }\n}\n\nlet customAlphabet = (alphabet, size = 21) =>\n customRandom(alphabet, size, random)\n\nlet nanoid = (size = 21) => {\n // `|=` convert `size` to number to prevent `valueOf` abusing and pool pollution\n fillPool((size |= 0))\n let id = ''\n // We are reading directly from the random pool to avoid creating new array\n for (let i = poolOffset - size; i < poolOffset; i++) {\n // It is incorrect to use bytes exceeding the alphabet size.\n // The following mask reduces the random byte in the 0-255 value\n // range to the 0-63 value range. Therefore, adding hacks, such\n // as empty string fallback or magic numbers, is unneccessary because\n // the bitmask trims bytes down to the alphabet size.\n id += urlAlphabet[pool[i] & 63]\n }\n return id\n}\n\nexport { nanoid, customAlphabet, customRandom, urlAlphabet, random }\n","// cspell:ignore Streamable\n\nimport {\n type ReadableStream,\n TransformStream,\n TransformStreamDefaultController,\n} from 'stream/web'\n\nimport {\n type LanguageModelV1,\n type LanguageModelV1CallWarning,\n type LanguageModelV1FinishReason,\n type LanguageModelV1FunctionTool,\n type LanguageModelV1FunctionToolCall,\n type LanguageModelV1Prompt,\n type LanguageModelV1StreamPart,\n} from '@ai-sdk/provider'\nimport type {\n AxAgentic,\n AxAIService,\n AxChatRequest,\n AxChatResponse,\n AxChatResponseResult,\n AxFunction,\n AxFunctionJSONSchema,\n AxGenIn,\n AxGenOut,\n} from '@ax-llm/ax/index.js'\nimport type { CoreMessage } from 'ai'\nimport { customAlphabet } from 'nanoid'\nimport type { ReactNode } from 'react'\nimport { z } from 'zod'\n\ntype Writeable<T> = { -readonly [P in keyof T]: T[P] }\ntype AxChatRequestChatPrompt = Writeable<AxChatRequest['chatPrompt'][0]>\n\ntype AxConfig = {\n fetch?: typeof fetch\n}\n\ntype Streamable = ReactNode | Promise<ReactNode>\ntype Renderer<T> = (\n args: T\n) =>\n | Streamable\n | Generator<Streamable, Streamable, void>\n | AsyncGenerator<Streamable, Streamable, void>\n\nconst nanoid = customAlphabet(\n '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz',\n 7\n)\n\nexport class AxAgentProvider<IN extends AxGenIn, OUT extends AxGenOut> {\n private readonly config?: AxConfig\n private readonly funcInfo: AxFunction\n private generateFunction: Renderer<OUT>\n private updateState: (msgs: readonly CoreMessage[]) => void\n\n constructor({\n agent,\n updateState,\n generate,\n config,\n }: Readonly<{\n agent: AxAgentic\n updateState: (msgs: readonly CoreMessage[]) => void\n generate: Renderer<OUT>\n config?: Readonly<AxConfig>\n }>) {\n this.funcInfo = agent.getFunction()\n this.generateFunction = generate\n this.updateState = updateState\n this.config = config\n }\n\n get description() {\n return this.funcInfo.description\n }\n\n get parameters(): z.ZodTypeAny {\n const schema = this.funcInfo.parameters ?? {\n type: 'object',\n properties: {},\n }\n\n return convertToZodSchema(schema)\n }\n\n get generate(): Renderer<IN> {\n const fn = async (input: IN) => {\n const res = (await this.funcInfo.func(input)) as OUT\n const toolCallId = nanoid()\n\n this.updateState([\n {\n role: 'assistant',\n content: [\n {\n type: 'tool-call',\n toolName: this.funcInfo.name,\n toolCallId,\n args: input,\n },\n ],\n },\n {\n role: 'tool',\n content: [\n {\n type: 'tool-result',\n toolName: this.funcInfo.name,\n toolCallId,\n result: res,\n },\n ],\n },\n ])\n\n return this.generateFunction(res)\n }\n return fn as Renderer<IN>\n }\n}\n\nexport class AxAIProvider implements LanguageModelV1 {\n readonly specificationVersion = 'v1'\n readonly defaultObjectGenerationMode = 'json'\n\n private readonly ai: AxAIService\n private readonly config?: AxConfig\n\n public modelId: string\n\n constructor(ai: AxAIService, config?: Readonly<AxConfig>) {\n this.ai = ai\n this.config = config\n this.modelId = this.ai.getModelInfo().name\n }\n\n get provider(): string {\n return this.ai.getName()\n }\n\n async doGenerate(\n options: Readonly<Parameters<LanguageModelV1['doGenerate']>[0]>\n ): Promise<Awaited<ReturnType<LanguageModelV1['doGenerate']>>> {\n const { req, warnings } = createChatRequest(options)\n const res = (await this.ai.chat(req)) as AxChatResponse\n const choice = res.results.at(0)\n\n if (!choice) {\n throw new Error('No choice returned')\n }\n\n return {\n text: choice.content ?? undefined,\n toolCalls: choice.functionCalls?.map((tc) => ({\n toolCallType: 'function',\n toolCallId: tc.id,\n toolName: tc.function.name,\n args:\n typeof tc.function.params === 'string'\n ? tc.function.params\n : JSON.stringify(tc.function.params),\n })),\n finishReason: mapAxFinishReason(choice.finishReason),\n usage: {\n promptTokens: res.modelUsage?.promptTokens ?? 0,\n completionTokens: res.modelUsage?.completionTokens ?? 0,\n },\n rawCall: { rawPrompt: '', rawSettings: req.modelConfig ?? {} },\n warnings,\n }\n }\n\n async doStream(\n options: Readonly<Parameters<LanguageModelV1['doStream']>[0]>\n ): Promise<Awaited<ReturnType<LanguageModelV1['doStream']>>> {\n const { req, warnings } = createChatRequest(options)\n\n const res = (await this.ai.chat(req, {\n stream: true,\n })) as ReadableStream<AxChatResponse>\n\n return {\n stream: res.pipeThrough(new AxToSDKTransformer()),\n rawCall: { rawPrompt: '', rawSettings: req.modelConfig ?? {} },\n warnings,\n }\n }\n}\n\nfunction prepareToolsAndToolChoice(\n mode: Readonly<\n Parameters<LanguageModelV1['doGenerate']>[0]['mode'] & { type: 'regular' }\n >\n): Pick<AxChatRequest, 'functions' | 'functionCall'> {\n // when the tools array is empty, change it to undefined to prevent errors:\n if (!mode.tools || mode.tools.length === 0) {\n return {}\n }\n\n const tools = mode.tools as Array<LanguageModelV1FunctionTool>\n const functions = tools.map((f) => ({\n name: f.name,\n description: 'description' in f ? (f.description ?? '') : '',\n parameters: f.parameters as AxFunctionJSONSchema,\n }))\n\n const toolChoice = mode.toolChoice\n if (!toolChoice) {\n return { functions }\n }\n\n const type = toolChoice.type\n\n switch (type) {\n case 'auto':\n return { functions, functionCall: 'auto' }\n case 'none':\n return { functions, functionCall: 'none' }\n case 'required':\n return { functions, functionCall: 'required' }\n case 'tool':\n return {\n functions,\n functionCall: {\n type: 'function',\n function: { name: toolChoice.toolName },\n },\n }\n default: {\n const _exhaustiveCheck: never = type\n throw new Error(`Unsupported tool choice type: ${_exhaustiveCheck}`)\n }\n }\n}\n\nfunction convertToAxChatPrompt(\n prompt: Readonly<LanguageModelV1Prompt>\n): AxChatRequest['chatPrompt'] {\n const messages: AxChatRequest['chatPrompt'] = []\n\n for (const { role, content } of prompt) {\n switch (role) {\n case 'system': {\n messages.push({ role: 'system', content })\n break\n }\n\n case 'user': {\n messages.push({\n role: 'user',\n content: content.map((part) => {\n switch (part.type) {\n case 'text': {\n return { type: 'text', text: part.text }\n }\n case 'image': {\n if (!part.mimeType) {\n throw new Error('Image part must have a mimeType')\n }\n if (!ArrayBuffer.isView(part.image)) {\n throw new Error('Image part must have an ArrayBuffer')\n }\n const image = Buffer.from(part.image).toString('base64')\n return {\n type: 'image',\n mimeType: part.mimeType,\n image,\n }\n }\n default:\n throw new Error(`Unsupported part: ${part}`)\n // case 'audio': {\n // if (!part.data) {\n // throw new Error('Audio part must have a audio');\n // }\n // if (!ArrayBuffer.isView(part.data)) {\n // throw new Error('Audio part must have an ArrayBuffer');\n // }\n // const data = Buffer.from(part.data).toString('base64');\n // return {\n // type: 'audio',\n // format: 'wav',\n // data\n // };\n // }\n }\n }),\n })\n break\n }\n\n case 'assistant': {\n let text = ''\n const toolCalls: Extract<\n AxChatRequestChatPrompt,\n { role: 'assistant' }\n >['functionCalls'] = []\n\n for (const part of content) {\n switch (part.type) {\n case 'text': {\n text += part.text\n break\n }\n case 'tool-call': {\n toolCalls.push({\n id: part.toolCallId,\n type: 'function',\n function: {\n name: part.toolName,\n params: part.args as Record<string, unknown>,\n },\n })\n break\n }\n\n default: {\n const _exhaustiveCheck: never = part\n throw new Error(`Unsupported part: ${_exhaustiveCheck}`)\n }\n }\n }\n\n const functionCalls = toolCalls.length === 0 ? undefined : toolCalls\n\n if (functionCalls || text.length > 0) {\n messages.push({\n role: 'assistant',\n content: text,\n functionCalls,\n })\n }\n\n break\n }\n case 'tool': {\n for (const part of content) {\n messages.push({\n role: 'function' as const,\n functionId: part.toolCallId,\n result: JSON.stringify(part.result, null, 2),\n })\n }\n break\n }\n default: {\n const _exhaustiveCheck: never = role\n throw new Error(`Unsupported role: ${_exhaustiveCheck}`)\n }\n }\n }\n\n return messages\n}\n\nfunction mapAxFinishReason(\n finishReason: AxChatResponseResult['finishReason']\n): LanguageModelV1FinishReason {\n switch (finishReason) {\n case 'stop':\n return 'stop'\n case 'length':\n return 'length'\n case 'function_call':\n return 'tool-calls'\n default:\n return 'other'\n }\n}\n\nfunction createChatRequest({\n mode,\n prompt,\n maxTokens,\n temperature,\n topP,\n frequencyPenalty,\n presencePenalty,\n //seed,\n}: Readonly<Parameters<LanguageModelV1['doGenerate']>[0]>): {\n req: AxChatRequest\n warnings: LanguageModelV1CallWarning[]\n} {\n const req: AxChatRequest = {\n chatPrompt: convertToAxChatPrompt(prompt),\n ...(frequencyPenalty != null ? { frequencyPenalty } : {}),\n ...(presencePenalty != null ? { presencePenalty } : {}),\n ...(maxTokens != null ? { maxTokens } : {}),\n ...(temperature != null ? { temperature } : {}),\n ...(topP != null ? { topP } : {}),\n }\n\n const warnings: LanguageModelV1CallWarning[] = []\n\n switch (mode.type) {\n case 'regular': {\n return {\n req: { ...req, ...prepareToolsAndToolChoice(mode) },\n warnings,\n }\n }\n\n case 'object-json': {\n return {\n req,\n warnings,\n }\n }\n\n case 'object-tool': {\n const tool = {\n type: 'function',\n function: {\n name: mode.tool.name,\n params: mode.tool.parameters,\n },\n }\n return {\n req: { ...req, ...tool },\n warnings,\n }\n }\n\n default: {\n throw new Error(`Unsupported type`)\n }\n }\n}\n\nclass AxToSDKTransformer extends TransformStream<\n AxChatResponse,\n LanguageModelV1StreamPart\n> {\n private usage: Extract<\n LanguageModelV1StreamPart,\n { type: 'finish' }\n >['usage'] = {\n promptTokens: 0,\n completionTokens: 0,\n }\n\n private finishReason: Extract<\n LanguageModelV1StreamPart,\n { type: 'finish' }\n >['finishReason'] = 'other'\n\n private functionCalls: LanguageModelV1FunctionToolCall[] = []\n\n constructor() {\n const transformer = {\n transform: (\n chunk: Readonly<AxChatResponse>,\n controller: TransformStreamDefaultController<LanguageModelV1StreamPart>\n ) => {\n const choice = chunk.results.at(0)\n if (!choice) {\n const val = {\n type: 'finish' as const,\n finishReason: this.finishReason,\n usage: this.usage,\n }\n controller.enqueue(val)\n return\n }\n\n if (chunk.modelUsage) {\n this.usage = {\n promptTokens:\n this.usage.promptTokens + chunk.modelUsage.promptTokens,\n completionTokens:\n this.usage.completionTokens + chunk.modelUsage?.completionTokens,\n }\n }\n\n if (choice.functionCalls) {\n for (const fc of choice.functionCalls) {\n const index = this.functionCalls.findIndex(\n (f) => f.toolCallId === fc.id\n )\n if (index === -1) {\n this.functionCalls.push({\n toolCallType: 'function' as const,\n toolCallId: fc.id,\n toolName: fc.function.name,\n args:\n typeof fc.function.params === 'string'\n ? fc.function.params\n : JSON.stringify(fc.function.params),\n })\n } else {\n const obj = this.functionCalls[index]\n if (!obj) {\n continue\n }\n if (typeof fc.function.params === 'string') {\n obj.args = (obj.args ?? '') + fc.function.params\n } else {\n obj.args = JSON.stringify(fc.function.params)\n }\n }\n this.finishReason = 'tool-calls'\n }\n }\n\n if (choice.content && choice.content.length > 0) {\n controller.enqueue({\n type: 'text-delta',\n textDelta: choice.content ?? '',\n })\n this.finishReason = mapAxFinishReason(choice.finishReason)\n }\n },\n flush: (\n controller: TransformStreamDefaultController<LanguageModelV1StreamPart>\n ) => {\n for (const fc of this.functionCalls) {\n const tc = {\n type: 'tool-call' as const,\n ...fc,\n }\n controller.enqueue(tc)\n }\n\n const val = {\n type: 'finish' as const,\n finishReason: this.finishReason,\n usage: this.usage,\n }\n controller.enqueue(val)\n controller.terminate()\n },\n }\n\n super(transformer)\n }\n}\n\ntype AnyZod =\n | z.AnyZodObject\n | z.ZodString\n | z.ZodNumber\n | z.ZodBoolean\n | z.ZodArray<AnyZod>\n | z.ZodOptional<AnyZod>\n\nfunction convertToZodSchema(\n jsonSchema: Readonly<AxFunctionJSONSchema>\n): AnyZod {\n const { type, properties, required, items } = jsonSchema\n\n switch (type) {\n case 'string':\n return z.string()\n case 'number':\n return z.number()\n case 'boolean':\n return z.boolean()\n case 'array':\n if (!items) {\n throw new Error(\"Array type must have 'items' property.\")\n }\n return z.array(convertToZodSchema(items))\n case 'object': {\n if (!properties) {\n throw new Error(\"Object type must have 'properties' property.\")\n }\n const shape: Record<string, AnyZod> = {}\n\n for (const [key, value] of Object.entries(properties)) {\n const schema = convertToZodSchema(value)\n let val = required?.includes(key) ? schema : schema.optional()\n val = value.description ? val.describe(value.description) : val\n shape[key] = val\n }\n return z.object(shape)\n }\n default:\n throw new Error(`Unsupported type: ${type}`)\n }\n}\n"],"mappings":";AAAA,OAAO,YAAY;AASnB,IAAM,uBAAuB;AAC7B,IAAI;AAAJ,IAAU;AAEV,IAAI,WAAW,WAAS;AACtB,MAAI,CAAC,QAAQ,KAAK,SAAS,OAAO;AAChC,WAAO,OAAO,YAAY,QAAQ,oBAAoB;AACtD,WAAO,eAAe,IAAI;AAC1B,iBAAa;AAAA,EACf,WAAW,aAAa,QAAQ,KAAK,QAAQ;AAC3C,WAAO,eAAe,IAAI;AAC1B,iBAAa;AAAA,EACf;AACA,gBAAc;AAChB;AAEA,IAAI,SAAS,WAAS;AAEpB,WAAU,SAAS,CAAE;AACrB,SAAO,KAAK,SAAS,aAAa,OAAO,UAAU;AACrD;AAEA,IAAI,eAAe,CAAC,UAAU,aAAa,cAAc;AAKvD,MAAI,QAAQ,KAAM,KAAK,KAAK,MAAO,SAAS,SAAS,IAAK,CAAC,KAAM;AAajE,MAAI,OAAO,KAAK,KAAM,MAAM,OAAO,cAAe,SAAS,MAAM;AAEjE,SAAO,CAAC,OAAO,gBAAgB;AAC7B,QAAI,KAAK;AACT,WAAO,MAAM;AACX,UAAI,QAAQ,UAAU,IAAI;AAE1B,UAAI,IAAI;AACR,aAAO,KAAK;AAEV,cAAM,SAAS,MAAM,CAAC,IAAI,IAAI,KAAK;AACnC,YAAI,GAAG,WAAW,KAAM,QAAO;AAAA,MACjC;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAI,iBAAiB,CAAC,UAAU,OAAO,OACrC,aAAa,UAAU,MAAM,MAAM;;;ACnCrC,SAAS,SAAS;AAiBlB,IAAM,SAAS;AAAA,EACb;AAAA,EACA;AACF;AAEO,IAAM,kBAAN,MAAgE;AAAA,EACpD;AAAA,EACA;AAAA,EACT;AAAA,EACA;AAAA,EAER,YAAY;AAAA,IACV;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAKI;AACF,SAAK,WAAW,MAAM,YAAY;AAClC,SAAK,mBAAmB;AACxB,SAAK,cAAc;AACnB,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,IAAI,cAAc;AAChB,WAAO,KAAK,SAAS;AAAA,EACvB;AAAA,EAEA,IAAI,aAA2B;AAC7B,UAAM,SAAS,KAAK,SAAS,cAAc;AAAA,MACzC,MAAM;AAAA,MACN,YAAY,CAAC;AAAA,IACf;AAEA,WAAO,mBAAmB,MAAM;AAAA,EAClC;AAAA,EAEA,IAAI,WAAyB;AAC3B,UAAM,KAAK,OAAO,UAAc;AAC9B,YAAM,MAAO,MAAM,KAAK,SAAS,KAAK,KAAK;AAC3C,YAAM,aAAa,OAAO;AAE1B,WAAK,YAAY;AAAA,QACf;AAAA,UACE,MAAM;AAAA,UACN,SAAS;AAAA,YACP;AAAA,cACE,MAAM;AAAA,cACN,UAAU,KAAK,SAAS;AAAA,cACxB;AAAA,cACA,MAAM;AAAA,YACR;AAAA,UACF;AAAA,QACF;AAAA,QACA;AAAA,UACE,MAAM;AAAA,UACN,SAAS;AAAA,YACP;AAAA,cACE,MAAM;AAAA,cACN,UAAU,KAAK,SAAS;AAAA,cACxB;AAAA,cACA,QAAQ;AAAA,YACV;AAAA,UACF;AAAA,QACF;AAAA,MACF,CAAC;AAED,aAAO,KAAK,iBAAiB,GAAG;AAAA,IAClC;AACA,WAAO;AAAA,EACT;AACF;AA0aA,SAAS,mBACP,YACQ;AACR,QAAM,EAAE,MAAM,YAAY,UAAU,MAAM,IAAI;AAE9C,UAAQ,MAAM;AAAA,IACZ,KAAK;AACH,aAAO,EAAE,OAAO;AAAA,IAClB,KAAK;AACH,aAAO,EAAE,OAAO;AAAA,IAClB,KAAK;AACH,aAAO,EAAE,QAAQ;AAAA,IACnB,KAAK;AACH,UAAI,CAAC,OAAO;AACV,cAAM,IAAI,MAAM,wCAAwC;AAAA,MAC1D;AACA,aAAO,EAAE,MAAM,mBAAmB,KAAK,CAAC;AAAA,IAC1C,KAAK,UAAU;AACb,UAAI,CAAC,YAAY;AACf,cAAM,IAAI,MAAM,8CAA8C;AAAA,MAChE;AACA,YAAM,QAAgC,CAAC;AAEvC,iBAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,UAAU,GAAG;AACrD,cAAM,SAAS,mBAAmB,KAAK;AACvC,YAAI,MAAM,UAAU,SAAS,GAAG,IAAI,SAAS,OAAO,SAAS;AAC7D,cAAM,MAAM,cAAc,IAAI,SAAS,MAAM,WAAW,IAAI;AAC5D,cAAM,GAAG,IAAI;AAAA,MACf;AACA,aAAO,EAAE,OAAO,KAAK;AAAA,IACvB;AAAA,IACA;AACE,YAAM,IAAI,MAAM,qBAAqB,IAAI,EAAE;AAAA,EAC/C;AACF;","names":[]}
1
+ {"version":3,"sources":["../provider.ts","../../../node_modules/nanoid/index.js"],"sourcesContent":["// cspell:ignore Streamable\n\nimport {\n type ReadableStream,\n TransformStream,\n TransformStreamDefaultController,\n} from 'stream/web'\n\nimport {\n type LanguageModelV1,\n type LanguageModelV1CallWarning,\n type LanguageModelV1FinishReason,\n type LanguageModelV1FunctionTool,\n type LanguageModelV1FunctionToolCall,\n type LanguageModelV1Prompt,\n type LanguageModelV1StreamPart,\n} from '@ai-sdk/provider'\nimport type {\n AxAgentic,\n AxAIService,\n AxChatRequest,\n AxChatResponse,\n AxChatResponseResult,\n AxFunction,\n AxFunctionJSONSchema,\n AxGenIn,\n AxGenOut,\n} from '@ax-llm/ax/index.js'\nimport type { CoreMessage } from 'ai'\nimport { customAlphabet } from 'nanoid'\nimport type { ReactNode } from 'react'\nimport { z } from 'zod'\n\ntype Writeable<T> = { -readonly [P in keyof T]: T[P] }\ntype AxChatRequestChatPrompt = Writeable<AxChatRequest['chatPrompt'][0]>\n\ntype AxConfig = {\n fetch?: typeof fetch\n}\n\ntype Streamable = ReactNode | Promise<ReactNode>\ntype Renderer<T> = (\n args: T\n) =>\n | Streamable\n | Generator<Streamable, Streamable, void>\n | AsyncGenerator<Streamable, Streamable, void>\n\nconst nanoid = customAlphabet(\n '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz',\n 7\n)\n\nexport class AxAgentProvider<IN extends AxGenIn, OUT extends AxGenOut> {\n private readonly config?: AxConfig\n private readonly funcInfo: AxFunction\n private generateFunction: Renderer<OUT>\n private updateState: (msgs: readonly CoreMessage[]) => void\n\n constructor({\n agent,\n updateState,\n generate,\n config,\n }: Readonly<{\n agent: AxAgentic\n updateState: (msgs: readonly CoreMessage[]) => void\n generate: Renderer<OUT>\n config?: Readonly<AxConfig>\n }>) {\n this.funcInfo = agent.getFunction()\n this.generateFunction = generate\n this.updateState = updateState\n this.config = config\n }\n\n get description() {\n return this.funcInfo.description\n }\n\n get parameters(): z.ZodTypeAny {\n const schema = this.funcInfo.parameters ?? {\n type: 'object',\n properties: {},\n }\n\n return convertToZodSchema(schema)\n }\n\n get generate(): Renderer<IN> {\n const fn = async (input: IN) => {\n const res = (await this.funcInfo.func(input)) as OUT\n const toolCallId = nanoid()\n\n this.updateState([\n {\n role: 'assistant',\n content: [\n {\n type: 'tool-call',\n toolName: this.funcInfo.name,\n toolCallId,\n args: input,\n },\n ],\n },\n {\n role: 'tool',\n content: [\n {\n type: 'tool-result',\n toolName: this.funcInfo.name,\n toolCallId,\n result: res,\n },\n ],\n },\n ])\n\n return this.generateFunction(res)\n }\n return fn as Renderer<IN>\n }\n}\n\nexport class AxAIProvider implements LanguageModelV1 {\n readonly specificationVersion = 'v1'\n readonly defaultObjectGenerationMode = 'json'\n\n private readonly ai: AxAIService\n private readonly config?: AxConfig\n\n public modelId: string\n\n constructor(ai: AxAIService, config?: Readonly<AxConfig>) {\n this.ai = ai\n this.config = config\n this.modelId = this.ai.getModelInfo().name\n }\n\n get provider(): string {\n return this.ai.getName()\n }\n\n async doGenerate(\n options: Readonly<Parameters<LanguageModelV1['doGenerate']>[0]>\n ): Promise<Awaited<ReturnType<LanguageModelV1['doGenerate']>>> {\n const { req, warnings } = createChatRequest(options)\n const res = (await this.ai.chat(req)) as AxChatResponse\n const choice = res.results.at(0)\n\n if (!choice) {\n throw new Error('No choice returned')\n }\n\n return {\n text: choice.content ?? undefined,\n toolCalls: choice.functionCalls?.map((tc) => ({\n toolCallType: 'function',\n toolCallId: tc.id,\n toolName: tc.function.name,\n args:\n typeof tc.function.params === 'string'\n ? tc.function.params\n : JSON.stringify(tc.function.params),\n })),\n finishReason: mapAxFinishReason(choice.finishReason),\n usage: {\n promptTokens: res.modelUsage?.promptTokens ?? 0,\n completionTokens: res.modelUsage?.completionTokens ?? 0,\n },\n rawCall: { rawPrompt: '', rawSettings: req.modelConfig ?? {} },\n warnings,\n }\n }\n\n async doStream(\n options: Readonly<Parameters<LanguageModelV1['doStream']>[0]>\n ): Promise<Awaited<ReturnType<LanguageModelV1['doStream']>>> {\n const { req, warnings } = createChatRequest(options)\n\n const res = (await this.ai.chat(req, {\n stream: true,\n })) as ReadableStream<AxChatResponse>\n\n return {\n stream: res.pipeThrough(new AxToSDKTransformer()),\n rawCall: { rawPrompt: '', rawSettings: req.modelConfig ?? {} },\n warnings,\n }\n }\n}\n\nfunction prepareToolsAndToolChoice(\n mode: Readonly<\n Parameters<LanguageModelV1['doGenerate']>[0]['mode'] & { type: 'regular' }\n >\n): Pick<AxChatRequest, 'functions' | 'functionCall'> {\n // when the tools array is empty, change it to undefined to prevent errors:\n if (!mode.tools || mode.tools.length === 0) {\n return {}\n }\n\n const tools = mode.tools as Array<LanguageModelV1FunctionTool>\n const functions = tools.map((f) => ({\n name: f.name,\n description: 'description' in f ? (f.description ?? '') : '',\n parameters: f.parameters as AxFunctionJSONSchema,\n }))\n\n const toolChoice = mode.toolChoice\n if (!toolChoice) {\n return { functions }\n }\n\n const type = toolChoice.type\n\n switch (type) {\n case 'auto':\n return { functions, functionCall: 'auto' }\n case 'none':\n return { functions, functionCall: 'none' }\n case 'required':\n return { functions, functionCall: 'required' }\n case 'tool':\n return {\n functions,\n functionCall: {\n type: 'function',\n function: { name: toolChoice.toolName },\n },\n }\n default: {\n const _exhaustiveCheck: never = type\n throw new Error(`Unsupported tool choice type: ${_exhaustiveCheck}`)\n }\n }\n}\n\nfunction convertToAxChatPrompt(\n prompt: Readonly<LanguageModelV1Prompt>\n): AxChatRequest['chatPrompt'] {\n const messages: AxChatRequest['chatPrompt'] = []\n\n for (const { role, content } of prompt) {\n switch (role) {\n case 'system': {\n messages.push({ role: 'system', content })\n break\n }\n\n case 'user': {\n messages.push({\n role: 'user',\n content: content.map((part) => {\n switch (part.type) {\n case 'text': {\n return { type: 'text', text: part.text }\n }\n case 'image': {\n if (!part.mimeType) {\n throw new Error('Image part must have a mimeType')\n }\n if (!ArrayBuffer.isView(part.image)) {\n throw new Error('Image part must have an ArrayBuffer')\n }\n const image = Buffer.from(part.image).toString('base64')\n return {\n type: 'image',\n mimeType: part.mimeType,\n image,\n }\n }\n default:\n throw new Error(`Unsupported part: ${part}`)\n // case 'audio': {\n // if (!part.data) {\n // throw new Error('Audio part must have a audio');\n // }\n // if (!ArrayBuffer.isView(part.data)) {\n // throw new Error('Audio part must have an ArrayBuffer');\n // }\n // const data = Buffer.from(part.data).toString('base64');\n // return {\n // type: 'audio',\n // format: 'wav',\n // data\n // };\n // }\n }\n }),\n })\n break\n }\n\n case 'assistant': {\n let text = ''\n const toolCalls: Extract<\n AxChatRequestChatPrompt,\n { role: 'assistant' }\n >['functionCalls'] = []\n\n for (const part of content) {\n switch (part.type) {\n case 'text': {\n text += part.text\n break\n }\n case 'tool-call': {\n toolCalls.push({\n id: part.toolCallId,\n type: 'function',\n function: {\n name: part.toolName,\n params: part.args as Record<string, unknown>,\n },\n })\n break\n }\n\n default: {\n const _exhaustiveCheck: never = part\n throw new Error(`Unsupported part: ${_exhaustiveCheck}`)\n }\n }\n }\n\n const functionCalls = toolCalls.length === 0 ? undefined : toolCalls\n\n if (functionCalls || text.length > 0) {\n messages.push({\n role: 'assistant',\n content: text,\n functionCalls,\n })\n }\n\n break\n }\n case 'tool': {\n for (const part of content) {\n messages.push({\n role: 'function' as const,\n functionId: part.toolCallId,\n result: JSON.stringify(part.result, null, 2),\n })\n }\n break\n }\n default: {\n const _exhaustiveCheck: never = role\n throw new Error(`Unsupported role: ${_exhaustiveCheck}`)\n }\n }\n }\n\n return messages\n}\n\nfunction mapAxFinishReason(\n finishReason: AxChatResponseResult['finishReason']\n): LanguageModelV1FinishReason {\n switch (finishReason) {\n case 'stop':\n return 'stop'\n case 'length':\n return 'length'\n case 'function_call':\n return 'tool-calls'\n default:\n return 'other'\n }\n}\n\nfunction createChatRequest({\n mode,\n prompt,\n maxTokens,\n temperature,\n topP,\n frequencyPenalty,\n presencePenalty,\n //seed,\n}: Readonly<Parameters<LanguageModelV1['doGenerate']>[0]>): {\n req: AxChatRequest\n warnings: LanguageModelV1CallWarning[]\n} {\n const req: AxChatRequest = {\n chatPrompt: convertToAxChatPrompt(prompt),\n ...(frequencyPenalty != null ? { frequencyPenalty } : {}),\n ...(presencePenalty != null ? { presencePenalty } : {}),\n ...(maxTokens != null ? { maxTokens } : {}),\n ...(temperature != null ? { temperature } : {}),\n ...(topP != null ? { topP } : {}),\n }\n\n const warnings: LanguageModelV1CallWarning[] = []\n\n switch (mode.type) {\n case 'regular': {\n return {\n req: { ...req, ...prepareToolsAndToolChoice(mode) },\n warnings,\n }\n }\n\n case 'object-json': {\n return {\n req,\n warnings,\n }\n }\n\n case 'object-tool': {\n const tool = {\n type: 'function',\n function: {\n name: mode.tool.name,\n params: mode.tool.parameters,\n },\n }\n return {\n req: { ...req, ...tool },\n warnings,\n }\n }\n\n default: {\n throw new Error(`Unsupported type`)\n }\n }\n}\n\nclass AxToSDKTransformer extends TransformStream<\n AxChatResponse,\n LanguageModelV1StreamPart\n> {\n private usage: Extract<\n LanguageModelV1StreamPart,\n { type: 'finish' }\n >['usage'] = {\n promptTokens: 0,\n completionTokens: 0,\n }\n\n private finishReason: Extract<\n LanguageModelV1StreamPart,\n { type: 'finish' }\n >['finishReason'] = 'other'\n\n private functionCalls: LanguageModelV1FunctionToolCall[] = []\n\n constructor() {\n const transformer = {\n transform: (\n chunk: Readonly<AxChatResponse>,\n controller: TransformStreamDefaultController<LanguageModelV1StreamPart>\n ) => {\n const choice = chunk.results.at(0)\n if (!choice) {\n const val = {\n type: 'finish' as const,\n finishReason: this.finishReason,\n usage: this.usage,\n }\n controller.enqueue(val)\n return\n }\n\n if (chunk.modelUsage) {\n this.usage = {\n promptTokens:\n this.usage.promptTokens + chunk.modelUsage.promptTokens,\n completionTokens:\n this.usage.completionTokens + chunk.modelUsage?.completionTokens,\n }\n }\n\n if (choice.functionCalls) {\n for (const fc of choice.functionCalls) {\n const index = this.functionCalls.findIndex(\n (f) => f.toolCallId === fc.id\n )\n if (index === -1) {\n this.functionCalls.push({\n toolCallType: 'function' as const,\n toolCallId: fc.id,\n toolName: fc.function.name,\n args:\n typeof fc.function.params === 'string'\n ? fc.function.params\n : JSON.stringify(fc.function.params),\n })\n } else {\n const obj = this.functionCalls[index]\n if (!obj) {\n continue\n }\n if (typeof fc.function.params === 'string') {\n obj.args = (obj.args ?? '') + fc.function.params\n } else {\n obj.args = JSON.stringify(fc.function.params)\n }\n }\n this.finishReason = 'tool-calls'\n }\n }\n\n if (choice.content && choice.content.length > 0) {\n controller.enqueue({\n type: 'text-delta',\n textDelta: choice.content ?? '',\n })\n this.finishReason = mapAxFinishReason(choice.finishReason)\n }\n },\n flush: (\n controller: TransformStreamDefaultController<LanguageModelV1StreamPart>\n ) => {\n for (const fc of this.functionCalls) {\n const tc = {\n type: 'tool-call' as const,\n ...fc,\n }\n controller.enqueue(tc)\n }\n\n const val = {\n type: 'finish' as const,\n finishReason: this.finishReason,\n usage: this.usage,\n }\n controller.enqueue(val)\n controller.terminate()\n },\n }\n\n super(transformer)\n }\n}\n\ntype AnyZod =\n | z.AnyZodObject\n | z.ZodString\n | z.ZodNumber\n | z.ZodBoolean\n | z.ZodArray<AnyZod>\n | z.ZodOptional<AnyZod>\n\nfunction convertToZodSchema(\n jsonSchema: Readonly<AxFunctionJSONSchema>\n): AnyZod {\n const { type, properties, required, items } = jsonSchema\n\n switch (type) {\n case 'string':\n return z.string()\n case 'number':\n return z.number()\n case 'boolean':\n return z.boolean()\n case 'array':\n if (!items) {\n throw new Error(\"Array type must have 'items' property.\")\n }\n return z.array(convertToZodSchema(items))\n case 'object': {\n if (!properties) {\n throw new Error(\"Object type must have 'properties' property.\")\n }\n const shape: Record<string, AnyZod> = {}\n\n for (const [key, value] of Object.entries(properties)) {\n const schema = convertToZodSchema(value)\n let val = required?.includes(key) ? schema : schema.optional()\n val = value.description ? val.describe(value.description) : val\n shape[key] = val\n }\n return z.object(shape)\n }\n default:\n throw new Error(`Unsupported type: ${type}`)\n }\n}\n","import crypto from 'crypto'\n\nimport { urlAlphabet } from './url-alphabet/index.js'\n\n// It is best to make fewer, larger requests to the crypto module to\n// avoid system call overhead. So, random numbers are generated in a\n// pool. The pool is a Buffer that is larger than the initial random\n// request size by this multiplier. The pool is enlarged if subsequent\n// requests exceed the maximum buffer size.\nconst POOL_SIZE_MULTIPLIER = 128\nlet pool, poolOffset\n\nlet fillPool = bytes => {\n if (!pool || pool.length < bytes) {\n pool = Buffer.allocUnsafe(bytes * POOL_SIZE_MULTIPLIER)\n crypto.randomFillSync(pool)\n poolOffset = 0\n } else if (poolOffset + bytes > pool.length) {\n crypto.randomFillSync(pool)\n poolOffset = 0\n }\n poolOffset += bytes\n}\n\nlet random = bytes => {\n // `|=` convert `bytes` to number to prevent `valueOf` abusing and pool pollution\n fillPool((bytes |= 0))\n return pool.subarray(poolOffset - bytes, poolOffset)\n}\n\nlet customRandom = (alphabet, defaultSize, getRandom) => {\n // First, a bitmask is necessary to generate the ID. The bitmask makes bytes\n // values closer to the alphabet size. The bitmask calculates the closest\n // `2^31 - 1` number, which exceeds the alphabet size.\n // For example, the bitmask for the alphabet size 30 is 31 (00011111).\n let mask = (2 << (31 - Math.clz32((alphabet.length - 1) | 1))) - 1\n // Though, the bitmask solution is not perfect since the bytes exceeding\n // the alphabet size are refused. Therefore, to reliably generate the ID,\n // the random bytes redundancy has to be satisfied.\n\n // Note: every hardware random generator call is performance expensive,\n // because the system call for entropy collection takes a lot of time.\n // So, to avoid additional system calls, extra bytes are requested in advance.\n\n // Next, a step determines how many random bytes to generate.\n // The number of random bytes gets decided upon the ID size, mask,\n // alphabet size, and magic number 1.6 (using 1.6 peaks at performance\n // according to benchmarks).\n let step = Math.ceil((1.6 * mask * defaultSize) / alphabet.length)\n\n return (size = defaultSize) => {\n let id = ''\n while (true) {\n let bytes = getRandom(step)\n // A compact alternative for `for (let i = 0; i < step; i++)`.\n let i = step\n while (i--) {\n // Adding `|| ''` refuses a random byte that exceeds the alphabet size.\n id += alphabet[bytes[i] & mask] || ''\n if (id.length === size) return id\n }\n }\n }\n}\n\nlet customAlphabet = (alphabet, size = 21) =>\n customRandom(alphabet, size, random)\n\nlet nanoid = (size = 21) => {\n // `|=` convert `size` to number to prevent `valueOf` abusing and pool pollution\n fillPool((size |= 0))\n let id = ''\n // We are reading directly from the random pool to avoid creating new array\n for (let i = poolOffset - size; i < poolOffset; i++) {\n // It is incorrect to use bytes exceeding the alphabet size.\n // The following mask reduces the random byte in the 0-255 value\n // range to the 0-63 value range. Therefore, adding hacks, such\n // as empty string fallback or magic numbers, is unneccessary because\n // the bitmask trims bytes down to the alphabet size.\n id += urlAlphabet[pool[i] & 63]\n }\n return id\n}\n\nexport { nanoid, customAlphabet, customRandom, urlAlphabet, random }\n"],"mappings":";AAEA;AAAA,EAEE;AAAA,OAEK;;;ACNP,OAAO,YAAY;AASnB,IAAM,uBAAuB;AAC7B,IAAI;AAAJ,IAAU;AAEV,IAAI,WAAW,WAAS;AACtB,MAAI,CAAC,QAAQ,KAAK,SAAS,OAAO;AAChC,WAAO,OAAO,YAAY,QAAQ,oBAAoB;AACtD,WAAO,eAAe,IAAI;AAC1B,iBAAa;AAAA,EACf,WAAW,aAAa,QAAQ,KAAK,QAAQ;AAC3C,WAAO,eAAe,IAAI;AAC1B,iBAAa;AAAA,EACf;AACA,gBAAc;AAChB;AAEA,IAAI,SAAS,WAAS;AAEpB,WAAU,SAAS,CAAE;AACrB,SAAO,KAAK,SAAS,aAAa,OAAO,UAAU;AACrD;AAEA,IAAI,eAAe,CAAC,UAAU,aAAa,cAAc;AAKvD,MAAI,QAAQ,KAAM,KAAK,KAAK,MAAO,SAAS,SAAS,IAAK,CAAC,KAAM;AAajE,MAAI,OAAO,KAAK,KAAM,MAAM,OAAO,cAAe,SAAS,MAAM;AAEjE,SAAO,CAAC,OAAO,gBAAgB;AAC7B,QAAI,KAAK;AACT,WAAO,MAAM;AACX,UAAI,QAAQ,UAAU,IAAI;AAE1B,UAAI,IAAI;AACR,aAAO,KAAK;AAEV,cAAM,SAAS,MAAM,CAAC,IAAI,IAAI,KAAK;AACnC,YAAI,GAAG,WAAW,KAAM,QAAO;AAAA,MACjC;AAAA,IACF;AAAA,EACF;AACF;AAEA,IAAI,iBAAiB,CAAC,UAAU,OAAO,OACrC,aAAa,UAAU,MAAM,MAAM;;;ADnCrC,SAAS,SAAS;AAiBlB,IAAM,SAAS;AAAA,EACb;AAAA,EACA;AACF;AAEO,IAAM,kBAAN,MAAgE;AAAA,EACpD;AAAA,EACA;AAAA,EACT;AAAA,EACA;AAAA,EAER,YAAY;AAAA,IACV;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAKI;AACF,SAAK,WAAW,MAAM,YAAY;AAClC,SAAK,mBAAmB;AACxB,SAAK,cAAc;AACnB,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,IAAI,cAAc;AAChB,WAAO,KAAK,SAAS;AAAA,EACvB;AAAA,EAEA,IAAI,aAA2B;AAC7B,UAAM,SAAS,KAAK,SAAS,cAAc;AAAA,MACzC,MAAM;AAAA,MACN,YAAY,CAAC;AAAA,IACf;AAEA,WAAO,mBAAmB,MAAM;AAAA,EAClC;AAAA,EAEA,IAAI,WAAyB;AAC3B,UAAM,KAAK,OAAO,UAAc;AAC9B,YAAM,MAAO,MAAM,KAAK,SAAS,KAAK,KAAK;AAC3C,YAAM,aAAa,OAAO;AAE1B,WAAK,YAAY;AAAA,QACf;AAAA,UACE,MAAM;AAAA,UACN,SAAS;AAAA,YACP;AAAA,cACE,MAAM;AAAA,cACN,UAAU,KAAK,SAAS;AAAA,cACxB;AAAA,cACA,MAAM;AAAA,YACR;AAAA,UACF;AAAA,QACF;AAAA,QACA;AAAA,UACE,MAAM;AAAA,UACN,SAAS;AAAA,YACP;AAAA,cACE,MAAM;AAAA,cACN,UAAU,KAAK,SAAS;AAAA,cACxB;AAAA,cACA,QAAQ;AAAA,YACV;AAAA,UACF;AAAA,QACF;AAAA,MACF,CAAC;AAED,aAAO,KAAK,iBAAiB,GAAG;AAAA,IAClC;AACA,WAAO;AAAA,EACT;AACF;AAEO,IAAM,eAAN,MAA8C;AAAA,EAC1C,uBAAuB;AAAA,EACvB,8BAA8B;AAAA,EAEtB;AAAA,EACA;AAAA,EAEV;AAAA,EAEP,YAAY,IAAiB,QAA6B;AACxD,SAAK,KAAK;AACV,SAAK,SAAS;AACd,SAAK,UAAU,KAAK,GAAG,aAAa,EAAE;AAAA,EACxC;AAAA,EAEA,IAAI,WAAmB;AACrB,WAAO,KAAK,GAAG,QAAQ;AAAA,EACzB;AAAA,EAEA,MAAM,WACJ,SAC6D;AAC7D,UAAM,EAAE,KAAK,SAAS,IAAI,kBAAkB,OAAO;AACnD,UAAM,MAAO,MAAM,KAAK,GAAG,KAAK,GAAG;AACnC,UAAM,SAAS,IAAI,QAAQ,GAAG,CAAC;AAE/B,QAAI,CAAC,QAAQ;AACX,YAAM,IAAI,MAAM,oBAAoB;AAAA,IACtC;AAEA,WAAO;AAAA,MACL,MAAM,OAAO,WAAW;AAAA,MACxB,WAAW,OAAO,eAAe,IAAI,CAAC,QAAQ;AAAA,QAC5C,cAAc;AAAA,QACd,YAAY,GAAG;AAAA,QACf,UAAU,GAAG,SAAS;AAAA,QACtB,MACE,OAAO,GAAG,SAAS,WAAW,WAC1B,GAAG,SAAS,SACZ,KAAK,UAAU,GAAG,SAAS,MAAM;AAAA,MACzC,EAAE;AAAA,MACF,cAAc,kBAAkB,OAAO,YAAY;AAAA,MACnD,OAAO;AAAA,QACL,cAAc,IAAI,YAAY,gBAAgB;AAAA,QAC9C,kBAAkB,IAAI,YAAY,oBAAoB;AAAA,MACxD;AAAA,MACA,SAAS,EAAE,WAAW,IAAI,aAAa,IAAI,eAAe,CAAC,EAAE;AAAA,MAC7D;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,SACJ,SAC2D;AAC3D,UAAM,EAAE,KAAK,SAAS,IAAI,kBAAkB,OAAO;AAEnD,UAAM,MAAO,MAAM,KAAK,GAAG,KAAK,KAAK;AAAA,MACnC,QAAQ;AAAA,IACV,CAAC;AAED,WAAO;AAAA,MACL,QAAQ,IAAI,YAAY,IAAI,mBAAmB,CAAC;AAAA,MAChD,SAAS,EAAE,WAAW,IAAI,aAAa,IAAI,eAAe,CAAC,EAAE;AAAA,MAC7D;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,0BACP,MAGmD;AAEnD,MAAI,CAAC,KAAK,SAAS,KAAK,MAAM,WAAW,GAAG;AAC1C,WAAO,CAAC;AAAA,EACV;AAEA,QAAM,QAAQ,KAAK;AACnB,QAAM,YAAY,MAAM,IAAI,CAAC,OAAO;AAAA,IAClC,MAAM,EAAE;AAAA,IACR,aAAa,iBAAiB,IAAK,EAAE,eAAe,KAAM;AAAA,IAC1D,YAAY,EAAE;AAAA,EAChB,EAAE;AAEF,QAAM,aAAa,KAAK;AACxB,MAAI,CAAC,YAAY;AACf,WAAO,EAAE,UAAU;AAAA,EACrB;AAEA,QAAM,OAAO,WAAW;AAExB,UAAQ,MAAM;AAAA,IACZ,KAAK;AACH,aAAO,EAAE,WAAW,cAAc,OAAO;AAAA,IAC3C,KAAK;AACH,aAAO,EAAE,WAAW,cAAc,OAAO;AAAA,IAC3C,KAAK;AACH,aAAO,EAAE,WAAW,cAAc,WAAW;AAAA,IAC/C,KAAK;AACH,aAAO;AAAA,QACL;AAAA,QACA,cAAc;AAAA,UACZ,MAAM;AAAA,UACN,UAAU,EAAE,MAAM,WAAW,SAAS;AAAA,QACxC;AAAA,MACF;AAAA,IACF,SAAS;AACP,YAAM,mBAA0B;AAChC,YAAM,IAAI,MAAM,iCAAiC,gBAAgB,EAAE;AAAA,IACrE;AAAA,EACF;AACF;AAEA,SAAS,sBACP,QAC6B;AAC7B,QAAM,WAAwC,CAAC;AAE/C,aAAW,EAAE,MAAM,QAAQ,KAAK,QAAQ;AACtC,YAAQ,MAAM;AAAA,MACZ,KAAK,UAAU;AACb,iBAAS,KAAK,EAAE,MAAM,UAAU,QAAQ,CAAC;AACzC;AAAA,MACF;AAAA,MAEA,KAAK,QAAQ;AACX,iBAAS,KAAK;AAAA,UACZ,MAAM;AAAA,UACN,SAAS,QAAQ,IAAI,CAAC,SAAS;AAC7B,oBAAQ,KAAK,MAAM;AAAA,cACjB,KAAK,QAAQ;AACX,uBAAO,EAAE,MAAM,QAAQ,MAAM,KAAK,KAAK;AAAA,cACzC;AAAA,cACA,KAAK,SAAS;AACZ,oBAAI,CAAC,KAAK,UAAU;AAClB,wBAAM,IAAI,MAAM,iCAAiC;AAAA,gBACnD;AACA,oBAAI,CAAC,YAAY,OAAO,KAAK,KAAK,GAAG;AACnC,wBAAM,IAAI,MAAM,qCAAqC;AAAA,gBACvD;AACA,sBAAM,QAAQ,OAAO,KAAK,KAAK,KAAK,EAAE,SAAS,QAAQ;AACvD,uBAAO;AAAA,kBACL,MAAM;AAAA,kBACN,UAAU,KAAK;AAAA,kBACf;AAAA,gBACF;AAAA,cACF;AAAA,cACA;AACE,sBAAM,IAAI,MAAM,qBAAqB,IAAI,EAAE;AAAA,YAe/C;AAAA,UACF,CAAC;AAAA,QACH,CAAC;AACD;AAAA,MACF;AAAA,MAEA,KAAK,aAAa;AAChB,YAAI,OAAO;AACX,cAAM,YAGe,CAAC;AAEtB,mBAAW,QAAQ,SAAS;AAC1B,kBAAQ,KAAK,MAAM;AAAA,YACjB,KAAK,QAAQ;AACX,sBAAQ,KAAK;AACb;AAAA,YACF;AAAA,YACA,KAAK,aAAa;AAChB,wBAAU,KAAK;AAAA,gBACb,IAAI,KAAK;AAAA,gBACT,MAAM;AAAA,gBACN,UAAU;AAAA,kBACR,MAAM,KAAK;AAAA,kBACX,QAAQ,KAAK;AAAA,gBACf;AAAA,cACF,CAAC;AACD;AAAA,YACF;AAAA,YAEA,SAAS;AACP,oBAAM,mBAA0B;AAChC,oBAAM,IAAI,MAAM,qBAAqB,gBAAgB,EAAE;AAAA,YACzD;AAAA,UACF;AAAA,QACF;AAEA,cAAM,gBAAgB,UAAU,WAAW,IAAI,SAAY;AAE3D,YAAI,iBAAiB,KAAK,SAAS,GAAG;AACpC,mBAAS,KAAK;AAAA,YACZ,MAAM;AAAA,YACN,SAAS;AAAA,YACT;AAAA,UACF,CAAC;AAAA,QACH;AAEA;AAAA,MACF;AAAA,MACA,KAAK,QAAQ;AACX,mBAAW,QAAQ,SAAS;AAC1B,mBAAS,KAAK;AAAA,YACZ,MAAM;AAAA,YACN,YAAY,KAAK;AAAA,YACjB,QAAQ,KAAK,UAAU,KAAK,QAAQ,MAAM,CAAC;AAAA,UAC7C,CAAC;AAAA,QACH;AACA;AAAA,MACF;AAAA,MACA,SAAS;AACP,cAAM,mBAA0B;AAChC,cAAM,IAAI,MAAM,qBAAqB,gBAAgB,EAAE;AAAA,MACzD;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AACT;AAEA,SAAS,kBACP,cAC6B;AAC7B,UAAQ,cAAc;AAAA,IACpB,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AACH,aAAO;AAAA,IACT,KAAK;AACH,aAAO;AAAA,IACT;AACE,aAAO;AAAA,EACX;AACF;AAEA,SAAS,kBAAkB;AAAA,EACzB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAEF,GAGE;AACA,QAAM,MAAqB;AAAA,IACzB,YAAY,sBAAsB,MAAM;AAAA,IACxC,GAAI,oBAAoB,OAAO,EAAE,iBAAiB,IAAI,CAAC;AAAA,IACvD,GAAI,mBAAmB,OAAO,EAAE,gBAAgB,IAAI,CAAC;AAAA,IACrD,GAAI,aAAa,OAAO,EAAE,UAAU,IAAI,CAAC;AAAA,IACzC,GAAI,eAAe,OAAO,EAAE,YAAY,IAAI,CAAC;AAAA,IAC7C,GAAI,QAAQ,OAAO,EAAE,KAAK,IAAI,CAAC;AAAA,EACjC;AAEA,QAAM,WAAyC,CAAC;AAEhD,UAAQ,KAAK,MAAM;AAAA,IACjB,KAAK,WAAW;AACd,aAAO;AAAA,QACL,KAAK,EAAE,GAAG,KAAK,GAAG,0BAA0B,IAAI,EAAE;AAAA,QAClD;AAAA,MACF;AAAA,IACF;AAAA,IAEA,KAAK,eAAe;AAClB,aAAO;AAAA,QACL;AAAA,QACA;AAAA,MACF;AAAA,IACF;AAAA,IAEA,KAAK,eAAe;AAClB,YAAM,OAAO;AAAA,QACX,MAAM;AAAA,QACN,UAAU;AAAA,UACR,MAAM,KAAK,KAAK;AAAA,UAChB,QAAQ,KAAK,KAAK;AAAA,QACpB;AAAA,MACF;AACA,aAAO;AAAA,QACL,KAAK,EAAE,GAAG,KAAK,GAAG,KAAK;AAAA,QACvB;AAAA,MACF;AAAA,IACF;AAAA,IAEA,SAAS;AACP,YAAM,IAAI,MAAM,kBAAkB;AAAA,IACpC;AAAA,EACF;AACF;AAEA,IAAM,qBAAN,cAAiC,gBAG/B;AAAA,EACQ,QAGK;AAAA,IACX,cAAc;AAAA,IACd,kBAAkB;AAAA,EACpB;AAAA,EAEQ,eAGY;AAAA,EAEZ,gBAAmD,CAAC;AAAA,EAE5D,cAAc;AACZ,UAAM,cAAc;AAAA,MAClB,WAAW,CACT,OACA,eACG;AACH,cAAM,SAAS,MAAM,QAAQ,GAAG,CAAC;AACjC,YAAI,CAAC,QAAQ;AACX,gBAAM,MAAM;AAAA,YACV,MAAM;AAAA,YACN,cAAc,KAAK;AAAA,YACnB,OAAO,KAAK;AAAA,UACd;AACA,qBAAW,QAAQ,GAAG;AACtB;AAAA,QACF;AAEA,YAAI,MAAM,YAAY;AACpB,eAAK,QAAQ;AAAA,YACX,cACE,KAAK,MAAM,eAAe,MAAM,WAAW;AAAA,YAC7C,kBACE,KAAK,MAAM,mBAAmB,MAAM,YAAY;AAAA,UACpD;AAAA,QACF;AAEA,YAAI,OAAO,eAAe;AACxB,qBAAW,MAAM,OAAO,eAAe;AACrC,kBAAM,QAAQ,KAAK,cAAc;AAAA,cAC/B,CAAC,MAAM,EAAE,eAAe,GAAG;AAAA,YAC7B;AACA,gBAAI,UAAU,IAAI;AAChB,mBAAK,cAAc,KAAK;AAAA,gBACtB,cAAc;AAAA,gBACd,YAAY,GAAG;AAAA,gBACf,UAAU,GAAG,SAAS;AAAA,gBACtB,MACE,OAAO,GAAG,SAAS,WAAW,WAC1B,GAAG,SAAS,SACZ,KAAK,UAAU,GAAG,SAAS,MAAM;AAAA,cACzC,CAAC;AAAA,YACH,OAAO;AACL,oBAAM,MAAM,KAAK,cAAc,KAAK;AACpC,kBAAI,CAAC,KAAK;AACR;AAAA,cACF;AACA,kBAAI,OAAO,GAAG,SAAS,WAAW,UAAU;AAC1C,oBAAI,QAAQ,IAAI,QAAQ,MAAM,GAAG,SAAS;AAAA,cAC5C,OAAO;AACL,oBAAI,OAAO,KAAK,UAAU,GAAG,SAAS,MAAM;AAAA,cAC9C;AAAA,YACF;AACA,iBAAK,eAAe;AAAA,UACtB;AAAA,QACF;AAEA,YAAI,OAAO,WAAW,OAAO,QAAQ,SAAS,GAAG;AAC/C,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN,WAAW,OAAO,WAAW;AAAA,UAC/B,CAAC;AACD,eAAK,eAAe,kBAAkB,OAAO,YAAY;AAAA,QAC3D;AAAA,MACF;AAAA,MACA,OAAO,CACL,eACG;AACH,mBAAW,MAAM,KAAK,eAAe;AACnC,gBAAM,KAAK;AAAA,YACT,MAAM;AAAA,YACN,GAAG;AAAA,UACL;AACA,qBAAW,QAAQ,EAAE;AAAA,QACvB;AAEA,cAAM,MAAM;AAAA,UACV,MAAM;AAAA,UACN,cAAc,KAAK;AAAA,UACnB,OAAO,KAAK;AAAA,QACd;AACA,mBAAW,QAAQ,GAAG;AACtB,mBAAW,UAAU;AAAA,MACvB;AAAA,IACF;AAEA,UAAM,WAAW;AAAA,EACnB;AACF;AAUA,SAAS,mBACP,YACQ;AACR,QAAM,EAAE,MAAM,YAAY,UAAU,MAAM,IAAI;AAE9C,UAAQ,MAAM;AAAA,IACZ,KAAK;AACH,aAAO,EAAE,OAAO;AAAA,IAClB,KAAK;AACH,aAAO,EAAE,OAAO;AAAA,IAClB,KAAK;AACH,aAAO,EAAE,QAAQ;AAAA,IACnB,KAAK;AACH,UAAI,CAAC,OAAO;AACV,cAAM,IAAI,MAAM,wCAAwC;AAAA,MAC1D;AACA,aAAO,EAAE,MAAM,mBAAmB,KAAK,CAAC;AAAA,IAC1C,KAAK,UAAU;AACb,UAAI,CAAC,YAAY;AACf,cAAM,IAAI,MAAM,8CAA8C;AAAA,MAChE;AACA,YAAM,QAAgC,CAAC;AAEvC,iBAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,UAAU,GAAG;AACrD,cAAM,SAAS,mBAAmB,KAAK;AACvC,YAAI,MAAM,UAAU,SAAS,GAAG,IAAI,SAAS,OAAO,SAAS;AAC7D,cAAM,MAAM,cAAc,IAAI,SAAS,MAAM,WAAW,IAAI;AAC5D,cAAM,GAAG,IAAI;AAAA,MACf;AACA,aAAO,EAAE,OAAO,KAAK;AAAA,IACvB;AAAA,IACA;AACE,YAAM,IAAI,MAAM,qBAAqB,IAAI,EAAE;AAAA,EAC/C;AACF;","names":[]}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@ax-llm/ax-ai-sdk-provider",
3
- "version": "11.0.17",
3
+ "version": "11.0.19",
4
4
  "type": "module",
5
5
  "description": "Ax AI SDK Provider for the Vercel AI SDK",
6
6
  "repository": {
@@ -14,7 +14,7 @@
14
14
  "keywords": [],
15
15
  "dependencies": {
16
16
  "@ai-sdk/provider-utils": "^2.0.7 ",
17
- "@ax-llm/ax": "11.0.17",
17
+ "@ax-llm/ax": "11.0.19",
18
18
  "ai": "^4.0.33",
19
19
  "zod": "^3.23.8"
20
20
  },