Update api/endpoints.py

api/endpoints.py

@@ -1,13 +1,14 @@
-from fastapi import APIRouter, HTTPException, Query
+from fastapi import APIRouter, HTTPException, Query, status
 from typing import Optional
 import random
+import logging
 
 from .openrouter import OpenRouterClient
 from .utils import get_openrouter_client
 
 router = APIRouter()
+logger = logging.getLogger(__name__)
 
-# Available free models
 FREE_MODELS = [
     "openai/gpt-3.5-turbo",
     "anthropic/claude-instant-v1",
@@ -20,26 +21,30 @@ FREE_MODELS = [
 async def chat_with_model(
     message: str,
     model: Optional[str] = None,
-    temperature: float = 0.7,
-    max_tokens: int = 1024
+    temperature: float = Query(0.7, ge=0, le=2),
+    max_tokens: int = Query(1024, ge=1, le=4096)
 ):
     """
     Chat with a specified model or let the system choose one
-    
-    Parameters:
-    - message: The user's input message
-    - model: Optional model identifier (default: random)
-    - temperature: Creativity parameter (0-2)
-    - max_tokens: Maximum length of response
     """
+    if not message.strip():
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail="Message cannot be empty"
+        )
+    
     client = get_openrouter_client()
     
     if not model:
         model = random.choice(FREE_MODELS)
     elif model not in FREE_MODELS:
-        raise HTTPException(status_code=400, detail="Model not available in free tier")
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail=f"Model not available in free tier. Available models: {', '.join(FREE_MODELS)}"
+        )
     
     try:
+        logger.info(f"Processing chat request with model: {model}")
         response = client.chat_completion(
             model=model,
             messages=[{"role": "user", "content": message}],
@@ -52,48 +57,8 @@ async def chat_with_model(
             "usage": response.get("usage", {})
         }
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-
-@router.get("/models")
-async def list_available_models():
-    """
-    List all available free models
-    """
-    return {
-        "models": FREE_MODELS,
-        "count": len(FREE_MODELS),
-        "description": "These models are available in the free tier"
-    }
-
-@router.post("/smart-chat")
-async def smart_chat_with_model(message: str):
-    """
-    Smart chat endpoint that selects the best model based on the input
-    
-    Parameters:
-    - message: The user's input message
-    """
-    client = get_openrouter_client()
-    
-    # Model selection logic
-    if any(word in message.lower() for word in ["code", "program", "python", "javascript"]):
-        model = "openai/gpt-3.5-turbo"
-    elif any(word in message.lower() for word in ["story", "creative", "write", "poem"]):
-        model = "anthropic/claude-instant-v1"
-    elif any(word in message.lower() for word in ["fact", "science", "history"]):
-        model = "google/palm-2-chat-bison"
-    else:
-        model = random.choice(FREE_MODELS)
-    
-    try:
-        response = client.chat_completion(
-            model=model,
-            messages=[{"role": "user", "content": message}]
-        )
-        return {
-            "model": model,
-            "response": response["choices"][0]["message"]["content"],
-            "reason": f"Selected {model} based on input content"
-        }
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        logger.error(f"Error processing chat request: {str(e)}")
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail="Error processing your request"
+        )