Updated API method.

taskiq-python · s3rius · Apr 19, 2025 · Apr 19, 2025 · Apr 19, 2025 · Apr 19, 2025
commit e51558378745ca9dd25be325d0123796bb79adac
diff --git a/taskiq/api/scheduler.py b/taskiq/api/scheduler.py
@@ -1,10 +1,14 @@
+from datetime import timedelta
+from typing import Optional
+
 from taskiq.cli.scheduler.run import run_scheduler_loop
 from taskiq.scheduler.scheduler import TaskiqScheduler
 
 
 async def run_scheduler_task(
     scheduler: TaskiqScheduler,
     run_startup: bool = False,
+    interval: Optional[timedelta] = None,
 ) -> None:
     """
     Run scheduler task.
@@ -20,4 +24,4 @@ async def run_scheduler_task(
     if run_startup:
         await scheduler.startup()
     while True:
-        await run_scheduler_loop(scheduler)
+        await run_scheduler_loop(scheduler, interval)
diff --git a/taskiq/cli/scheduler/run.py b/taskiq/cli/scheduler/run.py
@@ -154,6 +154,7 @@ async def run_scheduler_loop(  # noqa: C901
     and runs tasks when needed.
 
     :param scheduler: current scheduler.
+    :param interval: interval to check for schedule updates.
     """
     loop = asyncio.get_event_loop()
     running_schedules: Dict[str, asyncio.Task[Any]] = {}