Fix relative import errors in engines package for HuggingFace deployment

engines/base.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from dataclasses import dataclass
 from typing import Any, Dict, Protocol
 
-from ..service.tasks import TaskRecord
+from service.tasks import TaskRecord
 
 
 @dataclass(frozen=True)

engines/mediacrawler.py

@@ -22,7 +22,7 @@ except Exception as e:
     sync_playwright = None
     _PLAYWRIGHT_IMPORT_ERROR = str(e)
 
-from ..service.tasks import TaskRecord
+from service.tasks import TaskRecord
 from .base import EngineRunOutput
 
 try:
@@ -819,7 +819,7 @@ class MediaCrawlerEngine:
         return unique
 
     def _pick_proxy(self, attempt: int) -> str | None:
-        from ..service.proxy_pool import proxy_pool
+        from service.proxy_pool import proxy_pool
         dynamic_proxy = proxy_pool.get_random_proxy()
         if dynamic_proxy:
             return dynamic_proxy
@@ -846,7 +846,7 @@ class MediaCrawlerEngine:
                         self._humanize(page)
                         res = job(page)
                         if proxy:
-                            from ..service.proxy_pool import proxy_pool
+                            from service.proxy_pool import proxy_pool
 
                             proxy_pool.report_success(proxy)
                         return res
@@ -860,7 +860,7 @@ class MediaCrawlerEngine:
                 if proxy:
                     if reason in ("timeout", "proxy_failed", "rate"):
                         try:
-                            from ..service.proxy_pool import proxy_pool
+                            from service.proxy_pool import proxy_pool
 
                             proxy_pool.report_failure(proxy, reason)
                         except Exception:

engines/spider_xhs.py

@@ -10,7 +10,7 @@ from apis.xhs_pc_apis import XHS_Apis
 from xhs_utils.data_util import handle_note_info, handle_user_info
 from xhs_utils.response_guard import get_dict, get_list
 
-from ..service.tasks import TaskRecord
+from service.tasks import TaskRecord
 from .base import EngineRunOutput
 
 
@@ -117,7 +117,7 @@ class SpiderXHSEngine:
             explicit_proxy = payload.get("proxy") or payload.get("proxies")
             proxy_value = explicit_proxy or self.proxy
             if not proxy_value:
-                from ..service.proxy_pool import proxy_pool
+                from service.proxy_pool import proxy_pool
 
                 proxy_value = proxy_pool.get_random_proxy()
             if proxy_value:
@@ -156,7 +156,7 @@ class SpiderXHSEngine:
 
         try:
             if proxy_used:
-                from ..service.proxy_pool import proxy_pool
+                from service.proxy_pool import proxy_pool
 
                 if ok:
                     proxy_pool.report_success(proxy_used)