diff --git a/apps/api/app/dependencies.py b/apps/api/app/dependencies.py index e0c0d57ac..a1d45a039 100644 --- a/apps/api/app/dependencies.py +++ b/apps/api/app/dependencies.py @@ -1,3 +1,5 @@ +from typing import Generator + import redis from app.config import settings from fastapi import Depends @@ -40,7 +42,8 @@ from packages.ports.workspace_repository import WorkspaceRepository _engine, _SessionLocal = build_session_factory(settings.DATABASE_URL) -def get_db_session(): +def get_db_session() -> Generator[Session, None, None]: + """获取数据库会话""" session: Session = _SessionLocal() try: yield session @@ -120,13 +123,13 @@ def get_workspace_invitation_repository( return SQLAlchemyWorkspaceInvitationRepository(session) -def get_auth_session_store(): - if not settings.ENABLE_REDIS_SESSIONS: +def get_auth_session_store() -> NoopSessionStore | SessionStore: + if not settings.ENABLE_REDIS_SESSION: return NoopSessionStore() return SessionStore(redis_client=redis.from_url(settings.REDIS_URL, decode_responses=True)) -def get_auth_email_service(): +def get_auth_email_service() -> NoopEmailService: if not settings.ENABLE_EMAIL_DELIVERY: return NoopEmailService() return get_email_service(