@@ -38,6 +38,32 @@ option ruby_package = "Google::Cloud::AIPlatform::V1beta1";
3838// one type of Tool (e.g FunctionDeclaration, Retrieval or
3939// GoogleSearchRetrieval).
4040message Tool {
41+ // These are available confidence level user can set to block malicious urls
42+ // with chosen confidence and above. For understanding different confidence of
43+ // webrisk, please refer to
44+ // https://cloud.google.com/web-risk/docs/reference/rpc/google.cloud.webrisk.v1eap1#confidencelevel
45+ enum PhishBlockThreshold {
46+ // Defaults to unspecified.
47+ PHISH_BLOCK_THRESHOLD_UNSPECIFIED = 0 ;
48+
49+ // Blocks Low and above confidence URL that is risky.
50+ BLOCK_LOW_AND_ABOVE = 30 ;
51+
52+ // Blocks Medium and above confidence URL that is risky.
53+ BLOCK_MEDIUM_AND_ABOVE = 40 ;
54+
55+ // Blocks High and above confidence URL that is risky.
56+ BLOCK_HIGH_AND_ABOVE = 50 ;
57+
58+ // Blocks Higher and above confidence URL that is risky.
59+ BLOCK_HIGHER_AND_ABOVE = 55 ;
60+
61+ // Blocks Very high and above confidence URL that is risky.
62+ BLOCK_VERY_HIGH_AND_ABOVE = 60 ;
63+
64+ // Blocks Extremely high confidence URL that is risky.
65+ BLOCK_ONLY_EXTREMELY_HIGH = 100 ;
66+ }
4167 // GoogleSearch tool type.
4268 // Tool to support Google Search in Model. Powered by Google.
4369 message GoogleSearch {
@@ -46,6 +72,11 @@ message Tool {
4672 // Example: ["amazon.com", "facebook.com"].
4773 repeated string exclude_domains = 3
4874 [(google.api.field_behavior ) = OPTIONAL ];
75+
76+ // Optional. Sites with confidence level chosen & above this value will be
77+ // blocked from the search results.
78+ optional PhishBlockThreshold blocking_confidence = 4
79+ [(google.api.field_behavior ) = OPTIONAL ];
4980 }
5081
5182 // Tool that executes code generated by the model, and automatically returns
@@ -452,6 +483,11 @@ message EnterpriseWebSearch {
452483 // Optional. List of domains to be excluded from the search results.
453484 // The default limit is 2000 domains.
454485 repeated string exclude_domains = 1 [(google.api.field_behavior ) = OPTIONAL ];
486+
487+ // Optional. Sites with confidence level chosen & above this value will be
488+ // blocked from the search results.
489+ optional Tool.PhishBlockThreshold blocking_confidence = 4
490+ [(google.api.field_behavior ) = OPTIONAL ];
455491}
456492
457493// Describes the options to customize dynamic retrieval.
0 commit comments