@ax-llm/ax 11.0.6 → 11.0.8

package/index.cjs

@@ -2657,7 +2657,7 @@ var safetySettings = [
   }
 ];
 var axAIGoogleGeminiDefaultConfig = () => structuredClone({
-  model: "gemini-1.5-pro" /* Gemini15Pro */,
+  model: "gemini-2.0-flash" /* Gemini20Flash */,
   embedModel: "text-embedding-004" /* TextEmbedding004 */,
   safetySettings,
   ...axBaseAIDefaultConfig()
@@ -4596,6 +4596,9 @@ var LRUCache = class {
 };
 var globalPrefixCache = new LRUCache(500);
 function matchesContent(content, prefix, startIndex = 0, prefixCache = globalPrefixCache) {
+  if (/^\s*$/.test(content)) {
+    return -3;
+  }
   const exactMatchIndex = content.indexOf(prefix, startIndex);
   if (exactMatchIndex !== -1) {
     return exactMatchIndex;
@@ -5320,6 +5323,9 @@ var streamingExtractValues = (sig, values, xstate, content, streamingValidation
       // Field is not found, continue to the next field
       case -2:
         return true;
+      // Partial match at end, skip and gather more content
+      case -3:
+        return true;
     }
     let prefixLen = prefix.length;
     if (xstate.currField) {
@@ -6089,6 +6095,43 @@ var AxGen = class extends AxProgramWithSignature {
 };
 
 // prompts/agent.ts
+function processChildAgentFunction(childFunction, parentValues, parentInputKeys, modelList, options) {
+  let processedFunction = { ...childFunction };
+  if (processedFunction.parameters) {
+    const childKeys = processedFunction.parameters.properties ? Object.keys(processedFunction.parameters.properties) : [];
+    const commonKeys = parentInputKeys.filter((key) => childKeys.includes(key)).filter((key) => key !== "model");
+    const injectionKeys = commonKeys.filter(
+      (key) => !options.excludeFieldsFromPassthrough.includes(key)
+    );
+    if (injectionKeys.length > 0) {
+      processedFunction.parameters = removePropertiesFromSchema(
+        processedFunction.parameters,
+        injectionKeys
+      );
+      const originalFunc = processedFunction.func;
+      processedFunction.func = (childArgs, funcOptions) => {
+        const updatedChildArgs = {
+          ...childArgs,
+          ...pick(parentValues, injectionKeys)
+        };
+        if (options.debug && injectionKeys.length > 0) {
+          process.stdout.write(
+            `Function Params: ${JSON.stringify(updatedChildArgs, null, 2)}`
+          );
+        }
+        return originalFunc(updatedChildArgs, funcOptions);
+      };
+    }
+    return processedFunction;
+  }
+  if (modelList && !options.disableSmartModelRouting && options.canConfigureSmartModelRouting) {
+    processedFunction.parameters = addModelParameter(
+      processedFunction.parameters,
+      modelList
+    );
+  }
+  return processedFunction;
+}
 var AxAgent = class {
   ai;
   signature;
@@ -6096,6 +6139,7 @@ var AxAgent = class {
   functions;
   agents;
   disableSmartModelRouting;
+  excludeFieldsFromPassthrough;
   name;
   description;
   subAgentList;
@@ -6112,6 +6156,7 @@ var AxAgent = class {
     this.agents = agents;
     this.functions = functions;
     this.disableSmartModelRouting = options?.disableSmartModelRouting;
+    this.excludeFieldsFromPassthrough = options?.excludeFieldsFromPassthrough ?? [];
     this.signature = new AxSignature(signature);
     this.signature.setDescription(description);
     if (!name || name.length < 5) {
@@ -6180,15 +6225,34 @@ var AxAgent = class {
   }
   getFeatures() {
     return {
-      canConfigureSmartModelRouting: this.ai !== void 0
+      canConfigureSmartModelRouting: this.ai === void 0,
+      excludeFieldsFromPassthrough: this.excludeFieldsFromPassthrough
     };
   }
-  init(parentAi, options) {
+  /**
+   * Initializes the agent's execution context, processing child agents and their functions.
+   */
+  init(parentAi, values, options) {
     const ai = this.ai ?? parentAi;
     const mm = ai?.getModelList();
-    const agentFuncs = this.agents?.map((a) => a.getFunction())?.map(
-      (f) => mm && !this.disableSmartModelRouting && this.agents?.find((a) => a.getFunction().name === f.name)?.getFeatures().canConfigureSmartModelRouting ? { ...f, parameters: addModelParameter(f.parameters, mm) } : f
-    );
+    const parentSchema = this.signature.getInputFields();
+    const parentKeys = parentSchema.map((p) => p.name);
+    const agentFuncs = this.agents?.map((agent) => {
+      const f = agent.getFeatures();
+      const processOptions = {
+        debug: ai?.getOptions()?.debug ?? false,
+        disableSmartModelRouting: !!this.disableSmartModelRouting,
+        excludeFieldsFromPassthrough: f.excludeFieldsFromPassthrough,
+        canConfigureSmartModelRouting: f.canConfigureSmartModelRouting
+      };
+      return processChildAgentFunction(
+        agent.getFunction(),
+        values,
+        parentKeys,
+        mm,
+        processOptions
+      );
+    });
     const functions = [
       ...options?.functions ?? this.functions ?? [],
       ...agentFuncs ?? []
@@ -6196,16 +6260,33 @@ var AxAgent = class {
     return { ai, functions };
   }
   async forward(parentAi, values, options) {
-    const { ai, functions } = this.init(parentAi, options);
+    const { ai, functions } = this.init(parentAi, values, options);
     return await this.program.forward(ai, values, { ...options, functions });
   }
   async *streamingForward(parentAi, values, options) {
-    const { ai, functions } = this.init(parentAi, options);
+    const { ai, functions } = this.init(parentAi, values, options);
     return yield* this.program.streamingForward(ai, values, {
       ...options,
       functions
     });
   }
+  /**
+   * Updates the agent's description.
+   * This updates both the stored description and the function's description.
+   *
+   * @param description - New description for the agent (must be at least 20 characters)
+   * @throws Error if description is too short
+   */
+  setDescription(description) {
+    if (!description || description.length < 20) {
+      throw new Error(
+        "Agent description must be at least 20 characters (explain in detail what the agent does)"
+      );
+    }
+    this.description = description;
+    this.signature.setDescription(description);
+    this.func.description = description;
+  }
 };
 function toCamelCase(inputString) {
   const words = inputString.split(/[^a-zA-Z0-9]/);
@@ -6230,7 +6311,7 @@ function addModelParameter(parameters, models) {
   const modelProperty = {
     type: "string",
     enum: models.map((m) => m.key),
-    description: `The AI model to use for this function call. Available options: ${models.map((m) => `${m.key}: ${m.description}`).join(" | ")}`
+    description: `The AI model to use for this function call. Available options: ${models.map((m) => `\`${m.key}\` ${m.description}`).join(", ")}`
   };
   const newProperties = {
     ...baseSchema.properties ?? {},
@@ -6243,6 +6324,34 @@ function addModelParameter(parameters, models) {
     required: newRequired
   };
 }
+function removePropertiesFromSchema(schema, keys) {
+  const newSchema = structuredClone(schema);
+  if (newSchema.properties) {
+    for (const key of keys) {
+      delete newSchema.properties[key];
+    }
+  }
+  if (Array.isArray(newSchema.required)) {
+    const filteredRequired = newSchema.required.filter(
+      (r) => !keys.includes(r)
+    );
+    Object.defineProperty(newSchema, "required", {
+      value: filteredRequired,
+      writable: true,
+      configurable: true
+    });
+  }
+  return newSchema;
+}
+function pick(obj, keys) {
+  const result = {};
+  for (const key of keys) {
+    if (key in obj) {
+      result[key] = obj[key];
+    }
+  }
+  return result;
+}
 
 // docs/tika.ts
 var import_node_fs = require("fs");