feat: Support more image sizes and resolutions

PiperOrigin-RevId: 875269060

Author: Annhiluc

google/genai/_interactions/types/google_search_call_content.py

@@ -15,6 +15,7 @@
 
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
+from typing import Optional
 from typing_extensions import Literal
 
 from .._models import BaseModel
@@ -33,3 +34,6 @@ class GoogleSearchCallContent(BaseModel):
     """The arguments to pass to Google Search."""
 
     type: Literal["google_search_call"]
+
+    search_type: Optional[Literal["web_search", "image_search"]] = None
+    """The type of search grounding enabled."""

google/genai/_interactions/types/google_search_call_content_param.py

@@ -34,3 +34,6 @@ class GoogleSearchCallContentParam(TypedDict, total=False):
     """The arguments to pass to Google Search."""
 
     type: Required[Literal["google_search_call"]]
+
+    search_type: Literal["web_search", "image_search"]
+    """The type of search grounding enabled."""

google/genai/_interactions/types/image_config.py

@@ -26,6 +26,8 @@
 class ImageConfig(BaseModel):
     """The configuration for image interaction."""
 
-    aspect_ratio: Optional[Literal["1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9", "21:9"]] = None
+    aspect_ratio: Optional[
+        Literal["1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9", "21:9", "1:8", "8:1", "1:4", "4:1"]
+    ] = None
 
-    image_size: Optional[Literal["1K", "2K", "4K"]] = None
+    image_size: Optional[Literal["1K", "2K", "4K", "512"]] = None

google/genai/_interactions/types/image_config_param.py

@@ -25,6 +25,8 @@
 class ImageConfigParam(TypedDict, total=False):
     """The configuration for image interaction."""
 
-    aspect_ratio: Literal["1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9", "21:9"]
+    aspect_ratio: Literal[
+        "1:1", "2:3", "3:2", "3:4", "4:3", "4:5", "5:4", "9:16", "16:9", "21:9", "1:8", "8:1", "1:4", "4:1"
+    ]
 
-    image_size: Literal["1K", "2K", "4K"]
+    image_size: Literal["1K", "2K", "4K", "512"]

google/genai/_interactions/types/tool.py

@@ -33,6 +33,9 @@ class GoogleSearch(BaseModel):
 
     type: Literal["google_search"]
 
+    search_types: Optional[List[Literal["web_search", "image_search"]]] = None
+    """The types of search grounding to enable."""
+
 
 class CodeExecution(BaseModel):
     """A tool that can be used by the model to execute code."""

google/genai/_interactions/types/tool_param.py

@@ -17,7 +17,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, Union, Iterable
+from typing import Dict, List, Union, Iterable
 from typing_extensions import Literal, Required, Annotated, TypeAlias, TypedDict
 
 from .._types import SequenceNotStr
@@ -33,6 +33,9 @@ class GoogleSearch(TypedDict, total=False):
 
     type: Required[Literal["google_search"]]
 
+    search_types: List[Literal["web_search", "image_search"]]
+    """The types of search grounding to enable."""
+
 
 class CodeExecution(TypedDict, total=False):
     """A tool that can be used by the model to execute code."""