Small type fix for 3.9

dreadnode · Aug 28, 2024 · 44d4e1d · 44d4e1d
1 parent 6c27f7e
commit 44d4e1d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pyproject.toml b/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"]

diff --git a/rigging/chat.py b/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__})"
                 )