@axiom-lattice/pg-stores 1.0.69 → 1.0.71

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/pg-stores",
-  "version": "1.0.69",
+  "version": "1.0.71",
   "description": "PG stores implementation for Axiom Lattice framework",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -22,8 +22,8 @@
   "dependencies": {
     "pg": "^8.16.3",
     "uuid": "^9.0.1",
-    "@axiom-lattice/core": "2.1.78",
-    "@axiom-lattice/protocols": "2.1.40"
+    "@axiom-lattice/core": "2.1.80",
+    "@axiom-lattice/protocols": "2.1.42"
   },
   "devDependencies": {
     "@types/node": "^20.11.24",

package/src/stores/PostgreSQLWorkflowTrackingStore.ts

@@ -188,6 +188,19 @@ export class PostgreSQLWorkflowTrackingStore implements WorkflowTrackingStore {
     };
   }
 
+  async upsertRunStep(request: CreateRunStepRequest): Promise<RunStep> {
+    await this.ensureInitialized();
+    // Look up by (run_id, step_type, step_name) — the idempotency key
+    const result = await this.pool.query(
+      `SELECT * FROM lattice_workflow_steps WHERE run_id = $1 AND step_type = $2 AND step_name = $3`,
+      [request.runId, request.stepType, request.stepName]
+    );
+    if (result.rows.length > 0) {
+      return this.mapRowToRunStep(result.rows[0]);
+    }
+    return this.createRunStep(request);
+  }
+
   async updateRunStep(runId: string, stepId: string, updates: UpdateRunStepRequest): Promise<RunStep | null> {
     await this.ensureInitialized();
     const now = new Date();