Merge pull request #293 from PrefectHQ/nits

2 small things
PrefectHQ · Sep 9, 2024 · b323232 · b323232
2 parents eba6ac5 + 4e5603f
commit b323232
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: