diff --git a/app/api/routers/schedule_type.py b/app/api/routers/schedule_type.py index 6bbeb00..09d4cf3 100644 --- a/app/api/routers/schedule_type.py +++ b/app/api/routers/schedule_type.py @@ -9,7 +9,7 @@ from sqlalchemy.orm import Session from typing import List from app.core.config import get_db -from app.api.deps import get_current_user +from app.api.deps import get_current_user_or_apikey from app.models.models import User from app.models.agent import Agent from app.models.schedule_type import ScheduleType @@ -68,7 +68,7 @@ def _require_schedule_manage(db: Session, user: User) -> User: ) def list_schedule_types( db: Session = Depends(get_db), - current_user: User = Depends(get_current_user), + current_user: User = Depends(get_current_user_or_apikey), ): _require_schedule_read(db, current_user) return db.query(ScheduleType).all() @@ -82,7 +82,7 @@ def list_schedule_types( def create_schedule_type( payload: ScheduleTypeCreate, db: Session = Depends(get_db), - current_user: User = Depends(get_current_user), + current_user: User = Depends(get_current_user_or_apikey), ): _require_schedule_manage(db, current_user) @@ -112,7 +112,7 @@ def update_schedule_type( schedule_type_id: int, payload: ScheduleTypeUpdate, db: Session = Depends(get_db), - current_user: User = Depends(get_current_user), + current_user: User = Depends(get_current_user_or_apikey), ): _require_schedule_manage(db, current_user) @@ -135,7 +135,7 @@ def update_schedule_type( def delete_schedule_type( schedule_type_id: int, db: Session = Depends(get_db), - current_user: User = Depends(get_current_user), + current_user: User = Depends(get_current_user_or_apikey), ): _require_schedule_manage(db, current_user) @@ -192,7 +192,7 @@ def assign_schedule_type( agent_id: str, payload: AgentScheduleTypeAssign, db: Session = Depends(get_db), - current_user: User = Depends(get_current_user), + current_user: User = Depends(get_current_user_or_apikey), ): _require_schedule_manage(db, current_user)