Small type fix for 3.9

Author: monoxgas

pyproject.toml

@@ -20,6 +20,7 @@ pandas = "^2.2.2"
 eval-type-backport = "^0.2.0" # For 3.9 future annotations
 elasticsearch = "^8.13.2"
 xmltodict = "^0.13.0"
+types-colorama = "^0.4.15.20240311"
 
 vllm = { version = "0.5.0", optional = true }
 transformers = { version = "^4.41.0", optional = true }
@@ -31,7 +32,7 @@ click = { version = "^8.1.7", optional = true }
 httpx = { version = "^0.27.0", optional = true }
 aiodocker = { version = "^0.22.2", optional = true }
 websockets = { version = "^13.0", optional = true }
-types-colorama = "^0.4.15.20240311"
+
 
 [tool.poetry.extras]
 examples = ["asyncssh", "types-requests", "click", "httpx", "aiodocker", "websockets"]

rigging/chat.py

@@ -1024,7 +1024,7 @@ async def _post_run(self, chats: list[Chat], on_failed: FailMode) -> ChatList:
         for then_callback in self.then_callbacks:
             coros = [then_callback(chat) for chat in chats]
             new_chats = await asyncio.gather(*coros)
-            if not all(isinstance(c, Chat | None) for c in new_chats):
+            if not all(isinstance(c, Chat) or c is None for c in new_chats):
                 raise ValueError(
                     f".then() callback must return a Chat object or None ({then_callback.__call__.__name__})"
                 )