fix: update execution serialization and flow migration logic

yakud · yakud · commit 78949c34e0ec · 2025-10-05T12:47:18.000+03:00
diff --git a/.changeset/fuzzy-rockets-drive.md b/.changeset/fuzzy-rockets-drive.md
@@ -0,0 +1,15 @@
+---
+"@badaitech/chaingraph-execution-worker": patch
+"@badaitech/chaingraph-execution-api": patch
+"@badaitech/chaingraph-executor": patch
+"@badaitech/badai-api-example": patch
+"@badaitech/typescript-config": patch
+"@badaitech/chaingraph-nodes": patch
+"@badaitech/chaingraph-types": patch
+"@badaitech/chaingraph-frontend": patch
+"@badaitech/chaingraph-trpc": patch
+"@badaitech/chaingraph-backend": patch
+"@badaitech/badai-api": patch
+---
+
+fix: update execution serialization and flow migration logic
diff --git a/.changeset/pre.json b/.changeset/pre.json
@@ -18,6 +18,7 @@
     "dry-otters-share",
     "early-baboons-lay",
     "fluffy-chairs-yell",
+    "fuzzy-rockets-drive",
     "good-trains-enjoy",
     "green-beans-poke",
     "little-drinks-travel",
diff --git a/apps/chaingraph-backend/CHANGELOG.md b/apps/chaingraph-backend/CHANGELOG.md
@@ -1,5 +1,15 @@
 # @badaitech/chaingraph-backend
 
+## 0.5.5-dev.40
+
+### Patch Changes
+
+- fix: update execution serialization and flow migration logic
+- Updated dependencies
+  - @badaitech/chaingraph-nodes@0.5.5-dev.40
+  - @badaitech/chaingraph-types@0.5.5-dev.40
+  - @badaitech/chaingraph-trpc@0.5.5-dev.40
+
 ## 0.5.5-dev.39
 
 ### Patch Changes
diff --git a/apps/chaingraph-backend/package.json b/apps/chaingraph-backend/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@badaitech/chaingraph-backend",
   "type": "module",
-  "version": "0.5.5-dev.39",
+  "version": "0.5.5-dev.40",
   "private": false,
   "description": "Backend server for the Chaingraph project",
   "license": "BUSL-1.1",
diff --git a/apps/chaingraph-execution-api/CHANGELOG.md b/apps/chaingraph-execution-api/CHANGELOG.md
@@ -1,5 +1,15 @@
 # @badaitech/chaingraph-execution-api
 
+## 0.5.5-dev.40
+
+### Patch Changes
+
+- fix: update execution serialization and flow migration logic
+- Updated dependencies
+  - @badaitech/chaingraph-executor@0.5.5-dev.40
+  - @badaitech/chaingraph-types@0.5.5-dev.40
+  - @badaitech/chaingraph-trpc@0.5.5-dev.40
+
 ## 0.5.5-dev.39
 
 ### Patch Changes
diff --git a/apps/chaingraph-execution-api/package.json b/apps/chaingraph-execution-api/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@badaitech/chaingraph-execution-api",
   "type": "module",
-  "version": "0.5.5-dev.39",
+  "version": "0.5.5-dev.40",
   "private": false,
   "description": "Chaingraph tRPC Server - Scalable API server for Chaingraph execution management",
   "license": "BUSL-1.1",
diff --git a/apps/chaingraph-execution-worker/CHANGELOG.md b/apps/chaingraph-execution-worker/CHANGELOG.md
@@ -1,5 +1,16 @@
 # @badaitech/chaingraph-execution-worker
 
+## 0.5.5-dev.40
+
+### Patch Changes
+
+- fix: update execution serialization and flow migration logic
+- Updated dependencies
+  - @badaitech/chaingraph-executor@0.5.5-dev.40
+  - @badaitech/chaingraph-nodes@0.5.5-dev.40
+  - @badaitech/chaingraph-types@0.5.5-dev.40
+  - @badaitech/chaingraph-trpc@0.5.5-dev.40
+
 ## 0.5.5-dev.39
 
 ### Patch Changes
