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.0Entrypoint
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 integerfunc (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, imageuri: str, modelname: str, endpointname: str, region: str, machinetype: str, acceleratortype: str, acceleratorcount: int, minreplicas: int, maxreplicas: int) -> str:
return await (
dag.gcp_vertex_ai()
.deploymodel(gcloud, imageuri, modelname, endpointname, region, machinetype, acceleratortype, acceleratorcount, minreplicas, maxreplicas)
)@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 stringfunc (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()
.listendpoints(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 stringfunc (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()
.listmodels(gcloud, region)
)@func()
async example(gcloud: Container, region: string): Promise<string> {
return dag
.gcpVertexAi()
.listModels(gcloud, region)
}