format

Author: retanoj

litellm/google_genai/adapters/handler.py

@@ -38,8 +38,8 @@ def _prepare_completion_kwargs(
         completion_kwargs: Dict[str, Any] = dict(completion_request)
 
         # feed metadata for custom callback
-        if 'metadata' in extra_kwargs:
-            completion_kwargs['metadata'] = extra_kwargs['metadata']
+        if "metadata" in extra_kwargs:
+            completion_kwargs["metadata"] = extra_kwargs["metadata"]
 
         if stream:
             completion_kwargs["stream"] = stream

litellm/proxy/google_endpoints/endpoints.py

@@ -181,9 +181,11 @@ async def google_count_tokens(request: Request, model_name: str):
     from litellm.proxy._types import TokenCountRequest
 
     # Translate contents to openai format messages using the adapter
-    messages = (GoogleGenAIAdapter()
-                .translate_generate_content_to_completion(model_name, contents)
-                .get("messages", []))
+    messages = (
+        GoogleGenAIAdapter()
+        .translate_generate_content_to_completion(model_name, contents)
+        .get("messages", [])
+    )
 
     token_request = TokenCountRequest(
         model=model_name,
@@ -209,7 +211,7 @@ async def google_count_tokens(request: Request, model_name: str):
                 totalTokens=token_response.total_tokens or 0,
                 promptTokensDetails=[],
             )
-    
+
     #########################################################
     # Return the response in the well known format
     #########################################################