small signature update

chatbot_model.py

@@ -112,7 +112,10 @@ class EncoderModel(tf.keras.Model):
         
 class RetrievalChatbot(DeviceAwareModel):
     """Retrieval-based chatbot using pretrained embeddings and FAISS for similarity search."""
-    def __init__(self, config: ChatbotConfig, dialogues: List[dict] = [], device: str = None, strategy=None, reranker: Optional[CrossEncoderReranker] = None, summarizer: Optional[Summarizer] = None):
+    def __init__(self, config: ChatbotConfig, dialogues: List[dict] = [], device: str = None,
+                 strategy=None, reranker: Optional[CrossEncoderReranker] = None,
+                 summarizer: Optional[Summarizer] = None
+                 ):
         self.config = config
         self.strategy = strategy
         self.setup_device(device)
@@ -991,6 +994,16 @@ class RetrievalChatbot(DeviceAwareModel):
         conversation_parts.append(f"{self.special_tokens['user']} {query}")
         return "\n".join(conversation_parts)
 
+
+
+
+
+
+
+
+
+
+
 # def prepare_dataset(
 #     self, 
 #     dialogues: List[dict],