Skip to content

Commit cd6f4c9

Browse files
author
Abraham Lazaro Martinez
committed
typo
1 parent 13dd6a5 commit cd6f4c9

File tree

2 files changed

+14
-3
lines changed

2 files changed

+14
-3
lines changed

py/plugins/vertex-ai/src/genkit/plugins/vertex_ai/model_garden/model_garden.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def __init__(
5757
model: str,
5858
location: str,
5959
project_id: str,
60-
registry: GenkitRegistry | None,
60+
registry: GenkitRegistry,
6161
) -> None:
6262
"""Initializes the ModelGarden instance.
6363

py/plugins/vertex-ai/src/genkit/plugins/vertex_ai/model_garden/modelgarden_plugin.py

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,19 @@ def __init__(
4545
project_id: str | None = None,
4646
location: str | None = None,
4747
models: list[str] | None = None,
48-
):
49-
"""Initialize the plugin by registering actions with the registry."""
48+
) -> None:
49+
"""Initializes the plugin and sets up its configuration.
50+
51+
This constructor prepares the plugin by assigning the Google Cloud project ID,
52+
location, and a list of models to be used.
53+
54+
Args:
55+
project_id: The Google Cloud project ID to use. If not provided, it attempts
56+
to load from the `GCLOUD_PROJECT` environment variable.
57+
location: The Google Cloud region to use for services. If not provided,
58+
it defaults to `DEFAULT_REGION`.
59+
models: An optional list of model names to register with the plugin.
60+
"""
5061
self.project_id = project_id if project_id is not None else os.getenv(const.GCLOUD_PROJECT)
5162
self.location = location if location is not None else const.DEFAULT_REGION
5263
self.models = models

0 commit comments

Comments
 (0)