init project

rag_core/embedder.py

@@ -1,11 +1,22 @@
 import requests
+import time
+import logging
 from rag_core.utils import log_timed
 
 @log_timed("gửi API tạo embedding")
-def get_embedding(text: str):
-    response = requests.post(
-        "https://vietcat-phobertnode.hf.space/embed",
-        json={"text": text},
-        timeout=10
-    )
-    return response.json()["embedding"]
+def get_embedding(text: str, retries: int = 3):
+    for i in range(retries):
+        try:
+            response = requests.post(
+                "https://vietcat-phobertnode.hf.space/embed",
+                json={"text": text},
+                timeout=30  # Tăng từ 10 lên 30 giây
+            )
+            response.raise_for_status()  # Nếu không 200 -> raise exception
+            return response.json()["embedding"]
+        except requests.exceptions.RequestException as e:
+            logging.warning(f"Lỗi embedding (lần {i+1}/{retries}): {e}")
+            if i < retries - 1:
+                time.sleep(2)  # Đợi 2s rồi thử lại
+            else:
+                raise RuntimeError(f"Không thể lấy embedding sau {retries} lần thử.")