nambrogi
/
pomice
mirror of https://github.com/cloudwithax/pomice.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore: rename to type adapter instead

This commit is contained in:
NiceAesth 2024-02-23 15:30:42 +02:00
parent ac49c81609
commit d486245428
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pomice/models/payloads.py
View File

@ -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)

8
pomice/pool.py
View File

@ -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):