@mastra/server 0.0.0-fix-persist-session-cache-option-mcp-server-20251031182703 → 0.0.0-fix-9244-clickhouse-metadata-20251104220626

package/dist/chunk-BANGX6SC.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/server/handlers/scores.ts"],"names":["__export","handleError"],"mappings":";;;;;;AAAA,IAAA,cAAA,GAAA;AAAAA,0BAAA,CAAA,cAAA,EAAA;AAAA,EAAA,gBAAA,EAAA,MAAA,gBAAA;AAAA,EAAA,kBAAA,EAAA,MAAA,kBAAA;AAAA,EAAA,2BAAA,EAAA,MAAA,2BAAA;AAAA,EAAA,wBAAA,EAAA,MAAA,wBAAA;AAAA,EAAA,2BAAA,EAAA,MAAA,2BAAA;AAAA,EAAA,gBAAA,EAAA,MAAA;AAAA,CAAA,CAAA;AAMA,eAAe,qBAAA,CAAsB;AAAA,EACnC,MAAA;AAAA,EACA;AACF,CAAA,EAEG;AACD,EAAA,MAAM,MAAA,GAAS,OAAO,UAAA,EAAW;AACjC,EAAA,MAAM,SAAA,GAAY,OAAO,aAAA,EAAc;AAEvC,EAAA,MAAM,UAAA,uBAAiB,GAAA,EAGrB;AAEF,EAAA,KAAA,MAAW,CAAC,OAAA,EAAS,KAAK,KAAK,MAAA,CAAO,OAAA,CAAQ,MAAM,CAAA,EAAG;AACrD,IAAA,MAAM,OAAA,GACH,MAAM,KAAA,CAAM,WAAA,CAAY;AAAA,MACvB;AAAA,KACD,KAAM,EAAC;AAEV,IAAA,IAAI,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA,CAAE,SAAS,CAAA,EAAG;AACnC,MAAA,KAAA,MAAW,CAAC,SAAA,EAAW,MAAM,KAAK,MAAA,CAAO,OAAA,CAAQ,OAAO,CAAA,EAAG;AACzD,QAAA,MAAM,QAAA,GAAW,OAAO,MAAA,CAAO,EAAA;AAC/B,QAAA,IAAI,UAAA,CAAW,GAAA,CAAI,QAAQ,CAAA,EAAG;AAC5B,UAAA,UAAA,CAAW,GAAA,CAAI,QAAQ,CAAA,EAAG,QAAA,CAAS,KAAK,OAAO,CAAA;AAC/C,UAAA,UAAA,CAAW,IAAI,QAAQ,CAAA,EAAG,UAAA,CAAW,IAAA,CAAK,MAAM,IAAI,CAAA;AAAA,QACtD,CAAA,MAAO;AACL,UAAA,UAAA,CAAW,IAAI,QAAA,EAAU;AAAA,YACvB,aAAa,EAAC;AAAA,YACd,GAAG,MAAA;AAAA,YACH,UAAA,EAAY,CAAC,KAAA,CAAM,IAAI,CAAA;AAAA,YACvB,QAAA,EAAU,CAAC,OAAO,CAAA;AAAA,YAClB,YAAA,EAAc;AAAA,WACf,CAAA;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,EAAA,KAAA,MAAW,CAAC,UAAA,EAAY,QAAQ,KAAK,MAAA,CAAO,OAAA,CAAQ,SAAS,CAAA,EAAG;AAC9D,IAAA,MAAM,OAAA,GACH,MAAM,QAAA,CAAS,WAAA,CAAY;AAAA,MAC1B;AAAA,KACD,KAAM,EAAC;AAEV,IAAA,IAAI,MAAA,CAAO,IAAA,CAAK,OAAO,CAAA,CAAE,SAAS,CAAA,EAAG;AACnC,MAAA,KAAA,MAAW,CAAC,SAAA,EAAW,MAAM,KAAK,MAAA,CAAO,OAAA,CAAQ,OAAO,CAAA,EAAG;AACzD,QAAA,MAAM,UAAA,GAAa,OAAO,MAAA,CAAO,IAAA;AACjC,QAAA,IAAI,UAAA,CAAW,GAAA,CAAI,UAAU,CAAA,EAAG;AAC9B,UAAA,UAAA,CAAW,GAAA,CAAI,UAAU,CAAA,EAAG,WAAA,CAAY,KAAK,UAAU,CAAA;AAAA,QACzD,CAAA,MAAO;AACL,UAAA,UAAA,CAAW,IAAI,UAAA,EAAY;AAAA,YACzB,UAAU,EAAC;AAAA,YACX,YAAY,EAAC;AAAA,YACb,GAAG,MAAA;AAAA,YACH,WAAA,EAAa,CAAC,UAAU,CAAA;AAAA,YACxB,YAAA,EAAc;AAAA,WACf,CAAA;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,EAAA,MAAM,iBAAA,GAAoB,MAAM,MAAA,CAAO,WAAA,EAAY;AACnD,EAAA,KAAA,MAAW,CAAC,WAAW,MAAM,CAAA,IAAK,OAAO,OAAA,CAAQ,iBAAA,IAAqB,EAAE,CAAA,EAAG;AACzE,IAAA,MAAM,WAAW,MAAA,CAAO,EAAA;AACxB,IAAA,IAAI,UAAA,CAAW,GAAA,CAAI,QAAQ,CAAA,EAAG;AAC5B,MAAA,UAAA,CAAW,GAAA,CAAI,QAAQ,CAAA,CAAG,YAAA,GAAe,IAAA;AAAA,IAC3C,CAAA,MAAO;AACL,MAAA,UAAA,CAAW,IAAI,QAAA,EAAU;AAAA,QACvB,MAAA;AAAA,QACA,UAAU,EAAC;AAAA,QACX,YAAY,EAAC;AAAA,QACb,aAAa,EAAC;AAAA,QACd,YAAA,EAAc;AAAA,OACf,CAAA;AAAA,IACH;AAAA,EACF;AAEA,EAAA,OAAO,MAAA,CAAO,WAAA,CAAY,UAAA,CAAW,OAAA,EAAS,CAAA;AAChD;AAEA,eAAsB,kBAAA,CAAmB,EAAE,MAAA,EAAQ,cAAA,EAAe,EAAiD;AACjH,EAAA,MAAM,OAAA,GAAU,MAAM,qBAAA,CAAsB;AAAA,IAC1C,MAAA;AAAA,IACA;AAAA,GACD,CAAA;AAED,EAAA,OAAO,OAAA;AACT;AAEA,eAAsB,gBAAA,CAAiB;AAAA,EACrC,MAAA;AAAA,EACA,QAAA;AAAA,EACA;AACF,CAAA,EAAmE;AACjE,EAAA,MAAM,OAAA,GAAU,MAAM,qBAAA,CAAsB;AAAA,IAC1C,MAAA;AAAA,IACA;AAAA,GACD,CAAA;AAED,EAAA,MAAM,MAAA,GAAS,QAAQ,QAAQ,CAAA;AAE/B,EAAA,IAAI,CAAC,MAAA,EAAQ;AACX,IAAA,OAAO,IAAA;AAAA,EACT;AAEA,EAAA,OAAO,MAAA;AACT;AAEA,eAAsB,wBAAA,CAAyB;AAAA,EAC7C,MAAA;AAAA,EACA,KAAA;AAAA,EACA;AACF,CAAA,EAA+D;AAC7D,EAAA,IAAI;AACF,IAAA,MAAM,YAAA,GAAgB,MAAM,MAAA,CAAO,UAAA,IAAc,iBAAA,GAAoB;AAAA,MACnE,KAAA;AAAA,MACA;AAAA,KACD,CAAA,IAAM,EAAE,UAAA,EAAY,EAAE,OAAO,CAAA,EAAG,IAAA,EAAM,CAAA,EAAG,OAAA,EAAS,GAAG,OAAA,EAAS,KAAA,EAAM,EAAG,MAAA,EAAQ,EAAC,EAAE;AACnF,IAAA,OAAO;AAAA,MACL,YAAY,YAAA,CAAa,UAAA;AAAA,MACzB,MAAA,EAAQ,YAAA,CAAa,MAAA,CAAO,GAAA,CAAI,CAAA,KAAA,MAAU,EAAE,GAAG,KAAA,EAAO,GAAG,eAAA,CAAgB,KAAA,CAAM,OAAO,GAAE,CAAE;AAAA,KAC5F;AAAA,EACF,SAAS,KAAA,EAAO;AACd,IAAA,OAAOC,6BAAA,CAAY,OAAO,gCAAgC,CAAA;AAAA,EAC5D;AACF;AAEA,eAAsB,2BAAA,CAA4B;AAAA,EAChD,MAAA;AAAA,EACA,QAAA;AAAA,EACA,UAAA;AAAA,EACA,QAAA;AAAA,EACA;AACF,CAAA,EAA0G;AACxG,EAAA,IAAI;AACF,IAAA,MAAM,YAAA,GAAgB,MAAM,MAAA,CAAO,UAAA,IAAc,oBAAA,GAAuB;AAAA,MACtE,QAAA;AAAA,MACA,UAAA;AAAA,MACA,QAAA;AAAA,MACA;AAAA,KACD,CAAA,IAAM,EAAE,UAAA,EAAY,EAAE,OAAO,CAAA,EAAG,IAAA,EAAM,CAAA,EAAG,OAAA,EAAS,GAAG,OAAA,EAAS,KAAA,EAAM,EAAG,MAAA,EAAQ,EAAC,EAAE;AACnF,IAAA,OAAO;AAAA,MACL,YAAY,YAAA,CAAa,UAAA;AAAA,MACzB,MAAA,EAAQ,YAAA,CAAa,MAAA,CAAO,GAAA,CAAI,CAAA,KAAA,MAAU,EAAE,GAAG,KAAA,EAAO,GAAG,eAAA,CAAgB,KAAA,CAAM,OAAO,GAAE,CAAE;AAAA,KAC5F;AAAA,EACF,SAAS,KAAA,EAAO;AACd,IAAA,OAAOA,6BAAA,CAAY,OAAO,mCAAmC,CAAA;AAAA,EAC/D;AACF;AAEA,eAAsB,2BAAA,CAA4B;AAAA,EAChD,MAAA;AAAA,EACA,QAAA;AAAA,EACA,UAAA;AAAA,EACA;AACF,CAAA,EAAsF;AACpF,EAAA,IAAI;AACF,IAAA,IAAI,aAAA,GAAgB,QAAA;AAEpB,IAAA,IAAI,eAAe,OAAA,EAAS;AAC1B,MAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,YAAA,CAAa,QAAQ,CAAA;AAC1C,MAAA,aAAA,GAAgB,KAAA,CAAM,EAAA;AAAA,IACxB,CAAA,MAAA,IAAW,eAAe,UAAA,EAAY;AACpC,MAAA,MAAM,QAAA,GAAW,MAAA,CAAO,eAAA,CAAgB,QAAQ,CAAA;AAChD,MAAA,aAAA,GAAgB,QAAA,CAAS,EAAA;AAAA,IAC3B;AAEA,IAAA,MAAM,YAAA,GAAgB,MAAM,MAAA,CAAO,UAAA,IAAc,oBAAA,GAAuB;AAAA,MACtE,QAAA,EAAU,aAAA;AAAA,MACV,UAAA;AAAA,MACA;AAAA,KACD,CAAA,IAAM,EAAE,UAAA,EAAY,EAAE,OAAO,CAAA,EAAG,IAAA,EAAM,CAAA,EAAG,OAAA,EAAS,GAAG,OAAA,EAAS,KAAA,EAAM,EAAG,MAAA,EAAQ,EAAC,EAAE;AAEnF,IAAA,OAAO;AAAA,MACL,YAAY,YAAA,CAAa,UAAA;AAAA,MACzB,MAAA,EAAQ,YAAA,CAAa,MAAA,CAAO,GAAA,CAAI,CAAA,KAAA,MAAU,EAAE,GAAG,KAAA,EAAO,GAAG,eAAA,CAAgB,KAAA,CAAM,OAAO,GAAE,CAAE;AAAA,KAC5F;AAAA,EACF,SAAS,KAAA,EAAO;AACd,IAAA,OAAOA,6BAAA,CAAY,OAAO,mCAAmC,CAAA;AAAA,EAC/D;AACF;AAGA,SAAS,gBAAgB,iBAAA,EAA4B;AACnD,EAAA,IAAI,CAAC,iBAAA,EAAmB;AACtB,IAAA,OAAO,EAAC;AAAA,EACV;AAEA,EAAA,MAAM,CAAC,OAAA,EAAS,MAAM,CAAA,GAAI,iBAAA,CAAkB,MAAM,GAAG,CAAA;AAErD,EAAA,OAAO;AAAA,IACL,GAAI,OAAA,GAAU,EAAE,OAAA,KAAY,EAAC;AAAA,IAC7B,GAAI,MAAA,GAAS,EAAE,MAAA,KAAW;AAAC,GAC7B;AACF;AAEA,eAAsB,gBAAA,CAAiB,EAAE,MAAA,EAAQ,KAAA,EAAM,EAAsC;AAC3F,EAAA,IAAI;AACF,IAAA,MAAM,MAAA,GAAU,MAAM,MAAA,CAAO,UAAA,IAAc,SAAA,GAAY,KAAK,KAAM,EAAC;AACnE,IAAA,OAAO,MAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAOA,6BAAA,CAAY,OAAO,oBAAoB,CAAA;AAAA,EAChD;AACF","file":"chunk-BANGX6SC.cjs","sourcesContent":["import type { MastraScorerEntry, ScoreRowData } from '@mastra/core/evals';\nimport type { RequestContext } from '@mastra/core/request-context';\nimport type { StoragePagination } from '@mastra/core/storage';\nimport type { Context } from '../types';\nimport { handleError } from './error';\n\nasync function listScorersFromSystem({\n  mastra,\n  requestContext,\n}: Context & {\n  requestContext: RequestContext;\n}) {\n  const agents = mastra.listAgents();\n  const workflows = mastra.listWorkflows();\n\n  const scorersMap = new Map<\n    string,\n    MastraScorerEntry & { agentIds: string[]; agentNames: string[]; workflowIds: string[]; isRegistered: boolean }\n  >();\n\n  for (const [agentId, agent] of Object.entries(agents)) {\n    const scorers =\n      (await agent.listScorers({\n        requestContext,\n      })) || {};\n\n    if (Object.keys(scorers).length > 0) {\n      for (const [_scorerId, scorer] of Object.entries(scorers)) {\n        const scorerId = scorer.scorer.id;\n        if (scorersMap.has(scorerId)) {\n          scorersMap.get(scorerId)?.agentIds.push(agentId);\n          scorersMap.get(scorerId)?.agentNames.push(agent.name);\n        } else {\n          scorersMap.set(scorerId, {\n            workflowIds: [],\n            ...scorer,\n            agentNames: [agent.name],\n            agentIds: [agentId],\n            isRegistered: false,\n          });\n        }\n      }\n    }\n  }\n\n  for (const [workflowId, workflow] of Object.entries(workflows)) {\n    const scorers =\n      (await workflow.listScorers({\n        requestContext,\n      })) || {};\n\n    if (Object.keys(scorers).length > 0) {\n      for (const [_scorerId, scorer] of Object.entries(scorers)) {\n        const scorerName = scorer.scorer.name;\n        if (scorersMap.has(scorerName)) {\n          scorersMap.get(scorerName)?.workflowIds.push(workflowId);\n        } else {\n          scorersMap.set(scorerName, {\n            agentIds: [],\n            agentNames: [],\n            ...scorer,\n            workflowIds: [workflowId],\n            isRegistered: false,\n          });\n        }\n      }\n    }\n  }\n\n  const registeredScorers = await mastra.listScorers();\n  for (const [_scorerId, scorer] of Object.entries(registeredScorers || {})) {\n    const scorerId = scorer.id;\n    if (scorersMap.has(scorerId)) {\n      scorersMap.get(scorerId)!.isRegistered = true;\n    } else {\n      scorersMap.set(scorerId, {\n        scorer: scorer,\n        agentIds: [],\n        agentNames: [],\n        workflowIds: [],\n        isRegistered: true,\n      });\n    }\n  }\n\n  return Object.fromEntries(scorersMap.entries());\n}\n\nexport async function listScorersHandler({ mastra, requestContext }: Context & { requestContext: RequestContext }) {\n  const scorers = await listScorersFromSystem({\n    mastra,\n    requestContext,\n  });\n\n  return scorers;\n}\n\nexport async function getScorerHandler({\n  mastra,\n  scorerId,\n  requestContext,\n}: Context & { scorerId: string; requestContext: RequestContext }) {\n  const scorers = await listScorersFromSystem({\n    mastra,\n    requestContext,\n  });\n\n  const scorer = scorers[scorerId];\n\n  if (!scorer) {\n    return null;\n  }\n\n  return scorer;\n}\n\nexport async function listScoresByRunIdHandler({\n  mastra,\n  runId,\n  pagination,\n}: Context & { runId: string; pagination: StoragePagination }) {\n  try {\n    const scoreResults = (await mastra.getStorage()?.listScoresByRunId?.({\n      runId,\n      pagination,\n    })) || { pagination: { total: 0, page: 0, perPage: 0, hasMore: false }, scores: [] };\n    return {\n      pagination: scoreResults.pagination,\n      scores: scoreResults.scores.map(score => ({ ...score, ...getTraceDetails(score.traceId) })),\n    };\n  } catch (error) {\n    return handleError(error, 'Error getting scores by run id');\n  }\n}\n\nexport async function listScoresByScorerIdHandler({\n  mastra,\n  scorerId,\n  pagination,\n  entityId,\n  entityType,\n}: Context & { scorerId: string; pagination: StoragePagination; entityId?: string; entityType?: string }) {\n  try {\n    const scoreResults = (await mastra.getStorage()?.listScoresByScorerId?.({\n      scorerId,\n      pagination,\n      entityId,\n      entityType,\n    })) || { pagination: { total: 0, page: 0, perPage: 0, hasMore: false }, scores: [] };\n    return {\n      pagination: scoreResults.pagination,\n      scores: scoreResults.scores.map(score => ({ ...score, ...getTraceDetails(score.traceId) })),\n    };\n  } catch (error) {\n    return handleError(error, 'Error getting scores by scorer id');\n  }\n}\n\nexport async function listScoresByEntityIdHandler({\n  mastra,\n  entityId,\n  entityType,\n  pagination,\n}: Context & { entityId: string; entityType: string; pagination: StoragePagination }) {\n  try {\n    let entityIdToUse = entityId;\n\n    if (entityType === 'AGENT') {\n      const agent = mastra.getAgentById(entityId);\n      entityIdToUse = agent.id;\n    } else if (entityType === 'WORKFLOW') {\n      const workflow = mastra.getWorkflowById(entityId);\n      entityIdToUse = workflow.id;\n    }\n\n    const scoreResults = (await mastra.getStorage()?.listScoresByEntityId?.({\n      entityId: entityIdToUse,\n      entityType,\n      pagination,\n    })) || { pagination: { total: 0, page: 0, perPage: 0, hasMore: false }, scores: [] };\n\n    return {\n      pagination: scoreResults.pagination,\n      scores: scoreResults.scores.map(score => ({ ...score, ...getTraceDetails(score.traceId) })),\n    };\n  } catch (error) {\n    return handleError(error, 'Error getting scores by entity id');\n  }\n}\n\n// Legacy function to get trace and span details\nfunction getTraceDetails(traceIdWithSpanId?: string) {\n  if (!traceIdWithSpanId) {\n    return {};\n  }\n\n  const [traceId, spanId] = traceIdWithSpanId.split('-');\n\n  return {\n    ...(traceId ? { traceId } : {}),\n    ...(spanId ? { spanId } : {}),\n  };\n}\n\nexport async function saveScoreHandler({ mastra, score }: Context & { score: ScoreRowData }) {\n  try {\n    const scores = (await mastra.getStorage()?.saveScore?.(score)) || [];\n    return scores;\n  } catch (error) {\n    return handleError(error, 'Error saving score');\n  }\n}\n"]}

