@mastra/lance 1.0.4 → 1.0.5

package/dist/index.js

@@ -1,9 +1,9 @@
 import { connect, Index } from '@lancedb/lancedb';
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MemoryStorage, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, createStorageErrorId, normalizePerPage, calculatePagination, ScoresStorage, SCORERS_SCHEMA, TABLE_SCORERS, WorkflowsStorage, ensureDate, TABLE_WORKFLOW_SNAPSHOT, MastraCompositeStore, createVectorErrorId, getDefaultValue } from '@mastra/core/storage';
-import { MessageList } from '@mastra/core/agent';
+import { BackgroundTasksStorage, TABLE_SCHEMAS, TABLE_BACKGROUND_TASKS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, createStorageErrorId, normalizePerPage, calculatePagination, ScoresStorage, SCORERS_SCHEMA, TABLE_SCORERS, WorkflowsStorage, ensureDate, TABLE_WORKFLOW_SNAPSHOT, MastraCompositeStore, createVectorErrorId, getDefaultValue } from '@mastra/core/storage';
 import { MastraBase } from '@mastra/core/base';
 import { Utf8, Float64, Binary, Float32, Int32, Field, Schema } from 'apache-arrow';
+import { MessageList } from '@mastra/core/agent';
 import { saveScorePayloadSchema } from '@mastra/core/evals';
 import { MastraVector } from '@mastra/core/vector';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
@@ -587,7 +587,188 @@ var LanceDB = class extends MastraBase {
   }
 };
 
