refactor: use pydantic v2 API

ApeWorX · Nov 24, 2024 · 35167ae · 35167ae
1 parent e949e04
commit 35167ae
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/silverback/main.py b/silverback/main.py
@@ -125,7 +125,9 @@ def __init__(self, settings: Settings | None = None):
         ):
             settings.NEW_BLOCK_TIMEOUT = int(timedelta(days=1).total_seconds())
 
-        settings_str = "\n  ".join(f'{key}="{val}"' for key, val in settings.dict().items() if val)
+        settings_str = "\n  ".join(
+            f'{key}="{val}"' for key, val in settings.model_dump().items() if val
+        )
         logger.info(f"Loading Silverback Bot with settings:\n  {settings_str}")
 
         self.broker = settings.get_broker()

diff --git a/silverback/state.py b/silverback/state.py
@@ -46,7 +46,7 @@ async def init(self, bot_id: SilverbackID) -> StateSnapshot | None:
         self.state_backup_file = data_folder / "state.json"
 
         return (
-            StateSnapshot.parse_file(self.state_backup_file)
+            StateSnapshot.model_validate_json(self.state_backup_file.read_text())
             if self.state_backup_file.exists()
             else None
         )