fix: use fallback when closing off active step

ranst91 · ranst91 · commit 6132f0c22302 · 2025-09-01T17:30:47.000+02:00
diff --git a/typescript-sdk/integrations/langgraph/python/ag_ui_langgraph/agent.py b/typescript-sdk/integrations/langgraph/python/ag_ui_langgraph/agent.py
@@ -728,7 +728,7 @@ def end_step(self):
         dispatch = self._dispatch_event(
             StepFinishedEvent(
                 type=EventType.STEP_FINISHED,
-                step_name=self.active_run["node_name"]
+                step_name=self.active_run["node_name"] or self.active_step
             )
         )
 
diff --git a/typescript-sdk/integrations/langgraph/src/agent.ts b/typescript-sdk/integrations/langgraph/src/agent.ts
@@ -977,7 +977,7 @@ export class LangGraphAgent extends AbstractAgent {
     }
     this.dispatchEvent({
       type: EventType.STEP_FINISHED,
-      stepName: this.activeRun!.nodeName!,
+      stepName: this.activeRun!.nodeName! ?? this.activeStep,
     });
     this.activeRun!.nodeName = undefined;
     this.activeStep = undefined;

Original file line number	Diff line number	Diff line change
`@@ -728,7 +728,7 @@ def end_step(self):`
`728`	`728`	`dispatch = self._dispatch_event(`
`729`	`729`	`StepFinishedEvent(`
`730`	`730`	`type=EventType.STEP_FINISHED,`
`731`		`- step_name=self.active_run["node_name"]`
	`731`	`+ step_name=self.active_run["node_name"] or self.active_step`
`732`	`732`	`)`
`733`	`733`	`)`
`734`	`734`
Original file line number	Diff line number	Diff line change
`@@ -977,7 +977,7 @@ export class LangGraphAgent extends AbstractAgent {`
`977`	`977`	`}`
`978`	`978`	`this.dispatchEvent({`
`979`	`979`	`type: EventType.STEP_FINISHED,`
`980`		`- stepName: this.activeRun!.nodeName!,`
	`980`	`+ stepName: this.activeRun!.nodeName! ?? this.activeStep,`
`981`	`981`	`});`
`982`	`982`	`this.activeRun!.nodeName = undefined;`
`983`	`983`	`this.activeStep = undefined;`