reverting utils.py

rabbitmq-community · May 23, 2024 · 4be9a95 · 4be9a95
1 parent 106bece
commit 4be9a95
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/rstream/utils.py b/rstream/utils.py
@@ -6,7 +6,7 @@
 from dataclasses import dataclass
 from typing import Any, Callable, Generator, Optional
 
-from .amqp import Message, _MessageProtocol
+from .amqp import AMQPMessage, _MessageProtocol
 
 
 @dataclass
@@ -54,7 +54,7 @@ class FilterConfiguration:
     def __init__(
         self,
         values_to_filter: list[str],
-        predicate: Optional[Callable[[Message], bool]] = None,
+        predicate: Optional[Callable[[AMQPMessage], bool]] = None,
         match_unfiltered: bool = False,
     ):
         self._values_to_filter = values_to_filter
@@ -64,7 +64,7 @@ def __init__(
     def values(self) -> list[str]:
         return self._values_to_filter
 
-    def post_filler(self) -> Optional[Callable[[Message], bool]]:
+    def post_filler(self) -> Optional[Callable[[AMQPMessage], bool]]:
         return self._predicate
 
     def match_unfiltered(self) -> bool: