@ax-llm/ax-ai-sdk-provider 9.0.34 → 9.0.36

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/cjs/index.d.ts CHANGED
@@ -1,16 +1,28 @@
1
1
  import { type LanguageModelV1, type LanguageModelV1FinishReason } from '@ai-sdk/provider';
2
- import type { AxAIService, AxChatResponseResult } from '@ax-llm/ax/index.js';
3
- type AxChatConfig = {
4
- generateId: () => string;
2
+ import type { AxAgent, AxAIService, AxChatResponseResult, AxFunctionJSONSchema, AxGenIn, AxGenOut } from '@ax-llm/ax/index.js';
3
+ type AxConfig = {
5
4
  fetch?: typeof fetch;
6
5
  };
7
- export declare class AxAgentFramework implements LanguageModelV1 {
6
+ interface RenderTool<IN> {
7
+ description?: string;
8
+ parameters?: AxFunctionJSONSchema;
9
+ generate?: ((input: IN) => Promise<unknown>) | undefined;
10
+ }
11
+ export declare class AxAgentProvider<IN extends AxGenIn> implements RenderTool<IN> {
12
+ private readonly config?;
13
+ private readonly funcInfo;
14
+ constructor(agent: Readonly<AxAgent<IN, AxGenOut>>, config?: Readonly<AxConfig>);
15
+ get description(): string;
16
+ get parameters(): AxFunctionJSONSchema;
17
+ get generate(): ((input: IN) => Promise<unknown>) | undefined;
18
+ }
19
+ export declare class AxAIProvider implements LanguageModelV1 {
8
20
  readonly specificationVersion = "v1";
9
21
  readonly defaultObjectGenerationMode = "json";
10
22
  private readonly ai;
11
- private readonly config;
23
+ private readonly config?;
12
24
  modelId: string;
13
- constructor(ai: AxAIService, config: Readonly<AxChatConfig>);
25
+ constructor(ai: AxAIService, config?: Readonly<AxConfig>);
14
26
  get provider(): string;
15
27
  doGenerate(options: Readonly<Parameters<LanguageModelV1['doGenerate']>[0]>): Promise<Awaited<ReturnType<LanguageModelV1['doGenerate']>>>;
16
28
  doStream(options: Readonly<Parameters<LanguageModelV1['doStream']>[0]>): Promise<Awaited<ReturnType<LanguageModelV1['doStream']>>>;
package/cjs/index.js CHANGED
@@ -1,10 +1,37 @@
1
1
  "use strict";
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.AxAgentFramework = void 0;
3
+ exports.AxAIProvider = exports.AxAgentProvider = void 0;
4
4
  exports.mapAxFinishReason = mapAxFinishReason;
5
5
  const web_1 = require("stream/web");
6
6
  const provider_1 = require("@ai-sdk/provider");
7
- class AxAgentFramework {
7
+ class AxAgentProvider {
8
+ config;
9
+ funcInfo;
10
+ constructor(agent, config) {
11
+ this.config = config;
12
+ this.funcInfo = agent.getFunction();
13
+ }
14
+ get description() {
15
+ return this.funcInfo.description;
16
+ }
17
+ get parameters() {
18
+ return (this.funcInfo.parameters ?? {
19
+ type: 'object',
20
+ properties: {}
21
+ });
22
+ }
23
+ get generate() {
24
+ if (!this.funcInfo.func) {
25
+ return undefined;
26
+ }
27
+ const agentFunc = this.funcInfo.func;
28
+ return async (input) => {
29
+ return (await agentFunc(input));
30
+ };
31
+ }
32
+ }
33
+ exports.AxAgentProvider = AxAgentProvider;
34
+ class AxAIProvider {
8
35
  specificationVersion = 'v1';
9
36
  defaultObjectGenerationMode = 'json';
10
37
  ai;
@@ -44,7 +71,9 @@ class AxAgentFramework {
44
71
  }
45
72
  async doStream(options) {
46
73
  const { req, warnings } = createChatRequest(options);
47
- const res = (await this.ai.chat(req));
74
+ const res = (await this.ai.chat(req, {
75
+ stream: true
76
+ }));
48
77
  return {
49
78
  stream: res.pipeThrough(new AxToSDKTransformer()),
50
79
  rawCall: { rawPrompt: '', rawSettings: req.modelConfig ?? {} },
@@ -52,7 +81,7 @@ class AxAgentFramework {
52
81
  };
53
82
  }
54
83
  }
55
- exports.AxAgentFramework = AxAgentFramework;
84
+ exports.AxAIProvider = AxAIProvider;
56
85
  function prepareToolsAndToolChoice(mode) {
57
86
  // when the tools array is empty, change it to undefined to prevent errors:
58
87
  const tools = mode.tools?.length ? mode.tools : undefined;
@@ -145,11 +174,14 @@ function convertToAxChatPrompt(prompt) {
145
174
  }
146
175
  }
147
176
  }
148
- messages.push({
149
- role: 'assistant',
150
- content: text,
151
- functionCalls: toolCalls
152
- });
177
+ const functionCalls = toolCalls.length === 0 ? undefined : toolCalls;
178
+ if (functionCalls || text.length > 0) {
179
+ messages.push({
180
+ role: 'assistant',
181
+ content: text,
182
+ functionCalls
183
+ });
184
+ }
153
185
  break;
154
186
  }
155
187
  case 'tool': {
@@ -229,35 +261,47 @@ class AxToSDKTransformer extends web_1.TransformStream {
229
261
  completionTokens: 0
230
262
  };
231
263
  finishReason = 'other';
232
- transform(chunk, controller) {
233
- const choice = chunk.results.at(0);
234
- if (!choice) {
235
- throw new Error('No choice returned');
236
- }
237
- if (choice.functionCalls) {
238
- for (const tc of choice.functionCalls) {
264
+ constructor() {
265
+ const transformer = {
266
+ transform: (chunk, controller) => {
267
+ const choice = chunk.results.at(0);
268
+ if (!choice) {
269
+ return;
270
+ }
271
+ if (choice.functionCalls) {
272
+ for (const tc of choice.functionCalls) {
273
+ controller.enqueue({
274
+ type: 'tool-call',
275
+ toolCallType: 'function',
276
+ toolCallId: tc.id,
277
+ toolName: tc.function.name,
278
+ args: JSON.stringify(tc.function.params)
279
+ });
280
+ this.finishReason = 'tool-calls';
281
+ }
282
+ }
283
+ if (choice.content && choice.content.length > 0) {
284
+ controller.enqueue({
285
+ type: 'text-delta',
286
+ textDelta: choice.content ?? ''
287
+ });
288
+ }
289
+ this.finishReason = mapAxFinishReason(choice.finishReason);
290
+ },
291
+ flush: (controller) => {
239
292
  controller.enqueue({
240
- type: 'tool-call',
241
- toolCallType: 'function',
242
- toolCallId: tc.id,
243
- toolName: tc.function.name,
244
- args: JSON.stringify(tc.function.params)
293
+ type: 'finish',
294
+ finishReason: this.finishReason,
295
+ usage: this.usage
245
296
  });
246
- this.finishReason = 'tool-calls';
247
297
  }
248
- }
249
- controller.enqueue({
250
- type: 'text-delta',
251
- textDelta: choice.content ?? ''
252
- });
253
- this.finishReason = mapAxFinishReason(choice.finishReason);
254
- }
255
- flush(controller) {
256
- controller.enqueue({
257
- type: 'finish',
258
- finishReason: this.finishReason,
259
- usage: this.usage
260
- });
298
+ };
299
+ super(transformer);
300
+ this.usage = {
301
+ promptTokens: 0,
302
+ completionTokens: 0
303
+ };
304
+ this.finishReason = 'other';
261
305
  }
262
306
  }
263
307
  //# sourceMappingURL=index.js.map
package/cjs/index.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../index.ts"],"names":[],"mappings":";;;AA2OA,8CAaC;AAxPD,oCAIoB;AAEpB,+CAO0B;AAgB1B,MAAa,gBAAgB;IAClB,oBAAoB,GAAG,IAAI,CAAC;IAC5B,2BAA2B,GAAG,MAAM,CAAC;IAE7B,EAAE,CAAc;IAChB,MAAM,CAAe;IAE/B,OAAO,CAAS;IAEvB,YAAY,EAAe,EAAE,MAA8B;QACzD,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC;IAC7C,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,UAAU,CACd,OAA+D;QAE/D,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAmB,CAAC;QACxD,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAEjC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC;QAED,OAAO;YACL,IAAI,EAAE,MAAM,CAAC,OAAO,IAAI,SAAS;YACjC,SAAS,EAAE,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC5C,YAAY,EAAE,UAAU;gBACxB,UAAU,EAAE,EAAE,CAAC,EAAE;gBACjB,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;gBAC1B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAO,CAAC;aAC1C,CAAC,CAAC;YACH,YAAY,EAAE,iBAAiB,CAAC,MAAM,CAAC,YAAY,CAAC;YACpD,KAAK,EAAE;gBACL,YAAY,EAAE,GAAG,CAAC,UAAU,EAAE,YAAY,IAAI,CAAC;gBAC/C,gBAAgB,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,IAAI,CAAC;aACxD;YACD,OAAO,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE,EAAE;YAC9D,QAAQ;SACT,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,OAA6D;QAE7D,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAmC,CAAC;QAExE,OAAO;YACL,MAAM,EAAE,GAAG,CAAC,WAAW,CAAC,IAAI,kBAAkB,EAAE,CAAC;YACjD,OAAO,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE,EAAE;YAC9D,QAAQ;SACT,CAAC;IACJ,CAAC;CACF;AA5DD,4CA4DC;AAED,SAAS,yBAAyB,CAChC,IAEC;IAED,2EAA2E;IAC3E,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;IAE1D,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;QAClB,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC;IACtD,CAAC;IAED,MAAM,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACvC,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE;YACR,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B;KACF,CAAC,CAAC,CAAC;IAEJ,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;IAEnC,IAAI,UAAU,IAAI,IAAI,EAAE,CAAC;QACvB,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC;IACxD,CAAC;IAED,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;IAE7B,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,MAAM,CAAC;QACZ,KAAK,MAAM;YACT,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;QACnD,KAAK,UAAU;YACb,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QAEpD,+CAA+C;QAC/C,iEAAiE;QACjE,KAAK,MAAM;YACT,OAAO;gBACL,KAAK,EAAE,WAAW,CAAC,MAAM,CACvB,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,UAAU,CAAC,QAAQ,CACrD;gBACD,WAAW,EAAE,KAAK;aACnB,CAAC;QACJ,OAAO,CAAC,CAAC,CAAC;YACR,MAAM,gBAAgB,GAAU,IAAI,CAAC;YACrC,MAAM,IAAI,KAAK,CAAC,iCAAiC,gBAAgB,EAAE,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,qBAAqB,CAC5B,MAAuC;IAEvC,MAAM,QAAQ,GAAgC,EAAE,CAAC;IAEjD,KAAK,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,MAAM,EAAE,CAAC;QACvC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC3C,MAAM;YACR,CAAC;YAED,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO;yBACb,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;wBACZ,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;4BAClB,KAAK,MAAM,CAAC,CAAC,CAAC;gCACZ,OAAO,IAAI,CAAC,IAAI,CAAC;4BACnB,CAAC;4BACD,KAAK,OAAO,CAAC,CAAC,CAAC;gCACb,MAAM,IAAI,wCAA6B,CAAC;oCACtC,aAAa,EAAE,YAAY;iCAC5B,CAAC,CAAC;4BACL,CAAC;wBACH,CAAC;oBACH,CAAC,CAAC;yBACD,IAAI,CAAC,EAAE,CAAC;iBACZ,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;YAED,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,IAAI,IAAI,GAAG,EAAE,CAAC;gBACd,MAAM,SAAS,GAGM,EAAE,CAAC;gBAExB,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;oBAC3B,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;wBAClB,KAAK,MAAM,CAAC,CAAC,CAAC;4BACZ,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;4BAClB,MAAM;wBACR,CAAC;wBACD,KAAK,WAAW,CAAC,CAAC,CAAC;4BACjB,SAAS,CAAC,IAAI,CAAC;gCACb,EAAE,EAAE,IAAI,CAAC,UAAU;gCACnB,IAAI,EAAE,UAAU;gCAChB,QAAQ,EAAE;oCACR,IAAI,EAAE,IAAI,CAAC,QAAQ;oCACnB,MAAM,EAAE,IAAI,CAAC,IAA+B;iCAC7C;6BACF,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBACD,OAAO,CAAC,CAAC,CAAC;4BACR,MAAM,gBAAgB,GAAU,IAAI,CAAC;4BACrC,MAAM,IAAI,KAAK,CAAC,qBAAqB,gBAAgB,EAAE,CAAC,CAAC;wBAC3D,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,IAAI;oBACb,aAAa,EAAE,SAAS;iBACzB,CAAC,CAAC;gBAEH,MAAM;YACR,CAAC;YACD,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,KAAK,MAAM,YAAY,IAAI,OAAO,EAAE,CAAC;oBACnC,QAAQ,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,YAAY,CAAC,QAAQ;wBAC3B,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,MAAM,CAAC;qBAC7C,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM;YACR,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACR,MAAM,gBAAgB,GAAU,IAAI,CAAC;gBACrC,MAAM,IAAI,KAAK,CAAC,qBAAqB,gBAAgB,EAAE,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAgB,iBAAiB,CAC/B,YAAkD;IAElD,QAAQ,YAAY,EAAE,CAAC;QACrB,KAAK,MAAM;YACT,OAAO,MAAM,CAAC;QAChB,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB,KAAK,eAAe;YAClB,OAAO,YAAY,CAAC;QACtB;YACE,OAAO,OAAO,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,EACzB,IAAI,EACJ,MAAM,EACN,SAAS,EACT,WAAW,EACX,IAAI,EACJ,gBAAgB,EAChB,eAAe;AACf,MAAM;EACiD;IAIvD,MAAM,GAAG,GAAkB;QACzB,UAAU,EAAE,qBAAqB,CAAC,MAAM,CAAC;QACzC,GAAG,CAAC,gBAAgB,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACzD,GAAG,CAAC,eAAe,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACvD,GAAG,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3C,GAAG,CAAC,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/C,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAClC,CAAC;IAEF,MAAM,QAAQ,GAAiC,EAAE,CAAC;IAElD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,OAAO;gBACL,GAAG,EAAE,EAAE,GAAG,GAAG,EAAE,GAAG,yBAAyB,CAAC,IAAI,CAAC,EAAE;gBACnD,QAAQ;aACT,CAAC;QACJ,CAAC;QAED,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,OAAO;gBACL,GAAG;gBACH,QAAQ;aACT,CAAC;QACJ,CAAC;QAED,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,OAAO;gBACL,GAAG,EAAE,EAAE,GAAG,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE;gBAC7B,QAAQ;aACT,CAAC;QACJ,CAAC;QAED,KAAK,gBAAgB,CAAC,CAAC,CAAC;YACtB,MAAM,IAAI,wCAA6B,CAAC;gBACtC,aAAa,EAAE,qBAAqB;aACrC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,CAAC,CAAC,CAAC;YACR,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,kBAAmB,SAAQ,qBAGhC;IACS,KAAK,GAGA;QACX,YAAY,EAAE,CAAC;QACf,gBAAgB,EAAE,CAAC;KACpB,CAAC;IAEM,YAAY,GAGA,OAAO,CAAC;IAE5B,SAAS,CACP,KAA+B,EAC/B,UAAuE;QAEvE,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;YACzB,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;gBACtC,UAAU,CAAC,OAAO,CAAC;oBACjB,IAAI,EAAE,WAAW;oBACjB,YAAY,EAAE,UAAU;oBACxB,UAAU,EAAE,EAAE,CAAC,EAAE;oBACjB,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;oBAC1B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAO,CAAC;iBAC1C,CAAC,CAAC;gBACH,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;YACnC,CAAC;QACH,CAAC;QAED,UAAU,CAAC,OAAO,CAAC;YACjB,IAAI,EAAE,YAAY;YAClB,SAAS,EAAE,MAAM,CAAC,OAAO,IAAI,EAAE;SAChC,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,GAAG,iBAAiB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;IAC7D,CAAC;IAED,KAAK,CACH,UAAuE;QAEvE,UAAU,CAAC,OAAO,CAAC;YACjB,IAAI,EAAE,QAAQ;YACd,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC,CAAC;IACL,CAAC;CACF"}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../index.ts"],"names":[],"mappings":";;;AA8RA,8CAaC;AA3SD,oCAIoB;AAEpB,+CAO0B;AA0B1B,MAAa,eAAe;IACT,MAAM,CAAY;IAClB,QAAQ,CAAa;IAEtC,YACE,KAAsC,EACtC,MAA2B;QAE3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IACtC,CAAC;IACD,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC;IACnC,CAAC;IACD,IAAI,UAAU;QACZ,OAAO,CACL,IAAI,CAAC,QAAQ,CAAC,UAAU,IAAI;YAC1B,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACf,CACF,CAAC;IACJ,CAAC;IACD,IAAI,QAAQ;QACV,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACxB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;QAErC,OAAO,KAAK,EAAE,KAAS,EAAqB,EAAE;YAC5C,OAAO,CAAC,MAAM,SAAS,CAAC,KAAK,CAAC,CAAa,CAAC;QAC9C,CAAC,CAAC;IACJ,CAAC;CACF;AAhCD,0CAgCC;AAED,MAAa,YAAY;IACd,oBAAoB,GAAG,IAAI,CAAC;IAC5B,2BAA2B,GAAG,MAAM,CAAC;IAE7B,EAAE,CAAc;IAChB,MAAM,CAAY;IAE5B,OAAO,CAAS;IAEvB,YAAY,EAAe,EAAE,MAA2B;QACtD,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC;IAC7C,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,UAAU,CACd,OAA+D;QAE/D,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAmB,CAAC;QACxD,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAEjC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC;QAED,OAAO;YACL,IAAI,EAAE,MAAM,CAAC,OAAO,IAAI,SAAS;YACjC,SAAS,EAAE,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC5C,YAAY,EAAE,UAAU;gBACxB,UAAU,EAAE,EAAE,CAAC,EAAE;gBACjB,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;gBAC1B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAO,CAAC;aAC1C,CAAC,CAAC;YACH,YAAY,EAAE,iBAAiB,CAAC,MAAM,CAAC,YAAY,CAAC;YACpD,KAAK,EAAE;gBACL,YAAY,EAAE,GAAG,CAAC,UAAU,EAAE,YAAY,IAAI,CAAC;gBAC/C,gBAAgB,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,IAAI,CAAC;aACxD;YACD,OAAO,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE,EAAE;YAC9D,QAAQ;SACT,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,OAA6D;QAE7D,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAErD,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE;YACnC,MAAM,EAAE,IAAI;SACb,CAAC,CAAmC,CAAC;QAEtC,OAAO;YACL,MAAM,EAAE,GAAG,CAAC,WAAW,CAAC,IAAI,kBAAkB,EAAE,CAAC;YACjD,OAAO,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE,EAAE;YAC9D,QAAQ;SACT,CAAC;IACJ,CAAC;CACF;AA/DD,oCA+DC;AAED,SAAS,yBAAyB,CAChC,IAEC;IAED,2EAA2E;IAC3E,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;IAE1D,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;QAClB,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC;IACtD,CAAC;IAED,MAAM,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACvC,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE;YACR,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B;KACF,CAAC,CAAC,CAAC;IAEJ,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;IAEnC,IAAI,UAAU,IAAI,IAAI,EAAE,CAAC;QACvB,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC;IACxD,CAAC;IAED,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;IAE7B,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,MAAM,CAAC;QACZ,KAAK,MAAM;YACT,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;QACnD,KAAK,UAAU;YACb,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QAEpD,+CAA+C;QAC/C,iEAAiE;QACjE,KAAK,MAAM;YACT,OAAO;gBACL,KAAK,EAAE,WAAW,CAAC,MAAM,CACvB,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,UAAU,CAAC,QAAQ,CACrD;gBACD,WAAW,EAAE,KAAK;aACnB,CAAC;QACJ,OAAO,CAAC,CAAC,CAAC;YACR,MAAM,gBAAgB,GAAU,IAAI,CAAC;YACrC,MAAM,IAAI,KAAK,CAAC,iCAAiC,gBAAgB,EAAE,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,qBAAqB,CAC5B,MAAuC;IAEvC,MAAM,QAAQ,GAAgC,EAAE,CAAC;IAEjD,KAAK,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,MAAM,EAAE,CAAC;QACvC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC3C,MAAM;YACR,CAAC;YAED,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO;yBACb,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;wBACZ,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;4BAClB,KAAK,MAAM,CAAC,CAAC,CAAC;gCACZ,OAAO,IAAI,CAAC,IAAI,CAAC;4BACnB,CAAC;4BACD,KAAK,OAAO,CAAC,CAAC,CAAC;gCACb,MAAM,IAAI,wCAA6B,CAAC;oCACtC,aAAa,EAAE,YAAY;iCAC5B,CAAC,CAAC;4BACL,CAAC;wBACH,CAAC;oBACH,CAAC,CAAC;yBACD,IAAI,CAAC,EAAE,CAAC;iBACZ,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;YAED,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,IAAI,IAAI,GAAG,EAAE,CAAC;gBACd,MAAM,SAAS,GAGM,EAAE,CAAC;gBAExB,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;oBAC3B,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;wBAClB,KAAK,MAAM,CAAC,CAAC,CAAC;4BACZ,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;4BAClB,MAAM;wBACR,CAAC;wBACD,KAAK,WAAW,CAAC,CAAC,CAAC;4BACjB,SAAS,CAAC,IAAI,CAAC;gCACb,EAAE,EAAE,IAAI,CAAC,UAAU;gCACnB,IAAI,EAAE,UAAU;gCAChB,QAAQ,EAAE;oCACR,IAAI,EAAE,IAAI,CAAC,QAAQ;oCACnB,MAAM,EAAE,IAAI,CAAC,IAA+B;iCAC7C;6BACF,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBACD,OAAO,CAAC,CAAC,CAAC;4BACR,MAAM,gBAAgB,GAAU,IAAI,CAAC;4BACrC,MAAM,IAAI,KAAK,CAAC,qBAAqB,gBAAgB,EAAE,CAAC,CAAC;wBAC3D,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,MAAM,aAAa,GAAG,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;gBAErE,IAAI,aAAa,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACrC,QAAQ,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,IAAI;wBACb,aAAa;qBACd,CAAC,CAAC;gBACL,CAAC;gBAED,MAAM;YACR,CAAC;YACD,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,KAAK,MAAM,YAAY,IAAI,OAAO,EAAE,CAAC;oBACnC,QAAQ,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,YAAY,CAAC,QAAQ;wBAC3B,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,MAAM,CAAC;qBAC7C,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM;YACR,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACR,MAAM,gBAAgB,GAAU,IAAI,CAAC;gBACrC,MAAM,IAAI,KAAK,CAAC,qBAAqB,gBAAgB,EAAE,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAgB,iBAAiB,CAC/B,YAAkD;IAElD,QAAQ,YAAY,EAAE,CAAC;QACrB,KAAK,MAAM;YACT,OAAO,MAAM,CAAC;QAChB,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB,KAAK,eAAe;YAClB,OAAO,YAAY,CAAC;QACtB;YACE,OAAO,OAAO,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,EACzB,IAAI,EACJ,MAAM,EACN,SAAS,EACT,WAAW,EACX,IAAI,EACJ,gBAAgB,EAChB,eAAe;AACf,MAAM;EACiD;IAIvD,MAAM,GAAG,GAAkB;QACzB,UAAU,EAAE,qBAAqB,CAAC,MAAM,CAAC;QACzC,GAAG,CAAC,gBAAgB,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACzD,GAAG,CAAC,eAAe,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACvD,GAAG,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3C,GAAG,CAAC,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/C,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAClC,CAAC;IAEF,MAAM,QAAQ,GAAiC,EAAE,CAAC;IAElD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,OAAO;gBACL,GAAG,EAAE,EAAE,GAAG,GAAG,EAAE,GAAG,yBAAyB,CAAC,IAAI,CAAC,EAAE;gBACnD,QAAQ;aACT,CAAC;QACJ,CAAC;QAED,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,OAAO;gBACL,GAAG;gBACH,QAAQ;aACT,CAAC;QACJ,CAAC;QAED,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,OAAO;gBACL,GAAG,EAAE,EAAE,GAAG,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE;gBAC7B,QAAQ;aACT,CAAC;QACJ,CAAC;QAED,KAAK,gBAAgB,CAAC,CAAC,CAAC;YACtB,MAAM,IAAI,wCAA6B,CAAC;gBACtC,aAAa,EAAE,qBAAqB;aACrC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,CAAC,CAAC,CAAC;YACR,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,kBAAmB,SAAQ,qBAGhC;IACS,KAAK,GAGA;QACX,YAAY,EAAE,CAAC;QACf,gBAAgB,EAAE,CAAC;KACpB,CAAC;IAEM,YAAY,GAGA,OAAO,CAAC;IAC5B;QACE,MAAM,WAAW,GAAG;YAClB,SAAS,EAAE,CACT,KAA+B,EAC/B,UAAuE,EACvE,EAAE;gBACF,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBACnC,IAAI,CAAC,MAAM,EAAE,CAAC;oBACZ,OAAO;gBACT,CAAC;gBAED,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;oBACzB,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;wBACtC,UAAU,CAAC,OAAO,CAAC;4BACjB,IAAI,EAAE,WAAW;4BACjB,YAAY,EAAE,UAAU;4BACxB,UAAU,EAAE,EAAE,CAAC,EAAE;4BACjB,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;4BAC1B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;yBACzC,CAAC,CAAC;wBACH,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;oBACnC,CAAC;gBACH,CAAC;gBAED,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAChD,UAAU,CAAC,OAAO,CAAC;wBACjB,IAAI,EAAE,YAAY;wBAClB,SAAS,EAAE,MAAM,CAAC,OAAO,IAAI,EAAE;qBAChC,CAAC,CAAC;gBACL,CAAC;gBACD,IAAI,CAAC,YAAY,GAAG,iBAAiB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YAC7D,CAAC;YACD,KAAK,EAAE,CACL,UAAuE,EACvE,EAAE;gBACF,UAAU,CAAC,OAAO,CAAC;oBACjB,IAAI,EAAE,QAAQ;oBACd,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;iBAClB,CAAC,CAAC;YACL,CAAC;SACF,CAAC;QAEF,KAAK,CAAC,WAAW,CAAC,CAAC;QAEnB,IAAI,CAAC,KAAK,GAAG;YACX,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,CAAC;SACpB,CAAC;QACF,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC;IAC9B,CAAC;CACF"}
package/index.d.ts CHANGED
@@ -1,16 +1,28 @@
1
1
  import { type LanguageModelV1, type LanguageModelV1FinishReason } from '@ai-sdk/provider';
2
- import type { AxAIService, AxChatResponseResult } from '@ax-llm/ax/index.js';
3
- type AxChatConfig = {
4
- generateId: () => string;
2
+ import type { AxAgent, AxAIService, AxChatResponseResult, AxFunctionJSONSchema, AxGenIn, AxGenOut } from '@ax-llm/ax/index.js';
3
+ type AxConfig = {
5
4
  fetch?: typeof fetch;
6
5
  };
7
- export declare class AxAgentFramework implements LanguageModelV1 {
6
+ interface RenderTool<IN> {
7
+ description?: string;
8
+ parameters?: AxFunctionJSONSchema;
9
+ generate?: ((input: IN) => Promise<unknown>) | undefined;
10
+ }
11
+ export declare class AxAgentProvider<IN extends AxGenIn> implements RenderTool<IN> {
12
+ private readonly config?;
13
+ private readonly funcInfo;
14
+ constructor(agent: Readonly<AxAgent<IN, AxGenOut>>, config?: Readonly<AxConfig>);
15
+ get description(): string;
16
+ get parameters(): AxFunctionJSONSchema;
17
+ get generate(): ((input: IN) => Promise<unknown>) | undefined;
18
+ }
19
+ export declare class AxAIProvider implements LanguageModelV1 {
8
20
  readonly specificationVersion = "v1";
9
21
  readonly defaultObjectGenerationMode = "json";
10
22
  private readonly ai;
11
- private readonly config;
23
+ private readonly config?;
12
24
  modelId: string;
13
- constructor(ai: AxAIService, config: Readonly<AxChatConfig>);
25
+ constructor(ai: AxAIService, config?: Readonly<AxConfig>);
14
26
  get provider(): string;
15
27
  doGenerate(options: Readonly<Parameters<LanguageModelV1['doGenerate']>[0]>): Promise<Awaited<ReturnType<LanguageModelV1['doGenerate']>>>;
16
28
  doStream(options: Readonly<Parameters<LanguageModelV1['doStream']>[0]>): Promise<Awaited<ReturnType<LanguageModelV1['doStream']>>>;
package/index.js CHANGED
@@ -1,6 +1,32 @@
1
1
  import { TransformStream, TransformStreamDefaultController } from 'stream/web';
2
2
  import { UnsupportedFunctionalityError } from '@ai-sdk/provider';
3
- export class AxAgentFramework {
3
+ export class AxAgentProvider {
4
+ config;
5
+ funcInfo;
6
+ constructor(agent, config) {
7
+ this.config = config;
8
+ this.funcInfo = agent.getFunction();
9
+ }
10
+ get description() {
11
+ return this.funcInfo.description;
12
+ }
13
+ get parameters() {
14
+ return (this.funcInfo.parameters ?? {
15
+ type: 'object',
16
+ properties: {}
17
+ });
18
+ }
19
+ get generate() {
20
+ if (!this.funcInfo.func) {
21
+ return undefined;
22
+ }
23
+ const agentFunc = this.funcInfo.func;
24
+ return async (input) => {
25
+ return (await agentFunc(input));
26
+ };
27
+ }
28
+ }
29
+ export class AxAIProvider {
4
30
  specificationVersion = 'v1';
5
31
  defaultObjectGenerationMode = 'json';
6
32
  ai;
@@ -40,7 +66,9 @@ export class AxAgentFramework {
40
66
  }
41
67
  async doStream(options) {
42
68
  const { req, warnings } = createChatRequest(options);
43
- const res = (await this.ai.chat(req));
69
+ const res = (await this.ai.chat(req, {
70
+ stream: true
71
+ }));
44
72
  return {
45
73
  stream: res.pipeThrough(new AxToSDKTransformer()),
46
74
  rawCall: { rawPrompt: '', rawSettings: req.modelConfig ?? {} },
@@ -140,11 +168,14 @@ function convertToAxChatPrompt(prompt) {
140
168
  }
141
169
  }
142
170
  }
143
- messages.push({
144
- role: 'assistant',
145
- content: text,
146
- functionCalls: toolCalls
147
- });
171
+ const functionCalls = toolCalls.length === 0 ? undefined : toolCalls;
172
+ if (functionCalls || text.length > 0) {
173
+ messages.push({
174
+ role: 'assistant',
175
+ content: text,
176
+ functionCalls
177
+ });
178
+ }
148
179
  break;
149
180
  }
150
181
  case 'tool': {
@@ -224,35 +255,47 @@ class AxToSDKTransformer extends TransformStream {
224
255
  completionTokens: 0
225
256
  };
226
257
  finishReason = 'other';
227
- transform(chunk, controller) {
228
- const choice = chunk.results.at(0);
229
- if (!choice) {
230
- throw new Error('No choice returned');
231
- }
232
- if (choice.functionCalls) {
233
- for (const tc of choice.functionCalls) {
258
+ constructor() {
259
+ const transformer = {
260
+ transform: (chunk, controller) => {
261
+ const choice = chunk.results.at(0);
262
+ if (!choice) {
263
+ return;
264
+ }
265
+ if (choice.functionCalls) {
266
+ for (const tc of choice.functionCalls) {
267
+ controller.enqueue({
268
+ type: 'tool-call',
269
+ toolCallType: 'function',
270
+ toolCallId: tc.id,
271
+ toolName: tc.function.name,
272
+ args: JSON.stringify(tc.function.params)
273
+ });
274
+ this.finishReason = 'tool-calls';
275
+ }
276
+ }
277
+ if (choice.content && choice.content.length > 0) {
278
+ controller.enqueue({
279
+ type: 'text-delta',
280
+ textDelta: choice.content ?? ''
281
+ });
282
+ }
283
+ this.finishReason = mapAxFinishReason(choice.finishReason);
284
+ },
285
+ flush: (controller) => {
234
286
  controller.enqueue({
235
- type: 'tool-call',
236
- toolCallType: 'function',
237
- toolCallId: tc.id,
238
- toolName: tc.function.name,
239
- args: JSON.stringify(tc.function.params)
287
+ type: 'finish',
288
+ finishReason: this.finishReason,
289
+ usage: this.usage
240
290
  });
241
- this.finishReason = 'tool-calls';
242
291
  }
243
- }
244
- controller.enqueue({
245
- type: 'text-delta',
246
- textDelta: choice.content ?? ''
247
- });
248
- this.finishReason = mapAxFinishReason(choice.finishReason);
249
- }
250
- flush(controller) {
251
- controller.enqueue({
252
- type: 'finish',
253
- finishReason: this.finishReason,
254
- usage: this.usage
255
- });
292
+ };
293
+ super(transformer);
294
+ this.usage = {
295
+ promptTokens: 0,
296
+ completionTokens: 0
297
+ };
298
+ this.finishReason = 'other';
256
299
  }
257
300
  }
258
301
  //# sourceMappingURL=index.js.map
package/index.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,eAAe,EACf,gCAAgC,EACjC,MAAM,YAAY,CAAC;AAEpB,OAAO,EAML,6BAA6B,EAC9B,MAAM,kBAAkB,CAAC;AAgB1B,MAAM,OAAO,gBAAgB;IAClB,oBAAoB,GAAG,IAAI,CAAC;IAC5B,2BAA2B,GAAG,MAAM,CAAC;IAE7B,EAAE,CAAc;IAChB,MAAM,CAAe;IAE/B,OAAO,CAAS;IAEvB,YAAY,EAAe,EAAE,MAA8B;QACzD,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC;IAC7C,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,UAAU,CACd,OAA+D;QAE/D,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAmB,CAAC;QACxD,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAEjC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC;QAED,OAAO;YACL,IAAI,EAAE,MAAM,CAAC,OAAO,IAAI,SAAS;YACjC,SAAS,EAAE,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC5C,YAAY,EAAE,UAAU;gBACxB,UAAU,EAAE,EAAE,CAAC,EAAE;gBACjB,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;gBAC1B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAO,CAAC;aAC1C,CAAC,CAAC;YACH,YAAY,EAAE,iBAAiB,CAAC,MAAM,CAAC,YAAY,CAAC;YACpD,KAAK,EAAE;gBACL,YAAY,EAAE,GAAG,CAAC,UAAU,EAAE,YAAY,IAAI,CAAC;gBAC/C,gBAAgB,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,IAAI,CAAC;aACxD;YACD,OAAO,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE,EAAE;YAC9D,QAAQ;SACT,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,OAA6D;QAE7D,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAmC,CAAC;QAExE,OAAO;YACL,MAAM,EAAE,GAAG,CAAC,WAAW,CAAC,IAAI,kBAAkB,EAAE,CAAC;YACjD,OAAO,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE,EAAE;YAC9D,QAAQ;SACT,CAAC;IACJ,CAAC;CACF;AAED,SAAS,yBAAyB,CAChC,IAEC;IAED,2EAA2E;IAC3E,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;IAE1D,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;QAClB,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC;IACtD,CAAC;IAED,MAAM,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACvC,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE;YACR,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B;KACF,CAAC,CAAC,CAAC;IAEJ,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;IAEnC,IAAI,UAAU,IAAI,IAAI,EAAE,CAAC;QACvB,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC;IACxD,CAAC;IAED,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;IAE7B,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,MAAM,CAAC;QACZ,KAAK,MAAM;YACT,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;QACnD,KAAK,UAAU;YACb,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QAEpD,+CAA+C;QAC/C,iEAAiE;QACjE,KAAK,MAAM;YACT,OAAO;gBACL,KAAK,EAAE,WAAW,CAAC,MAAM,CACvB,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,UAAU,CAAC,QAAQ,CACrD;gBACD,WAAW,EAAE,KAAK;aACnB,CAAC;QACJ,OAAO,CAAC,CAAC,CAAC;YACR,MAAM,gBAAgB,GAAU,IAAI,CAAC;YACrC,MAAM,IAAI,KAAK,CAAC,iCAAiC,gBAAgB,EAAE,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,qBAAqB,CAC5B,MAAuC;IAEvC,MAAM,QAAQ,GAAgC,EAAE,CAAC;IAEjD,KAAK,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,MAAM,EAAE,CAAC;QACvC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC3C,MAAM;YACR,CAAC;YAED,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO;yBACb,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;wBACZ,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;4BAClB,KAAK,MAAM,CAAC,CAAC,CAAC;gCACZ,OAAO,IAAI,CAAC,IAAI,CAAC;4BACnB,CAAC;4BACD,KAAK,OAAO,CAAC,CAAC,CAAC;gCACb,MAAM,IAAI,6BAA6B,CAAC;oCACtC,aAAa,EAAE,YAAY;iCAC5B,CAAC,CAAC;4BACL,CAAC;wBACH,CAAC;oBACH,CAAC,CAAC;yBACD,IAAI,CAAC,EAAE,CAAC;iBACZ,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;YAED,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,IAAI,IAAI,GAAG,EAAE,CAAC;gBACd,MAAM,SAAS,GAGM,EAAE,CAAC;gBAExB,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;oBAC3B,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;wBAClB,KAAK,MAAM,CAAC,CAAC,CAAC;4BACZ,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;4BAClB,MAAM;wBACR,CAAC;wBACD,KAAK,WAAW,CAAC,CAAC,CAAC;4BACjB,SAAS,CAAC,IAAI,CAAC;gCACb,EAAE,EAAE,IAAI,CAAC,UAAU;gCACnB,IAAI,EAAE,UAAU;gCAChB,QAAQ,EAAE;oCACR,IAAI,EAAE,IAAI,CAAC,QAAQ;oCACnB,MAAM,EAAE,IAAI,CAAC,IAA+B;iCAC7C;6BACF,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBACD,OAAO,CAAC,CAAC,CAAC;4BACR,MAAM,gBAAgB,GAAU,IAAI,CAAC;4BACrC,MAAM,IAAI,KAAK,CAAC,qBAAqB,gBAAgB,EAAE,CAAC,CAAC;wBAC3D,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,IAAI;oBACb,aAAa,EAAE,SAAS;iBACzB,CAAC,CAAC;gBAEH,MAAM;YACR,CAAC;YACD,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,KAAK,MAAM,YAAY,IAAI,OAAO,EAAE,CAAC;oBACnC,QAAQ,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,YAAY,CAAC,QAAQ;wBAC3B,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,MAAM,CAAC;qBAC7C,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM;YACR,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACR,MAAM,gBAAgB,GAAU,IAAI,CAAC;gBACrC,MAAM,IAAI,KAAK,CAAC,qBAAqB,gBAAgB,EAAE,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,YAAkD;IAElD,QAAQ,YAAY,EAAE,CAAC;QACrB,KAAK,MAAM;YACT,OAAO,MAAM,CAAC;QAChB,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB,KAAK,eAAe;YAClB,OAAO,YAAY,CAAC;QACtB;YACE,OAAO,OAAO,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,EACzB,IAAI,EACJ,MAAM,EACN,SAAS,EACT,WAAW,EACX,IAAI,EACJ,gBAAgB,EAChB,eAAe;AACf,MAAM;EACiD;IAIvD,MAAM,GAAG,GAAkB;QACzB,UAAU,EAAE,qBAAqB,CAAC,MAAM,CAAC;QACzC,GAAG,CAAC,gBAAgB,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACzD,GAAG,CAAC,eAAe,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACvD,GAAG,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3C,GAAG,CAAC,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/C,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAClC,CAAC;IAEF,MAAM,QAAQ,GAAiC,EAAE,CAAC;IAElD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,OAAO;gBACL,GAAG,EAAE,EAAE,GAAG,GAAG,EAAE,GAAG,yBAAyB,CAAC,IAAI,CAAC,EAAE;gBACnD,QAAQ;aACT,CAAC;QACJ,CAAC;QAED,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,OAAO;gBACL,GAAG;gBACH,QAAQ;aACT,CAAC;QACJ,CAAC;QAED,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,OAAO;gBACL,GAAG,EAAE,EAAE,GAAG,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE;gBAC7B,QAAQ;aACT,CAAC;QACJ,CAAC;QAED,KAAK,gBAAgB,CAAC,CAAC,CAAC;YACtB,MAAM,IAAI,6BAA6B,CAAC;gBACtC,aAAa,EAAE,qBAAqB;aACrC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,CAAC,CAAC,CAAC;YACR,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,kBAAmB,SAAQ,eAGhC;IACS,KAAK,GAGA;QACX,YAAY,EAAE,CAAC;QACf,gBAAgB,EAAE,CAAC;KACpB,CAAC;IAEM,YAAY,GAGA,OAAO,CAAC;IAE5B,SAAS,CACP,KAA+B,EAC/B,UAAuE;QAEvE,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;YACzB,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;gBACtC,UAAU,CAAC,OAAO,CAAC;oBACjB,IAAI,EAAE,WAAW;oBACjB,YAAY,EAAE,UAAU;oBACxB,UAAU,EAAE,EAAE,CAAC,EAAE;oBACjB,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;oBAC1B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAO,CAAC;iBAC1C,CAAC,CAAC;gBACH,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;YACnC,CAAC;QACH,CAAC;QAED,UAAU,CAAC,OAAO,CAAC;YACjB,IAAI,EAAE,YAAY;YAClB,SAAS,EAAE,MAAM,CAAC,OAAO,IAAI,EAAE;SAChC,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,GAAG,iBAAiB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;IAC7D,CAAC;IAED,KAAK,CACH,UAAuE;QAEvE,UAAU,CAAC,OAAO,CAAC;YACjB,IAAI,EAAE,QAAQ;YACd,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC,CAAC;IACL,CAAC;CACF"}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,eAAe,EACf,gCAAgC,EACjC,MAAM,YAAY,CAAC;AAEpB,OAAO,EAML,6BAA6B,EAC9B,MAAM,kBAAkB,CAAC;AA0B1B,MAAM,OAAO,eAAe;IACT,MAAM,CAAY;IAClB,QAAQ,CAAa;IAEtC,YACE,KAAsC,EACtC,MAA2B;QAE3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IACtC,CAAC;IACD,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC;IACnC,CAAC;IACD,IAAI,UAAU;QACZ,OAAO,CACL,IAAI,CAAC,QAAQ,CAAC,UAAU,IAAI;YAC1B,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACf,CACF,CAAC;IACJ,CAAC;IACD,IAAI,QAAQ;QACV,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACxB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;QAErC,OAAO,KAAK,EAAE,KAAS,EAAqB,EAAE;YAC5C,OAAO,CAAC,MAAM,SAAS,CAAC,KAAK,CAAC,CAAa,CAAC;QAC9C,CAAC,CAAC;IACJ,CAAC;CACF;AAED,MAAM,OAAO,YAAY;IACd,oBAAoB,GAAG,IAAI,CAAC;IAC5B,2BAA2B,GAAG,MAAM,CAAC;IAE7B,EAAE,CAAc;IAChB,MAAM,CAAY;IAE5B,OAAO,CAAS;IAEvB,YAAY,EAAe,EAAE,MAA2B;QACtD,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,IAAI,CAAC;IAC7C,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,UAAU,CACd,OAA+D;QAE/D,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAmB,CAAC;QACxD,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAEjC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC;QAED,OAAO;YACL,IAAI,EAAE,MAAM,CAAC,OAAO,IAAI,SAAS;YACjC,SAAS,EAAE,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC5C,YAAY,EAAE,UAAU;gBACxB,UAAU,EAAE,EAAE,CAAC,EAAE;gBACjB,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;gBAC1B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAO,CAAC;aAC1C,CAAC,CAAC;YACH,YAAY,EAAE,iBAAiB,CAAC,MAAM,CAAC,YAAY,CAAC;YACpD,KAAK,EAAE;gBACL,YAAY,EAAE,GAAG,CAAC,UAAU,EAAE,YAAY,IAAI,CAAC;gBAC/C,gBAAgB,EAAE,GAAG,CAAC,UAAU,EAAE,gBAAgB,IAAI,CAAC;aACxD;YACD,OAAO,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE,EAAE;YAC9D,QAAQ;SACT,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,OAA6D;QAE7D,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAErD,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE;YACnC,MAAM,EAAE,IAAI;SACb,CAAC,CAAmC,CAAC;QAEtC,OAAO;YACL,MAAM,EAAE,GAAG,CAAC,WAAW,CAAC,IAAI,kBAAkB,EAAE,CAAC;YACjD,OAAO,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,IAAI,EAAE,EAAE;YAC9D,QAAQ;SACT,CAAC;IACJ,CAAC;CACF;AAED,SAAS,yBAAyB,CAChC,IAEC;IAED,2EAA2E;IAC3E,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;IAE1D,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;QAClB,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC;IACtD,CAAC;IAED,MAAM,WAAW,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACvC,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE;YACR,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B;KACF,CAAC,CAAC,CAAC;IAEJ,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;IAEnC,IAAI,UAAU,IAAI,IAAI,EAAE,CAAC;QACvB,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC;IACxD,CAAC;IAED,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC;IAE7B,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,MAAM,CAAC;QACZ,KAAK,MAAM;YACT,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;QACnD,KAAK,UAAU;YACb,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QAEpD,+CAA+C;QAC/C,iEAAiE;QACjE,KAAK,MAAM;YACT,OAAO;gBACL,KAAK,EAAE,WAAW,CAAC,MAAM,CACvB,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,UAAU,CAAC,QAAQ,CACrD;gBACD,WAAW,EAAE,KAAK;aACnB,CAAC;QACJ,OAAO,CAAC,CAAC,CAAC;YACR,MAAM,gBAAgB,GAAU,IAAI,CAAC;YACrC,MAAM,IAAI,KAAK,CAAC,iCAAiC,gBAAgB,EAAE,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,qBAAqB,CAC5B,MAAuC;IAEvC,MAAM,QAAQ,GAAgC,EAAE,CAAC;IAEjD,KAAK,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,MAAM,EAAE,CAAC;QACvC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC3C,MAAM;YACR,CAAC;YAED,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO;yBACb,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;wBACZ,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;4BAClB,KAAK,MAAM,CAAC,CAAC,CAAC;gCACZ,OAAO,IAAI,CAAC,IAAI,CAAC;4BACnB,CAAC;4BACD,KAAK,OAAO,CAAC,CAAC,CAAC;gCACb,MAAM,IAAI,6BAA6B,CAAC;oCACtC,aAAa,EAAE,YAAY;iCAC5B,CAAC,CAAC;4BACL,CAAC;wBACH,CAAC;oBACH,CAAC,CAAC;yBACD,IAAI,CAAC,EAAE,CAAC;iBACZ,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;YAED,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,IAAI,IAAI,GAAG,EAAE,CAAC;gBACd,MAAM,SAAS,GAGM,EAAE,CAAC;gBAExB,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;oBAC3B,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;wBAClB,KAAK,MAAM,CAAC,CAAC,CAAC;4BACZ,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;4BAClB,MAAM;wBACR,CAAC;wBACD,KAAK,WAAW,CAAC,CAAC,CAAC;4BACjB,SAAS,CAAC,IAAI,CAAC;gCACb,EAAE,EAAE,IAAI,CAAC,UAAU;gCACnB,IAAI,EAAE,UAAU;gCAChB,QAAQ,EAAE;oCACR,IAAI,EAAE,IAAI,CAAC,QAAQ;oCACnB,MAAM,EAAE,IAAI,CAAC,IAA+B;iCAC7C;6BACF,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBACD,OAAO,CAAC,CAAC,CAAC;4BACR,MAAM,gBAAgB,GAAU,IAAI,CAAC;4BACrC,MAAM,IAAI,KAAK,CAAC,qBAAqB,gBAAgB,EAAE,CAAC,CAAC;wBAC3D,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,MAAM,aAAa,GAAG,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;gBAErE,IAAI,aAAa,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACrC,QAAQ,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,IAAI;wBACb,aAAa;qBACd,CAAC,CAAC;gBACL,CAAC;gBAED,MAAM;YACR,CAAC;YACD,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,KAAK,MAAM,YAAY,IAAI,OAAO,EAAE,CAAC;oBACnC,QAAQ,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,YAAY,CAAC,QAAQ;wBAC3B,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,MAAM,CAAC;qBAC7C,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM;YACR,CAAC;YACD,OAAO,CAAC,CAAC,CAAC;gBACR,MAAM,gBAAgB,GAAU,IAAI,CAAC;gBACrC,MAAM,IAAI,KAAK,CAAC,qBAAqB,gBAAgB,EAAE,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,YAAkD;IAElD,QAAQ,YAAY,EAAE,CAAC;QACrB,KAAK,MAAM;YACT,OAAO,MAAM,CAAC;QAChB,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB,KAAK,eAAe;YAClB,OAAO,YAAY,CAAC;QACtB;YACE,OAAO,OAAO,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,EACzB,IAAI,EACJ,MAAM,EACN,SAAS,EACT,WAAW,EACX,IAAI,EACJ,gBAAgB,EAChB,eAAe;AACf,MAAM;EACiD;IAIvD,MAAM,GAAG,GAAkB;QACzB,UAAU,EAAE,qBAAqB,CAAC,MAAM,CAAC;QACzC,GAAG,CAAC,gBAAgB,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACzD,GAAG,CAAC,eAAe,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACvD,GAAG,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3C,GAAG,CAAC,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/C,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAClC,CAAC;IAEF,MAAM,QAAQ,GAAiC,EAAE,CAAC;IAElD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,OAAO;gBACL,GAAG,EAAE,EAAE,GAAG,GAAG,EAAE,GAAG,yBAAyB,CAAC,IAAI,CAAC,EAAE;gBACnD,QAAQ;aACT,CAAC;QACJ,CAAC;QAED,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,OAAO;gBACL,GAAG;gBACH,QAAQ;aACT,CAAC;QACJ,CAAC;QAED,KAAK,aAAa,CAAC,CAAC,CAAC;YACnB,OAAO;gBACL,GAAG,EAAE,EAAE,GAAG,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE;gBAC7B,QAAQ;aACT,CAAC;QACJ,CAAC;QAED,KAAK,gBAAgB,CAAC,CAAC,CAAC;YACtB,MAAM,IAAI,6BAA6B,CAAC;gBACtC,aAAa,EAAE,qBAAqB;aACrC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,CAAC,CAAC,CAAC;YACR,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,kBAAmB,SAAQ,eAGhC;IACS,KAAK,GAGA;QACX,YAAY,EAAE,CAAC;QACf,gBAAgB,EAAE,CAAC;KACpB,CAAC;IAEM,YAAY,GAGA,OAAO,CAAC;IAC5B;QACE,MAAM,WAAW,GAAG;YAClB,SAAS,EAAE,CACT,KAA+B,EAC/B,UAAuE,EACvE,EAAE;gBACF,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBACnC,IAAI,CAAC,MAAM,EAAE,CAAC;oBACZ,OAAO;gBACT,CAAC;gBAED,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;oBACzB,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;wBACtC,UAAU,CAAC,OAAO,CAAC;4BACjB,IAAI,EAAE,WAAW;4BACjB,YAAY,EAAE,UAAU;4BACxB,UAAU,EAAE,EAAE,CAAC,EAAE;4BACjB,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;4BAC1B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;yBACzC,CAAC,CAAC;wBACH,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;oBACnC,CAAC;gBACH,CAAC;gBAED,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAChD,UAAU,CAAC,OAAO,CAAC;wBACjB,IAAI,EAAE,YAAY;wBAClB,SAAS,EAAE,MAAM,CAAC,OAAO,IAAI,EAAE;qBAChC,CAAC,CAAC;gBACL,CAAC;gBACD,IAAI,CAAC,YAAY,GAAG,iBAAiB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YAC7D,CAAC;YACD,KAAK,EAAE,CACL,UAAuE,EACvE,EAAE;gBACF,UAAU,CAAC,OAAO,CAAC;oBACjB,IAAI,EAAE,QAAQ;oBACd,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;iBAClB,CAAC,CAAC;YACL,CAAC;SACF,CAAC;QAEF,KAAK,CAAC,WAAW,CAAC,CAAC;QAEnB,IAAI,CAAC,KAAK,GAAG;YACX,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,CAAC;SACpB,CAAC;QACF,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC;IAC9B,CAAC;CACF"}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@ax-llm/ax-ai-sdk-provider",
3
- "version": "9.0.34",
3
+ "version": "9.0.36",
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": "^1.0.2",
17
- "@ax-llm/ax": "9.0.36"
17
+ "@ax-llm/ax": "9.0.38"
18
18
  },
19
19
  "bugs": {
20
20
  "url": "https://github.com/@ax-llm/ax/issues"