diff --git a/litellm/google_genai/main.py b/litellm/google_genai/main.py index db1a11341fb..25d7de9f710 100644 --- a/litellm/google_genai/main.py +++ b/litellm/google_genai/main.py @@ -199,10 +199,7 @@ class GenerateContentHelper: explicit_vertex_location: Final = VertexBase.explicit_vertex_ai_location( MappingProxyType( - { - key: getattr(litellm_params, key, None) - for key in ("vertex_location", "vertex_ai_location") - } + {key: getattr(litellm_params, key, None) for key in ("vertex_location", "vertex_ai_location")} ) ) vertex_location_params: Final = ( diff --git a/tests/unit/google_genai/test_google_genai_main.py b/tests/unit/google_genai/test_google_genai_main.py index 275264e2542..31e2f56092c 100644 --- a/tests/unit/google_genai/test_google_genai_main.py +++ b/tests/unit/google_genai/test_google_genai_main.py @@ -262,7 +262,7 @@ async def test_native_fields_forwarded_on_async_stream(): def _generate_content_logging_obj(call_id: str): - from datetime import datetime + from datetime import datetime, timezone from litellm.litellm_core_utils.litellm_logging import Logging @@ -271,7 +271,7 @@ def _generate_content_logging_obj(call_id: str): messages=[], stream=False, call_type="generate_content", - start_time=datetime(2025, 1, 1), + start_time=datetime(2025, 1, 1, tzinfo=timezone.utc), litellm_call_id=call_id, function_id=call_id, )