-// src/storage/domains/memory/index.ts
+// src/storage/domains/background-tasks/index.ts
+function serializeJson(v) {
+  if (typeof v === "object" && v != null) return JSON.stringify(v);
+  return v ?? void 0;
+}
+function toRecord(task) {
+  return {
+    id: task.id,
+    tool_call_id: task.toolCallId,
+    tool_name: task.toolName,
+    agent_id: task.agentId,
+    thread_id: task.threadId ?? "",
+    resource_id: task.resourceId ?? "",
+    run_id: task.runId,
+    status: task.status,
+    args: serializeJson(task.args),
+    result: serializeJson(task.result),
+    error: serializeJson(task.error),
+    retry_count: task.retryCount,
+    max_retries: task.maxRetries,
+    timeout_ms: task.timeoutMs,
+    createdAt: task.createdAt,
+    startedAt: task.startedAt ?? /* @__PURE__ */ new Date(0),
+    completedAt: task.completedAt ?? /* @__PURE__ */ new Date(0)
+  };
+}
+function fromRecord(row) {
+  const parseJson = (val) => {
+    if (val == null || val === "") return void 0;
+    if (typeof val === "string") {
+      try {
+        return JSON.parse(val);
+      } catch {
+        return val;
+      }
+    }
+    return val;
+  };
+  const startedAt = row.startedAt instanceof Date ? row.startedAt : row.startedAt ? new Date(row.startedAt) : void 0;
+  const completedAt = row.completedAt instanceof Date ? row.completedAt : row.completedAt ? new Date(row.completedAt) : void 0;
+  return {
+    id: String(row.id),
+    status: String(row.status),
+    toolName: String(row.tool_name),
+    toolCallId: String(row.tool_call_id),
+    args: parseJson(row.args) ?? {},
+    agentId: String(row.agent_id),
+    threadId: row.thread_id && row.thread_id !== "" ? String(row.thread_id) : void 0,
+    resourceId: row.resource_id && row.resource_id !== "" ? String(row.resource_id) : void 0,
+    runId: row.run_id ?? "",
+    result: parseJson(row.result),
+    error: parseJson(row.error),
+    retryCount: Number(row.retry_count ?? 0),
+    maxRetries: Number(row.max_retries ?? 0),
+    timeoutMs: Number(row.timeout_ms ?? 3e5),
+    createdAt: row.createdAt instanceof Date ? row.createdAt : new Date(row.createdAt),
+    startedAt: startedAt && startedAt.getTime() > 0 ? startedAt : void 0,
+    completedAt: completedAt && completedAt.getTime() > 0 ? completedAt : void 0
+  };
+}
+function escapeStr(val) {
+  return val.replace(/'/g, "''");
+}
+var StoreBackgroundTasksLance = class extends BackgroundTasksStorage {
+  client;
+  #db;
+  constructor(config) {
+    super();
+    const client = resolveLanceConfig(config);
+    this.client = client;
+    this.#db = new LanceDB({ client });
+  }
+  async init() {
+    await this.#db.createTable({
+      tableName: TABLE_BACKGROUND_TASKS,
+      schema: TABLE_SCHEMAS[TABLE_BACKGROUND_TASKS]
+    });
+  }
+  async dangerouslyClearAll() {
+    await this.#db.clearTable({ tableName: TABLE_BACKGROUND_TASKS });
+  }
+  async createTask(task) {
+    const table = await this.client.openTable(TABLE_BACKGROUND_TASKS);
+    await table.add([toRecord(task)], { mode: "append" });
+  }
+  async updateTask(taskId, update) {
+    const existing = await this.getTask(taskId);
+    if (!existing) return;
+    const merged = { ...existing };
+    if ("status" in update) merged.status = update.status;
+    if ("result" in update) merged.result = update.result;
+    if ("error" in update) merged.error = update.error;
+    if ("retryCount" in update) merged.retryCount = update.retryCount;
+    if ("startedAt" in update) merged.startedAt = update.startedAt;
+    if ("completedAt" in update) merged.completedAt = update.completedAt;
+    const table = await this.client.openTable(TABLE_BACKGROUND_TASKS);
+    await table.delete(`id = '${escapeStr(taskId)}'`);
+    await table.add([toRecord(merged)], { mode: "append" });
+  }
+  async getTask(taskId) {
+    try {
+      const table = await this.client.openTable(TABLE_BACKGROUND_TASKS);
+      const records = await table.query().where(`id = '${escapeStr(taskId)}'`).limit(1).toArray();
+      if (records.length === 0) return null;
+      return fromRecord(records[0]);
+    } catch {
+      return null;
+    }
+  }
+  async listTasks(filter) {
+    try {
+      const table = await this.client.openTable(TABLE_BACKGROUND_TASKS);
+      const conditions = [];
+      if (filter.status) {
+        const statuses = Array.isArray(filter.status) ? filter.status : [filter.status];
+        const inList = statuses.map((s) => `'${escapeStr(s)}'`).join(", ");
+        conditions.push(`status IN (${inList})`);
+      }
+      if (filter.agentId) conditions.push(`agent_id = '${escapeStr(filter.agentId)}'`);
+      if (filter.threadId) conditions.push(`thread_id = '${escapeStr(filter.threadId)}'`);
+      if (filter.resourceId) conditions.push(`resource_id = '${escapeStr(filter.resourceId)}'`);
+      if (filter.runId) conditions.push(`run_id = '${escapeStr(filter.runId)}'`);
+      if (filter.toolName) conditions.push(`tool_name = '${escapeStr(filter.toolName)}'`);
+      let query = table.query();
+      if (conditions.length > 0) {
+        query = query.where(conditions.join(" AND "));
+      }
+      const records = await query.toArray();
+      let tasks = records.map(fromRecord);
+      const dateCol = filter.dateFilterBy ?? "createdAt";
+      if (filter.fromDate) {
+        tasks = tasks.filter((t) => {
+          const val = t[dateCol];
+          return val != null && val >= filter.fromDate;
+        });
+      }
+      if (filter.toDate) {
+        tasks = tasks.filter((t) => {
+          const val = t[dateCol];
+          return val != null && val < filter.toDate;
+        });
+      }
+      const orderBy = filter.orderBy ?? "createdAt";
+      const direction = filter.orderDirection ?? "asc";
+      tasks.sort((a, b) => {
+        const aVal = a[orderBy]?.getTime() ?? 0;
+        const bVal = b[orderBy]?.getTime() ?? 0;
+        return direction === "asc" ? aVal - bVal : bVal - aVal;
+      });
+      const total = tasks.length;
+      if (filter.page != null && filter.perPage != null) {
+        const start = filter.page * filter.perPage;
+        tasks = tasks.slice(start, start + filter.perPage);
+      } else if (filter.perPage != null) {
+        tasks = tasks.slice(0, filter.perPage);
+      }
+      return { tasks, total };
+    } catch {
+      return { tasks: [], total: 0 };
+    }
+  }
+  async deleteTask(taskId) {
+    const table = await this.client.openTable(TABLE_BACKGROUND_TASKS);
+    await table.delete(`id = '${escapeStr(taskId)}'`);
+  }
+  async deleteTasks(filter) {
+    const { tasks } = await this.listTasks(filter);
+    if (tasks.length === 0) return;
+    const table = await this.client.openTable(TABLE_BACKGROUND_TASKS);
+    for (const task of tasks) {
+      await table.delete(`id = '${escapeStr(task.id)}'`);
+    }
+  }
+  async getRunningCount() {
+    const { total } = await this.listTasks({ status: "running" });
+    return total;
+  }
+  async getRunningCountByAgent(agentId) {
+    const { total } = await this.listTasks({ status: "running", agentId });
+    return total;
+  }
+};
 var StoreMemoryLance = class extends MemoryStorage {
   client;
   #db;
@@ -2008,7 +2189,8 @@ var LanceStorage = class _LanceStorage extends MastraCompositeStore {
       instance.stores = {
         workflows: new StoreWorkflowsLance({ client: instance.lanceClient }),
         scores: new StoreScoresLance({ client: instance.lanceClient }),
-        memory: new StoreMemoryLance({ client: instance.lanceClient })
+        memory: new StoreMemoryLance({ client: instance.lanceClient }),
+        backgroundTasks: new StoreBackgroundTasksLance({ client: instance.lanceClient })
       };
       return instance;
     } catch (e) {
@@ -2050,7 +2232,8 @@ var LanceStorage = class _LanceStorage extends MastraCompositeStore {
     instance.stores = {
       workflows: new StoreWorkflowsLance({ client }),
       scores: new StoreScoresLance({ client }),
-      memory: new StoreMemoryLance({ client })
+      memory: new StoreMemoryLance({ client }),
+      backgroundTasks: new StoreBackgroundTasksLance({ client })
     };
     return instance;
   }
@@ -3395,6 +3578,6 @@ var LanceVectorStore = class _LanceVectorStore extends MastraVector {
   }
 };
 
-export { LanceStorage, LanceVectorStore, StoreMemoryLance, StoreScoresLance, StoreWorkflowsLance };
+export { LanceStorage, LanceVectorStore, StoreBackgroundTasksLance, StoreMemoryLance, StoreScoresLance, StoreWorkflowsLance };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map