diff --git a/ai_datastream/agent/langgraph/streamer.py b/ai_datastream/agent/langgraph/streamer.py
index 4755748..9f54d79 100644
--- a/ai_datastream/agent/langgraph/streamer.py
+++ b/ai_datastream/agent/langgraph/streamer.py
@@ -37,8 +37,11 @@ def __init__(self, agent: CompiledGraph, thread_id: Union[str, None] = None):
         self.status = StreamStatus.INIT
 
     def _handle_stream_finish(self) -> Generator[DataStreamPart, None, None]:
-        snapshot = self.agent.get_state(self.config)
-        if snapshot.next:
+        try:
+            snapshot = self.agent.get_state(self.config)
+        except ValueError:
+            snapshot = None
+        if snapshot and snapshot.next:
             self.status = StreamStatus.INTERRUPTED
             return
         if self.converter.current_message_id is not None: