@ax-llm/ax 9.0.26 → 9.0.29

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,mBAAU;IAC1C,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,5 +1,6 @@
 import type { ReadableStream } from 'stream/web';
 import type { AxTracer } from '../trace/index.js';
+export type AxAIModelMap = Record<string, string>;
 export type AxModelInfo = {
     name: string;
     currency?: string;
@@ -144,15 +145,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;
     };
+    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,29 +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';
-export type AxAIModelMap<T> = Record<string, T>;
-export interface AxAIOptions<TM = string, TEM = string> {
-    modelMap?: AxAIModelMap<TM>;
-    embedModelMap?: AxAIModelMap<TEM>;
-}
-export type AxAIArgs = (AxAIOpenAIArgs & AxAIOptions<AxAIOpenAIModel, AxAIOpenAIEmbedModel>) | (AxAIAzureOpenAIArgs & AxAIOptions<AxAIOpenAIModel, AxAIOpenAIEmbedModel>) | (AxAITogetherArgs & AxAIOptions<string, string>) | (AxAIAnthropicArgs & AxAIOptions<AxAIAnthropicModel, AxAIAnthropicModel>) | (AxAIGroqArgs & AxAIOptions<AxAIGroqModel, AxAIGroqModel>) | (AxAIGoogleGeminiArgs & AxAIOptions<AxAIGoogleGeminiModel, AxAIGoogleGeminiEmbedModel>) | (AxAICohereArgs & AxAIOptions<AxAICohereModel, AxAICohereEmbedModel>) | (AxAIHuggingFaceArgs & AxAIOptions<AxAIHuggingFaceModel, AxAIHuggingFaceModel>) | (AxAIMistralArgs & AxAIOptions<AxAIMistralModel, AxAIMistralModel>) | (AxAIDeepSeekArgs & AxAIOptions<AxAIDeepSeekModel, AxAIDeepSeekModel>) | (AxAIOllamaArgs & AxAIOptions<string, string>);
+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;
-    private options;
     constructor(options: Readonly<AxAIArgs>);
     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

