@mastra/deployer 0.2.5 → 0.2.6-alpha.1

package/dist/server/index.js

@@ -16,6 +16,7 @@ import { logger } from 'hono/logger';
 import { HTTPException } from 'hono/http-exception';
 import { z, ZodFirstPartyTypeKind, ZodOptional } from 'zod';
 import { Agent } from '@mastra/core/agent';
+import { TABLE_TRACES } from '@mastra/core/storage';
 import { isVercelTool } from '@mastra/core';
 import { streamText } from 'hono/streaming';
 
@@ -170,11 +171,11 @@ function writeFromReadableStream(stream, writable) {
     writable.off("close", cancel);
     writable.off("error", cancel);
   });
-  function cancel(error) {
-    reader.cancel(error).catch(() => {
+  function cancel(error2) {
+    reader.cancel(error2).catch(() => {
     });
-    if (error) {
-      writable.destroy(error);
+    if (error2) {
+      writable.destroy(error2);
     }
   }
   function onDrain() {
@@ -2830,9 +2831,9 @@ var zodToJsonSchema = (schema, options) => {
 
 // ../../node_modules/.pnpm/zod-to-json-schema@3.24.4_zod@3.24.2/node_modules/zod-to-json-schema/dist/esm/index.js
 var esm_default = zodToJsonSchema;
-function handleError(error, defaultMessage) {
-  console.error(defaultMessage, error);
-  const apiError = error;
+function handleError(error2, defaultMessage) {
+  console.error(defaultMessage, error2);
+  const apiError = error2;
   throw new HTTPException(apiError.status || 500, {
     message: apiError.message || defaultMessage
   });
@@ -2882,8 +2883,8 @@ async function getAgentsHandler(c2) {
       return acc;
     }, {});
     return c2.json(serializedAgents);
-  } catch (error) {
-    return handleError(error, "Error getting agents");
+  } catch (error2) {
+    return handleError(error2, "Error getting agents");
   }
 }
 async function getAgentByIdHandler(c2) {
@@ -2910,8 +2911,8 @@ async function getAgentByIdHandler(c2) {
       provider: agent.llm?.getProvider(),
       modelId: agent.llm?.getModelId()
     });
-  } catch (error) {
-    return handleError(error, "Error getting agent");
+  } catch (error2) {
+    return handleError(error2, "Error getting agent");
   }
 }
 async function getEvalsByAgentIdHandler(c2) {
@@ -2926,8 +2927,8 @@ async function getEvalsByAgentIdHandler(c2) {
       instructions: agent.instructions,
       evals
     });
-  } catch (error) {
-    return handleError(error, "Error getting test evals");
+  } catch (error2) {
+    return handleError(error2, "Error getting test evals");
   }
 }
 async function getLiveEvalsByAgentIdHandler(c2) {
@@ -2942,8 +2943,8 @@ async function getLiveEvalsByAgentIdHandler(c2) {
       instructions: agent.instructions,
       evals
     });
-  } catch (error) {
-    return handleError(error, "Error getting live evals");
+  } catch (error2) {
+    return handleError(error2, "Error getting live evals");
   }
 }
 async function generateHandler(c2) {
@@ -2959,8 +2960,8 @@ async function generateHandler(c2) {
     const finalResourceId = resourceId ?? resourceid;
     const result = await agent.generate(messages, { threadId, resourceId: finalResourceId, output, runId, ...rest });
     return c2.json(result);
-  } catch (error) {
-    return handleError(error, "Error generating from agent");
+  } catch (error2) {
+    return handleError(error2, "Error generating from agent");
   }
 }
 async function streamGenerateHandler(c2) {
@@ -2984,13 +2985,13 @@ async function streamGenerateHandler(c2) {
     const streamResponse = output ? streamResult.toTextStreamResponse() : streamResult.toDataStreamResponse({
       sendUsage: true,
       sendReasoning: true,
-      getErrorMessage: (error) => {
-        return `An error occurred while processing your request. ${error instanceof Error ? error.message : JSON.stringify(error)}`;
+      getErrorMessage: (error2) => {
+        return `An error occurred while processing your request. ${error2 instanceof Error ? error2.message : JSON.stringify(error2)}`;
       }
     });
     return streamResponse;
-  } catch (error) {
-    return handleError(error, "Error streaming from agent");
+  } catch (error2) {
+    return handleError(error2, "Error streaming from agent");
   }
 }
 async function setAgentInstructionsHandler(c2) {
@@ -3016,8 +3017,8 @@ async function setAgentInstructionsHandler(c2) {
       },
       200
     );
-  } catch (error) {
-    return handleError(error, "Error setting agent instructions");
+  } catch (error2) {
+    return handleError(error2, "Error setting agent instructions");
   }
 }
 
