fix: fix both lg hitl post interrupt calls

Author: ranst91

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

@@ -118,7 +118,12 @@ async def _handle_stream_events(self, input: RunAgentInput) -> AsyncGenerator[st
         config["configurable"] = {**(config.get('configurable', {})), "thread_id": thread_id}
 
         agent_state = await self.graph.aget_state(config)
-        self.active_run["mode"] = "continue" if thread_id and self.active_run.get("node_name") != "__end__" and self.active_run.get("node_name") else "start"
+        resume_input = forwarded_props.get('command', {}).get('resume', None)
+
+        if resume_input is None and thread_id and self.active_run.get("node_name") != "__end__" and self.active_run.get("node_name"):
+            self.active_run["mode"] = "continue"
+        else:
+            self.active_run["mode"] = "start"
 
         prepared_stream_response = await self.prepare_stream(input=input, agent_state=agent_state, config=config)
 
@@ -264,6 +269,7 @@ async def prepare_stream(self, input: RunAgentInput, agent_state: State, config:
         interrupts = agent_state.tasks[0].interrupts if agent_state.tasks and len(agent_state.tasks) > 0 else []
         has_active_interrupts = len(interrupts) > 0
         resume_input = forwarded_props.get('command', {}).get('resume', None)
+
         self.active_run["schema_keys"] = self.get_schema_keys(config)
 
         non_system_messages = [msg for msg in langchain_messages if not isinstance(msg, SystemMessage)]

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

@@ -257,11 +257,11 @@ export class LangGraphAgent extends AbstractAgent {
     this.activeRun!.graphInfo = await this.client.assistants.getGraph(this.assistant.assistant_id);
 
     const mode =
-      threadId && this.activeRun!.nodeName != "__end__" && this.activeRun!.nodeName
+      !forwardedProps?.command?.resume && threadId && this.activeRun!.nodeName != "__end__" && this.activeRun!.nodeName
         ? "continue"
         : "start";
 
-    if (mode === "continue" && !forwardedProps?.command?.resume) {
+    if (mode === "continue") {
       const nodeBefore = this.activeRun!.graphInfo.edges.find(e => e.target === this.activeRun!.nodeName);
       await this.client.threads.updateState(threadId, {
         values: inputState,
@@ -447,7 +447,7 @@ export class LangGraphAgent extends AbstractAgent {
           event: chunkData,
         });
 
-        this.handleSingleEvent(chunkData, state);
+        this.handleSingleEvent(chunkData);
       }
 
       state = await this.client.threads.getState(threadId);
@@ -500,7 +500,7 @@ export class LangGraphAgent extends AbstractAgent {
     }
   }
 
-  handleSingleEvent(event: any, state: State): void {
+  handleSingleEvent(event: any): void {
     switch (event.event) {
       case LangGraphEventTypes.OnChatModelStream:
         let shouldEmitMessages = event.metadata["emit-messages"] ?? true;