mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-10-23 10:20:01 +00:00
feat(core): Upgrade pydantic to 2.x (#1428)
This commit is contained in:
@@ -9,7 +9,7 @@ from typing import Dict, Optional, Type
|
||||
class Serializable(ABC):
|
||||
"""The serializable abstract class."""
|
||||
|
||||
serializer: Optional["Serializer"] = None
|
||||
_serializer: Optional["Serializer"] = None
|
||||
|
||||
@abstractmethod
|
||||
def to_dict(self) -> Dict:
|
||||
@@ -21,11 +21,12 @@ class Serializable(ABC):
|
||||
Returns:
|
||||
bytes: The byte array after serialization
|
||||
"""
|
||||
if self.serializer is None:
|
||||
if self._serializer is None:
|
||||
raise ValueError(
|
||||
"Serializer is not set. Please set the serializer before serialization."
|
||||
"Serializer is not set. Please set the serializer before "
|
||||
"serialization."
|
||||
)
|
||||
return self.serializer.serialize(self)
|
||||
return self._serializer.serialize(self)
|
||||
|
||||
def set_serializer(self, serializer: "Serializer") -> None:
|
||||
"""Set the serializer for current serializable object.
|
||||
@@ -33,7 +34,7 @@ class Serializable(ABC):
|
||||
Args:
|
||||
serializer (Serializer): The serializer to set
|
||||
"""
|
||||
self.serializer = serializer
|
||||
self._serializer = serializer
|
||||
|
||||
|
||||
class Serializer(ABC):
|
||||
|
Reference in New Issue
Block a user