Files
Imago/app/schemas/auth.py
T
bruno 9bb440707c
CI / Lint & Format (push) Failing after 11s
CI / Tests (push) Has been skipped
CI / Security Scan (push) Failing after 8s
CI / Docker Build (push) Has been skipped
AI Vision: multi-provider config, model testing, per-client AI preferences
Backend:
- Add ai_provider + ai_model columns to APIClient model
- New admin_ai router: GET /admin/api/ai/status + POST /admin/api/ai/test
- Update ai_vision.py: provider_override + model_override support
- Update pipeline: load client AI prefs for image processing
- ClientUpdate/ClientResponse schemas include ai_provider/ai_model

Frontend:
- New AIConfigSection component: provider list, model dropdowns,
  test button with live results (no page refresh needed)
- AI types (ai.types.ts) + API (api/ai.ts)
- APIClient types extended with ai_provider/ai_model
2026-06-25 10:30:50 -04:00

92 lines
3.3 KiB
Python

"""
Schémas Pydantic — authentification et gestion des clients API
"""
from datetime import datetime
from typing import List, Optional, Set
from pydantic import BaseModel, ConfigDict, Field, field_validator
from app.models.client import ClientPlan
# Scopes valides — toute valeur hors de cet ensemble est rejetée
VALID_SCOPES: Set[str] = {
"images:read",
"images:write",
"images:delete",
"ai:use",
"admin",
}
# ─────────────────────────────────────────────────────────────
# Requêtes
# ─────────────────────────────────────────────────────────────
class ClientCreate(BaseModel):
"""Créer un nouveau client API."""
name: str = Field(..., min_length=1, max_length=256, description="Nom de l'application cliente")
scopes: List[str] = Field(
default=["images:read", "images:write", "images:delete"],
description="Permissions accordées",
)
plan: ClientPlan = Field(default=ClientPlan.FREE, description="Plan tarifaire")
@field_validator("scopes")
@classmethod
def validate_scopes(cls, v):
invalid = set(v) - VALID_SCOPES
if invalid:
raise ValueError(
f"Scopes invalides : {invalid}. "
f"Scopes valides : {sorted(VALID_SCOPES)}"
)
return v
class ClientUpdate(BaseModel):
"""Modifier un client API existant."""
name: Optional[str] = Field(None, min_length=1, max_length=256)
scopes: Optional[List[str]] = None
plan: Optional[ClientPlan] = None
is_active: Optional[bool] = None
ai_provider: Optional[str] = Field(None, description="Fournisseur AI préféré (openrouter, gemini)")
ai_model: Optional[str] = Field(None, description="Modèle AI préféré")
# ─────────────────────────────────────────────────────────────
# Réponses
# ─────────────────────────────────────────────────────────────
class ClientResponse(BaseModel):
"""Réponse de base pour un client API."""
model_config = ConfigDict(from_attributes=True)
id: str
name: str
scopes: List[str]
plan: ClientPlan
is_active: bool
ai_provider: Optional[str] = None
ai_model: Optional[str] = None
created_at: Optional[datetime] = None
updated_at: Optional[datetime] = None
class ClientCreateResponse(ClientResponse):
"""
Réponse après création d'un client.
La clé API est retournée EN CLAIR une seule fois.
"""
api_key: str = Field(
...,
description="Clé API en clair — stockez-la, elle ne sera plus jamais affichée",
)
class KeyRotateResponse(BaseModel):
"""Réponse après rotation de la clé API."""
id: str
api_key: str = Field(
...,
description="Nouvelle clé API en clair — stockez-la, elle ne sera plus jamais affichée",
)
message: str = "Clé API régénérée avec succès — l'ancienne clé est désormais invalide"