@ax-llm/ax 11.0.8 → 11.0.9

package/index.d.cts

@@ -326,6 +326,7 @@ declare class AxBaseAI<TModel, TEmbedModel, TChatRequest, TEmbedRequest, TChatRe
     getMetrics(): AxAIServiceMetrics;
     chat(req: Readonly<AxChatRequest<TModel>>, options?: Readonly<AxAIPromptConfig & AxAIServiceActionOptions<TModel, TEmbedModel>>): Promise<AxChatResponse | ReadableStream$1<AxChatResponse>>;
     private _chat1;
+    private cleanupFunctionSchema;
     private _chat2;
     embed(req: Readonly<AxEmbedRequest<TEmbedModel>>, options?: Readonly<AxAIServiceActionOptions<TModel, TEmbedModel>>): Promise<AxEmbedResponse>;
     private _embed1;
@@ -1637,14 +1638,12 @@ interface AxAgentFeatures {
  */
 declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut = AxGenOut> implements AxAgentic {
     private ai?;
-    private signature;
     private program;
     private functions?;
     private agents?;
     private disableSmartModelRouting?;
     private excludeFieldsFromPassthrough;
     private name;
-    private description;
     private subAgentList?;
     private func;
     constructor({ ai, name, description, signature, agents, functions, }: Readonly<{

package/index.d.ts

@@ -326,6 +326,7 @@ declare class AxBaseAI<TModel, TEmbedModel, TChatRequest, TEmbedRequest, TChatRe
     getMetrics(): AxAIServiceMetrics;
     chat(req: Readonly<AxChatRequest<TModel>>, options?: Readonly<AxAIPromptConfig & AxAIServiceActionOptions<TModel, TEmbedModel>>): Promise<AxChatResponse | ReadableStream$1<AxChatResponse>>;
     private _chat1;
+    private cleanupFunctionSchema;
     private _chat2;
     embed(req: Readonly<AxEmbedRequest<TEmbedModel>>, options?: Readonly<AxAIServiceActionOptions<TModel, TEmbedModel>>): Promise<AxEmbedResponse>;
     private _embed1;
@@ -1637,14 +1638,12 @@ interface AxAgentFeatures {
  */
 declare class AxAgent<IN extends AxGenIn, OUT extends AxGenOut = AxGenOut> implements AxAgentic {
     private ai?;
-    private signature;
     private program;
     private functions?;
     private agents?;
     private disableSmartModelRouting?;
     private excludeFieldsFromPassthrough;
     private name;
-    private description;
     private subAgentList?;
     private func;
     constructor({ ai, name, description, signature, agents, functions, }: Readonly<{

package/index.js

@@ -803,7 +803,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;
@@ -856,13 +857,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,
@@ -6032,14 +6051,12 @@ function processChildAgentFunction(childFunction, parentValues, parentInputKeys,
 }
 var AxAgent = class {
   ai;
-  signature;
   program;
   functions;
   agents;
   disableSmartModelRouting;
   excludeFieldsFromPassthrough;
   name;
-  description;
   subAgentList;
   func;
   constructor({
@@ -6055,8 +6072,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}`
@@ -6067,17 +6084,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();
@@ -6133,7 +6149,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();
@@ -6181,8 +6197,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;
   }
 };