Skip to content

Commit c222f4b

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
feat: Add Webpage Browser Extension to Vertex AI Extension SDK
PiperOrigin-RevId: 642015460
1 parent 3ffb6cb commit c222f4b

File tree

1 file changed

+21
-1
lines changed

1 file changed

+21
-1
lines changed

vertexai/extensions/_extensions.py

+21-1
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,23 @@
6868
},
6969
},
7070
},
71+
"webpage_browser": {
72+
"display_name": "Webpage Browser",
73+
"description": "This extension fetches the content of a webpage",
74+
"manifest": {
75+
"name": "webpage_browser",
76+
"description": "Vertex Webpage Browser Extension",
77+
"api_spec": {
78+
"open_api_gcs_uri": (
79+
"gs://vertex-extension-public/webpage_browser.yaml"
80+
),
81+
},
82+
"auth_config": {
83+
"auth_type": "GOOGLE_SERVICE_ACCOUNT_AUTH",
84+
"google_service_account_config": {},
85+
},
86+
},
87+
},
7188
}
7289

7390

@@ -243,7 +260,8 @@ def from_hub(
243260
Args:
244261
name (str):
245262
Required. The name of the extension in the hub to be created.
246-
Supported values are "code_interpreter" and "vertex_ai_search".
263+
Supported values are "code_interpreter", "vertex_ai_search" and
264+
"webpage_browser".
247265
runtime_config (Union[dict[str, Any], RuntimeConfig]):
248266
Optional. Runtime config controlling the runtime behavior of
249267
the Extension. Defaults to None.
@@ -286,6 +304,8 @@ def from_hub(
286304
"vertex_ai_search_runtime_config is required for "
287305
"vertex_ai_search extension"
288306
)
307+
elif name == "webpage_browser":
308+
pass
289309
else:
290310
raise ValueError(f"Unsupported 1P extension name: {name}")
291311
extension_info = _VERTEX_EXTENSION_HUB[name]

0 commit comments

Comments
 (0)