scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 114

Handle Pydantic v2 anyOf in field schema type extraction

Browse source
This commit is contained in:
Ryan Crabbe 2026-02-28 16:26:14 -08:00
parent d09675ba7e
commit c72e1d1065
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
litellm/proxy/management_endpoints/config_override_endpoints.py
View file

@ -86,6 +86,16 @@ def _get_current_env_values(env_var_mapping: Dict[str, str]) -> Dict[str, Any]:
return values
def _extract_field_type(field_info: Dict[str, Any]) -> str:
"""Extract the non-null type from a Pydantic v2 JSON schema field."""
if "type" in field_info:
return field_info["type"]
for option in field_info.get("anyOf", []):
if option.get("type") != "null":
return option.get("type", "string")
return "string"
def _build_field_schema(model_class: type) -> Dict[str, Any]:
"""Build field_schema dict from a Pydantic model for UI rendering."""
schema = TypeAdapter(model_class).json_schema(by_alias=True)
@ -93,7 +103,7 @@ def _build_field_schema(model_class: type) -> Dict[str, Any]:
for field_name, field_info in schema.get("properties", {}).items():
properties[field_name] = {
"description": field_info.get("description", ""),
"type": field_info.get("type", "string"),
"type": _extract_field_type(field_info),
}
return {
"description": schema.get("description", ""),