Merge remote-tracking branch 'origin/add-vertex-ai-model-COPY' into v…

…ertex-ai-model-UPLOAD
BBBmau · Oct 23, 2024 · b6c68f0 · b6c68f0
2 parents 83849bc + 2900c02
commit b6c68f0
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/mmv1/products/vertexai/Models.yaml b/mmv1/products/vertexai/Models.yaml
@@ -37,6 +37,12 @@ async:
   result:
     path: 'response'
     resource_inside_response: true
+  status:
+    path: 'done'
+    complete: true
+    allowed:
+      - true
+      - false
   error:
     path: 'error'
     message: 'message'

diff --git a/mmv1/templates/terraform/encoders/vertex_ai_models.tmpl b/mmv1/templates/terraform/encoders/vertex_ai_models.tmpl
@@ -3,4 +3,20 @@ if obj["sourceModel"] != nil {
     delete(obj, "metadata")
 }
 
+newObj := make(map[string]interface{})
+name := ""
+if v, ok := d.GetOk("model_id"); ok {
+	name = v.(string)
+    delete(obj,"modelId")
+} else {
+	name = d.Get("name").(string)
+}
+newObj["modelId"] = name
+if v, ok := d.GetOk("parent_model"); ok {
+    newObj["parentModel"] = v
+    delete(obj,"parentModel")
+}
+newObj["model"] = obj
+return newObj, nil
+
 return obj, nil