fix: imports

Author: giancarloromeo

services/notifications/src/simcore_service_notifications/clients/celery.py

@@ -6,10 +6,14 @@
 from celery_library.types import register_celery_types, register_pydantic_types
 from fastapi import FastAPI
 from fastapi_lifespan_manager import State
+from models_library.rpc.notifications.messages import (
+    EmailRecipient,
+    NotificationMessage,
+    SMSRecipient,
+)
 from settings_library.celery import CelerySettings
 
 from ..core.settings import ApplicationSettings
-from ..models.schemas import EmailRecipient, NotificationMessage, SMSRecipient
 
 _logger = logging.getLogger(__name__)
 

services/notifications/src/simcore_service_notifications/modules/celery/_email_tasks.py

@@ -3,8 +3,10 @@
 import logging
 
 from celery import Task  # type: ignore[import-untyped]
-
-from ...models.schemas import EmailRecipient, NotificationMessage
+from models_library.rpc.notifications.messages import (
+    EmailRecipient,
+    NotificationMessage,
+)
 
 _logger = logging.getLogger(__name__)
 

services/notifications/src/simcore_service_notifications/modules/celery/tasks.py

@@ -4,9 +4,13 @@
 from celery import Celery  # type: ignore[import-untyped]
 from celery_library.task import register_task
 from celery_library.types import register_celery_types, register_pydantic_types
+from models_library.rpc.notifications.messages import (
+    EmailRecipient,
+    NotificationMessage,
+    SMSRecipient,
+)
 from servicelib.logging_utils import log_context
 
-from ...models.schemas import EmailRecipient, NotificationMessage, SMSRecipient
 from ...modules.celery._email_tasks import EMAIL_CHANNEL_NAME, send_email
 
 _logger = logging.getLogger(__name__)

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

@@ -1,10 +1,9 @@
 from enum import StrEnum
 
+from models_library.rpc.notifications.messages import NotificationMessage, Recipient
 from servicelib.celery.models import TaskContext, TaskMetadata
 from servicelib.celery.task_manager import TaskManager
 
-from ..models.schemas import NotificationMessage, Recipient
-
 
 class TaskQueues(StrEnum):
     DEFAULT = "notifications.default"