Formatted to fix the linter

Author: Kamal-Moha

pydantic_ai_slim/pydantic_ai/models/google.py

@@ -91,7 +91,7 @@
         'Please install `google-genai` to use the Google model, '
         'you can use the `google` optional group — `pip install "pydantic-ai-slim[google]"`'
     ) from _import_error
-    
+
 LatestGoogleModelNames = Literal[
     'gemini-2.0-flash',
     'gemini-2.0-flash-lite',
@@ -575,7 +575,7 @@ async def _map_user_prompt(self, part: UserPromptPart) -> list[PartDict]:
                                 media_type=item.media_type,
                                 identifier=item.identifier,
                             )
-                        )                    
+                        )
                     else:
                         inline_data_dict: BlobDict = {'data': item.data, 'mime_type': item.media_type}
                         part_dict: PartDict = {'inline_data': inline_data_dict}
@@ -607,7 +607,7 @@ async def _map_user_prompt(self, part: UserPromptPart) -> list[PartDict]:
                     if item.vendor_metadata:  # pragma: no branch
                         part_dict['video_metadata'] = cast(VideoMetadataDict, item.vendor_metadata)
                     content.append(part_dict)
-                
+
                 elif isinstance(item, FileUrl):
                     if item.force_download or (
                         # google-gla does not support passing file urls directly, except for youtube videos
@@ -626,7 +626,7 @@ async def _map_user_prompt(self, part: UserPromptPart) -> list[PartDict]:
                         content.append({'file_data': file_data_dict})  # pragma: lax no cover
                 else:
                     assert_never(item)
-            
+
             return content
 
     @staticmethod
@@ -641,15 +641,15 @@ def _is_text_like_media_type(media_type: str) -> bool:
         )
 
     @staticmethod
-    def _inline_text_file_part(text: str, *, media_type: str, identifier: str) -> ChatCompletionContentPartTextParam:
+    def _inline_text_file_part(text: str, *, media_type: str, identifier: str):
         text = '\n'.join(
             [
                 f'-----BEGIN FILE id="{identifier}" type="{media_type}"-----',
                 text,
                 f'-----END FILE id="{identifier}"-----',
             ]
         )
-        return {"text": text}
+        return {'text': text}
 
     def _map_response_schema(self, o: OutputObjectDefinition) -> dict[str, Any]:
         response_schema = o.json_schema.copy()

tests/models/test_google.py

@@ -778,17 +778,19 @@ async def test_google_model_text_document_url_input(allow_model_requests: None,
         'The main content of the TXT file is an explanation of the placeholder name "John Doe" (and related variations) and its usage in legal contexts, popular culture, and other situations where the identity of a person is unknown or needs to be withheld. The document also includes the purpose of the file and other file type information.\n'
     )
 
+
 async def test_google_model_json_document_url_input(allow_model_requests: None, google_provider: GoogleProvider):
     m = GoogleModel('gemini-2.5-pro', provider=google_provider)
     agent = Agent(m, system_prompt='You are a helpful chatbot.')
 
     json_document_url = DocumentUrl(url='https://kamalscraping-collab.github.io/sample-data/sample_transcript.json')
-    
+
     result = await agent.run(['What is the main content of this document?', json_document_url])
     assert result.output == snapshot(
-        "Based on the JSON data provided, the document contains the log of a conversation between a user and an AI assistant.\n"
+        'Based on the JSON data provided, the document contains the log of a conversation between a user and an AI assistant.\n'
     )
 
+
 async def test_google_model_text_as_binary_content_input(allow_model_requests: None, google_provider: GoogleProvider):
     m = GoogleModel('gemini-2.0-flash', provider=google_provider)
     agent = Agent(m, system_prompt='You are a helpful chatbot.')