mastra 0.1.57-unstable.56 → 0.1.57-unstable.88

package/dist/templates/netlify.js

@@ -2,8 +2,21 @@ import express from 'express';
 import { join } from 'path';
 import serverless from 'serverless-http';
 import { stringify } from 'superjson';
+import { pathToFileURL } from 'url';
 import zodToJsonSchema from 'zod-to-json-schema';
 const { mastra } = await import(join(process.cwd(), 'mastra.mjs'));
+const mastraToolsPaths = process.env.MASTRA_TOOLS_PATH;
+const toolImports = mastraToolsPaths
+    ? await Promise.all(mastraToolsPaths.split(',').map(async (toolPath) => {
+        return import(pathToFileURL(toolPath).href);
+    }))
+    : [];
+const tools = toolImports.reduce((acc, toolModule) => {
+    Object.entries(toolModule).forEach(([key, tool]) => {
+        acc[key] = tool;
+    });
+    return acc;
+}, {});
 const app = express();
 app.use(express.json());
 const validateBody = async (body) => {
@@ -24,7 +37,24 @@ app.get('/', (_req, res) => {
 app.get('/api/agents', async (_req, res) => {
     try {
         const agents = mastra.getAgents();
-        res.json(agents);
+        const serializedAgents = Object.entries(agents).reduce((acc, [_id, _agent]) => {
+            const agent = _agent;
+            const serializedAgentTools = Object.entries(agent?.tools || {}).reduce((acc, [key, tool]) => {
+                const _tool = tool;
+                acc[key] = {
+                    ..._tool,
+                    inputSchema: _tool.inputSchema ? stringify(zodToJsonSchema(_tool.inputSchema)) : undefined,
+                    outputSchema: _tool.outputSchema ? stringify(zodToJsonSchema(_tool.outputSchema)) : undefined,
+                };
+                return acc;
+            }, {});
+            acc[_id] = {
+                ...agent,
+                tools: serializedAgentTools,
+            };
+            return acc;
+        }, {});
+        res.json(serializedAgents);
     }
     catch (error) {
         const apiError = error;
@@ -37,7 +67,19 @@ app.get('/api/agents/:agentId', async (req, res) => {
     try {
         const agentId = req.params.agentId;
         const agent = mastra.getAgent(agentId);
-        res.json(agent);
+        const serializedAgentTools = Object.entries(agent?.tools || {}).reduce((acc, [key, tool]) => {
+            const _tool = tool;
+            acc[key] = {
+                ..._tool,
+                inputSchema: _tool.inputSchema ? stringify(zodToJsonSchema(_tool.inputSchema)) : undefined,
+                outputSchema: _tool.outputSchema ? stringify(zodToJsonSchema(_tool.outputSchema)) : undefined,
+            };
+            return acc;
+        }, {});
+        res.json({
+            ...agent,
+            tools: serializedAgentTools,
+        });
     }
     catch (error) {
         const apiError = error;
@@ -46,11 +88,11 @@ app.get('/api/agents/:agentId', async (req, res) => {
         return;
     }
 });
-app.post('/api/agents/:agentId/text', async (req, res) => {
+app.post('/api/agents/:agentId/generate', async (req, res) => {
     try {
         const agentId = req.params.agentId;
         const agent = mastra.getAgent(agentId);
-        const messages = req.body.messages;
+        const { messages, threadId, resourceid } = req.body;
         const { ok, errorResponse } = await validateBody({
             messages,
         });
@@ -58,11 +100,7 @@ app.post('/api/agents/:agentId/text', async (req, res) => {
             res.status(400).json({ error: errorResponse });
             return;
         }
-        if (!Array.isArray(messages)) {
-            res.status(400).json({ error: { messages: 'Messages should be an array' } });
-            return;
-        }
-        const result = await agent.generate(messages);
+        const result = await agent.generate(messages, { threadId, resourceid });
         res.json(result);
     }
     catch (error) {
@@ -76,7 +114,7 @@ app.post('/api/agents/:agentId/stream', async (req, res) => {
     try {
         const agentId = req.params.agentId;
         const agent = mastra.getAgent(agentId);
-        const messages = req.body.messages;
+        const { messages, threadId, resourceid } = req.body;
         const { ok, errorResponse } = await validateBody({
             messages,
         });
@@ -84,11 +122,10 @@ app.post('/api/agents/:agentId/stream', async (req, res) => {
             res.status(400).json({ error: errorResponse });
             return;
         }
-        if (!Array.isArray(messages)) {
-            res.status(400).json({ error: { messages: 'Messages should be an array' } });
-            return;
-        }
-        const streamResult = await agent.generate(messages, { stream: true });
+        const streamResult = await agent.stream(messages, {
+            threadId,
+            resourceid,
+        });
         streamResult.pipeDataStreamToResponse(res);
     }
     catch (error) {
@@ -102,8 +139,7 @@ app.post('/api/agents/:agentId/text-object', async (req, res) => {
     try {
         const agentId = req.params.agentId;
         const agent = mastra.getAgent(agentId);
-        const messages = req.body.messages;
-        const schema = req.body.schema;
+        const { messages, schema, threadId, resourceid } = req.body;
         const { ok, errorResponse } = await validateBody({
             messages,
             schema,
@@ -112,11 +148,7 @@ app.post('/api/agents/:agentId/text-object', async (req, res) => {
             res.status(400).json({ error: errorResponse });
             return;
         }
-        if (!Array.isArray(messages)) {
-            res.status(400).json({ error: { messages: 'Messages should be an array' } });
-            return;
-        }
-        const result = await agent.generate(messages, { schema });
+        const result = await agent.generate(messages, { output: schema, threadId, resourceid });
         res.json(result);
     }
     catch (error) {
@@ -132,8 +164,7 @@ app.post('/api/agents/:agentId/stream-object', async (req, res) => {
     try {
         const agentId = req.params.agentId;
         const agent = mastra.getAgent(agentId);
-        const messages = req.body.messages;
-        const schema = req.body.schema;
+        const { messages, schema, threadId, resourceid } = req.body;
         const { ok, errorResponse } = await validateBody({
             messages,
             schema,
@@ -142,11 +173,7 @@ app.post('/api/agents/:agentId/stream-object', async (req, res) => {
             res.status(400).json({ error: errorResponse });
             return;
         }
-        if (!Array.isArray(messages)) {
-            res.status(400).json({ error: { messages: 'Messages should be an array' } });
-            return;
-        }
-        const streamResult = await agent.generate(messages, { schema, stream: true });
+        const streamResult = await agent.stream(messages, { output: schema, threadId, resourceid });
         streamResult.pipeTextStreamToResponse(res);
     }
     catch (error) {
@@ -158,9 +185,31 @@ app.post('/api/agents/:agentId/stream-object', async (req, res) => {
         return;
     }
 });
+app.post('/api/agents/:agentId/tools/:toolId/execute', async (req, res) => {
+    try {
+        const agentId = req.params.agentId;
+        const toolId = req.params.toolId;
+        const agent = mastra.getAgent(agentId);
+        const tool = Object.values(agent?.tools || {}).find((tool) => tool.id === toolId);
+        const result = await tool.execute({
+            context: {
+                ...req.body,
+            },
+            mastra,
+            runId: agentId,
+        });
+        res.json(result);
+    }
+    catch (error) {
+        const apiError = error;
+        console.error('Error executing tool', apiError);
+        res.status(apiError.status || 500).json({ error: apiError.message || 'Error executing tool' });
+        return;
+    }
+});
 app.get('/api/workflows', async (_req, res) => {
     try {
-        const workflows = mastra.getWorkflows();
+        const workflows = mastra.getWorkflows({ serialized: true });
         res.json(workflows);
     }
     catch (error) {
@@ -175,9 +224,23 @@ app.get('/api/workflows/:workflowId', async (req, res) => {
         const workflowId = req.params.workflowId;
         const workflow = mastra.getWorkflow(workflowId);
         const triggerSchema = workflow.triggerSchema;
+        const stepGraph = workflow.stepGraph;
+        const stepSubscriberGraph = workflow.stepSubscriberGraph;
+        const serializedSteps = Object.entries(workflow.steps).reduce((acc, [key, step]) => {
+            const _step = step;
+            acc[key] = {
+                ..._step,
+                inputSchema: _step.inputSchema ? stringify(zodToJsonSchema(_step.inputSchema)) : undefined,
+                outputSchema: _step.outputSchema ? stringify(zodToJsonSchema(_step.outputSchema)) : undefined,
+            };
+            return acc;
+        }, {});
         res.json({
             ...workflow,
             triggerSchema: triggerSchema ? stringify(zodToJsonSchema(triggerSchema)) : undefined,
+            steps: serializedSteps,
+            stepGraph,
+            stepSubscriberGraph,
         });
     }
     catch (error) {
@@ -190,8 +253,7 @@ app.get('/api/workflows/:workflowId', async (req, res) => {
 app.post('/api/workflows/:workflowId/execute', async (req, res) => {
     try {
         const workflowId = req.params.workflowId;
-        const workflow = mastra.workflows.get(workflowId);
-        console.log('req.body', req.body);
+        const workflow = mastra.getWorkflow(workflowId);
         const result = await workflow.execute(req.body);
         res.json(result);
     }
@@ -202,14 +264,35 @@ app.post('/api/workflows/:workflowId/execute', async (req, res) => {
         return;
     }
 });
-app.get('/api/memory/threads/get-by-resourceid/:resourceid', async (req, res) => {
+app.get('/api/memory/status', async (_req, res) => {
     try {
-        const resourceid = req.params.resourceid;
+        const memory = mastra.memory;
+        if (!memory) {
+            res.json({ result: false });
+            return;
+        }
+        res.json({ result: true });
+    }
+    catch (error) {
+        const apiError = error;
+        console.error('Error getting memory status', apiError);
+        res.status(apiError.status || 500).json({ error: apiError.message || 'Error getting memory status' });
+        return;
+    }
+});
+app.get('/api/memory/threads', async (req, res) => {
+    try {
+        const resourceid = req.query.resourceid;
         const memory = mastra.memory;
         if (!memory) {
             res.status(400).json({ error: 'Memory is not initialized' });
             return;
         }
+        const { ok, errorResponse } = await validateBody({ resourceid });
+        if (!ok) {
+            res.status(400).json({ error: errorResponse });
+            return;
+        }
         const threads = await memory.getThreadsByResourceId({ resourceid });
         res.json(threads);
     }
@@ -485,4 +568,86 @@ app.post('/api/syncs/:syncId/execute', async (req, res) => {
         return;
     }
 });
+app.get('/api/logs', async (_req, res) => {
+    try {
+        const logs = await mastra.getLogs();
+        res.json(logs);
+    }
+    catch (error) {
+        const apiError = error;
+        console.error('Error getting logs', apiError);
+        res.status(apiError.status || 500).json({ error: apiError.message || 'Error getting logs' });
+        return;
+    }
+});
+app.get('/api/logs/:runId', async (req, res) => {
+    try {
+        const runId = req.params.runId;
+        const logs = await mastra.getLogsByRunId(runId);
+        res.json(logs);
+    }
+    catch (error) {
+        const apiError = error;
+        console.error('Error getting logs', apiError);
+        res.status(apiError.status || 500).json({ error: apiError.message || 'Error getting logs' });
+        return;
+    }
+});
+app.get('/api/tools', async (_req, res) => {
+    if (tools) {
+        const serializedTools = Object.entries(tools).reduce((acc, [id, _tool]) => {
+            const tool = _tool;
+            acc[id] = {
+                ...tool,
+                inputSchema: tool.inputSchema ? stringify(zodToJsonSchema(tool.inputSchema)) : undefined,
+                outputSchema: tool.outputSchema ? stringify(zodToJsonSchema(tool.outputSchema)) : undefined,
+            };
+            return acc;
+        }, {});
+        res.json(serializedTools);
+    }
+    else {
+        res.status(200).json({});
+    }
+});
+app.get('/api/tools/:toolId', async (req, res) => {
+    const toolId = req.params.toolId;
+    const tool = Object.values(tools || {}).find((tool) => tool.id === toolId);
+    if (tool) {
+        const serializedTool = {
+            ...tool,
+            inputSchema: tool.inputSchema ? stringify(zodToJsonSchema(tool.inputSchema)) : undefined,
+            outputSchema: tool.outputSchema ? stringify(zodToJsonSchema(tool.outputSchema)) : undefined,
+        };
+        res.json(serializedTool);
+    }
+    else {
+        res.status(404).json({ error: 'Tool not found' });
+    }
+});
+app.post('/api/tools/:toolId/execute', async (req, res) => {
+    try {
+        const toolId = req.params.toolId;
+        const tool = Object.values(tools || {}).find((tool) => tool.id === toolId);
+        if (!tool) {
+            res.status(404).json({ error: 'Tool not found' });
+            return;
+        }
+        const { input } = req.body;
+        const result = await tool.execute({
+            context: {
+                ...input,
+            },
+            mastra,
+            runId: mastra.runId,
+        });
+        res.json(result);
+    }
+    catch (error) {
+        const apiError = error;
+        console.error('Error executing tool', apiError);
+        res.status(apiError.status || 500).json({ error: apiError.message || 'Error executing tool' });
+        return;
+    }
+});
 export const handler = serverless(app);

package/dist/templates/worker.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"worker.d.ts","sourceRoot":"","sources":["../../src/templates/worker.ts"],"names":[],"mappings":"AAwBA,UAAU,gBAAgB;IACxB,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;IACvC,sBAAsB,IAAI,IAAI,CAAC;CAChC;;mBAk0BsB,OAAO,OAAO,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,gBAAgB;;AADlF,wBAQE"}
+{"version":3,"file":"worker.d.ts","sourceRoot":"","sources":["../../src/templates/worker.ts"],"names":[],"mappings":"AA0CA,UAAU,gBAAgB;IACxB,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;IACvC,sBAAsB,IAAI,IAAI,CAAC;CAChC;;mBAshCsB,OAAO,OAAO,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,gBAAgB;;AADlF,wBAQE"}