@ax-llm/ax 11.0.60 → 11.0.61

package/index.cjs

@@ -8704,9 +8704,21 @@ function processChildAgentFunction(childFunction, parentValues, parentInputKeys,
       );
       const originalFunc = processedFunction.func;
       processedFunction.func = async (childArgs, funcOptions) => {
+        let valuesToInject = {};
+        if (Array.isArray(parentValues)) {
+          const lastUserMessage = parentValues.filter((msg) => msg.role === "user").pop();
+          if (lastUserMessage) {
+            valuesToInject = pick(
+              lastUserMessage.values,
+              injectionKeys
+            );
+          }
+        } else {
+          valuesToInject = pick(parentValues, injectionKeys);
+        }
         const updatedChildArgs = {
           ...childArgs,
-          ...pick(parentValues, injectionKeys)
+          ...valuesToInject
         };
         if (options.debug && injectionKeys.length > 0) {
           const ai = funcOptions?.ai;
@@ -13695,7 +13707,17 @@ var AxRAG = class extends AxChainOfThought {
     this.queryFn = queryFn;
     this.register(this.genQuery);
   }
-  async forward(ai, { question }, options) {
+  async forward(ai, values, options) {
+    let question;
+    if (Array.isArray(values)) {
+      const lastUserMessage = values.filter((msg) => msg.role === "user").pop();
+      if (!lastUserMessage) {
+        throw new Error("No user message found in values array");
+      }
+      question = lastUserMessage.values.question;
+    } else {
+      question = values.question;
+    }
     let hop = 0;
     let context3 = [];
     while (hop < this.maxHops) {