diff --git a/src/ax/dsp/generate.ts b/src/ax/dsp/generate.ts
index eb885fb..524fb9d 100644
--- a/src/ax/dsp/generate.ts
+++ b/src/ax/dsp/generate.ts
@@ -348,7 +348,7 @@ export class AxGen<
         const funcs = parseFunctions(ai, result.functionCalls, values);
 
         if (funcs) {
-          await this.processFunctions(funcs, mem, sessionId, traceId);
+          await this.processFunctions(ai, funcs, mem, sessionId, traceId);
         }
       }