@mastra/libsql 1.9.1-alpha.0 → 1.10.0-alpha.0

package/dist/index.js

@@ -1,6 +1,6 @@
 import { createClient } from '@libsql/client';
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { createVectorErrorId, AgentsStorage, AGENTS_SCHEMA, TABLE_AGENTS, AGENT_VERSIONS_SCHEMA, TABLE_AGENT_VERSIONS, createStorageErrorId, normalizePerPage, calculatePagination, BackgroundTasksStorage, TABLE_SCHEMAS, TABLE_BACKGROUND_TASKS, BlobStore, TABLE_SKILL_BLOBS, SKILL_BLOBS_SCHEMA, ChannelsStorage, TABLE_CHANNEL_INSTALLATIONS, TABLE_CHANNEL_CONFIG, DatasetsStorage, DATASETS_SCHEMA, TABLE_DATASETS, DATASET_ITEMS_SCHEMA, TABLE_DATASET_ITEMS, DATASET_VERSIONS_SCHEMA, TABLE_DATASET_VERSIONS, ensureDate, safelyParseJSON, TABLE_EXPERIMENT_RESULTS, TABLE_EXPERIMENTS, ExperimentsStorage, EXPERIMENTS_SCHEMA, EXPERIMENT_RESULTS_SCHEMA, MCPClientsStorage, MCP_CLIENTS_SCHEMA, TABLE_MCP_CLIENTS, MCP_CLIENT_VERSIONS_SCHEMA, TABLE_MCP_CLIENT_VERSIONS, MCPServersStorage, MCP_SERVERS_SCHEMA, TABLE_MCP_SERVERS, MCP_SERVER_VERSIONS_SCHEMA, TABLE_MCP_SERVER_VERSIONS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ObservabilityStorage, SPAN_SCHEMA, TABLE_SPANS, listTracesArgsSchema, toTraceSpans, PromptBlocksStorage, PROMPT_BLOCKS_SCHEMA, TABLE_PROMPT_BLOCKS, PROMPT_BLOCK_VERSIONS_SCHEMA, TABLE_PROMPT_BLOCK_VERSIONS, ScorerDefinitionsStorage, SCORER_DEFINITIONS_SCHEMA, TABLE_SCORER_DEFINITIONS, SCORER_DEFINITION_VERSIONS_SCHEMA, TABLE_SCORER_DEFINITION_VERSIONS, ScoresStorage, SCORERS_SCHEMA, TABLE_SCORERS, transformScoreRow, SkillsStorage, SKILLS_SCHEMA, TABLE_SKILLS, SKILL_VERSIONS_SCHEMA, TABLE_SKILL_VERSIONS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, WorkspacesStorage, WORKSPACES_SCHEMA, TABLE_WORKSPACES, WORKSPACE_VERSIONS_SCHEMA, TABLE_WORKSPACE_VERSIONS, MastraCompositeStore, getSqlType, TraceStatus } from '@mastra/core/storage';
+import { createVectorErrorId, AgentsStorage, AGENTS_SCHEMA, TABLE_AGENTS, AGENT_VERSIONS_SCHEMA, TABLE_AGENT_VERSIONS, createStorageErrorId, normalizePerPage, calculatePagination, BackgroundTasksStorage, TABLE_SCHEMAS, TABLE_BACKGROUND_TASKS, BlobStore, TABLE_SKILL_BLOBS, SKILL_BLOBS_SCHEMA, ChannelsStorage, TABLE_CHANNEL_INSTALLATIONS, TABLE_CHANNEL_CONFIG, DatasetsStorage, DATASETS_SCHEMA, TABLE_DATASETS, DATASET_ITEMS_SCHEMA, TABLE_DATASET_ITEMS, DATASET_VERSIONS_SCHEMA, TABLE_DATASET_VERSIONS, ensureDate, safelyParseJSON, TABLE_EXPERIMENT_RESULTS, TABLE_EXPERIMENTS, ExperimentsStorage, EXPERIMENTS_SCHEMA, EXPERIMENT_RESULTS_SCHEMA, MCPClientsStorage, MCP_CLIENTS_SCHEMA, TABLE_MCP_CLIENTS, MCP_CLIENT_VERSIONS_SCHEMA, TABLE_MCP_CLIENT_VERSIONS, MCPServersStorage, MCP_SERVERS_SCHEMA, TABLE_MCP_SERVERS, MCP_SERVER_VERSIONS_SCHEMA, TABLE_MCP_SERVER_VERSIONS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ObservabilityStorage, SPAN_SCHEMA, TABLE_SPANS, listTracesArgsSchema, toTraceSpans, PromptBlocksStorage, PROMPT_BLOCKS_SCHEMA, TABLE_PROMPT_BLOCKS, PROMPT_BLOCK_VERSIONS_SCHEMA, TABLE_PROMPT_BLOCK_VERSIONS, SchedulesStorage, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, ScorerDefinitionsStorage, SCORER_DEFINITIONS_SCHEMA, TABLE_SCORER_DEFINITIONS, SCORER_DEFINITION_VERSIONS_SCHEMA, TABLE_SCORER_DEFINITION_VERSIONS, ScoresStorage, SCORERS_SCHEMA, TABLE_SCORERS, transformScoreRow, SkillsStorage, SKILLS_SCHEMA, TABLE_SKILLS, SKILL_VERSIONS_SCHEMA, TABLE_SKILL_VERSIONS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, WorkspacesStorage, WORKSPACES_SCHEMA, TABLE_WORKSPACES, WORKSPACE_VERSIONS_SCHEMA, TABLE_WORKSPACE_VERSIONS, MastraCompositeStore, getSqlType, TraceStatus } from '@mastra/core/storage';
 import { parseSqlIdentifier, parseFieldKey } from '@mastra/core/utils';
 import { MastraVector, validateTopK, validateUpsertInput } from '@mastra/core/vector';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