package/dist/{chunk-BMOVKZO7.cjs → chunk-BHQH5QXD.cjs}

@@ -1,9 +1,9 @@
 'use strict';
 
-var chunkLZKJT7QP_cjs = require('./chunk-LZKJT7QP.cjs');
-var chunkG4PUALCE_cjs = require('./chunk-G4PUALCE.cjs');
-var chunkRE4RPXT2_cjs = require('./chunk-RE4RPXT2.cjs');
-var chunk7NADHFD2_cjs = require('./chunk-7NADHFD2.cjs');
+var chunkGU4EWMZB_cjs = require('./chunk-GU4EWMZB.cjs');
+var chunkBTWIR2B7_cjs = require('./chunk-BTWIR2B7.cjs');
+var chunkV5WWQN7P_cjs = require('./chunk-V5WWQN7P.cjs');
+var chunk64ITUOXI_cjs = require('./chunk-64ITUOXI.cjs');
 var chunkO7I5CWRX_cjs = require('./chunk-O7I5CWRX.cjs');
 var tools = require('@mastra/core/tools');
 var zodToJson = require('@mastra/core/utils/zod-to-json');
@@ -27,8 +27,8 @@ async function listToolsHandler({ tools }) {
         const tool = _tool;
         acc[id] = {
           ...tool,
-          inputSchema: tool.inputSchema ? chunkLZKJT7QP_cjs.stringify(zodToJson.zodToJsonSchema(tool.inputSchema)) : void 0,
-          outputSchema: tool.outputSchema ? chunkLZKJT7QP_cjs.stringify(zodToJson.zodToJsonSchema(tool.outputSchema)) : void 0
+          inputSchema: tool.inputSchema ? chunkGU4EWMZB_cjs.stringify(zodToJson.zodToJsonSchema(tool.inputSchema)) : void 0,
+          outputSchema: tool.outputSchema ? chunkGU4EWMZB_cjs.stringify(zodToJson.zodToJsonSchema(tool.outputSchema)) : void 0
         };
         return acc;
       },
