codereview fix

Author: Guiners

genai/live/live_audiogen_with_txt.py

@@ -24,9 +24,15 @@ async def generate_content() -> None:
     import numpy as np
     import scipy.io.wavfile as wavfile
     from google import genai
-    from google.genai.types import (Content, LiveConnectConfig, Modality, Part,
-                                    PrebuiltVoiceConfig, SpeechConfig,
-                                    VoiceConfig)
+    from google.genai.types import (
+        Content,
+        LiveConnectConfig,
+        Modality,
+        Part,
+        PrebuiltVoiceConfig,
+        SpeechConfig,
+        VoiceConfig,
+    )
 
     client = genai.Client()
     model = "gemini-2.0-flash-live-preview-04-09"

genai/live/live_code_exec_with_txt.py

@@ -18,8 +18,14 @@
 async def generate_content() -> list[str]:
     # [START googlegenaisdk_live_code_exec_with_txt]
     from google import genai
-    from google.genai.types import (Content, LiveConnectConfig, Modality, Part,
-                                    Tool, ToolCodeExecution)
+    from google.genai.types import (
+        Content,
+        LiveConnectConfig,
+        Modality,
+        Part,
+        Tool,
+        ToolCodeExecution,
+    )
 
     client = genai.Client()
     model_id = "gemini-2.0-flash-live-preview-04-09"

genai/live/live_func_call_with_txt.py

@@ -20,9 +20,15 @@
 async def generate_content() -> list[FunctionResponse]:
     # [START googlegenaisdk_live_func_call_with_txt]
     from google import genai
-    from google.genai.types import (Content, FunctionDeclaration,
-                                    FunctionResponse, LiveConnectConfig,
-                                    Modality, Part, Tool)
+    from google.genai.types import (
+        Content,
+        FunctionDeclaration,
+        FunctionResponse,
+        LiveConnectConfig,
+        Modality,
+        Part,
+        Tool,
+    )
 
     client = genai.Client()
     model_id = "gemini-2.0-flash-live-preview-04-09"

genai/live/live_ground_googsearch_with_txt.py

@@ -19,8 +19,14 @@
 async def generate_content() -> list[str]:
     # [START googlegenaisdk_live_ground_googsearch_with_txt]
     from google import genai
-    from google.genai.types import (Content, GoogleSearch, LiveConnectConfig,
-                                    Modality, Part, Tool)
+    from google.genai.types import (
+        Content,
+        GoogleSearch,
+        LiveConnectConfig,
+        Modality,
+        Part,
+        Tool,
+    )
 
     client = genai.Client()
     model_id = "gemini-2.0-flash-live-preview-04-09"

genai/live/live_structured_ouput_with_txt.py

@@ -30,8 +30,10 @@ def generate_content() -> CalendarEvent:
     import google.auth.transport.requests
     import openai
     from google.auth import default
-    from openai.types.chat import (ChatCompletionSystemMessageParam,
-                                   ChatCompletionUserMessageParam)
+    from openai.types.chat import (
+        ChatCompletionSystemMessageParam,
+        ChatCompletionUserMessageParam,
+    )
 
     project_id = os.environ["GOOGLE_CLOUD_PROJECT"]
     location = "us-central1"

genai/live/live_transcribe_with_audio.py

@@ -22,8 +22,13 @@
 async def generate_content() -> list[str]:
     # [START googlegenaisdk_live_transcribe_with_audio]
     from google import genai
-    from google.genai.types import (AudioTranscriptionConfig, Content,
-                                    LiveConnectConfig, Modality, Part)
+    from google.genai.types import (
+        AudioTranscriptionConfig,
+        Content,
+        LiveConnectConfig,
+        Modality,
+        Part,
+    )
 
     client = genai.Client()
     model = "gemini-live-2.5-flash-preview-native-audio"

genai/live/live_with_txt.py

@@ -18,8 +18,13 @@
 async def generate_content() -> list[str]:
     # [START googlegenaisdk_live_with_txt]
     from google import genai
-    from google.genai.types import (Content, HttpOptions, LiveConnectConfig,
-                                    Modality, Part)
+    from google.genai.types import (
+        Content,
+        HttpOptions,
+        LiveConnectConfig,
+        Modality,
+        Part,
+    )
 
     client = genai.Client(http_options=HttpOptions(api_version="v1beta1"))
     model_id = "gemini-2.0-flash-live-preview-04-09"

genai/live/test_live_examples.py

@@ -39,31 +39,28 @@
 # The project name is included in the CICD pipeline
 # os.environ['GOOGLE_CLOUD_PROJECT'] = "add-your-project-name"
 
+
 @pytest.fixture()
 def mock_rag_components(mocker):
-   mock_client_cls = mocker.patch("google.genai.Client")
-
-
-   class AsyncIterator:
-       def __aiter__(self):
-           return self
-
-
-       async def __anext__(self):
-           if not hasattr(self, "used"):
-               self.used = True
-               return mocker.MagicMock(
-                   text="Mariusz Pudzianowski won in 2002, 2003, 2005, 2007, and 2008."
-               )
-           raise StopAsyncIteration
+    mock_client_cls = mocker.patch("google.genai.Client")
 
+    class AsyncIterator:
+        def __aiter__(self):
+            return self
 
-   mock_session = mocker.AsyncMock()
-   mock_session.__aenter__.return_value = mock_session
-   mock_session.receive = lambda: AsyncIterator()
+        async def __anext__(self):
+            if not hasattr(self, "used"):
+                self.used = True
+                return mocker.MagicMock(
+                    text="Mariusz Pudzianowski won in 2002, 2003, 2005, 2007, and 2008."
+                )
+            raise StopAsyncIteration
 
+    mock_session = mocker.AsyncMock()
+    mock_session.__aenter__.return_value = mock_session
+    mock_session.receive = lambda: AsyncIterator()
 
-   mock_client_cls.return_value.aio.live.connect.return_value = mock_session
+    mock_client_cls.return_value.aio.live.connect.return_value = mock_session
 
 
 @pytest.mark.asyncio
@@ -129,4 +126,4 @@ async def test_live_structured_ouput_with_txt() -> None:
 
 @pytest.mark.asyncio
 async def test_live_ground_ragengine_with_txt(mock_rag_components) -> None:
-   assert await live_ground_ragengine_with_txt.generate_content("test")
+    assert await live_ground_ragengine_with_txt.generate_content("test")