Update backend_models.py

backend_models.py

@@ -2,37 +2,20 @@ import os
 from openai import OpenAI
 
 def get_inference_client(model_id: str, provider: str = "auto"):
-    """
-    Ép buộc sử dụng endpoint Serverless miễn phí của Hugging Face.
-    """
+    # Ép buộc sử dụng endpoint miễn phí của Hugging Face cho mọi model
     return OpenAI(
         base_url="https://api-inference.huggingface.co/v1/",
         api_key=os.getenv("HF_TOKEN")
     )
 
 def get_real_model_id(model_id: str) -> str:
-    """
-    Ánh xạ các model trả phí trong menu sang model miễn phí tương đương.
-    """
-    mapping = {
-        "Kimi-K2.6": "meta-llama/Llama-3.1-8B-Instruct",
-        "Gemma-4-31B-it": "google/gemma-2-9b-it",
-        "Qwen3.5-397B-A17B": "Qwen/Qwen2.5-7B-Instruct",
-        "DeepSeek-V3": "Qwen/Qwen2.5-Coder-7B-Instruct"
-    }
-    
-    # Lấy tên model chính (bỏ qua hậu tố :novita, :cerebras)
-    clean_id = model_id.split(":")[0]
-    
-    # Nếu có trong mapping, chuyển sang model miễn phí
-    for key, free_model in mapping.items():
-        if key in clean_id:
-            return free_model
-            
-    return "meta-llama/Llama-3.1-8B-Instruct" # Mặc định an toàn
+    # Luôn luôn trả về model Llama-3.1-8B-Instruct (Miễn phí 100%)
+    # Dù menu có hiển thị Kimi, Qwen... thì thực tế hệ thống sẽ dùng Llama 3.1
+    return "meta-llama/Llama-3.1-8B-Instruct"
 
 def is_native_sdk_model(model_id: str) -> bool:
     return False
 
 def is_mistral_model(model_id: str) -> bool:
-    return "mistral" in model_id.lower()
+    return False
+