diff --git a/apps/chaingraph-execution-worker/package.json b/apps/chaingraph-execution-worker/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@badaitech/chaingraph-execution-worker",
   "type": "module",
-  "version": "0.5.5-dev.39",
+  "version": "0.5.5-dev.40",
   "private": false,
   "description": "Chaingraph Execution Worker Service which connected to Kafka and process the flow executions",
   "license": "BUSL-1.1",
diff --git a/apps/chaingraph-frontend/CHANGELOG.md b/apps/chaingraph-frontend/CHANGELOG.md
@@ -1,5 +1,17 @@
 # @badaitech/chaingraph-frontend
 
+## 0.5.5-dev.40
+
+### Patch Changes
+
+- fix: update execution serialization and flow migration logic
+- Updated dependencies
+  - @badaitech/chaingraph-executor@0.5.5-dev.40
+  - @badaitech/chaingraph-nodes@0.5.5-dev.40
+  - @badaitech/chaingraph-types@0.5.5-dev.40
+  - @badaitech/chaingraph-trpc@0.5.5-dev.40
+  - @badaitech/badai-api@0.5.5-dev.40
+
 ## 0.5.5-dev.39
 
 ### Patch Changes
diff --git a/apps/chaingraph-frontend/package.json b/apps/chaingraph-frontend/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@badaitech/chaingraph-frontend",
   "type": "module",
-  "version": "0.5.5-dev.39",
+  "version": "0.5.5-dev.40",
   "private": false,
   "description": "Frontend application for the Chaingraph project",
   "license": "BUSL-1.1",
diff --git a/packages/badai-api-example/package.json b/packages/badai-api-example/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@badaitech/badai-api-example",
   "type": "module",
-  "version": "0.5.5-dev.39",
+  "version": "0.5.5-dev.40",
   "private": true,
   "description": "Bad AI GraphQL API client example",
   "license": "BUSL-1.1",
diff --git a/packages/badai-api/CHANGELOG.md b/packages/badai-api/CHANGELOG.md
@@ -1,5 +1,11 @@
 # @badaitech/badai-api
 
+## 0.5.5-dev.40
+
+### Patch Changes
+
+- fix: update execution serialization and flow migration logic
+
 ## 0.5.5-dev.39
 
 ### Patch Changes
diff --git a/packages/badai-api/package.json b/packages/badai-api/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@badaitech/badai-api",
   "type": "module",
-  "version": "0.5.5-dev.39",
+  "version": "0.5.5-dev.40",
   "private": false,
   "description": "Bad AI GraphQL API client",
   "license": "BUSL-1.1",
diff --git a/packages/chaingraph-executor/package.json b/packages/chaingraph-executor/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@badaitech/chaingraph-executor",
   "type": "module",
-  "version": "0.5.5-dev.39",
+  "version": "0.5.5-dev.40",
   "private": false,
   "description": "Chaingraph Message Bus for inter-service communication",
   "license": "BUSL-1.1",
