vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Suppress Pydantic serialization warnings from litellm ModelResponse (#4557)

Browse Source
This commit is contained in:
pedrohsdb
2026-01-26 16:41:03 -08:00
committed by GitHub
parent 72b9fe960f
commit b92447df35
1 changed files with 17 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
skyvern/forge/sdk/api/llm/api_handler_factory.py
View File

@@ -2,6 +2,7 @@ import copy
import dataclasses
import json
import time
import warnings
from asyncio import CancelledError
from typing import Any, AsyncIterator, Protocol, runtime_checkable
@@ -54,6 +55,19 @@ CHECK_USER_GOAL_PROMPT_NAMES = {"check-user-goal", "check-user-goal-with-termina
EXTRACT_ACTION_DEFAULT_THINKING_BUDGET = 512
def _safe_model_dump_json(response: ModelResponse, indent: int = 2) -> str:
"""
Call model_dump_json() while suppressing Pydantic serialization warnings.
litellm's ModelResponse has Union types (Choices | StreamingChoices) that cause
Pydantic to emit warnings during serialization. These are cosmetic and don't
affect the serialized output.
"""
with warnings.catch_warnings():
warnings.filterwarnings("ignore", category=UserWarning, message=".*Pydantic.*Serializ.*")
return response.model_dump_json(indent=indent)
@runtime_checkable
class RouterWithModelList(Protocol):
model_list: list[dict[str, Any]]
@@ -671,7 +685,7 @@ class LLMAPIHandlerFactory:
)
raise LLMProviderError(llm_key) from e
llm_response_json = response.model_dump_json(indent=2)
llm_response_json = _safe_model_dump_json(response)
if should_persist_llm_artifacts:
artifacts.append(
await app.ARTIFACT_MANAGER.prepare_llm_artifact(
@@ -1063,7 +1077,7 @@ class LLMAPIHandlerFactory:
)
raise LLMProviderError(llm_key) from e
llm_response_json = response.model_dump_json(indent=2)
llm_response_json = _safe_model_dump_json(response)
if should_persist_llm_artifacts:
artifacts.append(
await app.ARTIFACT_MANAGER.prepare_llm_artifact(
@@ -1442,7 +1456,7 @@ class LLMCaller:
LOG.exception("LLM request failed unexpectedly", llm_key=self.llm_key)
raise LLMProviderError(self.llm_key) from e
llm_response_json = response.model_dump_json(indent=2)
llm_response_json = _safe_model_dump_json(response)
if should_persist_llm_artifacts:
artifacts.append(
await app.ARTIFACT_MANAGER.prepare_llm_artifact(