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

fix(gemini): append empty choice to model response when there are no candidates

Browse source
This commit is contained in:
Sian Cao 2025-04-21 18:40:42 +08:00
parent ebfff975d4
commit c958bbe706
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py
View file

@ -949,6 +949,8 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig):
) = self._process_candidates(
_candidates, model_response, litellm_params
)
else:
model_response.choices.append(litellm.Choices())
usage = self._calculate_usage(completion_response=completion_response)
setattr(model_response, "usage", usage)