update messenger instance

app/chat_channel.py

@@ -5,6 +5,7 @@ from .embedding import EmbeddingClient
 from .facebook import FacebookClient
 from .config import get_settings
 from app.message_processor import MessageProcessor
+from typing import Optional
 
 class ChatChannel:
     def __init__(self, page_id: str, channel_type: str = "facebook"):
@@ -18,9 +19,9 @@ class ChatChannel:
             base_url=settings.gemini_base_url,
             model=settings.gemini_models_list[0] if settings.gemini_models_list else "gemini-2.5-flash"
         )
-        self.reranker = Reranker()
+        self.facebook = FacebookClient(settings.facebook_app_secret, page_id=None, page_token=None, sender_id=None)
+        self.reranker = Reranker(facebook_client=self.facebook)
         self.embedder = EmbeddingClient()
-        self.facebook = FacebookClient(settings.facebook_app_secret)
         self.message_processor = MessageProcessor(self)
 
     def get_page_token(self, force_refresh=False):

app/facebook.py

@@ -9,7 +9,7 @@ from loguru import logger
 from .utils import timing_decorator_async, timing_decorator_sync
 
 class FacebookClient:
-    def __init__(self, app_secret: str):
+    def __init__(self, app_secret: str, page_id: Optional[str] = None, page_token: Optional[str] = None, sender_id: Optional[str] = None):
         """
         Khởi tạo FacebookClient với app_secret.
         Input: app_secret (str) - Facebook App Secret.
@@ -17,6 +17,22 @@ class FacebookClient:
         """
         self.app_secret = app_secret
         self._client = httpx.AsyncClient()
+        self.page_id = page_id
+        self.page_token = page_token
+        self.sender_id = sender_id
+
+    def update_context(self, page_id: Optional[str] = None, page_token: Optional[str] = None, sender_id: Optional[str] = None):
+        """
+        Cập nhật các thông tin context (page_id, page_token, sender_id) của client.
+        Input: page_id (str), page_token (str), sender_id (str)
+        Output: None
+        """
+        if page_id is not None:
+            self.page_id = page_id
+        if page_token is not None:
+            self.page_token = page_token
+        if sender_id is not None:
+            self.sender_id = sender_id
 
     @timing_decorator_async
     async def verify_webhook(self, token: str, challenge: str, verify_token: str) -> int:
@@ -48,21 +64,17 @@ class FacebookClient:
         return hmac.compare_digest(signature[7:], expected)
 
     @timing_decorator_async
-    async def send_message(self, page_access_token: str, recipient_id: str, message: str) -> Dict[str, Any]:
-        """
-        Gửi message tới user qua Facebook Messenger API.
-        Input: page_access_token (str), recipient_id (str), message (str)
-        Output: dict (response từ Facebook API)
-        """
-
+    async def send_message(self, page_access_token: Optional[str] = None, recipient_id: Optional[str] = None, message: str = "") -> Dict[str, Any]:
+        page_access_token = page_access_token or self.page_token
+        recipient_id = recipient_id or self.sender_id
+        if not page_access_token or not recipient_id:
+            raise ValueError("FacebookClient: page_access_token and recipient_id must not be None when sending a message.")
         logger.debug(f"Đang gửi tin nhắn đến Facebook Messenger....\n\t{message}")
         url = f"https://graph.facebook.com/v18.0/me/messages?access_token={page_access_token}"
-        
         payload = {
             "recipient": {"id": recipient_id},
             "message": {"text": message}
         }
-
         try:
             response = await self._client.post(url, json=payload)
             response.raise_for_status()

app/message_processor.py

@@ -109,6 +109,7 @@ class MessageProcessor:
 
         # Get page access token (cache)
         page_token = self.channel.get_page_token()
+        self.channel.facebook.update_context(page_id=page_id, page_token=page_token, sender_id=sender_id)
         if page_token:
             logger.info(f"[DEBUG] page_token: {page_token[:10]} ... {page_token[-10:]}")
         else:
@@ -117,13 +118,14 @@ class MessageProcessor:
             return
         # Gửi message Facebook, nếu lỗi token expired thì invalidate và thử lại một lần
         try:
-            await self.channel.facebook.send_message(page_token, message_data['sender_id'], "Ok, để mình check. Bạn chờ mình chút xíu nhé!")
+            await self.channel.facebook.send_message(message="Ok, để mình check. Bạn chờ mình chút xíu nhé!")
         except Exception as e:
             if "expired" in str(e).lower():
                 logger.warning("[FACEBOOK] Token expired, invalidate and refresh")
                 self.channel.invalidate_page_token()
                 page_token = self.channel.get_page_token(force_refresh=True)
