Connector: vertexai schema (#17632)

* vertexai connection schema

* remove yaml file

openmetadata-service/src/main/resources/json/data/testConnections/mlmodel/vertexai.json

@@ -0,0 +1,14 @@
+{
+    "name": "VertexAI",
+    "displayName": "VertexAI Test Connection",
+    "description": "Test Connection to validate the access against models and basic metadata extraction",
+    "steps": [
+      {
+        "name": "GetModels",
+        "description": "Validate that we can get the instances with the given credentials.",
+        "errorMessage": "Failed to get VertexAI models, please validate to the credentials of service account",
+        "shortCircuit": true,
+        "mandatory": true
+      }
+    ]
+}

openmetadata-spec/src/main/resources/json/schema/entity/services/connections/mlmodel/vertexaiConnection.json

@@ -0,0 +1,38 @@
+{
+  "$id": "https://open-metadata.org/schema/entity/services/connections/mlmodel/vertexaiConnection.json",
+  "$schema": "http://json-schema.org/draft-07/schema#",
+  "title": "VertexAIConnection",
+  "description": "Google VertexAI Connection Config",
+  "type": "object",
+  "javaType": "org.openmetadata.schema.services.connections.mlmodel.VertexAIConnection",
+  "definitions": {
+    "vertexAIType": {
+      "description": "Service type.",
+      "type": "string",
+      "enum": ["VertexAI"],
+      "default": "VertexAI"
+    }
+  },
+  "properties": {
+    "type": {
+      "title": "Service Type",
+      "description": "Service Type",
+      "$ref": "#/definitions/vertexAIType",
+      "default": "VertexAI"
+    },
+    "credentials": {
+      "title": "GCP Credentials",
+      "description": "GCP Credentials",
+      "$ref": "../../../../security/credentials/gcpCredentials.json"
+    },
+    "location": {
+      "title": "Project location",
+      "description": "location/region of google cloud project",
+      "type": "string"
+    }
+  },
+  "additionalProperties": false,
+  "required": [
+    "credentials", "location"
+  ]
+}

openmetadata-spec/src/main/resources/json/schema/entity/services/mlmodelService.json

@@ -14,7 +14,7 @@
       "description": "Type of MlModel service",
       "type": "string",
       "javaInterfaces": ["org.openmetadata.schema.EnumInterface"],
-      "enum": ["Mlflow", "Sklearn", "CustomMlModel", "SageMaker"],
+      "enum": ["Mlflow", "Sklearn", "CustomMlModel", "SageMaker", "VertexAI"],
       "javaEnums": [
         {
           "name": "Mlflow"
@@ -27,6 +27,9 @@
         },
         {
           "name": "SageMaker"
+        },
+        {
+          "name": "VertexAI"
         }
       ]
     },
@@ -52,6 +55,9 @@
             },
             {
               "$ref": "./connections/mlmodel/sageMakerConnection.json"
+            },
+            {
+              "$ref": "./connections/mlmodel/vertexaiConnection.json"
             }
           ]
         }