scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

style(google_genai): ruff format and tz-aware fixed timestamp

Browse source
This commit is contained in:
Ankit Jha 2026-10-01 21:28:14 +05:30
parent 849baa8f8f
commit 162d60cf38
No known key found for this signature in database
2 changed files with 3 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/google_genai/main.py
View file

@ -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 = (

4
tests/unit/google_genai/test_google_genai_main.py
View file

@ -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,
)