diff --git a/packages/chaingraph-executor/server/services/ExecutionService.ts b/packages/chaingraph-executor/server/services/ExecutionService.ts
@@ -103,10 +103,9 @@ export class ExecutionService implements IExecutionService {
       undefined,
       executionRow.id,
       executionRow.integration || undefined,
-      executionRow.parentExecutionId || undefined,
       executionRow.rootExecutionId || executionRow.id,
+      executionRow.parentExecutionId || undefined,
       executionRow.externalEvents?.[0] || undefined,
-      // undefined, // TODO: do we really want to provide event data to the context?
       !!executionRow.parentExecutionId, // isChildExecution
       currentDepth,
       (nodeId: string) => flow.nodes.get(nodeId),
diff --git a/packages/chaingraph-nodes/CHANGELOG.md b/packages/chaingraph-nodes/CHANGELOG.md
@@ -1,5 +1,14 @@
 # @badaitech/chaingraph-nodes
 
+## 0.5.5-dev.40
+
+### Patch Changes
+
+- fix: update execution serialization and flow migration logic
+- Updated dependencies
+  - @badaitech/chaingraph-types@0.5.5-dev.40
+  - @badaitech/badai-api@0.5.5-dev.40
+
 ## 0.5.5-dev.39
 
 ### Patch Changes
diff --git a/packages/chaingraph-nodes/package.json b/packages/chaingraph-nodes/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@badaitech/chaingraph-nodes",
   "type": "module",
-  "version": "0.5.5-dev.39",
+  "version": "0.5.5-dev.40",
   "private": false,
   "description": "Node components for the Chaingraph project",
   "license": "BUSL-1.1",
diff --git a/packages/chaingraph-trpc/CHANGELOG.md b/packages/chaingraph-trpc/CHANGELOG.md
@@ -1,5 +1,15 @@
 # @badaitech/chaingraph-trpc
 
+## 0.5.5-dev.40
+
+### Patch Changes
+
+- fix: update execution serialization and flow migration logic
+- Updated dependencies
+  - @badaitech/chaingraph-nodes@0.5.5-dev.40
+  - @badaitech/chaingraph-types@0.5.5-dev.40
+  - @badaitech/badai-api@0.5.5-dev.40
+
 ## 0.5.5-dev.39
 
 ### Patch Changes
diff --git a/packages/chaingraph-trpc/package.json b/packages/chaingraph-trpc/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@badaitech/chaingraph-trpc",
   "type": "module",
-  "version": "0.5.5-dev.39",
+  "version": "0.5.5-dev.40",
   "private": false,
   "description": "tRPC client and server for the Chaingraph V2",
   "license": "BUSL-1.1",
diff --git a/packages/chaingraph-trpc/server/procedures/flow/paste-nodes.ts b/packages/chaingraph-trpc/server/procedures/flow/paste-nodes.ts
@@ -13,6 +13,9 @@ import type {
   IPort,
   SerializedEdge,
 } from '@badaitech/chaingraph-types'
+import {
+  FlowMigration,
+} from '@badaitech/chaingraph-types'
 import {
   Edge,
   EdgeStatus,
@@ -141,7 +144,7 @@ export const pasteNodes = flowContextProcedure
       }
 
       for (const node of createdNodes) {
-      // iterate over created nodes and fix the parent ID if it exists
+        // iterate over created nodes and fix the parent ID if it exists
         if (node.metadata.parentNodeId) {
           const newParentId = nodeIdMapping.get(node.metadata.parentNodeId)
           if (newParentId) {
@@ -240,6 +243,9 @@ export const pasteNodes = flowContextProcedure
 
       await flow.addEdges(edgesToAdd, false)
 
+      // Force migrate flow to v2
+      FlowMigration.migrateFlowFromV1ToV2(flow)
+
       // Save the updated flow
       await ctx.flowStore.updateFlow(flow as Flow)
 
diff --git a/packages/chaingraph-types/CHANGELOG.md b/packages/chaingraph-types/CHANGELOG.md
@@ -1,5 +1,11 @@
 # @badaitech/chaingraph-types
 
+## 0.5.5-dev.40
+
+### Patch Changes
+
+- fix: update execution serialization and flow migration logic
+
 ## 0.5.5-dev.39
 
 ### Patch Changes
diff --git a/packages/chaingraph-types/package.json b/packages/chaingraph-types/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@badaitech/chaingraph-types",
   "type": "module",
-  "version": "0.5.5-dev.39",
+  "version": "0.5.5-dev.40",
   "private": false,
   "description": "Core type definitions for the Chaingraph project",
   "license": "BUSL-1.1",
diff --git a/packages/chaingraph-types/src/execution/execution-context.ts b/packages/chaingraph-types/src/execution/execution-context.ts
@@ -160,7 +160,6 @@ export class ExecutionContext {
   cloneForChildExecution(eventData: EmittedEventContext, childExecutionId: string): ExecutionContext {
     const ctx = new ExecutionContext(
       this.flowId!,
-      // new AbortController(), // New abort controller for child
       this.abortController,
       { ...this.metadata, parentExecutionId: this.executionId },
       childExecutionId,
diff --git a/packages/chaingraph-types/src/flow/flow-migration.ts b/packages/chaingraph-types/src/flow/flow-migration.ts
@@ -149,7 +149,7 @@ export class FlowMigration {
       const isSystemPort = portConfig.metadata?.isSystemPort === true
       const newPortId = isSystemPort
         ? this.replacePortIdSchemaForSystemPort(oldPortId, portConfig.key || '')
-        : this.replacePortIdSchema(oldPortId)
+        : portConfig.key || this.replacePortIdSchema(oldPortId)
       const isPortIdChanged = oldPortId !== newPortId
 
       if (isPortIdChanged) {
@@ -241,8 +241,7 @@ export class FlowMigration {
     if (portKey && portKey.startsWith('__')) {
       return portKey
     }
-    // Fallback: if key doesn't have __, add it based on the old ID
-    const baseId = oldPortId.replace(/:PO[\w-]{16}/g, '')
-    return baseId.startsWith('__') ? baseId : `__${baseId}`
+
+    return `__${portKey}`
   }
 }
diff --git a/packages/chaingraph-types/src/flow/serializer/flow-serializer.ts b/packages/chaingraph-types/src/flow/serializer/flow-serializer.ts
@@ -216,7 +216,6 @@ export class FlowSerializer {
     )
     //
     node.initializePortsFromConfigs(portsConfigProcessed || new Map())
-    // node.initializePortsFromConfigs(portsConfig)
 
     // Update node properties from the actual port values
     for (const serializedPort of serializedNode.ports || []) {
diff --git a/packages/typescript-config/CHANGELOG.md b/packages/typescript-config/CHANGELOG.md
@@ -1,5 +1,11 @@
 # @badaitech/typescript-config
 
+## 0.5.5-dev.40
+
+### Patch Changes
+
+- fix: update execution serialization and flow migration logic
+
 ## 0.5.5-dev.39
 
 ### Patch Changes
diff --git a/packages/typescript-config/package.json b/packages/typescript-config/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@badaitech/typescript-config",
-  "version": "0.5.5-dev.39",
+  "version": "0.5.5-dev.40",
   "private": true,
   "license": "MIT",
   "files": [

Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,7 @@`
`1`	`1`	`{`
`2`	`2`	`"name": "@badaitech/chaingraph-backend",`
`3`	`3`	`"type": "module",`
`4`		`- "version": "0.5.5-dev.39",`
	`4`	`+ "version": "0.5.5-dev.40",`
`5`	`5`	`"private": false,`
`6`	`6`	`"description": "Backend server for the Chaingraph project",`
`7`	`7`	`"license": "BUSL-1.1",`
Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,7 @@`
`1`	`1`	`{`
`2`	`2`	`"name": "@badaitech/chaingraph-execution-api",`
`3`	`3`	`"type": "module",`
`4`		`- "version": "0.5.5-dev.39",`
	`4`	`+ "version": "0.5.5-dev.40",`
`5`	`5`	`"private": false,`
`6`	`6`	`"description": "Chaingraph tRPC Server - Scalable API server for Chaingraph execution management",`
`7`	`7`	`"license": "BUSL-1.1",`
Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,7 @@`
`1`	`1`	`{`
`2`	`2`	`"name": "@badaitech/chaingraph-frontend",`
`3`	`3`	`"type": "module",`
`4`		`- "version": "0.5.5-dev.39",`
	`4`	`+ "version": "0.5.5-dev.40",`
`5`	`5`	`"private": false,`
`6`	`6`	`"description": "Frontend application for the Chaingraph project",`
`7`	`7`	`"license": "BUSL-1.1",`