@ax-llm/ax 9.0.28 → 9.0.30

package/cjs/ai/together/api.js.map

@@ -1 +1 @@
-{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/together/api.ts"],"names":[],"mappings":";;;AAAA,wCAAmD;AACnD,6CAA8C;AAI9C,uCAAgD;AAIzC,MAAM,yBAAyB,GAAG,GAAqB,EAAE,CAC9D,eAAe,CAAC;IACd,2BAA2B;IAC3B,KAAK,EAAE,sCAAsC;IAC7C,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AALQ,QAAA,yBAAyB,6BAKjC;AASL,MAAa,YAAa,SAAQ,mBAA0B;IAC1D,YAAY,EACV,MAAM,EACN,MAAM,EACN,OAAO,EACkC;QACzC,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QACD,MAAM,OAAO,GAAG;YACd,GAAG,IAAA,iCAAyB,GAAE;YAC9B,GAAG,MAAM;SACV,CAAC;QACF,KAAK,CAAC;YACJ,MAAM;YACN,MAAM,EAAE,OAAO;YACf,OAAO;YACP,MAAM,EAAE,6BAA6B;YACrC,SAAS,EAAE,6BAAmB;SAC/B,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC5B,CAAC;CACF;AAvBD,oCAuBC"}
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../ai/together/api.ts"],"names":[],"mappings":";;;AAAA,wCAAmD;AACnD,6CAA8C;AAI9C,uCAAgD;AAIzC,MAAM,yBAAyB,GAAG,GAAqB,EAAE,CAC9D,eAAe,CAAC;IACd,2BAA2B;IAC3B,KAAK,EAAE,sCAAsC;IAC7C,GAAG,IAAA,+BAAqB,GAAE;CAC3B,CAAC,CAAC;AALQ,QAAA,yBAAyB,6BAKjC;AAUL,MAAa,YAAa,SAAQ,mBAAU;IAC1C,YAAY,EACV,MAAM,EACN,MAAM,EACN,OAAO,EACP,QAAQ,EACiC;QACzC,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QACD,MAAM,OAAO,GAAG;YACd,GAAG,IAAA,iCAAyB,GAAE;YAC9B,GAAG,MAAM;SACV,CAAC;QACF,KAAK,CAAC;YACJ,MAAM;YACN,MAAM,EAAE,OAAO;YACf,OAAO;YACP,MAAM,EAAE,6BAA6B;YACrC,SAAS,EAAE,6BAAmB;YAC9B,QAAQ;SACT,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC5B,CAAC;CACF;AAzBD,oCAyBC"}

package/cjs/ai/types.d.ts

@@ -1,6 +1,6 @@
 import type { ReadableStream } from 'stream/web';
 import type { AxTracer } from '../trace/index.js';
-import type { AxAIModelMap } from './base.js';
+export type AxAIModelMap = Record<string, string>;
 export type AxModelInfo = {
     name: string;
     currency?: string;
@@ -120,10 +120,14 @@ export type AxChatRequest = {
         };
     };
     modelConfig?: Readonly<AxModelConfig>;
+    model?: string;
 };
+export type AxInternalChatRequest = Omit<AxChatRequest, 'model'> & Required<Pick<AxChatRequest, 'model'>>;
 export type AxEmbedRequest = {
     texts?: readonly string[];
+    embedModel?: string;
 };
+export type AxInternalEmbedRequest = Omit<AxEmbedRequest, 'embedModel'> & Required<Pick<AxEmbedRequest, 'embedModel'>>;
 export type AxRateLimiterFunction = <T = unknown>(reqFunc: () => Promise<T | ReadableStream<T>>, info: Readonly<{
     modelUsage?: AxTokenUsage;
     embedModelUsage?: AxTokenUsage;
@@ -143,17 +147,14 @@ export type AxAIServiceActionOptions = {
 };
 export interface AxAIService {
     getName(): string;
-    getModelInfo(): Readonly<AxModelInfo & {
-        provider: string;
-    }>;
-    getEmbedModelInfo(): Readonly<AxModelInfo> | undefined;
+    getModelInfo(): Readonly<AxModelInfoWithProvider>;
+    getEmbedModelInfo(): Readonly<AxModelInfoWithProvider> | undefined;
     getModelConfig(): Readonly<AxModelConfig>;
     getFeatures(): {
         functions: boolean;
         streaming: boolean;
     };
-    setModelMap(modelMap: AxAIModelMap<string>): void;
-    setEmbedModelMap(embedModelMap: AxAIModelMap<string>): void;
+    getModelMap(): AxAIModelMap | undefined;
     chat(req: Readonly<AxChatRequest>, options?: Readonly<AxAIPromptConfig & AxAIServiceActionOptions>): Promise<AxChatResponse | ReadableStream<AxChatResponse>>;
     embed(req: Readonly<AxEmbedRequest>, options?: Readonly<AxAIServiceActionOptions & AxAIServiceActionOptions>): Promise<AxEmbedResponse>;
     setOptions(options: Readonly<AxAIServiceOptions>): void;

package/cjs/ai/wrap.d.ts

@@ -18,25 +18,22 @@ import { type AxAIOllamaArgs } from './ollama/api.js';
 import { type AxAIOpenAIArgs as AxAIOpenAIArgs } from './openai/api.js';
 import type { AxAIOpenAIEmbedModel, AxAIOpenAIModel } from './openai/types.js';
 import { type AxAITogetherArgs } from './together/api.js';
-import type { AxAIPromptConfig, AxAIService, AxAIServiceActionOptions, AxAIServiceOptions, AxChatRequest, AxChatResponse, AxEmbedRequest, AxEmbedResponse, AxModelConfig, AxModelInfo } from './types.js';
+import type { AxAIModelMap, AxAIPromptConfig, AxAIService, AxAIServiceActionOptions, AxAIServiceOptions, AxChatRequest, AxChatResponse, AxEmbedRequest, AxEmbedResponse, AxModelConfig, AxModelInfoWithProvider } from './types.js';
 export type AxAIArgs = AxAIOpenAIArgs | AxAIAzureOpenAIArgs | AxAITogetherArgs | AxAIAnthropicArgs | AxAIGroqArgs | AxAIGoogleGeminiArgs | AxAICohereArgs | AxAIHuggingFaceArgs | AxAIMistralArgs | AxAIDeepSeekArgs | AxAIOllamaArgs;
 export type AxAIModels = AxAIOpenAIModel | AxAIAnthropicModel | AxAIGroqModel | AxAIGoogleGeminiModel | AxAICohereModel | AxAIHuggingFaceModel | AxAIMistralModel | AxAIDeepSeekModel | string;
 export type AxAIEmbedModels = AxAIOpenAIEmbedModel | AxAIGoogleGeminiEmbedModel | AxAICohereEmbedModel | string;
 export declare class AxAI implements AxAIService {
     private ai;
     constructor(options: Readonly<AxAIArgs>);
-    setModelMap(modelMap: Readonly<Record<string, string>>): void;
-    setEmbedModelMap(modelMap: Readonly<Record<string, string>>): void;
     getName(): string;
-    getModelInfo(): Readonly<AxModelInfo & {
-        provider: string;
-    }>;
-    getEmbedModelInfo(): Readonly<AxModelInfo> | undefined;
+    getModelInfo(): Readonly<AxModelInfoWithProvider>;
+    getEmbedModelInfo(): Readonly<AxModelInfoWithProvider> | undefined;
     getModelConfig(): Readonly<AxModelConfig>;
     getFeatures(): {
         functions: boolean;
         streaming: boolean;
     };
+    getModelMap(): AxAIModelMap | undefined;
     chat(req: Readonly<AxChatRequest>, options?: Readonly<AxAIPromptConfig & AxAIServiceActionOptions>): Promise<AxChatResponse | ReadableStream<AxChatResponse>>;
     embed(req: Readonly<AxEmbedRequest>, options?: Readonly<AxAIServiceActionOptions & AxAIServiceActionOptions>): Promise<AxEmbedResponse>;
     setOptions(options: Readonly<AxAIServiceOptions>): void;

package/cjs/ai/wrap.js

@@ -53,12 +53,6 @@ class AxAI {
                 throw new Error(`Unknown AI`);
         }
     }
-    setModelMap(modelMap) {
-        this.ai.setModelMap(modelMap);
-    }
-    setEmbedModelMap(modelMap) {
-        this.ai.setEmbedModelMap(modelMap);
-    }
     getName() {
         return this.ai.getName();
     }
@@ -74,6 +68,9 @@ class AxAI {
     getFeatures() {
         return this.ai.getFeatures();
     }
+    getModelMap() {
+        return this.ai.getModelMap();
+    }
     async chat(req, options) {
         return await this.ai.chat(req, options);
     }

package/cjs/ai/wrap.js.map

@@ -1 +1 @@
-{"version":3,"file":"wrap.js","sourceRoot":"","sources":["../../../ai/wrap.ts"],"names":[],"mappings":";;;AAEA,+CAA2E;AAE3E,kDAG+B;AAC/B,4CAAkE;AAElE,8CAAwE;AAExE,mDAGgC;AAKhC,0CAA4D;AAE5D,iDAG8B;AAE9B,6CAAqE;AAErE,4CAAkE;AAClE,6CAGyB;AAEzB,+CAAwE;AA4CxE,MAAa,IAAI;IACP,EAAE,CAAc;IAExB,YAAY,OAA2B;QACrC,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,QAAQ;gBACX,IAAI,CAAC,EAAE,GAAG,IAAI,oBAAU,CAAC,OAAO,CAAC,CAAC;gBAClC,MAAM;YACR,KAAK,cAAc;gBACjB,IAAI,CAAC,EAAE,GAAG,IAAI,wBAAe,CAAC,OAAO,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,aAAa;gBAChB,IAAI,CAAC,EAAE,GAAG,IAAI,wBAAe,CAAC,OAAO,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,MAAM;gBACT,IAAI,CAAC,EAAE,GAAG,IAAI,iBAAQ,CAAC,OAAO,CAAC,CAAC;gBAChC,MAAM;YACR,KAAK,UAAU;gBACb,IAAI,CAAC,EAAE,GAAG,IAAI,sBAAY,CAAC,OAAO,CAAC,CAAC;gBACpC,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,EAAE,GAAG,IAAI,mBAAU,CAAC,OAAO,CAAC,CAAC;gBAClC,MAAM;YACR,KAAK,eAAe;gBAClB,IAAI,CAAC,EAAE,GAAG,IAAI,yBAAgB,CAAC,OAAO,CAAC,CAAC;gBACxC,MAAM;YACR,KAAK,WAAW;gBACd,IAAI,CAAC,EAAE,GAAG,IAAI,sBAAa,CAAC,OAAO,CAAC,CAAC;gBACrC,MAAM;YACR,KAAK,SAAS;gBACZ,IAAI,CAAC,EAAE,GAAG,IAAI,oBAAW,CAAC,OAAO,CAAC,CAAC;gBACnC,MAAM;YACR,KAAK,UAAU;gBACb,IAAI,CAAC,EAAE,GAAG,IAAI,qBAAY,CAAC,OAAO,CAAC,CAAC;gBACpC,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,EAAE,GAAG,IAAI,mBAAU,CAAC,OAAO,CAAC,CAAC;gBAClC,MAAM;YACR;gBACE,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,WAAW,CAAC,QAA0C;QACpD,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC;IAED,gBAAgB,CAAC,QAA0C;QACzD,IAAI,CAAC,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IACrC,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC;IAChC,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC;IACrC,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC;IAClC,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,IAAI,CACR,GAA4B,EAC5B,OAA+D;QAE/D,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,KAAK,CACT,GAA6B,EAC7B,OAAuE;QAEvE,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED,UAAU,CAAC,OAAqC;QAC9C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;CACF;AAxFD,oBAwFC"}
+{"version":3,"file":"wrap.js","sourceRoot":"","sources":["../../../ai/wrap.ts"],"names":[],"mappings":";;;AAEA,+CAA2E;AAE3E,kDAG+B;AAC/B,4CAAkE;AAElE,8CAAwE;AAExE,mDAGgC;AAKhC,0CAA4D;AAE5D,iDAG8B;AAE9B,6CAAqE;AAErE,4CAAkE;AAClE,6CAGyB;AAEzB,+CAAwE;AA6CxE,MAAa,IAAI;IACP,EAAE,CAAc;IAExB,YAAY,OAA2B;QACrC,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,QAAQ;gBACX,IAAI,CAAC,EAAE,GAAG,IAAI,oBAAU,CAAC,OAAO,CAAC,CAAC;gBAClC,MAAM;YACR,KAAK,cAAc;gBACjB,IAAI,CAAC,EAAE,GAAG,IAAI,wBAAe,CAAC,OAAO,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,aAAa;gBAChB,IAAI,CAAC,EAAE,GAAG,IAAI,wBAAe,CAAC,OAAO,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,MAAM;gBACT,IAAI,CAAC,EAAE,GAAG,IAAI,iBAAQ,CAAC,OAAO,CAAC,CAAC;gBAChC,MAAM;YACR,KAAK,UAAU;gBACb,IAAI,CAAC,EAAE,GAAG,IAAI,sBAAY,CAAC,OAAO,CAAC,CAAC;gBACpC,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,EAAE,GAAG,IAAI,mBAAU,CAAC,OAAO,CAAC,CAAC;gBAClC,MAAM;YACR,KAAK,eAAe;gBAClB,IAAI,CAAC,EAAE,GAAG,IAAI,yBAAgB,CAAC,OAAO,CAAC,CAAC;gBACxC,MAAM;YACR,KAAK,WAAW;gBACd,IAAI,CAAC,EAAE,GAAG,IAAI,sBAAa,CAAC,OAAO,CAAC,CAAC;gBACrC,MAAM;YACR,KAAK,SAAS;gBACZ,IAAI,CAAC,EAAE,GAAG,IAAI,oBAAW,CAAC,OAAO,CAAC,CAAC;gBACnC,MAAM;YACR,KAAK,UAAU;gBACb,IAAI,CAAC,EAAE,GAAG,IAAI,qBAAY,CAAC,OAAO,CAAC,CAAC;gBACpC,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,EAAE,GAAG,IAAI,mBAAU,CAAC,OAAO,CAAC,CAAC;gBAClC,MAAM;YACR;gBACE,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC;IAC3B,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC;IAChC,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC;IACrC,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC;IAClC,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;IAC/B,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,IAAI,CACR,GAA4B,EAC5B,OAA+D;QAE/D,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,KAAK,CACT,GAA6B,EAC7B,OAAuE;QAEvE,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED,UAAU,CAAC,OAAqC;QAC9C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;CACF;AApFD,oBAoFC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ax-llm/ax",
-  "version": "9.0.28",
+  "version": "9.0.30",
   "type": "module",
   "description": "The best library to work with LLMs",
   "repository": {