fix: emit the manual state when manually emit state event is discovered (#186)

Author: ranst91

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

@@ -602,7 +602,7 @@ async def _handle_single_event(self, event: Any, state: State) -> AsyncGenerator
             elif event["name"] == CustomEventNames.ManuallyEmitState:
                 self.active_run["manually_emitted_state"] = event["data"]
                 yield self._dispatch_event(
-                    StateSnapshotEvent(type=EventType.STATE_SNAPSHOT, snapshot=self.get_state_snapshot(state), raw_event=event)
+                    StateSnapshotEvent(type=EventType.STATE_SNAPSHOT, snapshot=self.get_state_snapshot(self.active_run["manually_emitted_state"]), raw_event=event)
                 )
 
             yield self._dispatch_event(

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

@@ -609,7 +609,7 @@ export class LangGraphAgent extends AbstractAgent {
           this.activeRun!.manuallyEmittedState = event.data;
           this.dispatchEvent({
             type: EventType.STATE_SNAPSHOT,
-            snapshot: this.getStateSnapshot(state),
+            snapshot: this.getStateSnapshot(this.activeRun!.manuallyEmittedState!),
             rawEvent: event,
           });
         }