@@ -36,23 +36,23 @@ async function listToolsHandler({ tools }) {
     );
     return serializedTools;
   } catch (error) {
-    return chunkRE4RPXT2_cjs.handleError(error, "Error getting tools");
+    return chunkV5WWQN7P_cjs.handleError(error, "Error getting tools");
   }
 }
 async function getToolByIdHandler({ tools, toolId }) {
   try {
     const tool = Object.values(tools || {}).find((tool2) => tool2.id === toolId);
     if (!tool) {
-      throw new chunk7NADHFD2_cjs.HTTPException(404, { message: "Tool not found" });
+      throw new chunk64ITUOXI_cjs.HTTPException(404, { message: "Tool not found" });
     }
     const serializedTool = {
       ...tool,
-      inputSchema: tool.inputSchema ? chunkLZKJT7QP_cjs.stringify(zodToJson.zodToJsonSchema(tool.inputSchema)) : void 0,
-      outputSchema: tool.outputSchema ? chunkLZKJT7QP_cjs.stringify(zodToJson.zodToJsonSchema(tool.outputSchema)) : void 0
+      inputSchema: tool.inputSchema ? chunkGU4EWMZB_cjs.stringify(zodToJson.zodToJsonSchema(tool.inputSchema)) : void 0,
+      outputSchema: tool.outputSchema ? chunkGU4EWMZB_cjs.stringify(zodToJson.zodToJsonSchema(tool.outputSchema)) : void 0
     };
     return serializedTool;
   } catch (error) {
-    return chunkRE4RPXT2_cjs.handleError(error, "Error getting tool");
+    return chunkV5WWQN7P_cjs.handleError(error, "Error getting tool");
   }
 }
 function executeToolHandler(tools$1) {
@@ -65,33 +65,36 @@ function executeToolHandler(tools$1) {
   }) => {
     try {
       if (!toolId) {
-        throw new chunk7NADHFD2_cjs.HTTPException(400, { message: "Tool ID is required" });
+        throw new chunk64ITUOXI_cjs.HTTPException(400, { message: "Tool ID is required" });
       }
       const tool = Object.values(tools$1 || {}).find((tool2) => tool2.id === toolId);
       if (!tool) {
-        throw new chunk7NADHFD2_cjs.HTTPException(404, { message: "Tool not found" });
+        throw new chunk64ITUOXI_cjs.HTTPException(404, { message: "Tool not found" });
       }
       if (!tool?.execute) {
-        throw new chunk7NADHFD2_cjs.HTTPException(400, { message: "Tool is not executable" });
+        throw new chunk64ITUOXI_cjs.HTTPException(400, { message: "Tool is not executable" });
       }
-      chunkG4PUALCE_cjs.validateBody({ data });
+      chunkBTWIR2B7_cjs.validateBody({ data });
       if (tools.isVercelTool(tool)) {
         const result2 = await tool.execute(data);
         return result2;
       }
-      const result = await tool.execute({
-        context: data,
+      const result = await tool.execute(data, {
         mastra,
-        runId,
         requestContext,
         // TODO: Pass proper tracing context when server API supports tracing
         tracingContext: { currentSpan: void 0 },
-        suspend: async () => {
-        }
+        ...runId ? {
+          workflow: {
+            runId,
+            suspend: async () => {
+            }
+          }
+        } : {}
       });
       return result;
     } catch (error) {
-      return chunkRE4RPXT2_cjs.handleError(error, "Error executing tool");
+      return chunkV5WWQN7P_cjs.handleError(error, "Error executing tool");
     }
   };
 }
@@ -104,21 +107,21 @@ async function getAgentToolHandler({
   try {
     const agent = agentId ? mastra.getAgent(agentId) : null;
     if (!agent) {
-      throw new chunk7NADHFD2_cjs.HTTPException(404, { message: "Agent not found" });
+      throw new chunk64ITUOXI_cjs.HTTPException(404, { message: "Agent not found" });
     }
     const agentTools = await agent.listTools({ requestContext });
     const tool = Object.values(agentTools || {}).find((tool2) => tool2.id === toolId);
     if (!tool) {
-      throw new chunk7NADHFD2_cjs.HTTPException(404, { message: "Tool not found" });
+      throw new chunk64ITUOXI_cjs.HTTPException(404, { message: "Tool not found" });
     }
     const serializedTool = {
       ...tool,
-      inputSchema: tool.inputSchema ? chunkLZKJT7QP_cjs.stringify(zodToJson.zodToJsonSchema(tool.inputSchema)) : void 0,
-      outputSchema: tool.outputSchema ? chunkLZKJT7QP_cjs.stringify(zodToJson.zodToJsonSchema(tool.outputSchema)) : void 0
+      inputSchema: tool.inputSchema ? chunkGU4EWMZB_cjs.stringify(zodToJson.zodToJsonSchema(tool.inputSchema)) : void 0,
+      outputSchema: tool.outputSchema ? chunkGU4EWMZB_cjs.stringify(zodToJson.zodToJsonSchema(tool.outputSchema)) : void 0
     };
     return serializedTool;
   } catch (error) {
-    return chunkRE4RPXT2_cjs.handleError(error, "Error getting agent tool");
+    return chunkV5WWQN7P_cjs.handleError(error, "Error getting agent tool");
   }
 }
 async function executeAgentToolHandler({
@@ -131,29 +134,31 @@ async function executeAgentToolHandler({
   try {
     const agent = agentId ? mastra.getAgent(agentId) : null;
     if (!agent) {
-      throw new chunk7NADHFD2_cjs.HTTPException(404, { message: "Tool not found" });
+      throw new chunk64ITUOXI_cjs.HTTPException(404, { message: "Tool not found" });
     }
     const agentTools = await agent.listTools({ requestContext });
     const tool = Object.values(agentTools || {}).find((tool2) => tool2.id === toolId);
     if (!tool) {
-      throw new chunk7NADHFD2_cjs.HTTPException(404, { message: "Tool not found" });
+      throw new chunk64ITUOXI_cjs.HTTPException(404, { message: "Tool not found" });
     }
     if (!tool?.execute) {
-      throw new chunk7NADHFD2_cjs.HTTPException(400, { message: "Tool is not executable" });
+      throw new chunk64ITUOXI_cjs.HTTPException(400, { message: "Tool is not executable" });
     }
-    const result = await tool.execute({
-      context: data,
-      requestContext,
+    const result = await tool.execute(data, {
       mastra,
-      runId: agentId,
+      requestContext,
       // TODO: Pass proper tracing context when server API supports tracing
       tracingContext: { currentSpan: void 0 },
-      suspend: async () => {
+      agent: {
+        messages: [],
+        toolCallId: "",
+        suspend: async () => {
+        }
       }
     });
     return result;
   } catch (error) {
-    return chunkRE4RPXT2_cjs.handleError(error, "Error executing tool");
+    return chunkV5WWQN7P_cjs.handleError(error, "Error executing tool");
   }
 }
 
@@ -163,5 +168,5 @@ exports.getAgentToolHandler = getAgentToolHandler;
 exports.getToolByIdHandler = getToolByIdHandler;
 exports.listToolsHandler = listToolsHandler;
 exports.tools_exports = tools_exports;
-//# sourceMappingURL=chunk-BMOVKZO7.cjs.map
-//# sourceMappingURL=chunk-BMOVKZO7.cjs.map
+//# sourceMappingURL=chunk-BHQH5QXD.cjs.map
+//# sourceMappingURL=chunk-BHQH5QXD.cjs.map

package/dist/chunk-BHQH5QXD.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/server/handlers/tools.ts"],"names":["__export","stringify","zodToJsonSchema","handleError","tool","HTTPException","tools","validateBody","isVercelTool","result"],"mappings":";;;;;;;;;;;AAAA,IAAA,aAAA,GAAA;AAAAA,0BAAA,CAAA,aAAA,EAAA;AAAA,EAAA,uBAAA,EAAA,MAAA,uBAAA;AAAA,EAAA,kBAAA,EAAA,MAAA,kBAAA;AAAA,EAAA,mBAAA,EAAA,MAAA,mBAAA;AAAA,EAAA,kBAAA,EAAA,MAAA,kBAAA;AAAA,EAAA,gBAAA,EAAA,MAAA;AAAA,CAAA,CAAA;AAkBA,eAAsB,gBAAA,CAAiB,EAAE,KAAA,EAAM,EAAgC;AAC7E,EAAA,IAAI;AACF,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,OAAO,EAAC;AAAA,IACV;AAEA,IAAA,MAAM,eAAA,GAAkB,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA,CAAE,MAAA;AAAA,MAC5C,CAAC,GAAA,EAAK,CAAC,EAAA,EAAI,KAAK,CAAA,KAAM;AACpB,QAAA,MAAM,IAAA,GAAO,KAAA;AACb,QAAA,GAAA,CAAI,EAAE,CAAA,GAAI;AAAA,UACR,GAAG,IAAA;AAAA,UACH,WAAA,EAAa,KAAK,WAAA,GAAcC,2BAAA,CAAUC,0BAAgB,IAAA,CAAK,WAAW,CAAC,CAAA,GAAI,MAAA;AAAA,UAC/E,YAAA,EAAc,KAAK,YAAA,GAAeD,2BAAA,CAAUC,0BAAgB,IAAA,CAAK,YAAY,CAAC,CAAA,GAAI;AAAA,SACpF;AACA,QAAA,OAAO,GAAA;AAAA,MACT,CAAA;AAAA,MACA;AAAC,KACH;AAEA,IAAA,OAAO,eAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAOC,6BAAA,CAAY,OAAO,qBAAqB,CAAA;AAAA,EACjD;AACF;AAEA,eAAsB,kBAAA,CAAmB,EAAE,KAAA,EAAO,MAAA,EAAO,EAA2C;AAClG,EAAA,IAAI;AACF,IAAA,MAAM,IAAA,GAAO,MAAA,CAAO,MAAA,CAAO,KAAA,IAAS,EAAE,CAAA,CAAE,IAAA,CAAK,CAACC,KAAAA,KAAcA,KAAAA,CAAK,EAAA,KAAO,MAAM,CAAA;AAE9E,IAAA,IAAI,CAAC,IAAA,EAAM;AACT,MAAA,MAAM,IAAIC,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,kBAAkB,CAAA;AAAA,IAC5D;AAEA,IAAA,MAAM,cAAA,GAAiB;AAAA,MACrB,GAAG,IAAA;AAAA,MACH,WAAA,EAAa,KAAK,WAAA,GAAcJ,2BAAA,CAAUC,0BAAgB,IAAA,CAAK,WAAW,CAAC,CAAA,GAAI,MAAA;AAAA,MAC/E,YAAA,EAAc,KAAK,YAAA,GAAeD,2BAAA,CAAUC,0BAAgB,IAAA,CAAK,YAAY,CAAC,CAAA,GAAI;AAAA,KACpF;AAEA,IAAA,OAAO,cAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAOC,6BAAA,CAAY,OAAO,oBAAoB,CAAA;AAAA,EAChD;AACF;AAEO,SAAS,mBAAmBG,OAAA,EAA8B;AAC/D,EAAA,OAAO,OAAO;AAAA,IACZ,MAAA;AAAA,IACA,KAAA;AAAA,IACA,MAAA;AAAA,IACA,IAAA;AAAA,IACA;AAAA,GACF,KAGM;AACJ,IAAA,IAAI;AACF,MAAA,IAAI,CAAC,MAAA,EAAQ;AACX,QAAA,MAAM,IAAID,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,uBAAuB,CAAA;AAAA,MACjE;AAEA,MAAA,MAAM,IAAA,GAAO,MAAA,CAAO,MAAA,CAAOC,OAAA,IAAS,EAAE,CAAA,CAAE,IAAA,CAAK,CAACF,KAAAA,KAAcA,KAAAA,CAAK,EAAA,KAAO,MAAM,CAAA;AAE9E,MAAA,IAAI,CAAC,IAAA,EAAM;AACT,QAAA,MAAM,IAAIC,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,kBAAkB,CAAA;AAAA,MAC5D;AAEA,MAAA,IAAI,CAAC,MAAM,OAAA,EAAS;AAClB,QAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,0BAA0B,CAAA;AAAA,MACpE;AAEA,MAAAE,8BAAA,CAAa,EAAE,MAAM,CAAA;AAErB,MAAA,IAAIC,kBAAA,CAAa,IAAI,CAAA,EAAG;AACtB,QAAA,MAAMC,OAAAA,GAAS,MAAO,IAAA,CAAa,OAAA,CAAQ,IAAI,CAAA;AAC/C,QAAA,OAAOA,OAAAA;AAAA,MACT;AAEA,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,OAAA,CAAQ,IAAA,EAAO;AAAA,QACvC,MAAA;AAAA,QACA,cAAA;AAAA;AAAA,QAEA,cAAA,EAAgB,EAAE,WAAA,EAAa,MAAA,EAAU;AAAA,QACzC,GAAI,KAAA,GACA;AAAA,UACE,QAAA,EAAU;AAAA,YACR,KAAA;AAAA,YACA,SAAS,YAAY;AAAA,YAAC;AAAA;AACxB,YAEF;AAAC,OACN,CAAA;AACD,MAAA,OAAO,MAAA;AAAA,IACT,SAAS,KAAA,EAAO;AACd,MAAA,OAAON,6BAAA,CAAY,OAAO,sBAAsB,CAAA;AAAA,IAClD;AAAA,EACF,CAAA;AACF;AAEA,eAAsB,mBAAA,CAAoB;AAAA,EACxC,MAAA;AAAA,EACA,OAAA;AAAA,EACA,MAAA;AAAA,EACA;AACF,CAAA,EAGG;AACD,EAAA,IAAI;AACF,IAAA,MAAM,KAAA,GAAQ,OAAA,GAAU,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA,GAAI,IAAA;AACnD,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,MAAM,IAAIE,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,mBAAmB,CAAA;AAAA,IAC7D;AAEA,IAAA,MAAM,aAAa,MAAM,KAAA,CAAM,SAAA,CAAU,EAAE,gBAAgB,CAAA;AAE3D,IAAA,MAAM,IAAA,GAAO,MAAA,CAAO,MAAA,CAAO,UAAA,IAAc,EAAE,CAAA,CAAE,IAAA,CAAK,CAACD,KAAAA,KAAcA,KAAAA,CAAK,EAAA,KAAO,MAAM,CAAA;AAEnF,IAAA,IAAI,CAAC,IAAA,EAAM;AACT,MAAA,MAAM,IAAIC,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,kBAAkB,CAAA;AAAA,IAC5D;AAEA,IAAA,MAAM,cAAA,GAAiB;AAAA,MACrB,GAAG,IAAA;AAAA,MACH,WAAA,EAAa,KAAK,WAAA,GAAcJ,2BAAA,CAAUC,0BAAgB,IAAA,CAAK,WAAW,CAAC,CAAA,GAAI,MAAA;AAAA,MAC/E,YAAA,EAAc,KAAK,YAAA,GAAeD,2BAAA,CAAUC,0BAAgB,IAAA,CAAK,YAAY,CAAC,CAAA,GAAI;AAAA,KACpF;AAEA,IAAA,OAAO,cAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAOC,6BAAA,CAAY,OAAO,0BAA0B,CAAA;AAAA,EACtD;AACF;AAEA,eAAsB,uBAAA,CAAwB;AAAA,EAC5C,MAAA;AAAA,EACA,OAAA;AAAA,EACA,MAAA;AAAA,EACA,IAAA;AAAA,EACA;AACF,CAAA,EAIG;AACD,EAAA,IAAI;AACF,IAAA,MAAM,KAAA,GAAQ,OAAA,GAAU,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA,GAAI,IAAA;AACnD,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,MAAM,IAAIE,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,kBAAkB,CAAA;AAAA,IAC5D;AAEA,IAAA,MAAM,aAAa,MAAM,KAAA,CAAM,SAAA,CAAU,EAAE,gBAAgB,CAAA;AAE3D,IAAA,MAAM,IAAA,GAAO,MAAA,CAAO,MAAA,CAAO,UAAA,IAAc,EAAE,CAAA,CAAE,IAAA,CAAK,CAACD,KAAAA,KAAcA,KAAAA,CAAK,EAAA,KAAO,MAAM,CAAA;AAEnF,IAAA,IAAI,CAAC,IAAA,EAAM;AACT,MAAA,MAAM,IAAIC,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,kBAAkB,CAAA;AAAA,IAC5D;AAEA,IAAA,IAAI,CAAC,MAAM,OAAA,EAAS;AAClB,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,0BAA0B,CAAA;AAAA,IACpE;AAOA,IAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,OAAA,CAAQ,IAAA,EAAM;AAAA,MACtC,MAAA;AAAA,MACA,cAAA;AAAA;AAAA,MAEA,cAAA,EAAgB,EAAE,WAAA,EAAa,MAAA,EAAU;AAAA,MACzC,KAAA,EAAO;AAAA,QACL,UAAU,EAAC;AAAA,QACX,UAAA,EAAY,EAAA;AAAA,QACZ,SAAS,YAAY;AAAA,QAAC;AAAA;AACxB,KACD,CAAA;AAED,IAAA,OAAO,MAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAOF,6BAAA,CAAY,OAAO,sBAAsB,CAAA;AAAA,EAClD;AACF","file":"chunk-BHQH5QXD.cjs","sourcesContent":["import type { RequestContext } from '@mastra/core/di';\nimport type { ToolAction, VercelTool } from '@mastra/core/tools';\nimport { isVercelTool } from '@mastra/core/tools';\nimport { zodToJsonSchema } from '@mastra/core/utils/zod-to-json';\nimport { stringify } from 'superjson';\nimport { HTTPException } from '../http-exception';\nimport type { Context } from '../types';\n\nimport { handleError } from './error';\nimport { validateBody } from './utils';\n\ninterface ToolsContext extends Context {\n  tools?: Record<string, ToolAction | VercelTool>;\n  toolId?: string;\n  runId?: string;\n}\n\n// Tool handlers\nexport async function listToolsHandler({ tools }: Pick<ToolsContext, 'tools'>) {\n  try {\n    if (!tools) {\n      return {};\n    }\n\n    const serializedTools = Object.entries(tools).reduce(\n      (acc, [id, _tool]) => {\n        const tool = _tool as any;\n        acc[id] = {\n          ...tool,\n          inputSchema: tool.inputSchema ? stringify(zodToJsonSchema(tool.inputSchema)) : undefined,\n          outputSchema: tool.outputSchema ? stringify(zodToJsonSchema(tool.outputSchema)) : undefined,\n        };\n        return acc;\n      },\n      {} as Record<string, any>,\n    );\n\n    return serializedTools;\n  } catch (error) {\n    return handleError(error, 'Error getting tools');\n  }\n}\n\nexport async function getToolByIdHandler({ tools, toolId }: Pick<ToolsContext, 'tools' | 'toolId'>) {\n  try {\n    const tool = Object.values(tools || {}).find((tool: any) => tool.id === toolId) as any;\n\n    if (!tool) {\n      throw new HTTPException(404, { message: 'Tool not found' });\n    }\n\n    const serializedTool = {\n      ...tool,\n      inputSchema: tool.inputSchema ? stringify(zodToJsonSchema(tool.inputSchema)) : undefined,\n      outputSchema: tool.outputSchema ? stringify(zodToJsonSchema(tool.outputSchema)) : undefined,\n    };\n\n    return serializedTool;\n  } catch (error) {\n    return handleError(error, 'Error getting tool');\n  }\n}\n\nexport function executeToolHandler(tools: ToolsContext['tools']) {\n  return async ({\n    mastra,\n    runId,\n    toolId,\n    data,\n    requestContext,\n  }: Pick<ToolsContext, 'mastra' | 'toolId' | 'runId'> & {\n    data?: unknown;\n    requestContext: RequestContext;\n  }) => {\n    try {\n      if (!toolId) {\n        throw new HTTPException(400, { message: 'Tool ID is required' });\n      }\n\n      const tool = Object.values(tools || {}).find((tool: any) => tool.id === toolId) as any;\n\n      if (!tool) {\n        throw new HTTPException(404, { message: 'Tool not found' });\n      }\n\n      if (!tool?.execute) {\n        throw new HTTPException(400, { message: 'Tool is not executable' });\n      }\n\n      validateBody({ data });\n\n      if (isVercelTool(tool)) {\n        const result = await (tool as any).execute(data);\n        return result;\n      }\n\n      const result = await tool.execute(data!, {\n        mastra,\n        requestContext,\n        // TODO: Pass proper tracing context when server API supports tracing\n        tracingContext: { currentSpan: undefined },\n        ...(runId\n          ? {\n              workflow: {\n                runId,\n                suspend: async () => {},\n              },\n            }\n          : {}),\n      });\n      return result;\n    } catch (error) {\n      return handleError(error, 'Error executing tool');\n    }\n  };\n}\n\nexport async function getAgentToolHandler({\n  mastra,\n  agentId,\n  toolId,\n  requestContext,\n}: Pick<ToolsContext, 'mastra' | 'toolId'> & {\n  agentId?: string;\n  requestContext: RequestContext;\n}) {\n  try {\n    const agent = agentId ? mastra.getAgent(agentId) : null;\n    if (!agent) {\n      throw new HTTPException(404, { message: 'Agent not found' });\n    }\n\n    const agentTools = await agent.listTools({ requestContext });\n\n    const tool = Object.values(agentTools || {}).find((tool: any) => tool.id === toolId) as any;\n\n    if (!tool) {\n      throw new HTTPException(404, { message: 'Tool not found' });\n    }\n\n    const serializedTool = {\n      ...tool,\n      inputSchema: tool.inputSchema ? stringify(zodToJsonSchema(tool.inputSchema)) : undefined,\n      outputSchema: tool.outputSchema ? stringify(zodToJsonSchema(tool.outputSchema)) : undefined,\n    };\n\n    return serializedTool;\n  } catch (error) {\n    return handleError(error, 'Error getting agent tool');\n  }\n}\n\nexport async function executeAgentToolHandler({\n  mastra,\n  agentId,\n  toolId,\n  data,\n  requestContext,\n}: Pick<ToolsContext, 'mastra' | 'toolId'> & {\n  agentId?: string;\n  data: any;\n  requestContext: RequestContext;\n}) {\n  try {\n    const agent = agentId ? mastra.getAgent(agentId) : null;\n    if (!agent) {\n      throw new HTTPException(404, { message: 'Tool not found' });\n    }\n\n    const agentTools = await agent.listTools({ requestContext });\n\n    const tool = Object.values(agentTools || {}).find((tool: any) => tool.id === toolId) as any;\n\n    if (!tool) {\n      throw new HTTPException(404, { message: 'Tool not found' });\n    }\n\n    if (!tool?.execute) {\n      throw new HTTPException(400, { message: 'Tool is not executable' });\n    }\n\n    // if (isVercelTool(tool)) {\n    //   const result = await (tool as any).execute(data);\n    //   return result;\n    // }\n\n    const result = await tool.execute(data, {\n      mastra,\n      requestContext,\n      // TODO: Pass proper tracing context when server API supports tracing\n      tracingContext: { currentSpan: undefined },\n      agent: {\n        messages: [],\n        toolCallId: '',\n        suspend: async () => {},\n      },\n    });\n\n    return result;\n  } catch (error) {\n    return handleError(error, 'Error executing tool');\n  }\n}\n"]}

package/dist/{chunk-G4PUALCE.cjs → chunk-BTWIR2B7.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunk7NADHFD2_cjs = require('./chunk-7NADHFD2.cjs');
+var chunk64ITUOXI_cjs = require('./chunk-64ITUOXI.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 chunk7NADHFD2_cjs.HTTPException(400, { message: Object.values(errorResponse)[0] });
+    throw new chunk64ITUOXI_cjs.HTTPException(400, { message: Object.values(errorResponse)[0] });
   }
 }
 function sanitizeBody(body, disallowedKeys) {
@@ -24,5 +24,5 @@ function sanitizeBody(body, disallowedKeys) {
 
 exports.sanitizeBody = sanitizeBody;
 exports.validateBody = validateBody;
-//# sourceMappingURL=chunk-G4PUALCE.cjs.map
-//# sourceMappingURL=chunk-G4PUALCE.cjs.map
+//# sourceMappingURL=chunk-BTWIR2B7.cjs.map
+//# sourceMappingURL=chunk-BTWIR2B7.cjs.map

package/dist/{chunk-G4PUALCE.cjs.map → chunk-BTWIR2B7.cjs.map}

@@ -1 +1 @@
-{"version":3,"sources":["../src/server/handlers/utils.ts"],"names":["HTTPException"],"mappings":";;;;;AAGO,SAAS,aAAa,IAAA,EAA+B;AAC1D,EAAA,MAAM,aAAA,GAAgB,MAAA,CAAO,OAAA,CAAQ,IAAI,CAAA,CAAE,MAAA,CAA+B,CAAC,GAAA,EAAK,CAAC,GAAA,EAAK,KAAK,CAAA,KAAM;AAC/F,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,GAAA,CAAI,GAAG,CAAA,GAAI,CAAA,UAAA,EAAa,GAAG,CAAA,aAAA,CAAA;AAAA,IAC7B;AACA,IAAA,OAAO,GAAA;AAAA,EACT,CAAA,EAAG,EAAE,CAAA;AAEL,EAAA,IAAI,MAAA,CAAO,IAAA,CAAK,aAAa,CAAA,CAAE,SAAS,CAAA,EAAG;AACzC,IAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,MAAA,CAAO,MAAA,CAAO,aAAa,CAAA,CAAE,CAAC,CAAA,EAAG,CAAA;AAAA,EAC3E;AACF;AAOO,SAAS,YAAA,CAAa,MAA+B,cAAA,EAA0B;AACpF,EAAA,KAAA,MAAW,OAAO,cAAA,EAAgB;AAChC,IAAA,IAAI,OAAO,IAAA,EAAM;AACf,MAAA,OAAO,KAAK,GAAG,CAAA;AAAA,IACjB;AAAA,EACF;AACF","file":"chunk-G4PUALCE.cjs","sourcesContent":["import { HTTPException } from '../http-exception';\n\n// Validation helper\nexport function validateBody(body: Record<string, unknown>) {\n  const errorResponse = Object.entries(body).reduce<Record<string, string>>((acc, [key, value]) => {\n    if (!value) {\n      acc[key] = `Argument \"${key}\" is required`;\n    }\n    return acc;\n  }, {});\n\n  if (Object.keys(errorResponse).length > 0) {\n    throw new HTTPException(400, { message: Object.values(errorResponse)[0] });\n  }\n}\n\n/**\n * sanitizes the body by removing disallowed keys.\n * @param body body to sanitize\n * @param disallowedKeys keys to remove from the body\n */\nexport function sanitizeBody(body: Record<string, unknown>, disallowedKeys: string[]) {\n  for (const key of disallowedKeys) {\n    if (key in body) {\n      delete body[key];\n    }\n  }\n}\n"]}
+{"version":3,"sources":["../src/server/handlers/utils.ts"],"names":["HTTPException"],"mappings":";;;;;AAGO,SAAS,aAAa,IAAA,EAA+B;AAC1D,EAAA,MAAM,aAAA,GAAgB,MAAA,CAAO,OAAA,CAAQ,IAAI,CAAA,CAAE,MAAA,CAA+B,CAAC,GAAA,EAAK,CAAC,GAAA,EAAK,KAAK,CAAA,KAAM;AAC/F,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,GAAA,CAAI,GAAG,CAAA,GAAI,CAAA,UAAA,EAAa,GAAG,CAAA,aAAA,CAAA;AAAA,IAC7B;AACA,IAAA,OAAO,GAAA;AAAA,EACT,CAAA,EAAG,EAAE,CAAA;AAEL,EAAA,IAAI,MAAA,CAAO,IAAA,CAAK,aAAa,CAAA,CAAE,SAAS,CAAA,EAAG;AACzC,IAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,MAAA,CAAO,MAAA,CAAO,aAAa,CAAA,CAAE,CAAC,CAAA,EAAG,CAAA;AAAA,EAC3E;AACF;AAOO,SAAS,YAAA,CAAa,MAA+B,cAAA,EAA0B;AACpF,EAAA,KAAA,MAAW,OAAO,cAAA,EAAgB;AAChC,IAAA,IAAI,OAAO,IAAA,EAAM;AACf,MAAA,OAAO,KAAK,GAAG,CAAA;AAAA,IACjB;AAAA,EACF;AACF","file":"chunk-BTWIR2B7.cjs","sourcesContent":["import { HTTPException } from '../http-exception';\n\n// Validation helper\nexport function validateBody(body: Record<string, unknown>) {\n  const errorResponse = Object.entries(body).reduce<Record<string, string>>((acc, [key, value]) => {\n    if (!value) {\n      acc[key] = `Argument \"${key}\" is required`;\n    }\n    return acc;\n  }, {});\n\n  if (Object.keys(errorResponse).length > 0) {\n    throw new HTTPException(400, { message: Object.values(errorResponse)[0] });\n  }\n}\n\n/**\n * sanitizes the body by removing disallowed keys.\n * @param body body to sanitize\n * @param disallowedKeys keys to remove from the body\n */\nexport function sanitizeBody(body: Record<string, unknown>, disallowedKeys: string[]) {\n  for (const key of disallowedKeys) {\n    if (key in body) {\n      delete body[key];\n    }\n  }\n}\n"]}

package/dist/{chunk-6562SOHJ.js → chunk-DNUO33HF.js}

@@ -1,4 +1,4 @@
-import { stringify } from './chunk-GQRJOUV7.js';
+import { stringify } from './chunk-LF2ZLOFP.js';
 import { zodToJsonSchema } from '@mastra/core/utils/zod-to-json';
 
 function getSteps(steps, path) {
@@ -61,7 +61,6 @@ var WorkflowRegistry = class {
         workflow.__registerPrimitives({
           logger: mastra.getLogger(),
           storage: mastra.getStorage(),
-          memory: mastra.getMemory(),
           agents: mastra.listAgents(),
           tts: mastra.getTTS(),
           vectors: mastra.getVectors()
@@ -126,5 +125,5 @@ function convertInstructionsToString(message) {
 }
 
 export { WorkflowRegistry, convertInstructionsToString, getWorkflowInfo };
-//# sourceMappingURL=chunk-6562SOHJ.js.map
-//# sourceMappingURL=chunk-6562SOHJ.js.map
+//# sourceMappingURL=chunk-DNUO33HF.js.map
+//# sourceMappingURL=chunk-DNUO33HF.js.map

package/dist/chunk-DNUO33HF.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/server/utils.ts"],"names":[],"mappings":";;;AAKA,SAAS,QAAA,CAAS,OAA0C,IAAA,EAAe;AACzE,EAAA,OAAO,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA,CAAE,MAAA,CAAY,CAAC,GAAA,EAAK,CAAC,GAAA,EAAK,IAAI,CAAA,KAAM;AAC7D,IAAA,MAAM,UAAU,IAAA,GAAO,CAAA,EAAG,IAAI,CAAA,CAAA,EAAI,GAAG,CAAA,CAAA,GAAK,GAAA;AAC1C,IAAA,GAAA,CAAI,OAAO,CAAA,GAAI;AAAA,MACb,IAAI,IAAA,CAAK,EAAA;AAAA,MACT,aAAa,IAAA,CAAK,WAAA;AAAA,MAClB,WAAA,EAAa,KAAK,WAAA,GAAc,SAAA,CAAU,gBAAgB,IAAA,CAAK,WAAW,CAAC,CAAA,GAAI,MAAA;AAAA,MAC/E,YAAA,EAAc,KAAK,YAAA,GAAe,SAAA,CAAU,gBAAgB,IAAA,CAAK,YAAY,CAAC,CAAA,GAAI,MAAA;AAAA,MAClF,YAAA,EAAc,KAAK,YAAA,GAAe,SAAA,CAAU,gBAAgB,IAAA,CAAK,YAAY,CAAC,CAAA,GAAI,MAAA;AAAA,MAClF,aAAA,EAAe,KAAK,aAAA,GAAgB,SAAA,CAAU,gBAAgB,IAAA,CAAK,aAAa,CAAC,CAAA,GAAI,MAAA;AAAA,MACrF,UAAA,EAAY,KAAK,SAAA,KAAc,UAAA;AAAA,MAC/B,WAAW,IAAA,CAAK;AAAA,KAClB;AAEA,IAAA,IAAI,IAAA,CAAK,SAAA,KAAc,UAAA,IAAc,IAAA,CAAK,KAAA,EAAO;AAC/C,MAAA,MAAM,cAAc,QAAA,CAAS,IAAA,CAAK,KAAA,EAAO,OAAO,KAAK,EAAC;AACtD,MAAA,GAAA,GAAM,EAAE,GAAG,GAAA,EAAK,GAAG,WAAA,EAAY;AAAA,IACjC;AAEA,IAAA,OAAO,GAAA;AAAA,EACT,CAAA,EAAG,EAAE,CAAA;AACP;AAEO,SAAS,gBAAgB,QAAA,EAAkC;AAChE,EAAA,OAAO;AAAA,IACL,MAAM,QAAA,CAAS,IAAA;AAAA,IACf,aAAa,QAAA,CAAS,WAAA;AAAA,IACtB,KAAA,EAAO,MAAA,CAAO,OAAA,CAAQ,QAAA,CAAS,KAAK,CAAA,CAAE,MAAA,CAAY,CAAC,GAAA,EAAK,CAAC,GAAA,EAAK,IAAI,CAAA,KAAM;AACtE,MAAA,GAAA,CAAI,GAAG,CAAA,GAAI;AAAA,QACT,IAAI,IAAA,CAAK,EAAA;AAAA,QACT,aAAa,IAAA,CAAK,WAAA;AAAA,QAClB,WAAA,EAAa,KAAK,WAAA,GAAc,SAAA,CAAU,gBAAgB,IAAA,CAAK,WAAW,CAAC,CAAA,GAAI,MAAA;AAAA,QAC/E,YAAA,EAAc,KAAK,YAAA,GAAe,SAAA,CAAU,gBAAgB,IAAA,CAAK,YAAY,CAAC,CAAA,GAAI,MAAA;AAAA,QAClF,YAAA,EAAc,KAAK,YAAA,GAAe,SAAA,CAAU,gBAAgB,IAAA,CAAK,YAAY,CAAC,CAAA,GAAI,MAAA;AAAA,QAClF,aAAA,EAAe,KAAK,aAAA,GAAgB,SAAA,CAAU,gBAAgB,IAAA,CAAK,aAAa,CAAC,CAAA,GAAI,MAAA;AAAA,QACrF,WAAW,IAAA,CAAK;AAAA,OAClB;AACA,MAAA,OAAO,GAAA;AAAA,IACT,CAAA,EAAG,EAAE,CAAA;AAAA,IACL,QAAA,EAAU,QAAA,CAAS,QAAA,CAAS,KAAK,KAAK,EAAC;AAAA,IACvC,WAAW,QAAA,CAAS,mBAAA;AAAA,IACpB,WAAA,EAAa,SAAS,WAAA,GAAc,SAAA,CAAU,gBAAgB,QAAA,CAAS,WAAW,CAAC,CAAA,GAAI,MAAA;AAAA,IACvF,YAAA,EAAc,SAAS,YAAA,GAAe,SAAA,CAAU,gBAAgB,QAAA,CAAS,YAAY,CAAC,CAAA,GAAI;AAAA,GAC5F;AACF;AAMO,IAAM,mBAAN,MAAuB;AAAA,EAC5B,OAAe,sBAAgD,EAAC;AAAA;AAAA;AAAA;AAAA,EAKhE,OAAO,yBAAA,CAA0B,EAAA,EAAY,QAAA,EAA0B;AACrE,IAAA,IAAA,CAAK,mBAAA,CAAoB,EAAE,CAAA,GAAI,QAAA;AAAA,EACjC;AAAA;AAAA;AAAA;AAAA,EAKA,OAAO,0BAAA,CAA2B,SAAA,EAAqC,MAAA,EAAoB;AACzF,IAAA,KAAA,MAAW,CAAC,EAAA,EAAI,QAAQ,KAAK,MAAA,CAAO,OAAA,CAAQ,SAAS,CAAA,EAAG;AAEtD,MAAA,IAAI,MAAA,EAAQ;AACV,QAAA,QAAA,CAAS,iBAAiB,MAAM,CAAA;AAChC,QAAA,QAAA,CAAS,oBAAA,CAAqB;AAAA,UAC5B,MAAA,EAAQ,OAAO,SAAA,EAAU;AAAA,UACzB,OAAA,EAAS,OAAO,UAAA,EAAW;AAAA,UAC3B,MAAA,EAAQ,OAAO,UAAA,EAAW;AAAA,UAC1B,GAAA,EAAK,OAAO,MAAA,EAAO;AAAA,UACnB,OAAA,EAAS,OAAO,UAAA;AAAW,SAC5B,CAAA;AAAA,MACH;AACA,MAAA,IAAA,CAAK,mBAAA,CAAoB,EAAE,CAAA,GAAI,QAAA;AAAA,IACjC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,OAAO,YAAY,UAAA,EAA0C;AAC3D,IAAA,OAAO,IAAA,CAAK,oBAAoB,UAAU,CAAA;AAAA,EAC5C;AAAA;AAAA;AAAA;AAAA,EAKA,OAAO,eAAA,GAA4C;AACjD,IAAA,OAAO,EAAE,GAAG,IAAA,CAAK,mBAAA,EAAoB;AAAA,EACvC;AAAA;AAAA;AAAA;AAAA,EAKA,OAAO,yBAAyB,UAAA,EAA0B;AACxD,IAAA,OAAO,IAAA,CAAK,oBAAoB,UAAU,CAAA;AAAA,EAC5C;AAAA;AAAA;AAAA;AAAA,EAIA,OAAO,OAAA,GAAgB;AAErB,IAAA,IAAA,CAAK,sBAAsB,EAAC;AAAA,EAC9B;AAAA;AAAA;AAAA;AAAA,EAKA,OAAO,uBAAuB,UAAA,EAA6B;AACzD,IAAA,OAAO,cAAc,IAAA,CAAK,mBAAA;AAAA,EAC5B;AAAA;AAAA;AAAA;AAAA,EAKA,OAAO,wBAAA,GAAqC;AAC1C,IAAA,OAAO,MAAA,CAAO,IAAA,CAAK,IAAA,CAAK,mBAAmB,CAAA;AAAA,EAC7C;AACF;AAEO,SAAS,4BAA4B,OAAA,EAAgC;AAC1E,EAAA,IAAI,CAAC,OAAA,EAAS;AACZ,IAAA,OAAO,EAAA;AAAA,EACT;AAEA,EAAA,IAAI,OAAO,YAAY,QAAA,EAAU;AAC/B,IAAA,OAAO,OAAA;AAAA,EACT;AAEA,EAAA,IAAI,KAAA,CAAM,OAAA,CAAQ,OAAO,CAAA,EAAG;AAC1B,IAAA,OAAO,OAAA,CACJ,IAAI,CAAA,CAAA,KAAK;AACR,MAAA,IAAI,OAAO,MAAM,QAAA,EAAU;AACzB,QAAA,OAAO,CAAA;AAAA,MACT;AAEA,MAAA,OAAO,OAAO,CAAA,CAAE,OAAA,KAAY,QAAA,GAAW,EAAE,OAAA,GAAU,EAAA;AAAA,IACrD,CAAC,CAAA,CACA,MAAA,CAAO,aAAW,OAAO,CAAA,CACzB,KAAK,IAAI,CAAA;AAAA,EACd;AAGA,EAAA,OAAO,OAAO,OAAA,CAAQ,OAAA,KAAY,QAAA,GAAW,QAAQ,OAAA,GAAU,EAAA;AACjE","file":"chunk-DNUO33HF.js","sourcesContent":["import type { SystemMessage } from '@mastra/core/llm';\nimport { zodToJsonSchema } from '@mastra/core/utils/zod-to-json';\nimport type { StepWithComponent, Workflow, WorkflowInfo } from '@mastra/core/workflows';\nimport { stringify } from 'superjson';\n\nfunction getSteps(steps: Record<string, StepWithComponent>, path?: string) {\n  return Object.entries(steps).reduce<any>((acc, [key, step]) => {\n    const fullKey = path ? `${path}.${key}` : key;\n    acc[fullKey] = {\n      id: step.id,\n      description: step.description,\n      inputSchema: step.inputSchema ? stringify(zodToJsonSchema(step.inputSchema)) : undefined,\n      outputSchema: step.outputSchema ? stringify(zodToJsonSchema(step.outputSchema)) : undefined,\n      resumeSchema: step.resumeSchema ? stringify(zodToJsonSchema(step.resumeSchema)) : undefined,\n      suspendSchema: step.suspendSchema ? stringify(zodToJsonSchema(step.suspendSchema)) : undefined,\n      isWorkflow: step.component === 'WORKFLOW',\n      component: step.component,\n    };\n\n    if (step.component === 'WORKFLOW' && step.steps) {\n      const nestedSteps = getSteps(step.steps, fullKey) || {};\n      acc = { ...acc, ...nestedSteps };\n    }\n\n    return acc;\n  }, {});\n}\n\nexport function getWorkflowInfo(workflow: Workflow): WorkflowInfo {\n  return {\n    name: workflow.name,\n    description: workflow.description,\n    steps: Object.entries(workflow.steps).reduce<any>((acc, [key, step]) => {\n      acc[key] = {\n        id: step.id,\n        description: step.description,\n        inputSchema: step.inputSchema ? stringify(zodToJsonSchema(step.inputSchema)) : undefined,\n        outputSchema: step.outputSchema ? stringify(zodToJsonSchema(step.outputSchema)) : undefined,\n        resumeSchema: step.resumeSchema ? stringify(zodToJsonSchema(step.resumeSchema)) : undefined,\n        suspendSchema: step.suspendSchema ? stringify(zodToJsonSchema(step.suspendSchema)) : undefined,\n        component: step.component,\n      };\n      return acc;\n    }, {}),\n    allSteps: getSteps(workflow.steps) || {},\n    stepGraph: workflow.serializedStepGraph,\n    inputSchema: workflow.inputSchema ? stringify(zodToJsonSchema(workflow.inputSchema)) : undefined,\n    outputSchema: workflow.outputSchema ? stringify(zodToJsonSchema(workflow.outputSchema)) : undefined,\n  };\n}\n\n/**\n * Workflow Registry for temporarily registering additional workflows\n * that are not part of the user's Mastra instance (e.g., internal template workflows)\n */\nexport class WorkflowRegistry {\n  private static additionalWorkflows: Record<string, Workflow> = {};\n\n  /**\n   * Register a workflow temporarily\n   */\n  static registerTemporaryWorkflow(id: string, workflow: Workflow): void {\n    this.additionalWorkflows[id] = workflow;\n  }\n\n  /**\n   * Register all workflows from map\n   */\n  static registerTemporaryWorkflows(workflows: Record<string, Workflow>, mastra?: any): void {\n    for (const [id, workflow] of Object.entries(workflows)) {\n      // Register Mastra instance with the workflow if provided\n      if (mastra) {\n        workflow.__registerMastra(mastra);\n        workflow.__registerPrimitives({\n          logger: mastra.getLogger(),\n          storage: mastra.getStorage(),\n          agents: mastra.listAgents(),\n          tts: mastra.getTTS(),\n          vectors: mastra.getVectors(),\n        });\n      }\n      this.additionalWorkflows[id] = workflow;\n    }\n  }\n\n  /**\n   * Get a workflow by ID from the registry (returns undefined if not found)\n   */\n  static getWorkflow(workflowId: string): Workflow | undefined {\n    return this.additionalWorkflows[workflowId];\n  }\n\n  /**\n   * Get all workflows from the registry\n   */\n  static getAllWorkflows(): Record<string, Workflow> {\n    return { ...this.additionalWorkflows };\n  }\n\n  /**\n   * Clean up a temporary workflow\n   */\n  static cleanupTemporaryWorkflow(workflowId: string): void {\n    delete this.additionalWorkflows[workflowId];\n  }\n  /**\n   * Clean up all registered workflows\n   */\n  static cleanup(): void {\n    // Clear all workflows (since we register all agent-builder workflows each time)\n    this.additionalWorkflows = {};\n  }\n\n  /**\n   * Check if a workflow ID is a valid agent-builder workflow\n   */\n  static isAgentBuilderWorkflow(workflowId: string): boolean {\n    return workflowId in this.additionalWorkflows;\n  }\n\n  /**\n   * Get all registered temporary workflow IDs (for debugging)\n   */\n  static getRegisteredWorkflowIds(): string[] {\n    return Object.keys(this.additionalWorkflows);\n  }\n}\n\nexport function convertInstructionsToString(message: SystemMessage): string {\n  if (!message) {\n    return '';\n  }\n\n  if (typeof message === 'string') {\n    return message;\n  }\n\n  if (Array.isArray(message)) {\n    return message\n      .map(m => {\n        if (typeof m === 'string') {\n          return m;\n        }\n        // Safely extract content from message objects\n        return typeof m.content === 'string' ? m.content : '';\n      })\n      .filter(content => content) // Remove empty strings\n      .join('\\n');\n  }\n\n  // Handle single message object - safely extract content\n  return typeof message.content === 'string' ? message.content : '';\n}\n"]}

package/dist/{chunk-CXVOX4WB.js → chunk-FKZVCLOY.js}

@@ -1,6 +1,6 @@
-import { validateBody } from './chunk-4QCXUEAT.js';
-import { handleError } from './chunk-CY4TP3FK.js';
-import { HTTPException } from './chunk-MMROOK5J.js';
+import { validateBody } from './chunk-SZIFSF4T.js';
+import { handleError } from './chunk-UXGQZUYZ.js';
+import { HTTPException } from './chunk-6QWQZI4Q.js';
 import { __export } from './chunk-PR4QN5HX.js';
 import { convertMessages } from '@mastra/core/agent';
 import { RequestContext } from '@mastra/core/di';
@@ -15,9 +15,9 @@ __export(memory_exports, {
   getMemoryConfigHandler: () => getMemoryConfigHandler,
   getMemoryStatusHandler: () => getMemoryStatusHandler,
   getMessagesHandler: () => getMessagesHandler,
-  getMessagesPaginatedHandler: () => getMessagesPaginatedHandler,
   getThreadByIdHandler: () => getThreadByIdHandler,
   getWorkingMemoryHandler: () => getWorkingMemoryHandler,
+  listMessagesHandler: () => listMessagesHandler,
   listThreadsHandler: () => listThreadsHandler,
   saveMessagesHandler: () => saveMessagesHandler,
   searchMemoryHandler: () => searchMemoryHandler,
@@ -61,9 +61,8 @@ async function getMemoryFromContext({
   if (agent) {
     return await agent?.getMemory({
       requestContext: requestContext ?? new RequestContext()
-    }) || mastra.getMemory();
+    });
   }
-  return mastra.getMemory();
 }
 async function getMemoryStatusHandler({
   mastra,
@@ -101,10 +100,9 @@ async function listThreadsHandler({
   agentId,
   resourceId,
   requestContext,
-  offset,
-  limit,
-  orderBy,
-  sortDirection
+  page,
+  perPage,
+  orderBy
 }) {
   try {
     const memory = await getMemoryFromContext({ mastra, agentId, requestContext });
@@ -114,10 +112,9 @@ async function listThreadsHandler({
     validateBody({ resourceId });
     const result = await memory.listThreadsByResourceId({
       resourceId,
-      offset,
-      limit,
-      orderBy,
-      sortDirection
+      page,
+      perPage,
+      orderBy
     });
     return result;
   } catch (error) {
@@ -260,12 +257,15 @@ async function deleteThreadHandler({
     return handleError(error, "Error deleting thread");
   }
 }
-async function getMessagesPaginatedHandler({
+async function listMessagesHandler({
   mastra,
   threadId,
   resourceId,
-  selectBy,
-  format
+  perPage,
+  page,
+  orderBy,
+  include,
+  filter
 }) {
   try {
     validateBody({ threadId });
@@ -277,7 +277,15 @@ async function getMessagesPaginatedHandler({
     if (!thread) {
       throw new HTTPException(404, { message: "Thread not found" });
     }
-    const result = await storage.getMessagesPaginated({ threadId, resourceId, selectBy, format });
+    const result = await storage.listMessages({
+      threadId,
+      resourceId,
+      perPage,
+      page,
+      orderBy,
+      include,
+      filter
+    });
     return result;
   } catch (error) {
     return handleError(error, "Error getting messages");
@@ -299,6 +307,9 @@ async function getMessagesHandler({
     if (!memory) {
       throw new HTTPException(400, { message: "Memory is not initialized" });
     }
+    if (!threadId) {
+      throw new HTTPException(400, { message: "No threadId found" });
+    }
     const thread = await memory.getThreadById({ threadId });
     if (!thread) {
       throw new HTTPException(404, { message: "Thread not found" });
@@ -307,8 +318,8 @@ async function getMessagesHandler({
       threadId,
       ...limit && { selectBy: { last: limit } }
     });
-    const uiMessages = convertMessages(result.messagesV2).to("AIV5.UI");
-    return { messages: result.messages, uiMessages, legacyMessages: result.uiMessages };
+    const uiMessages = convertMessages(result.messages).to("AIV5.UI");
+    return { messages: result.messages, uiMessages };
   } catch (error) {
     return handleError(error, "Error getting messages");
   }
@@ -429,8 +440,13 @@ async function searchMemoryHandler({
       }
     }
     if (!threadId) {
-      const threads2 = await memory.getThreadsByResourceId({ resourceId });
-      if (threads2.length === 0) {
+      const { threads } = await memory.listThreadsByResourceId({
+        resourceId,
+        page: 0,
+        perPage: 1,
+        orderBy: { field: "updatedAt", direction: "DESC" }
+      });
+      if (threads.length === 0) {
         return {
           results: [],
           count: 0,
@@ -439,7 +455,7 @@ async function searchMemoryHandler({
           searchType: hasSemanticRecall ? "semantic" : "text"
         };
       }
-      threadId = threads2[0].id;
+      threadId = threads[0].id;
     }
     const beforeRange = typeof config.semanticRecall === `boolean` ? 2 : typeof config.semanticRecall?.messageRange === `number` ? config.semanticRecall.messageRange : config.semanticRecall?.messageRange.before || 2;
     const afterRange = typeof config.semanticRecall === `boolean` ? 2 : typeof config.semanticRecall?.messageRange === `number` ? config.semanticRecall.messageRange : config.semanticRecall?.messageRange.after || 2;
@@ -456,13 +472,16 @@ async function searchMemoryHandler({
       vectorMessageSearch: searchQuery,
       config
     });
-    const threads = await memory.getThreadsByResourceId({ resourceId });
-    const threadMap = new Map(threads.map((t) => [t.id, t]));
-    for (const msg of result.messagesV2) {
+    const threadIds = Array.from(
+      new Set(result.messages.map((m) => m.threadId || threadId).filter(Boolean))
+    );
+    const fetched = await Promise.all(threadIds.map((id) => memory.getThreadById({ threadId: id })));
+    const threadMap = new Map(fetched.filter(Boolean).map((t) => [t.id, t]));
+    for (const msg of result.messages) {
       const content = typeof msg.content.content === `string` ? msg.content.content : msg.content.parts?.map((p) => p.type === "text" ? p.text : "").join(" ") || "";
       const msgThreadId = msg.threadId || threadId;
       const thread = threadMap.get(msgThreadId);
-      const threadMessages = (await memory.query({ threadId: msgThreadId })).uiMessages;
+      const threadMessages = (await memory.query({ threadId: msgThreadId })).messages;
       const messageIndex = threadMessages.findIndex((m) => m.id === msg.id);
       const searchResult = {
         id: msg.id,
@@ -503,6 +522,6 @@ async function searchMemoryHandler({
   }
 }
 
-export { createThreadHandler, deleteMessagesHandler, deleteThreadHandler, getMemoryConfigHandler, getMemoryStatusHandler, getMessagesHandler, getMessagesPaginatedHandler, getThreadByIdHandler, getWorkingMemoryHandler, listThreadsHandler, memory_exports, saveMessagesHandler, searchMemoryHandler, updateThreadHandler, updateWorkingMemoryHandler };
-//# sourceMappingURL=chunk-CXVOX4WB.js.map
-//# sourceMappingURL=chunk-CXVOX4WB.js.map
+export { createThreadHandler, deleteMessagesHandler, deleteThreadHandler, getMemoryConfigHandler, getMemoryStatusHandler, getMessagesHandler, getThreadByIdHandler, getWorkingMemoryHandler, listMessagesHandler, listThreadsHandler, memory_exports, saveMessagesHandler, searchMemoryHandler, updateThreadHandler, updateWorkingMemoryHandler };
+//# sourceMappingURL=chunk-FKZVCLOY.js.map
+//# sourceMappingURL=chunk-FKZVCLOY.js.map

package/dist/chunk-FKZVCLOY.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/server/handlers/memory.ts"],"names":["agents"],"mappings":";;;;;;;;;AAAA,IAAA,cAAA,GAAA;AAAA,QAAA,CAAA,cAAA,EAAA;AAAA,EAAA,mBAAA,EAAA,MAAA,mBAAA;AAAA,EAAA,qBAAA,EAAA,MAAA,qBAAA;AAAA,EAAA,mBAAA,EAAA,MAAA,mBAAA;AAAA,EAAA,sBAAA,EAAA,MAAA,sBAAA;AAAA,EAAA,sBAAA,EAAA,MAAA,sBAAA;AAAA,EAAA,kBAAA,EAAA,MAAA,kBAAA;AAAA,EAAA,oBAAA,EAAA,MAAA,oBAAA;AAAA,EAAA,uBAAA,EAAA,MAAA,uBAAA;AAAA,EAAA,mBAAA,EAAA,MAAA,mBAAA;AAAA,EAAA,kBAAA,EAAA,MAAA,kBAAA;AAAA,EAAA,mBAAA,EAAA,MAAA,mBAAA;AAAA,EAAA,mBAAA,EAAA,MAAA,mBAAA;AAAA,EAAA,mBAAA,EAAA,MAAA,mBAAA;AAAA,EAAA,0BAAA,EAAA,MAAA;AAAA,CAAA,CAAA;AAmBA,eAAe,oBAAA,CAAqB;AAAA,EAClC,MAAA;AAAA,EACA,OAAA;AAAA,EACA;AACF,CAAA,EAA2G;AACzG,EAAA,MAAM,MAAA,GAAS,OAAO,SAAA,EAAU;AAChC,EAAA,IAAI,KAAA;AACJ,EAAA,IAAI,OAAA,EAAS;AACX,IAAA,IAAI;AACF,MAAA,KAAA,GAAQ,MAAA,CAAO,SAAS,OAAO,CAAA;AAAA,IACjC,SAAS,KAAA,EAAO;AACd,MAAA,MAAA,CAAO,KAAA,CAAM,+DAA+D,KAAK,CAAA;AAAA,IACnF;AAAA,EACF;AACA,EAAA,IAAI,OAAA,IAAW,CAAC,KAAA,EAAO;AACrB,IAAA,MAAA,CAAO,KAAA,CAAM,6CAAA,EAA+C,EAAE,OAAA,EAAS,CAAA;AACvE,IAAA,MAAM,MAAA,GAAS,OAAO,UAAA,EAAW;AACjC,IAAA,IAAI,OAAO,IAAA,CAAK,MAAA,IAAU,EAAE,EAAE,MAAA,EAAQ;AACpC,MAAA,KAAA,MAAW,CAAC,CAAA,EAAG,EAAE,KAAK,MAAA,CAAO,OAAA,CAAQ,MAAM,CAAA,EAAG;AAC5C,QAAA,IAAI;AACF,UAAA,MAAMA,OAAAA,GAAS,MAAM,EAAA,CAAG,UAAA,EAAW;AAEnC,UAAA,IAAIA,OAAAA,CAAO,OAAO,CAAA,EAAG;AACnB,YAAA,KAAA,GAAQA,QAAO,OAAO,CAAA;AACtB,YAAA;AAAA,UACF;AAAA,QACF,SAAS,KAAA,EAAO;AACd,UAAA,MAAA,CAAO,KAAA,CAAM,kCAAkC,KAAK,CAAA;AAAA,QACtD;AAAA,MACF;AAAA,IACF;AAEA,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,mBAAmB,CAAA;AAAA,IAC7D;AAAA,EACF;AAEA,EAAA,IAAI,KAAA,EAAO;AACT,IAAA,OAAO,MAAM,OAAO,SAAA,CAAU;AAAA,MAC5B,cAAA,EAAgB,cAAA,IAAkB,IAAI,cAAA;AAAe,KACtD,CAAA;AAAA,EACH;AACF;AAGA,eAAsB,sBAAA,CAAuB;AAAA,EAC3C,MAAA;AAAA,EACA,OAAA;AAAA,EACA;AACF,CAAA,EAAiE;AAC/D,EAAA,IAAI;AACF,IAAA,MAAM,SAAS,MAAM,oBAAA,CAAqB,EAAE,MAAA,EAAQ,OAAA,EAAS,gBAAgB,CAAA;AAE7E,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,OAAO,EAAE,QAAQ,KAAA,EAAM;AAAA,IACzB;AAEA,IAAA,OAAO,EAAE,QAAQ,IAAA,EAAK;AAAA,EACxB,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,6BAA6B,CAAA;AAAA,EACzD;AACF;AAEA,eAAsB,sBAAA,CAAuB;AAAA,EAC3C,MAAA;AAAA,EACA,OAAA;AAAA,EACA;AACF,CAAA,EAAiE;AAC/D,EAAA,IAAI;AACF,IAAA,MAAM,SAAS,MAAM,oBAAA,CAAqB,EAAE,MAAA,EAAQ,OAAA,EAAS,gBAAgB,CAAA;AAE7E,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,6BAA6B,CAAA;AAAA,IACvE;AAGA,IAAA,MAAM,MAAA,GAAS,MAAA,CAAO,qBAAA,CAAsB,EAAE,CAAA;AAE9C,IAAA,OAAO,EAAE,MAAA,EAAO;AAAA,EAClB,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,oCAAoC,CAAA;AAAA,EAChE;AACF;AAEA,eAAsB,kBAAA,CAAmB;AAAA,EACvC,MAAA;AAAA,EACA,OAAA;AAAA,EACA,UAAA;AAAA,EACA,cAAA;AAAA,EACA,IAAA;AAAA,EACA,OAAA;AAAA,EACA;AACF,CAAA,EAIG;AACD,EAAA,IAAI;AACF,IAAA,MAAM,SAAS,MAAM,oBAAA,CAAqB,EAAE,MAAA,EAAQ,OAAA,EAAS,gBAAgB,CAAA;AAE7E,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,6BAA6B,CAAA;AAAA,IACvE;AAEA,IAAA,YAAA,CAAa,EAAE,YAAY,CAAA;AAE3B,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,uBAAA,CAAwB;AAAA,MAClD,UAAA;AAAA,MACA,IAAA;AAAA,MACA,OAAA;AAAA,MACA;AAAA,KACD,CAAA;AACD,IAAA,OAAO,MAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,uBAAuB,CAAA;AAAA,EACnD;AACF;AAEA,eAAsB,oBAAA,CAAqB;AAAA,EACzC,MAAA;AAAA,EACA,OAAA;AAAA,EACA,QAAA;AAAA,EACA;AACF,CAAA,EAA8E;AAC5E,EAAA,IAAI;AACF,IAAA,YAAA,CAAa,EAAE,UAAU,CAAA;AAEzB,IAAA,MAAM,SAAS,MAAM,oBAAA,CAAqB,EAAE,MAAA,EAAQ,OAAA,EAAS,gBAAgB,CAAA;AAC7E,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,6BAA6B,CAAA;AAAA,IACvE;AAEA,IAAA,MAAM,SAAS,MAAM,MAAA,CAAO,aAAA,CAAc,EAAE,UAAqB,CAAA;AACjE,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,oBAAoB,CAAA;AAAA,IAC9D;AAEA,IAAA,OAAO,MAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,sBAAsB,CAAA;AAAA,EAClD;AACF;AAEA,eAAsB,mBAAA,CAAoB;AAAA,EACxC,MAAA;AAAA,EACA,OAAA;AAAA,EACA,IAAA;AAAA,EACA;AACF,CAAA,EAIG;AACD,EAAA,IAAI;AACF,IAAA,MAAM,SAAS,MAAM,oBAAA,CAAqB,EAAE,MAAA,EAAQ,OAAA,EAAS,gBAAgB,CAAA;AAE7E,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,6BAA6B,CAAA;AAAA,IACvE;AAEA,IAAA,IAAI,CAAC,MAAM,QAAA,EAAU;AACnB,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,yBAAyB,CAAA;AAAA,IACnE;AAEA,IAAA,IAAI,CAAC,KAAA,CAAM,OAAA,CAAQ,IAAA,CAAK,QAAQ,CAAA,EAAG;AACjC,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,+BAA+B,CAAA;AAAA,IACzE;AAGA,IAAA,MAAM,eAAA,GAAkB,IAAA,CAAK,QAAA,CAAS,MAAA,CAAO,CAAA,OAAA,KAAW,CAAC,OAAA,CAAQ,QAAA,IAAY,CAAC,OAAA,CAAQ,UAAU,CAAA;AAChG,IAAA,IAAI,eAAA,CAAgB,SAAS,CAAA,EAAG;AAC9B,MAAA,MAAM,IAAI,cAAc,GAAA,EAAK;AAAA,QAC3B,OAAA,EAAS,CAAA,6DAAA,EAAgE,eAAA,CAAgB,MAAM,CAAA,oBAAA;AAAA,OAChG,CAAA;AAAA,IACH;AAEA,IAAA,MAAM,iBAAA,GAAoB,IAAA,CAAK,QAAA,CAAS,GAAA,CAAI,CAAA,OAAA,MAAY;AAAA,MACtD,GAAG,OAAA;AAAA,MACH,EAAA,EAAI,OAAA,CAAQ,EAAA,IAAM,MAAA,CAAO,UAAA,EAAW;AAAA,MACpC,SAAA,EAAW,QAAQ,SAAA,GAAY,IAAI,KAAK,OAAA,CAAQ,SAAS,CAAA,mBAAI,IAAI,IAAA;AAAK,KACxE,CAAE,CAAA;AAEF,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,YAAA,CAAa,EAAE,UAAU,iBAAA,EAAmB,YAAA,EAAc,EAAC,EAAG,CAAA;AAC1F,IAAA,OAAO,MAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,uBAAuB,CAAA;AAAA,EACnD;AACF;AAEA,eAAsB,mBAAA,CAAoB;AAAA,EACxC,MAAA;AAAA,EACA,OAAA;AAAA,EACA,IAAA;AAAA,EACA;AACF,CAAA,EAEG;AACD,EAAA,IAAI;AACF,IAAA,MAAM,SAAS,MAAM,oBAAA,CAAqB,EAAE,MAAA,EAAQ,OAAA,EAAS,gBAAgB,CAAA;AAE7E,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,6BAA6B,CAAA;AAAA,IACvE;AAEA,IAAA,YAAA,CAAa,EAAE,UAAA,EAAY,IAAA,EAAM,UAAA,EAAY,CAAA;AAE7C,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,YAAA,CAAa;AAAA,MACvC,YAAY,IAAA,EAAM,UAAA;AAAA,MAClB,OAAO,IAAA,EAAM,KAAA;AAAA,MACb,UAAU,IAAA,EAAM,QAAA;AAAA,MAChB,UAAU,IAAA,EAAM;AAAA,KACjB,CAAA;AACD,IAAA,OAAO,MAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,+BAA+B,CAAA;AAAA,EAC3D;AACF;AAEA,eAAsB,mBAAA,CAAoB;AAAA,EACxC,MAAA;AAAA,EACA,OAAA;AAAA,EACA,QAAA;AAAA,EACA,IAAA;AAAA,EACA;AACF,CAAA,EAEG;AACD,EAAA,IAAI;AACF,IAAA,MAAM,SAAS,MAAM,oBAAA,CAAqB,EAAE,MAAA,EAAQ,OAAA,EAAS,gBAAgB,CAAA;AAE7E,IAAA,IAAI,CAAC,IAAA,EAAM;AACT,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,oBAAoB,CAAA;AAAA,IAC9D;AAEA,IAAA,MAAM,EAAE,KAAA,EAAO,QAAA,EAAU,UAAA,EAAW,GAAI,IAAA;AACxC,IAAA,MAAM,SAAA,uBAAgB,IAAA,EAAK;AAE3B,IAAA,YAAA,CAAa,EAAE,UAAU,CAAA;AAEzB,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,6BAA6B,CAAA;AAAA,IACvE;AAEA,IAAA,MAAM,SAAS,MAAM,MAAA,CAAO,aAAA,CAAc,EAAE,UAAqB,CAAA;AACjE,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,oBAAoB,CAAA;AAAA,IAC9D;AAEA,IAAA,MAAM,aAAA,GAAgB;AAAA,MACpB,GAAG,MAAA;AAAA,MACH,KAAA,EAAO,SAAS,MAAA,CAAO,KAAA;AAAA,MACvB,QAAA,EAAU,YAAY,MAAA,CAAO,QAAA;AAAA,MAC7B,UAAA,EAAY,cAAc,MAAA,CAAO,UAAA;AAAA,MACjC,WAAW,MAAA,CAAO,SAAA;AAAA,MAClB;AAAA,KACF;AAEA,IAAA,MAAM,SAAS,MAAM,MAAA,CAAO,WAAW,EAAE,MAAA,EAAQ,eAAe,CAAA;AAChE,IAAA,OAAO,MAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,uBAAuB,CAAA;AAAA,EACnD;AACF;AAEA,eAAsB,mBAAA,CAAoB;AAAA,EACxC,MAAA;AAAA,EACA,OAAA;AAAA,EACA,QAAA;AAAA,EACA;AACF,CAAA,EAA8E;AAC5E,EAAA,IAAI;AACF,IAAA,YAAA,CAAa,EAAE,UAAU,CAAA;AAEzB,IAAA,MAAM,SAAS,MAAM,oBAAA,CAAqB,EAAE,MAAA,EAAQ,OAAA,EAAS,gBAAgB,CAAA;AAC7E,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,6BAA6B,CAAA;AAAA,IACvE;AAEA,IAAA,MAAM,SAAS,MAAM,MAAA,CAAO,aAAA,CAAc,EAAE,UAAqB,CAAA;AACjE,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,oBAAoB,CAAA;AAAA,IAC9D;AAEA,IAAA,MAAM,MAAA,CAAO,aAAa,QAAS,CAAA;AACnC,IAAA,OAAO,EAAE,QAAQ,gBAAA,EAAiB;AAAA,EACpC,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,uBAAuB,CAAA;AAAA,EACnD;AACF;AAEA,eAAsB,mBAAA,CAAoB;AAAA,EACxC,MAAA;AAAA,EACA,QAAA;AAAA,EACA,UAAA;AAAA,EACA,OAAA;AAAA,EACA,IAAA;AAAA,EACA,OAAA;AAAA,EACA,OAAA;AAAA,EACA;AACF,CAAA,EAA6D;AAC3D,EAAA,IAAI;AACF,IAAA,YAAA,CAAa,EAAE,UAAU,CAAA;AAEzB,IAAA,MAAM,OAAA,GAAU,OAAO,UAAA,EAAW;AAElC,IAAA,IAAI,CAAC,OAAA,EAAS;AACZ,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,8BAA8B,CAAA;AAAA,IACxE;AAEA,IAAA,MAAM,SAAS,MAAM,OAAA,CAAQ,aAAA,CAAc,EAAE,UAAqB,CAAA;AAElE,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,oBAAoB,CAAA;AAAA,IAC9D;AAEA,IAAA,MAAM,MAAA,GAAS,MAAM,OAAA,CAAQ,YAAA,CAAa;AAAA,MACxC,QAAA;AAAA,MACA,UAAA;AAAA,MACA,OAAA;AAAA,MACA,IAAA;AAAA,MACA,OAAA;AAAA,MACA,OAAA;AAAA,MACA;AAAA,KACD,CAAA;AACD,IAAA,OAAO,MAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,wBAAwB,CAAA;AAAA,EACpD;AACF;AAEA,eAAsB,kBAAA,CAAmB;AAAA,EACvC,MAAA;AAAA,EACA,OAAA;AAAA,EACA,QAAA;AAAA,EACA,KAAA;AAAA,EACA;AACF,CAAA,EAEG;AACD,EAAA,IAAI,KAAA,KAAU,WAAc,CAAC,MAAA,CAAO,UAAU,KAAK,CAAA,IAAK,SAAS,CAAA,CAAA,EAAI;AACnE,IAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,6CAA6C,CAAA;AAAA,EACvF;AACA,EAAA,IAAI;AACF,IAAA,YAAA,CAAa,EAAE,UAAU,CAAA;AAEzB,IAAA,MAAM,SAAS,MAAM,oBAAA,CAAqB,EAAE,MAAA,EAAQ,OAAA,EAAS,gBAAgB,CAAA;AAE7E,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,6BAA6B,CAAA;AAAA,IACvE;AAEA,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,qBAAqB,CAAA;AAAA,IAC/D;AAEA,IAAA,MAAM,SAAS,MAAM,MAAA,CAAO,aAAA,CAAc,EAAE,UAAoB,CAAA;AAChE,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,oBAAoB,CAAA;AAAA,IAC9D;AAEA,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,KAAA,CAAM;AAAA,MAChC,QAAA;AAAA,MACA,GAAI,KAAA,IAAS,EAAE,UAAU,EAAE,IAAA,EAAM,OAAM;AAAE,KAC1C,CAAA;AACD,IAAA,MAAM,aAAa,eAAA,CAAgB,MAAA,CAAO,QAAQ,CAAA,CAAE,GAAG,SAAS,CAAA;AAChE,IAAA,OAAO,EAAE,QAAA,EAAU,MAAA,CAAO,QAAA,EAAU,UAAA,EAAW;AAAA,EACjD,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,wBAAwB,CAAA;AAAA,EACpD;AACF;AAOA,eAAsB,uBAAA,CAAwB;AAAA,EAC5C,MAAA;AAAA,EACA,OAAA;AAAA,EACA,QAAA;AAAA,EACA,UAAA;AAAA,EACA,cAAA;AAAA,EACA;AACF,CAAA,EAGG;AACD,EAAA,IAAI;AACF,IAAA,MAAM,SAAS,MAAM,oBAAA,CAAqB,EAAE,MAAA,EAAQ,OAAA,EAAS,gBAAgB,CAAA;AAC7E,IAAA,YAAA,CAAa,EAAE,UAAU,CAAA;AACzB,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,6BAA6B,CAAA;AAAA,IACvE;AACA,IAAA,MAAM,SAAS,MAAM,MAAA,CAAO,aAAA,CAAc,EAAE,UAAqB,CAAA;AACjE,IAAA,MAAM,YAAA,GAAe,CAAC,CAAC,MAAA;AACvB,IAAA,MAAM,WAAW,MAAM,MAAA,CAAO,wBAAA,CAAyB,EAAE,cAAc,CAAA;AACvE,IAAA,MAAM,qBAAA,GACJ,QAAA,EAAU,MAAA,KAAW,MAAA,GACjB,EAAE,GAAG,QAAA,EAAU,OAAA,EAAS,IAAA,CAAK,UAAU,uBAAA,CAAwB,QAAA,CAAS,OAAO,CAAC,GAAE,GAClF,QAAA;AACN,IAAA,MAAM,aAAA,GAAgB,MAAM,MAAA,CAAO,gBAAA,CAAiB,EAAE,QAAA,EAAqB,UAAA,EAAY,cAAc,CAAA;AACrG,IAAA,MAAM,MAAA,GAAS,MAAA,CAAO,qBAAA,CAAsB,YAAA,IAAgB,EAAE,CAAA;AAC9D,IAAA,MAAM,SAAS,MAAA,CAAO,aAAA,EAAe,KAAA,KAAU,QAAA,IAAY,aAAa,UAAA,GAAa,QAAA;AACrF,IAAA,OAAO,EAAE,aAAA,EAAe,MAAA,EAAQ,qBAAA,EAAuB,YAAA,EAAa;AAAA,EACtE,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,8BAA8B,CAAA;AAAA,EAC1D;AACF;AAOA,eAAsB,0BAAA,CAA2B;AAAA,EAC/C,MAAA;AAAA,EACA,OAAA;AAAA,EACA,QAAA;AAAA,EACA,IAAA;AAAA,EACA;AACF,CAAA,EAEG;AACD,EAAA,IAAI;AACF,IAAA,YAAA,CAAa,EAAE,UAAU,CAAA;AACzB,IAAA,MAAM,SAAS,MAAM,oBAAA,CAAqB,EAAE,MAAA,EAAQ,OAAA,EAAS,gBAAgB,CAAA;AAC7E,IAAA,MAAM,EAAE,UAAA,EAAY,YAAA,EAAc,aAAA,EAAc,GAAI,IAAA;AACpD,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,6BAA6B,CAAA;AAAA,IACvE;AACA,IAAA,MAAM,SAAS,MAAM,MAAA,CAAO,aAAA,CAAc,EAAE,UAAqB,CAAA;AACjE,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,oBAAoB,CAAA;AAAA,IAC9D;AAEA,IAAA,MAAM,OAAO,mBAAA,CAAoB,EAAE,UAAqB,UAAA,EAAY,aAAA,EAAe,cAAc,CAAA;AACjG,IAAA,OAAO,EAAE,SAAS,IAAA,EAAK;AAAA,EACzB,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,+BAA+B,CAAA;AAAA,EAC3D;AACF;AA4BA,eAAsB,qBAAA,CAAsB;AAAA,EAC1C,MAAA;AAAA,EACA,OAAA;AAAA,EACA,UAAA;AAAA,EACA;AACF,CAAA,EAEG;AACD,EAAA,IAAI;AACF,IAAA,IAAI,UAAA,KAAe,MAAA,IAAa,UAAA,KAAe,IAAA,EAAM;AACnD,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,0BAA0B,CAAA;AAAA,IACpE;AAEA,IAAA,MAAM,SAAS,MAAM,oBAAA,CAAqB,EAAE,MAAA,EAAQ,OAAA,EAAS,gBAAgB,CAAA;AAC7E,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,6BAA6B,CAAA;AAAA,IACvE;AAGA,IAAA,MAAM,MAAA,CAAO,eAAe,UAAiB,CAAA;AAG7C,IAAA,IAAI,KAAA,GAAQ,CAAA;AACZ,IAAA,IAAI,KAAA,CAAM,OAAA,CAAQ,UAAU,CAAA,EAAG;AAC7B,MAAA,KAAA,GAAQ,UAAA,CAAW,MAAA;AAAA,IACrB;AAEA,IAAA,OAAO,EAAE,OAAA,EAAS,IAAA,EAAM,OAAA,EAAS,CAAA,EAAG,KAAK,CAAA,QAAA,EAAW,KAAA,KAAU,CAAA,GAAI,EAAA,GAAK,GAAG,CAAA,qBAAA,CAAA,EAAwB;AAAA,EACpG,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,yBAAyB,CAAA;AAAA,EACrD;AACF;AAEA,eAAsB,mBAAA,CAAoB;AAAA,EACxC,MAAA;AAAA,EACA,OAAA;AAAA,EACA,WAAA;AAAA,EACA,UAAA;AAAA,EACA,QAAA;AAAA,EACA,KAAA,GAAQ,EAAA;AAAA,EACR,cAAA;AAAA,EACA;AACF,CAAA,EAM6D;AAC3D,EAAA,IAAI;AACF,IAAA,YAAA,CAAa,EAAE,WAAA,EAAa,UAAA,EAAY,CAAA;AAExC,IAAA,MAAM,SAAS,MAAM,oBAAA,CAAqB,EAAE,MAAA,EAAQ,OAAA,EAAS,gBAAgB,CAAA;AAC7E,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,6BAA6B,CAAA;AAAA,IACvE;AAGA,IAAA,MAAM,MAAA,GAAS,MAAA,CAAO,qBAAA,CAAsB,YAAA,IAAgB,EAAE,CAAA;AAC9D,IAAA,MAAM,iBAAA,GAAoB,CAAC,CAAC,MAAA,EAAQ,cAAA;AACpC,IAAA,MAAM,aAAA,GACJ,OAAO,MAAA,EAAQ,cAAA,KAAmB,WAAW,MAAA,EAAQ,cAAA,EAAgB,UAAU,QAAA,GAAW,IAAA;AAG5F,IAAA,IAAI,QAAA,IAAY,CAAC,aAAA,EAAe;AAC9B,MAAA,MAAM,SAAS,MAAM,MAAA,CAAO,aAAA,CAAc,EAAE,UAAU,CAAA;AACtD,MAAA,IAAI,CAAC,MAAA,EAAQ;AACX,QAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,oBAAoB,CAAA;AAAA,MAC9D;AACA,MAAA,IAAI,MAAA,CAAO,eAAe,UAAA,EAAY;AACpC,QAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,oDAAoD,CAAA;AAAA,MAC9F;AAAA,IACF;AAEA,IAAA,MAAM,gBAAgC,EAAC;AAGvC,IAAA,IAAI,QAAA,IAAY,CAAC,aAAA,EAAe;AAC9B,MAAA,MAAM,SAAS,MAAM,MAAA,CAAO,aAAA,CAAc,EAAE,UAAU,CAAA;AACtD,MAAA,IAAI,CAAC,MAAA,EAAQ;AAEX,QAAA,OAAO;AAAA,UACL,SAAS,EAAC;AAAA,UACV,KAAA,EAAO,CAAA;AAAA,UACP,KAAA,EAAO,WAAA;AAAA,UACP,WAAA,EAAa,gBAAgB,UAAA,GAAa,QAAA;AAAA,UAC1C,UAAA,EAAY,oBAAoB,UAAA,GAAa;AAAA,SAC/C;AAAA,MACF;AAAA,IACF;AAGA,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,MAAM,EAAE,OAAA,EAAQ,GAAI,MAAM,OAAO,uBAAA,CAAwB;AAAA,QACvD,UAAA;AAAA,QACA,IAAA,EAAM,CAAA;AAAA,QACN,OAAA,EAAS,CAAA;AAAA,QACT,OAAA,EAAS,EAAE,KAAA,EAAO,WAAA,EAAa,WAAW,MAAA;AAAO,OAClD,CAAA;AAED,MAAA,IAAI,OAAA,CAAQ,WAAW,CAAA,EAAG;AACxB,QAAA,OAAO;AAAA,UACL,SAAS,EAAC;AAAA,UACV,KAAA,EAAO,CAAA;AAAA,UACP,KAAA,EAAO,WAAA;AAAA,UACP,WAAA,EAAa,gBAAgB,UAAA,GAAa,QAAA;AAAA,UAC1C,UAAA,EAAY,oBAAoB,UAAA,GAAa;AAAA,SAC/C;AAAA,MACF;AAGA,MAAA,QAAA,GAAW,OAAA,CAAQ,CAAC,CAAA,CAAG,EAAA;AAAA,IACzB;AAEA,IAAA,MAAM,cACJ,OAAO,MAAA,CAAO,cAAA,KAAmB,CAAA,OAAA,CAAA,GAC7B,IACA,OAAO,MAAA,CAAO,cAAA,EAAgB,YAAA,KAAiB,WAC7C,MAAA,CAAO,cAAA,CAAe,eACtB,MAAA,CAAO,cAAA,EAAgB,aAAa,MAAA,IAAU,CAAA;AACtD,IAAA,MAAM,aACJ,OAAO,MAAA,CAAO,cAAA,KAAmB,CAAA,OAAA,CAAA,GAC7B,IACA,OAAO,MAAA,CAAO,cAAA,EAAgB,YAAA,KAAiB,WAC7C,MAAA,CAAO,cAAA,CAAe,eACtB,MAAA,CAAO,cAAA,EAAgB,aAAa,KAAA,IAAS,CAAA;AAErD,IAAA,IAAI,aAAA,IAAiB,OAAO,cAAA,EAAgB;AAC1C,MAAA,MAAA,CAAO,cAAA,GACL,OAAO,MAAA,CAAO,cAAA,KAAmB,CAAA,OAAA,CAAA;AAAA;AAAA;AAAA,QAG7B,EAAE,YAAA,EAAc,CAAA,EAAG,IAAA,EAAM,CAAA,EAAG,OAAO,UAAA;AAAW,UAC9C,EAAE,GAAG,MAAA,CAAO,cAAA,EAAgB,cAAc,CAAA,EAAE;AAAA,IACpD;AAIA,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,gBAAA,CAAiB;AAAA,MAC3C,QAAA;AAAA,MACA,UAAA;AAAA,MACA,mBAAA,EAAqB,WAAA;AAAA,MACrB;AAAA,KACD,CAAA;AAID,IAAA,MAAM,YAAY,KAAA,CAAM,IAAA;AAAA,MACtB,IAAI,GAAA,CAAI,MAAA,CAAO,QAAA,CAAS,GAAA,CAAI,CAAC,CAAA,KAAuB,CAAA,CAAE,QAAA,IAAY,QAAS,CAAA,CAAE,MAAA,CAAO,OAAO,CAAC;AAAA,KAC9F;AACA,IAAA,MAAM,OAAA,GAAU,MAAM,OAAA,CAAQ,GAAA,CAAI,UAAU,GAAA,CAAI,CAAC,EAAA,KAAe,MAAA,CAAO,cAAc,EAAE,QAAA,EAAU,EAAA,EAAI,CAAC,CAAC,CAAA;AACvG,IAAA,MAAM,SAAA,GAAY,IAAI,GAAA,CAAI,OAAA,CAAQ,OAAO,OAAO,CAAA,CAAE,GAAA,CAAI,CAAA,CAAA,KAAK,CAAC,CAAA,CAAG,EAAA,EAAI,CAAE,CAAC,CAAC,CAAA;AAGvE,IAAA,KAAA,MAAW,GAAA,IAAO,OAAO,QAAA,EAAU;AACjC,MAAA,MAAM,OAAA,GACJ,OAAO,GAAA,CAAI,OAAA,CAAQ,OAAA,KAAY,WAC3B,GAAA,CAAI,OAAA,CAAQ,OAAA,GACZ,GAAA,CAAI,OAAA,CAAQ,KAAA,EAAO,IAAI,CAAC,CAAA,KAAY,CAAA,CAAE,IAAA,KAAS,MAAA,GAAS,CAAA,CAAE,OAAO,EAAG,CAAA,CAAE,IAAA,CAAK,GAAG,CAAA,IAAK,EAAA;AAEzF,MAAA,MAAM,WAAA,GAAc,IAAI,QAAA,IAAY,QAAA;AACpC,MAAA,MAAM,MAAA,GAAS,SAAA,CAAU,GAAA,CAAI,WAAW,CAAA;AAGxC,MAAA,MAAM,cAAA,GAAA,CAAkB,MAAM,MAAA,CAAO,KAAA,CAAM,EAAE,QAAA,EAAU,WAAA,EAAa,CAAA,EAAG,QAAA;AACvE,MAAA,MAAM,eAAe,cAAA,CAAe,SAAA,CAAU,OAAK,CAAA,CAAE,EAAA,KAAO,IAAI,EAAE,CAAA;AAElE,MAAA,MAAM,YAAA,GAA6B;AAAA,QACjC,IAAI,GAAA,CAAI,EAAA;AAAA,QACR,MAAM,GAAA,CAAI,IAAA;AAAA,QACV,OAAA;AAAA,QACA,WAAW,GAAA,CAAI,SAAA;AAAA,QACf,QAAA,EAAU,WAAA;AAAA,QACV,WAAA,EAAa,QAAQ,KAAA,IAAS;AAAA,OAChC;AAEA,MAAA,IAAI,iBAAiB,EAAA,EAAI;AACvB,QAAA,YAAA,CAAa,OAAA,GAAU;AAAA,UACrB,MAAA,EAAQ,cAAA,CAAe,KAAA,CAAM,IAAA,CAAK,GAAA,CAAI,CAAA,EAAG,YAAA,GAAe,WAAW,CAAA,EAAG,YAAY,CAAA,CAAE,GAAA,CAAI,CAAA,CAAA,MAAM;AAAA,YAC5F,IAAI,CAAA,CAAE,EAAA;AAAA,YACN,MAAM,CAAA,CAAE,IAAA;AAAA,YACR,SAAS,CAAA,CAAE,OAAA;AAAA,YACX,SAAA,EAAW,CAAA,CAAE,SAAA,oBAAa,IAAI,IAAA;AAAK,WACrC,CAAE,CAAA;AAAA,UACF,KAAA,EAAO,cAAA,CAAe,KAAA,CAAM,YAAA,GAAe,CAAA,EAAG,eAAe,UAAA,GAAa,CAAC,CAAA,CAAE,GAAA,CAAI,CAAA,CAAA,MAAM;AAAA,YACrF,IAAI,CAAA,CAAE,EAAA;AAAA,YACN,MAAM,CAAA,CAAE,IAAA;AAAA,YACR,SAAS,CAAA,CAAE,OAAA;AAAA,YACX,SAAA,EAAW,CAAA,CAAE,SAAA,oBAAa,IAAI,IAAA;AAAK,WACrC,CAAE;AAAA,SACJ;AAAA,MACF;AAEA,MAAA,aAAA,CAAc,KAAK,YAAY,CAAA;AAAA,IACjC;AAGA,IAAA,MAAM,aAAA,GAAgB,cACnB,IAAA,CAAK,CAAC,GAAG,CAAA,KAAM,IAAI,IAAA,CAAK,CAAA,CAAE,SAAS,CAAA,CAAE,SAAQ,GAAI,IAAI,IAAA,CAAK,CAAA,CAAE,SAAS,CAAA,CAAE,SAAS,CAAA,CAChF,KAAA,CAAM,CAAA,EAAG,KAAK,CAAA;AAEjB,IAAA,OAAO;AAAA,MACL,OAAA,EAAS,aAAA;AAAA,MACT,OAAO,aAAA,CAAc,MAAA;AAAA,MACrB,KAAA,EAAO,WAAA;AAAA,MACP,WAAA,EAAa,gBAAgB,UAAA,GAAa,QAAA;AAAA,MAC1C,UAAA,EAAY,oBAAoB,UAAA,GAAa;AAAA,KAC/C;AAAA,EACF,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,wBAAwB,CAAA;AAAA,EACpD;AACF","file":"chunk-FKZVCLOY.js","sourcesContent":["import { convertMessages } from '@mastra/core/agent';\nimport type { MastraDBMessage } from '@mastra/core/agent';\nimport { RequestContext } from '@mastra/core/di';\nimport type { MastraMemory } from '@mastra/core/memory';\nimport type { StorageListMessagesInput, StorageOrderBy } from '@mastra/core/storage';\nimport { generateEmptyFromSchema } from '@mastra/core/utils';\nimport { HTTPException } from '../http-exception';\nimport type { Context } from '../types';\n\nimport { handleError } from './error';\nimport { validateBody } from './utils';\n\ninterface MemoryContext extends Context {\n  agentId?: string;\n  resourceId?: string;\n  threadId?: string;\n  requestContext?: RequestContext;\n}\n\nasync function getMemoryFromContext({\n  mastra,\n  agentId,\n  requestContext,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'requestContext'>): Promise<MastraMemory | null | undefined> {\n  const logger = mastra.getLogger();\n  let agent;\n  if (agentId) {\n    try {\n      agent = mastra.getAgent(agentId);\n    } catch (error) {\n      logger.debug('Error getting agent from mastra, searching agents for agent', error);\n    }\n  }\n  if (agentId && !agent) {\n    logger.debug('Agent not found, searching agents for agent', { agentId });\n    const agents = mastra.listAgents();\n    if (Object.keys(agents || {}).length) {\n      for (const [_, ag] of Object.entries(agents)) {\n        try {\n          const agents = await ag.listAgents();\n\n          if (agents[agentId]) {\n            agent = agents[agentId];\n            break;\n          }\n        } catch (error) {\n          logger.debug('Error getting agent from agent', error);\n        }\n      }\n    }\n\n    if (!agent) {\n      throw new HTTPException(404, { message: 'Agent not found' });\n    }\n  }\n\n  if (agent) {\n    return await agent?.getMemory({\n      requestContext: requestContext ?? new RequestContext(),\n    });\n  }\n}\n\n// Memory handlers\nexport async function getMemoryStatusHandler({\n  mastra,\n  agentId,\n  requestContext,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'requestContext'>) {\n  try {\n    const memory = await getMemoryFromContext({ mastra, agentId, requestContext });\n\n    if (!memory) {\n      return { result: false };\n    }\n\n    return { result: true };\n  } catch (error) {\n    return handleError(error, 'Error getting memory status');\n  }\n}\n\nexport async function getMemoryConfigHandler({\n  mastra,\n  agentId,\n  requestContext,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'requestContext'>) {\n  try {\n    const memory = await getMemoryFromContext({ mastra, agentId, requestContext });\n\n    if (!memory) {\n      throw new HTTPException(400, { message: 'Memory is not initialized' });\n    }\n\n    // Get the merged configuration (defaults + custom)\n    const config = memory.getMergedThreadConfig({});\n\n    return { config };\n  } catch (error) {\n    return handleError(error, 'Error getting memory configuration');\n  }\n}\n\nexport async function listThreadsHandler({\n  mastra,\n  agentId,\n  resourceId,\n  requestContext,\n  page,\n  perPage,\n  orderBy,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'resourceId' | 'requestContext'> & {\n  page: number;\n  perPage: number | false;\n  orderBy?: StorageOrderBy;\n}) {\n  try {\n    const memory = await getMemoryFromContext({ mastra, agentId, requestContext });\n\n    if (!memory) {\n      throw new HTTPException(400, { message: 'Memory is not initialized' });\n    }\n\n    validateBody({ resourceId });\n\n    const result = await memory.listThreadsByResourceId({\n      resourceId: resourceId!,\n      page,\n      perPage,\n      orderBy,\n    });\n    return result;\n  } catch (error) {\n    return handleError(error, 'Error listing threads');\n  }\n}\n\nexport async function getThreadByIdHandler({\n  mastra,\n  agentId,\n  threadId,\n  requestContext,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'threadId' | 'requestContext'>) {\n  try {\n    validateBody({ threadId });\n\n    const memory = await getMemoryFromContext({ mastra, agentId, requestContext });\n    if (!memory) {\n      throw new HTTPException(400, { message: 'Memory is not initialized' });\n    }\n\n    const thread = await memory.getThreadById({ threadId: threadId! });\n    if (!thread) {\n      throw new HTTPException(404, { message: 'Thread not found' });\n    }\n\n    return thread;\n  } catch (error) {\n    return handleError(error, 'Error getting thread');\n  }\n}\n\nexport async function saveMessagesHandler({\n  mastra,\n  agentId,\n  body,\n  requestContext,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'requestContext'> & {\n  body: {\n    messages: Parameters<MastraMemory['saveMessages']>[0]['messages'];\n  };\n}) {\n  try {\n    const memory = await getMemoryFromContext({ mastra, agentId, requestContext });\n\n    if (!memory) {\n      throw new HTTPException(400, { message: 'Memory is not initialized' });\n    }\n\n    if (!body?.messages) {\n      throw new HTTPException(400, { message: 'Messages are required' });\n    }\n\n    if (!Array.isArray(body.messages)) {\n      throw new HTTPException(400, { message: 'Messages should be an array' });\n    }\n\n    // Validate that all messages have threadId and resourceId\n    const invalidMessages = body.messages.filter(message => !message.threadId || !message.resourceId);\n    if (invalidMessages.length > 0) {\n      throw new HTTPException(400, {\n        message: `All messages must have threadId and resourceId fields. Found ${invalidMessages.length} invalid message(s).`,\n      });\n    }\n\n    const processedMessages = body.messages.map(message => ({\n      ...message,\n      id: message.id || memory.generateId(),\n      createdAt: message.createdAt ? new Date(message.createdAt) : new Date(),\n    }));\n\n    const result = await memory.saveMessages({ messages: processedMessages, memoryConfig: {} });\n    return result;\n  } catch (error) {\n    return handleError(error, 'Error saving messages');\n  }\n}\n\nexport async function createThreadHandler({\n  mastra,\n  agentId,\n  body,\n  requestContext,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'requestContext'> & {\n  body?: Omit<Parameters<MastraMemory['createThread']>[0], 'resourceId'> & { resourceId?: string };\n}) {\n  try {\n    const memory = await getMemoryFromContext({ mastra, agentId, requestContext });\n\n    if (!memory) {\n      throw new HTTPException(400, { message: 'Memory is not initialized' });\n    }\n\n    validateBody({ resourceId: body?.resourceId });\n\n    const result = await memory.createThread({\n      resourceId: body?.resourceId!,\n      title: body?.title,\n      metadata: body?.metadata,\n      threadId: body?.threadId,\n    });\n    return result;\n  } catch (error) {\n    return handleError(error, 'Error saving thread to memory');\n  }\n}\n\nexport async function updateThreadHandler({\n  mastra,\n  agentId,\n  threadId,\n  body,\n  requestContext,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'threadId' | 'requestContext'> & {\n  body?: Parameters<MastraMemory['saveThread']>[0]['thread'];\n}) {\n  try {\n    const memory = await getMemoryFromContext({ mastra, agentId, requestContext });\n\n    if (!body) {\n      throw new HTTPException(400, { message: 'Body is required' });\n    }\n\n    const { title, metadata, resourceId } = body;\n    const updatedAt = new Date();\n\n    validateBody({ threadId });\n\n    if (!memory) {\n      throw new HTTPException(400, { message: 'Memory is not initialized' });\n    }\n\n    const thread = await memory.getThreadById({ threadId: threadId! });\n    if (!thread) {\n      throw new HTTPException(404, { message: 'Thread not found' });\n    }\n\n    const updatedThread = {\n      ...thread,\n      title: title || thread.title,\n      metadata: metadata || thread.metadata,\n      resourceId: resourceId || thread.resourceId,\n      createdAt: thread.createdAt,\n      updatedAt,\n    };\n\n    const result = await memory.saveThread({ thread: updatedThread });\n    return result;\n  } catch (error) {\n    return handleError(error, 'Error updating thread');\n  }\n}\n\nexport async function deleteThreadHandler({\n  mastra,\n  agentId,\n  threadId,\n  requestContext,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'threadId' | 'requestContext'>) {\n  try {\n    validateBody({ threadId });\n\n    const memory = await getMemoryFromContext({ mastra, agentId, requestContext });\n    if (!memory) {\n      throw new HTTPException(400, { message: 'Memory is not initialized' });\n    }\n\n    const thread = await memory.getThreadById({ threadId: threadId! });\n    if (!thread) {\n      throw new HTTPException(404, { message: 'Thread not found' });\n    }\n\n    await memory.deleteThread(threadId!);\n    return { result: 'Thread deleted' };\n  } catch (error) {\n    return handleError(error, 'Error deleting thread');\n  }\n}\n\nexport async function listMessagesHandler({\n  mastra,\n  threadId,\n  resourceId,\n  perPage,\n  page,\n  orderBy,\n  include,\n  filter,\n}: Pick<MemoryContext, 'mastra'> & StorageListMessagesInput) {\n  try {\n    validateBody({ threadId });\n\n    const storage = mastra.getStorage();\n\n    if (!storage) {\n      throw new HTTPException(400, { message: 'Storage is not initialized' });\n    }\n\n    const thread = await storage.getThreadById({ threadId: threadId! });\n\n    if (!thread) {\n      throw new HTTPException(404, { message: 'Thread not found' });\n    }\n\n    const result = await storage.listMessages({\n      threadId: threadId!,\n      resourceId,\n      perPage,\n      page,\n      orderBy,\n      include,\n      filter,\n    });\n    return result;\n  } catch (error) {\n    return handleError(error, 'Error getting messages');\n  }\n}\n\nexport async function getMessagesHandler({\n  mastra,\n  agentId,\n  threadId,\n  limit,\n  requestContext,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'threadId' | 'requestContext'> & {\n  limit?: number;\n}) {\n  if (limit !== undefined && (!Number.isInteger(limit) || limit <= 0)) {\n    throw new HTTPException(400, { message: 'Invalid limit: must be a positive integer' });\n  }\n  try {\n    validateBody({ threadId });\n\n    const memory = await getMemoryFromContext({ mastra, agentId, requestContext });\n\n    if (!memory) {\n      throw new HTTPException(400, { message: 'Memory is not initialized' });\n    }\n\n    if (!threadId) {\n      throw new HTTPException(400, { message: 'No threadId found' });\n    }\n\n    const thread = await memory.getThreadById({ threadId: threadId });\n    if (!thread) {\n      throw new HTTPException(404, { message: 'Thread not found' });\n    }\n\n    const result = await memory.query({\n      threadId: threadId,\n      ...(limit && { selectBy: { last: limit } }),\n    });\n    const uiMessages = convertMessages(result.messages).to('AIV5.UI');\n    return { messages: result.messages, uiMessages };\n  } catch (error) {\n    return handleError(error, 'Error getting messages');\n  }\n}\n\n/**\n * Handler to get the working memory for a thread (optionally resource-scoped).\n * @returns workingMemory - the working memory for the thread\n * @returns source - thread or resource\n */\nexport async function getWorkingMemoryHandler({\n  mastra,\n  agentId,\n  threadId,\n  resourceId,\n  requestContext,\n  memoryConfig,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'threadId' | 'requestContext'> & {\n  resourceId?: Parameters<MastraMemory['getWorkingMemory']>[0]['resourceId'];\n  memoryConfig?: Parameters<MastraMemory['getWorkingMemory']>[0]['memoryConfig'];\n}) {\n  try {\n    const memory = await getMemoryFromContext({ mastra, agentId, requestContext });\n    validateBody({ threadId });\n    if (!memory) {\n      throw new HTTPException(400, { message: 'Memory is not initialized' });\n    }\n    const thread = await memory.getThreadById({ threadId: threadId! });\n    const threadExists = !!thread;\n    const template = await memory.getWorkingMemoryTemplate({ memoryConfig });\n    const workingMemoryTemplate =\n      template?.format === 'json'\n        ? { ...template, content: JSON.stringify(generateEmptyFromSchema(template.content)) }\n        : template;\n    const workingMemory = await memory.getWorkingMemory({ threadId: threadId!, resourceId, memoryConfig });\n    const config = memory.getMergedThreadConfig(memoryConfig || {});\n    const source = config.workingMemory?.scope !== 'thread' && resourceId ? 'resource' : 'thread';\n    return { workingMemory, source, workingMemoryTemplate, threadExists };\n  } catch (error) {\n    return handleError(error, 'Error getting working memory');\n  }\n}\n\n/**\n * Handler to update the working memory for a thread (optionally resource-scoped).\n * @param threadId - the thread id\n * @param body - the body containing the working memory to update and the resource id (optional)\n */\nexport async function updateWorkingMemoryHandler({\n  mastra,\n  agentId,\n  threadId,\n  body,\n  requestContext,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'threadId' | 'requestContext'> & {\n  body: Omit<Parameters<MastraMemory['updateWorkingMemory']>[0], 'threadId'>;\n}) {\n  try {\n    validateBody({ threadId });\n    const memory = await getMemoryFromContext({ mastra, agentId, requestContext });\n    const { resourceId, memoryConfig, workingMemory } = body;\n    if (!memory) {\n      throw new HTTPException(400, { message: 'Memory is not initialized' });\n    }\n    const thread = await memory.getThreadById({ threadId: threadId! });\n    if (!thread) {\n      throw new HTTPException(404, { message: 'Thread not found' });\n    }\n\n    await memory.updateWorkingMemory({ threadId: threadId!, resourceId, workingMemory, memoryConfig });\n    return { success: true };\n  } catch (error) {\n    return handleError(error, 'Error updating working memory');\n  }\n}\n\ninterface SearchResult {\n  id: string;\n  role: string;\n  content: any;\n  createdAt: Date;\n  threadId?: string;\n  threadTitle?: string;\n  score?: number;\n  context?: {\n    before?: SearchResult[];\n    after?: SearchResult[];\n  };\n}\n\ninterface SearchResponse {\n  results: SearchResult[];\n  count: number;\n  query: string;\n  searchScope?: string;\n  searchType?: string;\n}\n\n/**\n * Handler to delete one or more messages.\n * @param messageIds - Can be a single ID, array of IDs, or objects with ID property\n */\nexport async function deleteMessagesHandler({\n  mastra,\n  agentId,\n  messageIds,\n  requestContext,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'requestContext'> & {\n  messageIds: string | string[] | { id: string } | { id: string }[];\n}) {\n  try {\n    if (messageIds === undefined || messageIds === null) {\n      throw new HTTPException(400, { message: 'messageIds is required' });\n    }\n\n    const memory = await getMemoryFromContext({ mastra, agentId, requestContext });\n    if (!memory) {\n      throw new HTTPException(400, { message: 'Memory is not initialized' });\n    }\n\n    // Delete the messages - let the memory method handle validation\n    await memory.deleteMessages(messageIds as any);\n\n    // Count messages for response\n    let count = 1;\n    if (Array.isArray(messageIds)) {\n      count = messageIds.length;\n    }\n\n    return { success: true, message: `${count} message${count === 1 ? '' : 's'} deleted successfully` };\n  } catch (error) {\n    return handleError(error, 'Error deleting messages');\n  }\n}\n\nexport async function searchMemoryHandler({\n  mastra,\n  agentId,\n  searchQuery,\n  resourceId,\n  threadId,\n  limit = 20,\n  requestContext,\n  memoryConfig,\n}: Pick<MemoryContext, 'mastra' | 'agentId' | 'requestContext'> & {\n  searchQuery: string;\n  resourceId: string;\n  threadId?: string;\n  limit?: number;\n  memoryConfig?: any;\n}): Promise<SearchResponse | ReturnType<typeof handleError>> {\n  try {\n    validateBody({ searchQuery, resourceId });\n\n    const memory = await getMemoryFromContext({ mastra, agentId, requestContext });\n    if (!memory) {\n      throw new HTTPException(400, { message: 'Memory is not initialized' });\n    }\n\n    // Get memory configuration first to check scope\n    const config = memory.getMergedThreadConfig(memoryConfig || {});\n    const hasSemanticRecall = !!config?.semanticRecall;\n    const resourceScope =\n      typeof config?.semanticRecall === 'object' ? config?.semanticRecall?.scope !== 'thread' : true;\n\n    // Only validate thread ownership if we're in thread scope\n    if (threadId && !resourceScope) {\n      const thread = await memory.getThreadById({ threadId });\n      if (!thread) {\n        throw new HTTPException(404, { message: 'Thread not found' });\n      }\n      if (thread.resourceId !== resourceId) {\n        throw new HTTPException(403, { message: 'Thread does not belong to the specified resource' });\n      }\n    }\n\n    const searchResults: SearchResult[] = [];\n\n    // If threadId is provided and scope is thread-based, check if the thread exists\n    if (threadId && !resourceScope) {\n      const thread = await memory.getThreadById({ threadId });\n      if (!thread) {\n        // Thread doesn't exist yet (new unsaved thread) - return empty results\n        return {\n          results: [],\n          count: 0,\n          query: searchQuery,\n          searchScope: resourceScope ? 'resource' : 'thread',\n          searchType: hasSemanticRecall ? 'semantic' : 'text',\n        };\n      }\n    }\n\n    // If no threadId provided, get one from the resource\n    if (!threadId) {\n      const { threads } = await memory.listThreadsByResourceId({\n        resourceId,\n        page: 0,\n        perPage: 1,\n        orderBy: { field: 'updatedAt', direction: 'DESC' },\n      });\n\n      if (threads.length === 0) {\n        return {\n          results: [],\n          count: 0,\n          query: searchQuery,\n          searchScope: resourceScope ? 'resource' : 'thread',\n          searchType: hasSemanticRecall ? 'semantic' : 'text',\n        };\n      }\n\n      // Use first thread - Memory class will handle scope internally\n      threadId = threads[0]!.id;\n    }\n\n    const beforeRange =\n      typeof config.semanticRecall === `boolean`\n        ? 2\n        : typeof config.semanticRecall?.messageRange === `number`\n          ? config.semanticRecall.messageRange\n          : config.semanticRecall?.messageRange.before || 2;\n    const afterRange =\n      typeof config.semanticRecall === `boolean`\n        ? 2\n        : typeof config.semanticRecall?.messageRange === `number`\n          ? config.semanticRecall.messageRange\n          : config.semanticRecall?.messageRange.after || 2;\n\n    if (resourceScope && config.semanticRecall) {\n      config.semanticRecall =\n        typeof config.semanticRecall === `boolean`\n          ? // make message range 0 so we can highlight the matches in search, message range will include other messages, not the matching ones\n            // and we add prev/next messages in a special section on each message anyway\n            { messageRange: 0, topK: 2, scope: 'resource' }\n          : { ...config.semanticRecall, messageRange: 0 };\n    }\n\n    // Single call to rememberMessages - just like the agent does\n    // The Memory class handles scope (thread vs resource) internally\n    const result = await memory.rememberMessages({\n      threadId,\n      resourceId,\n      vectorMessageSearch: searchQuery,\n      config,\n    });\n\n    // Get all threads to build context and show which thread each message is from\n    // Fetch threads by IDs from the actual messages to avoid truncation\n    const threadIds = Array.from(\n      new Set(result.messages.map((m: MastraDBMessage) => m.threadId || threadId!).filter(Boolean)),\n    );\n    const fetched = await Promise.all(threadIds.map((id: string) => memory.getThreadById({ threadId: id })));\n    const threadMap = new Map(fetched.filter(Boolean).map(t => [t!.id, t!]));\n\n    // Process each message in the results\n    for (const msg of result.messages) {\n      const content =\n        typeof msg.content.content === `string`\n          ? msg.content.content\n          : msg.content.parts?.map((p: any) => (p.type === 'text' ? p.text : '')).join(' ') || '';\n\n      const msgThreadId = msg.threadId || threadId;\n      const thread = threadMap.get(msgThreadId);\n\n      // Get thread messages for context\n      const threadMessages = (await memory.query({ threadId: msgThreadId })).messages;\n      const messageIndex = threadMessages.findIndex(m => m.id === msg.id);\n\n      const searchResult: SearchResult = {\n        id: msg.id,\n        role: msg.role,\n        content,\n        createdAt: msg.createdAt,\n        threadId: msgThreadId,\n        threadTitle: thread?.title || msgThreadId,\n      };\n\n      if (messageIndex !== -1) {\n        searchResult.context = {\n          before: threadMessages.slice(Math.max(0, messageIndex - beforeRange), messageIndex).map(m => ({\n            id: m.id,\n            role: m.role,\n            content: m.content,\n            createdAt: m.createdAt || new Date(),\n          })),\n          after: threadMessages.slice(messageIndex + 1, messageIndex + afterRange + 1).map(m => ({\n            id: m.id,\n            role: m.role,\n            content: m.content,\n            createdAt: m.createdAt || new Date(),\n          })),\n        };\n      }\n\n      searchResults.push(searchResult);\n    }\n\n    // Sort by date (newest first) and limit\n    const sortedResults = searchResults\n      .sort((a, b) => new Date(b.createdAt).getTime() - new Date(a.createdAt).getTime())\n      .slice(0, limit);\n\n    return {\n      results: sortedResults,\n      count: sortedResults.length,\n      query: searchQuery,\n      searchScope: resourceScope ? 'resource' : 'thread',\n      searchType: hasSemanticRecall ? 'semantic' : 'text',\n    };\n  } catch (error) {\n    return handleError(error, 'Error searching memory');\n  }\n}\n"]}