From 4e5603f35ac72d53d12235066a360da4f475e015 Mon Sep 17 00:00:00 2001
From: zzstoatzz <thrast36@gmail.com>
Date: Mon, 9 Sep 2024 10:50:35 -0500
Subject: [PATCH] 2 small things

---
 all_code.txt                     | 1 -
 src/controlflow/events/events.py | 1 +
 2 files changed, 1 insertion(+), 1 deletion(-)
 delete mode 100644 all_code.txt

diff --git a/all_code.txt b/all_code.txt
deleted file mode 100644
index 8b13789..0000000
--- a/all_code.txt
+++ /dev/null
@@ -1 +0,0 @@
-
diff --git a/src/controlflow/events/events.py b/src/controlflow/events/events.py
index 90b6910..74eaa2b 100644
--- 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: