@mastra/core 0.4.2-alpha.0 → 0.4.2-alpha.2

package/dist/agent/index.d.ts

@@ -1,15 +1,15 @@
 import 'ai';
 import 'json-schema';
 import 'zod';
-export { A as Agent } from '../base-BdNHFjqH.js';
-import '../base-eWkcLLSb.js';
+export { A as Agent } from '../base-hk-xmLC1.js';
+import '../base-D90KQ4XI.js';
 import '../types-m9RryK9a.js';
 import '../voice/index.js';
 import 'sift';
 import '../vector/index.js';
 import '@opentelemetry/api';
-import '../index-C5uPdbs4.js';
-import 'pino';
+import '../index-mKY1XrpK.js';
 import 'stream';
+import 'pino';
 import '@opentelemetry/sdk-trace-base';
 import '../tts/index.js';

package/dist/agent/index.js

@@ -1 +1 @@
-export { Agent } from '../chunk-ORM3OCHX.js';
+export { Agent } from '../chunk-7TG2Y45H.js';

package/dist/{base-eWkcLLSb.d.ts → base-D90KQ4XI.d.ts}

@@ -1,6 +1,6 @@
 import * as _opentelemetry_api from '@opentelemetry/api';
 import { Tracer, Span } from '@opentelemetry/api';
-import { R as RegisteredLogger, L as Logger } from './index-C5uPdbs4.js';
+import { R as RegisteredLogger, L as Logger } from './index-mKY1XrpK.js';
 import { SpanExporter } from '@opentelemetry/sdk-trace-base';
 
 /** Sampling strategy configuration for OpenTelemetry */

package/dist/{base-BdNHFjqH.d.ts → base-hk-xmLC1.d.ts}

@@ -1,13 +1,13 @@
 import * as ai from 'ai';
 import { EmbeddingModel, CoreMessage as CoreMessage$1, Message, UserContent, AssistantContent, LanguageModelV1, GenerateTextResult, GenerateObjectResult, StreamTextResult, StreamObjectResult, CoreSystemMessage as CoreSystemMessage$1, CoreAssistantMessage as CoreAssistantMessage$1, CoreUserMessage as CoreUserMessage$1, CoreToolMessage as CoreToolMessage$1, EmbedResult as EmbedResult$1, EmbedManyResult as EmbedManyResult$1, LanguageModel as LanguageModel$1, DeepPartial, ToolContent } from 'ai';
-import { M as MastraBase, T as Telemetry } from './base-eWkcLLSb.js';
+import { M as MastraBase, T as Telemetry } from './base-D90KQ4XI.js';
 import { a as Metric, M as MetricResult, T as TestInfo } from './types-m9RryK9a.js';
 import { Query } from 'sift';
 import { z, ZodSchema } from 'zod';
 import { JSONSchema7 } from 'json-schema';
 import { MastraVector } from './vector/index.js';
 import { CompositeVoice } from './voice/index.js';
-import { B as BaseLogMessage, R as RegisteredLogger, L as Logger, d as Run } from './index-C5uPdbs4.js';
+import { B as BaseLogMessage, R as RegisteredLogger, L as Logger, d as Run } from './index-mKY1XrpK.js';
 import { MastraTTS } from './tts/index.js';
 
 /**
@@ -196,6 +196,7 @@ type StepDef<TStepId extends TSteps[number]['id'], TSteps extends IAction<any, a
     snapshotOnTimeout?: boolean;
     when?: Condition<any, any> | ((args: {
         context: WorkflowContext;
+        mastra?: MastraPrimitives;
     }) => Promise<boolean>);
     data: TSchemaIn;
     handler: (args: ActionContext<TSchemaIn>) => Promise<z.infer<TSchemaOut>>;
@@ -214,6 +215,7 @@ interface StepConfig<TStep extends IAction<any, any, any, any>, CondStep extends
     snapshotOnTimeout?: boolean;
     when?: Condition<CondStep, TTriggerSchema> | ((args: {
         context: WorkflowContext<TTriggerSchema>;
+        mastra?: MastraPrimitives;
     }) => Promise<boolean>);
     variables?: StepInputType<TStep, 'inputSchema'> extends never ? Record<string, VariableReference<VarStep, TTriggerSchema>> : {
         [K in keyof StepInputType<TStep, 'inputSchema'>]?: VariableReference<VarStep, TTriggerSchema>;
@@ -238,7 +240,7 @@ interface WorkflowContext<TTrigger extends z.ZodType<any> = any> {
     steps: Record<string, StepResult<any>>;
     triggerData: z.infer<TTrigger>;
     attempts: Record<string, number>;
-    getStepPayload: <T = unknown>(stepId: string) => T | undefined;
+    getStepResult: <T = unknown>(stepId: string) => T | undefined;
 }
 interface WorkflowLogMessage extends BaseLogMessage {
     type: typeof RegisteredLogger.WORKFLOW;
@@ -435,14 +437,14 @@ interface IAction<TId extends string, TSchemaIn extends z.ZodSchema | undefined,
     outputSchema?: TSchemaOut;
     mastra?: MastraPrimitives;
     payload?: TSchemaIn extends z.ZodSchema ? Partial<z.infer<TSchemaIn>> : unknown;
-    execute: (context: TContext) => Promise<TSchemaOut extends z.ZodSchema ? z.infer<TSchemaOut> : unknown>;
+    execute?: (context: TContext) => Promise<TSchemaOut extends z.ZodSchema ? z.infer<TSchemaOut> : unknown>;
     [key: string]: any;
 }
 
 type CoreTool = {
     description?: string;
     parameters: ZodSchema;
-    execute: (params: any) => Promise<any>;
+    execute?: (params: any) => Promise<any>;
 };
 interface ToolExecutionContext<TSchemaIn extends z.ZodSchema | undefined = undefined, TContext extends WorkflowContext = WorkflowContext> extends IExecutionContext<TSchemaIn, TContext> {
 }
@@ -475,6 +477,7 @@ interface AgentGenerateOptions<Z extends ZodSchema | JSONSchema7 | undefined = u
     output?: OutputType | Z;
     temperature?: number;
     toolChoice?: 'auto' | 'required';
+    experimental_output?: Z;
 }
 interface AgentStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefined> {
     toolsets?: ToolsetsInput;
@@ -489,6 +492,7 @@ interface AgentStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = und
     output?: OutputType | Z;
     temperature?: number;
     toolChoice?: 'auto' | 'required';
+    experimental_output?: Z;
 }
 
 type LanguageModel = LanguageModelV1;
@@ -520,18 +524,8 @@ type StructuredOutput = {
 };
 type GenerateReturn<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = Z extends undefined ? GenerateTextResult<any, any> : GenerateObjectResult<any>;
 type StreamReturn<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = Z extends undefined ? StreamTextResult<any, any> : StreamObjectResult<any, any, any>;
-type OutputType = 'text' | StructuredOutput;
-type LLMStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = {
-    runId?: string;
-    onFinish?: (result: string) => Promise<void> | void;
-    onStepFinish?: (step: string) => void;
-    maxSteps?: number;
-    tools?: ToolsInput;
-    convertedTools?: Record<string, CoreTool>;
-    output?: OutputType | Z;
-    temperature?: number;
-};
-type LLMTextOptions = {
+type OutputType = 'text' | StructuredOutput | ZodSchema | JSONSchema7 | undefined;
+type LLMTextOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = {
     tools?: ToolsInput;
     convertedTools?: Record<string, CoreTool>;
     messages: CoreMessage[];
@@ -539,11 +533,23 @@ type LLMTextOptions = {
     toolChoice?: 'auto' | 'required';
     maxSteps?: number;
     temperature?: number;
+    experimental_output?: Z;
 } & Run;
 type LLMTextObjectOptions<T> = LLMTextOptions & {
     structuredOutput: JSONSchema7 | z.ZodType<T> | StructuredOutput;
 };
-type LLMInnerStreamOptions = {
+type LLMStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = {
+    runId?: string;
+    onFinish?: (result: string) => Promise<void> | void;
+    onStepFinish?: (step: string) => void;
+    maxSteps?: number;
+    tools?: ToolsInput;
+    convertedTools?: Record<string, CoreTool>;
+    output?: OutputType | Z;
+    temperature?: number;
+    experimental_output?: Z;
+};
+type LLMInnerStreamOptions<Z extends ZodSchema | JSONSchema7 | undefined = undefined> = {
     tools?: ToolsInput;
     convertedTools?: Record<string, CoreTool>;
     messages: CoreMessage[];
@@ -552,6 +558,7 @@ type LLMInnerStreamOptions = {
     maxSteps?: number;
     temperature?: number;
     toolChoice?: 'auto' | 'required';
+    experimental_output?: Z;
 } & Run;
 type LLMStreamObjectOptions<T> = LLMInnerStreamOptions & {
     structuredOutput: JSONSchema7 | z.ZodType<T> | StructuredOutput;
@@ -568,10 +575,10 @@ declare class MastraLLMBase extends MastraBase {
     getModel(): ai.LanguageModelV1;
     convertToMessages(messages: string | string[] | CoreMessage$1[]): CoreMessage$1[];
     __registerPrimitives(p: MastraPrimitives): void;
-    __text(input: LLMTextOptions): Promise<GenerateTextResult<any, any>>;
+    __text<Z extends ZodSchema | JSONSchema7 | undefined>(input: LLMTextOptions<Z>): Promise<GenerateTextResult<any, any>>;
     __textObject<T>(input: LLMTextObjectOptions<T>): Promise<GenerateObjectResult<T>>;
     generate<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], options?: LLMStreamOptions<Z>): Promise<GenerateReturn<Z>>;
-    __stream(input: LLMInnerStreamOptions): Promise<StreamTextResult<any, any>>;
+    __stream<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(input: LLMInnerStreamOptions<Z>): Promise<StreamTextResult<any, any>>;
     __streamObject<T>(input: LLMStreamObjectOptions<T>): Promise<StreamObjectResult<DeepPartial<T>, T, never>>;
     stream<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], options?: LLMStreamOptions<Z>): Promise<StreamReturn<Z>>;
 }
@@ -659,8 +666,8 @@ declare class Agent<TTools extends Record<string, ToolAction<any, any, any, any>
             outputText: string;
         }) => Promise<void>;
     };
-    generate<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], { context, threadId: threadIdInFn, memoryOptions, resourceId, maxSteps, onStepFinish, runId, toolsets, output, temperature, toolChoice, }?: AgentGenerateOptions<Z>): Promise<GenerateReturn<Z>>;
-    stream<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], { context, threadId: threadIdInFn, memoryOptions, resourceId, maxSteps, onFinish, onStepFinish, runId, toolsets, output, temperature, toolChoice, }?: AgentStreamOptions<Z>): Promise<StreamReturn<Z>>;
+    generate<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], { context, threadId: threadIdInFn, memoryOptions, resourceId, maxSteps, onStepFinish, runId, toolsets, output, temperature, toolChoice, experimental_output, }?: AgentGenerateOptions<Z>): Promise<GenerateReturn<Z>>;
+    stream<Z extends ZodSchema | JSONSchema7 | undefined = undefined>(messages: string | string[] | CoreMessage$1[], { context, threadId: threadIdInFn, memoryOptions, resourceId, maxSteps, onFinish, onStepFinish, runId, toolsets, output, temperature, toolChoice, experimental_output, }?: AgentStreamOptions<Z>): Promise<StreamReturn<Z>>;
     /**
      * Convert text to speech using the configured voice provider
      * @param input Text or text stream to convert to speech
@@ -891,4 +898,4 @@ declare abstract class MastraStorage extends MastraBase {
     __getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
 }
 
-export { type WorkflowContext as $, Agent as A, type BaseStructuredOutputType as B, type CoreMessage as C, type LLMInnerStreamOptions as D, type EvalRow as E, type LLMStreamObjectOptions as F, type GenerateReturn as G, type MessageType as H, type IAction as I, type StorageThreadType as J, type MessageResponse as K, type LanguageModel as L, MastraStorage as M, type MemoryConfig as N, type OutputType as O, type SharedMemoryConfig as P, type CoreTool as Q, type RetryConfig as R, type StepExecutionContext as S, type ToolAction as T, type StepNode as U, type VariableReference as V, type WorkflowOptions as W, type BaseCondition as X, type ActionContext as Y, type StepDef as Z, type StepCondition as _, MastraMemory as a, type WorkflowLogMessage as a0, type WorkflowEvent as a1, type ResolverFunctionInput as a2, type ResolverFunctionOutput as a3, type SubscriberFunctionOutput as a4, type DependencyCheckOutput as a5, type WorkflowActors as a6, type WorkflowActionParams as a7, type WorkflowActions as a8, type WorkflowState as a9, type StepId as aa, type ExtractSchemaFromStep as ab, type ExtractStepResult as ac, type StepInputType as ad, type ExtractSchemaType as ae, type PathsToStringProps as af, type TABLE_NAMES as ag, TABLE_WORKFLOW_SNAPSHOT as ah, TABLE_EVALS as ai, TABLE_MESSAGES as aj, TABLE_THREADS as ak, TABLE_TRACES as al, type StepAction as b, type MastraPrimitives as c, type StepVariableType as d, type StepConfig as e, type StepResult as f, type WorkflowRunState as g, type StepGraph as h, type AgentConfig as i, type ToolExecutionContext as j, type StorageColumn as k, type WorkflowRow as l, type StorageGetMessagesArg as m, type CoreSystemMessage as n, type CoreAssistantMessage as o, type CoreUserMessage as p, type CoreToolMessage as q, type EmbedResult as r, type EmbedManyResult as s, type StructuredOutputType as t, type StructuredOutputArrayItem as u, type StructuredOutput as v, type StreamReturn as w, type LLMStreamOptions as x, type LLMTextOptions as y, type LLMTextObjectOptions as z };
+export { type StepCondition as $, Agent as A, type BaseStructuredOutputType as B, type CoreMessage as C, type LLMTextOptions as D, type EvalRow as E, type LLMTextObjectOptions as F, type GenerateReturn as G, type LLMStreamOptions as H, type IAction as I, type LLMInnerStreamOptions as J, type LLMStreamObjectOptions as K, type LanguageModel as L, MastraStorage as M, type MessageResponse as N, type OutputType as O, type MemoryConfig as P, type SharedMemoryConfig as Q, type RetryConfig as R, type StorageColumn as S, type TABLE_NAMES as T, type CoreTool as U, type StepNode as V, type WorkflowOptions as W, type VariableReference as X, type BaseCondition as Y, type ActionContext as Z, type StepDef as _, MastraMemory as a, type WorkflowContext as a0, type WorkflowLogMessage as a1, type WorkflowEvent as a2, type ResolverFunctionInput as a3, type ResolverFunctionOutput as a4, type SubscriberFunctionOutput as a5, type DependencyCheckOutput as a6, type WorkflowActors as a7, type WorkflowActionParams as a8, type WorkflowActions as a9, type WorkflowState as aa, type StepId as ab, type ExtractSchemaFromStep as ac, type ExtractStepResult as ad, type StepInputType as ae, type ExtractSchemaType as af, type PathsToStringProps as ag, TABLE_WORKFLOW_SNAPSHOT as ah, TABLE_EVALS as ai, TABLE_MESSAGES as aj, TABLE_THREADS as ak, TABLE_TRACES as al, type StorageThreadType as b, type MessageType as c, type StorageGetMessagesArg as d, type StepExecutionContext as e, type StepAction as f, type MastraPrimitives as g, type StepVariableType as h, type StepConfig as i, type StepResult as j, type WorkflowRunState as k, type StepGraph as l, type ToolAction as m, type AgentConfig as n, type ToolExecutionContext as o, type WorkflowRow as p, type CoreSystemMessage as q, type CoreAssistantMessage as r, type CoreUserMessage as s, type CoreToolMessage as t, type EmbedResult as u, type EmbedManyResult as v, type StructuredOutputType as w, type StructuredOutputArrayItem as x, type StructuredOutput as y, type StreamReturn as z };

package/dist/base.d.ts

@@ -1,6 +1,6 @@
 import '@opentelemetry/api';
-import './index-C5uPdbs4.js';
-export { M as MastraBase } from './base-eWkcLLSb.js';
-import 'pino';
+import './index-mKY1XrpK.js';
+export { M as MastraBase } from './base-D90KQ4XI.js';
 import 'stream';
+import 'pino';
 import '@opentelemetry/sdk-trace-base';

package/dist/base.js

@@ -1 +1 @@
-export { MastraBase } from './chunk-6RDA4JYW.js';
+export { MastraBase } from './chunk-OZ4XVJ6F.js';

package/dist/bundler/index.d.ts

@@ -1,8 +1,8 @@
-import { M as MastraBase } from '../base-eWkcLLSb.js';
+import { M as MastraBase } from '../base-D90KQ4XI.js';
 import '@opentelemetry/api';
-import '../index-C5uPdbs4.js';
-import 'pino';
+import '../index-mKY1XrpK.js';
 import 'stream';
+import 'pino';
 import '@opentelemetry/sdk-trace-base';
 
 interface IBundler {

package/dist/bundler/index.js

@@ -1 +1 @@
-export { MastraBundler } from '../chunk-YUOADWSP.js';
+export { MastraBundler } from '../chunk-SVEAENO7.js';

package/dist/{chunk-LKOVXFLE.js → chunk-33GSTUNK.js}

@@ -1,7 +1,7 @@
-import { MastraVector } from './chunk-7NPRQT5A.js';
+import { MastraVector } from './chunk-55NFNRKO.js';
 import { BaseFilterTranslator } from './chunk-SIFBBGY6.js';
-import { createClient } from '@libsql/client';
 import { join } from 'path';
+import { createClient } from '@libsql/client';
 
 // src/vector/libsql/filter.ts
 var LibSQLFilterTranslator = class extends BaseFilterTranslator {
@@ -414,7 +414,7 @@ var processOperator = (key, operator, operatorValue) => {
 };
 
 // src/vector/libsql/index.ts
-var DefaultVectorDB = class extends MastraVector {
+var LibSQLVector = class extends MastraVector {
   turso;
   constructor({
     connectionUrl,
@@ -617,4 +617,4 @@ var DefaultVectorDB = class extends MastraVector {
   }
 };
 
-export { DefaultVectorDB };
+export { LibSQLVector };

package/dist/{chunk-Z735LDV7.js → chunk-4YRYBCOZ.js}

@@ -1,4 +1,4 @@
-import { MastraBundler } from './chunk-YUOADWSP.js';
+import { MastraBundler } from './chunk-SVEAENO7.js';
 
 // src/deployer/index.ts
 var MastraDeployer = class extends MastraBundler {

package/dist/chunk-55NFNRKO.js

@@ -0,0 +1,10 @@
+import { MastraBase } from './chunk-OZ4XVJ6F.js';
+
+// src/vector/index.ts
+var MastraVector = class extends MastraBase {
+  constructor() {
+    super({ name: "MastraVector", component: "VECTOR" });
+  }
+};
+
+export { MastraVector };

package/dist/chunk-5XPCMNGW.js

@@ -0,0 +1,215 @@
+import { TABLE_WORKFLOW_SNAPSHOT, TABLE_EVALS, TABLE_MESSAGES, TABLE_THREADS, TABLE_TRACES } from './chunk-RG66XEJT.js';
+import { MastraBase } from './chunk-OZ4XVJ6F.js';
+
+// src/storage/base.ts
+var MastraStorage = class extends MastraBase {
+  /** @deprecated import from { TABLE_WORKFLOW_SNAPSHOT } '@mastra/core/storage' instead */
+  static TABLE_WORKFLOW_SNAPSHOT = TABLE_WORKFLOW_SNAPSHOT;
+  /** @deprecated import from { TABLE_EVALS } '@mastra/core/storage' instead */
+  static TABLE_EVALS = TABLE_EVALS;
+  /** @deprecated import from { TABLE_MESSAGES } '@mastra/core/storage' instead */
+  static TABLE_MESSAGES = TABLE_MESSAGES;
+  /** @deprecated import from { TABLE_THREADS } '@mastra/core/storage' instead */
+  static TABLE_THREADS = TABLE_THREADS;
+  /** @deprecated import { TABLE_TRACES } from '@mastra/core/storage' instead */
+  static TABLE_TRACES = TABLE_TRACES;
+  hasInitialized = null;
+  constructor({ name }) {
+    super({
+      component: "STORAGE",
+      name
+    });
+  }
+  async __batchInsert({
+    tableName,
+    records
+  }) {
+    await this.init();
+    return this.batchInsert({ tableName, records });
+  }
+  async __getThreadById({ threadId }) {
+    await this.init();
+    return this.getThreadById({ threadId });
+  }
+  async __getThreadsByResourceId({ resourceId }) {
+    await this.init();
+    return this.getThreadsByResourceId({ resourceId });
+  }
+  async __saveThread({ thread }) {
+    await this.init();
+    return this.saveThread({ thread });
+  }
+  async __updateThread({
+    id,
+    title,
+    metadata
+  }) {
+    await this.init();
+    return this.updateThread({ id, title, metadata });
+  }
+  async __deleteThread({ threadId }) {
+    await this.init();
+    return this.deleteThread({ threadId });
+  }
+  async __getMessages({ threadId, selectBy, threadConfig }) {
+    await this.init();
+    return this.getMessages({ threadId, selectBy, threadConfig });
+  }
+  async __saveMessages({ messages }) {
+    await this.init();
+    return this.saveMessages({ messages });
+  }
+  async __getTraces({
+    scope,
+    page,
+    perPage,
+    attributes
+  }) {
+    await this.init();
+    return this.getTraces({ scope, page, perPage, attributes });
+  }
+  async init() {
+    if (await this.hasInitialized) {
+      return;
+    }
+    this.hasInitialized = Promise.all([
+      this.createTable({
+        tableName: TABLE_WORKFLOW_SNAPSHOT,
+        schema: {
+          workflow_name: {
+            type: "text"
+          },
+          run_id: {
+            type: "text"
+          },
+          snapshot: {
+            type: "text"
+          },
+          createdAt: {
+            type: "timestamp"
+          },
+          updatedAt: {
+            type: "timestamp"
+          }
+        }
+      }),
+      this.createTable({
+        tableName: TABLE_EVALS,
+        schema: {
+          input: {
+            type: "text"
+          },
+          output: {
+            type: "text"
+          },
+          result: {
+            type: "jsonb"
+          },
+          agent_name: {
+            type: "text"
+          },
+          metric_name: {
+            type: "text"
+          },
+          instructions: {
+            type: "text"
+          },
+          test_info: {
+            type: "jsonb",
+            nullable: true
+          },
+          global_run_id: {
+            type: "text"
+          },
+          run_id: {
+            type: "text"
+          },
+          created_at: {
+            type: "timestamp"
+          }
+        }
+      }),
+      this.createTable({
+        tableName: TABLE_THREADS,
+        schema: {
+          id: { type: "text", nullable: false, primaryKey: true },
+          resourceId: { type: "text", nullable: false },
+          title: { type: "text", nullable: false },
+          metadata: { type: "text", nullable: true },
+          createdAt: { type: "timestamp", nullable: false },
+          updatedAt: { type: "timestamp", nullable: false }
+        }
+      }),
+      this.createTable({
+        tableName: TABLE_MESSAGES,
+        schema: {
+          id: { type: "text", nullable: false, primaryKey: true },
+          thread_id: { type: "text", nullable: false },
+          content: { type: "text", nullable: false },
+          role: { type: "text", nullable: false },
+          type: { type: "text", nullable: false },
+          createdAt: { type: "timestamp", nullable: false }
+        }
+      }),
+      this.createTable({
+        tableName: TABLE_TRACES,
+        schema: {
+          id: { type: "text", nullable: false, primaryKey: true },
+          parentSpanId: { type: "text", nullable: true },
+          name: { type: "text", nullable: false },
+          traceId: { type: "text", nullable: false },
+          scope: { type: "text", nullable: false },
+          kind: { type: "integer", nullable: false },
+          attributes: { type: "jsonb", nullable: true },
+          status: { type: "jsonb", nullable: true },
+          events: { type: "jsonb", nullable: true },
+          links: { type: "jsonb", nullable: true },
+          other: { type: "text", nullable: true },
+          startTime: { type: "bigint", nullable: false },
+          endTime: { type: "bigint", nullable: false },
+          createdAt: { type: "timestamp", nullable: false }
+        }
+      })
+    ]).then(() => true);
+    await this.hasInitialized;
+  }
+  async persistWorkflowSnapshot({
+    workflowName,
+    runId,
+    snapshot
+  }) {
+    await this.init();
+    const data = {
+      workflow_name: workflowName,
+      run_id: runId,
+      snapshot,
+      createdAt: /* @__PURE__ */ new Date(),
+      updatedAt: /* @__PURE__ */ new Date()
+    };
+    this.logger.debug("Persisting workflow snapshot", { workflowName, runId, data });
+    await this.insert({
+      tableName: TABLE_WORKFLOW_SNAPSHOT,
+      record: data
+    });
+  }
+  async loadWorkflowSnapshot({
+    workflowName,
+    runId
+  }) {
+    if (!this.hasInitialized) {
+      await this.init();
+    }
+    this.logger.debug("Loading workflow snapshot", { workflowName, runId });
+    const d = await this.load({
+      tableName: TABLE_WORKFLOW_SNAPSHOT,
+      keys: { workflow_name: workflowName, run_id: runId }
+    });
+    return d ? d.snapshot : null;
+  }
+  async __getEvalsByAgentName(agentName, type) {
+    await this.init();
+    return this.getEvalsByAgentName(agentName, type);
+  }
+};
+
+export { MastraStorage };