add AI and generation routers, models, and OpenRouter service integration with tests

Co-authored-by: Copilot <copilot@github.com>
2026-04-27 18:12:53 +02:00
parent 3ee4ed7e7f
commit 05309f26b4
7 changed files with 826 additions and 0 deletions
@@ -0,0 +1,63 @@
+"""AI router: model listing and chat completions via OpenRouter."""
+from fastapi import APIRouter, Depends, HTTPException, status
+
+from backend.app.dependencies import get_current_user
+from backend.app.models.ai import ChatRequest, ChatResponse, ModelInfo
+from backend.app.services import openrouter
+
+router = APIRouter(prefix="/ai", tags=["ai"])
+
+
+@router.get("/models", response_model=list[ModelInfo])
+async def get_models(_: dict = Depends(get_current_user)) -> list[ModelInfo]:
+    """List available AI models from OpenRouter."""
+    try:
+        raw = await openrouter.list_models()
+    except Exception as exc:
+        raise HTTPException(
+            status_code=status.HTTP_502_BAD_GATEWAY,
+            detail=f"OpenRouter error: {exc}",
+        )
+    return [
+        ModelInfo(
+            id=m.get("id", ""),
+            name=m.get("name", m.get("id", "")),
+            context_length=m.get("context_length"),
+            pricing=m.get("pricing"),
+        )
+        for m in raw
+    ]
+
+
+@router.post("/chat", response_model=ChatResponse)
+async def chat(
+    body: ChatRequest,
+    _: dict = Depends(get_current_user),
+) -> ChatResponse:
+    """Send a chat completion request through OpenRouter."""
+    try:
+        result = await openrouter.chat_completion(
+            model=body.model,
+            messages=[m.model_dump() for m in body.messages],
+            temperature=body.temperature,
+            max_tokens=body.max_tokens,
+        )
+    except Exception as exc:
+        raise HTTPException(
+            status_code=status.HTTP_502_BAD_GATEWAY,
+            detail=f"OpenRouter error: {exc}",
+        )
+
+    try:
+        choice = result["choices"][0]
+        return ChatResponse(
+            id=result["id"],
+            model=result.get("model", body.model),
+            content=choice["message"]["content"],
+            usage=result.get("usage"),
+        )
+    except (KeyError, IndexError) as exc:
+        raise HTTPException(
+            status_code=status.HTTP_502_BAD_GATEWAY,
+            detail=f"Unexpected response format from OpenRouter: {exc}",
+        )