@@ -14,43 +14,40 @@ const api_js_10 = require("./openai/api.js");
 const api_js_11 = require("./together/api.js");
 class AxAI {
     ai;
-    options;
     constructor(options) {
-        const { modelMap, embedModelMap, ...args } = options;
-        this.options = { modelMap, embedModelMap };
-        switch (args.name) {
+        switch (options.name) {
             case 'openai':
-                this.ai = new api_js_10.AxAIOpenAI(args);
+                this.ai = new api_js_10.AxAIOpenAI(options);
                 break;
             case 'azure-openai':
-                this.ai = new api_js_2.AxAIAzureOpenAI(args);
+                this.ai = new api_js_2.AxAIAzureOpenAI(options);
                 break;
             case 'huggingface':
-                this.ai = new api_js_7.AxAIHuggingFace(args);
+                this.ai = new api_js_7.AxAIHuggingFace(options);
                 break;
             case 'groq':
-                this.ai = new api_js_6.AxAIGroq(args);
+                this.ai = new api_js_6.AxAIGroq(options);
                 break;
             case 'together':
-                this.ai = new api_js_11.AxAITogether(args);
+                this.ai = new api_js_11.AxAITogether(options);
                 break;
             case 'cohere':
-                this.ai = new api_js_3.AxAICohere(args);
+                this.ai = new api_js_3.AxAICohere(options);
                 break;
             case 'google-gemini':
-                this.ai = new api_js_5.AxAIGoogleGemini(args);
+                this.ai = new api_js_5.AxAIGoogleGemini(options);
                 break;
             case 'anthropic':
-                this.ai = new api_js_1.AxAIAnthropic(args);
+                this.ai = new api_js_1.AxAIAnthropic(options);
                 break;
             case 'mistral':
-                this.ai = new api_js_8.AxAIMistral(args);
+                this.ai = new api_js_8.AxAIMistral(options);
                 break;
             case 'deepseek':
-                this.ai = new api_js_4.AxAIDeepSeek(args);
+                this.ai = new api_js_4.AxAIDeepSeek(options);
                 break;
             case 'ollama':
-                this.ai = new api_js_9.AxAIOllama(args);
+                this.ai = new api_js_9.AxAIOllama(options);
                 break;
             default:
                 throw new Error(`Unknown AI`);
@@ -71,24 +68,13 @@ class AxAI {
     getFeatures() {
         return this.ai.getFeatures();
     }
+    getModelMap() {
+        return this.ai.getModelMap();
+    }
     async chat(req, options) {
-        if (this.options?.modelMap && req.model) {
-            const model = this.options.modelMap[req.model];
-            if (!model || model.length === 0) {
-                throw new Error(`Model not found in model map: ${req.model}`);
-            }
-            return await this.ai.chat({ ...req, model }, options);
-        }
         return await this.ai.chat(req, options);
     }
     async embed(req, options) {
-        if (this.options?.embedModelMap && req.embedModel) {
-            const embedModel = this.options.embedModelMap[req.embedModel];
-            if (!embedModel || embedModel.length === 0) {
-                throw new Error(`Model not found in embed model map: ${req.embedModel}`);
-            }
-            return await this.ai.embed({ ...req, embedModel }, options);
-        }
         return await this.ai.embed(req, options);
     }
     setOptions(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;AAqDxE,MAAa,IAAI;IACP,EAAE,CAAc;IAChB,OAAO,CAAc;IAE7B,YAAY,OAA2B;QACrC,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,IAAI,EAAE,GAAG,OAAO,CAAC;QACrD,IAAI,CAAC,OAAO,GAAG,EAAE,QAAQ,EAAE,aAAa,EAAE,CAAC;QAE3C,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,QAAQ;gBACX,IAAI,CAAC,EAAE,GAAG,IAAI,oBAAU,CAAC,IAAI,CAAC,CAAC;gBAC/B,MAAM;YACR,KAAK,cAAc;gBACjB,IAAI,CAAC,EAAE,GAAG,IAAI,wBAAe,CAAC,IAAI,CAAC,CAAC;gBACpC,MAAM;YACR,KAAK,aAAa;gBAChB,IAAI,CAAC,EAAE,GAAG,IAAI,wBAAe,CAAC,IAAI,CAAC,CAAC;gBACpC,MAAM;YACR,KAAK,MAAM;gBACT,IAAI,CAAC,EAAE,GAAG,IAAI,iBAAQ,CAAC,IAAI,CAAC,CAAC;gBAC7B,MAAM;YACR,KAAK,UAAU;gBACb,IAAI,CAAC,EAAE,GAAG,IAAI,sBAAY,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,EAAE,GAAG,IAAI,mBAAU,CAAC,IAAI,CAAC,CAAC;gBAC/B,MAAM;YACR,KAAK,eAAe;gBAClB,IAAI,CAAC,EAAE,GAAG,IAAI,yBAAgB,CAAC,IAAI,CAAC,CAAC;gBACrC,MAAM;YACR,KAAK,WAAW;gBACd,IAAI,CAAC,EAAE,GAAG,IAAI,sBAAa,CAAC,IAAI,CAAC,CAAC;gBAClC,MAAM;YACR,KAAK,SAAS;gBACZ,IAAI,CAAC,EAAE,GAAG,IAAI,oBAAW,CAAC,IAAI,CAAC,CAAC;gBAChC,MAAM;YACR,KAAK,UAAU;gBACb,IAAI,CAAC,EAAE,GAAG,IAAI,qBAAY,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,EAAE,GAAG,IAAI,mBAAU,CAAC,IAAI,CAAC,CAAC;gBAC/B,MAAM;YACR;gBACE,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IACD,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,IAAI,IAAI,CAAC,OAAO,EAAE,QAAQ,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACxC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC/C,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjC,MAAM,IAAI,KAAK,CAAC,iCAAiC,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC;YAChE,CAAC;YACD,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,KAAK,EAAE,EAAE,OAAO,CAAC,CAAC;QACxD,CAAC;QACD,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,KAAK,CACT,GAA6B,EAC7B,OAAuE;QAEvE,IAAI,IAAI,CAAC,OAAO,EAAE,aAAa,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;YAClD,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAC9D,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAI,KAAK,CACb,uCAAuC,GAAG,CAAC,UAAU,EAAE,CACxD,CAAC;YACJ,CAAC;YACD,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,GAAG,GAAG,EAAE,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;QAC9D,CAAC;QACD,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;AAnGD,oBAmGC"}
+{"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.26",
+  "version": "9.0.29",
   "type": "module",
   "description": "The best library to work with LLMs",
   "repository": {