@axiom-lattice/pg-stores 2.0.6 → 2.0.8

package/dist/index.mjs

@@ -4015,6 +4015,26 @@ var enforceUniqueEvalProjectName = {
     `);
   }
 };
+var addTaskIdToEvalRuns = {
+  version: 164,
+  name: "add_task_id_to_eval_runs",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_eval_runs
+        ADD COLUMN IF NOT EXISTS task_id TEXT
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_eval_runs_task_id
+        ON lattice_eval_runs(task_id)
+    `);
+  },
+  down: async (client) => {
+    await client.query(`DROP INDEX IF EXISTS idx_lattice_eval_runs_task_id`);
+    await client.query(`
+      ALTER TABLE lattice_eval_runs DROP COLUMN IF EXISTS task_id
+    `);
+  }
+};
 var evalMigrations = [
   createEvalProjectsTable,
   createEvalSuitesTable,
@@ -4025,7 +4045,8 @@ var evalMigrations = [
   addEnvToEvalRuns,
   addInterruptColumnsToEval,
   addInterruptPolicyToEvalCases,
-  enforceUniqueEvalProjectName
+  enforceUniqueEvalProjectName,
+  addTaskIdToEvalRuns
 ];
 
 // src/stores/PostgreSQLEvalStore.ts
@@ -4153,6 +4174,7 @@ var PostgreSQLEvalStore = class {
       holdout: row.holdout,
       envProjectId: row.env_project_id,
       envWorkspaceId: row.env_workspace_id,
+      taskId: row.task_id,
       createdAt: new Date(row.created_at),
       startedAt: row.started_at ? new Date(row.started_at) : void 0,
       completedAt: row.completed_at ? new Date(row.completed_at) : void 0
@@ -4520,7 +4542,7 @@ var PostgreSQLEvalStore = class {
       `SELECT id, project_id, tenant_id,
               status, concurrency, total_cases,
               passed_cases, failed_cases, interrupted_cases, avg_score,
-              error, holdout, env_project_id, env_workspace_id, created_at, started_at, completed_at
+              error, holdout, env_project_id, env_workspace_id, task_id, created_at, started_at, completed_at
        FROM lattice_eval_runs
        WHERE ${conditions.join(" AND ")}
        ORDER BY created_at DESC`,
@@ -4535,7 +4557,7 @@ var PostgreSQLEvalStore = class {
       `SELECT id, project_id, tenant_id,
               status, concurrency, total_cases,
               passed_cases, failed_cases, interrupted_cases, avg_score,
-              error, holdout, env_project_id, env_workspace_id, created_at, started_at, completed_at
+              error, holdout, env_project_id, env_workspace_id, task_id, created_at, started_at, completed_at
        FROM lattice_eval_runs
        WHERE id = $1 AND tenant_id = $2`,
       [id, tenantId]
@@ -4548,12 +4570,12 @@ var PostgreSQLEvalStore = class {
     const actualId = id || uuidv4();
     const { rows } = await this.pool.query(
       `INSERT INTO lattice_eval_runs
-        (id, project_id, tenant_id, status, concurrency, total_cases, passed_cases, failed_cases, avg_score, holdout, env_project_id, env_workspace_id, started_at)
-       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13)
+        (id, project_id, tenant_id, status, concurrency, total_cases, passed_cases, failed_cases, avg_score, holdout, env_project_id, env_workspace_id, task_id, started_at)
+       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14)
        RETURNING id, project_id, tenant_id,
                  status, concurrency, total_cases,
                  passed_cases, failed_cases, interrupted_cases, avg_score,
-                 error, holdout, env_project_id, env_workspace_id, created_at, started_at, completed_at`,
+                 error, holdout, env_project_id, env_workspace_id, task_id, created_at, started_at, completed_at`,
       [
         actualId,
         projectId,
@@ -4567,6 +4589,7 @@ var PostgreSQLEvalStore = class {
         data.holdout ?? false,
         data.envProjectId || null,
         data.envWorkspaceId || null,
+        data.taskId || null,
         /* @__PURE__ */ new Date()
       ]
     );
@@ -4614,7 +4637,7 @@ var PostgreSQLEvalStore = class {
        RETURNING id, project_id, tenant_id,
                  status, concurrency, total_cases,
                  passed_cases, failed_cases, interrupted_cases, avg_score,
-                 error, holdout, env_project_id, env_workspace_id, created_at, started_at, completed_at`,
+                 error, holdout, env_project_id, env_workspace_id, task_id, created_at, started_at, completed_at`,
       vals
     );
     return rows.length ? this.mapRowToRun(rows[0]) : null;
@@ -6570,9 +6593,40 @@ var addTaskProjectFieldsMigration = {
     `);
   }
 };
+var addFilesToTasks = {
+  version: 141,
+  name: "add_files_to_tasks",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_tasks
+        ADD COLUMN IF NOT EXISTS files JSONB
+    `);
+  },
+  down: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_tasks DROP COLUMN IF EXISTS files
+    `);
+  }
+};
+var taskMigrations = [
+  createTasksTable,
+  addTaskFieldsMigration,
+  addTaskProjectFieldsMigration,
+  addFilesToTasks
+];
 
 // src/stores/PostgreSQLTaskStore.ts
 import { v4 as uuidv42 } from "uuid";
+function parseTaskFiles(raw) {
+  if (!raw) return void 0;
+  try {
+    const parsed = typeof raw === "string" ? JSON.parse(raw) : raw;
+    if (!Array.isArray(parsed)) return void 0;
+    return parsed;
+  } catch {
+    return void 0;
+  }
+}
 function mapRowToTask(row) {
   return {
     id: row.id,
@@ -6594,6 +6648,7 @@ function mapRowToTask(row) {
     failureReason: row.failure_reason ?? void 0,
     workspaceId: row.workspace_id ?? void 0,
     projectId: row.project_id ?? void 0,
+    files: parseTaskFiles(row.files),
     createdAt: new Date(row.created_at),
     updatedAt: new Date(row.updated_at)
   };
@@ -6620,6 +6675,7 @@ var PostgreSQLTaskStore = class {
     this.migrationManager.register(createTasksTable);
     this.migrationManager.register(addTaskFieldsMigration);
     this.migrationManager.register(addTaskProjectFieldsMigration);
+    this.migrationManager.register(addFilesToTasks);
     if (options.autoMigrate !== false) {
       this.initialize().catch((error) => {
         console.error("Failed to initialize PostgreSQLTaskStore:", error);
@@ -6649,8 +6705,8 @@ var PostgreSQLTaskStore = class {
     const id = uuidv42();
     const now = (/* @__PURE__ */ new Date()).toISOString();
     await this.pool.query(
-      `INSERT INTO lattice_tasks (id, tenant_id, owner_type, owner_id, title, description, status, priority, due_date, metadata, parent_id, source_id, context, require_review, dependencies, result, failure_reason, workspace_id, project_id, created_at, updated_at)
-       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21)`,
+      `INSERT INTO lattice_tasks (id, tenant_id, owner_type, owner_id, title, description, status, priority, due_date, metadata, parent_id, source_id, context, require_review, dependencies, result, failure_reason, workspace_id, project_id, files, created_at, updated_at)
+       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21, $22)`,
       [
         id,
         params.tenantId,
@@ -6671,6 +6727,7 @@ var PostgreSQLTaskStore = class {
         params.failureReason ?? null,
         params.workspaceId || null,
         params.projectId || null,
+        params.files ? JSON.stringify(params.files) : null,
         now,
         now
       ]
@@ -6775,6 +6832,10 @@ var PostgreSQLTaskStore = class {
       setClauses.push(`metadata = $${paramIndex++}`);
       params.push(JSON.stringify(updates.metadata));
     }
+    if (updates.files !== void 0) {
+      setClauses.push(`files = $${paramIndex++}`);
+      params.push(JSON.stringify(updates.files));
+    }
     if (updates.parentId !== void 0) {
       setClauses.push(`parent_id = $${paramIndex++}`);
       params.push(updates.parentId);
@@ -7954,14 +8015,14 @@ async function createPgStoreConfig(connectionString) {
   mm.register(createA2AApiKeysTable);
   mm.register(addWorkspaceProjectToQueue);
   mm.register(addAssistantOwnerUserId);
-  mm.register(createTasksTable);
+  for (const m of taskMigrations) {
+    mm.register(m);
+  }
   mm.register(createMenuItemsTable);
   mm.register(createSharedResourcesTable);
   mm.register(addFileContentType);
   mm.register(createCollectionsTable);
-  mm.register(addTaskFieldsMigration);
   mm.register(createTaskWorkItemsMigration);
-  mm.register(addTaskProjectFieldsMigration);
   mm.register(addWorkItemProjectFieldsMigration);
   mm.register(createConnectionConfigsTable);
   mm.register(addWorkflowRunsTenantStatusUpdatedIndex);
@@ -8585,6 +8646,7 @@ export {
   addScheduleTenantId,
   addSkillTenantId,
   addStepThreadId,
+  addTaskIdToEvalRuns,
   addThreadTenantId,
   addWorkflowRunsTenantStatusUpdatedIndex,
   alterChannelInstallationsTable,