@mastra/server 1.68.0-alpha.0 → 1.68.0-alpha.11

package/dist/{observability-new-endpoints-pIq_OWa8.js → observability-new-endpoints-CWfdrnzJ.js}

@@ -1,10 +1,11 @@
 import { t as HTTPException } from "./http-exception-CyfxZB6W.js";
-import { a as wrapSchemaForQueryParams, i as pickParams, n as createRoute } from "./route-builder-BukcT_Kp.js";
-import { l as MASTRA_USER_KEY, m as isStudioClientTypeHeader } from "./constants-0jFCWoex.js";
+import { a as wrapSchemaForQueryParams, i as pickParams, n as createRoute } from "./route-builder-DDOjv0sM.js";
+import { d as MASTRA_USER_ROLES_KEY, l as MASTRA_USER_KEY, m as isStudioClientTypeHeader } from "./constants-0jFCWoex.js";
 import { handleError } from "./server/handlers/error.js";
 import { paginationArgsSchema as paginationArgsSchema$1 } from "./server/handlers/observability-list-query-schemas.js";
-import { NEW_ROUTE_DEFS, OBSERVABILITY_LIST_ENDPOINTS, assertObservabilityDeltaSupported, assertObservabilityTraceQuerySupported, createObservabilityListQuerySchema, getObservabilityStore } from "./server/handlers/observability-shared.js";
+import { NEW_ROUTE_DEFS, OBSERVABILITY_LIST_ENDPOINTS, assertObservabilityDeltaSupported, assertObservabilityThreadQuerySupported, assertObservabilityTraceQueryDiscoverySupported, assertObservabilityTraceQuerySupported, createObservabilityListQuerySchema, getObservabilityStore, supportsTraceQueryDiscoveryCore } from "./server/handlers/observability-shared.js";
 import { prepareAuthorEnrichment } from "./server/handlers/author-enrichment.js";
+import { getCallerPermissions } from "./server/handlers/authorship.js";
 import { z } from "zod/v4";
 import * as coreStorage from "@mastra/core/storage";
 import { coreFeatures } from "@mastra/core/features";
