chore: Add GoogleGenAIChatGenerator examples, set safety_settings (#1901)

* Add examples, set safety_settings

* Add print in examples

Author: vblagoje

integrations/google_genai/examples/chatgenerator_example.py

@@ -0,0 +1,32 @@
+# To run this example, you will need to
+# 1) set `GOOGLE_API_KEY` environment variable
+# 2) install the google_genai_haystack integration: pip install google-genai-haystack
+# Note: if you change the model, update the model-specific inference parameters.
+
+
+from haystack.dataclasses import ChatMessage
+
+from haystack_integrations.components.generators.google_genai import GoogleGenAIChatGenerator
+
+generator = GoogleGenAIChatGenerator(
+    model="gemini-2.0-flash",
+    # model-specific inference parameters
+    generation_kwargs={
+        "temperature": 0.7,
+    },
+)
+
+system_prompt = """
+You are a helpful assistant that helps users learn more about Google Cloud services.
+Your audience is engineers with a decent technical background.
+Be very concise and specific in your answers, keeping them short.
+You may use technical terms, jargon, and abbreviations that are common among practitioners.
+"""
+
+messages = [
+    ChatMessage.from_system(system_prompt),
+    ChatMessage.from_user("Which service should I use to train custom Machine Learning models?"),
+]
+
+results = generator.run(messages)
+print(results["replies"][0].text)

integrations/google_genai/pyproject.toml

@@ -142,6 +142,8 @@ ban-relative-imports = "parents"
 [tool.ruff.lint.per-file-ignores]
 # Tests can use magic values, assertions, and relative imports
 "tests/**/*" = ["PLR2004", "S101", "TID252"]
+# Examples can use print statements
+"examples/**/*" = ["T201"]
 
 [tool.coverage.run]
 source = ["haystack_integrations"]

integrations/google_genai/src/haystack_integrations/components/generators/google_genai/chat/chat_generator.py

@@ -509,6 +509,10 @@ def run(
             if system_instruction:
                 config_params["system_instruction"] = system_instruction
 
+            # Add safety settings if provided
+            if safety_settings:
+                config_params["safety_settings"] = safety_settings
+
             # Add tools if provided
             if tools:
                 config_params["tools"] = _convert_tools_to_google_genai_format(tools)
@@ -593,6 +597,10 @@ async def run_async(
             if system_instruction:
                 config_params["system_instruction"] = system_instruction
 
+            # Add safety settings if provided
+            if safety_settings:
+                config_params["safety_settings"] = safety_settings
+
             # Add tools if provided
             if tools:
                 config_params["tools"] = _convert_tools_to_google_genai_format(tools)