@mastra/core 0.9.0 → 0.9.1-alpha.1

package/dist/chunk-HNEE7IF4.js

@@ -0,0 +1,60 @@
+import { simulateReadableStream } from 'ai';
+import { MockLanguageModelV1 } from 'ai/test';
+
+// src/llm/model/mock.ts
+function createMockModel({
+  objectGenerationMode,
+  mockText,
+  spyGenerate,
+  spyStream
+}) {
+  const mockModel = new MockLanguageModelV1({
+    defaultObjectGenerationMode: objectGenerationMode,
+    doGenerate: async (props) => {
+      if (spyGenerate) {
+        spyGenerate(props);
+      }
+      if (objectGenerationMode === "json") {
+        return {
+          rawCall: { rawPrompt: null, rawSettings: {} },
+          finishReason: "stop",
+          usage: { promptTokens: 10, completionTokens: 20 },
+          text: JSON.stringify(mockText)
+        };
+      }
+      return {
+        rawCall: { rawPrompt: null, rawSettings: {} },
+        finishReason: "stop",
+        usage: { promptTokens: 10, completionTokens: 20 },
+        text: typeof mockText === "string" ? mockText : JSON.stringify(mockText)
+      };
+    },
+    doStream: async (props) => {
+      if (spyStream) {
+        spyStream(props);
+      }
+      const text = typeof mockText === "string" ? mockText : JSON.stringify(mockText);
+      const chunks = text.split(" ").map((word) => ({
+        type: "text-delta",
+        textDelta: word + " "
+      }));
+      return {
+        stream: simulateReadableStream({
+          chunks: [
+            ...chunks,
+            {
+              type: "finish",
+              finishReason: "stop",
+              logprobs: void 0,
+              usage: { completionTokens: 10, promptTokens: 3 }
+            }
+          ]
+        }),
+        rawCall: { rawPrompt: null, rawSettings: {} }
+      };
+    }
+  });
+  return mockModel;
+}
+
+export { createMockModel };

