@mastra/memory 0.1.0-alpha.71 → 0.1.0-alpha.73

package/CHANGELOG.md

@@ -1,5 +1,29 @@
 # @mastra/memory
 
+## 0.1.0-alpha.73
+
+### Minor Changes
+
+- d7d465a: Breaking change for Memory: embeddings: {} has been replaced with embedder: new OpenAIEmbedder() (or whichever embedder you want - check the docs)
+
+### Patch Changes
+
+- Updated dependencies [d7d465a]
+- Updated dependencies [d7d465a]
+- Updated dependencies [2017553]
+- Updated dependencies [a10b7a3]
+- Updated dependencies [16e5b04]
+  - @mastra/core@0.2.0-alpha.91
+
+## 0.1.0-alpha.72
+
+### Patch Changes
+
+- Updated dependencies [8151f44]
+- Updated dependencies [e897f1c]
+- Updated dependencies [3700be1]
+  - @mastra/core@0.2.0-alpha.90
+
 ## 0.1.0-alpha.71
 
 ### Minor Changes

package/dist/index.d.ts

@@ -8,7 +8,9 @@ import { Message } from 'ai';
  * and message injection.
  */
 declare class Memory extends MastraMemory {
-    constructor(config: SharedMemoryConfig);
+    constructor(config: SharedMemoryConfig & {
+        embeddings?: any;
+    });
     query({ threadId, selectBy, threadConfig, }: StorageGetMessagesArg): Promise<{
         messages: CoreMessage[];
         uiMessages: Message[];

package/dist/index.js

@@ -4,6 +4,11 @@ import { MastraMemory } from '@mastra/core/memory';
 // src/index.ts
 var Memory = class extends MastraMemory {
   constructor(config) {
+    if (config.embeddings) {
+      throw new Error(
+        'The `embeddings` option is deprecated. Please use `embedder` instead. Example: new Memory({ embedder: new OpenAIEmbedder({ model: "text-embedding-3-small" }) })'
+      );
+    }
     super({ name: "Memory", ...config });
     this.defaultWorkingMemoryTemplate = `
 <user>
@@ -46,9 +51,10 @@ var Memory = class extends MastraMemory {
       messageRange: config?.semanticRecall?.messageRange || { before: 2, after: 2 }
     };
     if (selectBy?.vectorSearchString && this.vector) {
-      const { embeddings } = await this.vector.embed(selectBy.vectorSearchString, this.parseEmbeddingOptions());
+      const embedder = this.getEmbedder();
+      const { embedding } = await embedder.embed(selectBy.vectorSearchString);
       await this.vector.createIndex("memory_messages", 1536);
-      vectorResults = await this.vector.query("memory_messages", embeddings[0], vectorConfig.topK, {
+      vectorResults = await this.vector.query("memory_messages", embedding, vectorConfig.topK, {
         thread_id: threadId
       });
     }
@@ -136,14 +142,19 @@ var Memory = class extends MastraMemory {
       await this.vector.createIndex("memory_messages", 1536);
       for (const message of messages) {
         if (typeof message.content !== `string`) continue;
-        const { embeddings } = await this.vector.embed(message.content, this.parseEmbeddingOptions());
-        await this.vector.upsert("memory_messages", embeddings, [
-          {
-            text: message.content,
-            message_id: message.id,
-            thread_id: message.threadId
-          }
-        ]);
+        const embedder = this.getEmbedder();
+        const { embedding } = await embedder.embed(message.content);
+        await this.vector.upsert(
+          "memory_messages",
+          [embedding],
+          [
+            {
+              text: message.content,
+              message_id: message.id,
+              thread_id: message.threadId
+            }
+          ]
+        );
       }
     }
     return this.storage.__saveMessages({ messages });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/memory",
-  "version": "0.1.0-alpha.71",
+  "version": "0.1.0-alpha.73",
   "description": "",
   "type": "module",
   "main": "./dist/index.js",
@@ -36,7 +36,7 @@
     "pg-pool": "^3.7.0",
     "postgres": "^3.4.5",
     "redis": "^4.7.0",
-    "@mastra/core": "^0.2.0-alpha.89"
+    "@mastra/core": "^0.2.0-alpha.91"
   },
   "devDependencies": {
     "@babel/preset-env": "^7.26.0",
@@ -48,7 +48,8 @@
     "vitest": "^3.0.4"
   },
   "scripts": {
-    "build": "tsup src/index.ts --format esm --dts --clean --treeshake",
+    "check": "tsc --noEmit",
+    "build": "pnpm run check && tsup src/index.ts --format esm --dts --clean --treeshake",
     "dev": "tsup src/index.ts --format esm --dts --clean --watch",
     "test:integration": "cd integration-tests && pnpm run test",
     "test": "pnpm test:integration"

package/src/index.ts

@@ -8,7 +8,19 @@ import { Message as AiMessage } from 'ai';
  * and message injection.
  */
 export class Memory extends MastraMemory {
-  constructor(config: SharedMemoryConfig) {
+  constructor(
+    config: SharedMemoryConfig & {
+      /* @deprecated use embedder instead */
+      embeddings?: any;
+    },
+  ) {
+    // Check for deprecated embeddings object
+    if (config.embeddings) {
+      throw new Error(
+        'The `embeddings` option is deprecated. Please use `embedder` instead. Example: new Memory({ embedder: new OpenAIEmbedder({ model: "text-embedding-3-small" }) })',
+      );
+    }
+
     super({ name: 'Memory', ...config });
 
     const mergedConfig = this.getMergedThreadConfig({
@@ -54,10 +66,11 @@ export class Memory extends MastraMemory {
           };
 
     if (selectBy?.vectorSearchString && this.vector) {
-      const { embeddings } = await this.vector.embed(selectBy.vectorSearchString, this.parseEmbeddingOptions());
+      const embedder = this.getEmbedder();
+      const { embedding } = await embedder.embed(selectBy.vectorSearchString);
 
       await this.vector.createIndex('memory_messages', 1536);
-      vectorResults = await this.vector.query('memory_messages', embeddings[0]!, vectorConfig.topK, {
+      vectorResults = await this.vector.query('memory_messages', embedding, vectorConfig.topK, {
         thread_id: threadId,
       });
     }
@@ -183,7 +196,7 @@ export class Memory extends MastraMemory {
   async saveMessages({ messages }: { messages: MessageType[] }): Promise<MessageType[]> {
     // First save working memory from any messages
     await this.saveWorkingMemory(messages);
-    
+
     // Then strip working memory tags from all messages
     this.mutateMessagesToHideWorkingMemory(messages);
 
@@ -191,14 +204,19 @@ export class Memory extends MastraMemory {
       await this.vector.createIndex('memory_messages', 1536);
       for (const message of messages) {
         if (typeof message.content !== `string`) continue;
-        const { embeddings } = await this.vector.embed(message.content, this.parseEmbeddingOptions());
-        await this.vector.upsert('memory_messages', embeddings, [
-          {
-            text: message.content,
-            message_id: message.id,
-            thread_id: message.threadId,
-          },
-        ]);
+        const embedder = this.getEmbedder();
+        const { embedding } = await embedder.embed(message.content);
+        await this.vector.upsert(
+          'memory_messages',
+          [embedding],
+          [
+            {
+              text: message.content,
+              message_id: message.id,
+              thread_id: message.threadId,
+            },
+          ],
+        );
       }
     }