@@ -34,6 +35,8 @@ let EntityType$1 = /* @__PURE__ */ function(EntityType) {
 	EntityType["WORKFLOW_STEP"] = "workflow_step";
 	/** Tool */
 	EntityType["TOOL"] = "tool";
+	/** MCP server */
+	EntityType["MCP_SERVER"] = "mcp_server";
 	/** Workflow */
 	EntityType["WORKFLOW_RUN"] = "workflow_run";
 	/** Memory */
@@ -869,7 +872,7 @@ const listFeedbackResponseSchema = listFeedbackResponseSchema$1.extend({ feedbac
 //#endregion
 //#region src/server/handlers/observability-new-endpoints.ts
 function createNewRoute(def, config) {
-	const { handler, preserveHttpExceptions, ...schemas } = config;
+	const { handler, preserveHttpExceptions, isCoreSupported, onUnsupportedCore, ...schemas } = config;
 	return createRoute({
 		...def,
 		...schemas,
@@ -878,7 +881,10 @@ function createNewRoute(def, config) {
 		requiresAuth: true,
 		handler: (async (params) => {
 			try {
-				if (!coreFeatures.has("observability:v1.13.2")) throw new HTTPException(501, { message: "New observability endpoints require @mastra/core >= 1.13.2, please upgrade." });
+				if (!coreFeatures.has("observability:v1.13.2") || isCoreSupported && !isCoreSupported()) {
+					if (onUnsupportedCore) onUnsupportedCore();
+					throw new HTTPException(501, { message: "New observability endpoints require @mastra/core >= 1.13.2, please upgrade." });
+				}
 				return await handler(params);
 			} catch (error) {
 				if (preserveHttpExceptions && error instanceof HTTPException) throw error;
@@ -921,6 +927,14 @@ const traceQueryTimeoutErrorSchema = z.object({
 	code: z.literal("TRACE_QUERY_EXECUTION_TIMEOUT"),
 	message: z.string()
 }).strict();
+const traceQueryResourceLimitErrorSchema = z.object({
+	code: z.literal("TRACE_QUERY_RESOURCE_LIMIT"),
+	message: z.string()
+}).strict();
+const traceQueryDiscoveryUnsupportedErrorSchema = z.object({
+	code: z.literal("TRACE_QUERY_DISCOVERY_UNSUPPORTED"),
+	message: z.string()
+}).strict();
 const traceQueryValidationError = (error) => ({
 	status: 422,
 	body: {
@@ -942,22 +956,41 @@ function throwTraceQueryError(status, body) {
 		})
 	});
 }
+const throwTraceQueryDiscoveryCoreUnsupported = () => throwTraceQueryError(501, {
+	code: "TRACE_QUERY_DISCOVERY_UNSUPPORTED",
+	message: "Trace query discovery requires a newer @mastra/core. Please upgrade."
+});
 const QUERY_TRACES = createNewRoute(NEW_ROUTE_DEFS.QUERY_TRACES, {
 	bodySchema: coreStorage.traceQueryRequestSchema,
 	responseSchema: coreStorage.traceQueryResponseSchema,
 	onValidationError: traceQueryValidationError,
 	maxBodySize: 256 * 1024,
 	preserveHttpExceptions: true,
-	handler: async ({ mastra, timeRange, where, group, orderBy, page }) => {
+	handler: async ({ mastra, requestContext, timeRange, where, group, orderBy, page, pagination, mode, after, limit }) => {
 		let plan;
 		try {
+			const user = requestContext.get(MASTRA_USER_KEY);
+			const userId = user && typeof user === "object" && "id" in user ? user.id : void 0;
+			const roles = requestContext.get(MASTRA_USER_ROLES_KEY);
+			const authorizationBinding = mode === "delta" || pagination !== void 0 ? JSON.stringify({
+				userId: typeof userId === "string" || typeof userId === "number" ? userId : null,
+				resourceId: requestContext.get("mastra__resourceId") ?? null,
+				organizationId: requestContext.get("organizationId") ?? null,
+				authMode: requestContext.get("mastra__authMode") ?? null,
+				permissions: [...new Set(getCallerPermissions(requestContext))].sort(),
+				roles: Array.isArray(roles) ? [...new Set(roles.filter((role) => typeof role === "string"))].sort() : []
+			}) : void 0;
 			plan = coreStorage.planTraceQuery({
 				timeRange,
 				where,
 				group,
 				orderBy,
-				page
-			});
+				page,
+				pagination,
+				mode,
+				after,
+				limit
+			}, { authorizationBinding });
 		} catch (error) {
 			if (error instanceof coreStorage.TraceQueryValidationError) throwTraceQueryError(422, {
 				code: error.code,
@@ -974,6 +1007,7 @@ const QUERY_TRACES = createNewRoute(NEW_ROUTE_DEFS.QUERY_TRACES, {
 		try {
 			observabilityStore = await getObservabilityStore(mastra);
 			assertObservabilityTraceQuerySupported(observabilityStore);
+			if (plan.paginationMode === "delta" && !observabilityStore.getFeatures()?.includes("delta-polling")) throw new HTTPException(501, { message: "This storage provider does not support observability delta polling" });
 		} catch (error) {
 			if (error instanceof HTTPException && error.status === 501) throwTraceQueryError(501, {
 				code: "TRACE_QUERY_UNSUPPORTED",
@@ -984,6 +1018,10 @@ const QUERY_TRACES = createNewRoute(NEW_ROUTE_DEFS.QUERY_TRACES, {
 		try {
 			return await observabilityStore.queryTraces(plan);
 		} catch (error) {
+			if (error instanceof coreStorage.TraceQueryCursorError) throwTraceQueryError(error.code === "TRACE_QUERY_CURSOR_CONFLICT" ? 409 : 400, {
+				code: error.code,
+				message: error.message
+			});
 			if (error instanceof coreStorage.TraceQueryExecutionError) throwTraceQueryError(504, {
 				code: error.code,
 				message: error.message
@@ -1018,6 +1056,195 @@ if (QUERY_TRACES.openapi) {
 		content: { "application/json": { schema: traceQueryTimeoutErrorSchema } }
 	};
 }
+const GET_TRACE_QUERY_FIELDS = createNewRoute(NEW_ROUTE_DEFS.GET_TRACE_QUERY_FIELDS, {
+	bodySchema: coreStorage.getTraceQueryFieldsArgsSchema,
+	responseSchema: coreStorage.getTraceQueryFieldsResponseSchema,
+	onValidationError: traceQueryValidationError,
+	maxBodySize: 256 * 1024,
+	preserveHttpExceptions: true,
+	isCoreSupported: supportsTraceQueryDiscoveryCore,
+	onUnsupportedCore: throwTraceQueryDiscoveryCoreUnsupported,
+	handler: async ({ mastra, timeRange, predicateScope, search, limit }) => {
+		const args = {
+			timeRange,
+			predicateScope,
+			search,
+			limit
+		};
+		const plan = coreStorage.planTraceQueryObservedFields(args);
+		let observabilityStore;
+		try {
+			observabilityStore = await getObservabilityStore(mastra);
+			assertObservabilityTraceQueryDiscoverySupported(observabilityStore);
+		} catch (error) {
+			if (error instanceof HTTPException && error.status === 501) throwTraceQueryError(501, {
+				code: "TRACE_QUERY_DISCOVERY_UNSUPPORTED",
+				message: error.message
+			});
+			throw error;
+		}
+		try {
+			const observed = await observabilityStore.getTraceQueryObservedFields(plan);
+			return {
+				canonicalFields: coreStorage.getTraceQueryCanonicalFieldDescriptors(predicateScope, search),
+				...observed
+			};
+		} catch (error) {
+			if (error instanceof coreStorage.TraceQueryResourceLimitError) throwTraceQueryError(503, {
+				code: error.code,
+				message: error.message
+			});
+			if (error instanceof coreStorage.TraceQueryExecutionError) throwTraceQueryError(504, {
+				code: error.code,
+				message: error.message
+			});
+			throw error;
+		}
+	}
+});
+const GET_TRACE_QUERY_VALUES = createNewRoute(NEW_ROUTE_DEFS.GET_TRACE_QUERY_VALUES, {
+	bodySchema: coreStorage.getTraceQueryValuesArgsSchema,
+	responseSchema: coreStorage.getTraceQueryValuesResponseSchema,
+	onValidationError: traceQueryValidationError,
+	maxBodySize: 256 * 1024,
+	preserveHttpExceptions: true,
+	isCoreSupported: supportsTraceQueryDiscoveryCore,
+	onUnsupportedCore: throwTraceQueryDiscoveryCoreUnsupported,
+	handler: async ({ mastra, timeRange, predicateScope, path, search, limit }) => {
+		const plan = coreStorage.planTraceQueryValues({
+			timeRange,
+			predicateScope,
+			path,
+			search,
+			limit
+		});
+		let observabilityStore;
+		try {
+			observabilityStore = await getObservabilityStore(mastra);
+			assertObservabilityTraceQueryDiscoverySupported(observabilityStore);
+		} catch (error) {
+			if (error instanceof HTTPException && error.status === 501) throwTraceQueryError(501, {
+				code: "TRACE_QUERY_DISCOVERY_UNSUPPORTED",
+				message: error.message
+			});
+			throw error;
+		}
+		try {
+			return await observabilityStore.getTraceQueryValues(plan);
+		} catch (error) {
+			if (error instanceof coreStorage.TraceQueryResourceLimitError) throwTraceQueryError(503, {
+				code: error.code,
+				message: error.message
+			});
+			if (error instanceof coreStorage.TraceQueryExecutionError) throwTraceQueryError(504, {
+				code: error.code,
+				message: error.message
+			});
+			throw error;
+		}
+	}
+});
+for (const route of [GET_TRACE_QUERY_FIELDS, GET_TRACE_QUERY_VALUES]) {
+	if (!route.openapi) continue;
+	route.openapi.responses[413] = {
+		description: "Request body exceeds 256 KiB",
+		content: { "application/json": { schema: traceQueryBodyTooLargeErrorSchema } }
+	};
+	route.openapi.responses[422] = {
+		description: "Structurally or semantically invalid trace-query discovery request",
+		content: { "application/json": { schema: traceQueryValidationResponseSchema } }
+	};
+	route.openapi.responses[501] = {
+		description: "The installed Core or configured observability store does not support trace-query discovery",
+		content: { "application/json": { schema: traceQueryDiscoveryUnsupportedErrorSchema } }
+	};
+	route.openapi.responses[503] = {
+		description: "Trace-query discovery exceeded the configured database resource limit",
+		content: { "application/json": { schema: traceQueryResourceLimitErrorSchema } }
+	};
+	route.openapi.responses[504] = {
+		description: "Trace-query discovery exceeded the configured database execution timeout",
+		content: { "application/json": { schema: traceQueryTimeoutErrorSchema } }
+	};
+}
+const QUERY_THREADS = createNewRoute(NEW_ROUTE_DEFS.QUERY_THREADS, {
+	bodySchema: coreStorage.queryThreadsInputSchema,
+	responseSchema: coreStorage.queryThreadsResultSchema,
+	onValidationError: traceQueryValidationError,
+	maxBodySize: 256 * 1024,
+	preserveHttpExceptions: true,
+	onUnsupportedCore: () => throwTraceQueryError(501, {
+		code: "TRACE_QUERY_UNSUPPORTED",
+		message: "Thread queries require a newer @mastra/core with observability thread-query support. Please upgrade."
+	}),
+	handler: async ({ mastra, traces, where, page }) => {
+		let plan;
+		try {
+			plan = coreStorage.planThreadQuery({
+				traces,
+				where,
+				page
+			});
+		} catch (error) {
+			if (error instanceof coreStorage.TraceQueryValidationError) throwTraceQueryError(422, {
+				code: error.code,
+				message: error.message,
+				issues: error.issues
+			});
+			if (error instanceof coreStorage.TraceQueryCursorError) throwTraceQueryError(error.code === "TRACE_QUERY_CURSOR_CONFLICT" ? 409 : 400, {
+				code: error.code,
+				message: error.message
+			});
+			throw error;
+		}
+		let observabilityStore;
+		try {
+			observabilityStore = await getObservabilityStore(mastra);
+			assertObservabilityThreadQuerySupported(observabilityStore);
+		} catch (error) {
+			if (error instanceof HTTPException && error.status === 501) throwTraceQueryError(501, {
+				code: "TRACE_QUERY_UNSUPPORTED",
+				message: error.message
+			});
+			throw error;
+		}
+		try {
+			return await observabilityStore.queryThreads(plan);
+		} catch (error) {
+			if (error instanceof coreStorage.TraceQueryExecutionError) throwTraceQueryError(504, {
+				code: error.code,
+				message: error.message
+			});
+			throw error;
+		}
+	}
+});
+if (QUERY_THREADS.openapi) {
+	QUERY_THREADS.openapi.responses[400] = {
+		description: "Malformed JSON or malformed cursor",
+		content: { "application/json": { schema: z.union([traceQueryMalformedBodyErrorSchema, traceQueryMalformedCursorErrorSchema]) } }
+	};
+	QUERY_THREADS.openapi.responses[409] = {
+		description: "Cursor does not match the normalized query",
+		content: { "application/json": { schema: traceQueryCursorConflictErrorSchema } }
+	};
+	QUERY_THREADS.openapi.responses[413] = {
+		description: "Request body exceeds 256 KiB",
+		content: { "application/json": { schema: traceQueryBodyTooLargeErrorSchema } }
+	};
+	QUERY_THREADS.openapi.responses[422] = {
+		description: "Structurally or semantically invalid thread query",
+		content: { "application/json": { schema: traceQueryValidationResponseSchema } }
+	};
+	QUERY_THREADS.openapi.responses[501] = {
+		description: "The configured observability store does not support thread queries",
+		content: { "application/json": { schema: traceQueryUnsupportedErrorSchema } }
+	};
+	QUERY_THREADS.openapi.responses[504] = {
+		description: "Thread query exceeded the configured database execution timeout",
+		content: { "application/json": { schema: traceQueryTimeoutErrorSchema } }
+	};
+}
 const LIST_LOGS = createNewRoute(NEW_ROUTE_DEFS.LIST_LOGS, {
 	queryParamSchema: createObservabilityListQuerySchema(logsFilterSchema, logsOrderBySchema),
 	responseSchema: listLogsResponseSchema,
@@ -1370,6 +1597,7 @@ const GET_TAGS = createNewRoute(NEW_ROUTE_DEFS.GET_TAGS, {
 });
 const NEW_ROUTES = {
 	QUERY_TRACES,
+	QUERY_THREADS,
 	LIST_LOGS,
 	LIST_SCORES,
 	CREATE_SCORE,
@@ -1402,6 +1630,6 @@ const NEW_ROUTES = {
 	GET_TAGS
 };
 //#endregion
-export { NEW_ROUTES as A, GET_SCORE_TIME_SERIES as C, LIST_LOGS as D, LIST_FEEDBACK as E, UPDATE_FEEDBACK_REVIEW_STATUS as M, LIST_METRICS as O, GET_SCORE_PERCENTILES as S, GET_TAGS as T, GET_METRIC_PERCENTILES as _, GET_ENTITY_NAMES as a, GET_SCORE_AGGREGATE as b, GET_FEEDBACK_AGGREGATE as c, GET_FEEDBACK_TIME_SERIES as d, GET_METRIC_AGGREGATE as f, GET_METRIC_NAMES as g, GET_METRIC_LABEL_VALUES as h, DELETE_SCORES as i, QUERY_TRACES as j, LIST_SCORES as k, GET_FEEDBACK_BREAKDOWN as l, GET_METRIC_LABEL_KEYS as m, CREATE_SCORE as n, GET_ENTITY_TYPES as o, GET_METRIC_BREAKDOWN as p, DELETE_FEEDBACK as r, GET_ENVIRONMENTS as s, CREATE_FEEDBACK as t, GET_FEEDBACK_PERCENTILES as u, GET_METRIC_TIME_SERIES as v, GET_SERVICE_NAMES as w, GET_SCORE_BREAKDOWN as x, GET_SCORE as y };
+export { LIST_METRICS as A, GET_SCORE_TIME_SERIES as C, GET_TRACE_QUERY_VALUES as D, GET_TRACE_QUERY_FIELDS as E, UPDATE_FEEDBACK_REVIEW_STATUS as F, NEW_ROUTES as M, QUERY_THREADS as N, LIST_FEEDBACK as O, QUERY_TRACES as P, GET_SCORE_PERCENTILES as S, GET_TAGS as T, GET_METRIC_PERCENTILES as _, GET_ENTITY_NAMES as a, GET_SCORE_AGGREGATE as b, GET_FEEDBACK_AGGREGATE as c, GET_FEEDBACK_TIME_SERIES as d, GET_METRIC_AGGREGATE as f, GET_METRIC_NAMES as g, GET_METRIC_LABEL_VALUES as h, DELETE_SCORES as i, LIST_SCORES as j, LIST_LOGS as k, GET_FEEDBACK_BREAKDOWN as l, GET_METRIC_LABEL_KEYS as m, CREATE_SCORE as n, GET_ENTITY_TYPES as o, GET_METRIC_BREAKDOWN as p, DELETE_FEEDBACK as r, GET_ENVIRONMENTS as s, CREATE_FEEDBACK as t, GET_FEEDBACK_PERCENTILES as u, GET_METRIC_TIME_SERIES as v, GET_SERVICE_NAMES as w, GET_SCORE_BREAKDOWN as x, GET_SCORE as y };
 
-//# sourceMappingURL=observability-new-endpoints-pIq_OWa8.js.map
+//# sourceMappingURL=observability-new-endpoints-CWfdrnzJ.js.map