mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-24 14:28:01 +00:00
fix(proxy): catch callback metadata validation errors
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user