feat(interaction-api): Add Vertex AI Search and Enterprise Web Search to Interaction API For Vertex AI.

PiperOrigin-RevId: 893615421

Author: google-genai-bot

google/genai/_interactions/types/google_search_call_content.py

@@ -35,7 +35,7 @@ class GoogleSearchCallContent(BaseModel):
 
     type: Literal["google_search_call"]
 
-    search_type: Optional[Literal["web_search", "image_search"]] = None
+    search_type: Optional[Literal["web_search", "image_search", "enterprise_web_search"]] = None
     """The type of search grounding enabled."""
 
     signature: Optional[str] = None

google/genai/_interactions/types/google_search_call_content_param.py

@@ -39,7 +39,7 @@ class GoogleSearchCallContentParam(TypedDict, total=False):
 
     type: Required[Literal["google_search_call"]]
 
-    search_type: Literal["web_search", "image_search"]
+    search_type: Literal["web_search", "image_search", "enterprise_web_search"]
     """The type of search grounding enabled."""
 
     signature: Annotated[Union[str, Base64FileInput], PropertyInfo(format="base64")]

google/genai/_interactions/types/tool.py

@@ -34,6 +34,8 @@
     "GoogleSearch",
     "FileSearch",
     "GoogleMaps",
+    "Retrieval",
+    "RetrievalVertexAISearchConfig",
 ]
 
 
@@ -84,7 +86,7 @@ class GoogleSearch(BaseModel):
 
     type: Literal["google_search"]
 
-    search_types: Optional[List[Literal["web_search", "image_search"]]] = None
+    search_types: Optional[List[Literal["web_search", "image_search", "enterprise_web_search"]]] = None
     """The types of search grounding to enable."""
 
 
@@ -121,7 +123,29 @@ class GoogleMaps(BaseModel):
     """The longitude of the user's location."""
 
 
+class RetrievalVertexAISearchConfig(BaseModel):
+    """Used to specify configuration for VertexAISearch."""
+
+    datastores: Optional[List[str]] = None
+    """Optional. Used to specify Vertex AI Search datastores."""
+
+    engine: Optional[str] = None
+    """Optional. Used to specify Vertex AI Search engine."""
+
+
+class Retrieval(BaseModel):
+    """A tool that can be used by the model to search files."""
+
+    type: Literal["retrieval"]
+
+    retrieval_types: Optional[List[Literal["vertex_ai_search"]]] = None
+    """The types of file search to enable."""
+
+    vertex_ai_search_config: Optional[RetrievalVertexAISearchConfig] = None
+    """Used to specify configuration for VertexAISearch."""
+
+
 Tool: TypeAlias = Annotated[
-    Union[Function, CodeExecution, URLContext, ComputerUse, MCPServer, GoogleSearch, FileSearch, GoogleMaps],
+    Union[Function, CodeExecution, URLContext, ComputerUse, MCPServer, GoogleSearch, FileSearch, GoogleMaps, Retrieval],
     PropertyInfo(discriminator="type"),
 ]

google/genai/_interactions/types/tool_param.py

@@ -34,6 +34,8 @@
     "GoogleSearch",
     "FileSearch",
     "GoogleMaps",
+    "Retrieval",
+    "RetrievalVertexAISearchConfig",
 ]
 
 
@@ -84,7 +86,7 @@ class GoogleSearch(TypedDict, total=False):
 
     type: Required[Literal["google_search"]]
 
-    search_types: List[Literal["web_search", "image_search"]]
+    search_types: List[Literal["web_search", "image_search", "enterprise_web_search"]]
     """The types of search grounding to enable."""
 
 
@@ -121,6 +123,28 @@ class GoogleMaps(TypedDict, total=False):
     """The longitude of the user's location."""
 
 
+class RetrievalVertexAISearchConfig(TypedDict, total=False):
+    """Used to specify configuration for VertexAISearch."""
+
+    datastores: SequenceNotStr[str]
+    """Optional. Used to specify Vertex AI Search datastores."""
+
+    engine: str
+    """Optional. Used to specify Vertex AI Search engine."""
+
+
+class Retrieval(TypedDict, total=False):
+    """A tool that can be used by the model to search files."""
+
+    type: Required[Literal["retrieval"]]
+
+    retrieval_types: List[Literal["vertex_ai_search"]]
+    """The types of file search to enable."""
+
+    vertex_ai_search_config: RetrievalVertexAISearchConfig
+    """Used to specify configuration for VertexAISearch."""
+
+
 ToolParam: TypeAlias = Union[
-    FunctionParam, CodeExecution, URLContext, ComputerUse, MCPServer, GoogleSearch, FileSearch, GoogleMaps
+    FunctionParam, CodeExecution, URLContext, ComputerUse, MCPServer, GoogleSearch, FileSearch, GoogleMaps, Retrieval
 ]