Handle begin message

trypromptly · Oct 22, 2024 · ac56b05 · ac56b05
1 parent b9c30aa
commit ac56b05
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/llmstack/play/actor.py b/llmstack/play/actor.py
@@ -68,6 +68,10 @@ def on_receive(self, message: Message) -> Any:
             self.on_error(message.data.errors)
             return
 
+        if message.type == MessageType.BEGIN:
+            self.input(self._messages)
+            return
+
         if message.type == MessageType.CONTENT_STREAM_BEGIN:
             pass