Upload 22 files

backend/services/auto_discovery_service.py

@@ -8,6 +8,7 @@ Auto Discovery Service
 from __future__ import annotations
 
 import asyncio
+import inspect
 import json
 import logging
 import os
@@ -182,18 +183,59 @@ class AutoDiscoveryService:
         async with AsyncDDGS() as ddgs:
             for query in query_list:
                 try:
-                    async for entry in ddgs.atext(
+                    text_method = getattr(ddgs, "atext", None)
+                    if callable(text_method):
+                        async for entry in text_method(
+                            query,
+                            max_results=self.max_candidates_per_query,
+                        ):
+                            results.append(
+                                {
+                                    "query": query,
+                                    "title": entry.get("title", ""),
+                                    "url": entry.get("href") or entry.get("url") or "",
+                                    "snippet": entry.get("body", ""),
+                                }
+                            )
+                        continue
+
+                    text_method = getattr(ddgs, "text", None)
+                    if not callable(text_method):
+                        raise AttributeError("AsyncDDGS has no 'atext' or 'text' method")
+
+                    search_result = text_method(
                         query,
                         max_results=self.max_candidates_per_query,
-                    ):
-                        results.append(
-                            {
-                                "query": query,
-                                "title": entry.get("title", ""),
-                                "url": entry.get("href") or entry.get("url") or "",
-                                "snippet": entry.get("body", ""),
-                            }
+                    )
+
+                    if inspect.isawaitable(search_result):
+                        search_result = await search_result
+
+                    if hasattr(search_result, "__aiter__"):
+                        async for entry in search_result:
+                            results.append(
+                                {
+                                    "query": query,
+                                    "title": entry.get("title", ""),
+                                    "url": entry.get("href") or entry.get("url") or "",
+                                    "snippet": entry.get("body", ""),
+                                }
+                            )
+                    else:
+                        iterable = (
+                            search_result
+                            if isinstance(search_result, list)
+                            else list(search_result or [])
                         )
+                        for entry in iterable:
+                            results.append(
+                                {
+                                    "query": query,
+                                    "title": entry.get("title", ""),
+                                    "url": entry.get("href") or entry.get("url") or "",
+                                    "snippet": entry.get("body", ""),
+                                }
+                            )
                 except Exception as exc:  # pragma: no cover - وابسته به اینترنت
                     logger.warning("Failed to fetch results for query '%s': %s", query, exc)
 

provider_manager.py

@@ -31,15 +31,54 @@ class RotationStrategy(Enum):
     FASTEST_RESPONSE = "fastest_response"
 
 
-@dataclass
+@dataclass(init=False)
 class RateLimitInfo:
     """اطلاعات محدودیت نرخ"""
     requests_per_second: Optional[int] = None
     requests_per_minute: Optional[int] = None
     requests_per_hour: Optional[int] = None
     requests_per_day: Optional[int] = None
+    requests_per_week: Optional[int] = None
+    requests_per_month: Optional[int] = None
+    weight_per_minute: Optional[int] = None
     current_usage: int = 0
     reset_time: Optional[float] = None
+    extra_limits: Dict[str, Any] = field(default_factory=dict)
+
+    def __init__(
+        self,
+        requests_per_second: Optional[int] = None,
+        requests_per_minute: Optional[int] = None,
+        requests_per_hour: Optional[int] = None,
+        requests_per_day: Optional[int] = None,
+        requests_per_week: Optional[int] = None,
+        requests_per_month: Optional[int] = None,
+        weight_per_minute: Optional[int] = None,
+        current_usage: int = 0,
+        reset_time: Optional[float] = None,
+        **extra: Any,
+    ):
+        self.requests_per_second = requests_per_second
+        self.requests_per_minute = requests_per_minute
+        self.requests_per_hour = requests_per_hour
+        self.requests_per_day = requests_per_day
+        self.requests_per_week = requests_per_week
+        self.requests_per_month = requests_per_month
+        self.weight_per_minute = weight_per_minute
+        self.current_usage = current_usage
+        self.reset_time = reset_time
+        self.extra_limits = extra
+
+    @classmethod
+    def from_dict(cls, data: Optional[Dict[str, Any]]) -> "RateLimitInfo":
+        """ساخت نمونه از دیکشنری و مدیریت کلیدهای ناشناخته."""
+        if isinstance(data, cls):
+            return data
+
+        if not data:
+            return cls()
+
+        return cls(**data)
     
     def is_limited(self) -> bool:
         """بررسی محدودیت نرخ"""
@@ -90,7 +129,9 @@ class Provider:
     def __post_init__(self):
         """مقداردهی اولیه"""
         if isinstance(self.rate_limit, dict):
-            self.rate_limit = RateLimitInfo(**self.rate_limit)
+            self.rate_limit = RateLimitInfo.from_dict(self.rate_limit)
+        elif not isinstance(self.rate_limit, RateLimitInfo):
+            self.rate_limit = RateLimitInfo()
     
     @property
     def success_rate(self) -> float:
@@ -277,7 +318,7 @@ class ProviderManager:
             # بارگذاری ارائه‌دهندگان
             for provider_id, provider_data in config.get('providers', {}).items():
                 rate_limit_data = provider_data.get('rate_limit', {})
-                rate_limit = RateLimitInfo(**rate_limit_data)
+                rate_limit = RateLimitInfo.from_dict(rate_limit_data)
                 
                 provider = Provider(
                     provider_id=provider_id,