@@ -3061,8 +3062,8 @@ async function getLogsHandler(c2) {
     }
     const logs = await mastra.getLogs(transportId);
     return c2.json(logs);
-  } catch (error) {
-    return handleError(error, "Error getting logs");
+  } catch (error2) {
+    return handleError(error2, "Error getting logs");
   }
 }
 async function getLogsByRunIdHandler(c2) {
@@ -3075,8 +3076,8 @@ async function getLogsByRunIdHandler(c2) {
     }
     const logs = await mastra.getLogsByRunId({ runId, transportId });
     return c2.json(logs);
-  } catch (error) {
-    return handleError(error, "Error getting logs by run ID");
+  } catch (error2) {
+    return handleError(error2, "Error getting logs by run ID");
   }
 }
 async function getLogTransports(c2) {
@@ -3108,8 +3109,8 @@ async function getMemoryStatusHandler(c2) {
       return c2.json({ result: false });
     }
     return c2.json({ result: true });
-  } catch (error) {
-    return handleError(error, "Error getting memory status");
+  } catch (error2) {
+    return handleError(error2, "Error getting memory status");
   }
 }
 async function getThreadsHandler(c2) {
@@ -3124,8 +3125,8 @@ async function getThreadsHandler(c2) {
     }
     const threads = await memory.getThreadsByResourceId({ resourceId: resourceid });
     return c2.json(threads);
-  } catch (error) {
-    return handleError(error, "Error getting threads");
+  } catch (error2) {
+    return handleError(error2, "Error getting threads");
   }
 }
 async function getThreadByIdHandler(c2) {
@@ -3140,8 +3141,8 @@ async function getThreadByIdHandler(c2) {
       throw new HTTPException(404, { message: "Thread not found" });
     }
     return c2.json(thread);
-  } catch (error) {
-    return handleError(error, "Error getting thread");
+  } catch (error2) {
+    return handleError(error2, "Error getting thread");
   }
 }
 async function saveMessagesHandler(c2) {
@@ -3162,8 +3163,8 @@ async function saveMessagesHandler(c2) {
     }));
     const result = await memory.saveMessages({ messages: processedMessages, memoryConfig: {} });
     return c2.json(result);
-  } catch (error) {
-    return handleError(error, "Error saving messages");
+  } catch (error2) {
+    return handleError(error2, "Error saving messages");
   }
 }
 async function createThreadHandler(c2) {
@@ -3176,8 +3177,8 @@ async function createThreadHandler(c2) {
     validateBody({ resourceid });
     const result = await memory.createThread({ resourceId: resourceid, title, metadata, threadId });
     return c2.json(result);
-  } catch (error) {
-    return handleError(error, "Error saving thread to memory");
+  } catch (error2) {
+    return handleError(error2, "Error saving thread to memory");
   }
 }
 async function updateThreadHandler(c2) {
@@ -3203,8 +3204,8 @@ async function updateThreadHandler(c2) {
     };
     const result = await memory.saveThread({ thread: updatedThread });
     return c2.json(result);
-  } catch (error) {
-    return handleError(error, "Error updating thread");
+  } catch (error2) {
+    return handleError(error2, "Error updating thread");
   }
 }
 async function deleteThreadHandler(c2) {
@@ -3220,8 +3221,8 @@ async function deleteThreadHandler(c2) {
     }
     await memory.deleteThread(threadId);
     return c2.json({ result: "Thread deleted" });
-  } catch (error) {
-    return handleError(error, "Error deleting thread");
+  } catch (error2) {
+    return handleError(error2, "Error deleting thread");
   }
 }
 async function getMessagesHandler(c2) {
@@ -3237,8 +3238,8 @@ async function getMessagesHandler(c2) {
     }
     const result = await memory.query({ threadId });
     return c2.json(result);
-  } catch (error) {
-    return handleError(error, "Error getting messages");
+  } catch (error2) {
+    return handleError(error2, "Error getting messages");
   }
 }
 async function getNetworksHandler(c2) {
@@ -3264,8 +3265,8 @@ async function getNetworksHandler(c2) {
       };
     });
     return c2.json(serializedNetworks);
