@mastra/server 0.0.0-fix-message-embedding-20250506021742 → 0.0.0-fix-generate-title-20250616171351

package/dist/{chunk-Y3SV5XK4.js → chunk-4O23XCE5.js}

@@ -1,7 +1,9 @@
-import { stringify, esm_default } from './chunk-OMN3UI6X.js';
-import { validateBody } from './chunk-L7XE5QTW.js';
-import { handleError } from './chunk-3AHQ5RGN.js';
-import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
+import { stringify, esm_default } from './chunk-TGJMNUYJ.js';
+import { validateBody } from './chunk-H5PTF3Y4.js';
+import { handleError } from './chunk-M5ABIP7D.js';
+import { HTTPException } from './chunk-NYN7KFXL.js';
+import { __export } from './chunk-MLKGABMK.js';
+import { RuntimeContext } from '@mastra/core/runtime-context';
 
 // src/server/handlers/agents.ts
 var agents_exports = {};
@@ -30,13 +32,33 @@ async function getAgentsHandler({ mastra, runtimeContext }) {
           };
           return acc;
         }, {});
+        let serializedAgentWorkflows = {};
+        if ("getWorkflows" in agent) {
+          const logger = mastra.getLogger();
+          try {
+            const workflows = await agent.getWorkflows({ runtimeContext });
+            serializedAgentWorkflows = Object.entries(workflows || {}).reduce((acc, [key, workflow]) => {
+              return {
+                ...acc,
+                [key]: {
+                  name: workflow.name
+                }
+              };
+            }, {});
+          } catch (error) {
+            logger.error("Error getting workflows for agent", { agentName: agent.name, error });
+          }
+        }
         return {
           id,
           name: agent.name,
           instructions,
           tools: serializedAgentTools,
+          workflows: serializedAgentWorkflows,
           provider: llm?.getProvider(),
-          modelId: llm?.getModelId()
+          modelId: llm?.getModelId(),
+          defaultGenerateOptions: agent.getDefaultGenerateOptions(),
+          defaultStreamOptions: agent.getDefaultStreamOptions()
         };
       })
     );
