tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-07 00:24:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revert "fix(vertex): use module-level import for ModelResponseStream instead of type: ignore"

Browse Source 
This reverts commit 7e4f07c45d.
This commit is contained in:
Chesars
2026-02-22 09:43:08 -03:00
parent 7e4f07c45d
commit ca5fd1ecec
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/llms/vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py
+3 -4
View File
@@ -80,7 +80,6 @@ from litellm.types.utils import (
ChatCompletionTokenLogprob,
ChoiceLogprobs,
CompletionTokensDetailsWrapper,
ModelResponseStream,
PromptTokensDetailsWrapper,
TopLogprob,
Usage,
@@ -1932,7 +1931,7 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig):
@staticmethod
def _process_candidates( # noqa: PLR0915
_candidates: List[Candidates],
model_response: Union[ModelResponse, ModelResponseStream],
model_response: Union[ModelResponse, "ModelResponseStream"],
standard_optional_params: dict,
cumulative_tool_call_index: int = 0,
) -> Tuple[List[dict], List[dict], List, List, int]:
@@ -2094,7 +2093,7 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig):
chat_completion_logprobs=chat_completion_logprobs,
image_response=image_response,
)
model_response.choices.append(choice)
model_response.choices.append(choice) # type: ignore[arg-type]
elif isinstance(model_response, ModelResponse):
choice = litellm.Choices(
finish_reason=VertexGeminiConfig._check_finish_reason(
@@ -2105,7 +2104,7 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig):
logprobs=chat_completion_logprobs,
enhancements=None,
)
model_response.choices.append(choice)
model_response.choices.append(choice) # type: ignore[arg-type]
return (
grounding_metadata,