@axiom-lattice/pg-stores 1.0.89 → 1.0.90

package/dist/index.mjs

@@ -3302,6 +3302,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 +3340,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 +3470,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();
@@ -7526,6 +7573,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 +8189,7 @@ export {
   addSkillTenantId,
   addStepThreadId,
   addThreadTenantId,
+  addWorkflowRunsTenantStatusUpdatedIndex,
   alterChannelInstallationsTable,
   changeAssistantPrimaryKey,
   changeSkillPrimaryKey,