fix: dispatch step start when resuming an interrupt

Author: ranst91

typescript-sdk/integrations/langgraph/python/ag_ui_langgraph/agent.py

@@ -131,6 +131,12 @@ async def _handle_stream_events(self, input: RunAgentInput) -> AsyncGenerator[st
             RunStartedEvent(type=EventType.RUN_STARTED, thread_id=thread_id, run_id=self.active_run["id"])
         )
 
+        # In case of resume (interrupt), re-start resumed step
+        if resume_input and self.active_run.get("node_name"):
+            yield self._dispatch_event(
+                StepStartedEvent(type=EventType.STEP_STARTED, step_name=self.active_run.get("node_name"))
+            )
+
         state = prepared_stream_response["state"]
         stream = prepared_stream_response["stream"]
         config = prepared_stream_response["config"]

typescript-sdk/integrations/langgraph/src/agent.ts

@@ -169,7 +169,7 @@ export class LangGraphAgent extends AbstractAgent {
       return subscriber.error('No stream to regenerate');
     }
 
-    this.handleStreamEvents(preparedStream, threadId, subscriber, input, streamMode)
+    await this.handleStreamEvents(preparedStream, threadId, subscriber, input, streamMode)
   }
 
   async prepareRegenerateStream(input: RegenerateInput, streamMode: StreamMode | StreamMode[]) {
@@ -353,6 +353,11 @@ export class LangGraphAgent extends AbstractAgent {
         runId: this.activeRun!.id,
       });
 
+      // In case of resume (interrupt), re-start resumed step
+      if (forwardedProps?.command?.resume && this.activeRun!.nodeName) {
+        this.startStep(this.activeRun!.nodeName)
+      }
+
       for await (let streamResponseChunk of streamResponse) {
         // @ts-ignore
         if (!streamMode.includes(streamResponseChunk.event as StreamMode)) {