va/agent
0
0
Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

Add notification model #31

Merged
A8065384 merged 3 commits from notif-model into main 2026-03-10 23:50:41 +00:00
Conversation 0 Commits 3 Files Changed 3 +5 -15
1 changed files with 5 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 014ceb9da6 - Show all commits

20
src/va_agent/notifications.py
View File

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