@ax-llm/ax 11.0.8 → 11.0.9

package/index.cjs

@@ -905,7 +905,8 @@ var AxBaseAI = class {
     const startTime = performance.now();
     let isError = false;
     try {
-      return this._chat1(req, options);
+      const result = await this._chat1(req, options);
+      return result;
     } catch (error) {
       isError = true;
       throw error;
@@ -958,13 +959,31 @@ var AxBaseAI = class {
     }
     return await this._chat2(model, modelConfig, req, options);
   }
+  cleanupFunctionSchema(fn) {
+    const cleanFn = { ...fn };
+    if (cleanFn.parameters) {
+      const cleanParams = { ...cleanFn.parameters };
+      if (Array.isArray(cleanParams.required) && cleanParams.required.length === 0) {
+        delete cleanParams.required;
+      }
+      if (cleanParams.properties && Object.keys(cleanParams.properties).length === 0) {
+        delete cleanParams.properties;
+      }
+      if (Object.keys(cleanParams).length === 0 || Object.keys(cleanParams).length === 1 && cleanParams.type === "object") {
+        delete cleanFn.parameters;
+      } else {
+        cleanFn.parameters = cleanParams;
+      }
+    }
+    return cleanFn;
+  }
   async _chat2(model, modelConfig, chatReq, options, span) {
     if (!this.aiImpl.createChatReq) {
       throw new Error("generateChatReq not implemented");
     }
     let functions;
     if (chatReq.functions && chatReq.functions.length > 0) {
-      functions = chatReq.functions;
+      functions = chatReq.functions.map((fn2) => this.cleanupFunctionSchema(fn2));
     }
     const req = {
       ...chatReq,
@@ -6134,14 +6153,12 @@ function processChildAgentFunction(childFunction, parentValues, parentInputKeys,
 }
 var AxAgent = class {
   ai;
-  signature;
   program;
   functions;
   agents;
   disableSmartModelRouting;
   excludeFieldsFromPassthrough;
   name;
-  description;
   subAgentList;
   func;
   constructor({
@@ -6157,8 +6174,8 @@ var AxAgent = class {
     this.functions = functions;
     this.disableSmartModelRouting = options?.disableSmartModelRouting;
     this.excludeFieldsFromPassthrough = options?.excludeFieldsFromPassthrough ?? [];
-    this.signature = new AxSignature(signature);
-    this.signature.setDescription(description);
+    const sig = new AxSignature(signature);
+    sig.setDescription(description);
     if (!name || name.length < 5) {
       throw new Error(
         `Agent name must be at least 10 characters (more descriptive): ${name}`
@@ -6169,17 +6186,16 @@ var AxAgent = class {
         `Agent description must be at least 20 characters (explain in detail what the agent does): ${description}`
       );
     }
-    this.program = new AxGen(this.signature, options);
+    this.program = new AxGen(sig, options);
     for (const agent of agents ?? []) {
       this.program.register(agent);
     }
     this.name = name;
-    this.description = description;
     this.subAgentList = agents?.map((a) => a.getFunction().name).join(", ");
     this.func = {
       name: toCamelCase(this.name),
-      description: this.description,
-      parameters: this.signature.toJSONSchema(),
+      description,
+      parameters: sig.toJSONSchema(),
       func: () => this.forward
     };
     const mm = ai?.getModelList();
@@ -6235,7 +6251,7 @@ var AxAgent = class {
   init(parentAi, values, options) {
     const ai = this.ai ?? parentAi;
     const mm = ai?.getModelList();
-    const parentSchema = this.signature.getInputFields();
+    const parentSchema = this.program.getSignature().getInputFields();
     const parentKeys = parentSchema.map((p) => p.name);
     const agentFuncs = this.agents?.map((agent) => {
       const f = agent.getFeatures();
@@ -6283,8 +6299,7 @@ var AxAgent = class {
         "Agent description must be at least 20 characters (explain in detail what the agent does)"
       );
     }
-    this.description = description;
-    this.signature.setDescription(description);
+    this.program.getSignature().setDescription(description);
     this.func.description = description;
   }
 };