use simple form of messageRange everywhere

mastra-ai · TylerBarnes · Feb 4, 2025 · Feb 4, 2025 · Feb 3, 2025 · Feb 3, 2025
commit cbd516323b5865e31976130260083e2286056d00
diff --git a/docs/src/pages/docs/agents/01-agent-memory.mdx b/docs/src/pages/docs/agents/01-agent-memory.mdx
@@ -243,11 +243,8 @@ const embeddingOptions = {
 ```typescript
 const optionsConfig = {
   historySearch: {
-    topK: 3,           // Number of similar messages to find
-    messageRange: {     // Context around each result
-      before: 2,       // Context before each result
-      after: 2         // Context after each result
-    }
+    topK: 3,         // Number of similar messages to find
+    messageRange: 2, // Context around each result
   },
 };
 ```

diff --git a/docs/src/pages/examples/memory/memory-with-libsql.mdx b/docs/src/pages/examples/memory/memory-with-libsql.mdx
@@ -24,7 +24,7 @@ const memory = new Memory({
     lastMessages: 10,
     historySearch: {
       topK: 3,
-      messageRange: { before: 2, after: 2 },
+      messageRange: 2,
     },
   },
   embeddingOptions: {
@@ -90,7 +90,7 @@ const response3 = await memoryAgent.stream(
       lastMessages: false,
       historySearch: {
         topK: 3, // Get top 3 most relevant messages
-        messageRange: { before: 2, after: 2 }, // Include context around each match
+        messageRange: 2 // Include context around each match
       },
     },
   }

diff --git a/docs/src/pages/examples/memory/memory-with-pg.mdx b/docs/src/pages/examples/memory/memory-with-pg.mdx
@@ -34,7 +34,7 @@ const memory = new Memory({
     lastMessages: 10,
     historySearch: {
       topK: 3,
-      messageRange: { before: 2, after: 2 },
+      messageRange: 2,
     },
   },
   embeddingOptions: {

diff --git a/docs/src/pages/examples/memory/memory-with-upstash.mdx b/docs/src/pages/examples/memory/memory-with-upstash.mdx
@@ -26,7 +26,7 @@ const memory = new Memory({
     lastMessages: 10,
     historySearch: {
       topK: 3,
-      messageRange: { before: 2, after: 2 },
+      messageRange: 2,
     },
   },
   embeddingOptions: {
@@ -95,7 +95,7 @@ const response3 = await chefAgent.stream(
       lastMessages: 3, // Get last 3 messages for context
       historySearch: {
         topK: 2, // Also get 2 most relevant messages
-        messageRange: { before: 1, after: 1 }, // Include context around matches
+        messageRange: 2 // Include context around matches
       },
     }
   }

diff --git a/examples/memory-with-libsql/src/mastra/agents/index.ts b/examples/memory-with-libsql/src/mastra/agents/index.ts
@@ -16,7 +16,7 @@ const memory = new Memory({
     lastMessages: 100,
     historySearch: {
       topK: 2,
-      messageRange: { before: 2, after: 2 },
+      messageRange: 2,
     },
   },
   embeddingOptions: {

diff --git a/examples/memory-with-pg/src/mastra/agents/index.ts b/examples/memory-with-pg/src/mastra/agents/index.ts
@@ -23,7 +23,7 @@ export const memory = new Memory({
     lastMessages: 10,
     historySearch: {
       topK: 3,
-      messageRange: { before: 2, after: 2 },
+      messageRange: 2,
     },
   },
   embeddingOptions: {

diff --git a/examples/memory-with-upstash/src/mastra/index.ts b/examples/memory-with-upstash/src/mastra/index.ts
@@ -15,7 +15,7 @@ const memory = new Memory({
     lastMessages: 1,
     historySearch: {
       topK: 3,
-      messageRange: { before: 2, after: 2 },
+      messageRange: 2,
     },
   },
   embeddingOptions: {