@@ -8894,6 +8894,233 @@ var PromptBlocksLibSQL = class extends PromptBlocksStorage {
     };
   }
 };
+function parseJson2(val) {
+  if (val == null) return void 0;
+  if (typeof val === "string") {
+    try {
+      return JSON.parse(val);
+    } catch {
+      return val;
+    }
+  }
+  return val;
+}
+function toNumber(val) {
+  if (typeof val === "bigint") return Number(val);
+  return Number(val);
+}
+function rowToSchedule(row) {
+  const target = parseJson2(row.target);
+  if (!target) {
+    throw new Error(`Schedule row ${row.id} has invalid target`);
+  }
+  const schedule = {
+    id: String(row.id),
+    target,
+    cron: String(row.cron),
+    status: String(row.status),
+    nextFireAt: toNumber(row.next_fire_at),
+    createdAt: toNumber(row.created_at),
+    updatedAt: toNumber(row.updated_at)
+  };
+  if (row.timezone != null) schedule.timezone = String(row.timezone);
+  if (row.last_fire_at != null) schedule.lastFireAt = toNumber(row.last_fire_at);
+  if (row.last_run_id != null) schedule.lastRunId = String(row.last_run_id);
+  const metadata = parseJson2(row.metadata);
+  if (metadata !== void 0) schedule.metadata = metadata;
+  return schedule;
+}
+function rowToTrigger(row) {
+  const trigger = {
+    scheduleId: String(row.schedule_id),
+    runId: String(row.run_id),
+    scheduledFireAt: toNumber(row.scheduled_fire_at),
+    actualFireAt: toNumber(row.actual_fire_at),
+    status: String(row.status)
+  };
+  if (row.error != null) trigger.error = String(row.error);
+  return trigger;
+}
+var SchedulesLibSQL = class extends SchedulesStorage {
+  #db;
+  #client;
+  constructor(config) {
+    super();
+    const client = resolveClient(config);
+    this.#client = client;
+    this.#db = new LibSQLDB({ client, maxRetries: config.maxRetries, initialBackoffMs: config.initialBackoffMs });
+  }
+  async init() {
+    await this.#db.createTable({
+      tableName: TABLE_SCHEDULES,
+      schema: TABLE_SCHEMAS[TABLE_SCHEDULES]
+    });
+    await this.#db.createTable({
+      tableName: TABLE_SCHEDULE_TRIGGERS,
+      schema: TABLE_SCHEMAS[TABLE_SCHEDULE_TRIGGERS]
+    });
+  }
+  async dangerouslyClearAll() {
+    await this.#db.deleteData({ tableName: TABLE_SCHEDULE_TRIGGERS });
+    await this.#db.deleteData({ tableName: TABLE_SCHEDULES });
+  }
+  async createSchedule(schedule) {
+    const existing = await this.getSchedule(schedule.id);
+    if (existing) {
+      throw new Error(`Schedule with id "${schedule.id}" already exists`);
+    }
+    await this.#db.insert({
+      tableName: TABLE_SCHEDULES,
+      record: {
+        id: schedule.id,
+        target: schedule.target,
+        cron: schedule.cron,
+        timezone: schedule.timezone ?? null,
+        status: schedule.status,
+        next_fire_at: schedule.nextFireAt,
+        last_fire_at: schedule.lastFireAt ?? null,
+        last_run_id: schedule.lastRunId ?? null,
+        created_at: schedule.createdAt,
+        updated_at: schedule.updatedAt,
+        metadata: schedule.metadata ?? null
+      }
+    });
+    return schedule;
+  }
+  async getSchedule(id) {
+    const result = await this.#client.execute({
+      sql: `SELECT ${buildSelectColumns(TABLE_SCHEDULES)} FROM ${TABLE_SCHEDULES} WHERE id = ?`,
+      args: [id]
+    });
+    const row = result.rows[0];
+    return row ? rowToSchedule(row) : null;
+  }
+  async listSchedules(filter) {
+    const conditions = [];
+    const params = [];
+    if (filter?.status) {
+      conditions.push("status = ?");
+      params.push(filter.status);
+    }
+    if (filter?.workflowId) {
+      conditions.push("json_extract(target, '$.workflowId') = ?");
+      params.push(filter.workflowId);
+    }
+    const where = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
+    const result = await this.#client.execute({
+      sql: `SELECT ${buildSelectColumns(TABLE_SCHEDULES)} FROM ${TABLE_SCHEDULES} ${where} ORDER BY created_at ASC`,
+      args: params
+    });
+    return result.rows.map((r) => rowToSchedule(r));
+  }
+  async listDueSchedules(now, limit) {
+    const cap = limit ?? 100;
+    const result = await this.#client.execute({
+      sql: `SELECT ${buildSelectColumns(TABLE_SCHEDULES)} FROM ${TABLE_SCHEDULES}
+            WHERE status = ? AND next_fire_at <= ?
+            ORDER BY next_fire_at ASC
+            LIMIT ?`,
+      args: ["active", now, cap]
+    });
+    return result.rows.map((r) => rowToSchedule(r));
+  }
+  async updateSchedule(id, patch) {
+    const setClauses = [];
+    const params = [];
+    if ("cron" in patch && patch.cron !== void 0) {
+      setClauses.push("cron = ?");
+      params.push(patch.cron);
+    }
+    if ("timezone" in patch) {
+      setClauses.push("timezone = ?");
+      params.push(patch.timezone ?? null);
+    }
+    if ("status" in patch && patch.status !== void 0) {
+      setClauses.push("status = ?");
+      params.push(patch.status);
+    }
+    if ("nextFireAt" in patch && patch.nextFireAt !== void 0) {
+      setClauses.push("next_fire_at = ?");
+      params.push(patch.nextFireAt);
+    }
+    if ("target" in patch && patch.target !== void 0) {
+      setClauses.push("target = jsonb(?)");
+      params.push(JSON.stringify(patch.target));
+    }
+    if ("metadata" in patch) {
+      setClauses.push("metadata = jsonb(?)");
+      params.push(patch.metadata != null ? JSON.stringify(patch.metadata) : null);
+    }
+    setClauses.push("updated_at = ?");
+    params.push(Date.now());
+    params.push(id);
+    if (setClauses.length === 1) {
+      const existing = await this.getSchedule(id);
+      if (!existing) throw new Error(`Schedule ${id} not found`);
+      return existing;
+    }
+    await this.#client.execute({
+      sql: `UPDATE ${TABLE_SCHEDULES} SET ${setClauses.join(", ")} WHERE id = ?`,
+      args: params
+    });
+    const updated = await this.getSchedule(id);
+    if (!updated) throw new Error(`Schedule ${id} not found`);
+    return updated;
+  }
+  async updateScheduleNextFire(id, expectedNextFireAt, newNextFireAt, lastFireAt, lastRunId) {
+    const result = await this.#client.execute({
+      sql: `UPDATE ${TABLE_SCHEDULES}
+            SET next_fire_at = ?, last_fire_at = ?, last_run_id = ?, updated_at = ?
+            WHERE id = ? AND next_fire_at = ? AND status = ?`,
+      args: [newNextFireAt, lastFireAt, lastRunId, Date.now(), id, expectedNextFireAt, "active"]
+    });
+    return (result.rowsAffected ?? 0) > 0;
+  }
+  async deleteSchedule(id) {
+    await this.#client.execute({
+      sql: `DELETE FROM ${TABLE_SCHEDULE_TRIGGERS} WHERE schedule_id = ?`,
+      args: [id]
+    });
+    await this.#client.execute({
+      sql: `DELETE FROM ${TABLE_SCHEDULES} WHERE id = ?`,
+      args: [id]
+    });
+  }
+  async recordTrigger(trigger) {
+    await this.#db.insert({
+      tableName: TABLE_SCHEDULE_TRIGGERS,
+      record: {
+        schedule_id: trigger.scheduleId,
+        run_id: trigger.runId,
+        scheduled_fire_at: trigger.scheduledFireAt,
+        actual_fire_at: trigger.actualFireAt,
+        status: trigger.status,
+        error: trigger.error ?? null
+      }
+    });
+  }
+  async listTriggers(scheduleId, opts) {
+    const conditions = ["schedule_id = ?"];
+    const params = [scheduleId];
+    if (opts?.fromActualFireAt != null) {
+      conditions.push("actual_fire_at >= ?");
+      params.push(opts.fromActualFireAt);
+    }
+    if (opts?.toActualFireAt != null) {
+      conditions.push("actual_fire_at < ?");
+      params.push(opts.toActualFireAt);
+    }
+    const limitClause = opts?.limit != null ? `LIMIT ${Math.floor(opts.limit)}` : "";
+    const result = await this.#client.execute({
+      sql: `SELECT ${buildSelectColumns(TABLE_SCHEDULE_TRIGGERS)} FROM ${TABLE_SCHEDULE_TRIGGERS}
+            WHERE ${conditions.join(" AND ")}
+            ORDER BY actual_fire_at DESC
+            ${limitClause}`,
+      args: params
+    });
+    return result.rows.map((r) => rowToTrigger(r));
+  }
+};
 var ScorerDefinitionsLibSQL = class extends ScorerDefinitionsStorage {
   #db;
   #client;
@@ -11125,6 +11352,7 @@ var LibSQLStore = class extends MastraCompositeStore {
     const skills = new SkillsLibSQL(domainConfig);
     const blobs = new BlobsLibSQL(domainConfig);
     const backgroundTasks = new BackgroundTasksLibSQL(domainConfig);
+    const schedules = new SchedulesLibSQL(domainConfig);
     this.stores = {
       scores,
       workflows,
@@ -11141,7 +11369,8 @@ var LibSQLStore = class extends MastraCompositeStore {
       workspaces,
       skills,
       blobs,
-      backgroundTasks
+      backgroundTasks,
+      schedules
     };
   }
 };
@@ -11245,6 +11474,6 @@ Example Complex Query:
   ]
 }`;
 
-export { AgentsLibSQL, BackgroundTasksLibSQL, BlobsLibSQL, ChannelsLibSQL, DatasetsLibSQL, LibSQLStore as DefaultStorage, ExperimentsLibSQL, LIBSQL_PROMPT, LibSQLStore, LibSQLVector, MCPClientsLibSQL, MCPServersLibSQL, MemoryLibSQL, ObservabilityLibSQL, PromptBlocksLibSQL, ScorerDefinitionsLibSQL, ScoresLibSQL, SkillsLibSQL, WorkflowsLibSQL, WorkspacesLibSQL };
+export { AgentsLibSQL, BackgroundTasksLibSQL, BlobsLibSQL, ChannelsLibSQL, DatasetsLibSQL, LibSQLStore as DefaultStorage, ExperimentsLibSQL, LIBSQL_PROMPT, LibSQLStore, LibSQLVector, MCPClientsLibSQL, MCPServersLibSQL, MemoryLibSQL, ObservabilityLibSQL, PromptBlocksLibSQL, SchedulesLibSQL, ScorerDefinitionsLibSQL, ScoresLibSQL, SkillsLibSQL, WorkflowsLibSQL, WorkspacesLibSQL };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map