-  } catch (error) {
-    return handleError(error, "Error getting networks");
+  } catch (error2) {
+    return handleError(error2, "Error getting networks");
   }
 }
 async function getNetworkByIdHandler(c2) {
@@ -3297,8 +3298,8 @@ async function getNetworkByIdHandler(c2) {
       }
     };
     return c2.json(serializedNetwork);
-  } catch (error) {
-    return handleError(error, "Error getting network by ID");
+  } catch (error2) {
+    return handleError(error2, "Error getting network by ID");
   }
 }
 async function generateHandler2(c2) {
@@ -3314,8 +3315,8 @@ async function generateHandler2(c2) {
     const finalResourceId = resourceId ?? resourceid;
     const result = await network.generate(messages, { threadId, resourceId: finalResourceId, output, runId, ...rest });
     return c2.json(result);
-  } catch (error) {
-    return handleError(error, "Error generating from network");
+  } catch (error2) {
+    return handleError(error2, "Error generating from network");
   }
 }
 async function streamGenerateHandler2(c2) {
@@ -3339,13 +3340,13 @@ async function streamGenerateHandler2(c2) {
     const streamResponse = output ? streamResult.toTextStreamResponse() : streamResult.toDataStreamResponse({
       sendUsage: true,
       sendReasoning: true,
-      getErrorMessage: (error) => {
-        return `An error occurred while processing your request. ${error instanceof Error ? error.message : JSON.stringify(error)}`;
+      getErrorMessage: (error2) => {
+        return `An error occurred while processing your request. ${error2 instanceof Error ? error2.message : JSON.stringify(error2)}`;
       }
     });
     return streamResponse;
-  } catch (error) {
-    return handleError(error, "Error streaming from network");
+  } catch (error2) {
+    return handleError(error2, "Error streaming from network");
   }
 }
 async function generateSystemPromptHandler(c2) {
@@ -3381,8 +3382,8 @@ async function generateSystemPromptHandler(c2) {
 
         `
       ).join("");
-    } catch (error) {
-      mastra.getLogger().error(`Error fetching evals`, { error });
+    } catch (error2) {
+      mastra.getLogger().error(`Error fetching evals`, { error: error2 });
     }
     const ENHANCE_SYSTEM_PROMPT_INSTRUCTIONS = `
             You are an expert system prompt engineer, specialized in analyzing and enhancing instructions to create clear, effective, and comprehensive system prompts. Your goal is to help users transform their basic instructions into well-structured system prompts that will guide AI behavior effectively.
@@ -3453,8 +3454,8 @@ ${evalSummary}` : ""}
       }
     );
     return c2.json(result?.object || {});
-  } catch (error) {
-    return handleError(error, "Error generating system prompt");
+  } catch (error2) {
+    return handleError(error2, "Error generating system prompt");
   }
 }
 
@@ -3489,8 +3490,97 @@ async function getTelemetryHandler(c2) {
       attributes
     });
     return c2.json({ traces });
-  } catch (error) {
-    return handleError(error, "Error saving messages");
+  } catch (error2) {
+    return handleError(error2, "Error saving messages");
+  }
+}
+async function storeTelemetryHandler(c2) {
+  try {
+    const body = await c2.req.json();
+    const mastra = c2.get("mastra");
+    const storage = mastra.getStorage();
+    const now = /* @__PURE__ */ new Date();
+    const items = body?.resourceSpans?.[0]?.scopeSpans;
+    const allSpans = items.reduce((acc, scopedSpans) => {
+      const { scope, spans } = scopedSpans;
+      for (const span of spans) {
+        const {
+          spanId,
+          parentSpanId,
+          traceId,
+          name,
+          kind,
+          attributes,
+          status,
+          events,
+          links,
+          startTimeUnixNano,
+          endTimeUnixNano,
+          ...rest
+        } = span;
+        const startTime = Number(BigInt(startTimeUnixNano) / 1000n);
+        const endTime = Number(BigInt(endTimeUnixNano) / 1000n);
+        acc.push({
+          id: spanId,
+          parentSpanId,
+          traceId,
+          name,
+          scope: scope.name,
+          kind,
+          status: JSON.stringify(status),
+          events: JSON.stringify(events),
+          links: JSON.stringify(links),
+          attributes: JSON.stringify(
+            attributes.reduce((acc2, attr) => {
+              const valueKey = Object.keys(attr.value)[0];
+              if (valueKey) {
+                acc2[attr.key] = attr.value[valueKey];
+              }
+              return acc2;
+            }, {})
+          ),
+          startTime,
+          endTime,
+          other: JSON.stringify(rest),
+          createdAt: now
+        });
+      }
+      return acc;
+    }, []);
+    return storage.__batchInsert({
+      tableName: TABLE_TRACES,
+      records: allSpans
+    }).then(() => {
+      return c2.json(
+        {
+          status: "success",
+          message: "Traces received and processed successfully",
+          traceCount: body.resourceSpans?.length || 0
+        },
+        200
+      );
+    }).catch((e2) => {
+      return c2.json(
+        {
+          status: "error",
+          message: "Failed to process traces",
+          // @ts-ignore
+          error: error.message
+        },
+        500
+      );
+    });
+  } catch (error2) {
+    console.error("Error processing traces:", error2);
+    return c2.json(
+      {
+        status: "error",
+        message: "Failed to process traces",
+        // @ts-ignore
+        error: error2.message
+      },
+      500
+    );
   }
 }
 async function getToolsHandler(c2) {
@@ -3512,8 +3602,8 @@ async function getToolsHandler(c2) {
       {}
     );
     return c2.json(serializedTools);
-  } catch (error) {
-    return handleError(error, "Error getting tools");
+  } catch (error2) {
+    return handleError(error2, "Error getting tools");
   }
 }
 async function getToolByIdHandler(c2) {
@@ -3530,8 +3620,8 @@ async function getToolByIdHandler(c2) {
       outputSchema: tool.outputSchema ? stringify(esm_default(tool.outputSchema)) : void 0
     };
     return c2.json(serializedTool);
-  } catch (error) {
-    return handleError(error, "Error getting tool");
+  } catch (error2) {
+    return handleError(error2, "Error getting tool");
   }
 }
 function executeToolHandler(tools) {
@@ -3557,8 +3647,8 @@ function executeToolHandler(tools) {
         runId: mastra.runId
       });
       return c2.json(result);
-    } catch (error) {
-      return handleError(error, "Error executing tool");
+    } catch (error2) {
+      return handleError(error2, "Error executing tool");
     }
   };
 }
@@ -3586,8 +3676,8 @@ async function executeAgentToolHandler(c2) {
       runId: agentId
     });
     return c2.json(result);
-  } catch (error) {
-    return handleError(error, "Error executing tool");
+  } catch (error2) {
+    return handleError(error2, "Error executing tool");
   }
 }
 var getVector = (c2, vectorName) => {
@@ -3607,8 +3697,8 @@ async function upsertVectors(c2) {
     const vector = getVector(c2, vectorName);
     const result = await vector.upsert({ indexName, vectors, metadata, ids });
     return c2.json({ ids: result });
-  } catch (error) {
-    return handleError(error, "Error upserting vectors");
+  } catch (error2) {
+    return handleError(error2, "Error upserting vectors");
   }
 }
 async function createIndex(c2) {
@@ -3626,8 +3716,8 @@ async function createIndex(c2) {
     const vector = getVector(c2, vectorName);
     await vector.createIndex({ indexName, dimension, metric });
     return c2.json({ success: true });
-  } catch (error) {
-    return handleError(error, "Error creating index");
+  } catch (error2) {
+    return handleError(error2, "Error creating index");
   }
 }
 async function queryVectors(c2) {
@@ -3640,8 +3730,8 @@ async function queryVectors(c2) {
     const vector = getVector(c2, vectorName);
     const results = await vector.query({ indexName, queryVector, topK, filter, includeVector });
     return c2.json({ results });
-  } catch (error) {
-    return handleError(error, "Error querying vectors");
+  } catch (error2) {
+    return handleError(error2, "Error querying vectors");
   }
 }
 async function listIndexes(c2) {
@@ -3650,8 +3740,8 @@ async function listIndexes(c2) {
     const vector = getVector(c2, vectorName);
     const indexes = await vector.listIndexes();
     return c2.json({ indexes: indexes.filter(Boolean) });
-  } catch (error) {
-    return handleError(error, "Error listing indexes");
+  } catch (error2) {
+    return handleError(error2, "Error listing indexes");
   }
 }
 async function describeIndex(c2) {
@@ -3668,8 +3758,8 @@ async function describeIndex(c2) {
       count: stats.count,
       metric: stats.metric?.toLowerCase()
     });
-  } catch (error) {
-    return handleError(error, "Error describing index");
+  } catch (error2) {
+    return handleError(error2, "Error describing index");
   }
 }
 async function deleteIndex(c2) {
@@ -3682,8 +3772,8 @@ async function deleteIndex(c2) {
     const vector = getVector(c2, vectorName);
     await vector.deleteIndex(indexName);
     return c2.json({ success: true });
-  } catch (error) {
-    return handleError(error, "Error deleting index");
+  } catch (error2) {
+    return handleError(error2, "Error deleting index");
   }
 }
 async function getSpeakersHandler(c2) {
@@ -3699,8 +3789,8 @@ async function getSpeakersHandler(c2) {
     }
     const speakers = await agent.getSpeakers();
     return c2.json(speakers);
-  } catch (error) {
-    return handleError(error, "Error getting speakers");
+  } catch (error2) {
+    return handleError(error2, "Error getting speakers");
   }
 }
 async function speakHandler(c2) {
@@ -3720,8 +3810,8 @@ async function speakHandler(c2) {
     c2.header("Content-Type", `audio/${options?.filetype ?? "mp3"}`);
     c2.header("Transfer-Encoding", "chunked");
     return c2.body(audioStream);
-  } catch (error) {
-    return handleError(error, "Error generating speech");
+  } catch (error2) {
+    return handleError(error2, "Error generating speech");
   }
 }
 async function listenHandler(c2) {
@@ -3749,16 +3839,16 @@ async function listenHandler(c2) {
     let parsedOptions;
     try {
       parsedOptions = options ? JSON.parse(options) : {};
-    } catch (error) {
-      if (error instanceof SyntaxError) {
-        logger2.error("Invalid JSON in options:", error);
+    } catch (error2) {
+      if (error2 instanceof SyntaxError) {
+        logger2.error("Invalid JSON in options:", error2);
       }
       parsedOptions = {};
     }
     const transcription = await agent.voice.listen(audioStream, parsedOptions);
     return c2.json({ text: transcription });
-  } catch (error) {
-    return handleError(error, "Error transcribing speech");
+  } catch (error2) {
+    return handleError(error2, "Error transcribing speech");
   }
 }
 async function getWorkflowsHandler(c2) {
@@ -3786,8 +3876,8 @@ async function getWorkflowsHandler(c2) {
       return acc;
     }, {});
     return c2.json(_workflows);
-  } catch (error) {
-    return handleError(error, "Error getting workflows");
+  } catch (error2) {
+    return handleError(error2, "Error getting workflows");
   }
 }
 async function getWorkflowByIdHandler(c2) {
@@ -3818,8 +3908,8 @@ async function getWorkflowByIdHandler(c2) {
       serializedStepGraph,
       serializedStepSubscriberGraph
     });
-  } catch (error) {
-    return handleError(error, "Error getting workflow");
+  } catch (error2) {
+    return handleError(error2, "Error getting workflow");
   }
 }
 async function startAsyncWorkflowHandler(c2) {
@@ -3840,8 +3930,8 @@ async function startAsyncWorkflowHandler(c2) {
       triggerData: body
     });
     return c2.json(result);
-  } catch (error) {
-    return handleError(error, "Error executing workflow");
+  } catch (error2) {
+    return handleError(error2, "Error executing workflow");
   }
 }
 async function createRunHandler(c2) {
@@ -3896,8 +3986,9 @@ async function watchWorkflowHandler(c2) {
       c2,
       async (stream) => {
         return new Promise((_resolve, _reject) => {
-          let unwatch = run.watch(({ activePaths, context, runId: runId2, timestamp, suspendedSteps }) => {
-            void stream.write(JSON.stringify({ activePaths, context, runId: runId2, timestamp, suspendedSteps }) + "");
+          let unwatch = run.watch(({ activePaths, runId: runId2, timestamp, results }) => {
+            const activePathsObj = Object.fromEntries(activePaths);
+            void stream.write(JSON.stringify({ activePaths: activePathsObj, runId: runId2, timestamp, results }) + "");
           });
           stream.onAbort(() => {
             unwatch?.();
@@ -3910,8 +4001,8 @@ async function watchWorkflowHandler(c2) {
         await stream.close();
       }
     );
-  } catch (error) {
-    return handleError(error, "Error watching workflow");
+  } catch (error2) {
+    return handleError(error2, "Error watching workflow");
   }
 }
 async function resumeAsyncWorkflowHandler(c2) {
@@ -3933,8 +4024,8 @@ async function resumeAsyncWorkflowHandler(c2) {
       context
     });
     return c2.json(result);
-  } catch (error) {
-    return handleError(error, "Error resuming workflow step");
+  } catch (error2) {
+    return handleError(error2, "Error resuming workflow step");
   }
 }
 async function resumeWorkflowHandler(c2) {
@@ -3956,8 +4047,8 @@ async function resumeWorkflowHandler(c2) {
       context
     });
     return c2.json({ message: "Workflow run resumed" });
-  } catch (error) {
-    return handleError(error, "Error resuming workflow");
+  } catch (error2) {
+    return handleError(error2, "Error resuming workflow");
   }
 }
 
@@ -4088,7 +4179,7 @@ async function createHonoServer(mastra, options = {}) {
     cors({
       origin: "*",
       allowMethods: ["GET", "POST", "PUT", "DELETE", "OPTIONS"],
-      allowHeaders: ["Content-Type", "Authorization"],
+      allowHeaders: ["Content-Type", "Authorization", "x-mastra-client-type"],
       exposeHeaders: ["Content-Length", "X-Requested-With"],
       credentials: false,
       maxAge: 3600
@@ -4115,6 +4206,44 @@ async function createHonoServer(mastra, options = {}) {
     // 4.5 MB,
     onError: (c2) => c2.json({ error: "Request body too large" }, 413)
   };
+  const server = mastra.getServer();
+  const routes = server?.apiRoutes;
+  if (server?.middleware) {
+    const normalizedMiddlewares = Array.isArray(server.middleware) ? server.middleware : [server.middleware];
+    const middlewares = normalizedMiddlewares.map((middleware2) => {
+      if (typeof middleware2 === "function") {
+        return {
+          path: "*",
+          handler: middleware2
+        };
+      }
+      return middleware2;
+    });
+    for (const middleware2 of middlewares) {
+      app.use(middleware2.path, middleware2.handler);
+    }
+  }
+  if (routes) {
+    for (const route of routes) {
+      const middlewares = [];
+      if (route.middleware) {
+        middlewares.push(...Array.isArray(route.middleware) ? route.middleware : [route.middleware]);
+      }
+      if (route.openapi) {
+        middlewares.push(h(route.openapi));
+      }
+      console.log({ path: route.path, middlewares });
+      if (route.method === "GET") {
+        app.get(route.path, ...middlewares, route.handler);
+      } else if (route.method === "POST") {
+        app.post(route.path, ...middlewares, route.handler);
+      } else if (route.method === "PUT") {
+        app.put(route.path, ...middlewares, route.handler);
+      } else if (route.method === "DELETE") {
+        app.delete(route.path, ...middlewares, route.handler);
+      }
+    }
+  }
   app.get(
     "/api",
     h({
@@ -5236,6 +5365,19 @@ async function createHonoServer(mastra, options = {}) {
     }),
     getTelemetryHandler
   );
+  app.post(
+    "/api/telemetry",
+    h({
+      description: "Store telemetry",
+      tags: ["telemetry"],
+      responses: {
+        200: {
+          description: "Traces stored"
+        }
+      }
+    }),
+    storeTelemetryHandler
+  );
   app.get(
     "/api/workflows",
     h({