diff --git a/src/va_agent/notifications.py b/src/va_agent/notifications.py index 4177937..2e12eda 100644 --- a/src/va_agent/notifications.py +++ b/src/va_agent/notifications.py @@ -78,9 +78,7 @@ class NotificationDocument(BaseModel): class NotificationBackend(Protocol): """Backend-agnostic interface for notification storage.""" - async def get_recent_notifications( - self, phone_number: str - ) -> list[Notification]: + async def get_recent_notifications(self, phone_number: str) -> list[Notification]: """Return recent notifications for *phone_number*.""" ... @@ -113,9 +111,7 @@ class FirestoreNotificationBackend: self._max_to_notify = max_to_notify self._window_hours = window_hours - async def get_recent_notifications( - self, phone_number: str - ) -> list[Notification]: + async def get_recent_notifications(self, phone_number: str) -> list[Notification]: """Get recent notifications for a user. Retrieves notifications created within the configured time window, @@ -148,9 +144,7 @@ class FirestoreNotificationBackend: cutoff = time.time() - (self._window_hours * 3600) parsed = [ - n - for n in document.notificaciones - if n.timestamp_creacion >= cutoff + n for n in document.notificaciones if n.timestamp_creacion >= cutoff ] if not parsed: @@ -212,9 +206,7 @@ class RedisNotificationBackend: self._max_to_notify = max_to_notify self._window_hours = window_hours - async def get_recent_notifications( - self, phone_number: str - ) -> list[Notification]: + async def get_recent_notifications(self, phone_number: str) -> list[Notification]: """Get recent notifications for a user from Redis. Reads from the ``notification:{phone}`` key, parses the JSON @@ -246,9 +238,7 @@ class RedisNotificationBackend: cutoff = time.time() - (self._window_hours * 3600) parsed = [ - n - for n in document.notificaciones - if n.timestamp_creacion >= cutoff + n for n in document.notificaciones if n.timestamp_creacion >= cutoff ] if not parsed: