deps.py 803 B

12345678910111213141516171819
  1. from typing import Annotated
  2. from fastapi import Depends
  3. from sqlmodel.ext.asyncio.session import AsyncSession
  4. from gpustack.schemas.users import User
  5. from gpustack.api.auth import get_admin_user, get_current_user
  6. from gpustack.api.tenant import (
  7. TenantContext,
  8. get_tenant_context,
  9. require_platform_admin,
  10. )
  11. from gpustack.server.db import get_session
  12. from gpustack.schemas.common import ListParams
  13. SessionDep = Annotated[AsyncSession, Depends(get_session)]
  14. ListParamsDep = Annotated[ListParams, Depends(ListParams)]
  15. CurrentUserDep = Annotated[User, Depends(get_current_user)]
  16. CurrentAdminUserDep = Annotated[User, Depends(get_admin_user)]
  17. TenantContextDep = Annotated[TenantContext, Depends(get_tenant_context)]
  18. PlatformAdminDep = Annotated[TenantContext, Depends(require_platform_admin)]