@mastra/libsql 1.11.1 → 1.12.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, FavoritesStorage, FAVORITES_SCHEMA, TABLE_FAVORITES, TABLE_SKILLS, 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, 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, FavoritesStorage, FAVORITES_SCHEMA, TABLE_FAVORITES, TABLE_SKILLS, 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, SKILL_VERSIONS_SCHEMA, TABLE_SKILL_VERSIONS, ToolProviderConnectionsStorage, TOOL_PROVIDER_CONNECTIONS_SCHEMA, TABLE_TOOL_PROVIDER_CONNECTIONS, 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';
@@ -2273,7 +2273,15 @@ var AgentsLibSQL = class extends AgentsStorage {
     await this.#db.alterTable({
       tableName: TABLE_AGENT_VERSIONS,
       schema: AGENT_VERSIONS_SCHEMA,
-      ifNotExists: ["mcpClients", "requestContextSchema", "workspace", "skills", "skillsFormat", "browser"]
+      ifNotExists: [
+        "mcpClients",
+        "requestContextSchema",
+        "workspace",
+        "skills",
+        "skillsFormat",
+        "browser",
+        "toolProviders"
+      ]
     });
     await this.#migrateToolsToJsonbFormat();
     await this.#cleanupStaleDrafts();
@@ -2334,6 +2342,7 @@ var AgentsLibSQL = class extends 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,
@@ -2786,6 +2795,7 @@ var AgentsLibSQL = class extends 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,
@@ -3060,6 +3070,7 @@ var AgentsLibSQL = class extends 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"),
@@ -10978,6 +10989,197 @@ var SkillsLibSQL = class extends 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 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: TABLE_TOOL_PROVIDER_CONNECTIONS,
+      schema: TOOL_PROVIDER_CONNECTIONS_SCHEMA,
+      compositePrimaryKey: ["authorId", "providerId", "connectionId"]
+    });
+    await this.#client.execute(
+      `CREATE INDEX IF NOT EXISTS idx_tool_provider_connections_author ON "${TABLE_TOOL_PROVIDER_CONNECTIONS}" ("authorId", "providerId", "toolkit")`
+    );
+  }
+  async dangerouslyClearAll() {
+    await this.#client.execute(`DELETE FROM "${TABLE_TOOL_PROVIDER_CONNECTIONS}"`);
+  }
+  async getConnectionById({
+    authorId,
+    providerId,
+    connectionId
+  }) {
+    try {
+      const result = await this.#client.execute({
+        sql: `SELECT * FROM "${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) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "TOOL_PROVIDER_CONNECTION_GET", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { authorId, providerId, connectionId }
+        },
+        error
+      );
+    }
+  }
+  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 "${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 "${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 "${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) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "TOOL_PROVIDER_CONNECTION_UPSERT", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { authorId, providerId, connectionId }
+        },
+        error
+      );
+    }
+  }
+  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 "${TABLE_TOOL_PROVIDER_CONNECTIONS}"${whereClause}`,
+        args
+      });
+      return (result.rows ?? []).map((row) => rowToToolProviderConnection(row));
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "TOOL_PROVIDER_CONNECTION_LIST", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { authorId: authorId ?? "", providerId: providerId ?? "", toolkit: toolkit ?? "" }
+        },
+        error
+      );
+    }
+  }
+  async deleteConnection({
+    authorId,
+    providerId,
+    connectionId
+  }) {
+    try {
+      await this.#client.execute({
+        sql: `DELETE FROM "${TABLE_TOOL_PROVIDER_CONNECTIONS}" WHERE "authorId" = ? AND "providerId" = ? AND "connectionId" = ?`,
+        args: [authorId, providerId, connectionId]
+      });
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "TOOL_PROVIDER_CONNECTION_DELETE", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { authorId, providerId, connectionId }
+        },
+        error
+      );
+    }
+  }
+};
 var WorkflowsLibSQL = class extends WorkflowsStorage {
   #db;
   #client;
@@ -11914,6 +12116,7 @@ var LibSQLStore = class extends 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,
@@ -11932,7 +12135,8 @@ var LibSQLStore = class extends MastraCompositeStore {
       favorites,
       blobs,
       backgroundTasks,
-      schedules
+      schedules,
+      toolProviderConnections
     };
   }
   async applyLocalPragmas() {
@@ -11992,6 +12196,34 @@ var LibSQLStore = class extends 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
@@ -12093,6 +12325,6 @@ Example Complex Query:
   ]
 }`;
 
-export { AgentsLibSQL, BackgroundTasksLibSQL, BlobsLibSQL, ChannelsLibSQL, DatasetsLibSQL, LibSQLStore as DefaultStorage, ExperimentsLibSQL, FavoritesLibSQL, LIBSQL_PROMPT, LibSQLStore, LibSQLVector, MCPClientsLibSQL, MCPServersLibSQL, MemoryLibSQL, ObservabilityLibSQL, PromptBlocksLibSQL, SchedulesLibSQL, ScorerDefinitionsLibSQL, ScoresLibSQL, SkillsLibSQL, WorkflowsLibSQL, WorkspacesLibSQL };
+export { AgentsLibSQL, BackgroundTasksLibSQL, BlobsLibSQL, ChannelsLibSQL, DatasetsLibSQL, LibSQLStore as DefaultStorage, ExperimentsLibSQL, FavoritesLibSQL, LIBSQL_PROMPT, LibSQLStore, LibSQLVector, MCPClientsLibSQL, MCPServersLibSQL, MemoryLibSQL, ObservabilityLibSQL, PromptBlocksLibSQL, SchedulesLibSQL, ScorerDefinitionsLibSQL, ScoresLibSQL, SkillsLibSQL, ToolProviderConnectionsLibSQL, WorkflowsLibSQL, WorkspacesLibSQL };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map