Update src/routes/analytics_routes.py

src/routes/analytics_routes.py

@@ -18,6 +18,7 @@ from src.managers.chat_manager import ChatManager
 
 from typing import Any, Dict, List, Optional
 from src.utils.logger import Logger
+from src.utils.model_registry import MODEL_TIERS, get_model_tier as get_model_tier_from_registry
 
 # Initialize logger with console logging disabled
 logger = Logger("analytics_routes", see_time=True, console_log=False)
@@ -30,7 +31,7 @@ if os.getenv("ENVIRONMENT") == "production":
     logger.disable_logging()
 
 # Initialize chat manager
-chat_manager = ChatManager()
+chat_manager = ChatManager(db_url=os.getenv("DATABASE_URL"))
 
 # API Key security
 ADMIN_API_KEY = os.getenv("ADMIN_API_KEY", "default-admin-key-change-me")
@@ -64,74 +65,10 @@ async def verify_admin_api_key(
 active_dashboard_connections = set()
 active_user_connections = set()
 
-# Model tier definitions
-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"
-        ]
-    }
-}
-
 # Helper function to determine model tier
 def get_model_tier(model_name):
     """Determine which tier a model belongs to based on its name"""
-    model_name = model_name.lower()
-    
-    for tier_id, tier_info in MODEL_TIERS.items():
-        # Check if the model name matches or starts with any of the models in this tier
-        if any(model_name == model.lower() or model_name.startswith(model.lower()) 
-               for model in tier_info["models"]):
-            return tier_id
-    
-    # Default to tier 1 if no match is found
-    return "tier1"
+    return get_model_tier_from_registry(model_name)
 
 # Helper function to parse period parameter
 def get_date_range(period: str):