db api online

model_service_api.py

@@ -1,7 +1,7 @@
 import aiohttp
 import logging
 from typing import List, Dict, Optional
-from fastapi import FastAPI, HTTPException, Query
+from fastapi import FastAPI, HTTPException, Query, Path
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 from datetime import datetime, timedelta
@@ -71,6 +71,12 @@ cache = {
     "ttl_minutes": 15  # Cache for 15 minutes
 }
 
+# Supported categories
+SUPPORTED_CATEGORIES = [
+    "chat", "reasoning", "vision", "coding", "lightweight", 
+    "roleplay", "marketing", "programming", "image", "text"
+]
+
 class OpenRouterClient:
     """Client for interacting with OpenRouter API"""
     
@@ -111,7 +117,7 @@ class OpenRouterClient:
             params["category"] = category
         
         try:
-            logger.info("Fetching models from OpenRouter API...")
+            logger.info(f"Fetching models from OpenRouter API{f' with category: {category}' if category else ''}...")
             async with self.session.get(url, params=params) as response:
                 if response.status != 200:
                     logger.error(f"OpenRouter API returned status {response.status}")
@@ -194,11 +200,55 @@ async def root():
         "endpoints": {
             "free_models": "/api/free-models",
             "free_models_by_category": "/api/free-models?category=programming",
+            "category_route": "/api/category/{category}/free-models",
+            "free_model_names": "/api/free-models/names",
+            "specific_model": "/api/free-models/{model_id}",
+            "supported_categories": "/api/categories",
             "health": "/health"
         },
         "description": "This API fetches models from OpenRouter and filters for free models (prompt=0, completion=0)"
     }
 
+@app.get("/api/categories")
+async def get_supported_categories():
+    """Get list of supported categories"""
+    return {
+        "success": True,
+        "message": "List of supported categories",
+        "categories": SUPPORTED_CATEGORIES,
+        "usage": "Use these categories with /api/category/{category}/free-models"
+    }
+
+@app.get("/api/category/{category}/free-models", response_model=FreeModelsResponse)
+async def get_free_models_by_category(
+    category: str = Path(..., description="Category to filter by (e.g., chat, coding, vision)")
+):
+    """Get free models filtered by specific category"""
+    
+    # Validate category
+    if category.lower() not in [cat.lower() for cat in SUPPORTED_CATEGORIES]:
+        raise HTTPException(
+            status_code=400,
+            detail=f"Unsupported category '{category}'. Supported categories: {', '.join(SUPPORTED_CATEGORIES)}"
+        )
+    
+    try:
+        free_models = await get_free_models_data(category=category.lower())
+        
+        return FreeModelsResponse(
+            success=True,
+            message=f"Successfully retrieved {len(free_models)} free models in category '{category}'",
+            count=len(free_models),
+            models=free_models,
+            filtered_at=datetime.now().isoformat()
+        )
+    
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error(f"Error in get_free_models_by_category: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
+
 @app.get("/api/free-models", response_model=FreeModelsResponse)
 async def get_free_models(
     category: Optional[str] = Query(None, description="Filter by category (e.g., 'programming', 'chat')")
@@ -304,7 +354,6 @@ async def health_check():
             }
         )
 
-
 @app.get("/ping")
 async def ping():
     """Ping endpoint to check API status"""

requirements.txt

@@ -2,3 +2,4 @@ aiohttp==3.11.13
 uvicorn==0.34.0
 fastapi==0.115.5
 pydantic==2.10.3
+dotenv==0.9.9