@ax-llm/ax 9.0.0 → 9.0.3

package/build/module/src/ai/index.js

@@ -1,14 +1,14 @@
-import { Anthropic } from './anthropic/api.js';
-import { AzureOpenAI } from './azure-openai/api.js';
-import { Cohere } from './cohere/api.js';
-import { DeepSeek } from './deepseek/api.js';
-import { GoogleGemini } from './google-gemini/api.js';
-import { Groq } from './groq/api.js';
-import { HuggingFace } from './huggingface/api.js';
-import { Mistral } from './mistral/api.js';
-import { Ollama } from './ollama/api.js';
-import { OpenAI } from './openai/api.js';
-import { Together } from './together/api.js';
+import { AxAnthropic } from './anthropic/api.js';
+import { AxAzureOpenAI } from './azure-openai/api.js';
+import { AxCohere } from './cohere/api.js';
+import { AxDeepSeek } from './deepseek/api.js';
+import { AxGoogleGemini } from './google-gemini/api.js';
+import { AxGroq } from './groq/api.js';
+import { AxHuggingFace } from './huggingface/api.js';
+import { AxMistral } from './mistral/api.js';
+import { AxOllama } from './ollama/api.js';
+import { AxOpenAI } from './openai/api.js';
+import { AxTogether } from './together/api.js';
 export * from './openai/index.js';
 export * from './azure-openai/index.js';
 export * from './huggingface/index.js';
@@ -23,30 +23,30 @@ export * from './ollama/index.js';
 export * from './types.js';
 export * from './balance.js';
 export * from './base.js';
