@mastra/server 0.0.0-break-rename-vnext-legacy-20251002212351 → 0.0.0-bundle-recursion-20251030002519

package/dist/{chunk-52UP7DFT.cjs → chunk-2S4IMB6E.cjs}

@@ -2,12 +2,12 @@
 
 var chunkRE4RPXT2_cjs = require('./chunk-RE4RPXT2.cjs');
 var chunk7NADHFD2_cjs = require('./chunk-7NADHFD2.cjs');
-var chunkEMMSS5I5_cjs = require('./chunk-EMMSS5I5.cjs');
+var chunkO7I5CWRX_cjs = require('./chunk-O7I5CWRX.cjs');
 var scoreTraces = require('@mastra/core/scores/scoreTraces');
 
 // src/server/handlers/observability.ts
 var observability_exports = {};
-chunkEMMSS5I5_cjs.__export(observability_exports, {
+chunkO7I5CWRX_cjs.__export(observability_exports, {
   getAITraceHandler: () => getAITraceHandler,
   getAITracesPaginatedHandler: () => getAITracesPaginatedHandler,
   getScoresBySpan: () => getScoresBySpan,
@@ -126,5 +126,5 @@ exports.getAITracesPaginatedHandler = getAITracesPaginatedHandler;
 exports.getScoresBySpan = getScoresBySpan;
 exports.observability_exports = observability_exports;
 exports.scoreTracesHandler = scoreTracesHandler;
-//# sourceMappingURL=chunk-52UP7DFT.cjs.map
-//# sourceMappingURL=chunk-52UP7DFT.cjs.map
+//# sourceMappingURL=chunk-2S4IMB6E.cjs.map
+//# sourceMappingURL=chunk-2S4IMB6E.cjs.map

package/dist/{chunk-52UP7DFT.cjs.map → chunk-2S4IMB6E.cjs.map}

@@ -1 +1 @@
-{"version":3,"sources":["../src/server/handlers/observability.ts"],"names":["__export","HTTPException","handleError","scoreTraces"],"mappings":";;;;;;;;AAAA,IAAA,qBAAA,GAAA;AAAAA,0BAAA,CAAA,qBAAA,EAAA;AAAA,EAAA,iBAAA,EAAA,MAAA,iBAAA;AAAA,EAAA,2BAAA,EAAA,MAAA,2BAAA;AAAA,EAAA,eAAA,EAAA,MAAA,eAAA;AAAA,EAAA,kBAAA,EAAA,MAAA;AAAA,CAAA,CAAA;AAyBA,eAAsB,iBAAA,CAAkB,EAAE,MAAA,EAAQ,OAAA,EAAQ,EAA+C;AACvG,EAAA,IAAI;AACF,IAAA,IAAI,CAAC,OAAA,EAAS;AACZ,MAAA,MAAM,IAAIC,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,wBAAwB,CAAA;AAAA,IAClE;AAEA,IAAA,MAAM,OAAA,GAAU,OAAO,UAAA,EAAW;AAClC,IAAA,IAAI,CAAC,OAAA,EAAS;AACZ,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,4BAA4B,CAAA;AAAA,IACtE;AAEA,IAAA,MAAM,KAAA,GAAQ,MAAM,OAAA,CAAQ,UAAA,CAAW,OAAO,CAAA;AAE9C,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,MAAM,IAAIA,gCAAc,GAAA,EAAK,EAAE,SAAS,CAAA,eAAA,EAAkB,OAAO,eAAe,CAAA;AAAA,IAClF;AAEA,IAAA,OAAO,KAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAAC,6BAAA,CAAY,OAAO,wBAAwB,CAAA;AAAA,EAC7C;AACF;AAMA,eAAsB,2BAAA,CAA4B,EAAE,MAAA,EAAQ,IAAA,EAAK,EAAyB;AACxF,EAAA,IAAI;AACF,IAAA,MAAM,OAAA,GAAU,OAAO,UAAA,EAAW;AAClC,IAAA,IAAI,CAAC,OAAA,EAAS;AACZ,MAAA,MAAM,IAAID,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,4BAA4B,CAAA;AAAA,IACtE;AAEA,IAAA,IAAI,CAAC,IAAA,EAAM;AACT,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,4BAA4B,CAAA;AAAA,IACtE;AAEA,IAAA,MAAM,EAAE,OAAA,EAAS,UAAA,EAAW,GAAI,IAAA;AAEhC,IAAA,IAAI,UAAA,EAAY,IAAA,IAAQ,UAAA,CAAW,IAAA,GAAO,CAAA,EAAG;AAC3C,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,uCAAuC,CAAA;AAAA,IACjF;AAEA,IAAA,IAAI,UAAA,EAAY,OAAA,IAAW,UAAA,CAAW,OAAA,GAAU,CAAA,EAAG;AACjD,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,2CAA2C,CAAA;AAAA,IACrF;AAEA,IAAA,IAAI,YAAY,SAAA,EAAW;AACzB,MAAA,MAAM,EAAE,KAAA,EAAO,GAAA,EAAI,GAAI,UAAA,CAAW,SAAA;AAElC,MAAA,IAAI,KAAA,IAAS,EAAE,KAAA,YAAiB,IAAA,CAAA,EAAO;AACrC,QAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,qCAAqC,CAAA;AAAA,MAC/E;AAEA,MAAA,IAAI,GAAA,IAAO,EAAE,GAAA,YAAe,IAAA,CAAA,EAAO;AACjC,QAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,qCAAqC,CAAA;AAAA,MAC/E;AAAA,IACF;AAEA,IAAA,OAAO,QAAQ,oBAAA,CAAqB;AAAA,MAClC,UAAA;AAAA,MACA;AAAA,KACD,CAAA;AAAA,EACH,SAAS,KAAA,EAAO;AACd,IAAAC,6BAAA,CAAY,OAAO,mCAAmC,CAAA;AAAA,EACxD;AACF;AAMA,eAAsB,kBAAA,CAAmB,EAAE,MAAA,EAAQ,IAAA,EAAK,EAAuB;AAC7E,EAAA,IAAI;AACF,IAAA,IAAI,CAAC,IAAA,EAAM;AACT,MAAA,MAAM,IAAID,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,4BAA4B,CAAA;AAAA,IACtE;AAEA,IAAA,MAAM,EAAE,UAAA,EAAY,OAAA,EAAQ,GAAI,IAAA;AAEhC,IAAA,IAAI,CAAC,UAAA,EAAY;AACf,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,2BAA2B,CAAA;AAAA,IACrE;AAEA,IAAA,IAAI,CAAC,OAAA,IAAW,OAAA,CAAQ,MAAA,KAAW,CAAA,EAAG;AACpC,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,mCAAmC,CAAA;AAAA,IAC7E;AAEA,IAAA,MAAM,OAAA,GAAU,OAAO,UAAA,EAAW;AAClC,IAAA,IAAI,CAAC,OAAA,EAAS;AACZ,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,4BAA4B,CAAA;AAAA,IACtE;AAEA,IAAA,MAAM,MAAA,GAAS,MAAA,CAAO,eAAA,CAAgB,UAAU,CAAA;AAChD,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAIA,gCAAc,GAAA,EAAK,EAAE,SAAS,CAAA,QAAA,EAAW,UAAU,eAAe,CAAA;AAAA,IAC9E;AAEA,IAAA,MAAM,MAAA,GAAS,OAAO,SAAA,EAAU;AAChC,IAAAE,uBAAA,CAAY;AAAA,MACV,UAAA;AAAA,MACA,OAAA;AAAA,MACA;AAAA,KACD,CAAA,CAAE,KAAA,CAAM,CAAA,KAAA,KAAS;AAChB,MAAA,MAAA,EAAQ,KAAA,CAAM,CAAA,iCAAA,EAAoC,KAAA,CAAM,OAAO,IAAI,KAAK,CAAA;AAAA,IAC1E,CAAC,CAAA;AAGD,IAAA,OAAO;AAAA,MACL,MAAA,EAAQ,SAAA;AAAA,MACR,OAAA,EAAS,uBAAuB,OAAA,CAAQ,MAAM,IAAI,OAAA,CAAQ,MAAA,KAAW,CAAA,GAAI,OAAA,GAAU,QAAQ,CAAA,CAAA;AAAA,MAC3F,YAAY,OAAA,CAAQ;AAAA,KACtB;AAAA,EACF,SAAS,KAAA,EAAO;AACd,IAAAD,6BAAA,CAAY,OAAO,gCAAgC,CAAA;AAAA,EACrD;AACF;AAEA,eAAsB,eAAA,CAAgB;AAAA,EACpC,MAAA;AAAA,EACA,OAAA;AAAA,EACA,MAAA;AAAA,EACA;AACF,CAAA,EAAiF;AAC/E,EAAA,IAAI;AACF,IAAA,MAAM,OAAA,GAAU,OAAO,UAAA,EAAW;AAClC,IAAA,IAAI,CAAC,OAAA,EAAS;AACZ,MAAA,MAAM,IAAID,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,4BAA4B,CAAA;AAAA,IACtE;AAEA,IAAA,IAAI,CAAC,OAAA,IAAW,CAAC,MAAA,EAAQ;AACvB,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,qCAAqC,CAAA;AAAA,IAC/E;AAEA,IAAA,OAAO,MAAM,OAAA,CAAQ,eAAA,CAAgB,EAAE,OAAA,EAAS,MAAA,EAAQ,YAAY,CAAA;AAAA,EACtE,SAAS,KAAA,EAAO;AACd,IAAA,OAAOC,6BAAA,CAAY,OAAO,8BAA8B,CAAA;AAAA,EAC1D;AACF","file":"chunk-52UP7DFT.cjs","sourcesContent":["import type { AITracesPaginatedArg, StoragePagination } from '@mastra/core';\nimport { scoreTraces } from '@mastra/core/scores/scoreTraces';\nimport { HTTPException } from '../http-exception';\nimport type { Context } from '../types';\nimport { handleError } from './error';\n\ninterface ObservabilityContext extends Context {\n  traceId?: string;\n  body?: AITracesPaginatedArg;\n}\n\ninterface ScoreTracesContext extends Context {\n  body?: {\n    scorerName: string;\n    targets: Array<{\n      traceId: string;\n      spanId?: string;\n    }>;\n  };\n}\n\n/**\n * Get a complete AI trace by trace ID\n * Returns all spans in the trace with their parent-child relationships\n */\nexport async function getAITraceHandler({ mastra, traceId }: ObservabilityContext & { traceId: string }) {\n  try {\n    if (!traceId) {\n      throw new HTTPException(400, { message: 'Trace ID is required' });\n    }\n\n    const storage = mastra.getStorage();\n    if (!storage) {\n      throw new HTTPException(500, { message: 'Storage is not available' });\n    }\n\n    const trace = await storage.getAITrace(traceId);\n\n    if (!trace) {\n      throw new HTTPException(404, { message: `Trace with ID '${traceId}' not found` });\n    }\n\n    return trace;\n  } catch (error) {\n    handleError(error, 'Error getting AI trace');\n  }\n}\n\n/**\n * Get paginated AI traces with filtering and pagination\n * Returns only root spans (parent spans) for pagination, not child spans\n */\nexport async function getAITracesPaginatedHandler({ mastra, body }: ObservabilityContext) {\n  try {\n    const storage = mastra.getStorage();\n    if (!storage) {\n      throw new HTTPException(500, { message: 'Storage is not available' });\n    }\n\n    if (!body) {\n      throw new HTTPException(400, { message: 'Request body is required' });\n    }\n\n    const { filters, pagination } = body;\n\n    if (pagination?.page && pagination.page < 0) {\n      throw new HTTPException(400, { message: 'Page must be a non-negative integer' });\n    }\n\n    if (pagination?.perPage && pagination.perPage < 0) {\n      throw new HTTPException(400, { message: 'Per page must be a non-negative integer' });\n    }\n\n    if (pagination?.dateRange) {\n      const { start, end } = pagination.dateRange;\n\n      if (start && !(start instanceof Date)) {\n        throw new HTTPException(400, { message: 'Invalid date format in date range' });\n      }\n\n      if (end && !(end instanceof Date)) {\n        throw new HTTPException(400, { message: 'Invalid date format in date range' });\n      }\n    }\n\n    return storage.getAITracesPaginated({\n      pagination,\n      filters,\n    });\n  } catch (error) {\n    handleError(error, 'Error getting AI traces paginated');\n  }\n}\n\n/**\n * Score traces using a specified scorer\n * Fire-and-forget approach - returns immediately while scoring runs in background\n */\nexport async function scoreTracesHandler({ mastra, body }: ScoreTracesContext) {\n  try {\n    if (!body) {\n      throw new HTTPException(400, { message: 'Request body is required' });\n    }\n\n    const { scorerName, targets } = body;\n\n    if (!scorerName) {\n      throw new HTTPException(400, { message: 'Scorer Name is required' });\n    }\n\n    if (!targets || targets.length === 0) {\n      throw new HTTPException(400, { message: 'At least one target is required' });\n    }\n\n    const storage = mastra.getStorage();\n    if (!storage) {\n      throw new HTTPException(500, { message: 'Storage is not available' });\n    }\n\n    const scorer = mastra.getScorerByName(scorerName);\n    if (!scorer) {\n      throw new HTTPException(404, { message: `Scorer '${scorerName}' not found` });\n    }\n\n    const logger = mastra.getLogger();\n    scoreTraces({\n      scorerName,\n      targets,\n      mastra,\n    }).catch(error => {\n      logger?.error(`Background trace scoring failed: ${error.message}`, error);\n    });\n\n    // Return immediate response\n    return {\n      status: 'success',\n      message: `Scoring started for ${targets.length} ${targets.length === 1 ? 'trace' : 'traces'}`,\n      traceCount: targets.length,\n    };\n  } catch (error) {\n    handleError(error, 'Error processing trace scoring');\n  }\n}\n\nexport async function getScoresBySpan({\n  mastra,\n  traceId,\n  spanId,\n  pagination,\n}: Context & { traceId: string; spanId: string; pagination: StoragePagination }) {\n  try {\n    const storage = mastra.getStorage();\n    if (!storage) {\n      throw new HTTPException(500, { message: 'Storage is not available' });\n    }\n\n    if (!traceId || !spanId) {\n      throw new HTTPException(400, { message: 'Trace ID and span ID are required' });\n    }\n\n    return await storage.getScoresBySpan({ traceId, spanId, pagination });\n  } catch (error) {\n    return handleError(error, 'Error getting scores by span');\n  }\n}\n"]}
+{"version":3,"sources":["../src/server/handlers/observability.ts"],"names":["__export","HTTPException","handleError","scoreTraces"],"mappings":";;;;;;;;AAAA,IAAA,qBAAA,GAAA;AAAAA,0BAAA,CAAA,qBAAA,EAAA;AAAA,EAAA,iBAAA,EAAA,MAAA,iBAAA;AAAA,EAAA,2BAAA,EAAA,MAAA,2BAAA;AAAA,EAAA,eAAA,EAAA,MAAA,eAAA;AAAA,EAAA,kBAAA,EAAA,MAAA;AAAA,CAAA,CAAA;AAyBA,eAAsB,iBAAA,CAAkB,EAAE,MAAA,EAAQ,OAAA,EAAQ,EAA+C;AACvG,EAAA,IAAI;AACF,IAAA,IAAI,CAAC,OAAA,EAAS;AACZ,MAAA,MAAM,IAAIC,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,wBAAwB,CAAA;AAAA,IAClE;AAEA,IAAA,MAAM,OAAA,GAAU,OAAO,UAAA,EAAW;AAClC,IAAA,IAAI,CAAC,OAAA,EAAS;AACZ,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,4BAA4B,CAAA;AAAA,IACtE;AAEA,IAAA,MAAM,KAAA,GAAQ,MAAM,OAAA,CAAQ,UAAA,CAAW,OAAO,CAAA;AAE9C,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,MAAM,IAAIA,gCAAc,GAAA,EAAK,EAAE,SAAS,CAAA,eAAA,EAAkB,OAAO,eAAe,CAAA;AAAA,IAClF;AAEA,IAAA,OAAO,KAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAAC,6BAAA,CAAY,OAAO,wBAAwB,CAAA;AAAA,EAC7C;AACF;AAMA,eAAsB,2BAAA,CAA4B,EAAE,MAAA,EAAQ,IAAA,EAAK,EAAyB;AACxF,EAAA,IAAI;AACF,IAAA,MAAM,OAAA,GAAU,OAAO,UAAA,EAAW;AAClC,IAAA,IAAI,CAAC,OAAA,EAAS;AACZ,MAAA,MAAM,IAAID,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,4BAA4B,CAAA;AAAA,IACtE;AAEA,IAAA,IAAI,CAAC,IAAA,EAAM;AACT,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,4BAA4B,CAAA;AAAA,IACtE;AAEA,IAAA,MAAM,EAAE,OAAA,EAAS,UAAA,EAAW,GAAI,IAAA;AAEhC,IAAA,IAAI,UAAA,EAAY,IAAA,IAAQ,UAAA,CAAW,IAAA,GAAO,CAAA,EAAG;AAC3C,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,uCAAuC,CAAA;AAAA,IACjF;AAEA,IAAA,IAAI,UAAA,EAAY,OAAA,IAAW,UAAA,CAAW,OAAA,GAAU,CAAA,EAAG;AACjD,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,2CAA2C,CAAA;AAAA,IACrF;AAEA,IAAA,IAAI,YAAY,SAAA,EAAW;AACzB,MAAA,MAAM,EAAE,KAAA,EAAO,GAAA,EAAI,GAAI,UAAA,CAAW,SAAA;AAElC,MAAA,IAAI,KAAA,IAAS,EAAE,KAAA,YAAiB,IAAA,CAAA,EAAO;AACrC,QAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,qCAAqC,CAAA;AAAA,MAC/E;AAEA,MAAA,IAAI,GAAA,IAAO,EAAE,GAAA,YAAe,IAAA,CAAA,EAAO;AACjC,QAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,qCAAqC,CAAA;AAAA,MAC/E;AAAA,IACF;AAEA,IAAA,OAAO,QAAQ,oBAAA,CAAqB;AAAA,MAClC,UAAA;AAAA,MACA;AAAA,KACD,CAAA;AAAA,EACH,SAAS,KAAA,EAAO;AACd,IAAAC,6BAAA,CAAY,OAAO,mCAAmC,CAAA;AAAA,EACxD;AACF;AAMA,eAAsB,kBAAA,CAAmB,EAAE,MAAA,EAAQ,IAAA,EAAK,EAAuB;AAC7E,EAAA,IAAI;AACF,IAAA,IAAI,CAAC,IAAA,EAAM;AACT,MAAA,MAAM,IAAID,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,4BAA4B,CAAA;AAAA,IACtE;AAEA,IAAA,MAAM,EAAE,UAAA,EAAY,OAAA,EAAQ,GAAI,IAAA;AAEhC,IAAA,IAAI,CAAC,UAAA,EAAY;AACf,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,2BAA2B,CAAA;AAAA,IACrE;AAEA,IAAA,IAAI,CAAC,OAAA,IAAW,OAAA,CAAQ,MAAA,KAAW,CAAA,EAAG;AACpC,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,mCAAmC,CAAA;AAAA,IAC7E;AAEA,IAAA,MAAM,OAAA,GAAU,OAAO,UAAA,EAAW;AAClC,IAAA,IAAI,CAAC,OAAA,EAAS;AACZ,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,4BAA4B,CAAA;AAAA,IACtE;AAEA,IAAA,MAAM,MAAA,GAAS,MAAA,CAAO,eAAA,CAAgB,UAAU,CAAA;AAChD,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA,MAAM,IAAIA,gCAAc,GAAA,EAAK,EAAE,SAAS,CAAA,QAAA,EAAW,UAAU,eAAe,CAAA;AAAA,IAC9E;AAEA,IAAA,MAAM,MAAA,GAAS,OAAO,SAAA,EAAU;AAChC,IAAAE,uBAAA,CAAY;AAAA,MACV,UAAA;AAAA,MACA,OAAA;AAAA,MACA;AAAA,KACD,CAAA,CAAE,KAAA,CAAM,CAAA,KAAA,KAAS;AAChB,MAAA,MAAA,EAAQ,KAAA,CAAM,CAAA,iCAAA,EAAoC,KAAA,CAAM,OAAO,IAAI,KAAK,CAAA;AAAA,IAC1E,CAAC,CAAA;AAGD,IAAA,OAAO;AAAA,MACL,MAAA,EAAQ,SAAA;AAAA,MACR,OAAA,EAAS,uBAAuB,OAAA,CAAQ,MAAM,IAAI,OAAA,CAAQ,MAAA,KAAW,CAAA,GAAI,OAAA,GAAU,QAAQ,CAAA,CAAA;AAAA,MAC3F,YAAY,OAAA,CAAQ;AAAA,KACtB;AAAA,EACF,SAAS,KAAA,EAAO;AACd,IAAAD,6BAAA,CAAY,OAAO,gCAAgC,CAAA;AAAA,EACrD;AACF;AAEA,eAAsB,eAAA,CAAgB;AAAA,EACpC,MAAA;AAAA,EACA,OAAA;AAAA,EACA,MAAA;AAAA,EACA;AACF,CAAA,EAAiF;AAC/E,EAAA,IAAI;AACF,IAAA,MAAM,OAAA,GAAU,OAAO,UAAA,EAAW;AAClC,IAAA,IAAI,CAAC,OAAA,EAAS;AACZ,MAAA,MAAM,IAAID,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,4BAA4B,CAAA;AAAA,IACtE;AAEA,IAAA,IAAI,CAAC,OAAA,IAAW,CAAC,MAAA,EAAQ;AACvB,MAAA,MAAM,IAAIA,+BAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,qCAAqC,CAAA;AAAA,IAC/E;AAEA,IAAA,OAAO,MAAM,OAAA,CAAQ,eAAA,CAAgB,EAAE,OAAA,EAAS,MAAA,EAAQ,YAAY,CAAA;AAAA,EACtE,SAAS,KAAA,EAAO;AACd,IAAA,OAAOC,6BAAA,CAAY,OAAO,8BAA8B,CAAA;AAAA,EAC1D;AACF","file":"chunk-2S4IMB6E.cjs","sourcesContent":["import type { AITracesPaginatedArg, StoragePagination } from '@mastra/core';\nimport { scoreTraces } from '@mastra/core/scores/scoreTraces';\nimport { HTTPException } from '../http-exception';\nimport type { Context } from '../types';\nimport { handleError } from './error';\n\ninterface ObservabilityContext extends Context {\n  traceId?: string;\n  body?: AITracesPaginatedArg;\n}\n\ninterface ScoreTracesContext extends Context {\n  body?: {\n    scorerName: string;\n    targets: Array<{\n      traceId: string;\n      spanId?: string;\n    }>;\n  };\n}\n\n/**\n * Get a complete AI trace by trace ID\n * Returns all spans in the trace with their parent-child relationships\n */\nexport async function getAITraceHandler({ mastra, traceId }: ObservabilityContext & { traceId: string }) {\n  try {\n    if (!traceId) {\n      throw new HTTPException(400, { message: 'Trace ID is required' });\n    }\n\n    const storage = mastra.getStorage();\n    if (!storage) {\n      throw new HTTPException(500, { message: 'Storage is not available' });\n    }\n\n    const trace = await storage.getAITrace(traceId);\n\n    if (!trace) {\n      throw new HTTPException(404, { message: `Trace with ID '${traceId}' not found` });\n    }\n\n    return trace;\n  } catch (error) {\n    handleError(error, 'Error getting AI trace');\n  }\n}\n\n/**\n * Get paginated AI traces with filtering and pagination\n * Returns only root spans (parent spans) for pagination, not child spans\n */\nexport async function getAITracesPaginatedHandler({ mastra, body }: ObservabilityContext) {\n  try {\n    const storage = mastra.getStorage();\n    if (!storage) {\n      throw new HTTPException(500, { message: 'Storage is not available' });\n    }\n\n    if (!body) {\n      throw new HTTPException(400, { message: 'Request body is required' });\n    }\n\n    const { filters, pagination } = body;\n\n    if (pagination?.page && pagination.page < 0) {\n      throw new HTTPException(400, { message: 'Page must be a non-negative integer' });\n    }\n\n    if (pagination?.perPage && pagination.perPage < 0) {\n      throw new HTTPException(400, { message: 'Per page must be a non-negative integer' });\n    }\n\n    if (pagination?.dateRange) {\n      const { start, end } = pagination.dateRange;\n\n      if (start && !(start instanceof Date)) {\n        throw new HTTPException(400, { message: 'Invalid date format in date range' });\n      }\n\n      if (end && !(end instanceof Date)) {\n        throw new HTTPException(400, { message: 'Invalid date format in date range' });\n      }\n    }\n\n    return storage.getAITracesPaginated({\n      pagination,\n      filters,\n    });\n  } catch (error) {\n    handleError(error, 'Error getting AI traces paginated');\n  }\n}\n\n/**\n * Score traces using a specified scorer\n * Fire-and-forget approach - returns immediately while scoring runs in background\n */\nexport async function scoreTracesHandler({ mastra, body }: ScoreTracesContext) {\n  try {\n    if (!body) {\n      throw new HTTPException(400, { message: 'Request body is required' });\n    }\n\n    const { scorerName, targets } = body;\n\n    if (!scorerName) {\n      throw new HTTPException(400, { message: 'Scorer Name is required' });\n    }\n\n    if (!targets || targets.length === 0) {\n      throw new HTTPException(400, { message: 'At least one target is required' });\n    }\n\n    const storage = mastra.getStorage();\n    if (!storage) {\n      throw new HTTPException(500, { message: 'Storage is not available' });\n    }\n\n    const scorer = mastra.getScorerByName(scorerName);\n    if (!scorer) {\n      throw new HTTPException(404, { message: `Scorer '${scorerName}' not found` });\n    }\n\n    const logger = mastra.getLogger();\n    scoreTraces({\n      scorerName,\n      targets,\n      mastra,\n    }).catch(error => {\n      logger?.error(`Background trace scoring failed: ${error.message}`, error);\n    });\n\n    // Return immediate response\n    return {\n      status: 'success',\n      message: `Scoring started for ${targets.length} ${targets.length === 1 ? 'trace' : 'traces'}`,\n      traceCount: targets.length,\n    };\n  } catch (error) {\n    handleError(error, 'Error processing trace scoring');\n  }\n}\n\nexport async function getScoresBySpan({\n  mastra,\n  traceId,\n  spanId,\n  pagination,\n}: Context & { traceId: string; spanId: string; pagination: StoragePagination }) {\n  try {\n    const storage = mastra.getStorage();\n    if (!storage) {\n      throw new HTTPException(500, { message: 'Storage is not available' });\n    }\n\n    if (!traceId || !spanId) {\n      throw new HTTPException(400, { message: 'Trace ID and span ID are required' });\n    }\n\n    return await storage.getScoresBySpan({ traceId, spanId, pagination });\n  } catch (error) {\n    return handleError(error, 'Error getting scores by span');\n  }\n}\n"]}

package/dist/{chunk-HRG5HN4U.js → chunk-3PXWQX3J.js}

@@ -1,8 +1,8 @@
 import { sanitizeBody, validateBody } from './chunk-4QCXUEAT.js';
-import { stringify } from './chunk-LF2ZLOFP.js';
+import { stringify } from './chunk-GQRJOUV7.js';
 import { handleError } from './chunk-CY4TP3FK.js';
 import { HTTPException } from './chunk-MMROOK5J.js';
-import { __export } from './chunk-G3PMV62Z.js';
+import { __export } from './chunk-PR4QN5HX.js';
 import { PROVIDER_REGISTRY } from '@mastra/core/llm';
 import { RuntimeContext } from '@mastra/core/runtime-context';
 import { zodToJsonSchema } from '@mastra/core/utils/zod-to-json';
@@ -15,11 +15,13 @@ __export(agents_exports, {
   generateHandler: () => generateHandler,
   generateLegacyHandler: () => generateLegacyHandler,
   getAgentByIdHandler: () => getAgentByIdHandler,
+  getAgentFromSystem: () => getAgentFromSystem,
   getAgentsHandler: () => getAgentsHandler,
   getEvalsByAgentIdHandler: () => getEvalsByAgentIdHandler,
   getLiveEvalsByAgentIdHandler: () => getLiveEvalsByAgentIdHandler,
   getProvidersHandler: () => getProvidersHandler,
   getSerializedAgentTools: () => getSerializedAgentTools,
+  getSerializedProcessors: () => getSerializedProcessors,
   reorderAgentModelListHandler: () => reorderAgentModelListHandler,
   streamGenerateHandler: () => streamGenerateHandler,
   streamGenerateLegacyHandler: () => streamGenerateLegacyHandler,
@@ -33,20 +35,39 @@ async function getSerializedAgentTools(tools) {
     const _tool = tool;
     const toolId = _tool.id ?? `tool-${key}`;
     let inputSchemaForReturn = void 0;
-    if (_tool.inputSchema) {
-      if (_tool.inputSchema?.jsonSchema) {
-        inputSchemaForReturn = stringify(_tool.inputSchema.jsonSchema);
-      } else {
-        inputSchemaForReturn = stringify(zodToJsonSchema(_tool.inputSchema));
-      }
-    }
     let outputSchemaForReturn = void 0;
-    if (_tool.outputSchema) {
-      if (_tool.outputSchema?.jsonSchema) {
-        outputSchemaForReturn = stringify(_tool.outputSchema.jsonSchema);
-      } else {
-        outputSchemaForReturn = stringify(zodToJsonSchema(_tool.outputSchema));
+    try {
+      if (_tool.inputSchema) {
+        if (_tool.inputSchema && typeof _tool.inputSchema === "object" && "jsonSchema" in _tool.inputSchema) {
+          inputSchemaForReturn = stringify(_tool.inputSchema.jsonSchema);
+        } else if (typeof _tool.inputSchema === "function") {
+          const inputSchema = _tool.inputSchema();
+          if (inputSchema && inputSchema.jsonSchema) {
+            inputSchemaForReturn = stringify(inputSchema.jsonSchema);
+          }
+        } else if (_tool.inputSchema) {
+          inputSchemaForReturn = stringify(zodToJsonSchema(_tool.inputSchema));
+        }
       }
+      if (_tool.outputSchema) {
+        if (_tool.outputSchema && typeof _tool.outputSchema === "object" && "jsonSchema" in _tool.outputSchema) {
+          outputSchemaForReturn = stringify(_tool.outputSchema.jsonSchema);
+        } else if (typeof _tool.outputSchema === "function") {
+          const outputSchema = _tool.outputSchema();
+          if (outputSchema && outputSchema.jsonSchema) {
+            outputSchemaForReturn = stringify(outputSchema.jsonSchema);
+          }
+        } else if (_tool.outputSchema) {
+          outputSchemaForReturn = stringify(
+            zodToJsonSchema(_tool.outputSchema)
+          );
+        }
+      }
+    } catch (error) {
+      console.error(`Error getting serialized tool`, {
+        toolId: _tool.id,
+        error
+      });
     }
     acc[key] = {
       ..._tool,
@@ -57,6 +78,13 @@ async function getSerializedAgentTools(tools) {
     return acc;
   }, {});
 }
+function getSerializedProcessors(processors) {
+  return processors.map((processor) => {
+    return {
+      name: processor.name || processor.constructor.name
+    };
+  });
+}
 async function getSerializedAgentDefinition({
   agent,
   runtimeContext
@@ -64,12 +92,15 @@ async function getSerializedAgentDefinition({
   let serializedAgentAgents = {};
   if ("listAgents" in agent) {
     const agents = await agent.listAgents({ runtimeContext });
-    serializedAgentAgents = Object.entries(agents || {}).reduce((acc, [key, agent2]) => {
-      return {
-        ...acc,
-        [key]: { id: agent2.id, name: agent2.name }
-      };
-    }, {});
+    serializedAgentAgents = Object.entries(agents || {}).reduce(
+      (acc, [key, agent2]) => {
+        return {
+          ...acc,
+          [key]: { id: agent2.id, name: agent2.name }
+        };
+      },
+      {}
+    );
   }
   return serializedAgentAgents;
 }
@@ -82,8 +113,8 @@ async function formatAgentList({
   const instructions = await agent.getInstructions({ runtimeContext });
   const tools = await agent.getTools({ runtimeContext });
   const llm = await agent.getLLM({ runtimeContext });
-  const defaultGenerateOptions = await agent.getDefaultGenerateOptions({ runtimeContext });
-  const defaultStreamOptions = await agent.getDefaultStreamOptions({ runtimeContext });
+  const defaultGenerateOptions = await agent.getDefaultGenerateOptionsLegacy({ runtimeContext });
+  const defaultStreamOptions = await agent.getDefaultStreamOptionsLegacy({ runtimeContext });
   const serializedAgentTools = await getSerializedAgentTools(tools);
   let serializedAgentWorkflows = {};
   if ("getWorkflows" in agent) {
@@ -94,7 +125,7 @@ async function formatAgentList({
         return {
           ...acc,
           [key]: {
-            name: workflow.name
+            name: workflow.name || "Unnamed workflow"
           }
         };
       }, {});
@@ -103,6 +134,10 @@ async function formatAgentList({
     }
   }
   const serializedAgentAgents = await getSerializedAgentDefinition({ agent, runtimeContext });
+  const inputProcessors = await agent.getInputProcessors(runtimeContext);
+  const outputProcessors = await agent.getOutputProcessors(runtimeContext);
+  const serializedInputProcessors = getSerializedProcessors(inputProcessors);
+  const serializedOutputProcessors = getSerializedProcessors(outputProcessors);
   const model = llm?.getModel();
   const models = await agent.getModelList(runtimeContext);
   const modelList = models?.map((md) => ({
@@ -120,6 +155,8 @@ async function formatAgentList({
     agents: serializedAgentAgents,
     tools: serializedAgentTools,
     workflows: serializedAgentWorkflows,
+    inputProcessors: serializedInputProcessors,
+    outputProcessors: serializedOutputProcessors,
     provider: llm?.getProvider(),
     modelId: llm?.getModelId(),
     modelVersion: model?.specificationVersion,
@@ -128,7 +165,43 @@ async function formatAgentList({
     modelList
   };
 }
-async function getAgentsHandler({ mastra, runtimeContext }) {
+async function getAgentFromSystem({ mastra, agentId }) {
+  const logger = mastra.getLogger();
+  if (!agentId) {
+    throw new HTTPException(400, { message: "Agent ID is required" });
+  }
+  let agent;
+  try {
+    agent = mastra.getAgent(agentId);
+  } catch (error) {
+    logger.debug("Error getting agent from mastra, searching agents for agent", error);
+  }
+  if (!agent) {
+    logger.debug(`Agent ${agentId} not found, looking through sub-agents`);
+    const agents = mastra.getAgents();
+    if (Object.keys(agents || {}).length) {
+      for (const [_, ag] of Object.entries(agents)) {
+        try {
+          const subAgents = await ag.listAgents();
+          if (subAgents[agentId]) {
+            agent = subAgents[agentId];
+            break;
+          }
+        } catch (error) {
+          logger.debug("Error getting agent from agent", error);
+        }
+      }
+    }
+  }
+  if (!agent) {
+    throw new HTTPException(404, { message: `Agent with name ${agentId} not found` });
+  }
+  return agent;
+}
+async function getAgentsHandler({
+  mastra,
+  runtimeContext
+}) {
   try {
     const agents = mastra.getAgents();
     const serializedAgentsMap = await Promise.all(
@@ -162,65 +235,74 @@ async function formatAgent({
         return {
           ...acc,
           [key]: {
-            name: workflow.name,
-            steps: Object.entries(workflow.steps).reduce((acc2, [key2, step]) => {
-              return {
-                ...acc2,
-                [key2]: {
-                  id: step.id,
-                  description: step.description
-                }
-              };
-            }, {})
+            name: workflow.name || "Unnamed workflow",
+            steps: Object.entries(workflow.steps).reduce(
+              (acc2, [key2, step]) => {
+                return {
+                  ...acc2,
+                  [key2]: {
+                    id: step.id,
+                    description: step.description
+                  }
+                };
+              },
+              {}
+            )
           }
         };
       }, {});
     } catch (error) {
       logger.error("Error getting workflows for agent", { agentName: agent.name, error });
     }
-    let proxyRuntimeContext = runtimeContext;
-    if (isPlayground) {
-      proxyRuntimeContext = new Proxy(runtimeContext, {
-        get(target, prop) {
-          if (prop === "get") {
-            return function(key) {
-              const value = target.get(key);
-              return value ?? `<${key}>`;
-            };
-          }
-          return Reflect.get(target, prop);
+  }
+  let proxyRuntimeContext = runtimeContext;
+  if (isPlayground) {
+    proxyRuntimeContext = new Proxy(runtimeContext, {
+      get(target, prop) {
+        if (prop === "get") {
+          return function(key) {
+            const value = target.get(key);
+            return value ?? `<${key}>`;
+          };
         }
-      });
-    }
-    const instructions = await agent.getInstructions({ runtimeContext: proxyRuntimeContext });
-    const llm = await agent.getLLM({ runtimeContext });
-    const defaultGenerateOptions = await agent.getDefaultGenerateOptions({ runtimeContext: proxyRuntimeContext });
-    const defaultStreamOptions = await agent.getDefaultStreamOptions({ runtimeContext: proxyRuntimeContext });
-    const model = llm?.getModel();
-    const models = await agent.getModelList(runtimeContext);
-    const modelList = models?.map((md) => ({
-      ...md,
-      model: {
-        modelId: md.model.modelId,
-        provider: md.model.provider,
-        modelVersion: md.model.specificationVersion
+        return Reflect.get(target, prop);
       }
-    }));
-    const serializedAgentAgents = await getSerializedAgentDefinition({ agent, runtimeContext: proxyRuntimeContext });
-    return {
-      name: agent.name,
-      instructions,
-      tools: serializedAgentTools,
-      agents: serializedAgentAgents,
-      workflows: serializedAgentWorkflows,
-      provider: llm?.getProvider(),
-      modelId: llm?.getModelId(),
-      modelVersion: model?.specificationVersion,
-      modelList,
-      defaultGenerateOptions,
-      defaultStreamOptions
-    };
+    });
   }
+  const instructions = await agent.getInstructions({ runtimeContext: proxyRuntimeContext });
+  const llm = await agent.getLLM({ runtimeContext });
+  const defaultGenerateOptions = await agent.getDefaultGenerateOptionsLegacy({ runtimeContext: proxyRuntimeContext });
+  const defaultStreamOptions = await agent.getDefaultStreamOptionsLegacy({ runtimeContext: proxyRuntimeContext });
+  const model = llm?.getModel();
+  const models = await agent.getModelList(runtimeContext);
+  const modelList = models?.map((md) => ({
+    ...md,
+    model: {
+      modelId: md.model.modelId,
+      provider: md.model.provider,
+      modelVersion: md.model.specificationVersion
+    }
+  }));
+  const serializedAgentAgents = await getSerializedAgentDefinition({ agent, runtimeContext: proxyRuntimeContext });
+  const inputProcessors = await agent.getInputProcessors(proxyRuntimeContext);
+  const outputProcessors = await agent.getOutputProcessors(proxyRuntimeContext);
+  const serializedInputProcessors = getSerializedProcessors(inputProcessors);
+  const serializedOutputProcessors = getSerializedProcessors(outputProcessors);
+  return {
+    name: agent.name,
+    instructions,
+    tools: serializedAgentTools,
+    agents: serializedAgentAgents,
+    workflows: serializedAgentWorkflows,
+    inputProcessors: serializedInputProcessors,
+    outputProcessors: serializedOutputProcessors,
+    provider: llm?.getProvider(),
+    modelId: llm?.getModelId(),
+    modelVersion: model?.specificationVersion,
+    modelList,
+    defaultGenerateOptions,
+    defaultStreamOptions
+  };
 }
 async function getAgentByIdHandler({
   mastra,
@@ -229,10 +311,7 @@ async function getAgentByIdHandler({
   isPlayground = false
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     return formatAgent({ mastra, agent, runtimeContext, isPlayground });
   } catch (error) {
     return handleError(error, "Error getting agent");
@@ -244,7 +323,7 @@ async function getEvalsByAgentIdHandler({
   agentId
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
+    const agent = await getAgentFromSystem({ mastra, agentId });
     const evals = await mastra.getStorage()?.getEvalsByAgentName?.(agent.name, "test") || [];
     const instructions = await agent.getInstructions({ runtimeContext });
     return {
@@ -263,7 +342,7 @@ async function getLiveEvalsByAgentIdHandler({
   agentId
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
+    const agent = await getAgentFromSystem({ mastra, agentId });
     const evals = await mastra.getStorage()?.getEvalsByAgentName?.(agent.name, "live") || [];
     const instructions = await agent.getInstructions({ runtimeContext });
     return {
@@ -284,10 +363,7 @@ async function generateLegacyHandler({
   abortSignal
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     sanitizeBody(body, ["tools"]);
     const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalResourceId = resourceId ?? resourceid;
@@ -316,10 +392,7 @@ async function generateHandler({
   abortSignal
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     sanitizeBody(body, ["tools"]);
     const { messages, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalRuntimeContext = new RuntimeContext([
@@ -346,10 +419,7 @@ async function streamGenerateLegacyHandler({
   abortSignal
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalResourceId = resourceId ?? resourceid;
     const finalRuntimeContext = new RuntimeContext([
@@ -383,7 +453,7 @@ async function streamGenerateLegacyHandler({
     return handleError(error, "error streaming agent response");
   }
 }
-function streamGenerateHandler({
+async function streamGenerateHandler({
   mastra,
   runtimeContext,
   agentId,
@@ -391,10 +461,7 @@ function streamGenerateHandler({
   abortSignal
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     sanitizeBody(body, ["tools"]);
     const { messages, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalRuntimeContext = new RuntimeContext([
@@ -413,7 +480,7 @@ function streamGenerateHandler({
     return handleError(error, "error streaming agent response");
   }
 }
-function approveToolCallHandler({
+async function approveToolCallHandler({
   mastra,
   runtimeContext,
   agentId,
@@ -421,9 +488,12 @@ function approveToolCallHandler({
   abortSignal
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    if (!body.runId) {
+      throw new HTTPException(400, { message: "Run id is required" });
+    }
+    if (!body.toolCallId) {
+      throw new HTTPException(400, { message: "Tool call id is required" });
     }
     sanitizeBody(body, ["tools"]);
     const { runId, runtimeContext: agentRuntimeContext, ...rest } = body;
@@ -443,7 +513,7 @@ function approveToolCallHandler({
     return handleError(error, "error streaming agent response");
   }
 }
-function declineToolCallHandler({
+async function declineToolCallHandler({
   mastra,
   runtimeContext,
   agentId,
@@ -451,9 +521,12 @@ function declineToolCallHandler({
   abortSignal
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    if (!body.runId) {
+      throw new HTTPException(400, { message: "Run id is required" });
+    }
+    if (!body.toolCallId) {
+      throw new HTTPException(400, { message: "Tool call id is required" });
     }
     sanitizeBody(body, ["tools"]);
     const { runId, runtimeContext: agentRuntimeContext, ...rest } = body;
@@ -473,7 +546,7 @@ function declineToolCallHandler({
     return handleError(error, "error streaming agent response");
   }
 }
-function streamNetworkHandler({
+async function streamNetworkHandler({
   mastra,
   runtimeContext,
   agentId,
@@ -481,10 +554,7 @@ function streamNetworkHandler({
   // abortSignal,
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     sanitizeBody(body, ["tools"]);
     const { messages, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalRuntimeContext = new RuntimeContext([
@@ -496,7 +566,8 @@ function streamNetworkHandler({
       ...rest,
       memory: {
         thread: rest.thread ?? "",
-        resource: rest.resourceId ?? ""
+        resource: rest.resourceId ?? "",
+        options: rest.memory?.options ?? {}
       },
       runtimeContext: finalRuntimeContext
     });
@@ -513,10 +584,7 @@ async function streamUIMessageHandler({
   abortSignal
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     sanitizeBody(body, ["tools"]);
     const { messages, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalRuntimeContext = new RuntimeContext([
@@ -541,10 +609,7 @@ async function updateAgentModelHandler({
   body
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     const { modelId, provider } = body;
     const newModel = `${provider}/${modelId}`;
     agent.__updateModel({ model: newModel });
@@ -559,10 +624,7 @@ async function reorderAgentModelListHandler({
   body
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     const modelList = await agent.getModelList();
     if (!modelList || modelList.length === 0) {
       throw new HTTPException(400, { message: "Agent model list is not found or empty" });
@@ -580,10 +642,7 @@ async function updateAgentModelInModelListHandler({
   body
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     const { model: bodyModel, maxRetries, enabled } = body;
     if (!modelConfigId) {
       throw new HTTPException(400, { message: "Model id is required" });
@@ -628,6 +687,6 @@ async function getProvidersHandler() {
   }
 }
 
-export { agents_exports, approveToolCallHandler, declineToolCallHandler, generateHandler, generateLegacyHandler, getAgentByIdHandler, getAgentsHandler, getEvalsByAgentIdHandler, getLiveEvalsByAgentIdHandler, getProvidersHandler, getSerializedAgentTools, reorderAgentModelListHandler, streamGenerateHandler, streamGenerateLegacyHandler, streamNetworkHandler, streamUIMessageHandler, updateAgentModelHandler, updateAgentModelInModelListHandler };
-//# sourceMappingURL=chunk-HRG5HN4U.js.map
-//# sourceMappingURL=chunk-HRG5HN4U.js.map
+export { agents_exports, approveToolCallHandler, declineToolCallHandler, generateHandler, generateLegacyHandler, getAgentByIdHandler, getAgentFromSystem, getAgentsHandler, getEvalsByAgentIdHandler, getLiveEvalsByAgentIdHandler, getProvidersHandler, getSerializedAgentTools, getSerializedProcessors, reorderAgentModelListHandler, streamGenerateHandler, streamGenerateLegacyHandler, streamNetworkHandler, streamUIMessageHandler, updateAgentModelHandler, updateAgentModelInModelListHandler };
+//# sourceMappingURL=chunk-3PXWQX3J.js.map
+//# sourceMappingURL=chunk-3PXWQX3J.js.map