refactor: Add missing newline in GoogleGenAIChatGenerator and improve test formatting

Adwaithpv · Adwaithpv · commit 39e94b3f3532 · 2026-03-07T18:52:33.000+05:30
diff --git a/integrations/google_genai/src/haystack_integrations/components/generators/google_genai/chat/chat_generator.py b/integrations/google_genai/src/haystack_integrations/components/generators/google_genai/chat/chat_generator.py
@@ -151,6 +151,7 @@ def weather_function(city: str):
     response = chat_generator.run(messages=[chat_message])
     ```
     """
+
     SUPPORTED_MODELS: ClassVar[list[str]] = [
         "gemini-2.5-pro",
         "gemini-2.5-flash",
diff --git a/integrations/google_genai/tests/test_chat_generator.py b/integrations/google_genai/tests/test_chat_generator.py
@@ -47,11 +47,13 @@ def tools():
         )
     ]
 
+
 def test_supported_models_is_non_empty_list_of_strings():
     assert isinstance(GoogleGenAIChatGenerator.SUPPORTED_MODELS, list)
     assert GoogleGenAIChatGenerator.SUPPORTED_MODELS
     assert all(isinstance(model, str) and model for model in GoogleGenAIChatGenerator.SUPPORTED_MODELS)
 
+
 class TestGoogleGenAIChatGeneratorInitSerDe:
     def test_init_default(self, monkeypatch):
         monkeypatch.setenv("GOOGLE_API_KEY", "test-api-key")

Original file line number	Diff line number	Diff line change
`@@ -47,11 +47,13 @@ def tools():`
`47`	`47`	`)`
`48`	`48`	`]`
`49`	`49`
	`50`	`+`
`50`	`51`	`def test_supported_models_is_non_empty_list_of_strings():`
`51`	`52`	`assert isinstance(GoogleGenAIChatGenerator.SUPPORTED_MODELS, list)`
`52`	`53`	`assert GoogleGenAIChatGenerator.SUPPORTED_MODELS`
`53`	`54`	`assert all(isinstance(model, str) and model for model in GoogleGenAIChatGenerator.SUPPORTED_MODELS)`
`54`	`55`
	`56`	`+`
`55`	`57`	`class TestGoogleGenAIChatGeneratorInitSerDe:`
`56`	`58`	`def test_init_default(self, monkeypatch):`
`57`	`59`	`monkeypatch.setenv("GOOGLE_API_KEY", "test-api-key")`