fix(proxy): catch callback metadata validation errors

This commit is contained in:
user
2026-04-29 20:13:34 -07:00
parent 22c01adeb2
commit 1ebb192cbe
2 changed files with 4 additions and 2 deletions
+2 -1
View File
@@ -6,6 +6,7 @@ from collections import OrderedDict
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
from fastapi import Request
from pydantic import ValidationError as PydanticValidationError
from starlette.datastructures import Headers
import litellm
@@ -238,7 +239,7 @@ def _get_validated_callback_metadata(
) -> Optional[AddTeamCallback]:
try:
return AddTeamCallback(**item)
except ValueError as e:
except (PydanticValidationError, ValueError) as e:
verbose_proxy_logger.warning(
"Ignoring invalid %s callback metadata: %s",
source,
@@ -7,6 +7,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
import pytest
from fastapi import Request
from pydantic import ValidationError as PydanticValidationError
from starlette.datastructures import Headers
import litellm
@@ -1275,7 +1276,7 @@ def test_get_dynamic_logging_metadata_with_arize_team_logging():
def test_add_team_callback_rejects_env_reference():
with pytest.raises(ValueError) as exc_info:
with pytest.raises(PydanticValidationError) as exc_info:
AddTeamCallback(
callback_name="langfuse",
callback_type="success",