scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

test(vertex_ai): cover depth guard and non-dict nodes in oneOf conversion

Browse source
This commit is contained in:
WaleedH 2026-08-22 12:42:26 +03:00
parent aae2d8b66c
commit 48dc43132b
1 changed files with 26 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
tests/test_litellm/llms/vertex_ai/test_vertex_ai_common_utils.py
View file

@ -144,6 +144,32 @@ def test_build_vertex_schema_preserves_oneof_unions():
}
def test_oneof_conversion_with_excessive_nesting():
from litellm.llms.vertex_ai.common_utils import _convert_oneof_to_anyof
schema = {"type": "object", "properties": {}}
current = schema
for _ in range(DEFAULT_MAX_RECURSE_DEPTH + 1):
current["properties"] = {"nested": {"oneOf": [{"type": "string"}], "properties": {}}}
current = current["properties"]["nested"]
with pytest.raises(
ValueError,
match=f"Max depth of {DEFAULT_MAX_RECURSE_DEPTH} exceeded while processing schema. Please check the schema for excessive nesting.",
):
_convert_oneof_to_anyof(schema)
def test_oneof_conversion_ignores_non_dict_nodes():
from litellm.llms.vertex_ai.common_utils import _convert_oneof_to_anyof
schema = {"type": "object", "properties": {"tags": {"type": "array", "items": "string"}}}
_convert_oneof_to_anyof(schema)
assert schema == {"type": "object", "properties": {"tags": {"type": "array", "items": "string"}}}
def test_build_vertex_schema_converts_null_branch_inside_oneof():
from litellm.llms.vertex_ai.common_utils import _build_vertex_schema