@mastra/libsql 1.0.0 → 1.1.0-alpha.1

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, createStorageErrorId, normalizePerPage, calculatePagination, MemoryStorage, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ObservabilityStorage, SPAN_SCHEMA, TABLE_SPANS, listTracesArgsSchema, ScoresStorage, SCORERS_SCHEMA, TABLE_SCORERS, transformScoreRow, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, MastraCompositeStore, TraceStatus, getSqlType, safelyParseJSON } from '@mastra/core/storage';
+import { createVectorErrorId, AgentsStorage, AGENTS_SCHEMA, TABLE_AGENTS, AGENT_VERSIONS_SCHEMA, TABLE_AGENT_VERSIONS, createStorageErrorId, normalizePerPage, calculatePagination, MemoryStorage, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ObservabilityStorage, SPAN_SCHEMA, TABLE_SPANS, listTracesArgsSchema, toTraceSpans, ScoresStorage, SCORERS_SCHEMA, TABLE_SCORERS, transformScoreRow, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, MastraCompositeStore, TraceStatus, getSqlType, safelyParseJSON } 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';
@@ -2048,8 +2048,10 @@ var AgentsLibSQL = class extends AgentsStorage {
   }
   async init() {
     await this.#db.createTable({ tableName: TABLE_AGENTS, schema: AGENTS_SCHEMA });
+    await this.#db.createTable({ tableName: TABLE_AGENT_VERSIONS, schema: AGENT_VERSIONS_SCHEMA });
   }
   async dangerouslyClearAll() {
+    await this.#db.deleteData({ tableName: TABLE_AGENT_VERSIONS });
     await this.#db.deleteData({ tableName: TABLE_AGENTS });
   }
   parseJson(value, fieldName) {
@@ -2079,18 +2081,9 @@ var AgentsLibSQL = class extends AgentsStorage {
   parseRow(row) {
     return {
       id: row.id,
-      name: row.name,
-      description: row.description,
-      instructions: row.instructions,
-      model: this.parseJson(row.model, "model"),
-      tools: this.parseJson(row.tools, "tools"),
-      defaultOptions: this.parseJson(row.defaultOptions, "defaultOptions"),
-      workflows: this.parseJson(row.workflows, "workflows"),
-      agents: this.parseJson(row.agents, "agents"),
-      inputProcessors: this.parseJson(row.inputProcessors, "inputProcessors"),
-      outputProcessors: this.parseJson(row.outputProcessors, "outputProcessors"),
-      memory: this.parseJson(row.memory, "memory"),
-      scorers: this.parseJson(row.scorers, "scorers"),
+      status: row.status,
+      activeVersionId: row.activeVersionId,
+      authorId: row.authorId,
       metadata: this.parseJson(row.metadata, "metadata"),
       createdAt: new Date(row.createdAt),
       updatedAt: new Date(row.updatedAt)
@@ -2122,29 +2115,48 @@ var AgentsLibSQL = class extends AgentsStorage {
         tableName: TABLE_AGENTS,
         record: {
           id: agent.id,
-          name: agent.name,
-          description: agent.description ?? null,
-          instructions: agent.instructions,
-          model: agent.model,
-          tools: agent.tools ?? null,
-          defaultOptions: agent.defaultOptions ?? null,
-          workflows: agent.workflows ?? null,
-          agents: agent.agents ?? null,
-          inputProcessors: agent.inputProcessors ?? null,
-          outputProcessors: agent.outputProcessors ?? null,
-          memory: agent.memory ?? null,
-          scorers: agent.scorers ?? null,
+          status: "draft",
+          activeVersionId: null,
+          authorId: agent.authorId ?? null,
           metadata: agent.metadata ?? null,
           createdAt: now,
           updatedAt: now
         }
       });
-      return {
-        ...agent,
-        createdAt: now,
-        updatedAt: now
-      };
+      const { id: _id, authorId: _authorId, metadata: _metadata, ...snapshotConfig } = agent;
+      const versionId = crypto.randomUUID();
+      await this.createVersion({
+        id: versionId,
+        agentId: agent.id,
+        versionNumber: 1,
+        ...snapshotConfig,
+        changedFields: Object.keys(snapshotConfig),
+        changeMessage: "Initial version"
+      });
+      await this.#db.update({
+        tableName: TABLE_AGENTS,
+        keys: { id: agent.id },
+        data: {
+          activeVersionId: versionId,
+          status: "published",
+          updatedAt: /* @__PURE__ */ new Date()
+        }
+      });
+      const created = await this.getAgentById({ id: agent.id });
+      if (!created) {
+        throw new MastraError({
+          id: createStorageErrorId("LIBSQL", "CREATE_AGENT", "NOT_FOUND_AFTER_CREATE"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.SYSTEM,
+          text: `Agent ${agent.id} not found after creation`,
+          details: { agentId: agent.id }
+        });
+      }
+      return created;
     } catch (error) {
+      if (error instanceof MastraError) {
+        throw error;
+      }
       throw new MastraError(
         {
           id: createStorageErrorId("LIBSQL", "CREATE_AGENT", "FAILED"),
@@ -2171,18 +2183,11 @@ var AgentsLibSQL = class extends AgentsStorage {
       const data = {
         updatedAt: /* @__PURE__ */ new Date()
       };
-      if (updates.name !== void 0) data.name = updates.name;
-      if (updates.description !== void 0) data.description = updates.description;
-      if (updates.instructions !== void 0) data.instructions = updates.instructions;
-      if (updates.model !== void 0) data.model = updates.model;
-      if (updates.tools !== void 0) data.tools = updates.tools;
-      if (updates.defaultOptions !== void 0) data.defaultOptions = updates.defaultOptions;
-      if (updates.workflows !== void 0) data.workflows = updates.workflows;
-      if (updates.agents !== void 0) data.agents = updates.agents;
-      if (updates.inputProcessors !== void 0) data.inputProcessors = updates.inputProcessors;
-      if (updates.outputProcessors !== void 0) data.outputProcessors = updates.outputProcessors;
-      if (updates.memory !== void 0) data.memory = updates.memory;
-      if (updates.scorers !== void 0) data.scorers = updates.scorers;
+      if (updates.authorId !== void 0) data.authorId = updates.authorId;
+      if (updates.activeVersionId !== void 0) {
+        data.activeVersionId = updates.activeVersionId;
+        data.status = "published";
+      }
       if (updates.metadata !== void 0) {
         data.metadata = { ...existingAgent.metadata, ...updates.metadata };
       }
@@ -2221,6 +2226,7 @@ var AgentsLibSQL = class extends AgentsStorage {
   }
   async deleteAgent({ id }) {
     try {
+      await this.deleteVersionsByAgentId(id);
       await this.#db.delete({
         tableName: TABLE_AGENTS,
         keys: { id }
@@ -2290,6 +2296,284 @@ var AgentsLibSQL = class extends AgentsStorage {
       );
     }
   }
+  // ==========================================================================
+  // Agent Version Methods
+  // ==========================================================================
+  async createVersion(input) {
+    try {
+      const now = /* @__PURE__ */ new Date();
+      await this.#db.insert({
+        tableName: TABLE_AGENT_VERSIONS,
+        record: {
+          id: input.id,
+          agentId: input.agentId,
+          versionNumber: input.versionNumber,
+          name: input.name ?? null,
+          description: input.description ?? null,
+          instructions: input.instructions,
+          model: input.model,
+          tools: input.tools ?? null,
+          defaultOptions: input.defaultOptions ?? null,
+          workflows: input.workflows ?? null,
+          agents: input.agents ?? null,
+          integrationTools: input.integrationTools ?? null,
+          inputProcessors: input.inputProcessors ?? null,
+          outputProcessors: input.outputProcessors ?? null,
+          memory: input.memory ?? null,
+          scorers: input.scorers ?? null,
+          changedFields: input.changedFields ?? null,
+          changeMessage: input.changeMessage ?? null,
+          createdAt: now
+        }
+      });
+      return {
+        ...input,
+        createdAt: now
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "CREATE_VERSION", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { versionId: input.id, agentId: input.agentId }
+        },
+        error
+      );
+    }
+  }
+  async getVersion(id) {
+    try {
+      const result = await this.#db.select({
+        tableName: TABLE_AGENT_VERSIONS,
+        keys: { id }
+      });
+      if (!result) {
+        return null;
+      }
+      return this.parseVersionRow(result);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "GET_VERSION", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { versionId: id }
+        },
+        error
+      );
+    }
+  }
+  async getVersionByNumber(agentId, versionNumber) {
+    try {
+      const rows = await this.#db.selectMany({
+        tableName: TABLE_AGENT_VERSIONS,
+        whereClause: {
+          sql: "WHERE agentId = ? AND versionNumber = ?",
+          args: [agentId, versionNumber]
+        },
+        limit: 1
+      });
+      if (!rows || rows.length === 0) {
+        return null;
+      }
+      return this.parseVersionRow(rows[0]);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "GET_VERSION_BY_NUMBER", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId, versionNumber }
+        },
+        error
+      );
+    }
+  }
+  async getLatestVersion(agentId) {
+    try {
+      const rows = await this.#db.selectMany({
+        tableName: TABLE_AGENT_VERSIONS,
+        whereClause: {
+          sql: "WHERE agentId = ?",
+          args: [agentId]
+        },
+        orderBy: "versionNumber DESC",
+        limit: 1
+      });
+      if (!rows || rows.length === 0) {
+        return null;
+      }
+      return this.parseVersionRow(rows[0]);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "GET_LATEST_VERSION", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId }
+        },
+        error
+      );
+    }
+  }
+  async listVersions(input) {
+    const { agentId, page = 0, perPage: perPageInput, orderBy } = input;
+    if (page < 0) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "LIST_VERSIONS", "INVALID_PAGE"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { page }
+        },
+        new Error("page must be >= 0")
+      );
+    }
+    const perPage = normalizePerPage(perPageInput, 20);
+    const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
+    try {
+      const { field, direction } = this.parseVersionOrderBy(orderBy);
+      const total = await this.#db.selectTotalCount({
+        tableName: TABLE_AGENT_VERSIONS,
+        whereClause: {
+          sql: "WHERE agentId = ?",
+          args: [agentId]
+        }
+      });
+      if (total === 0) {
+        return {
+          versions: [],
+          total: 0,
+          page,
+          perPage: perPageForResponse,
+          hasMore: false
+        };
+      }
+      const limitValue = perPageInput === false ? total : perPage;
+      const rows = await this.#db.selectMany({
+        tableName: TABLE_AGENT_VERSIONS,
+        whereClause: {
+          sql: "WHERE agentId = ?",
+          args: [agentId]
+        },
+        orderBy: `"${field}" ${direction}`,
+        limit: limitValue,
+        offset
+      });
+      const versions = rows.map((row) => this.parseVersionRow(row));
+      return {
+        versions,
+        total,
+        page,
+        perPage: perPageForResponse,
+        hasMore: perPageInput === false ? false : offset + perPage < total
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "LIST_VERSIONS", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId }
+        },
+        error
+      );
+    }
+  }
+  async deleteVersion(id) {
+    try {
+      await this.#db.delete({
+        tableName: TABLE_AGENT_VERSIONS,
+        keys: { id }
+      });
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "DELETE_VERSION", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { versionId: id }
+        },
+        error
+      );
+    }
+  }
+  async deleteVersionsByAgentId(agentId) {
+    try {
+      const versions = await this.#db.selectMany({
+        tableName: TABLE_AGENT_VERSIONS,
+        whereClause: {
+          sql: "WHERE agentId = ?",
+          args: [agentId]
+        }
+      });
+      for (const version of versions) {
+        await this.#db.delete({
+          tableName: TABLE_AGENT_VERSIONS,
+          keys: { id: version.id }
+        });
+      }
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "DELETE_VERSIONS_BY_AGENT_ID", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId }
+        },
+        error
+      );
+    }
+  }
+  async countVersions(agentId) {
+    try {
+      const count = await this.#db.selectTotalCount({
+        tableName: TABLE_AGENT_VERSIONS,
+        whereClause: {
+          sql: "WHERE agentId = ?",
+          args: [agentId]
+        }
+      });
+      return count;
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "COUNT_VERSIONS", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { agentId }
+        },
+        error
+      );
+    }
+  }
+  // ==========================================================================
+  // Private Helper Methods
+  // ==========================================================================
+  parseVersionRow(row) {
+    return {
+      id: row.id,
+      agentId: row.agentId,
+      versionNumber: row.versionNumber,
+      name: row.name,
+      description: row.description,
+      instructions: row.instructions,
+      model: this.parseJson(row.model, "model"),
+      tools: this.parseJson(row.tools, "tools"),
+      defaultOptions: this.parseJson(row.defaultOptions, "defaultOptions"),
+      workflows: this.parseJson(row.workflows, "workflows"),
+      agents: this.parseJson(row.agents, "agents"),
+      integrationTools: this.parseJson(row.integrationTools, "integrationTools"),
+      inputProcessors: this.parseJson(row.inputProcessors, "inputProcessors"),
+      outputProcessors: this.parseJson(row.outputProcessors, "outputProcessors"),
+      memory: this.parseJson(row.memory, "memory"),
+      scorers: this.parseJson(row.scorers, "scorers"),
+      changedFields: this.parseJson(row.changedFields, "changedFields"),
+      changeMessage: row.changeMessage,
+      createdAt: new Date(row.createdAt)
+    };
+  }
 };
 var MemoryLibSQL = class extends MemoryStorage {
   #client;
@@ -3556,7 +3840,9 @@ var ObservabilityLibSQL = class extends ObservabilityStorage {
           perPage,
           hasMore: (page + 1) * perPage < count
         },
-        spans: spans.map((span) => transformFromSqlRow({ tableName: TABLE_SPANS, sqlRow: span }))
+        spans: toTraceSpans(
+          spans.map((span) => transformFromSqlRow({ tableName: TABLE_SPANS, sqlRow: span }))
+        )
       };
     } catch (error) {
       throw new MastraError(