@mastra/memory 1.15.1-alpha.0 → 1.15.1-alpha.1

package/dist/index.js

@@ -16431,6 +16431,25 @@ function deepMergeWorkingMemory(existing, update) {
   }
   return result;
 }
+function stripNullsFromOptional(value, schema) {
+  if (Array.isArray(value)) {
+    const itemSchema = schema.items ?? {};
+    return value.map((item) => stripNullsFromOptional(item, itemSchema));
+  }
+  if (typeof value === "object" && value !== null) {
+    const properties = schema.properties ?? {};
+    const required = schema.required ?? [];
+    const result = {};
+    for (const [key, propertyValue] of Object.entries(value)) {
+      if (propertyValue === null && !required.includes(key)) {
+        continue;
+      }
+      result[key] = stripNullsFromOptional(propertyValue, properties[key] ?? {});
+    }
+    return result;
+  }
+  return value;
+}
 var updateWorkingMemoryTool = (memoryConfig) => {
   const schema = memoryConfig?.workingMemory?.schema;
   let inputSchema = z.object({
@@ -16440,7 +16459,7 @@ var updateWorkingMemoryTool = (memoryConfig) => {
     const standardSchema2 = isStandardSchemaWithJSON$1(schema) ? schema : toStandardSchema$1(schema);
     const jsonSchema4 = standardSchemaToJSONSchema(standardSchema2, { io: "input" });
     delete jsonSchema4.$schema;
-    inputSchema = toStandardSchema$1({
+    const wrappedSchema = toStandardSchema$1({
       $schema: "http://json-schema.org/draft-07/schema#",
       type: "object",
       description: "The JSON formatted working memory content to store.",
@@ -16449,6 +16468,41 @@ var updateWorkingMemoryTool = (memoryConfig) => {
       },
       required: ["memory"]
     });
+    inputSchema = {
+      "~standard": {
+        version: 1,
+        vendor: "mastra",
+        validate: (value) => {
+          const wrappedResult = wrappedSchema["~standard"].validate(value);
+          if (wrappedResult instanceof Promise) {
+            return wrappedResult.then((result) => {
+              if (!("issues" in result) || !result.issues) {
+                return result;
+              }
+              if (!value || typeof value !== "object" || Array.isArray(value) || "memory" in value) {
+                return result;
+              }
+              return wrappedSchema["~standard"].validate({
+                memory: stripNullsFromOptional(value, jsonSchema4)
+              });
+            });
+          }
+          if (!("issues" in wrappedResult) || !wrappedResult.issues) {
+            return wrappedResult;
+          }
+          if (!value || typeof value !== "object" || Array.isArray(value) || "memory" in value) {
+            return wrappedResult;
+          }
+          return wrappedSchema["~standard"].validate({
+            memory: stripNullsFromOptional(value, jsonSchema4)
+          });
+        },
+        jsonSchema: {
+          input: (props) => wrappedSchema["~standard"].jsonSchema.input(props),
+          output: (props) => wrappedSchema["~standard"].jsonSchema.output(props)
+        }
+      }
+    };
   }
   const usesMergeSemantics = Boolean(schema);
   const description = schema ? `Update the working memory with new information. Data is merged with existing memory - only include fields you want to add or update. To preserve existing data, omit the field entirely. Arrays are replaced entirely when provided, so pass the complete array or omit it to keep the existing values.` : `Update the working memory with new information. Any data not included will be overwritten. Always pass data as string to the memory field. Never pass an object.`;