-export const AI = (name, options) => {
+export const axAI = (name, options) => {
     switch (name) {
         case 'openai':
-            return new OpenAI(options);
+            return new AxOpenAI(options);
         case 'azure-openai':
-            return new AzureOpenAI(options);
+            return new AxAzureOpenAI(options);
         case 'huggingface':
-            return new HuggingFace(options);
+            return new AxHuggingFace(options);
         case 'groq':
-            return new Groq(options);
+            return new AxGroq(options);
         case 'together':
-            return new Together(options);
+            return new AxTogether(options);
         case 'cohere':
-            return new Cohere(options);
+            return new AxCohere(options);
         case 'google-gemini':
-            return new GoogleGemini(options);
+            return new AxGoogleGemini(options);
         case 'anthropic':
-            return new Anthropic(options);
+            return new AxAnthropic(options);
         case 'mistral':
-            return new Mistral(options);
+            return new AxMistral(options);
         case 'deepseek':
-            return new DeepSeek(options);
+            return new AxDeepSeek(options);
         case 'ollama':
-            return new Ollama(options);
+            return new AxOllama(options);
         default:
             throw new Error(`Unknown AI ${name}`);
     }

package/build/module/src/ai/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/ai/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAsB,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,WAAW,EAAwB,MAAM,uBAAuB,CAAC;AAC1E,OAAO,EAAE,MAAM,EAAmB,MAAM,iBAAiB,CAAC;AAC1D,OAAO,EAAE,QAAQ,EAAqB,MAAM,mBAAmB,CAAC;AAChE,OAAO,EAAE,YAAY,EAAyB,MAAM,wBAAwB,CAAC;AAC7E,OAAO,EAAE,IAAI,EAAiB,MAAM,eAAe,CAAC;AACpD,OAAO,EAAE,WAAW,EAAwB,MAAM,sBAAsB,CAAC;AACzE,OAAO,EAAE,OAAO,EAAoB,MAAM,kBAAkB,CAAC;AAC7D,OAAO,EAAE,MAAM,EAAmB,MAAM,iBAAiB,CAAC;AAC1D,OAAO,EAAE,MAAM,EAAmB,MAAM,iBAAiB,CAAC;AAC1D,OAAO,EAAE,QAAQ,EAAqB,MAAM,mBAAmB,CAAC;AAEhE,cAAc,mBAAmB,CAAC;AAClC,cAAc,yBAAyB,CAAC;AACxC,cAAc,wBAAwB,CAAC;AACvC,cAAc,qBAAqB,CAAC;AACpC,cAAc,mBAAmB,CAAC;AAClC,cAAc,0BAA0B,CAAC;AACzC,cAAc,sBAAsB,CAAC;AACrC,cAAc,iBAAiB,CAAC;AAChC,cAAc,oBAAoB,CAAC;AACnC,cAAc,qBAAqB,CAAC;AACpC,cAAc,mBAAmB,CAAC;AAClC,cAAc,YAAY,CAAC;AAC3B,cAAc,cAAc,CAAC;AAC7B,cAAc,WAAW,CAAC;AAiB1B,MAAM,CAAC,MAAM,EAAE,GAAG,CAChB,IAAY,EACZ,OAWC,EACD,EAAE;IACF,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,QAAQ;YACX,OAAO,IAAI,MAAM,CAAC,OAAqB,CAAC,CAAC;QAC3C,KAAK,cAAc;YACjB,OAAO,IAAI,WAAW,CAAC,OAA0B,CAAC,CAAC;QACrD,KAAK,aAAa;YAChB,OAAO,IAAI,WAAW,CAAC,OAA0B,CAAC,CAAC;QACrD,KAAK,MAAM;YACT,OAAO,IAAI,IAAI,CAAC,OAAmB,CAAC,CAAC;QACvC,KAAK,UAAU;YACb,OAAO,IAAI,QAAQ,CAAC,OAAuB,CAAC,CAAC;QAC/C,KAAK,QAAQ;YACX,OAAO,IAAI,MAAM,CAAC,OAAqB,CAAC,CAAC;QAC3C,KAAK,eAAe;YAClB,OAAO,IAAI,YAAY,CAAC,OAA2B,CAAC,CAAC;QACvD,KAAK,WAAW;YACd,OAAO,IAAI,SAAS,CAAC,OAAwB,CAAC,CAAC;QACjD,KAAK,SAAS;YACZ,OAAO,IAAI,OAAO,CAAC,OAAsB,CAAC,CAAC;QAC7C,KAAK,UAAU;YACb,OAAO,IAAI,QAAQ,CAAC,OAAuB,CAAC,CAAC;QAC/C,KAAK,QAAQ;YACX,OAAO,IAAI,MAAM,CAAC,OAAqB,CAAC,CAAC;QAC3C;YACE,MAAM,IAAI,KAAK,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/ai/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAwB,MAAM,oBAAoB,CAAC;AACvE,OAAO,EAAE,aAAa,EAA0B,MAAM,uBAAuB,CAAC;AAC9E,OAAO,EAAE,QAAQ,EAAqB,MAAM,iBAAiB,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAuB,MAAM,mBAAmB,CAAC;AACpE,OAAO,EACL,cAAc,EAEf,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAqB,MAAM,EAAE,MAAM,eAAe,CAAC;AAC1D,OAAO,EAAE,aAAa,EAA0B,MAAM,sBAAsB,CAAC;AAC7E,OAAO,EAAE,SAAS,EAAsB,MAAM,kBAAkB,CAAC;AACjE,OAAO,EAAE,QAAQ,EAAqB,MAAM,iBAAiB,CAAC;AAC9D,OAAO,EAAE,QAAQ,EAAqB,MAAM,iBAAiB,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAuB,MAAM,mBAAmB,CAAC;AAEpE,cAAc,mBAAmB,CAAC;AAClC,cAAc,yBAAyB,CAAC;AACxC,cAAc,wBAAwB,CAAC;AACvC,cAAc,qBAAqB,CAAC;AACpC,cAAc,mBAAmB,CAAC;AAClC,cAAc,0BAA0B,CAAC;AACzC,cAAc,sBAAsB,CAAC;AACrC,cAAc,iBAAiB,CAAC;AAChC,cAAc,oBAAoB,CAAC;AACnC,cAAc,qBAAqB,CAAC;AACpC,cAAc,mBAAmB,CAAC;AAClC,cAAc,YAAY,CAAC;AAC3B,cAAc,cAAc,CAAC;AAC7B,cAAc,WAAW,CAAC;AAiB1B,MAAM,CAAC,MAAM,IAAI,GAAG,CAClB,IAAc,EACd,OAWC,EACD,EAAE;IACF,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,QAAQ;YACX,OAAO,IAAI,QAAQ,CAAC,OAAuB,CAAC,CAAC;QAC/C,KAAK,cAAc;YACjB,OAAO,IAAI,aAAa,CAAC,OAA4B,CAAC,CAAC;QACzD,KAAK,aAAa;YAChB,OAAO,IAAI,aAAa,CAAC,OAA4B,CAAC,CAAC;QACzD,KAAK,MAAM;YACT,OAAO,IAAI,MAAM,CAAC,OAAuB,CAAC,CAAC;QAC7C,KAAK,UAAU;YACb,OAAO,IAAI,UAAU,CAAC,OAAyB,CAAC,CAAC;QACnD,KAAK,QAAQ;YACX,OAAO,IAAI,QAAQ,CAAC,OAAuB,CAAC,CAAC;QAC/C,KAAK,eAAe;YAClB,OAAO,IAAI,cAAc,CAAC,OAA6B,CAAC,CAAC;QAC3D,KAAK,WAAW;YACd,OAAO,IAAI,WAAW,CAAC,OAA0B,CAAC,CAAC;QACrD,KAAK,SAAS;YACZ,OAAO,IAAI,SAAS,CAAC,OAAwB,CAAC,CAAC;QACjD,KAAK,UAAU;YACb,OAAO,IAAI,UAAU,CAAC,OAAyB,CAAC,CAAC;QACnD,KAAK,QAAQ;YACX,OAAO,IAAI,QAAQ,CAAC,OAAuB,CAAC,CAAC;QAC/C;YACE,MAAM,IAAI,KAAK,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC,CAAC"}

package/build/module/src/ai/mistral/api.d.ts

@@ -1,27 +1,15 @@
-import type { AIServiceOptions } from '../../text/types.js';
-import { OpenAI } from '../openai/api.js';
-import type { OpenAIConfig } from '../openai/types.js';
-type MistralConfig = OpenAIConfig;
-/**
- * Mistral: Default Model options for text generation
- * @export
- */
-export declare const MistralDefaultConfig: () => MistralConfig;
-/**
- * Mistral: Default model options to use the more advanced model
- * @export
- */
-export declare const MistralBestConfig: () => OpenAIConfig;
-export interface MistralArgs {
+import { AxOpenAI } from '../openai/api.js';
+import type { AxOpenAIConfig } from '../openai/types.js';
+import type { AxAIServiceOptions } from '../types.js';
+type MistralConfig = AxOpenAIConfig;
+export declare const axMistralDefaultConfig: () => MistralConfig;
+export declare const axMistralBestConfig: () => AxOpenAIConfig;
+export interface AxMistralArgs {
     apiKey: string;
     config: Readonly<MistralConfig>;
-    options?: Readonly<AIServiceOptions>;
+    options?: Readonly<AxAIServiceOptions>;
 }
-/**
- * Mistral: AI Service
- * @export
- */
-export declare class Mistral extends OpenAI {
-    constructor({ apiKey, config, options }: Readonly<MistralArgs>);
+export declare class AxMistral extends AxOpenAI {
+    constructor({ apiKey, config, options }: Readonly<AxMistralArgs>);
 }
 export {};

package/build/module/src/ai/mistral/api.js

@@ -1,28 +1,17 @@
-import { BaseAIDefaultConfig } from '../base.js';
-import { OpenAI } from '../openai/api.js';
-import { MistralModel } from './types.js';
-/**
- * Mistral: Default Model options for text generation
- * @export
- */
-export const MistralDefaultConfig = () => structuredClone({
-    model: MistralModel.MistralSmall,
-    ...BaseAIDefaultConfig()
+import { axBaseAIDefaultConfig } from '../base.js';
+import { AxOpenAI } from '../openai/api.js';
+import { axModelInfoMistral } from './info.js';
+import { AxMistralModel } from './types.js';
+export const axMistralDefaultConfig = () => structuredClone({
+    model: AxMistralModel.MistralSmall,
+    ...axBaseAIDefaultConfig()
 });
-/**
- * Mistral: Default model options to use the more advanced model
- * @export
- */
-export const MistralBestConfig = () => structuredClone({
-    ...MistralDefaultConfig(),
-    model: MistralModel.MistralLarge
+export const axMistralBestConfig = () => structuredClone({
+    ...axMistralDefaultConfig(),
+    model: AxMistralModel.MistralLarge
 });
-/**
- * Mistral: AI Service
- * @export
- */
-export class Mistral extends OpenAI {
-    constructor({ apiKey, config = MistralDefaultConfig(), options }) {
+export class AxMistral extends AxOpenAI {
+    constructor({ apiKey, config = axMistralDefaultConfig(), options }) {
         if (!apiKey || apiKey === '') {
             throw new Error('Mistral API key not set');
         }
@@ -30,7 +19,8 @@ export class Mistral extends OpenAI {
             apiKey,
             config,
             options,
-            apiURL: 'https://api.mistral.ai/v1'
+            apiURL: 'https://api.mistral.ai/v1',
+            modelInfo: axModelInfoMistral
         });
         super.setName('Mistral');
     }

package/build/module/src/ai/mistral/api.js.map

@@ -1 +1 @@
-{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/mistral/api.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAG1C,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAI1C;;;GAGG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,GAAkB,EAAE,CACtD,eAAe,CAAC;IACd,KAAK,EAAE,YAAY,CAAC,YAAY;IAChC,GAAG,mBAAmB,EAAE;CACzB,CAAC,CAAC;AAEL;;;GAGG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,GAAiB,EAAE,CAClD,eAAe,CAAC;IACd,GAAG,oBAAoB,EAAE;IACzB,KAAK,EAAE,YAAY,CAAC,YAAY;CACjC,CAAC,CAAC;AAQL;;;GAGG;AACH,MAAM,OAAO,OAAQ,SAAQ,MAAM;IACjC,YAAY,EACV,MAAM,EACN,MAAM,GAAG,oBAAoB,EAAE,EAC/B,OAAO,EACe;QACtB,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QAED,KAAK,CAAC;YACJ,MAAM;YACN,MAAM;YACN,OAAO;YACP,MAAM,EAAE,2BAA2B;SACpC,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC3B,CAAC;CACF"}
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/mistral/api.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AACnD,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAI5C,OAAO,EAAE,kBAAkB,EAAE,MAAM,WAAW,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAI5C,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAAkB,EAAE,CACxD,eAAe,CAAC;IACd,KAAK,EAAE,cAAc,CAAC,YAAY;IAClC,GAAG,qBAAqB,EAAE;CAC3B,CAAC,CAAC;AAEL,MAAM,CAAC,MAAM,mBAAmB,GAAG,GAAmB,EAAE,CACtD,eAAe,CAAC;IACd,GAAG,sBAAsB,EAAE;IAC3B,KAAK,EAAE,cAAc,CAAC,YAAY;CACnC,CAAC,CAAC;AAQL,MAAM,OAAO,SAAU,SAAQ,QAAQ;IACrC,YAAY,EACV,MAAM,EACN,MAAM,GAAG,sBAAsB,EAAE,EACjC,OAAO,EACiB;QACxB,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QAED,KAAK,CAAC;YACJ,MAAM;YACN,MAAM;YACN,OAAO;YACP,MAAM,EAAE,2BAA2B;YACnC,SAAS,EAAE,kBAAkB;SAC9B,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC3B,CAAC;CACF"}

package/build/module/src/ai/mistral/info.d.ts

@@ -1,2 +1,2 @@
-import type { TextModelInfo } from '../types.js';
-export declare const textModelInfos: TextModelInfo[];
+import type { AxModelInfo } from '../types.js';
+export declare const axModelInfoMistral: AxModelInfo[];

package/build/module/src/ai/mistral/info.js

@@ -1,31 +1,31 @@
-import { MistralModel } from './types.js';
-export const textModelInfos = [
+import { AxMistralModel } from './types.js';
+export const axModelInfoMistral = [
     {
-        name: MistralModel.Mistral7B,
+        name: AxMistralModel.Mistral7B,
         currency: 'USD',
         promptTokenCostPer1M: 0.25,
         completionTokenCostPer1M: 0.25
     },
     {
-        name: MistralModel.Mistral8x7B,
+        name: AxMistralModel.Mistral8x7B,
         currency: 'USD',
         promptTokenCostPer1M: 0.7,
         completionTokenCostPer1M: 0.7
     },
     {
-        name: MistralModel.MistralSmall,
+        name: AxMistralModel.MistralSmall,
         currency: 'USD',
         promptTokenCostPer1M: 2,
         completionTokenCostPer1M: 6
     },
     {
-        name: MistralModel.MistralMedium,
+        name: AxMistralModel.MistralMedium,
         currency: 'USD',
         promptTokenCostPer1M: 2.7,
         completionTokenCostPer1M: 8.1
     },
     {
-        name: MistralModel.MistralLarge,
+        name: AxMistralModel.MistralLarge,
         currency: 'USD',
         promptTokenCostPer1M: 8,
         completionTokenCostPer1M: 24

package/build/module/src/ai/mistral/info.js.map

@@ -1 +1 @@
-{"version":3,"file":"info.js","sourceRoot":"","sources":["../../../../../src/ai/mistral/info.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE1C,MAAM,CAAC,MAAM,cAAc,GAAoB;IAC7C;QACE,IAAI,EAAE,YAAY,CAAC,SAAS;QAC5B,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,IAAI;QAC1B,wBAAwB,EAAE,IAAI;KAC/B;IACD;QACE,IAAI,EAAE,YAAY,CAAC,WAAW;QAC9B,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,YAAY,CAAC,YAAY;QAC/B,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,CAAC;QACvB,wBAAwB,EAAE,CAAC;KAC5B;IACD;QACE,IAAI,EAAE,YAAY,CAAC,aAAa;QAChC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,YAAY,CAAC,YAAY;QAC/B,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,CAAC;QACvB,wBAAwB,EAAE,EAAE;KAC7B;CACF,CAAC"}
+{"version":3,"file":"info.js","sourceRoot":"","sources":["../../../../../src/ai/mistral/info.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAE5C,MAAM,CAAC,MAAM,kBAAkB,GAAkB;IAC/C;QACE,IAAI,EAAE,cAAc,CAAC,SAAS;QAC9B,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,IAAI;QAC1B,wBAAwB,EAAE,IAAI;KAC/B;IACD;QACE,IAAI,EAAE,cAAc,CAAC,WAAW;QAChC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,cAAc,CAAC,YAAY;QACjC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,CAAC;QACvB,wBAAwB,EAAE,CAAC;KAC5B;IACD;QACE,IAAI,EAAE,cAAc,CAAC,aAAa;QAClC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,cAAc,CAAC,YAAY;QACjC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,CAAC;QACvB,wBAAwB,EAAE,EAAE;KAC7B;CACF,CAAC"}

package/build/module/src/ai/mistral/types.d.ts

@@ -1,18 +1,10 @@
-/**
- * Mistral: Models for text generation
- * @export
- */
-export declare enum MistralModel {
+export declare enum AxMistralModel {
     Mistral7B = "open-mistral-7b",
     Mistral8x7B = "open-mixtral-8x7b",
     MistralSmall = "mistral-small-latest",
     MistralMedium = "mistral-medium-latest",
     MistralLarge = "mistral-large-latest"
 }
-/**
- * Mistral: Models for use in embeddings
- * @export
- */
-export declare enum MistralEmbedModels {
+export declare enum AxMistralEmbedModels {
     MistralEmbed = "mistral-embed"
 }

package/build/module/src/ai/mistral/types.js

@@ -1,21 +1,13 @@
-/**
- * Mistral: Models for text generation
- * @export
- */
-export var MistralModel;
-(function (MistralModel) {
-    MistralModel["Mistral7B"] = "open-mistral-7b";
-    MistralModel["Mistral8x7B"] = "open-mixtral-8x7b";
-    MistralModel["MistralSmall"] = "mistral-small-latest";
-    MistralModel["MistralMedium"] = "mistral-medium-latest";
-    MistralModel["MistralLarge"] = "mistral-large-latest";
-})(MistralModel || (MistralModel = {}));
-/**
- * Mistral: Models for use in embeddings
- * @export
- */
-export var MistralEmbedModels;
-(function (MistralEmbedModels) {
-    MistralEmbedModels["MistralEmbed"] = "mistral-embed";
-})(MistralEmbedModels || (MistralEmbedModels = {}));
+export var AxMistralModel;
+(function (AxMistralModel) {
+    AxMistralModel["Mistral7B"] = "open-mistral-7b";
+    AxMistralModel["Mistral8x7B"] = "open-mixtral-8x7b";
+    AxMistralModel["MistralSmall"] = "mistral-small-latest";
+    AxMistralModel["MistralMedium"] = "mistral-medium-latest";
+    AxMistralModel["MistralLarge"] = "mistral-large-latest";
+})(AxMistralModel || (AxMistralModel = {}));
+export var AxMistralEmbedModels;
+(function (AxMistralEmbedModels) {
+    AxMistralEmbedModels["MistralEmbed"] = "mistral-embed";
+})(AxMistralEmbedModels || (AxMistralEmbedModels = {}));
 //# sourceMappingURL=types.js.map

package/build/module/src/ai/mistral/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/mistral/types.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,MAAM,CAAN,IAAY,YAMX;AAND,WAAY,YAAY;IACtB,6CAA6B,CAAA;IAC7B,iDAAiC,CAAA;IACjC,qDAAqC,CAAA;IACrC,uDAAuC,CAAA;IACvC,qDAAqC,CAAA;AACvC,CAAC,EANW,YAAY,KAAZ,YAAY,QAMvB;AAED;;;GAGG;AACH,MAAM,CAAN,IAAY,kBAEX;AAFD,WAAY,kBAAkB;IAC5B,oDAA8B,CAAA;AAChC,CAAC,EAFW,kBAAkB,KAAlB,kBAAkB,QAE7B"}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/mistral/types.ts"],"names":[],"mappings":"AAAA,MAAM,CAAN,IAAY,cAMX;AAND,WAAY,cAAc;IACxB,+CAA6B,CAAA;IAC7B,mDAAiC,CAAA;IACjC,uDAAqC,CAAA;IACrC,yDAAuC,CAAA;IACvC,uDAAqC,CAAA;AACvC,CAAC,EANW,cAAc,KAAd,cAAc,QAMzB;AAED,MAAM,CAAN,IAAY,oBAEX;AAFD,WAAY,oBAAoB;IAC9B,sDAA8B,CAAA;AAChC,CAAC,EAFW,oBAAoB,KAApB,oBAAoB,QAE/B"}

package/build/module/src/ai/ollama/api.d.ts

@@ -1,24 +1,21 @@
-import type { AIServiceOptions } from '../../text/types.js';
-import { OpenAI } from '../openai/api.js';
-import type { OpenAIConfig } from '../openai/types.js';
-export type OllamaAIConfig = OpenAIConfig;
-/**
- * OllamaAI: Default Model options for text generation
- * @export
- */
-export declare const OllamaDefaultConfig: () => Omit<OllamaAIConfig, 'model'>;
-export declare const OllamaDefaultCreativeConfig: () => Omit<OllamaAIConfig, 'model'>;
-export type OllamaArgs = {
-    model: string;
+import { AxOpenAI } from '../openai/api.js';
+import type { AxOpenAIConfig } from '../openai/types.js';
+import type { AxAIServiceOptions } from '../types.js';
+export type AxOllamaAIConfig = AxOpenAIConfig;
+export declare const axOllamaDefaultConfig: () => AxOllamaAIConfig;
+export declare const axOllamaDefaultCreativeConfig: () => AxOllamaAIConfig;
+export type AxOllamaArgs = {
+    model?: string;
+    embedModel?: string;
     url?: string;
     apiKey?: string;
-    config?: Readonly<Omit<OllamaAIConfig, 'model'>>;
-    options?: Readonly<AIServiceOptions>;
+    config?: Readonly<AxOllamaAIConfig>;
+    options?: Readonly<AxAIServiceOptions>;
 };
 /**
  * OllamaAI: AI Service
  * @export
  */
-export declare class Ollama extends OpenAI {
-    constructor({ apiKey, url, model, config, options }: Readonly<OllamaArgs>);
+export declare class AxOllama extends AxOpenAI {
+    constructor({ apiKey, url, model, embedModel, config, options }: Readonly<AxOllamaArgs>);
 }

package/build/module/src/ai/ollama/api.js

@@ -1,25 +1,29 @@
-import { BaseAIDefaultConfig, BaseAIDefaultCreativeConfig } from '../base.js';
-import { OpenAI } from '../openai/api.js';
-/**
- * OllamaAI: Default Model options for text generation
- * @export
- */
-export const OllamaDefaultConfig = () => structuredClone({
-    ...BaseAIDefaultConfig()
+import { axBaseAIDefaultConfig, axBaseAIDefaultCreativeConfig } from '../base.js';
+import { AxOpenAI } from '../openai/api.js';
+export const axOllamaDefaultConfig = () => structuredClone({
+    ...axBaseAIDefaultConfig(),
+    model: 'nous-hermes2',
+    embedModel: 'all-minilm'
 });
-export const OllamaDefaultCreativeConfig = () => structuredClone({
-    ...BaseAIDefaultCreativeConfig()
+export const axOllamaDefaultCreativeConfig = () => structuredClone({
+    ...axBaseAIDefaultCreativeConfig(),
+    model: 'nous-hermes2',
+    embedModel: 'all-minilm'
 });
 /**
  * OllamaAI: AI Service
  * @export
  */
-export class Ollama extends OpenAI {
-    constructor({ apiKey = 'not-set', url = 'http://localhost:11434', model, config = OllamaDefaultConfig(), options }) {
+export class AxOllama extends AxOpenAI {
+    constructor({ apiKey = 'not-set', url = 'http://localhost:11434', model, embedModel, config = axOllamaDefaultConfig(), options }) {
         super({
             apiKey,
             options,
-            config: { ...config, model },
+            config: {
+                ...config,
+                ...(model ? { model } : {}),
+                ...(embedModel ? { embedModel } : {})
+            },
             apiURL: new URL('/v1', url).href
         });
         super.setName('Ollama');

package/build/module/src/ai/ollama/api.js.map

@@ -1 +1 @@
-{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/ollama/api.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,2BAA2B,EAAE,MAAM,YAAY,CAAC;AAC9E,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAK1C;;;GAGG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,GAAkC,EAAE,CACrE,eAAe,CAAC;IACd,GAAG,mBAAmB,EAAE;CACzB,CAAC,CAAC;AAEL,MAAM,CAAC,MAAM,2BAA2B,GAAG,GAAkC,EAAE,CAC7E,eAAe,CAAC;IACd,GAAG,2BAA2B,EAAE;CACjC,CAAC,CAAC;AAUL;;;GAGG;AACH,MAAM,OAAO,MAAO,SAAQ,MAAM;IAChC,YAAY,EACV,MAAM,GAAG,SAAS,EAClB,GAAG,GAAG,wBAAwB,EAC9B,KAAK,EACL,MAAM,GAAG,mBAAmB,EAAE,EAC9B,OAAO,EACc;QACrB,KAAK,CAAC;YACJ,MAAM;YACN,OAAO;YACP,MAAM,EAAE,EAAE,GAAG,MAAM,EAAE,KAAK,EAAE;YAC5B,MAAM,EAAE,IAAI,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI;SACjC,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC1B,CAAC;CACF"}
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/ollama/api.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,qBAAqB,EACrB,6BAA6B,EAC9B,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAM5C,MAAM,CAAC,MAAM,qBAAqB,GAAG,GAAqB,EAAE,CAC1D,eAAe,CAAC;IACd,GAAG,qBAAqB,EAAE;IAC1B,KAAK,EAAE,cAAc;IACrB,UAAU,EAAE,YAAY;CACzB,CAAC,CAAC;AAEL,MAAM,CAAC,MAAM,6BAA6B,GAAG,GAAqB,EAAE,CAClE,eAAe,CAAC;IACd,GAAG,6BAA6B,EAAE;IAClC,KAAK,EAAE,cAAc;IACrB,UAAU,EAAE,YAAY;CACzB,CAAC,CAAC;AAWL;;;GAGG;AACH,MAAM,OAAO,QAAS,SAAQ,QAAQ;IACpC,YAAY,EACV,MAAM,GAAG,SAAS,EAClB,GAAG,GAAG,wBAAwB,EAC9B,KAAK,EACL,UAAU,EACV,MAAM,GAAG,qBAAqB,EAAE,EAChC,OAAO,EACgB;QACvB,KAAK,CAAC;YACJ,MAAM;YACN,OAAO;YACP,MAAM,EAAE;gBACN,GAAG,MAAM;gBACT,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3B,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACtC;YACD,MAAM,EAAE,IAAI,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI;SACjC,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC1B,CAAC;CACF"}

package/build/module/src/ai/openai/api.d.ts

@@ -1,49 +1,28 @@
-import type { AIPromptConfig, AIServiceOptions } from '../../text/types.js';
-import type { AITextChatRequest, AITextEmbedRequest } from '../../types/index.js';
 import type { API } from '../../util/apicall.js';
-import { BaseAI } from '../base.js';
-import type { EmbedResponse, TextModelConfig, TextResponse } from '../types.js';
-import { type OpenAIChatRequest, type OpenAIChatResponse, type OpenAIChatResponseDelta, type OpenAIConfig, type OpenAIEmbedRequest, type OpenAIEmbedResponse } from './types.js';
-/**
- * OpenAI: Default Model options for text generation
- * @export
- */
-export declare const OpenAIDefaultConfig: () => OpenAIConfig;
-/**
- * OpenAI: Default model options to use the more advanced model
- * @export
- */
-export declare const OpenAIBestConfig: () => OpenAIConfig;
-/**
- * OpenAI: Default model options for more creative text generation
- * @export
- */
-export declare const OpenAICreativeConfig: () => OpenAIConfig;
-/**
- * OpenAI: Default model options for more fast text generation
- * @export
- */
-export declare const OpenAIFastConfig: () => OpenAIConfig;
-export interface OpenAIArgs {
+import { AxBaseAI } from '../base.js';
+import type { AxAIPromptConfig, AxAIServiceOptions, AxChatRequest, AxChatResponse, AxEmbedRequest, AxEmbedResponse, AxModelConfig, AxModelInfo } from '../types.js';
+import { type AxOpenAIChatRequest, type AxOpenAIChatResponse, type AxOpenAIChatResponseDelta, type AxOpenAIConfig, type AxOpenAIEmbedRequest, type AxOpenAIEmbedResponse } from './types.js';
+export declare const axOpenAIDefaultConfig: () => AxOpenAIConfig;
+export declare const axOpenAIBestConfig: () => AxOpenAIConfig;
+export declare const axOpenAICreativeConfig: () => AxOpenAIConfig;
+export declare const axOpenAIFastConfig: () => AxOpenAIConfig;
+export interface AxOpenAIArgs {
     apiKey: string;
     apiURL?: string;
-    config?: Readonly<OpenAIConfig>;
-    options?: Readonly<AIServiceOptions & {
+    config?: Readonly<AxOpenAIConfig>;
+    options?: Readonly<AxAIServiceOptions & {
         streamingUsage?: boolean;
     }>;
+    modelInfo?: Readonly<AxModelInfo[]>;
 }
-/**
- * OpenAI: AI Service
- * @export
- */
-export declare class OpenAI extends BaseAI<OpenAIChatRequest, OpenAIEmbedRequest, OpenAIChatResponse, OpenAIChatResponseDelta, OpenAIEmbedResponse> {
+export declare class AxOpenAI extends AxBaseAI<AxOpenAIChatRequest, AxOpenAIEmbedRequest, AxOpenAIChatResponse, AxOpenAIChatResponseDelta, AxOpenAIEmbedResponse> {
     private config;
     private streamingUsage;
-    constructor({ apiKey, config, options, apiURL }: Readonly<OpenAIArgs>);
-    getModelConfig(): TextModelConfig;
-    generateChatReq: (req: Readonly<AITextChatRequest>, _config: Readonly<AIPromptConfig>) => [API, OpenAIChatRequest];
-    generateEmbedReq: (req: Readonly<AITextEmbedRequest>) => [API, OpenAIEmbedRequest];
-    generateChatResp: (resp: Readonly<OpenAIChatResponse>) => TextResponse;
-    generateChatStreamResp: (resp: Readonly<OpenAIChatResponseDelta>, state: object) => TextResponse;
-    generateEmbedResp: (resp: Readonly<OpenAIEmbedResponse>) => EmbedResponse;
+    constructor({ apiKey, config, options, apiURL, modelInfo }: Readonly<AxOpenAIArgs>);
+    getModelConfig(): AxModelConfig;
+    generateChatReq: (req: Readonly<AxChatRequest>, _config: Readonly<AxAIPromptConfig>) => [API, AxOpenAIChatRequest];
+    generateEmbedReq: (req: Readonly<AxEmbedRequest>) => [API, AxOpenAIEmbedRequest];
+    generateChatResp: (resp: Readonly<AxOpenAIChatResponse>) => AxChatResponse;
+    generateChatStreamResp: (resp: Readonly<AxOpenAIChatResponseDelta>, state: object) => AxChatResponse;
+    generateEmbedResp: (resp: Readonly<AxOpenAIEmbedResponse>) => AxEmbedResponse;
 }