Fixed no-decorated context managers.

Signed-off-by: Pavel Kirilin <win10@list.ru>

Author: s3rius

taskiq_dependencies/dependency.py

@@ -1,11 +1,10 @@
 import inspect
 import uuid
+from contextlib import _AsyncGeneratorContextManager, _GeneratorContextManager
 from typing import (
     Any,
-    AsyncContextManager,
     AsyncGenerator,
     Callable,
-    ContextManager,
     Coroutine,
     Dict,
     Generator,
@@ -21,7 +20,7 @@
 
 @overload
 def Depends(
-    dependency: Optional[Callable[..., ContextManager[_T]]] = None,
+    dependency: Optional[Callable[..., "_GeneratorContextManager[_T]"]] = None,
     *,
     use_cache: bool = True,
     kwargs: Optional[Dict[str, Any]] = None,
@@ -31,7 +30,7 @@ def Depends(
 
 @overload
 def Depends(
-    dependency: Optional[Callable[..., AsyncContextManager[_T]]] = None,
+    dependency: Optional[Callable[..., "_AsyncGeneratorContextManager[_T]"]] = None,
     *,
     use_cache: bool = True,
     kwargs: Optional[Dict[str, Any]] = None,

taskiq_dependencies/utils.py

@@ -1,6 +1,6 @@
 import inspect
 import sys
-from contextlib import AsyncContextDecorator, ContextDecorator
+from contextlib import _AsyncGeneratorContextManager, _GeneratorContextManager
 from typing import Any, AsyncContextManager, ContextManager, Optional
 
 if sys.version_info >= (3, 10):
@@ -39,7 +39,7 @@ def iscontextmanager(obj: Any) -> TypeGuard[ContextManager[Any]]:
     :param obj: object to check.
     :return: bool that indicates whether the object is a context manager or not.
     """
-    return issubclass(obj.__class__, ContextDecorator)
+    return issubclass(obj.__class__, _GeneratorContextManager)
 
 
 def isasynccontextmanager(obj: Any) -> TypeGuard[AsyncContextManager[Any]]:
@@ -49,4 +49,4 @@ def isasynccontextmanager(obj: Any) -> TypeGuard[AsyncContextManager[Any]]:
     :param obj: object to check.
     :return: bool that indicates whether the object is a async context manager or not.
     """
-    return issubclass(obj.__class__, AsyncContextDecorator)
+    return issubclass(obj.__class__, _AsyncGeneratorContextManager)