Upload core/messaging.py with huggingface_hub

core/messaging.py

@@ -22,7 +22,9 @@ class MessageQueueInterface(ABC):
         """Publish a message to a topic/queue."""
 
     @abstractmethod
-    async def subscribe(self, topic: str, handler: Callable[[dict[str, Any]], None]) -> bool:
+    async def subscribe(
+        self, topic: str, handler: Callable[[dict[str, Any]], None]
+    ) -> bool:
         """Subscribe to a topic with a handler function."""
 
     @abstractmethod
@@ -63,7 +65,9 @@ class InMemoryMessageQueue(MessageQueueInterface):
         logger.debug(f"[MQ] Published to {topic}: {message.keys()}")
         return True
 
-    async def subscribe(self, topic: str, handler: Callable[[dict[str, Any]], None]) -> bool:
+    async def subscribe(
+        self, topic: str, handler: Callable[[dict[str, Any]], None]
+    ) -> bool:
         if topic not in self._subscribers:
             self._subscribers[topic] = []
         self._subscribers[topic].append(handler)