@mastra/libsql 1.11.1-alpha.0 → 1.12.0-alpha.0

package/dist/index.cjs

@@ -2275,7 +2275,15 @@ var AgentsLibSQL = class extends storage.AgentsStorage {
     await this.#db.alterTable({
       tableName: storage.TABLE_AGENT_VERSIONS,
       schema: storage.AGENT_VERSIONS_SCHEMA,
-      ifNotExists: ["mcpClients", "requestContextSchema", "workspace", "skills", "skillsFormat", "browser"]
+      ifNotExists: [
+        "mcpClients",
+        "requestContextSchema",
+        "workspace",
+        "skills",
+        "skillsFormat",
+        "browser",
+        "toolProviders"
+      ]
     });
     await this.#migrateToolsToJsonbFormat();
     await this.#cleanupStaleDrafts();
@@ -2336,6 +2344,7 @@ var AgentsLibSQL = class extends storage.AgentsStorage {
           workflows: row.workflows ?? null,
           agents: row.agents ?? null,
           integrationTools: row.integrationTools ?? null,
+          toolProviders: row.toolProviders ?? null,
           inputProcessors: row.inputProcessors ?? null,
           outputProcessors: row.outputProcessors ?? null,
           memory: row.memory ?? null,
@@ -2788,6 +2797,7 @@ var AgentsLibSQL = class extends storage.AgentsStorage {
           workflows: input.workflows ?? null,
           agents: input.agents ?? null,
           integrationTools: input.integrationTools ?? null,
+          toolProviders: input.toolProviders ?? null,
           inputProcessors: input.inputProcessors ?? null,
           outputProcessors: input.outputProcessors ?? null,
           memory: input.memory ?? null,
@@ -3062,6 +3072,7 @@ var AgentsLibSQL = class extends storage.AgentsStorage {
       workflows: this.parseJson(row.workflows, "workflows"),
       agents: this.parseJson(row.agents, "agents"),
       integrationTools: this.parseJson(row.integrationTools, "integrationTools"),
+      toolProviders: this.parseJson(row.toolProviders, "toolProviders"),
       inputProcessors: this.parseJson(row.inputProcessors, "inputProcessors"),
       outputProcessors: this.parseJson(row.outputProcessors, "outputProcessors"),
       memory: this.parseJson(row.memory, "memory"),
@@ -10980,6 +10991,197 @@ var SkillsLibSQL = class extends storage.SkillsStorage {
     };
   }
 };
+function normaliseScope(raw) {
+  const value = raw == null ? "per-author" : String(raw);
+  if (value === "shared") return "shared";
+  if (value === "caller-supplied") return "caller-supplied";
+  return "per-author";
+}
+function rowToToolProviderConnection(row) {
+  return {
+    authorId: String(row.authorId),
+    providerId: String(row.providerId),
+    toolkit: String(row.toolkit),
+    connectionId: String(row.connectionId),
+    label: row.label == null ? null : String(row.label),
+    scope: normaliseScope(row.scope),
+    createdAt: new Date(String(row.createdAt)),
+    updatedAt: new Date(String(row.updatedAt))
+  };
+}
+var ToolProviderConnectionsLibSQL = class extends storage.ToolProviderConnectionsStorage {
+  #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: storage.TABLE_TOOL_PROVIDER_CONNECTIONS,
+      schema: storage.TOOL_PROVIDER_CONNECTIONS_SCHEMA,
+      compositePrimaryKey: ["authorId", "providerId", "connectionId"]
+    });
+    await this.#client.execute(
+      `CREATE INDEX IF NOT EXISTS idx_tool_provider_connections_author ON "${storage.TABLE_TOOL_PROVIDER_CONNECTIONS}" ("authorId", "providerId", "toolkit")`
+    );
+  }
+  async dangerouslyClearAll() {
+    await this.#client.execute(`DELETE FROM "${storage.TABLE_TOOL_PROVIDER_CONNECTIONS}"`);
+  }
+  async getConnectionById({
+    authorId,
+    providerId,
+    connectionId
+  }) {
+    try {
+      const result = await this.#client.execute({
+        sql: `SELECT * FROM "${storage.TABLE_TOOL_PROVIDER_CONNECTIONS}" WHERE "authorId" = ? AND "providerId" = ? AND "connectionId" = ? LIMIT 1`,
+        args: [authorId, providerId, connectionId]
+      });
+      const row = result.rows?.[0];
+      if (!row) return null;
+      return rowToToolProviderConnection(row);
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: storage.createStorageErrorId("LIBSQL", "TOOL_PROVIDER_CONNECTION_GET", "FAILED"),
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { authorId, providerId, connectionId }
+        },
+        error$1
+      );
+    }
+  }
+  async upsertConnection(input) {
+    const { authorId, providerId, toolkit, connectionId, label } = input;
+    const now = /* @__PURE__ */ new Date();
+    const nowIso = now.toISOString();
+    const labelValue = label == null ? null : label;
+    try {
+      const tx = await this.#client.transaction("write");
+      try {
+        const existing = await tx.execute({
+          sql: `SELECT "createdAt", "scope" FROM "${storage.TABLE_TOOL_PROVIDER_CONNECTIONS}" WHERE "authorId" = ? AND "providerId" = ? AND "connectionId" = ? LIMIT 1`,
+          args: [authorId, providerId, connectionId]
+        });
+        const existingRow = existing.rows?.[0];
+        const createdAt = existingRow ? String(existingRow.createdAt) : nowIso;
+        const existingScope = existingRow && existingRow.scope != null ? normaliseScope(existingRow.scope) : void 0;
+        const scope = input.scope ?? existingScope ?? "per-author";
+        if (existingRow) {
+          await tx.execute({
+            sql: `UPDATE "${storage.TABLE_TOOL_PROVIDER_CONNECTIONS}" SET "toolkit" = ?, "label" = ?, "scope" = ?, "updatedAt" = ? WHERE "authorId" = ? AND "providerId" = ? AND "connectionId" = ?`,
+            args: [toolkit, labelValue, scope, nowIso, authorId, providerId, connectionId]
+          });
+        } else {
+          await tx.execute({
+            sql: `INSERT INTO "${storage.TABLE_TOOL_PROVIDER_CONNECTIONS}" ("authorId", "providerId", "toolkit", "connectionId", "label", "scope", "createdAt", "updatedAt") VALUES (?, ?, ?, ?, ?, ?, ?, ?)`,
+            args: [authorId, providerId, toolkit, connectionId, labelValue, scope, createdAt, nowIso]
+          });
+        }
+        await tx.commit();
+        return {
+          authorId,
+          providerId,
+          toolkit,
+          connectionId,
+          label: labelValue,
+          scope,
+          createdAt: new Date(createdAt),
+          updatedAt: now
+        };
+      } catch (error) {
+        if (!tx.closed) {
+          await tx.rollback();
+        }
+        throw error;
+      }
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: storage.createStorageErrorId("LIBSQL", "TOOL_PROVIDER_CONNECTION_UPSERT", "FAILED"),
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { authorId, providerId, connectionId }
+        },
+        error$1
+      );
+    }
+  }
+  async listConnectionsByAuthor({
+    authorId,
+    providerId,
+    toolkit,
+    scope
+  }) {
+    try {
+      const clauses = [];
+      const args = [];
+      if (authorId !== void 0) {
+        clauses.push('"authorId" = ?');
+        args.push(authorId);
+      }
+      if (providerId) {
+        clauses.push('"providerId" = ?');
+        args.push(providerId);
+      }
+      if (toolkit) {
+        clauses.push('"toolkit" = ?');
+        args.push(toolkit);
+      }
+      if (scope) {
+        clauses.push('"scope" = ?');
+        args.push(scope);
+      }
+      const whereClause = clauses.length > 0 ? ` WHERE ${clauses.join(" AND ")}` : "";
+      const result = await this.#client.execute({
+        sql: `SELECT * FROM "${storage.TABLE_TOOL_PROVIDER_CONNECTIONS}"${whereClause}`,
+        args
+      });
+      return (result.rows ?? []).map((row) => rowToToolProviderConnection(row));
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: storage.createStorageErrorId("LIBSQL", "TOOL_PROVIDER_CONNECTION_LIST", "FAILED"),
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { authorId: authorId ?? "", providerId: providerId ?? "", toolkit: toolkit ?? "" }
+        },
+        error$1
+      );
+    }
+  }
+  async deleteConnection({
+    authorId,
+    providerId,
+    connectionId
+  }) {
+    try {
+      await this.#client.execute({
+        sql: `DELETE FROM "${storage.TABLE_TOOL_PROVIDER_CONNECTIONS}" WHERE "authorId" = ? AND "providerId" = ? AND "connectionId" = ?`,
+        args: [authorId, providerId, connectionId]
+      });
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: storage.createStorageErrorId("LIBSQL", "TOOL_PROVIDER_CONNECTION_DELETE", "FAILED"),
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY,
+          details: { authorId, providerId, connectionId }
+        },
+        error$1
+      );
+    }
+  }
+};
 var WorkflowsLibSQL = class extends storage.WorkflowsStorage {
   #db;
   #client;
@@ -11916,6 +12118,7 @@ var LibSQLStore = class extends storage.MastraCompositeStore {
     const blobs = new BlobsLibSQL(domainConfig);
     const backgroundTasks = new BackgroundTasksLibSQL(domainConfig);
     const schedules = new SchedulesLibSQL(domainConfig);
+    const toolProviderConnections = new ToolProviderConnectionsLibSQL(domainConfig);
     this.stores = {
       scores,
       workflows,
@@ -11934,7 +12137,8 @@ var LibSQLStore = class extends storage.MastraCompositeStore {
       favorites,
       blobs,
       backgroundTasks,
-      schedules
+      schedules,
+      toolProviderConnections
     };
   }
   async applyLocalPragmas() {
@@ -11994,6 +12198,34 @@ var LibSQLStore = class extends storage.MastraCompositeStore {
     }
     await this.initDomainsSequentially();
   }
+  /**
+   * Closes the underlying libsql client, releasing all OS file handles.
+   *
+   * For local file databases, first runs PRAGMA wal_checkpoint(TRUNCATE) and
+   * switches back to journal_mode=DELETE so that Windows releases the -wal
+   * and -shm sidecar files promptly. Without this, the handles stay open
+   * until process exit, causing EBUSY errors when callers try to fs.rm the
+   * storage directory after Mastra.shutdown().
+   *
+   * Remote (Turso) databases skip the WAL pragmas and just close the client.
+   *
+   * Safe to call more than once; subsequent calls are no-ops.
+   */
+  async close() {
+    if (this.client.closed) {
+      return;
+    }
+    const isLocalFileDb = this.isLocalDb || this.client.protocol === "file";
+    if (isLocalFileDb) {
+      try {
+        await this.client.execute("PRAGMA wal_checkpoint(TRUNCATE);");
+        await this.client.execute("PRAGMA journal_mode=DELETE;");
+      } catch (err) {
+        this.logger.warn("LibSQLStore: Failed to checkpoint WAL before close.", err);
+      }
+    }
+    this.client.close();
+  }
 };
 
 // src/vector/prompt.ts
@@ -12115,6 +12347,7 @@ exports.SchedulesLibSQL = SchedulesLibSQL;
 exports.ScorerDefinitionsLibSQL = ScorerDefinitionsLibSQL;
 exports.ScoresLibSQL = ScoresLibSQL;
 exports.SkillsLibSQL = SkillsLibSQL;
+exports.ToolProviderConnectionsLibSQL = ToolProviderConnectionsLibSQL;
 exports.WorkflowsLibSQL = WorkflowsLibSQL;
 exports.WorkspacesLibSQL = WorkspacesLibSQL;
 //# sourceMappingURL=index.cjs.map