fixed DI

backend/src/shared/di_container.py

@@ -1,7 +1,7 @@
-from dishka import Container, Provider, Scope, provide
+from dishka import Container, Provider, Scope, provide, make_async_container
 from fastapi import Request
 from sqlalchemy.ext.asyncio import AsyncSession
-from collections.abc import AsyncIterator
+from contextlib import asynccontextmanager
 
 from src.infrastructure.database.base import AsyncSessionLocal
 from src.infrastructure.repositories.postgresql.user_repository import PostgreSQLUserRepository
@@ -39,7 +39,8 @@ from src.application.use_cases.rag_use_cases import RAGUseCases
 
 class DatabaseProvider(Provider):
     @provide(scope=Scope.REQUEST)
-    async def get_db(self) -> AsyncIterator[AsyncSession]:
+    @asynccontextmanager
+    async def get_db(self) -> AsyncSession:
         async with AsyncSessionLocal() as session:
             try:
                 yield session
@@ -94,8 +95,6 @@ class ServiceProvider(Provider):
     def get_parser_service(self, ocr_service: YandexOCRService) -> DocumentParserService:
         return DocumentParserService(ocr_service)
 
-
-class VectorServiceProvider(Provider):
     @provide(scope=Scope.APP)
     def get_qdrant_client(self) -> QdrantClient:
         return QdrantClient(host=settings.QDRANT_HOST, port=settings.QDRANT_PORT)
@@ -133,12 +132,6 @@ class VectorServiceProvider(Provider):
             splitter=text_splitter,
         )
 
-class AuthProvider(Provider):
-    @provide(scope=Scope.REQUEST)
-    async def get_current_user(self, request: Request, user_repo: IUserRepository) -> User:
-        from src.presentation.middleware.auth_middleware import get_current_user
-        return await get_current_user(request, user_repo)
-
 
 class UseCaseProvider(Provider):
     @provide(scope=Scope.REQUEST)
@@ -196,12 +189,10 @@ class UseCaseProvider(Provider):
 
 
 def create_container() -> Container:
-    container = Container()
-    container.add_provider(DatabaseProvider())
-    container.add_provider(RepositoryProvider())
-    container.add_provider(ServiceProvider())
-    container.add_provider(AuthProvider())
-    container.add_provider(UseCaseProvider())
-    container.add_provider(VectorServiceProvider())
-    return container
+    return make_async_container(
+        DatabaseProvider(),
+        RepositoryProvider(),
+        ServiceProvider(),
+        UseCaseProvider()
+    )