@@ -52,7 +74,8 @@ async function getAgentsHandler({ mastra, runtimeContext }) {
 async function getAgentByIdHandler({
   mastra,
   runtimeContext,
-  agentId
+  agentId,
+  isPlayground = false
 }) {
   try {
     const agent = mastra.getAgent(agentId);
@@ -69,14 +92,57 @@ async function getAgentByIdHandler({
       };
       return acc;
     }, {});
-    const instructions = await agent.getInstructions({ runtimeContext });
+    let serializedAgentWorkflows = {};
+    if ("getWorkflows" in agent) {
+      const logger = mastra.getLogger();
+      try {
+        const workflows = await agent.getWorkflows({ runtimeContext });
+        serializedAgentWorkflows = Object.entries(workflows || {}).reduce((acc, [key, workflow]) => {
+          return {
+            ...acc,
+            [key]: {
+              name: workflow.name,
+              steps: Object.entries(workflow.steps).reduce((acc2, [key2, step]) => {
+                return {
+                  ...acc2,
+                  [key2]: {
+                    id: step.id,
+                    description: step.description
+                  }
+                };
+              }, {})
+            }
+          };
+        }, {});
+      } catch (error) {
+        logger.error("Error getting workflows for agent", { agentName: agent.name, error });
+      }
+    }
+    let proxyRuntimeContext = runtimeContext;
+    if (isPlayground) {
+      proxyRuntimeContext = new Proxy(runtimeContext, {
+        get(target, prop) {
+          if (prop === "get") {
+            return function(key) {
+              const value = target.get(key);
+              return value ?? `<${key}>`;
+            };
+          }
+          return Reflect.get(target, prop);
+        }
+      });
+    }
+    const instructions = await agent.getInstructions({ runtimeContext: proxyRuntimeContext });
     const llm = await agent.getLLM({ runtimeContext });
     return {
       name: agent.name,
       instructions,
       tools: serializedAgentTools,
+      workflows: serializedAgentWorkflows,
       provider: llm?.getProvider(),
-      modelId: llm?.getModelId()
+      modelId: llm?.getModelId(),
+      defaultGenerateOptions: agent.getDefaultGenerateOptions(),
+      defaultStreamOptions: agent.getDefaultStreamOptions()
     };
   } catch (error) {
     return handleError(error, "Error getting agent");
@@ -131,14 +197,18 @@ async function generateHandler({
     if (!agent) {
       throw new HTTPException(404, { message: "Agent not found" });
     }
-    const { messages, resourceId, resourceid, ...rest } = body;
+    const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalResourceId = resourceId ?? resourceid;
+    const finalRuntimeContext = new RuntimeContext([
+      ...Array.from(runtimeContext.entries()),
+      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    ]);
     validateBody({ messages });
     const result = await agent.generate(messages, {
       ...rest,
       // @ts-expect-error TODO fix types
       resourceId: finalResourceId,
-      runtimeContext
+      runtimeContext: finalRuntimeContext
     });
     return result;
   } catch (error) {
@@ -156,20 +226,31 @@ async function streamGenerateHandler({
     if (!agent) {
       throw new HTTPException(404, { message: "Agent not found" });
     }
-    const { messages, resourceId, resourceid, ...rest } = body;
+    const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalResourceId = resourceId ?? resourceid;
+    const finalRuntimeContext = new RuntimeContext([
+      ...Array.from(runtimeContext.entries()),
+      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    ]);
     validateBody({ messages });
     const streamResult = await agent.stream(messages, {
       ...rest,
       // @ts-expect-error TODO fix types
       resourceId: finalResourceId,
-      runtimeContext
+      runtimeContext: finalRuntimeContext
     });
-    const streamResponse = rest.output ? streamResult.toTextStreamResponse() : streamResult.toDataStreamResponse({
+    const streamResponse = rest.output ? streamResult.toTextStreamResponse({
+      headers: {
+        "Transfer-Encoding": "chunked"
+      }
+    }) : streamResult.toDataStreamResponse({
       sendUsage: true,
       sendReasoning: true,
       getErrorMessage: (error) => {
         return `An error occurred while processing your request. ${error instanceof Error ? error.message : JSON.stringify(error)}`;
+      },
+      headers: {
+        "Transfer-Encoding": "chunked"
       }
     });
     return streamResponse;

package/dist/{chunk-4JINXASC.js → chunk-55DOQLP6.js}

@@ -1,5 +1,6 @@
-import { handleError } from './chunk-3AHQ5RGN.js';
-import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
+import { handleError } from './chunk-M5ABIP7D.js';
+import { HTTPException } from './chunk-NYN7KFXL.js';
+import { __export } from './chunk-MLKGABMK.js';
 
 // src/server/handlers/vector.ts
 var vector_exports = {};
@@ -90,7 +91,7 @@ async function describeIndex({
       throw new HTTPException(400, { message: "Index name is required" });
     }
     const vector = getVector(mastra, vectorName);
-    const stats = await vector.describeIndex(indexName);
+    const stats = await vector.describeIndex({ indexName });
     return {
       dimension: stats.dimension,
       count: stats.count,
@@ -110,7 +111,7 @@ async function deleteIndex({
       throw new HTTPException(400, { message: "Index name is required" });
     }
     const vector = getVector(mastra, vectorName);
-    await vector.deleteIndex(indexName);
+    await vector.deleteIndex({ indexName });
     return { success: true };
   } catch (error) {
     return handleError(error, "Error deleting index");

package/dist/{chunk-QN4KF3BH.cjs → chunk-57CJTIPW.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunkFV45V6WC_cjs = require('./chunk-FV45V6WC.cjs');
+var chunkOCWPVYNI_cjs = require('./chunk-OCWPVYNI.cjs');
 
 // src/server/handlers/utils.ts
 function validateBody(body) {
@@ -11,7 +11,7 @@ function validateBody(body) {
     return acc;
   }, {});
   if (Object.keys(errorResponse).length > 0) {
-    throw new chunkFV45V6WC_cjs.HTTPException(400, { message: Object.values(errorResponse)[0] });
+    throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: Object.values(errorResponse)[0] });
   }
 }
 

package/dist/{chunk-ZLBRQFDD.cjs → chunk-64U3UDTH.cjs}

@@ -1,11 +1,11 @@
 'use strict';
 
-var chunkFV45V6WC_cjs = require('./chunk-FV45V6WC.cjs');
+var chunkOCWPVYNI_cjs = require('./chunk-OCWPVYNI.cjs');
 
 // src/server/handlers/error.ts
 function handleError(error, defaultMessage) {
   const apiError = error;
-  throw new chunkFV45V6WC_cjs.HTTPException(apiError.status || 500, {
+  throw new chunkOCWPVYNI_cjs.HTTPException(apiError.status || 500, {
     message: apiError.message || defaultMessage
   });
 }

package/dist/{chunk-2FJURXCL.cjs → chunk-6TJSHFCJ.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,
@@ -22,7 +23,7 @@ function getMemoryFromContext({
 }) {
   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 memory = agent?.getMemory?.() || mastra.getMemory();
   return memory;
@@ -35,7 +36,7 @@ async function getMemoryStatusHandler({ mastra, agentId }) {
     }
     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({
@@ -46,13 +47,13 @@ async function getThreadsHandler({
   try {
     const memory = getMemoryFromContext({ mastra, agentId });
     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({
@@ -61,18 +62,18 @@ async function getThreadByIdHandler({
   threadId
 }) {
   try {
-    chunkQN4KF3BH_cjs.validateBody({ threadId });
+    chunk57CJTIPW_cjs.validateBody({ threadId });
     const memory = getMemoryFromContext({ mastra, agentId });
     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({
@@ -83,13 +84,13 @@ async function saveMessagesHandler({
   try {
     const memory = getMemoryFromContext({ mastra, agentId });
     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,7 +100,7 @@ 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({
@@ -110,9 +111,9 @@ async function createThreadHandler({
   try {
     const memory = getMemoryFromContext({ mastra, agentId });
     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,7 +122,7 @@ 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({
@@ -133,17 +134,17 @@ async function updateThreadHandler({
   try {
     const memory = getMemoryFromContext({ mastra, agentId });
     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,7 +157,7 @@ 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({
@@ -165,40 +166,47 @@ async function deleteThreadHandler({
   threadId
 }) {
   try {
-    chunkQN4KF3BH_cjs.validateBody({ threadId });
+    chunk57CJTIPW_cjs.validateBody({ threadId });
     const memory = getMemoryFromContext({ mastra, agentId });
     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
 }) {
+  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 });
+    chunk57CJTIPW_cjs.validateBody({ threadId });
     const memory = getMemoryFromContext({ mastra, agentId });
     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");
   }
 }
 

package/dist/chunk-75ZPJI57.cjs

@@ -0,0 +1,9 @@
+'use strict';
+
+var __defProp = Object.defineProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+
+exports.__export = __export;