Added graph ref to the ParamInfo.

Author: s3rius

taskiq_dependencies/ctx.py

@@ -89,7 +89,11 @@ def traverse_deps(  # noqa: C901
                 # If the user want to get ParamInfo,
                 # we get declaration of the current dependency.
                 if subdep.dependency == ParamInfo:
-                    kwargs[subdep.param_name] = ParamInfo(dep.param_name, dep.signature)
+                    kwargs[subdep.param_name] = ParamInfo(
+                        dep.param_name,
+                        self.graph,
+                        dep.signature,
+                    )
                     continue
                 if subdep.use_cache:
                     # If this dependency can be calculated, using cache,

taskiq_dependencies/utils.py

@@ -1,13 +1,16 @@
 import inspect
 import sys
 from contextlib import _AsyncGeneratorContextManager, _GeneratorContextManager
-from typing import Any, AsyncContextManager, ContextManager, Optional
+from typing import TYPE_CHECKING, Any, AsyncContextManager, ContextManager, Optional
 
 if sys.version_info >= (3, 10):
     from typing import TypeGuard
 else:
     from typing_extensions import TypeGuard
 
+if TYPE_CHECKING:
+    from taskiq_dependencies.graph import DependencyGraph
+
 
 class ParamInfo:
     """
@@ -23,9 +26,11 @@ class ParamInfo:
     def __init__(
         self,
         name: str,
+        graph: "DependencyGraph",
         signature: Optional[inspect.Parameter] = None,
     ) -> None:
         self.name = name
+        self.graph = graph
         self.definition = signature
 
     def __repr__(self) -> str: