fix: some minor issues (#342)

Co-authored-by: Anton <[email protected]>

Author: Sobes76rus

taskiq/depends/progress_tracker.py

@@ -7,12 +7,6 @@
 from taskiq.compat import IS_PYDANTIC2
 from taskiq.context import Context
 
-if IS_PYDANTIC2:
-    from pydantic import BaseModel as GenericModel
-else:
-    from pydantic.generics import GenericModel  # type: ignore[no-redef]
-
-
 _ProgressType = TypeVar("_ProgressType")
 
 
@@ -25,15 +19,26 @@ class TaskState(str, enum.Enum):
     RETRY = "RETRY"
 
 
-class TaskProgress(GenericModel, Generic[_ProgressType]):
+if IS_PYDANTIC2:
+    from pydantic import BaseModel, ConfigDict
+
+    class _TaskProgressConfig(BaseModel):
+        model_config = ConfigDict(arbitrary_types_allowed=True)
+
+else:
+    from pydantic.generics import GenericModel
+
+    class _TaskProgressConfig(GenericModel):  # type: ignore[no-redef]
+        class Config:
+            arbitrary_types_allowed = True
+
+
+class TaskProgress(_TaskProgressConfig, Generic[_ProgressType]):
     """Progress of task execution."""
 
     state: Union[TaskState, str]
     meta: Optional[_ProgressType]
 
-    class Config:
-        arbitrary_types_allowed = True
-
 
 class ProgressTracker(Generic[_ProgressType]):
     """Task's dependency to set progress."""

taskiq/receiver/receiver.py

@@ -401,8 +401,10 @@ def task_cb(task: "asyncio.Task[Any]") -> None:
             self.sem_prefetch.release()
             message = await queue.get()
             if message is QUEUE_DONE:
-                logger.info("Waiting for running tasks to complete.")
-                await asyncio.wait(tasks, timeout=self.wait_tasks_timeout)
+                # asyncio.wait will throw an error if there is nothing to wait for
+                if tasks:
+                    logger.info("Waiting for running tasks to complete.")
+                    await asyncio.wait(tasks, timeout=self.wait_tasks_timeout)
                 break
 
             task = asyncio.create_task(