init project

rag_core/embedder.py

@@ -4,20 +4,21 @@ import logging
 from rag_core.utils import log_timed
 
 @log_timed("gửi API tạo embedding")
-def get_embedding(text: str, retries: int = 3):
-    baseTimeout=30
+def get_embedding(text: str, retries: int = 3, base_timeout: int = 30):
     for i in range(retries):
         try:
+            current_timeout = base_timeout * (i + 1)
             response = requests.post(
                 "https://vietcat-phobertnode.hf.space/embed",
                 json={"text": text},
-                timeout=baseTimeout*(i+1)
+                timeout=current_timeout
             )
             response.raise_for_status()
             return response.json()["embedding"]
         except requests.exceptions.RequestException as e:
-            logging.warning(f"Lỗi embedding (lần {i+1}/{retries}): {e}")
+            logging.warning(f"Lỗi embedding (lần {i+1}/{retries}, timeout={current_timeout}s): {e}")
             if i < retries - 1:
                 time.sleep(2)
             else:
                 raise
+