phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(vertex_httpx.py): fix json schema call to pass in response_mime_type=="application/json"

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-21 15:22:22 -07:00
parent a583b95d85
commit 664c40a4c7
2 changed files with 14 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/tests/test_amazing_vertex_completion.py
View file

@ -1558,6 +1558,16 @@ async def test_gemini_pro_json_schema_args_sent_httpx_openai_schema(
"response_schema"
in mock_call.call_args.kwargs["json"]["generationConfig"]
)
assert (
"response_mime_type"
in mock_call.call_args.kwargs["json"]["generationConfig"]
)
assert (
mock_call.call_args.kwargs["json"]["generationConfig"][
"response_mime_type"
]
== "application/json"
)
else:
assert (
"response_schema"