package/dist/{chunk-NZDXKMDP.cjs → chunk-LABUWBKX.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-// src/di/index.ts
+// src/runtime-context/index.ts
 var RuntimeContext = class {
   registry = /* @__PURE__ */ new Map();
   constructor(iterable) {

package/dist/chunk-MUNFCOMB.cjs

@@ -0,0 +1,62 @@
+'use strict';
+
+var ai = require('ai');
+var test = require('ai/test');
+
+// src/llm/model/mock.ts
+function createMockModel({
+  objectGenerationMode,
+  mockText,
+  spyGenerate,
+  spyStream
+}) {
+  const mockModel = new test.MockLanguageModelV1({
+    defaultObjectGenerationMode: objectGenerationMode,
+    doGenerate: async (props) => {
+      if (spyGenerate) {
+        spyGenerate(props);
+      }
+      if (objectGenerationMode === "json") {
+        return {
+          rawCall: { rawPrompt: null, rawSettings: {} },
+          finishReason: "stop",
+          usage: { promptTokens: 10, completionTokens: 20 },
+          text: JSON.stringify(mockText)
+        };
+      }
+      return {
+        rawCall: { rawPrompt: null, rawSettings: {} },
+        finishReason: "stop",
+        usage: { promptTokens: 10, completionTokens: 20 },
+        text: typeof mockText === "string" ? mockText : JSON.stringify(mockText)
+      };
+    },
+    doStream: async (props) => {
+      if (spyStream) {
+        spyStream(props);
+      }
+      const text = typeof mockText === "string" ? mockText : JSON.stringify(mockText);
+      const chunks = text.split(" ").map((word) => ({
+        type: "text-delta",
+        textDelta: word + " "
+      }));
+      return {
+        stream: ai.simulateReadableStream({
+          chunks: [
+            ...chunks,
+            {
+              type: "finish",
+              finishReason: "stop",
+              logprobs: void 0,
+              usage: { completionTokens: 10, promptTokens: 3 }
+            }
+          ]
+        }),
+        rawCall: { rawPrompt: null, rawSettings: {} }
+      };
+    }
+  });
+  return mockModel;
+}
+
+exports.createMockModel = createMockModel;

package/dist/{chunk-GJWCFDFN.js → chunk-PS2ZF6MA.js}

@@ -1,4 +1,4 @@
-import { Agent } from './chunk-HARYMLZH.js';
+import { Agent } from './chunk-4K4DHXRF.js';
 import { CohereClient } from 'cohere-ai';
 
 var CohereRelevanceScorer = class {

package/dist/{chunk-M472GIT6.js → chunk-SGGPJWRQ.js}

@@ -1,4 +1,4 @@
-// src/di/index.ts
+// src/runtime-context/index.ts
 var RuntimeContext = class {
   registry = /* @__PURE__ */ new Map();
   constructor(iterable) {

package/dist/{chunk-7CSNWYGJ.cjs → chunk-U5DGGGS2.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunk53P5ZBJE_cjs = require('./chunk-53P5ZBJE.cjs');
+var chunkYTTBFAQJ_cjs = require('./chunk-YTTBFAQJ.cjs');
 var zod = require('zod');
 
 // src/integration/integration.ts
@@ -68,7 +68,7 @@ var OpenAPIToolset = class {
     const tools = Object.keys(clientMethods).reduce((acc, key) => {
       const comment = documentations[key]?.comment;
       const fallbackComment = `Execute ${key}`;
-      const tool = chunk53P5ZBJE_cjs.createTool({
+      const tool = chunkYTTBFAQJ_cjs.createTool({
         id: key,
         inputSchema: schemas[key] || zod.z.object({}),
         description: comment || fallbackComment,

package/dist/{chunk-W5IA5OGL.cjs → chunk-V7IE36YV.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunkWEYWYKLG_cjs = require('./chunk-WEYWYKLG.cjs');
+var chunk27PAET7X_cjs = require('./chunk-27PAET7X.cjs');
 var chunkRIBNDTA6_cjs = require('./chunk-RIBNDTA6.cjs');
 var chunkO7IW545H_cjs = require('./chunk-O7IW545H.cjs');
 var chunkRWTSGWWL_cjs = require('./chunk-RWTSGWWL.cjs');
@@ -16,6 +16,7 @@ exports.Mastra = class Mastra {
   #agents;
   #logger;
   #workflows;
+  #vnext_workflows;
   #tts;
   #deployer;
   #serverMiddleware = [];
@@ -66,13 +67,13 @@ exports.Mastra = class Mastra {
     this.#logger = logger;
     let storage = config?.storage;
     if (!storage) {
-      storage = new chunkWEYWYKLG_cjs.DefaultProxyStorage({
+      storage = new chunk27PAET7X_cjs.DefaultProxyStorage({
         config: {
           url: process.env.MASTRA_DEFAULT_STORAGE_URL || `:memory:`
         }
       });
     }
-    storage = chunkWEYWYKLG_cjs.augmentWithInit(storage);
+    storage = chunk27PAET7X_cjs.augmentWithInit(storage);
     this.#telemetry = chunkRIBNDTA6_cjs.Telemetry.init(config?.telemetry);
     if (this.#telemetry) {
       this.#storage = this.#telemetry.traceClass(storage, {
@@ -183,6 +184,22 @@ This is a warning for now, but will throw an error in the future
         }
       });
     }
+    this.#vnext_workflows = {};
+    if (config?.vnext_workflows) {
+      Object.entries(config.vnext_workflows).forEach(([key, workflow]) => {
+        workflow.__registerMastra(this);
+        workflow.__registerPrimitives({
+          logger: this.getLogger(),
+          telemetry: this.#telemetry,
+          storage: this.storage,
+          memory: this.memory,
+          agents,
+          tts: this.#tts,
+          vectors: this.#vectors
+        });
+        this.#vnext_workflows[key] = workflow;
+      });
+    }
     if (config?.server) {
       this.#server = config.server;
     }
@@ -227,6 +244,20 @@ This is a warning for now, but will throw an error in the future
     }
     return workflow;
   }
+  vnext_getWorkflow(id, {
+    serialized
+  } = {}) {
+    const workflow = this.#vnext_workflows?.[id];
+    if (!workflow) {
+      throw new Error(`Workflow with ID ${String(id)} not found`);
+    }
+    if (serialized) {
+      return {
+        name: workflow.name
+      };
+    }
+    return workflow;
+  }
   getWorkflows(props = {}) {
     if (props.serialized) {
       return Object.entries(this.#workflows).reduce((acc, [k, v]) => {
@@ -240,8 +271,31 @@ This is a warning for now, but will throw an error in the future
     }
     return this.#workflows;
   }
+  vnext_getWorkflows(props = {}) {
+    if (props.serialized) {
+      return Object.entries(this.#vnext_workflows).reduce((acc, [k, v]) => {
+        return {
+          ...acc,
+          [k]: {
+            name: v.name
+          }
+        };
+      }, {});
+    }
+    return this.#vnext_workflows;
+  }
   setStorage(storage) {
-    this.#storage = chunkWEYWYKLG_cjs.augmentWithInit(storage);
+    if (storage instanceof chunk27PAET7X_cjs.DefaultProxyStorage) {
+      this.#logger.warn(`Importing "DefaultStorage" from '@mastra/core/storage/libsql' is deprecated.
+
+Instead of:
+  import { DefaultStorage } from '@mastra/core/storage/libsql';
+
+Do:
+  import { LibSQLStore } from '@mastra/libsql';
+`);
+    }
+    this.#storage = chunk27PAET7X_cjs.augmentWithInit(storage);
   }
   setLogger({
     logger