address minor gaps

Author: icecrasher321

apps/sim/executor/execution/engine.ts

@@ -422,12 +422,7 @@ export class ExecutionEngine {
     )
 
     if (activatedEdges.length > 0) {
-      const existing = this.context.activatedEdges.get(nodeId)
-      if (existing) {
-        existing.push(...activatedEdges)
-      } else {
-        this.context.activatedEdges.set(nodeId, [...activatedEdges])
-      }
+      this.context.activatedEdges.set(nodeId, activatedEdges)
     }
 
     logger.info('Processing outgoing edges', {

apps/sim/executor/orchestrators/loop.ts

@@ -724,6 +724,7 @@ export class LoopOrchestrator {
     const output = { results: [] }
     const block = ctx.workflow?.blocks.find((b) => b.id === blockId)
     const blockName = block?.metadata?.name ?? blockType
+    const iterationContext = buildContainerIterationContext(ctx, blockId)
 
     ctx.blockLogs.push({
       blockId,
@@ -742,13 +743,19 @@ export class LoopOrchestrator {
     }
 
     if (this.contextExtensions?.onBlockComplete) {
-      this.contextExtensions.onBlockComplete(blockId, blockName, blockType, {
-        output,
-        executionTime: DEFAULTS.EXECUTION_TIME,
-        startedAt: now,
-        executionOrder,
-        endedAt: now,
-      })
+      this.contextExtensions.onBlockComplete(
+        blockId,
+        blockName,
+        blockType,
+        {
+          output,
+          executionTime: DEFAULTS.EXECUTION_TIME,
+          startedAt: now,
+          executionOrder,
+          endedAt: now,
+        },
+        iterationContext
+      )
     }
   }
 }

apps/sim/executor/orchestrators/parallel.ts

@@ -381,6 +381,7 @@ export class ParallelOrchestrator {
     const output = { results: [] }
     const block = ctx.workflow?.blocks.find((b) => b.id === blockId)
     const blockName = block?.metadata?.name ?? blockType
+    const iterationContext = buildContainerIterationContext(ctx, blockId)
 
     ctx.blockLogs.push({
       blockId,
@@ -399,13 +400,19 @@ export class ParallelOrchestrator {
     }
 
     if (this.contextExtensions?.onBlockComplete) {
-      this.contextExtensions.onBlockComplete(blockId, blockName, blockType, {
-        output,
-        executionTime: DEFAULTS.EXECUTION_TIME,
-        startedAt: now,
-        executionOrder,
-        endedAt: now,
-      })
+      this.contextExtensions.onBlockComplete(
+        blockId,
+        blockName,
+        blockType,
+        {
+          output,
+          executionTime: DEFAULTS.EXECUTION_TIME,
+          startedAt: now,
+          executionOrder,
+          endedAt: now,
+        },
+        iterationContext
+      )
     }
   }
 }