@mastra/libsql 1.1.0-alpha.1 → 1.1.0

package/dist/index.js

@@ -2041,14 +2041,122 @@ Note: This migration may take some time for large tables.
 // src/storage/domains/agents/index.ts
 var AgentsLibSQL = class extends AgentsStorage {
   #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.#migrateFromLegacySchema();
+    await this.#migrateVersionsSchema();
+    await this.#db.createTable({ tableName: TABLE_AGENTS, schema: AGENTS_SCHEMA });
+    await this.#db.createTable({ tableName: TABLE_AGENT_VERSIONS, schema: AGENT_VERSIONS_SCHEMA });
+    await this.#db.alterTable({
+      tableName: TABLE_AGENTS,
+      schema: AGENTS_SCHEMA,
+      ifNotExists: ["status", "authorId"]
+    });
+    await this.#cleanupStaleDrafts();
+  }
+  /**
+   * Migrates from the legacy flat agent schema (where config fields like name, instructions, model
+   * were stored directly on mastra_agents) to the new versioned schema (thin agent record + versions table).
+   * SQLite cannot drop columns or alter NOT NULL constraints, so we must recreate the table.
+   */
+  async #migrateFromLegacySchema() {
+    const legacyTable = `${TABLE_AGENTS}_legacy`;
+    const hasLegacyColumns = await this.#db.hasColumn(TABLE_AGENTS, "name");
+    if (hasLegacyColumns) {
+      await this.#client.execute({
+        sql: `ALTER TABLE "${TABLE_AGENTS}" RENAME TO "${legacyTable}"`
+      });
+      await this.#client.execute({
+        sql: `DROP TABLE IF EXISTS "${TABLE_AGENT_VERSIONS}"`
+      });
+    }
+    const legacyExists = await this.#db.hasColumn(legacyTable, "name");
+    if (!legacyExists) return;
+    const result = await this.#client.execute({
+      sql: `SELECT * FROM "${legacyTable}"`
+    });
+    const oldAgents = result.rows || [];
     await this.#db.createTable({ tableName: TABLE_AGENTS, schema: AGENTS_SCHEMA });
     await this.#db.createTable({ tableName: TABLE_AGENT_VERSIONS, schema: AGENT_VERSIONS_SCHEMA });
+    for (const row of oldAgents) {
+      const agentId = row.id;
+      if (!agentId) continue;
+      const versionId = crypto.randomUUID();
+      const now = /* @__PURE__ */ new Date();
+      await this.#db.insert({
+        tableName: TABLE_AGENTS,
+        record: {
+          id: agentId,
+          status: "published",
+          activeVersionId: versionId,
+          authorId: row.ownerId ?? row.authorId ?? null,
+          metadata: row.metadata ?? null,
+          createdAt: row.createdAt ?? now,
+          updatedAt: row.updatedAt ?? now
+        }
+      });
+      await this.#db.insert({
+        tableName: TABLE_AGENT_VERSIONS,
+        record: {
+          id: versionId,
+          agentId,
+          versionNumber: 1,
+          name: row.name ?? agentId,
+          description: row.description ?? null,
+          instructions: row.instructions ?? "",
+          model: row.model ?? "{}",
+          tools: row.tools ?? null,
+          defaultOptions: row.defaultOptions ?? null,
+          workflows: row.workflows ?? null,
+          agents: row.agents ?? null,
+          integrationTools: row.integrationTools ?? null,
+          inputProcessors: row.inputProcessors ?? null,
+          outputProcessors: row.outputProcessors ?? null,
+          memory: row.memory ?? null,
+          scorers: row.scorers ?? null,
+          changedFields: null,
+          changeMessage: "Migrated from legacy schema",
+          createdAt: row.createdAt ?? now
+        }
+      });
+    }
+    await this.#client.execute({
+      sql: `DROP TABLE IF EXISTS "${legacyTable}"`
+    });
+  }
+  /**
+   * Migrates the agent_versions table from the old snapshot-based schema (single `snapshot` JSON column)
+   * to the new flat schema (individual config columns). This handles the case where the agents table
+   * was already migrated but the versions table still has the old schema.
+   */
+  async #migrateVersionsSchema() {
+    const hasSnapshotColumn = await this.#db.hasColumn(TABLE_AGENT_VERSIONS, "snapshot");
+    if (!hasSnapshotColumn) return;
+    await this.#client.execute({
+      sql: `DROP TABLE IF EXISTS "${TABLE_AGENT_VERSIONS}"`
+    });
+    await this.#client.execute({
+      sql: `DROP TABLE IF EXISTS "${TABLE_AGENTS}_legacy"`
+    });
+  }
+  /**
+   * Removes stale draft agent records that have no activeVersionId.
+   * These are left behind when createAgent partially fails (inserts thin record
+   * but fails to create the version due to schema mismatch).
+   */
+  async #cleanupStaleDrafts() {
+    try {
+      await this.#client.execute({
+        sql: `DELETE FROM "${TABLE_AGENTS}" WHERE status = 'draft' AND activeVersionId IS NULL`
+      });
+    } catch {
+    }
   }
   async dangerouslyClearAll() {
     await this.#db.deleteData({ tableName: TABLE_AGENT_VERSIONS });