implement suggestions

Author: anakin87

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

@@ -138,6 +138,18 @@ def weather_function(city: str):
     messages = [ChatMessage.from_user("What's the weather in Paris?")]
     response = chat_generator_with_tools.run(messages=messages)
     ```
+
+    ### Usage example with FileContent embedded in a ChatMessage
+
+    ```python
+    from haystack.dataclasses import ChatMessage, FileContent
+    from haystack_integrations.components.generators.google_genai import GoogleGenAIChatGenerator
+
+    file_content = FileContent.from_url("https://arxiv.org/pdf/2309.08632")
+    chat_message = ChatMessage.from_user(content_parts=[file_content, "Summarize this paper in 100 words."])
+    chat_generator = GoogleGenAIChatGenerator()
+    response = chat_generator.run(messages=[chat_message])
+    ```
     """
 
     def __init__(
@@ -255,7 +267,6 @@ def _handle_streaming_response(
         try:
             chunks = []
 
-            chunk = None
             for i, chunk in enumerate(response_stream):
                 streaming_chunk = _convert_google_chunk_to_streaming_chunk(
                     chunk=chunk, index=i, component_info=component_info, model=self._model

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

@@ -233,9 +233,9 @@ def _convert_message_to_google_genai_format(message: ChatMessage) -> types.Conte
                 msg = f"{cls_name} is only supported for user messages"
                 raise ValueError(msg)
 
-            # Validate image MIME type and format
+            # MIME type validation: must be provided and, for images, one of the supported types
             if not content_part.mime_type:
-                msg = f"Mime type is required to use {cls_name} with GoogleGenAIChatGenerator"
+                msg = f"MIME type is required to use {cls_name} with GoogleGenAIChatGenerator"
                 raise ValueError(msg)
 
             if (
@@ -256,7 +256,6 @@ def _convert_message_to_google_genai_format(message: ChatMessage) -> types.Conte
                 )
                 bytes_data = base64.b64decode(base64_data)
 
-                # Create Part using from_bytes method
                 file_part = types.Part.from_bytes(data=bytes_data, mime_type=content_part.mime_type)
                 parts.append(file_part)