@mastra/server 0.0.0-extend-clickhouse-20250418135620 → 0.0.0-feat-tool-input-validation-20250731232758

package/dist/chunk-PIAJQ75M.js

@@ -0,0 +1,539 @@
+import { validateBody } from './chunk-RSEO4XPX.js';
+import { handleError } from './chunk-LF7P5PLR.js';
+import { HTTPException } from './chunk-LCM566I4.js';
+import { __export } from './chunk-MLKGABMK.js';
+import { generateEmptyFromSchema } from '@mastra/core';
+
+// src/server/handlers/memory.ts
+var memory_exports = {};
+__export(memory_exports, {
+  createThreadHandler: () => createThreadHandler,
+  deleteMessagesHandler: () => deleteMessagesHandler,
+  deleteThreadHandler: () => deleteThreadHandler,
+  getMemoryConfigHandler: () => getMemoryConfigHandler,
+  getMemoryStatusHandler: () => getMemoryStatusHandler,
+  getMessagesHandler: () => getMessagesHandler,
+  getMessagesPaginatedHandler: () => getMessagesPaginatedHandler,
+  getThreadByIdHandler: () => getThreadByIdHandler,
+  getThreadsHandler: () => getThreadsHandler,
+  getWorkingMemoryHandler: () => getWorkingMemoryHandler,
+  saveMessagesHandler: () => saveMessagesHandler,
+  searchMemoryHandler: () => searchMemoryHandler,
+  updateThreadHandler: () => updateThreadHandler,
+  updateWorkingMemoryHandler: () => updateWorkingMemoryHandler
+});
+async function getMemoryFromContext({
+  mastra,
+  agentId,
+  networkId,
+  runtimeContext
+}) {
+  const agent = agentId ? mastra.getAgent(agentId) : null;
+  if (agentId && !agent) {
+    throw new HTTPException(404, { message: "Agent not found" });
+  }
+  const network = networkId ? mastra.vnext_getNetwork(networkId) : null;
+  if (networkId && !network) {
+    throw new HTTPException(404, { message: "Network not found" });
+  }
+  if (agent) {
+    return await agent?.getMemory() || mastra.getMemory();
+  }
+  if (network) {
+    return await network?.getMemory({ runtimeContext }) || mastra.getMemory();
+  }
+  return mastra.getMemory();
+}
+async function getMemoryStatusHandler({
+  mastra,
+  agentId,
+  networkId,
+  runtimeContext
+}) {
+  try {
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
+    if (!memory) {
+      return { result: false };
+    }
+    return { result: true };
+  } catch (error) {
+    return handleError(error, "Error getting memory status");
+  }
+}
+async function getMemoryConfigHandler({
+  mastra,
+  agentId,
+  networkId,
+  runtimeContext
+}) {
+  try {
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
+    if (!memory) {
+      throw new HTTPException(400, { message: "Memory is not initialized" });
+    }
+    const config = memory.getMergedThreadConfig({});
+    return { config };
+  } catch (error) {
+    return handleError(error, "Error getting memory configuration");
+  }
+}
+async function getThreadsHandler({
+  mastra,
+  agentId,
+  resourceId,
+  networkId,
+  runtimeContext,
+  orderBy,
+  sortDirection
+}) {
+  try {
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
+    if (!memory) {
+      throw new HTTPException(400, { message: "Memory is not initialized" });
+    }
+    validateBody({ resourceId });
+    const threads = await memory.getThreadsByResourceId({
+      resourceId,
+      orderBy,
+      sortDirection
+    });
+    return threads;
+  } catch (error) {
+    return handleError(error, "Error getting threads");
+  }
+}
+async function getThreadByIdHandler({
+  mastra,
+  agentId,
+  threadId,
+  networkId,
+  runtimeContext
+}) {
+  try {
+    validateBody({ threadId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
+    if (!memory) {
+      throw new HTTPException(400, { message: "Memory is not initialized" });
+    }
+    const thread = await memory.getThreadById({ threadId });
+    if (!thread) {
+      throw new HTTPException(404, { message: "Thread not found" });
+    }
+    return thread;
+  } catch (error) {
+    return handleError(error, "Error getting thread");
+  }
+}
+async function saveMessagesHandler({
+  mastra,
+  agentId,
+  body,
+  networkId,
+  runtimeContext
+}) {
+  try {
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
+    if (!memory) {
+      throw new HTTPException(400, { message: "Memory is not initialized" });
+    }
+    if (!body?.messages) {
+      throw new HTTPException(400, { message: "Messages are required" });
+    }
+    if (!Array.isArray(body.messages)) {
+      throw new HTTPException(400, { message: "Messages should be an array" });
+    }
+    const invalidMessages = body.messages.filter((message) => !message.threadId || !message.resourceId);
+    if (invalidMessages.length > 0) {
+      throw new HTTPException(400, {
+        message: `All messages must have threadId and resourceId fields. Found ${invalidMessages.length} invalid message(s).`
+      });
+    }
+    const processedMessages = body.messages.map((message) => ({
+      ...message,
+      id: message.id || memory.generateId(),
+      createdAt: message.createdAt ? new Date(message.createdAt) : /* @__PURE__ */ new Date()
+    }));
+    const result = await memory.saveMessages({ messages: processedMessages, memoryConfig: {} });
+    return result;
+  } catch (error) {
+    return handleError(error, "Error saving messages");
+  }
+}
+async function createThreadHandler({
+  mastra,
+  agentId,
+  body,
+  networkId,
+  runtimeContext
+}) {
+  try {
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
+    if (!memory) {
+      throw new HTTPException(400, { message: "Memory is not initialized" });
+    }
+    validateBody({ resourceId: body?.resourceId });
+    const result = await memory.createThread({
+      resourceId: body?.resourceId,
+      title: body?.title,
+      metadata: body?.metadata,
+      threadId: body?.threadId
+    });
+    return result;
+  } catch (error) {
+    return handleError(error, "Error saving thread to memory");
+  }
+}
+async function updateThreadHandler({
+  mastra,
+  agentId,
+  threadId,
+  body,
+  networkId,
+  runtimeContext
+}) {
+  try {
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
+    if (!body) {
+      throw new HTTPException(400, { message: "Body is required" });
+    }
+    const { title, metadata, resourceId } = body;
+    const updatedAt = /* @__PURE__ */ new Date();
+    validateBody({ threadId });
+    if (!memory) {
+      throw new HTTPException(400, { message: "Memory is not initialized" });
+    }
+    const thread = await memory.getThreadById({ threadId });
+    if (!thread) {
+      throw new HTTPException(404, { message: "Thread not found" });
+    }
+    const updatedThread = {
+      ...thread,
+      title: title || thread.title,
+      metadata: metadata || thread.metadata,
+      resourceId: resourceId || thread.resourceId,
+      createdAt: thread.createdAt,
+      updatedAt
+    };
+    const result = await memory.saveThread({ thread: updatedThread });
+    return result;
+  } catch (error) {
+    return handleError(error, "Error updating thread");
+  }
+}
+async function deleteThreadHandler({
+  mastra,
+  agentId,
+  threadId,
+  networkId,
+  runtimeContext
+}) {
+  try {
+    validateBody({ threadId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
+    if (!memory) {
+      throw new HTTPException(400, { message: "Memory is not initialized" });
+    }
+    const thread = await memory.getThreadById({ threadId });
+    if (!thread) {
+      throw new HTTPException(404, { message: "Thread not found" });
+    }
+    await memory.deleteThread(threadId);
+    return { result: "Thread deleted" };
+  } catch (error) {
+    return handleError(error, "Error deleting thread");
+  }
+}
+async function getMessagesPaginatedHandler({
+  mastra,
+  threadId,
+  resourceId,
+  selectBy,
+  format
+}) {
+  try {
+    validateBody({ threadId });
+    const storage = mastra.getStorage();
+    if (!storage) {
+      throw new HTTPException(400, { message: "Storage is not initialized" });
+    }
+    const thread = await storage.getThreadById({ threadId });
+    if (!thread) {
+      throw new HTTPException(404, { message: "Thread not found" });
+    }
+    const result = await storage.getMessagesPaginated({ threadId, resourceId, selectBy, format });
+    return result;
+  } catch (error) {
+    return handleError(error, "Error getting messages");
+  }
+}
+async function getMessagesHandler({
+  mastra,
+  agentId,
+  threadId,
+  limit,
+  networkId,
+  runtimeContext
+}) {
+  if (limit !== void 0 && (!Number.isInteger(limit) || limit <= 0)) {
+    throw new HTTPException(400, { message: "Invalid limit: must be a positive integer" });
+  }
+  try {
+    validateBody({ threadId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
+    if (!memory) {
+      throw new HTTPException(400, { message: "Memory is not initialized" });
+    }
+    const thread = await memory.getThreadById({ threadId });
+    if (!thread) {
+      throw new HTTPException(404, { message: "Thread not found" });
+    }
+    const result = await memory.query({
+      threadId,
+      ...limit && { selectBy: { last: limit } }
+    });
+    return { messages: result.messages, uiMessages: result.uiMessages };
+  } catch (error) {
+    return handleError(error, "Error getting messages");
+  }
+}
+async function getWorkingMemoryHandler({
+  mastra,
+  agentId,
+  threadId,
+  resourceId,
+  networkId,
+  runtimeContext,
+  memoryConfig
+}) {
+  try {
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
+    validateBody({ threadId });
+    if (!memory) {
+      throw new HTTPException(400, { message: "Memory is not initialized" });
+    }
+    const thread = await memory.getThreadById({ threadId });
+    const threadExists = !!thread;
+    const template = await memory.getWorkingMemoryTemplate({ memoryConfig });
+    const workingMemoryTemplate = template?.format === "json" ? { ...template, content: JSON.stringify(generateEmptyFromSchema(template.content)) } : template;
+    const workingMemory = await memory.getWorkingMemory({ threadId, resourceId, memoryConfig });
+    const config = memory.getMergedThreadConfig(memoryConfig || {});
+    const source = config.workingMemory?.scope === "resource" && resourceId ? "resource" : "thread";
+    return { workingMemory, source, workingMemoryTemplate, threadExists };
+  } catch (error) {
+    return handleError(error, "Error getting working memory");
+  }
+}
+async function updateWorkingMemoryHandler({
+  mastra,
+  agentId,
+  threadId,
+  body,
+  networkId,
+  runtimeContext
+}) {
+  try {
+    validateBody({ threadId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
+    const { resourceId, memoryConfig, workingMemory } = body;
+    if (!memory) {
+      throw new HTTPException(400, { message: "Memory is not initialized" });
+    }
+    const thread = await memory.getThreadById({ threadId });
+    if (!thread) {
+      throw new HTTPException(404, { message: "Thread not found" });
+    }
+    await memory.updateWorkingMemory({ threadId, resourceId, workingMemory, memoryConfig });
+    return { success: true };
+  } catch (error) {
+    return handleError(error, "Error updating working memory");
+  }
+}
+async function deleteMessagesHandler({
+  mastra,
+  agentId,
+  messageIds,
+  networkId,
+  runtimeContext
+}) {
+  try {
+    if (messageIds === void 0 || messageIds === null) {
+      throw new HTTPException(400, { message: "messageIds is required" });
+    }
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
+    if (!memory) {
+      throw new HTTPException(400, { message: "Memory is not initialized" });
+    }
+    await memory.deleteMessages(messageIds);
+    let count = 1;
+    if (Array.isArray(messageIds)) {
+      count = messageIds.length;
+    }
+    return { success: true, message: `${count} message${count === 1 ? "" : "s"} deleted successfully` };
+  } catch (error) {
+    return handleError(error, "Error deleting messages");
+  }
+}
+async function searchMemoryHandler({
+  mastra,
+  agentId,
+  searchQuery,
+  resourceId,
+  threadId,
+  limit = 20,
+  networkId,
+  runtimeContext,
+  memoryConfig
+}) {
+  try {
+    validateBody({ searchQuery, resourceId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
+    if (!memory) {
+      throw new HTTPException(400, { message: "Memory is not initialized" });
+    }
+    const config = memory.getMergedThreadConfig(memoryConfig || {});
+    const hasSemanticRecall = !!config?.semanticRecall;
+    const resourceScope = typeof config?.semanticRecall === "object" && config?.semanticRecall?.scope === "resource";
+    if (threadId && !resourceScope) {
+      const thread = await memory.getThreadById({ threadId });
+      if (!thread) {
+        throw new HTTPException(404, { message: "Thread not found" });
+      }
+      if (thread.resourceId !== resourceId) {
+        throw new HTTPException(403, { message: "Thread does not belong to the specified resource" });
+      }
+    }
+    const searchResults = [];
+    const messageMap = /* @__PURE__ */ new Map();
+    if (threadId && !resourceScope) {
+      const thread = await memory.getThreadById({ threadId });
+      if (!thread) {
+        return {
+          results: [],
+          count: 0,
+          query: searchQuery,
+          searchScope: "thread",
+          searchType: hasSemanticRecall ? "semantic" : "text"
+        };
+      }
+    }
+    if (!threadId || resourceScope) {
+      const threads = await memory.getThreadsByResourceId({ resourceId });
+      if (threads.length === 0) {
+        return {
+          results: [],
+          count: 0,
+          query: searchQuery,
+          searchScope: "resource",
+          searchType: hasSemanticRecall ? "semantic" : "text"
+        };
+      }
+      for (const thread of threads) {
+        const result = await memory.rememberMessages({
+          threadId: thread.id,
+          resourceId,
+          vectorMessageSearch: searchQuery,
+          config
+        });
+        const threadMessages = (await memory.query({ threadId: thread.id })).uiMessages;
+        result.messagesV2.forEach((msg) => {
+          if (messageMap.has(msg.id)) return;
+          messageMap.set(msg.id, true);
+          const content = msg.content.content || msg.content.parts?.map((p) => p.type === "text" ? p.text : "").join(" ") || "";
+          if (!hasSemanticRecall && !content.toLowerCase().includes(searchQuery.toLowerCase())) {
+            return;
+          }
+          const messageIndex = threadMessages.findIndex((m) => m.id === msg.id);
+          const searchResult = {
+            id: msg.id,
+            role: msg.role,
+            content,
+            createdAt: msg.createdAt,
+            threadId: msg.threadId || thread.id,
+            threadTitle: thread.title || msg.threadId || thread.id
+          };
+          if (messageIndex !== -1) {
+            searchResult.context = {
+              before: threadMessages.slice(Math.max(0, messageIndex - 2), messageIndex).map((m) => ({
+                id: m.id,
+                role: m.role,
+                content: m.content,
+                createdAt: m.createdAt || /* @__PURE__ */ new Date()
+              })),
+              after: threadMessages.slice(messageIndex + 1, messageIndex + 3).map((m) => ({
+                id: m.id,
+                role: m.role,
+                content: m.content,
+                createdAt: m.createdAt || /* @__PURE__ */ new Date()
+              }))
+            };
+          }
+          searchResults.push(searchResult);
+        });
+      }
+    } else if (threadId) {
+      const thread = await memory.getThreadById({ threadId });
+      if (!thread) {
+        return {
+          results: [],
+          count: 0,
+          query: searchQuery,
+          searchScope: "thread",
+          searchType: hasSemanticRecall ? "semantic" : "text"
+        };
+      }
+      const result = await memory.rememberMessages({
+        threadId,
+        resourceId,
+        vectorMessageSearch: searchQuery,
+        config
+      });
+      const threadMessages = (await memory.query({ threadId })).uiMessages;
+      result.messagesV2.forEach((msg) => {
+        if (messageMap.has(msg.id)) return;
+        messageMap.set(msg.id, true);
+        const content = msg.content.content || msg.content.parts?.map((p) => p.type === "text" ? p.text : "").join(" ") || "";
+        if (!hasSemanticRecall && !content.toLowerCase().includes(searchQuery.toLowerCase())) {
+          return;
+        }
+        const messageIndex = threadMessages.findIndex((m) => m.id === msg.id);
+        const searchResult = {
+          id: msg.id,
+          role: msg.role,
+          content,
+          createdAt: msg.createdAt,
+          threadId,
+          threadTitle: thread?.title || threadId
+        };
+        if (messageIndex !== -1) {
+          searchResult.context = {
+            before: threadMessages.slice(Math.max(0, messageIndex - 2), messageIndex).map((m) => ({
+              id: m.id,
+              role: m.role,
+              content: m.content,
+              createdAt: m.createdAt || /* @__PURE__ */ new Date()
+            })),
+            after: threadMessages.slice(messageIndex + 1, messageIndex + 3).map((m) => ({
+              id: m.id,
+              role: m.role,
+              content: m.content,
+              createdAt: m.createdAt || /* @__PURE__ */ new Date()
+            }))
+          };
+        }
+        searchResults.push(searchResult);
+      });
+    }
+    const sortedResults = searchResults.sort((a, b) => new Date(b.createdAt).getTime() - new Date(a.createdAt).getTime()).slice(0, limit);
+    return {
+      results: sortedResults,
+      count: sortedResults.length,
+      query: searchQuery,
+      searchScope: resourceScope ? "resource" : "thread",
+      searchType: hasSemanticRecall ? "semantic" : "text"
+    };
+  } catch (error) {
+    return handleError(error, "Error searching memory");
+  }
+}
+
+export { createThreadHandler, deleteMessagesHandler, deleteThreadHandler, getMemoryConfigHandler, getMemoryStatusHandler, getMessagesHandler, getMessagesPaginatedHandler, getThreadByIdHandler, getThreadsHandler, getWorkingMemoryHandler, memory_exports, saveMessagesHandler, searchMemoryHandler, updateThreadHandler, updateWorkingMemoryHandler };

package/dist/chunk-PZQDCRPV.cjs

@@ -0,0 +1,16 @@
+'use strict';
+
+var chunk2KZFMI6P_cjs = require('./chunk-2KZFMI6P.cjs');
+
+// src/server/handlers/error.ts
+function handleError(error, defaultMessage) {
+  const apiError = error;
+  const apiErrorStatus = apiError.status || apiError.details?.status || 500;
+  throw new chunk2KZFMI6P_cjs.HTTPException(apiErrorStatus, {
+    message: apiError.message || defaultMessage,
+    stack: apiError.stack,
+    cause: apiError.cause
+  });
+}
+
+exports.handleError = handleError;

package/dist/chunk-QGX47B5D.cjs

@@ -0,0 +1,86 @@
+'use strict';
+
+var chunkCCGRCYWJ_cjs = require('./chunk-CCGRCYWJ.cjs');
+var chunkPZQDCRPV_cjs = require('./chunk-PZQDCRPV.cjs');
+var chunk75ZPJI57_cjs = require('./chunk-75ZPJI57.cjs');
+
+// src/server/handlers/logs.ts
+var logs_exports = {};
+chunk75ZPJI57_cjs.__export(logs_exports, {
+  getLogTransports: () => getLogTransports,
+  getLogsByRunIdHandler: () => getLogsByRunIdHandler,
+  getLogsHandler: () => getLogsHandler
+});
+async function getLogsHandler({
+  mastra,
+  transportId,
+  params
+}) {
+  try {
+    chunkCCGRCYWJ_cjs.validateBody({ transportId });
+    const { fromDate, toDate, logLevel, filters: _filters, page, perPage } = params || {};
+    const filters = _filters ? Object.fromEntries(
+      (Array.isArray(_filters) ? _filters : [_filters]).map((attr) => {
+        const [key, value] = attr.split(":");
+        return [key, value];
+      })
+    ) : void 0;
+    const logs = await mastra.getLogs(transportId, {
+      fromDate,
+      toDate,
+      logLevel,
+      filters,
+      page: page ? Number(page) : void 0,
+      perPage: perPage ? Number(perPage) : void 0
+    });
+    return logs;
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting logs");
+  }
+}
+async function getLogsByRunIdHandler({
+  mastra,
+  runId,
+  transportId,
+  params
+}) {
+  try {
+    chunkCCGRCYWJ_cjs.validateBody({ runId, transportId });
+    const { fromDate, toDate, logLevel, filters: _filters, page, perPage } = params || {};
+    const filters = _filters ? Object.fromEntries(
+      (Array.isArray(_filters) ? _filters : [_filters]).map((attr) => {
+        const [key, value] = attr.split(":");
+        return [key, value];
+      })
+    ) : void 0;
+    const logs = await mastra.getLogsByRunId({
+      runId,
+      transportId,
+      fromDate,
+      toDate,
+      logLevel,
+      filters,
+      page: page ? Number(page) : void 0,
+      perPage: perPage ? Number(perPage) : void 0
+    });
+    return logs;
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting logs by run ID");
+  }
+}
+async function getLogTransports({ mastra }) {
+  try {
+    const logger = mastra.getLogger();
+    const transports = logger.getTransports();
+    return {
+      transports: transports ? [...transports.keys()] : []
+    };
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting log Transports");
+  }
+}
+
+exports.getLogTransports = getLogTransports;
+exports.getLogsByRunIdHandler = getLogsByRunIdHandler;
+exports.getLogsHandler = getLogsHandler;
+exports.logs_exports = logs_exports;

package/dist/{chunk-WTHDCRMY.js → chunk-RG473F6Y.js}

@@ -1,5 +1,6 @@
-import { handleError } from './chunk-3AHQ5RGN.js';
-import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
+import { handleError } from './chunk-LF7P5PLR.js';
+import { HTTPException } from './chunk-LCM566I4.js';
+import { __export } from './chunk-MLKGABMK.js';
 
 // src/server/handlers/telemetry.ts
 var telemetry_exports = {};
@@ -15,12 +16,12 @@ async function getTelemetryHandler({ mastra, body }) {
       throw new HTTPException(400, { message: "Telemetry is not initialized" });
     }
     if (!storage) {
-      throw new HTTPException(400, { message: "Storage is not initialized" });
+      return [];
     }
     if (!body) {
       throw new HTTPException(400, { message: "Body is required" });
     }
-    const { name, scope, page, perPage, attribute } = body;
+    const { name, scope, page, perPage, attribute, fromDate, toDate } = body;
     const attributes = attribute ? Object.fromEntries(
       (Array.isArray(attribute) ? attribute : [attribute]).map((attr) => {
         const [key, value] = attr.split(":");
@@ -32,7 +33,9 @@ async function getTelemetryHandler({ mastra, body }) {
       scope,
       page: Number(page ?? 0),
       perPage: Number(perPage ?? 100),
-      attributes
+      attributes,
+      fromDate: fromDate ? new Date(fromDate) : void 0,
+      toDate: toDate ? new Date(toDate) : void 0
     });
     return traces;
   } catch (error2) {
@@ -44,7 +47,10 @@ async function storeTelemetryHandler({ mastra, body }) {
     const storage = mastra.getStorage();
     const logger = mastra.getLogger();
     if (!storage) {
-      throw new HTTPException(400, { message: "Storage is not initialized" });
+      return {
+        status: "error",
+        message: "Storage is not initialized"
+      };
     }
     const now = /* @__PURE__ */ new Date();
     const items = body?.resourceSpans?.[0]?.scopeSpans;