Fix: Pydantic 2 handle differently the errors

aleph-im · Nov 12, 2024 · c1bdd96 · c1bdd96
1 parent 6acdc3c
commit c1bdd96
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/aleph/vm/orchestrator/tasks.py b/src/aleph/vm/orchestrator/tasks.py
@@ -8,6 +8,7 @@
 
 import aiohttp
 import pydantic
+from pydantic import ValidationError
 from aiohttp import web
 from aleph_message.models import (
     AlephMessage,
@@ -73,10 +74,10 @@ async def subscribe_via_ws(url) -> AsyncIterable[AlephMessage]:
 
                     try:
                         yield parse_message(data)
-                    except pydantic.error_wrappers.ValidationError as error:
+                    except pydantic.ValidationError as error:
                         item_hash = data.get("item_hash", "ITEM_HASH_NOT_FOUND")
                         logger.warning(
-                            f"Invalid Aleph message: {item_hash} \n  {error.model_dump_json()}\n  {error.raw_errors}",
+                            f"Invalid Aleph message: {item_hash} \n  {error.errors}",
                             exc_info=False,
                         )
                         continue