Skip to content

Commit 5ab2463

Browse files
google-genai-botcopybara-github
authored andcommitted
chore(interaction-api): Add Vertex AI Search and Enterprise Web Search to Interaction API For Vertex AI.
PiperOrigin-RevId: 895604678
1 parent 227e509 commit 5ab2463

File tree

1 file changed

+38
-3
lines changed

1 file changed

+38
-3
lines changed

src/interactions/resources/interactions.ts

Lines changed: 38 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

16901691
export 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

Comments
 (0)