gcp-vertex-ai
Installation
Entrypoint
Types
GcpVertexAi

gcp-vertex-ai

Provides functions for uploading models, creating endpoints, deploying model versions,
and running predictions against Vertex AI services.

Installation

dagger install github.com/telchak/daggerverse/gcp-vertex-ai@v0.1.0

Entrypoint

Return Type

GcpVertexAi !

Example

dagger -m github.com/telchak/daggerverse/gcp-vertex-ai@010621c997378db92da5969584001be575c5e5a7 call \

func (m *MyModule) Example() *dagger.GcpVertexAi  {
	return dag.
			GcpVertexAi()
}

@function
def example() -> dagger.GcpVertexAi:
	return (
		dag.gcp_vertex_ai()
	)

@func()
example(): GcpVertexAi {
	return dag
		.gcpVertexAi()
}

Types

GcpVertexAi 🔗

Google Cloud Vertex AI utilities.

deployModel() 🔗

Deploy a containerized model to Vertex AI.

Return Type

String !

Arguments

Name	Type	Default Value	Description
gcloud	Container !	-	Authenticated gcloud container
imageUri	String !	-	Container image URI
modelName	String !	-	Model display name
endpointName	String !	-	Endpoint display name
region	String !	"us-central1"	GCP region
machineType	String !	"n1-standard-4"	VM machine type
acceleratorType	String !	"NVIDIA_TESLA_T4"	GPU type
acceleratorCount	Integer !	1	Number of GPUs
minReplicas	Integer !	1	Minimum replicas
maxReplicas	Integer !	3	Maximum replicas

Example

dagger -m github.com/telchak/daggerverse/gcp-vertex-ai@010621c997378db92da5969584001be575c5e5a7 call \
 deploy-model --gcloud IMAGE:TAG --image-uri string --model-name string --endpoint-name string --region string --machine-type string --accelerator-type string --accelerator-count integer --min-replicas integer --max-replicas integer

func (m *MyModule) Example(ctx context.Context, gcloud *dagger.Container, imageUri string, modelName string, endpointName string, region string, machineType string, acceleratorType string, acceleratorCount int, minReplicas int, maxReplicas int) string  {
	return dag.
			GcpVertexAi().
			DeployModel(ctx, gcloud, imageUri, modelName, endpointName, region, machineType, acceleratorType, acceleratorCount, minReplicas, maxReplicas)
}

@function
async def example(gcloud: dagger.Container, image_uri: str, model_name: str, endpoint_name: str, region: str, machine_type: str, accelerator_type: str, accelerator_count: int, min_replicas: int, max_replicas: int) -> str:
	return await (
		dag.gcp_vertex_ai()
		.deploy_model(gcloud, image_uri, model_name, endpoint_name, region, machine_type, accelerator_type, accelerator_count, min_replicas, max_replicas)
	)

@func()
async example(gcloud: Container, imageUri: string, modelName: string, endpointName: string, region: string, machineType: string, acceleratorType: string, acceleratorCount: number, minReplicas: number, maxReplicas: number): Promise<string> {
	return dag
		.gcpVertexAi()
		.deployModel(gcloud, imageUri, modelName, endpointName, region, machineType, acceleratorType, acceleratorCount, minReplicas, maxReplicas)
}

listEndpoints() 🔗

List all endpoints in Vertex AI.

Return Type

String !

Arguments

Name	Type	Default Value	Description
gcloud	Container !	-	Authenticated gcloud container
region	String !	"us-central1"	GCP region

Example

dagger -m github.com/telchak/daggerverse/gcp-vertex-ai@010621c997378db92da5969584001be575c5e5a7 call \
 list-endpoints --gcloud IMAGE:TAG --region string

func (m *MyModule) Example(ctx context.Context, gcloud *dagger.Container, region string) string  {
	return dag.
			GcpVertexAi().
			ListEndpoints(ctx, gcloud, region)
}

@function
async def example(gcloud: dagger.Container, region: str) -> str:
	return await (
		dag.gcp_vertex_ai()
		.list_endpoints(gcloud, region)
	)

@func()
async example(gcloud: Container, region: string): Promise<string> {
	return dag
		.gcpVertexAi()
		.listEndpoints(gcloud, region)
}

listModels() 🔗

List all models in Vertex AI.

Return Type

String !

Arguments

Name	Type	Default Value	Description
gcloud	Container !	-	Authenticated gcloud container
region	String !	"us-central1"	GCP region

Example

dagger -m github.com/telchak/daggerverse/gcp-vertex-ai@010621c997378db92da5969584001be575c5e5a7 call \
 list-models --gcloud IMAGE:TAG --region string

func (m *MyModule) Example(ctx context.Context, gcloud *dagger.Container, region string) string  {
	return dag.
			GcpVertexAi().
			ListModels(ctx, gcloud, region)
}

@function
async def example(gcloud: dagger.Container, region: str) -> str:
	return await (
		dag.gcp_vertex_ai()
		.list_models(gcloud, region)
	)

@func()
async example(gcloud: Container, region: string): Promise<string> {
	return dag
		.gcpVertexAi()
		.listModels(gcloud, region)
}