diff --git a/eval_protocol/__init__.py b/eval_protocol/__init__.py index da1760de..84e282af 100644 --- a/eval_protocol/__init__.py +++ b/eval_protocol/__init__.py @@ -13,6 +13,10 @@ import warnings from typing import TYPE_CHECKING +import litellm + +litellm.disable_add_transform_inline_image_block = True + warnings.filterwarnings("default", category=DeprecationWarning, module="eval_protocol") # Eager imports for symbols that conflict with module names - ONLY when pytest is running. diff --git a/eval_protocol/mcp/execution/policy.py b/eval_protocol/mcp/execution/policy.py index 5f0a2eb6..0b4aac4e 100644 --- a/eval_protocol/mcp/execution/policy.py +++ b/eval_protocol/mcp/execution/policy.py @@ -22,9 +22,6 @@ logger = logging.getLogger(__name__) -# Disable LiteLLM's document inlining feature that appends #transform=inline -# to base64 image URLs, which breaks base64 decoding -litellm.disable_add_transform_inline_image_block = True class LiteLLMPolicy(LLMBasePolicy): """