@mastra/server 0.0.0-separate-trace-data-from-component-20250501042644 → 0.0.0-support-d1-client-20250701191943

package/dist/{chunk-RBQASTUP.js → chunk-LTJNMM2Y.js}

@@ -1,6 +1,7 @@
-import { validateBody } from './chunk-L7XE5QTW.js';
-import { handleError } from './chunk-3AHQ5RGN.js';
-import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
+import { validateBody } from './chunk-H5PTF3Y4.js';
+import { handleError } from './chunk-QDOJJCS4.js';
+import { HTTPException } from './chunk-NYN7KFXL.js';
+import { __export } from './chunk-MLKGABMK.js';
 
 // src/server/handlers/memory.ts
 var memory_exports = {};
@@ -14,20 +15,36 @@ __export(memory_exports, {
   saveMessagesHandler: () => saveMessagesHandler,
   updateThreadHandler: () => updateThreadHandler
 });
-function getMemoryFromContext({
+async function getMemoryFromContext({
   mastra,
-  agentId
+  agentId,
+  networkId,
+  runtimeContext
 }) {
   const agent = agentId ? mastra.getAgent(agentId) : null;
   if (agentId && !agent) {
     throw new HTTPException(404, { message: "Agent not found" });
   }
-  const memory = agent?.getMemory?.() || mastra.getMemory();
-  return memory;
+  const network = networkId ? mastra.vnext_getNetwork(networkId) : null;
+  if (networkId && !network) {
+    throw new HTTPException(404, { message: "Network not found" });
+  }
+  if (agent) {
+    return agent?.getMemory() || mastra.getMemory();
+  }
+  if (network) {
+    return await network?.getMemory({ runtimeContext }) || mastra.getMemory();
+  }
+  return mastra.getMemory();
 }
-async function getMemoryStatusHandler({ mastra, agentId }) {
+async function getMemoryStatusHandler({
+  mastra,
+  agentId,
+  networkId,
+  runtimeContext
+}) {
   try {
-    const memory = getMemoryFromContext({ mastra, agentId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
       return { result: false };
     }
@@ -39,10 +56,12 @@ async function getMemoryStatusHandler({ mastra, agentId }) {
 async function getThreadsHandler({
   mastra,
   agentId,
-  resourceId
+  resourceId,
+  networkId,
+  runtimeContext
 }) {
   try {
-    const memory = getMemoryFromContext({ mastra, agentId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
       throw new HTTPException(400, { message: "Memory is not initialized" });
     }
@@ -56,11 +75,13 @@ async function getThreadsHandler({
 async function getThreadByIdHandler({
   mastra,
   agentId,
-  threadId
+  threadId,
+  networkId,
+  runtimeContext
 }) {
   try {
     validateBody({ threadId });
-    const memory = getMemoryFromContext({ mastra, agentId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
       throw new HTTPException(400, { message: "Memory is not initialized" });
     }
@@ -76,10 +97,12 @@ async function getThreadByIdHandler({
 async function saveMessagesHandler({
   mastra,
   agentId,
-  body
+  body,
+  networkId,
+  runtimeContext
 }) {
   try {
-    const memory = getMemoryFromContext({ mastra, agentId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
       throw new HTTPException(400, { message: "Memory is not initialized" });
     }
@@ -103,10 +126,12 @@ async function saveMessagesHandler({
 async function createThreadHandler({
   mastra,
   agentId,
-  body
+  body,
+  networkId,
+  runtimeContext
 }) {
   try {
-    const memory = getMemoryFromContext({ mastra, agentId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
       throw new HTTPException(400, { message: "Memory is not initialized" });
     }
@@ -126,10 +151,12 @@ async function updateThreadHandler({
   mastra,
   agentId,
   threadId,
-  body
+  body,
+  networkId,
+  runtimeContext
 }) {
   try {
-    const memory = getMemoryFromContext({ mastra, agentId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!body) {
       throw new HTTPException(400, { message: "Body is required" });
     }
@@ -160,11 +187,13 @@ async function updateThreadHandler({
 async function deleteThreadHandler({
   mastra,
   agentId,
-  threadId
+  threadId,
+  networkId,
+  runtimeContext
 }) {
   try {
     validateBody({ threadId });
-    const memory = getMemoryFromContext({ mastra, agentId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
       throw new HTTPException(400, { message: "Memory is not initialized" });
     }
@@ -181,11 +210,17 @@ async function deleteThreadHandler({
 async function getMessagesHandler({
   mastra,
   agentId,
-  threadId
+  threadId,
+  limit,
+  networkId,
+  runtimeContext
 }) {
+  if (limit !== void 0 && (!Number.isInteger(limit) || limit <= 0)) {
+    throw new HTTPException(400, { message: "Invalid limit: must be a positive integer" });
+  }
   try {
     validateBody({ threadId });
-    const memory = getMemoryFromContext({ mastra, agentId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
       throw new HTTPException(400, { message: "Memory is not initialized" });
     }
@@ -193,8 +228,11 @@ async function getMessagesHandler({
     if (!thread) {
       throw new HTTPException(404, { message: "Thread not found" });
     }
-    const result = await memory.query({ threadId });
-    return result;
+    const result = await memory.query({
+      threadId,
+      ...limit && { selectBy: { last: limit } }
+    });
+    return { messages: result.messages, uiMessages: result.uiMessages };
   } catch (error) {
     return handleError(error, "Error getting messages");
   }