Update src/utils/model_tier.py

src/utils/model_tier.py

@@ -1,60 +1,5 @@
-MODEL_TIERS = {
-    "tier1": {
-        "name": "Basic",
-        "credits": 1,
-        "models": [
-            "llama3-8b-8192",
-            "llama-3.2-1b-preview",
-            "llama-3.2-3b-preview",
-            "llama-3.2-11b-text-preview",
-            "llama-3.2-11b-vision-preview",
-            "llama-3.1-8b-instant",
-            "mixtral-8x7b-32768",
-            "gemma-7b-it",
-            "gemma2-9b-it",
-            "llama3-groq-8b-8192-tool-use-preview"
-        ]
-    },
-    "tier2": {
-        "name": "Standard",
-        "credits": 3,
-        "models": [
-            "gpt-4o-mini",
-            "o1-mini",
-            "o3-mini"
-        ]
-    },
-    "tier3": {
-        "name": "Premium",
-        "credits": 5,
-        "models": [
-            "gpt-3.5-turbo",
-            "gpt-4.5-preview",
-            "gpt-4",
-            "gpt-4o",
-            "o1",
-            "claude-3-opus-latest",
-            "claude-3-7-sonnet-latest",
-            "claude-3-5-sonnet-latest",
-            "claude-3-5-haiku-latest",
-            "deepseek-r1-distill-qwen-32b",
-            "deepseek-r1-distill-llama-70b",
-            "llama-3.3-70b-versatile",
-            "llama-3.3-70b-specdec",
-            "llama2-70b-4096",
-            "llama-3.2-90b-text-preview",
-            "llama-3.2-90b-vision-preview",
-            "llama3-70b-8192",
-            "llama-3.1-70b-versatile",
-            "llama-3.1-405b-reasoning",
-            "llama3-groq-70b-8192-tool-use-preview"
-        ]
-    }
-}
-
+from src.utils.model_registry import MODEL_TIERS, get_model_tier as get_model_tier_from_registry
 
+# For backward compatibility, keep the get_model_tier function exposed
 def get_model_tier(model_name):
-    for tier, tier_info in MODEL_TIERS.items():
-        if model_name in tier_info["models"]:
-            return tier
-    return "tier1"
+    return get_model_tier_from_registry(model_name)