@ax-llm/ax 11.0.1 → 11.0.2

package/index.cjs

@@ -6094,6 +6094,7 @@ var AxAgent = class {
   program;
   functions;
   agents;
+  disableSmartModelRouting;
   name;
   description;
   subAgentList;
@@ -6109,6 +6110,7 @@ var AxAgent = class {
     this.ai = ai;
     this.agents = agents;
     this.functions = functions;
+    this.disableSmartModelRouting = options?.disableSmartModelRouting;
     this.signature = new AxSignature(signature);
     this.signature.setDescription(description);
     if (!name || name.length < 5) {
@@ -6135,7 +6137,7 @@ var AxAgent = class {
       func: () => this.forward
     };
     const mm = ai?.getModelList();
-    if (mm) {
+    if (mm && !this.disableSmartModelRouting) {
       this.func.parameters = addModelParameter(this.func.parameters, mm);
     }
   }
@@ -6175,11 +6177,16 @@ var AxAgent = class {
       func: wrappedFunc
     };
   }
+  getFeatures() {
+    return {
+      canConfigureSmartModelRouting: this.ai !== void 0
+    };
+  }
   init(parentAi, options) {
     const ai = this.ai ?? parentAi;
     const mm = ai?.getModelList();
     const agentFuncs = this.agents?.map((a) => a.getFunction())?.map(
-      (f) => mm ? { ...f, parameters: addModelParameter(f.parameters, mm) } : f
+      (f) => mm && !this.disableSmartModelRouting && this.agents?.find((a) => a.getFunction().name === f.name)?.getFeatures().canConfigureSmartModelRouting ? { ...f, parameters: addModelParameter(f.parameters, mm) } : f
     );
     const functions = [
       ...options?.functions ?? this.functions ?? [],