@@ -1191,7 +1191,7 @@ export interface GoogleSearchCallContent {
11911191 /**
11921192 * The type of search grounding enabled.
11931193 */
1194- search_type ?: 'web_search' | 'image_search' ;
1194+ search_type ?: 'web_search' | 'image_search' | 'enterprise_web_search' ;
11951195
11961196 /**
11971197 * A signature hash for backend validation.
@@ -1685,7 +1685,8 @@ export type Tool =
16851685 | Tool . MCPServer
16861686 | Tool . GoogleSearch
16871687 | Tool . FileSearch
1688- | Tool . GoogleMaps ;
1688+ | Tool . GoogleMaps
1689+ | Tool . Retrieval ;
16891690
16901691export namespace Tool {
16911692 /**
@@ -1755,7 +1756,7 @@ export namespace Tool {
17551756 /**
17561757 * The types of search grounding to enable.
17571758 */
1758- search_types ?: Array < 'web_search' | 'image_search' > ;
1759+ search_types ?: Array < 'web_search' | 'image_search' | 'enterprise_web_search' > ;
17591760 }
17601761
17611762 /**
@@ -1802,6 +1803,40 @@ export namespace Tool {
18021803 */
18031804 longitude ?: number ;
18041805 }
1806+
1807+ /**
1808+ * A tool that can be used by the model to retrieve files.
1809+ */
1810+ export interface Retrieval {
1811+ type : 'retrieval' ;
1812+
1813+ /**
1814+ * The types of file retrieval to enable.
1815+ */
1816+ retrieval_types ?: Array < 'vertex_ai_search' > ;
1817+
1818+ /**
1819+ * Used to specify configuration for VertexAISearch.
1820+ */
1821+ vertex_ai_search_config ?: Retrieval . VertexAISearchConfig ;
1822+ }
1823+
1824+ export namespace Retrieval {
1825+ /**
1826+ * Used to specify configuration for VertexAISearch.
1827+ */
1828+ export interface VertexAISearchConfig {
1829+ /**
1830+ * Optional. Used to specify Vertex AI Search datastores.
1831+ */
1832+ datastores ?: Array < string > ;
1833+
1834+ /**
1835+ * Optional. Used to specify Vertex AI Search engine.
1836+ */
1837+ engine ?: string ;
1838+ }
1839+ }
18051840}
18061841
18071842/**
0 commit comments