@mastra/server 0.14.0-alpha.2 → 0.14.0-alpha.4

package/dist/{chunk-MG5HDGGH.js → chunk-6LHEW5YZ.js}

@@ -113,6 +113,7 @@ var require_secure_json_parse = __commonJS({
 var agents_exports = {};
 __export(agents_exports, {
   generateHandler: () => generateHandler,
+  generateVNextHandler: () => generateVNextHandler,
   getAgentByIdHandler: () => getAgentByIdHandler,
   getAgentsHandler: () => getAgentsHandler,
   getEvalsByAgentIdHandler: () => getEvalsByAgentIdHandler,
@@ -120,6 +121,7 @@ __export(agents_exports, {
   getSerializedAgentTools: () => getSerializedAgentTools,
   streamGenerateHandler: () => streamGenerateHandler,
   streamVNextGenerateHandler: () => streamVNextGenerateHandler,
+  streamVNextUIMessageHandler: () => streamVNextUIMessageHandler,
   updateAgentModelHandler: () => updateAgentModelHandler
 });
 
@@ -7549,6 +7551,7 @@ async function getAgentsHandler({ mastra, runtimeContext }) {
             logger.error("Error getting workflows for agent", { agentName: agent.name, error });
           }
         }
+        const model = llm?.getModel();
         return {
           id,
           name: agent.name,
@@ -7557,6 +7560,7 @@ async function getAgentsHandler({ mastra, runtimeContext }) {
           workflows: serializedAgentWorkflows,
           provider: llm?.getProvider(),
           modelId: llm?.getModelId(),
+          modelVersion: model?.specificationVersion,
           defaultGenerateOptions,
           defaultStreamOptions
         };
@@ -7628,6 +7632,7 @@ async function getAgentByIdHandler({
     const llm = await agent.getLLM({ runtimeContext });
     const defaultGenerateOptions = await agent.getDefaultGenerateOptions({ runtimeContext: proxyRuntimeContext });
     const defaultStreamOptions = await agent.getDefaultStreamOptions({ runtimeContext: proxyRuntimeContext });
+    const model = llm?.getModel();
     return {
       name: agent.name,
       instructions,
@@ -7635,6 +7640,7 @@ async function getAgentByIdHandler({
       workflows: serializedAgentWorkflows,
       provider: llm?.getProvider(),
       modelId: llm?.getModelId(),
+      modelVersion: model?.specificationVersion,
       defaultGenerateOptions,
       defaultStreamOptions
     };
@@ -7711,6 +7717,38 @@ async function generateHandler({
     return handleError(error, "Error generating from agent");
   }
 }
+async function generateVNextHandler({
+  mastra,
+  runtimeContext,
+  agentId,
+  body,
+  abortSignal
+}) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    if (!agent) {
+      throw new HTTPException(404, { message: "Agent not found" });
+    }
+    const { messages, runtimeContext: agentRuntimeContext, ...rest } = body;
+    const finalRuntimeContext = new RuntimeContext([
+      ...Array.from(runtimeContext.entries()),
+      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    ]);
+    validateBody({ messages });
+    const result = await agent.generateVNext(messages, {
+      ...rest,
+      runtimeContext: finalRuntimeContext,
+      format: rest.format || "mastra",
+      options: {
+        ...rest?.options ?? {},
+        abortSignal
+      }
+    });
+    return result;
+  } catch (error) {
+    return handleError(error, "Error generating from agent");
+  }
+}
 async function streamGenerateHandler({
   mastra,
   runtimeContext,
@@ -7777,13 +7815,49 @@ function streamVNextGenerateHandler({
     const streamResult = agent.streamVNext(messages, {
       ...rest,
       runtimeContext: finalRuntimeContext,
-      abortSignal
+      options: {
+        ...rest?.options ?? {},
+        abortSignal
+      },
+      format: body.format ?? "mastra"
     });
     return streamResult;
   } catch (error) {
     return handleError(error, "error streaming agent response");
   }
 }
+async function streamVNextUIMessageHandler({
+  mastra,
+  runtimeContext,
+  agentId,
+  body,
+  abortSignal
+}) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    if (!agent) {
+      throw new HTTPException(404, { message: "Agent not found" });
+    }
+    const { messages, runtimeContext: agentRuntimeContext, ...rest } = body;
+    const finalRuntimeContext = new RuntimeContext([
+      ...Array.from(runtimeContext.entries()),
+      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    ]);
+    validateBody({ messages });
+    const streamResult = await agent.streamVNext(messages, {
+      ...rest,
+      runtimeContext: finalRuntimeContext,
+      options: {
+        ...rest?.options ?? {},
+        abortSignal
+      },
+      format: "aisdk"
+    });
+    return streamResult.toUIMessageStreamResponse();
+  } catch (error) {
+    return handleError(error, "error streaming agent response");
+  }
+}
 function updateAgentModelHandler({
   mastra,
   agentId,
@@ -7810,6 +7884,6 @@ function updateAgentModelHandler({
   }
 }
 
-export { agents_exports, generateHandler, getAgentByIdHandler, getAgentsHandler, getEvalsByAgentIdHandler, getLiveEvalsByAgentIdHandler, getSerializedAgentTools, streamGenerateHandler, streamVNextGenerateHandler, updateAgentModelHandler };
-//# sourceMappingURL=chunk-MG5HDGGH.js.map
-//# sourceMappingURL=chunk-MG5HDGGH.js.map
+export { agents_exports, generateHandler, generateVNextHandler, getAgentByIdHandler, getAgentsHandler, getEvalsByAgentIdHandler, getLiveEvalsByAgentIdHandler, getSerializedAgentTools, streamGenerateHandler, streamVNextGenerateHandler, streamVNextUIMessageHandler, updateAgentModelHandler };
+//# sourceMappingURL=chunk-6LHEW5YZ.js.map
+//# sourceMappingURL=chunk-6LHEW5YZ.js.map