@ax-llm/ax 9.0.26 → 9.0.28

package/cjs/ai/wrap.js

@@ -14,48 +14,51 @@ 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`);
         }
     }
+    setModelMap(modelMap) {
+        this.ai.setModelMap(modelMap);
+    }
+    setEmbedModelMap(modelMap) {
+        this.ai.setEmbedModelMap(modelMap);
+    }
     getName() {
         return this.ai.getName();
     }
@@ -72,23 +75,9 @@ class AxAI {
         return this.ai.getFeatures();
     }
     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;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"}

package/package.json

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