From d486245428e1501b0fc83b4e8ce0276643f869d8 Mon Sep 17 00:00:00 2001 From: NiceAesth Date: Fri, 23 Feb 2024 15:30:42 +0200 Subject: [PATCH] chore: rename to type adapter instead --- pomice/models/payloads.py | 4 ++-- pomice/pool.py | 8 +++++--- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/pomice/models/payloads.py b/pomice/models/payloads.py index 92f1d7e..f445b8a 100644 --- a/pomice/models/payloads.py +++ b/pomice/models/payloads.py @@ -31,5 +31,5 @@ class ResumePayloadV4(BaseModel): resuming: bool = True -_ResumePayloadType = Union[ResumePayloadV3, ResumePayloadV4] -ResumePayloadType = lambda **kwargs: TypeAdapter(_ResumePayloadType).validate_python(kwargs) +ResumePayloadType = Union[ResumePayloadV3, ResumePayloadV4] +ResumePayloadTypeAdapter = lambda **kwargs: TypeAdapter(ResumePayloadType).validate_python(kwargs) diff --git a/pomice/pool.py b/pomice/pool.py index e52ffa0..6f21da8 100644 --- a/pomice/pool.py +++ b/pomice/pool.py @@ -42,7 +42,7 @@ from .routeplanner import RoutePlanner from .utils import ExponentialBackoff from .utils import NodeStats from .utils import Ping -from pomice.models.payloads import ResumePayloadType +from pomice.models.payloads import ResumePayloadTypeAdapter from pomice.models.payloads import ResumePayloadV4 from pomice.models.version import LavalinkVersion @@ -302,8 +302,10 @@ class Node: if not self._resume_key: return - data = ResumePayloadType( - version=self._version, timeout=self._resume_timeout, resuming_key=self._resume_key, + data = ResumePayloadTypeAdapter( + version=self._version, + timeout=self._resume_timeout, + resuming_key=self._resume_key, ).model_dump() if isinstance(data, ResumePayloadV4):