@ax-llm/ax 12.0.14 → 12.0.16

package/index.d.cts

@@ -2134,6 +2134,7 @@ declare class AxAIGrok extends AxAIOpenAIBase<AxAIGrokModel, AxAIGrokEmbedModels
 type AxMemoryData = {
     tags?: string[];
     role: AxChatRequest['chatPrompt'][number]['role'];
+    updatable?: boolean;
     chat: {
         index: number;
         value: Omit<AxChatRequest['chatPrompt'][number], 'role'>;
@@ -2319,6 +2320,7 @@ declare class AxMemory implements AxAIMemory {
     getLast(sessionId?: string): {
         tags?: string[];
         role: AxChatRequest["chatPrompt"][number]["role"];
+        updatable?: boolean;
         chat: {
             index: number;
             value: Omit<AxChatRequest["chatPrompt"][number], "role">;

package/index.d.ts

@@ -2134,6 +2134,7 @@ declare class AxAIGrok extends AxAIOpenAIBase<AxAIGrokModel, AxAIGrokEmbedModels
 type AxMemoryData = {
     tags?: string[];
     role: AxChatRequest['chatPrompt'][number]['role'];
+    updatable?: boolean;
     chat: {
         index: number;
         value: Omit<AxChatRequest['chatPrompt'][number], 'role'>;
@@ -2319,6 +2320,7 @@ declare class AxMemory implements AxAIMemory {
     getLast(sessionId?: string): {
         tags?: string[];
         role: AxChatRequest["chatPrompt"][number]["role"];
+        updatable?: boolean;
         chat: {
             index: number;
             value: Omit<AxChatRequest["chatPrompt"][number], "role">;

package/index.js

@@ -6124,9 +6124,10 @@ var MemoryImpl = class {
         }
       }
     };
-    if (!lastItem || lastItem.role !== "assistant") {
+    if (!lastItem || lastItem.role !== "assistant" || lastItem.role === "assistant" && !lastItem.updatable) {
       this.data.push({
         role: "assistant",
+        updatable: true,
         chat: [
           { index, value: structuredClone({ content, name, functionCalls }) }
         ]
@@ -6143,13 +6144,16 @@ var MemoryImpl = class {
       log();
       return;
     }
-    if ("content" in chat.value && typeof content === "string" && content.trim() !== "") {
+    if (typeof content === "string" && content.trim() !== "") {
+      ;
       chat.value.content = content;
     }
-    if ("name" in chat.value && name && name.trim() !== "") {
+    if (typeof name === "string" && name.trim() !== "") {
+      ;
       chat.value.name = name;
     }
-    if ("functionCalls" in chat.value && functionCalls && functionCalls.length > 0) {
+    if (Array.isArray(functionCalls) && functionCalls.length > 0) {
+      ;
       chat.value.functionCalls = functionCalls;
     }
     log();
@@ -7536,6 +7540,22 @@ function parseFunctionCalls(ai, functionCalls, values, model) {
   }));
   return funcs;
 }
+function createFunctionConfig(functionList, definedFunctionCall, firstStep) {
+  let functionCall = definedFunctionCall;
+  if (!firstStep && (functionCall === "required" || typeof functionCall === "function")) {
+    return { functions: [], functionCall: void 0 };
+  }
+  if (!functionList) {
+    return { functions: [], functionCall };
+  }
+  const functions = functionList.map((f2) => {
+    if ("toFunction" in f2) {
+      return f2.toFunction();
+    }
+    return f2;
+  }).flat();
+  return { functions, functionCall };
+}
 
 // dsp/processResponse.ts
 import "stream/web";
@@ -9984,7 +10004,8 @@ var AxGen = class extends AxProgramWithSignature {
     mem,
     options,
     traceContext,
-    firstStep
+    functions,
+    functionCall
   }) {
     const {
       sessionId,
@@ -9992,8 +10013,6 @@ var AxGen = class extends AxProgramWithSignature {
       model,
       rateLimiter,
       stream,
-      functions: _functions,
-      functionCall: _functionCall,
       thinkingTokenBudget,
       showThoughts
     } = options ?? {};
@@ -10004,11 +10023,6 @@ var AxGen = class extends AxProgramWithSignature {
     if (chatPrompt.length === 0) {
       throw new Error("No chat prompt found");
     }
-    const functions = _functions?.map((f2) => "toFunction" in f2 ? f2.toFunction() : f2)?.flat();
-    let functionCall = _functionCall ?? this.options?.functionCall;
-    if (!firstStep && (functionCall === "required" || typeof functionCall === "function")) {
-      functionCall = void 0;
-    }
     const modelConfig = {
       ...options?.modelConfig,
       ...options?.sampleCount ? { n: options.sampleCount } : {},
@@ -10045,18 +10059,24 @@ var AxGen = class extends AxProgramWithSignature {
     span,
     traceContext
   }) {
-    const { sessionId, traceId, functions: _functions } = options ?? {};
+    const { sessionId, traceId, functions: functionList } = options ?? {};
+    const definedFunctionCall = options?.functionCall ?? this.options?.functionCall;
     const strictMode = options?.strictMode ?? false;
     const model = options.model;
     const states = this.createStates(options.sampleCount ?? 1);
     const usage = this.usage;
-    const functions = _functions?.map((f2) => "toFunction" in f2 ? f2.toFunction() : f2)?.flat() ?? [];
+    const { functions, functionCall } = createFunctionConfig(
+      functionList,
+      definedFunctionCall,
+      firstStep
+    );
     const res = await this.forwardSendRequest({
       ai,
       mem,
       options,
       traceContext,
-      firstStep
+      functions,
+      functionCall
     });
     if (res instanceof ReadableStream3) {
       yield* processStreamingResponse({