@ax-llm/ax 11.0.7 → 11.0.8

package/index.js

@@ -2555,7 +2555,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()
@@ -4494,6 +4494,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;
@@ -5218,6 +5221,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) {
@@ -6001,14 +6007,20 @@ function processChildAgentFunction(childFunction, parentValues, parentInputKeys,
         injectionKeys
       );
       const originalFunc = processedFunction.func;
-      processedFunction.func = (childArgs, funcOptions) => originalFunc(
-        {
+      processedFunction.func = (childArgs, funcOptions) => {
+        const updatedChildArgs = {
           ...childArgs,
           ...pick(parentValues, injectionKeys)
-        },
-        funcOptions
-      );
+        };
+        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(
@@ -6121,11 +6133,12 @@ var AxAgent = class {
   init(parentAi, values, options) {
     const ai = this.ai ?? parentAi;
     const mm = ai?.getModelList();
-    const parentSchema = this.signature.toJSONSchema();
-    const parentKeys = parentSchema.properties ? Object.keys(parentSchema.properties) : [];
+    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
@@ -6196,7 +6209,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 ?? {},