Refactor settings.py: Simplify API configuration for free tier and remove account type checks

config/settings.py

@@ -42,12 +42,13 @@ if not HF_TOKEN:
 
 # API Configuration
 API_CONFIG = {
-    "inference_endpoint": os.getenv("HF_INFERENCE_ENDPOINT", "https://api-inference.huggingface.co"),
+    "inference_endpoint": "https://api-inference.huggingface.co",
     "token": HF_TOKEN,
-    "is_paid_tier": True,
+    "is_paid_tier": False,  # Принудительно устанавливаем бесплатный режим
     "timeout": 15,
+    "max_retries": 1,
     "headers": {
-        "X-Use-Cache": "false",
+        "X-Use-Cache": "true",  # Включаем кэширование для бесплатного тарифа
         "Content-Type": "application/json",
         "Authorization": f"Bearer {HF_TOKEN}"  
     }
@@ -55,47 +56,15 @@ API_CONFIG = {
 
 def check_account_type():
     """
-    Check HuggingFace account type and adjust settings accordingly
+    Simplified account check for free tier
     Returns:
         tuple: (is_pro: bool, account_type: str)
     """
-    try:
-        api = HfApi(token=HF_TOKEN)
-        user_info = api.whoami()
-        account_type = user_info.get("type", "free")
-        plan = user_info.get("plan", "free")
-        
-        is_pro = any([
-            plan.lower() == "pro",
-            account_type.lower() == "pro",
-            "pro" in str(user_info).lower()
-        ])
-        
-        return is_pro, account_type
-    except Exception as e:
-        logger.warning(f"Failed to check account type: {e}")
-        return False, "free"
+    return False, "free"
 
-# Check account type and adjust settings
-IS_PRO_ACCOUNT, ACCOUNT_TYPE = check_account_type()
-
-# Adjust API configuration based on account type
-API_CONFIG.update({
-    "is_paid_tier": IS_PRO_ACCOUNT,
-    "timeout": 30 if IS_PRO_ACCOUNT else 15,
-    "max_retries": 3 if IS_PRO_ACCOUNT else 1
-})
-
-# Define available models based on account type
-PRO_ONLY_MODELS = ["llama-7b", "mistral-7b"]
-FREE_MODELS = ["zephyr-7b"]
-
-# Filter available models based on account type
-if not IS_PRO_ACCOUNT:
-    # Remove pro-only models from MODELS dict
-    MODELS = {k: v for k, v in MODELS.items() if k not in PRO_ONLY_MODELS}
-    # Set default model to a free one
-    DEFAULT_MODEL = "zephyr-7b"
+# Устанавливаем базовые настройки для free tier
+IS_PRO_ACCOUNT, ACCOUNT_TYPE = False, "free"
+DEFAULT_MODEL = "zephyr-7b"  # Устанавливаем дефолтную бесплатную модель
 
 # Dataset configuration
 DATASET_ID = "Rulga/status-law-knowledge-base"