File tree Expand file tree Collapse file tree 2 files changed +14
-3
lines changed
py/plugins/vertex-ai/src/genkit/plugins/vertex_ai/model_garden Expand file tree Collapse file tree 2 files changed +14
-3
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments