refactor channel flow

app/channel.py

@@ -0,0 +1,71 @@
+from __future__ import annotations
+from typing import Dict, TYPE_CHECKING
+
+from loguru import logger
+
+from .config import get_settings
+from .message_processor import MessageProcessor
+
+if TYPE_CHECKING:
+    from .llm import LLMClient
+    from .sheets import SheetsClient
+    from .supabase_db import SupabaseClient
+    from .embedding import EmbeddingClient
+    from .reranker import Reranker
+
+
+class Channel:
+    """
+    Đại diện cho một kênh giao tiếp độc lập (ví dụ: một Page Facebook).
+    Mỗi channel quản lý nhiều cuộc hội thoại (conversations) với người dùng.
+    Nó sử dụng các client được chia sẻ từ ChannelManager.
+    """
+
+    def __init__(
+        self,
+        platform: str,
+        page_id: str,
+        llm_client: LLMClient,
+        sheets_client: SheetsClient,
+        supabase_client: SupabaseClient,
+        embedding_client: EmbeddingClient,
+        reranker: Reranker,
+    ):
+        self.platform = platform
+        self.page_id = page_id
+
+        # Gán các client được chia sẻ từ ChannelManager
+        self.llm = llm_client
+        self.sheets = sheets_client
+        self.supabase = supabase_client
+        self.embedder = embedding_client
+        self.reranker = reranker
+
+        self.conversations: Dict[str, MessageProcessor] = {}
+        self._page_token: str | None = None
+
+    def get_or_create_conversation(self, sender_id: str) -> MessageProcessor:
+        """Lấy hoặc tạo một cuộc hội thoại mới (MessageProcessor) cho một người dùng."""
+        if sender_id not in self.conversations:
+            # MessageProcessor sẽ sử dụng các client của Channel (self)
+            self.conversations[sender_id] = MessageProcessor(self, sender_id)
+        return self.conversations[sender_id]
+
+    def get_sheets_client(self) -> SheetsClient:
+        """Trả về instance của SheetsClient cho channel này."""
+        return self.sheets
+
+    def get_page_token(self, force_refresh: bool = False) -> str | None:
+        """Lấy page access token cho channel này."""
+        if not self._page_token or force_refresh:
+            page_tokens = get_settings().facebook_page_tokens
+            self._page_token = page_tokens.get(self.page_id) if page_tokens else None
+            if not self._page_token:
+                logger.warning(
+                    f"Không tìm thấy page token cho page_id: {self.page_id} trong config."
+                )
+        return self._page_token
+
+    def invalidate_page_token(self):
+        """Vô hiệu hóa page token hiện tại để buộc làm mới."""
+        self._page_token = None

app/channel_manager.py

@@ -1,13 +1,54 @@
-from app.chat_channel import ChatChannel
+from __future__ import annotations
+from typing import Dict, Optional, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .llm import LLMClient
+    from .sheets import SheetsClient
+    from .supabase_db import SupabaseClient
+    from .embedding import EmbeddingClient
+    from .reranker import Reranker
+    from .channel import Channel
 
 class ChannelManager:
+    """
+    Quản lý tất cả các channel (ví dụ: các Page Facebook) đang hoạt động.
+    Lưu trữ các client dùng chung để truyền cho các channel, tối ưu hóa tài nguyên.
+    """
     def __init__(self):
-        self.channels = {}  # {(channel_type, page_id): ChatChannel}
+        self.channels: Dict[str, Channel] = {}
+        # Khởi tạo các client dùng chung là None, sẽ được gán từ main.py
+        self.llm_client: Optional[LLMClient] = None
+        self.sheets_client: Optional[SheetsClient] = None
+        self.supabase_client: Optional[SupabaseClient] = None
+        self.embedding_client: Optional[EmbeddingClient] = None
+        self.reranker: Optional[Reranker] = None
+
+    def set_shared_clients(
+        self,
+        llm_client: LLMClient,
+        sheets_client: SheetsClient,
+        supabase_client: SupabaseClient,
+        embedding_client: EmbeddingClient,
+        reranker: Reranker,
+    ):
+        """Gán các client được chia sẻ từ main.py."""
+        self.llm_client = llm_client
+        self.sheets_client = sheets_client
+        self.supabase_client = supabase_client
+        self.embedding_client = embedding_client
+        self.reranker = reranker
+
+    def get_or_create_channel(self, platform: str, page_id: str) -> Channel:
+        """Lấy hoặc tạo một channel mới và truyền các client dùng chung vào."""
+        channel_key = f"{platform}:{page_id}"
+        if channel_key not in self.channels:
+            # Kiểm tra để đảm bảo các client đã được gán
+            if not all([self.llm_client, self.sheets_client, self.supabase_client, self.embedding_client, self.reranker]):
+                raise RuntimeError("Một hoặc nhiều client dùng chung chưa được thiết lập cho ChannelManager.")
 
-    def get_or_create_channel(self, channel_type: str, page_id: str) -> ChatChannel:
-        key = (channel_type, page_id)
-        if key not in self.channels:
-            self.channels[key] = ChatChannel(page_id, channel_type)
-        return self.channels[key]
+            from .channel import Channel
+            self.channels[channel_key] = Channel(platform=platform, page_id=page_id, llm_client=self.llm_client, sheets_client=self.sheets_client, supabase_client=self.supabase_client, embedding_client=self.embedding_client, reranker=self.reranker)
+        return self.channels[channel_key]
 
-channel_manager = ChannelManager() 
+# Khởi tạo instance singleton để sử dụng trong toàn bộ ứng dụng
+channel_manager = ChannelManager()

app/main.py

@@ -88,12 +88,24 @@ llm_client = create_llm_client(
     ),
 )
 
-reranker = Reranker()
+reranker = Reranker() # Khởi tạo Reranker
 
 # Khởi tạo LawDocumentChunker
 law_chunker = LawDocumentChunker()
 law_chunker.llm_client = llm_client
 
+# Gán các client dùng chung cho ChannelManager.
+# Đây là một bước refactor quan trọng theo nguyên tắc Dependency Injection.
+# Thay vì mỗi Channel tự tạo client, chúng ta tạo một lần và chia sẻ,
+# giúp tối ưu tài nguyên và làm cho code dễ quản lý, dễ test hơn.
+channel_manager.set_shared_clients(
+    llm_client=llm_client,
+    sheets_client=sheets_client,
+    supabase_client=supabase_client,
+    embedding_client=embedding_client,
+    reranker=reranker,
+)
+
 logger.info("[STARTUP] Mount health router...")
 app.include_router(health_router)