-                await self.channel.facebook.send_message(page_token, message_data['sender_id'], "Ok, để mình check. Bạn chờ mình chút xíu nhé!")
+                self.channel.facebook.update_context(page_id=page_id, page_token=page_token, sender_id=sender_id)
+                await self.channel.facebook.send_message(message="Ok, để mình check. Bạn chờ mình chút xíu nhé!")
             else:
                 raise
 
@@ -151,7 +153,7 @@ class MessageProcessor:
                 hanh_vi_vi_pham = hanh_vi_vi_pham.replace(kw, "")
             hanh_vi_vi_pham = hanh_vi_vi_pham.strip()
         logger.info(f"[DEBUG] Phương tiện: {keywords} - Hành vi: {hanh_vi_vi_pham} - Mục đích: {muc_dich}")
-        await self.channel.facebook.send_message(page_token, sender_id, f"... đang tìm kiếm hành vi {hanh_vi_vi_pham} .....")
+        await self.channel.facebook.send_message(message=f"... đang tìm kiếm hành vi {hanh_vi_vi_pham} .....")
         # 4. Update lại conversation với thông tin đầy đủ
         update_kwargs = {
             'conversation_id': conv['conversation_id'],
@@ -175,7 +177,7 @@ class MessageProcessor:
         response = await self.process_business_logic(conv, page_token)
         logger.info(f"[DEBUG] Message history sau khi process: {conv}")
         # 6. Gửi response và cập nhật final state
-        await self.channel.facebook.send_message(page_token, sender_id, response)
+        await self.channel.facebook.send_message(message=response)
         if hasattr(self.channel, 'sheets'):
             await loop.run_in_executor(None, lambda: self.channel.sheets.log_conversation(**conv))
         return
@@ -250,7 +252,7 @@ class MessageProcessor:
     async def format_search_results(self, question: str, matches: List[Dict[str, Any]], page_token: str, sender_id: str) -> str:
         if not matches:
             return "Không tìm thấy kết quả phù hợp."
-        await self.channel.facebook.send_message(page_token, sender_id, f"Tôi tìm thấy một số quy định rồi .....")
+        await self.channel.facebook.send_message(message=f"... đã tìm thấy một số quy định .....")
         try:
             reranked = await self.channel.reranker.rerank(question, matches, top_k=5)
             if reranked:
@@ -316,7 +318,7 @@ class MessageProcessor:
             "\n\nHãy trả lời ngắn gọn, dễ hiểu, trích dẫn rõ ràng thông tin từ các đoạn luật nếu cần."
             f"\n\nCâu hỏi của người dùng: {question}\n"
         )
-        await self.channel.facebook.send_message(page_token, sender_id, f"Được rồi, để tôi tóm tắt lại nhé .....")
+        await self.channel.facebook.send_message(message=f"... Được rồi, để tôi tóm tắt lại nhé .....")
         try:
             answer = await self.channel.llm.generate_text(prompt)
             if answer and answer.strip():

app/reranker.py

@@ -5,7 +5,7 @@ from loguru import logger
 import asyncio
 
 class Reranker:
-    def __init__(self):
+    def __init__(self, facebook_client=None):
         settings = get_settings()
         self.provider = getattr(settings, 'rerank_provider', settings.llm_provider)
         self.model = getattr(settings, 'rerank_model', settings.llm_model)
@@ -17,6 +17,7 @@ class Reranker:
         #     self.client = CohereClient(settings.cohere_api_key, model=self.model)
         else:
             raise NotImplementedError(f"Rerank provider {self.provider} not supported yet.")
+        self.facebook_client = facebook_client
 
     async def _score_doc(self, query: str, doc: Dict) -> Dict:
         """
@@ -84,6 +85,12 @@ class Reranker:
                 scored.append(result)
             
             logger.info(f"[RERANK] Completed batch {i//batch_size + 1}, processed {len(scored)} docs so far")
+            # Send Facebook message after each batch
+            if self.facebook_client:
+                try:
+                    await self.facebook_client.send_message(message=f"... tôi đang sắp xếp lại chút ...")
+                except Exception as e:
+                    logger.error(f"[RERANK][FACEBOOK] Error sending batch message: {e}")
         
         # Sort theo score và trả về top_k
         scored = sorted(scored, key=lambda x: x['rerank_score'], reverse=True)