class Multiwoven::Integrations::Source::VertexModel::Client

def create_connection(connection_config)
  Google::Cloud::AIPlatform::V1::EndpointService::Client.configure do |config|
    config.endpoint = build_url(GOOGLE_VERTEX_ENDPOINT_SERVICE_URL, connection_config)
    config.credentials = connection_config["credentials_json"]
  end
  Google::Cloud::AIPlatform::V1::PredictionService::Client.configure do |config|
    config.endpoint = build_url(GOOGLE_VERTEX_ENDPOINT_SERVICE_URL, connection_config)
    config.credentials = connection_config["credentials_json"]
  end
  @client = Google::Cloud::AIPlatform::V1::EndpointService::Client.new
  @endpoint = Google::Cloud::AIPlatform::V1::PredictionService::Client.new
end