fix: compatibility with Taskiq 0.13.0

pyproject.toml

@@ -29,18 +29,18 @@ repository = "https://github.com/taskiq-python/taskiq-nats"
 keywords = ["taskiq", "tasks", "distributed", "async", "nats", "result_backend"]
 requires-python = ">=3.10,<4"
 dependencies = [
-    "taskiq>=0.11.20,<1",
+    "taskiq>=0.13.0",
     "nats-py>=2.2.0",
 ]
 
 [dependency-groups]
 dev = [
     "pre-commit>=4.4.0",
     # lint
-    "ruff>=0.14.5",
-    "black>=25.11.0",
+    "ruff>=0.16.9",
+    "black>=26.5.1",
     # type check
-    "mypy>=1.18.2",
+    "mypy>=2.3.1",
     # tests
     "pytest>=9.0.1",
     "pytest-cov>=7.0.0",

taskiq_nats/broker.py

@@ -105,7 +105,7 @@ class BaseJetStreamBroker(
     be sure that messages are delivered to the workers.
     """
 
-    def __init__(
+    def __init__(  # noqa: PLR0917
         self,
         servers: str | list[str],
         subject: str = "taskiq_tasks",

taskiq_nats/result_backend.py

@@ -7,7 +7,6 @@
 from nats.js.object_store import ObjectStore
 from taskiq import AsyncResultBackend, ResultGetError
 from taskiq.abc.serializer import TaskiqSerializer
-from taskiq.compat import model_dump, model_validate
 from taskiq.result import TaskiqResult
 from taskiq.serializers import PickleSerializer
 
@@ -73,7 +72,7 @@ async def set_result(self, task_id: str, result: TaskiqResult[_ReturnType]) -> N
         """
         await self.object_store.put(
             name=task_id,
-            data=self.serializer.dumpb(model_dump(result)),
+            data=self.serializer.dumpb(result.model_dump(mode="json")),
         )
 
     async def is_result_ready(self, task_id: str) -> bool:
@@ -114,8 +113,9 @@ async def get_result(
                 name=task_id,
             )
 
-        taskiq_result: TaskiqResult[_ReturnType] = model_validate(
-            TaskiqResult[_ReturnType],
+        taskiq_result: TaskiqResult[_ReturnType] = TaskiqResult[
+            _ReturnType
+        ].model_validate(
             self.serializer.loadb(result.data),  # type: ignore[arg-type]
         )
 

taskiq_nats/schedule_source.py

@@ -8,7 +8,6 @@
 from nats.js.kv import KeyValue
 from taskiq import ScheduledTask, ScheduleSource
 from taskiq.abc.serializer import TaskiqSerializer
-from taskiq.compat import model_dump, model_validate
 from taskiq.serializers import PickleSerializer
 
 log = logging.getLogger(__name__)
@@ -85,7 +84,7 @@ async def add_schedule(self, schedule: ScheduledTask) -> None:
         """
         await self.kv.put(
             f"{self.prefix}.{schedule.schedule_id}",
-            self.serializer.dumpb(model_dump(schedule)),
+            self.serializer.dumpb(schedule.model_dump(mode="json")),
         )
 
     async def get_schedules(self) -> list[ScheduledTask]:
@@ -102,7 +101,7 @@ async def get_schedules(self) -> list[ScheduledTask]:
             return []
 
         return [
-            model_validate(ScheduledTask, self.serializer.loadb(schedule.value))
+            ScheduledTask.model_validate(self.serializer.loadb(schedule.value))
             for schedule in schedules
             if schedule and schedule.value
         ]