@axiom-lattice/local-stores 1.0.1 → 1.0.3

package/dist/index.mjs

@@ -278,6 +278,7 @@ CREATE TABLE IF NOT EXISTS lt_assistants (
   name TEXT NOT NULL,
   description TEXT,
   graph_definition TEXT NOT NULL,
+  owner_user_id TEXT,
   created_at TEXT NOT NULL,
   updated_at TEXT NOT NULL,
   PRIMARY KEY (tenant_id, id)
@@ -303,20 +304,22 @@ var LocalAssistantStore = class {
   async createAssistant(tenantId, id, data) {
     const now = nowISO();
     this.db.prepare(
-      `INSERT INTO lt_assistants (id, tenant_id, name, description, graph_definition, created_at, updated_at)
-       VALUES (?, ?, ?, ?, ?, ?, ?)
+      `INSERT INTO lt_assistants (id, tenant_id, name, description, graph_definition, owner_user_id, created_at, updated_at)
+       VALUES (?, ?, ?, ?, ?, ?, ?, ?)
        ON CONFLICT(tenant_id, id) DO UPDATE SET
          name = excluded.name,
          description = excluded.description,
          graph_definition = excluded.graph_definition,
+         owner_user_id = excluded.owner_user_id,
          updated_at = excluded.updated_at`
-    ).run(id, tenantId, data.name, data.description || null, JSON.stringify(data.graphDefinition), now, now);
+    ).run(id, tenantId, data.name, data.description || null, JSON.stringify(data.graphDefinition), data.ownerUserId || null, now, now);
     return {
       id,
       tenantId,
       name: data.name,
       description: data.description,
       graphDefinition: data.graphDefinition,
+      ownerUserId: data.ownerUserId,
       createdAt: parseISO(now),
       updatedAt: parseISO(now)
     };
@@ -338,6 +341,10 @@ var LocalAssistantStore = class {
       setClauses.push("graph_definition = ?");
       values.push(JSON.stringify(updates.graphDefinition));
     }
+    if (updates.ownerUserId !== void 0) {
+      setClauses.push("owner_user_id = ?");
+      values.push(updates.ownerUserId || null);
+    }
     if (setClauses.length === 0) return existing;
     const now = nowISO();
     setClauses.push("updated_at = ?");
@@ -360,6 +367,12 @@ var LocalAssistantStore = class {
     ).get(tenantId, id);
     return row !== void 0;
   }
+  async getByOwner(tenantId, userId) {
+    const row = this.db.prepare(
+      `SELECT * FROM lt_assistants WHERE tenant_id = ? AND owner_user_id = ? LIMIT 1`
+    ).get(tenantId, userId);
+    return row ? mapRowToAssistant(row) : null;
+  }
 };
 function mapRowToAssistant(row) {
   return {
@@ -368,6 +381,7 @@ function mapRowToAssistant(row) {
     name: row.name,
     description: row.description || void 0,
     graphDefinition: JSON.parse(row.graph_definition),
+    ownerUserId: row.owner_user_id || void 0,
     createdAt: parseISO(row.created_at),
     updatedAt: parseISO(row.updated_at)
   };
@@ -1356,6 +1370,7 @@ CREATE TABLE IF NOT EXISTS lt_workflow_steps (
   tenant_id TEXT NOT NULL,
   step_type TEXT NOT NULL,
   step_name TEXT NOT NULL,
+  thread_id TEXT,
   edge_from TEXT,
   edge_to TEXT,
   edge_purpose TEXT,
@@ -1371,6 +1386,8 @@ CREATE TABLE IF NOT EXISTS lt_workflow_steps (
   PRIMARY KEY (run_id, id)
 );
 CREATE INDEX IF NOT EXISTS idx_lt_ws_type ON lt_workflow_steps(run_id, step_type);
+-- Backfill for existing databases that don't have the column
+ALTER TABLE lt_workflow_steps ADD COLUMN IF NOT EXISTS thread_id TEXT;
 `;
 var LocalWorkflowTrackingStore = class {
   constructor(db) {
@@ -1475,14 +1492,15 @@ var LocalWorkflowTrackingStore = class {
     const now = nowISO();
     const id = `step_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
     this.db.prepare(
-      `INSERT INTO lt_workflow_steps (id, run_id, tenant_id, step_type, step_name, edge_from, edge_to, edge_purpose, input, status, started_at, created_at, updated_at)
-       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, 'running', ?, ?, ?)`
+      `INSERT INTO lt_workflow_steps (id, run_id, tenant_id, step_type, step_name, thread_id, edge_from, edge_to, edge_purpose, input, status, started_at, created_at, updated_at)
+       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, 'running', ?, ?, ?)`
     ).run(
       id,
       request.runId,
       request.tenantId,
       request.stepType,
       request.stepName,
+      request.threadId || null,
       request.edgeFrom || null,
       request.edgeTo || null,
       request.edgePurpose || null,
@@ -1497,6 +1515,7 @@ var LocalWorkflowTrackingStore = class {
       tenantId: request.tenantId,
       stepType: request.stepType,
       stepName: request.stepName,
+      threadId: request.threadId,
       edgeFrom: request.edgeFrom,
       edgeTo: request.edgeTo,
       edgePurpose: request.edgePurpose,
@@ -1507,6 +1526,22 @@ var LocalWorkflowTrackingStore = class {
       updatedAt: parseISO(now)
     };
   }
+  async upsertRunStep(request) {
+    const row = this.db.prepare(
+      `SELECT * FROM lt_workflow_steps WHERE run_id = ? AND step_type = ? AND step_name = ?`
+    ).get(request.runId, request.stepType, request.stepName);
+    if (row) {
+      const existing = mapRowToStep(row);
+      if (!existing.threadId && request.threadId) {
+        this.db.prepare(
+          `UPDATE lt_workflow_steps SET thread_id = ?, updated_at = ? WHERE run_id = ? AND id = ?`
+        ).run(request.threadId, nowISO(), request.runId, existing.id);
+        existing.threadId = request.threadId;
+      }
+      return existing;
+    }
+    return this.createRunStep(request);
+  }
   async updateRunStep(runId, stepId, updates) {
     const setClauses = [];
     const values = [];
@@ -1592,6 +1627,7 @@ function mapRowToStep(row) {
     tenantId: row.tenant_id,
     stepType: row.step_type,
     stepName: row.step_name,
+    threadId: row.thread_id,
     edgeFrom: row.edge_from || void 0,
     edgeTo: row.edge_to || void 0,
     edgePurpose: row.edge_purpose || void 0,
@@ -2377,6 +2413,19 @@ var LocalChannelInstallationStore = class {
     }
     return rows.map(mapRow);
   }
+  async getAllInstallations(channel) {
+    let rows;
+    if (channel) {
+      rows = this.db.prepare(
+        `SELECT * FROM lt_channel_installations WHERE channel = ? ORDER BY created_at DESC`
+      ).all(channel);
+    } else {
+      rows = this.db.prepare(
+        `SELECT * FROM lt_channel_installations ORDER BY created_at DESC`
+      ).all();
+    }
+    return rows.map(mapRow);
+  }
   async createInstallation(tenantId, installationId, data) {
     const now = nowISO();
     this.db.prepare(
@@ -2626,6 +2675,8 @@ CREATE TABLE IF NOT EXISTS lt_thread_message_queue (
   thread_id TEXT NOT NULL,
   tenant_id TEXT NOT NULL,
   assistant_id TEXT NOT NULL,
+  workspace_id TEXT,
+  project_id TEXT,
   message_content TEXT NOT NULL,
   message_type TEXT NOT NULL DEFAULT 'human',
   sequence_order INTEGER NOT NULL DEFAULT 0,
@@ -2651,13 +2702,15 @@ var LocalThreadMessageQueueStore = class {
     ).get(params.threadId);
     const nextSeq = seqRow.next_seq;
     this.db.prepare(
-      `INSERT INTO lt_thread_message_queue (id, thread_id, tenant_id, assistant_id, message_content, message_type, sequence_order, priority, command, custom_run_config, created_at)
-       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
+      `INSERT INTO lt_thread_message_queue (id, thread_id, tenant_id, assistant_id, workspace_id, project_id, message_content, message_type, sequence_order, priority, command, custom_run_config, created_at)
+       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
     ).run(
       id,
       params.threadId,
       params.tenantId,
       params.assistantId,
+      params.workspaceId || null,
+      params.projectId || null,
       JSON.stringify(params.content),
       params.type || "human",
       nextSeq,
@@ -2675,13 +2728,15 @@ var LocalThreadMessageQueueStore = class {
       `SELECT COALESCE(MAX(sequence_order), 0) + 1 as next_seq FROM lt_thread_message_queue WHERE thread_id = ?`
     ).get(params.threadId);
     this.db.prepare(
-      `INSERT INTO lt_thread_message_queue (id, thread_id, tenant_id, assistant_id, message_content, message_type, sequence_order, priority, command, custom_run_config, created_at)
-       VALUES (?, ?, ?, ?, ?, ?, ?, 100, ?, ?, ?)`
+      `INSERT INTO lt_thread_message_queue (id, thread_id, tenant_id, assistant_id, workspace_id, project_id, message_content, message_type, sequence_order, priority, command, custom_run_config, created_at)
+       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, 100, ?, ?, ?)`
     ).run(
       id,
       params.threadId,
       params.tenantId,
       params.assistantId,
+      params.workspaceId || null,
+      params.projectId || null,
       JSON.stringify(params.content),
       params.type || "human",
       seqRow.next_seq,
@@ -2711,12 +2766,14 @@ var LocalThreadMessageQueueStore = class {
   }
   async getThreadsWithPendingMessages() {
     const rows = this.db.prepare(
-      `SELECT DISTINCT tenant_id, assistant_id, thread_id FROM lt_thread_message_queue WHERE status IN ('pending', 'processing') ORDER BY thread_id`
+      `SELECT DISTINCT ON (thread_id) tenant_id, assistant_id, thread_id, workspace_id, project_id FROM lt_thread_message_queue WHERE status IN ('pending', 'processing') ORDER BY thread_id`
     ).all();
     return rows.map((r) => ({
       tenantId: r.tenant_id,
       assistantId: r.assistant_id,
-      threadId: r.thread_id
+      threadId: r.thread_id,
+      workspaceId: r.workspace_id || void 0,
+      projectId: r.project_id || void 0
     }));
   }
   async removeMessage(messageId) {
@@ -3249,6 +3306,192 @@ function mapRowToTask(row) {
   };
 }
 
+// src/stores/LocalTaskStore.ts
+import { randomUUID as randomUUID5 } from "crypto";
+var DDL19 = `
+CREATE TABLE IF NOT EXISTS lt_tasks (
+  id TEXT NOT NULL,
+  tenant_id TEXT NOT NULL,
+  owner_type TEXT NOT NULL DEFAULT 'user',
+  owner_id TEXT NOT NULL,
+  title TEXT NOT NULL,
+  description TEXT,
+  status TEXT NOT NULL DEFAULT 'pending',
+  priority TEXT NOT NULL DEFAULT 'medium',
+  due_date TEXT,
+  metadata TEXT,
+  parent_id TEXT,
+  source_id TEXT,
+  context TEXT,
+  created_at TEXT NOT NULL,
+  updated_at TEXT NOT NULL,
+  PRIMARY KEY (tenant_id, id)
+);
+`;
+function mapRowToTask2(row) {
+  return {
+    id: row.id,
+    tenantId: row.tenant_id,
+    ownerType: row.owner_type,
+    ownerId: row.owner_id,
+    title: row.title,
+    description: row.description ?? void 0,
+    status: row.status,
+    priority: row.priority,
+    dueDate: row.due_date ?? void 0,
+    metadata: row.metadata ? JSON.parse(row.metadata) : void 0,
+    parentId: row.parent_id ?? void 0,
+    sourceId: row.source_id ?? void 0,
+    context: row.context ? JSON.parse(row.context) : void 0,
+    createdAt: parseISO(row.created_at),
+    updatedAt: parseISO(row.updated_at)
+  };
+}
+var LocalTaskStore = class {
+  constructor(db) {
+    this.db = db;
+    ensureTable(db, DDL19);
+  }
+  async create(params) {
+    const id = randomUUID5();
+    const now = nowISO();
+    this.db.prepare(
+      `INSERT INTO lt_tasks (id, tenant_id, owner_type, owner_id, title, description, status, priority, due_date, metadata, parent_id, source_id, context, created_at, updated_at)
+       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
+    ).run(
+      id,
+      params.tenantId,
+      params.ownerType,
+      params.ownerId,
+      params.title,
+      params.description || null,
+      params.status || "pending",
+      params.priority || "medium",
+      params.dueDate || null,
+      params.metadata ? JSON.stringify(params.metadata) : null,
+      params.parentId || null,
+      params.sourceId || null,
+      params.context ? JSON.stringify(params.context) : null,
+      now,
+      now
+    );
+    return await this.getById(params.tenantId, id);
+  }
+  async getById(tenantId, id) {
+    const row = this.db.prepare(
+      `SELECT * FROM lt_tasks WHERE tenant_id = ? AND id = ?`
+    ).get(tenantId, id);
+    return row ? mapRowToTask2(row) : null;
+  }
+  async list(filter) {
+    const conditions = ["tenant_id = ?"];
+    const params = [filter.tenantId];
+    if (filter.ownerType) {
+      conditions.push("owner_type = ?");
+      params.push(filter.ownerType);
+    }
+    if (filter.ownerId) {
+      conditions.push("owner_id = ?");
+      params.push(filter.ownerId);
+    }
+    if (filter.status) {
+      conditions.push("status = ?");
+      params.push(filter.status);
+    }
+    if (filter.priority) {
+      conditions.push("priority = ?");
+      params.push(filter.priority);
+    }
+    if (filter.parentId) {
+      conditions.push("parent_id = ?");
+      params.push(filter.parentId);
+    }
+    if (filter.sourceId) {
+      conditions.push("source_id = ?");
+      params.push(filter.sourceId);
+    }
+    if (filter.metadata) {
+      for (const [key, value] of Object.entries(filter.metadata)) {
+        const safeKey = key.replace(/[^a-zA-Z0-9_]/g, "");
+        if (safeKey) {
+          conditions.push(`json_extract(metadata, '$.${safeKey}') = ?`);
+          params.push(String(value));
+        }
+      }
+    }
+    const where = conditions.join(" AND ");
+    const limit = filter.limit || 100;
+    const offset = filter.offset || 0;
+    const rows = this.db.prepare(
+      `SELECT * FROM lt_tasks WHERE ${where} ORDER BY created_at DESC LIMIT ? OFFSET ?`
+    ).all(...params, limit, offset);
+    return rows.map(mapRowToTask2);
+  }
+  async update(tenantId, id, updates) {
+    const existing = await this.getById(tenantId, id);
+    if (!existing) return null;
+    const now = nowISO();
+    const setClauses = ["updated_at = ?"];
+    const values = [now];
+    if (updates.title !== void 0) {
+      setClauses.push("title = ?");
+      values.push(updates.title);
+    }
+    if (updates.description !== void 0) {
+      setClauses.push("description = ?");
+      values.push(updates.description);
+    }
+    if (updates.status !== void 0) {
+      setClauses.push("status = ?");
+      values.push(updates.status);
+    }
+    if (updates.priority !== void 0) {
+      setClauses.push("priority = ?");
+      values.push(updates.priority);
+    }
+    if (updates.dueDate !== void 0) {
+      setClauses.push("due_date = ?");
+      values.push(updates.dueDate);
+    }
+    if (updates.metadata !== void 0) {
+      setClauses.push("metadata = ?");
+      values.push(JSON.stringify(updates.metadata));
+    }
+    if (updates.parentId !== void 0) {
+      setClauses.push("parent_id = ?");
+      values.push(updates.parentId);
+    }
+    if (updates.sourceId !== void 0) {
+      setClauses.push("source_id = ?");
+      values.push(updates.sourceId);
+    }
+    if (updates.context !== void 0) {
+      setClauses.push("context = ?");
+      values.push(JSON.stringify(updates.context));
+    }
+    if (updates.ownerType !== void 0) {
+      setClauses.push("owner_type = ?");
+      values.push(updates.ownerType);
+    }
+    if (updates.ownerId !== void 0) {
+      setClauses.push("owner_id = ?");
+      values.push(updates.ownerId);
+    }
+    if (setClauses.length === 1) return existing;
+    values.push(tenantId, id);
+    this.db.prepare(
+      `UPDATE lt_tasks SET ${setClauses.join(", ")} WHERE tenant_id = ? AND id = ?`
+    ).run(...values);
+    return this.getById(tenantId, id);
+  }
+  async delete(tenantId, id) {
+    const result = this.db.prepare(
+      `DELETE FROM lt_tasks WHERE tenant_id = ? AND id = ?`
+    ).run(tenantId, id);
+    return result.changes > 0;
+  }
+};
+
 // src/createLocalStoreConfig.ts
 async function createLocalStoreConfig(options = {}) {
   const dbPath = options.dbPath || "~/.axiom/lattice.db";
@@ -3272,6 +3515,7 @@ async function createLocalStoreConfig(options = {}) {
     threadMessageQueue: new LocalThreadMessageQueueStore(db),
     skill: new LocalSkillStore(db),
     schedule: new LocalScheduleStorage(db),
+    task: new LocalTaskStore(db),
     checkpoint: SqliteSaver.fromConnString(dbPath)
   };
 }
@@ -3288,6 +3532,7 @@ export {
   LocalProjectStore,
   LocalScheduleStorage,
   LocalSkillStore,
+  LocalTaskStore,
   LocalTenantStore,
   LocalThreadMessageQueueStore,
   LocalThreadStore,