org.openmetadata.schema.services.connections.mlmodel.VertexAIConnection Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of openmetadata-spec Show documentation
There is a newer version: 1.6.1

package org.openmetadata.schema.services.connections.mlmodel;

import java.util.HashMap;
import java.util.Map;
import javax.annotation.processing.Generated;
import javax.validation.Valid;
import javax.validation.constraints.NotNull;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.annotation.JsonPropertyDescription;
import com.fasterxml.jackson.annotation.JsonPropertyOrder;
import com.fasterxml.jackson.annotation.JsonValue;
import org.openmetadata.schema.security.credentials.GCPCredentials;


/**
 * VertexAIConnection
 * 
 * Google VertexAI Connection Config
 * 
 */
@JsonInclude(JsonInclude.Include.NON_NULL)
@JsonPropertyOrder({
    "type",
    "credentials",
    "location"
})
@Generated("jsonschema2pojo")
public class VertexAIConnection {

    /**
     * Service type.
     * 
     */
    @JsonProperty("type")
    @JsonPropertyDescription("Service type.")
    private VertexAIConnection.VertexAIType type = VertexAIConnection.VertexAIType.fromValue("VertexAI");
    /**
     * GCPCredentials
     * 

     * GCP credentials configs.
     * (Required)
     * 
     */
    @JsonProperty("credentials")
    @JsonPropertyDescription("GCP credentials configs.")
    @Valid
    @NotNull
    private GCPCredentials credentials;
    /**
     * Project location
     * 

     * location/region of google cloud project
     * (Required)
     * 
     */
    @JsonProperty("location")
    @JsonPropertyDescription("location/region of google cloud project")
    @NotNull
    private String location;

    /**
     * Service type.
     * 
     */
    @JsonProperty("type")
    public VertexAIConnection.VertexAIType getType() {
        return type;
    }

    /**
     * Service type.
     * 
     */
    @JsonProperty("type")
    public void setType(VertexAIConnection.VertexAIType type) {
        this.type = type;
    }

    public VertexAIConnection withType(VertexAIConnection.VertexAIType type) {
        this.type = type;
        return this;
    }

    /**
     * GCPCredentials
     * 

     * GCP credentials configs.
     * (Required)
     * 
     */
    @JsonProperty("credentials")
    public GCPCredentials getCredentials() {
        return credentials;
    }

    /**
     * GCPCredentials
     * 

     * GCP credentials configs.
     * (Required)
     * 
     */
    @JsonProperty("credentials")
    public void setCredentials(GCPCredentials credentials) {
        this.credentials = credentials;
    }

    public VertexAIConnection withCredentials(GCPCredentials credentials) {
        this.credentials = credentials;
        return this;
    }

    /**
     * Project location
     * 

     * location/region of google cloud project
     * (Required)
     * 
     */
    @JsonProperty("location")
    public String getLocation() {
        return location;
    }

    /**
     * Project location
     * 
     * location/region of google cloud project
     * (Required)
     * 
     */
    @JsonProperty("location")
    public void setLocation(String location) {
        this.location = location;
    }

    public VertexAIConnection withLocation(String location) {
        this.location = location;
        return this;
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder();
        sb.append(VertexAIConnection.class.getName()).append('@').append(Integer.toHexString(System.identityHashCode(this))).append('[');
        sb.append("type");
        sb.append('=');
        sb.append(((this.type == null)?"":this.type));
        sb.append(',');
        sb.append("credentials");
        sb.append('=');
        sb.append(((this.credentials == null)?"":this.credentials));
        sb.append(',');
        sb.append("location");
        sb.append('=');
        sb.append(((this.location == null)?"":this.location));
        sb.append(',');
        if (sb.charAt((sb.length()- 1)) == ',') {
            sb.setCharAt((sb.length()- 1), ']');
        } else {
            sb.append(']');
        }
        return sb.toString();
    }

    @Override
    public int hashCode() {
        int result = 1;
        result = ((result* 31)+((this.location == null)? 0 :this.location.hashCode()));
        result = ((result* 31)+((this.type == null)? 0 :this.type.hashCode()));
        result = ((result* 31)+((this.credentials == null)? 0 :this.credentials.hashCode()));
        return result;
    }

    @Override
    public boolean equals(Object other) {
        if (other == this) {
            return true;
        }
        if ((other instanceof VertexAIConnection) == false) {
            return false;
        }
        VertexAIConnection rhs = ((VertexAIConnection) other);
        return ((((this.location == rhs.location)||((this.location!= null)&&this.location.equals(rhs.location)))&&((this.type == rhs.type)||((this.type!= null)&&this.type.equals(rhs.type))))&&((this.credentials == rhs.credentials)||((this.credentials!= null)&&this.credentials.equals(rhs.credentials))));
    }


    /**
     * Service type.
     * 
     */
    @Generated("jsonschema2pojo")
    public enum VertexAIType {

        VERTEX_AI("VertexAI");
        private final String value;
        private final static Map CONSTANTS = new HashMap();

        static {
            for (VertexAIConnection.VertexAIType c: values()) {
                CONSTANTS.put(c.value, c);
            }
        }

        VertexAIType(String value) {
            this.value = value;
        }

        @Override
        public String toString() {
            return this.value;
        }

        @JsonValue
        public String value() {
            return this.value;
        }

        @JsonCreator
        public static VertexAIConnection.VertexAIType fromValue(String value) {
            VertexAIConnection.VertexAIType constant = CONSTANTS.get(value);
            if (constant == null) {
                throw new IllegalArgumentException(value);
            } else {
                return constant;
            }
        }

    }

}