From fb9f3ee24b9e8c4d8fc96dcd8b12f75bee727549 Mon Sep 17 00:00:00 2001 From: basil-k-aji-dev <70605804+basil-k-aji-dev@users.noreply.github.com> Date: Tue, 22 Sep 2026 20:51:10 +0530 Subject: [PATCH] fix(test): hoist the ModelResponseIterator import so ruff can resolve it F821: the annotation referenced a name imported inside the function, so it was undefined at module scope where ruff evaluates it. --- .../gemini/test_vertex_and_google_ai_studio_gemini.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/unit/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py b/tests/unit/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py index 70ce0dc24b0..c01353a3568 100644 --- a/tests/unit/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py +++ b/tests/unit/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py @@ -16,6 +16,7 @@ from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler from litellm.llms.gemini.chat.transformation import GoogleAIStudioGeminiConfig from litellm.llms.vertex_ai.common_utils import VertexAIError from litellm.llms.vertex_ai.gemini.vertex_and_google_ai_studio_gemini import ( + ModelResponseIterator, VertexGeminiConfig, ) from litellm.types.llms.vertex_ai import GeminiFinishReason, UsageMetadata @@ -6376,9 +6377,7 @@ _GROUNDING_URIS: Final = ( ) -def _gemini_stream_iterator() -> "ModelResponseIterator": - from litellm.llms.vertex_ai.gemini.vertex_and_google_ai_studio_gemini import ModelResponseIterator - +def _gemini_stream_iterator() -> ModelResponseIterator: logging_obj = MagicMock() logging_obj.optional_params = {} return ModelResponseIterator(streaming_response=iter([]), sync_stream=True, logging_obj=logging_obj)