@mastra/server 0.0.0-commonjs-20250414101718 → 0.0.0-custom-instrumentation-20250626084921

package/dist/{chunk-2FJURXCL.cjs → chunk-RQ2Z56EC.cjs}

@@ -1,12 +1,13 @@
 'use strict';
 
-var chunkQN4KF3BH_cjs = require('./chunk-QN4KF3BH.cjs');
-var chunkZLBRQFDD_cjs = require('./chunk-ZLBRQFDD.cjs');
-var chunkFV45V6WC_cjs = require('./chunk-FV45V6WC.cjs');
+var chunk57CJTIPW_cjs = require('./chunk-57CJTIPW.cjs');
+var chunk64U3UDTH_cjs = require('./chunk-64U3UDTH.cjs');
+var chunkOCWPVYNI_cjs = require('./chunk-OCWPVYNI.cjs');
+var chunk75ZPJI57_cjs = require('./chunk-75ZPJI57.cjs');
 
 // src/server/handlers/memory.ts
 var memory_exports = {};
-chunkFV45V6WC_cjs.__export(memory_exports, {
+chunk75ZPJI57_cjs.__export(memory_exports, {
   createThreadHandler: () => createThreadHandler,
   deleteThreadHandler: () => deleteThreadHandler,
   getMemoryStatusHandler: () => getMemoryStatusHandler,
@@ -16,80 +17,102 @@ chunkFV45V6WC_cjs.__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 chunkFV45V6WC_cjs.HTTPException(404, { message: "Agent not found" });
+    throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Agent not found" });
+  }
+  const network = networkId ? mastra.vnext_getNetwork(networkId) : null;
+  if (networkId && !network) {
+    throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Network not found" });
+  }
+  if (agent) {
+    return agent?.getMemory() || mastra.getMemory();
+  }
+  if (network) {
+    return await network?.getMemory({ runtimeContext }) || mastra.getMemory();
   }
-  const memory = agent?.getMemory?.() || mastra.getMemory();
-  return memory;
+  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 };
     }
     return { result: true };
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error getting memory status");
+    return chunk64U3UDTH_cjs.handleError(error, "Error getting memory status");
   }
 }
 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 chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
-    chunkQN4KF3BH_cjs.validateBody({ resourceId });
+    chunk57CJTIPW_cjs.validateBody({ resourceId });
     const threads = await memory.getThreadsByResourceId({ resourceId });
     return threads;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error getting threads");
+    return chunk64U3UDTH_cjs.handleError(error, "Error getting threads");
   }
 }
 async function getThreadByIdHandler({
   mastra,
   agentId,
-  threadId
+  threadId,
+  networkId,
+  runtimeContext
 }) {
   try {
-    chunkQN4KF3BH_cjs.validateBody({ threadId });
-    const memory = getMemoryFromContext({ mastra, agentId });
+    chunk57CJTIPW_cjs.validateBody({ threadId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
     const thread = await memory.getThreadById({ threadId });
     if (!thread) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Thread not found" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Thread not found" });
     }
     return thread;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error getting thread");
+    return chunk64U3UDTH_cjs.handleError(error, "Error getting thread");
   }
 }
 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 chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
     if (!body?.messages) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Messages are required" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Messages are required" });
     }
     if (!Array.isArray(body.messages)) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Messages should be an array" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Messages should be an array" });
     }
     const processedMessages = body.messages.map((message) => ({
       ...message,
@@ -99,20 +122,22 @@ async function saveMessagesHandler({
     const result = await memory.saveMessages({ messages: processedMessages, memoryConfig: {} });
     return result;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error saving messages");
+    return chunk64U3UDTH_cjs.handleError(error, "Error saving messages");
   }
 }
 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 chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
-    chunkQN4KF3BH_cjs.validateBody({ resourceId: body?.resourceId });
+    chunk57CJTIPW_cjs.validateBody({ resourceId: body?.resourceId });
     const result = await memory.createThread({
       resourceId: body?.resourceId,
       title: body?.title,
@@ -121,29 +146,31 @@ async function createThreadHandler({
     });
     return result;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error saving thread to memory");
+    return chunk64U3UDTH_cjs.handleError(error, "Error saving thread to memory");
   }
 }
 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 chunkFV45V6WC_cjs.HTTPException(400, { message: "Body is required" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Body is required" });
     }
     const { title, metadata, resourceId } = body;
     const updatedAt = /* @__PURE__ */ new Date();
-    chunkQN4KF3BH_cjs.validateBody({ threadId });
+    chunk57CJTIPW_cjs.validateBody({ threadId });
     if (!memory) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
     const thread = await memory.getThreadById({ threadId });
     if (!thread) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Thread not found" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Thread not found" });
     }
     const updatedThread = {
       ...thread,
@@ -156,49 +183,60 @@ async function updateThreadHandler({
     const result = await memory.saveThread({ thread: updatedThread });
     return result;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error updating thread");
+    return chunk64U3UDTH_cjs.handleError(error, "Error updating thread");
   }
 }
 async function deleteThreadHandler({
   mastra,
   agentId,
-  threadId
+  threadId,
+  networkId,
+  runtimeContext
 }) {
   try {
-    chunkQN4KF3BH_cjs.validateBody({ threadId });
-    const memory = getMemoryFromContext({ mastra, agentId });
+    chunk57CJTIPW_cjs.validateBody({ threadId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
     const thread = await memory.getThreadById({ threadId });
     if (!thread) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Thread not found" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Thread not found" });
     }
     await memory.deleteThread(threadId);
     return { result: "Thread deleted" };
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error deleting thread");
+    return chunk64U3UDTH_cjs.handleError(error, "Error deleting thread");
   }
 }
 async function getMessagesHandler({
   mastra,
   agentId,
-  threadId
+  threadId,
+  limit,
+  networkId,
+  runtimeContext
 }) {
+  if (limit !== void 0 && (!Number.isInteger(limit) || limit <= 0)) {
+    throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Invalid limit: must be a positive integer" });
+  }
   try {
-    chunkQN4KF3BH_cjs.validateBody({ threadId });
-    const memory = getMemoryFromContext({ mastra, agentId });
+    chunk57CJTIPW_cjs.validateBody({ threadId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
     const thread = await memory.getThreadById({ threadId });
     if (!thread) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Thread not found" });
+      throw new chunkOCWPVYNI_cjs.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 chunkZLBRQFDD_cjs.handleError(error, "Error getting messages");
+    return chunk64U3UDTH_cjs.handleError(error, "Error getting messages");
   }
 }