Skip to content

Commit 3457f5a

Browse files
Update changes
1 parent ef0fc2d commit 3457f5a

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

gcp/table_gcp_vertex_ai_model.go

+4-4
Original file line numberDiff line numberDiff line change
@@ -195,14 +195,14 @@ func tableGcpVertexAIModel(ctx context.Context) *plugin.Table {
195195
{
196196
Name: "akas",
197197
Type: proto.ColumnType_JSON,
198-
Transform: transform.FromP(gcpModelNameToAkas, "akas"),
198+
Transform: transform.FromP(gcpModelStandard, "akas"),
199199
Description: ColumnDescriptionAkas,
200200
},
201201
// Standard gcp columns
202202
{
203203
Name: "location",
204204
Type: proto.ColumnType_STRING,
205-
Transform: transform.FromP(gcpAIPlatformTurbotData, "Location"),
205+
Transform: transform.FromP(gcpModelStandard, "Location"),
206206
Description: ColumnDescriptionLocation,
207207
},
208208
{
@@ -300,7 +300,7 @@ func getAIPlatformModel(ctx context.Context, d *plugin.QueryData, h *plugin.Hydr
300300
getProjectCached := plugin.HydrateFunc(getProject).WithCache()
301301
projectId, err := getProjectCached(ctx, d, h)
302302
if err != nil {
303-
logger.Error("gcp_vertex_ai_endpoint.getAIPlatformEndpoint", "cache_error", err)
303+
logger.Error("gcp_vertex_ai_model.getAIPlatformModel", "cache_error", err)
304304
return nil, err
305305
}
306306
project := projectId.(string)
@@ -343,7 +343,7 @@ func getAIPlatformModel(ctx context.Context, d *plugin.QueryData, h *plugin.Hydr
343343

344344
/// TRANSFORM FUNCTIONS
345345

346-
func gcpModelNameToAkas(ctx context.Context, d *transform.TransformData) (interface{}, error) {
346+
func gcpModelStandard(ctx context.Context, d *transform.TransformData) (interface{}, error) {
347347
param := d.Param.(string)
348348
AIData := d.HydrateItem.(*aiplatformpb.Model)
349349
akas := []string{"gcp://aiplatform.googleapis.com/" + AIData.Name}

0 commit comments

Comments
 (0)