Create api/endpoints.py

api/endpoints.py

@@ -0,0 +1,99 @@
+from fastapi import APIRouter, HTTPException, Query
+from typing import Optional
+import random
+
+from .openrouter import OpenRouterClient
+from .utils import get_openrouter_client
+
+router = APIRouter()
+
+# Available free models
+FREE_MODELS = [
+    "openai/gpt-3.5-turbo",
+    "anthropic/claude-instant-v1",
+    "google/palm-2-chat-bison",
+    "meta-llama/llama-2-13b-chat",
+    "mistralai/mistral-7b-instruct"
+]
+
+@router.post("/chat")
+async def chat_with_model(
+    message: str,
+    model: Optional[str] = None,
+    temperature: float = 0.7,
+    max_tokens: int = 1024
+):
+    """
+    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
+    """
+    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")
+    
+    try:
+        response = client.chat_completion(
+            model=model,
+            messages=[{"role": "user", "content": message}],
+            temperature=temperature,
+            max_tokens=max_tokens
+        )
+        return {
+            "model": model,
+            "response": response["choices"][0]["message"]["content"],
+            "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))