Added depdencies override as in fastapi. (#228)

Author: s3rius

pyproject.toml

@@ -31,7 +31,7 @@ typing-extensions = ">=3.10.0.0"
 pydantic = ">=1.0,<=3.0"
 importlib-metadata = "*"
 pycron = "^3.0.0"
-taskiq_dependencies = "^1"
+taskiq_dependencies = ">=1.3.1,<2"
 anyio = ">=3"
 packaging = ">=19"
 # For prometheus metrics

taskiq/abc/broker.py

@@ -112,6 +112,7 @@ def __init__(
         ] = defaultdict(list)
         self.state = TaskiqState()
         self.custom_dependency_context: Dict[Any, Any] = {}
+        self.dependency_overrides: Dict[Any, Any] = {}
         # True only if broker runs in worker process.
         self.is_worker_process: bool = False
         # True only if broker runs in scheduler process.

taskiq/receiver/receiver.py

@@ -207,7 +207,10 @@ async def run_task(  # noqa: C901, PLR0912, PLR0915
                     TaskiqState: self.broker.state,
                 },
             )
-            dep_ctx = dependency_graph.async_ctx(broker_ctx)
+            dep_ctx = dependency_graph.async_ctx(
+                broker_ctx,
+                self.broker.dependency_overrides or None,
+            )
             # Resolve all function's dependencies.
 
         # Start a timer.