fix: recipient

Author: giancarloromeo

services/notifications/src/simcore_service_notifications/models/schemas.py

@@ -1,17 +1,25 @@
-from typing import Any, TypeAlias
+from typing import Annotated, Any, Literal, TypeAlias
 
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 
 
-class SMSRecipient(BaseModel):
+class _BaseRecipient(BaseModel):
+    type: str
+
+
+class SMSRecipient(_BaseRecipient):
+    type: Literal["sms"]
     phone_number: str
 
 
-class EmailRecipient(BaseModel):
-    email: str
+class EmailRecipient(_BaseRecipient):
+    type: Literal["email"]
+    address: str
 
 
-Recipient: TypeAlias = SMSRecipient | EmailRecipient
+Recipient: TypeAlias = Annotated[
+    EmailRecipient | SMSRecipient, Field(discriminator="type")
+]
 
 
 class NotificationMessage(BaseModel):

services/notifications/src/simcore_service_notifications/services/notifications_service.py

@@ -5,10 +5,6 @@
 
 from ..models.schemas import NotificationMessage, Recipient
 
-CHANNELS = {
-    "email": "notifications.email",
-}
-
 
 class TaskQueues(StrEnum):
     DEFAULT = "notifications.default"
@@ -23,8 +19,10 @@ async def send_notification(
     for recipient in recipients:
         await task_manager.send_task(
             TaskMetadata(
-                name="notifications.send_email",
+                name=f"notifications.{recipient.type}",
                 queue=TaskQueues.DEFAULT,
             ),
             task_context=TaskContext(),
+            message=message,
+            recipient=recipient,
         )