diff --git a/scripts/test_telegram_ingestion.py b/scripts/test_telegram_ingestion.py index 23f33d1..442788a 100644 --- a/scripts/test_telegram_ingestion.py +++ b/scripts/test_telegram_ingestion.py @@ -11,6 +11,7 @@ """ import sys +from collections.abc import Mapping from datetime import datetime, timedelta from pathlib import Path from typing import Any @@ -44,7 +45,7 @@ def _get_channel_config_value(channel: Any, field: str, default: Any) -> Any: from src.adapters.sqlite_repository import SQLiteRepository from src.adapters.telegram_client import TelegramClient from src.config.settings import get_settings -from src.domain.models import MessageSource +from src.domain.models import MessageSource, TelegramChannelConfig from src.domain.protocols import MessageClientProtocol from src.use_cases.ingest_telegram_messages import ingest_telegram_messages_use_case