Update backend_models.py

backend_models.py

@@ -3,21 +3,28 @@ from openai import OpenAI
 
 def get_inference_client(model_id: str, provider: str = "auto"):
     """
-    Ép hệ thống sử dụng endpoint miễn phí của Hugging Face
-    cho tất cả các model trong danh sách.
+    Ép tất cả các model trong danh sách (Kimi, Gemma, Qwen...) 
+    phải gọi vào endpoint miễn phí của Hugging Face.
     """
-    # Xóa bỏ hoàn toàn header X-HF-Bill-To để tránh lỗi 402
+    # Nếu nó đòi Novita, ta cứ phớt lờ và đưa nó sang endpoint miễn phí
     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:
-    # Trả về ID sạch, không có hậu tố :novita hay :cerebras
-    return model_id.split(":")[0]
+    """
+    Hàm này cực kỳ quan trọng:
+    Nó biến 'moonshotai/Kimi-K2.6:novita' thành 'moonshotai/Kimi-K2.6'
+    'google/gemma-4-31B:fastest' thành 'google/gemma-4-31B'
+    """
+    # Xóa hậu tố provider để API của Hugging Face nhận diện được model
+    base_id = model_id.split(":")[0]
+    return base_id
 
 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()
+