2 small things

PrefectHQ · Sep 9, 2024 · 4e5603f · 4e5603f
1 parent eba6ac5
commit 4e5603f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/all_code.txt b/all_code.txt
diff --git a/src/controlflow/events/events.py b/src/controlflow/events/events.py
@@ -101,6 +101,7 @@ def _message(cls, v):
     def _finalize(self):
         self.delta["name"] = self.agent.name
         self.snapshot["name"] = self.agent.name
+        return self
 
     @property
     def delta_message(self) -> AIMessageChunk: