@mastra/libsql 1.0.0 → 1.1.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, 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) {
@@ -2091,6 +2093,9 @@ var AgentsLibSQL = class extends AgentsStorage {
       outputProcessors: this.parseJson(row.outputProcessors, "outputProcessors"),
       memory: this.parseJson(row.memory, "memory"),
       scorers: this.parseJson(row.scorers, "scorers"),
+      integrationTools: this.parseJson(row.integrationTools, "integrationTools"),
+      ownerId: row.ownerId,
+      activeVersionId: row.activeVersionId,
       metadata: this.parseJson(row.metadata, "metadata"),
       createdAt: new Date(row.createdAt),
       updatedAt: new Date(row.updatedAt)
@@ -2134,6 +2139,9 @@ var AgentsLibSQL = class extends AgentsStorage {
           outputProcessors: agent.outputProcessors ?? null,
           memory: agent.memory ?? null,
           scorers: agent.scorers ?? null,
+          integrationTools: agent.integrationTools ?? null,
+          ownerId: agent.ownerId ?? null,
+          activeVersionId: agent.activeVersionId ?? null,
           metadata: agent.metadata ?? null,
           createdAt: now,
           updatedAt: now
@@ -2183,6 +2191,9 @@ var AgentsLibSQL = class extends AgentsStorage {
       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.integrationTools !== void 0) data.integrationTools = updates.integrationTools;
+      if (updates.ownerId !== void 0) data.ownerId = updates.ownerId;
+      if (updates.activeVersionId !== void 0) data.activeVersionId = updates.activeVersionId;
       if (updates.metadata !== void 0) {
         data.metadata = { ...existingAgent.metadata, ...updates.metadata };
       }
@@ -2221,6 +2232,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 +2302,262 @@ 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,
+          snapshot: input.snapshot,
+          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,
+      snapshot: this.parseJson(row.snapshot, "snapshot"),
+      changedFields: this.parseJson(row.changedFields, "changedFields"),
+      changeMessage: row.changeMessage,
+      createdAt: new Date(row.createdAt)
+    };
+  }
 };
 var MemoryLibSQL = class extends MemoryStorage {
   #client;
@@ -3556,7 +3824,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(