From bf325f6b2185cb42bc5d8f5713a64aecffb766d4 Mon Sep 17 00:00:00 2001 From: pewdiepie-archdaemon Date: Mon, 1 Jun 2026 10:47:11 +0900 Subject: [PATCH] Gate embedding routes behind admin auth --- routes/embedding_routes.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/routes/embedding_routes.py b/routes/embedding_routes.py index 67f78c8..ecdbfe0 100644 --- a/routes/embedding_routes.py +++ b/routes/embedding_routes.py @@ -6,8 +6,9 @@ import shutil import logging import asyncio from pathlib import Path -from fastapi import APIRouter, HTTPException, Form +from fastapi import APIRouter, HTTPException, Form, Depends from core.constants import BASE_DIR +from core.middleware import require_admin logger = logging.getLogger(__name__) @@ -97,7 +98,7 @@ def _save_custom_endpoint(data: dict): def setup_embedding_routes(): - router = APIRouter(prefix="/api/embeddings") + router = APIRouter(prefix="/api/embeddings", dependencies=[Depends(require_admin)]) @router.get("/models") def list_models():