From 7536854a118342822bbe3332a4ce3c8cf6212037 Mon Sep 17 00:00:00 2001 From: Emmanuel Ferdman Date: Wed, 7 May 2025 09:13:39 -0700 Subject: [PATCH] Resolve Pydantic deprecation warnings Signed-off-by: Emmanuel Ferdman --- src/codegate/types/anthropic/_generators.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/codegate/types/anthropic/_generators.py b/src/codegate/types/anthropic/_generators.py index 31362edd..a7a5a7fd 100644 --- a/src/codegate/types/anthropic/_generators.py +++ b/src/codegate/types/anthropic/_generators.py @@ -28,7 +28,7 @@ async def stream_generator(stream: AsyncIterator[Any]) -> AsyncIterator[str]: try: async for chunk in stream: try: - body = chunk.json(exclude_unset=True) + body = chunk.model_dump_json(exclude_unset=True) except Exception as e: logger.error("failed serializing payload", exc_info=e) err = MessageError( @@ -38,7 +38,7 @@ async def stream_generator(stream: AsyncIterator[Any]) -> AsyncIterator[str]: message=str(e), ), ) - body = err.json(exclude_unset=True) + body = err.model_dump_json(exclude_unset=True) yield f"event: error\ndata: {body}\n\n" data = f"event: {chunk.type}\ndata: {body}\n\n" @@ -56,7 +56,7 @@ async def stream_generator(stream: AsyncIterator[Any]) -> AsyncIterator[str]: message=str(e), ), ) - body = err.json(exclude_unset=True) + body = err.model_dump_json(exclude_unset=True) yield f"event: error\ndata: {body}\n\n"