@axiom-lattice/pg-stores 1.0.89 → 1.0.91

package/dist/index.mjs

@@ -1254,6 +1254,13 @@ var PostgreSQLConnectionStore = class {
   constructor(db) {
     this.db = db;
   }
+  async listByTenant(tenantId) {
+    const result = await this.db.query(
+      `SELECT * FROM ${TABLE} WHERE tenant_id = $1 ORDER BY created_at`,
+      [tenantId]
+    );
+    return result.rows.map((row) => this.mapRow(row));
+  }
   async listByType(tenantId, type) {
     const result = await this.db.query(
       `SELECT * FROM ${TABLE} WHERE tenant_id = $1 AND type = $2 ORDER BY created_at`,
@@ -3302,6 +3309,21 @@ var addStepThreadId = {
     `);
   }
 };
+var addWorkflowRunsTenantStatusUpdatedIndex = {
+  name: "add_workflow_runs_tenant_status_updated_index",
+  version: 161,
+  up: async (client) => {
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_workflow_runs_tenant_status
+        ON lattice_workflow_runs (tenant_id, status, updated_at DESC);
+    `);
+  },
+  down: async (client) => {
+    await client.query(`
+      DROP INDEX IF EXISTS idx_workflow_runs_tenant_status;
+    `);
+  }
+};
 
 // src/stores/PostgreSQLWorkflowTrackingStore.ts
 var PostgreSQLWorkflowTrackingStore = class {
@@ -3325,6 +3347,7 @@ var PostgreSQLWorkflowTrackingStore = class {
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createWorkflowTrackingTables);
     this.migrationManager.register(addStepThreadId);
+    this.migrationManager.register(addWorkflowRunsTenantStatusUpdatedIndex);
     if (options.autoMigrate !== false) {
       this.initialize().catch((error) => {
         console.error("Failed to initialize PostgreSQLWorkflowTrackingStore:", error);
@@ -3454,6 +3477,37 @@ var PostgreSQLWorkflowTrackingStore = class {
     );
     return result.rows.map(mapRowToWorkflowRun);
   }
+  async queryWorkflowRuns(tenantId, options = {}) {
+    await this.ensureInitialized();
+    const conditions = ["tenant_id = $1"];
+    const whereParams = [tenantId];
+    if (options.status) {
+      whereParams.push(options.status);
+      conditions.push(`status = $${whereParams.length}`);
+    }
+    if (options.assistantId) {
+      whereParams.push(options.assistantId);
+      conditions.push(`assistant_id = $${whereParams.length}`);
+    }
+    const where = conditions.join(" AND ");
+    const countResult = await this.pool.query(
+      `SELECT COUNT(*) AS count FROM lattice_workflow_runs WHERE ${where}`,
+      whereParams
+    );
+    const total = parseInt(countResult.rows[0].count, 10);
+    const selectParams = [...whereParams];
+    let sql = `SELECT * FROM lattice_workflow_runs WHERE ${where} ORDER BY updated_at DESC, id DESC`;
+    if (options.limit !== void 0) {
+      selectParams.push(options.limit);
+      sql += ` LIMIT $${selectParams.length}`;
+    }
+    if (options.offset !== void 0) {
+      selectParams.push(options.offset);
+      sql += ` OFFSET $${selectParams.length}`;
+    }
+    const result = await this.pool.query(sql, selectParams);
+    return { records: result.rows.map(mapRowToWorkflowRun), total };
+  }
   async createRunStep(request) {
     await this.ensureInitialized();
     const now = /* @__PURE__ */ new Date();
@@ -3613,7 +3667,7 @@ function mapRowToRunStep(row) {
     edgeTo: row.edge_to,
     edgePurpose: row.edge_purpose,
     input: safeParse(row.input, void 0),
-    output: safeParse(row.output, void 0),
+    output: typeof row.output === "string" ? safeParse(row.output, row.output) : row.output ?? void 0,
     status: row.status,
     errorMessage: row.error_message,
     startedAt: row.started_at,
@@ -4508,6 +4562,17 @@ var PostgreSQLEvalStore = class {
     );
     return rows.length ? this.mapRowToRunResult(rows[0]) : null;
   }
+  /** Delete a run result by ID */
+  async deleteRunResult(tenantId, id) {
+    await this.ensureInitialized();
+    const result = await this.pool.query(
+      `DELETE FROM lattice_eval_run_results
+       WHERE id = $1
+         AND run_id IN (SELECT id FROM lattice_eval_runs WHERE tenant_id = $2)`,
+      [id, tenantId]
+    );
+    return (result.rowCount ?? 0) > 0;
+  }
   /** Get a single run result by ID with tenant isolation */
   async getRunResultById(tenantId, id) {
     await this.ensureInitialized();
@@ -7526,6 +7591,7 @@ async function createPgStoreConfig(connectionString) {
   mm.register(addFileContentType);
   mm.register(createCollectionsTable);
   mm.register(createConnectionConfigsTable);
+  mm.register(addWorkflowRunsTenantStatusUpdatedIndex);
   await mm.migrate();
   const checkpoint = PostgresSaver.fromConnString(connectionString);
   checkpoint.setup().catch((err) => {
@@ -8141,6 +8207,7 @@ export {
   addSkillTenantId,
   addStepThreadId,
   addThreadTenantId,
+  addWorkflowRunsTenantStatusUpdatedIndex,
   alterChannelInstallationsTable,
   changeAssistantPrimaryKey,
   changeSkillPrimaryKey,