core

package
v0.2.0-alpha Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Apr 26, 2024 License: Apache-2.0 Imports: 13 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func BuildTypesMap 

func BuildTypesMap(cc grpc.ClientConnInterface, nameConfig mlmdtypes.MLMDTypeNamesConfig) (map[string]int64, error)

func NewModelRegistryService 

func NewModelRegistryService(cc grpc.ClientConnInterface, nameConfig mlmdtypes.MLMDTypeNamesConfig) (api.ModelRegistryApi, error)

NewModelRegistryService creates a new instance of the ModelRegistryService, initializing it with the provided gRPC client connection. It _assumes_ the necessary MLMD's Context, Artifact, Execution types etc. are already setup in the underlying MLMD service.

Parameters:

  • cc: A gRPC client connection to the underlying MLMD service

Types

type ModelRegistryService 

type ModelRegistryService struct {
	// contains filtered or unexported fields
}

ModelRegistryService is the core library of the model registry

func (*ModelRegistryService) GetArtifactById 

func (serv *ModelRegistryService) GetArtifactById(id string) (*openapi.Artifact, error)

func (*ModelRegistryService) GetArtifacts 

func (serv *ModelRegistryService) GetArtifacts(listOptions api.ListOptions, modelVersionId *string) (*openapi.ArtifactList, error)

func (*ModelRegistryService) GetInferenceServiceById 

func (serv *ModelRegistryService) GetInferenceServiceById(id string) (*openapi.InferenceService, error)

GetInferenceServiceById retrieves an inference service by its unique identifier (ID).

func (*ModelRegistryService) GetInferenceServiceByParams 

func (serv *ModelRegistryService) GetInferenceServiceByParams(name *string, servingEnvironmentId *string, externalId *string) (*openapi.InferenceService, error)

GetInferenceServiceByParams retrieves an inference service based on specified parameters, such as (name and serving environment ID), or external ID. If multiple or no serving environments are found, an error is returned accordingly.

func (*ModelRegistryService) GetInferenceServices 

func (serv *ModelRegistryService) GetInferenceServices(listOptions api.ListOptions, servingEnvironmentId *string, runtime *string) (*openapi.InferenceServiceList, error)

GetInferenceServices retrieves a list of inference services based on the provided list options and optional serving environment ID and runtime.

func (*ModelRegistryService) GetModelArtifactById 

func (serv *ModelRegistryService) GetModelArtifactById(id string) (*openapi.ModelArtifact, error)

GetModelArtifactById retrieves a model artifact by its unique identifier (ID).

func (*ModelRegistryService) GetModelArtifactByInferenceService 

func (serv *ModelRegistryService) GetModelArtifactByInferenceService(inferenceServiceId string) (*openapi.ModelArtifact, error)

GetModelArtifactByInferenceService retrieves the model artifact associated with the specified inference service ID.

func (*ModelRegistryService) GetModelArtifactByParams 

func (serv *ModelRegistryService) GetModelArtifactByParams(artifactName *string, modelVersionId *string, externalId *string) (*openapi.ModelArtifact, error)

GetModelArtifactByParams retrieves a model artifact based on specified parameters, such as (artifact name and model version ID), or external ID. If multiple or no model artifacts are found, an error is returned.

func (*ModelRegistryService) GetModelArtifacts 

func (serv *ModelRegistryService) GetModelArtifacts(listOptions api.ListOptions, modelVersionId *string) (*openapi.ModelArtifactList, error)

GetModelArtifacts retrieves a list of model artifacts based on the provided list options and optional model version ID.

func (*ModelRegistryService) GetModelVersionById 

func (serv *ModelRegistryService) GetModelVersionById(id string) (*openapi.ModelVersion, error)

GetModelVersionById retrieves a model version by its unique identifier (ID).

func (*ModelRegistryService) GetModelVersionByInferenceService 

func (serv *ModelRegistryService) GetModelVersionByInferenceService(inferenceServiceId string) (*openapi.ModelVersion, error)

GetModelVersionByInferenceService retrieves the model version associated with the specified inference service ID.

func (*ModelRegistryService) GetModelVersionByParams 

func (serv *ModelRegistryService) GetModelVersionByParams(versionName *string, registeredModelId *string, externalId *string) (*openapi.ModelVersion, error)

GetModelVersionByParams retrieves a model version based on specified parameters, such as (version name and registered model ID), or external ID. If multiple or no model versions are found, an error is returned.

func (*ModelRegistryService) GetModelVersions 

func (serv *ModelRegistryService) GetModelVersions(listOptions api.ListOptions, registeredModelId *string) (*openapi.ModelVersionList, error)

GetModelVersions retrieves a list of model versions based on the provided list options and optional registered model ID.

func (*ModelRegistryService) GetRegisteredModelById 

func (serv *ModelRegistryService) GetRegisteredModelById(id string) (*openapi.RegisteredModel, error)

GetRegisteredModelById retrieves a registered model by its unique identifier (ID).

func (*ModelRegistryService) GetRegisteredModelByInferenceService 

func (serv *ModelRegistryService) GetRegisteredModelByInferenceService(inferenceServiceId string) (*openapi.RegisteredModel, error)

GetRegisteredModelByInferenceService retrieves a registered model associated with the specified inference service ID.

func (*ModelRegistryService) GetRegisteredModelByParams 

func (serv *ModelRegistryService) GetRegisteredModelByParams(name *string, externalId *string) (*openapi.RegisteredModel, error)

GetRegisteredModelByParams retrieves a registered model based on specified parameters, such as name or external ID. If multiple or no registered models are found, an error is returned accordingly.

func (*ModelRegistryService) GetRegisteredModels 

func (serv *ModelRegistryService) GetRegisteredModels(listOptions api.ListOptions) (*openapi.RegisteredModelList, error)

GetRegisteredModels retrieves a list of registered models based on the provided list options.

func (*ModelRegistryService) GetServeModelById 

func (serv *ModelRegistryService) GetServeModelById(id string) (*openapi.ServeModel, error)

GetServeModelById retrieves a serve model by its unique identifier (ID).

func (*ModelRegistryService) GetServeModels 

func (serv *ModelRegistryService) GetServeModels(listOptions api.ListOptions, inferenceServiceId *string) (*openapi.ServeModelList, error)

GetServeModels retrieves a list of serve models based on the provided list options and optional inference service ID.

func (*ModelRegistryService) GetServingEnvironmentById 

func (serv *ModelRegistryService) GetServingEnvironmentById(id string) (*openapi.ServingEnvironment, error)

GetServingEnvironmentById retrieves a serving environment by its unique identifier (ID).

func (*ModelRegistryService) GetServingEnvironmentByParams 

func (serv *ModelRegistryService) GetServingEnvironmentByParams(name *string, externalId *string) (*openapi.ServingEnvironment, error)

GetServingEnvironmentByParams retrieves a serving environment based on specified parameters, such as name or external ID. If multiple or no serving environments are found, an error is returned accordingly.

func (*ModelRegistryService) GetServingEnvironments 

func (serv *ModelRegistryService) GetServingEnvironments(listOptions api.ListOptions) (*openapi.ServingEnvironmentList, error)

GetServingEnvironments retrieves a list of serving environments based on the provided list options.

func (*ModelRegistryService) UpsertArtifact 

func (serv *ModelRegistryService) UpsertArtifact(artifact *openapi.Artifact, modelVersionId *string) (*openapi.Artifact, error)

UpsertArtifact creates a new artifact if the provided artifact's ID is nil, or updates an existing artifact if the ID is provided. A model version ID must be provided to disambiguate between artifacts. Upon creation, new artifacts will be associated with their corresponding model version.

func (*ModelRegistryService) UpsertInferenceService 

func (serv *ModelRegistryService) UpsertInferenceService(inferenceService *openapi.InferenceService) (*openapi.InferenceService, error)

UpsertInferenceService creates a new inference service if the provided inference service's ID is nil, or updates an existing inference service if the ID is provided.

func (*ModelRegistryService) UpsertModelArtifact 

func (serv *ModelRegistryService) UpsertModelArtifact(modelArtifact *openapi.ModelArtifact, modelVersionId *string) (*openapi.ModelArtifact, error)

UpsertModelArtifact creates a new model artifact if the provided model artifact's ID is nil, or updates an existing model artifact if the ID is provided. If a model version ID is provided and the model artifact is newly created, establishes an explicit attribution between the model version and the created model artifact.

func (*ModelRegistryService) UpsertModelVersion 

func (serv *ModelRegistryService) UpsertModelVersion(modelVersion *openapi.ModelVersion, registeredModelId *string) (*openapi.ModelVersion, error)

UpsertModelVersion creates a new model version if the provided model version's ID is nil, or updates an existing model version if the ID is provided.

func (*ModelRegistryService) UpsertRegisteredModel 

func (serv *ModelRegistryService) UpsertRegisteredModel(registeredModel *openapi.RegisteredModel) (*openapi.RegisteredModel, error)

UpsertRegisteredModel creates a new registered model if the given registered model's ID is nil, or updates an existing registered model if the ID is provided.

func (*ModelRegistryService) UpsertServeModel 

func (serv *ModelRegistryService) UpsertServeModel(serveModel *openapi.ServeModel, inferenceServiceId *string) (*openapi.ServeModel, error)

UpsertServeModel creates a new serve model if the provided serve model's ID is nil, or updates an existing serve model if the ID is provided.

func (*ModelRegistryService) UpsertServingEnvironment 

func (serv *ModelRegistryService) UpsertServingEnvironment(servingEnvironment *openapi.ServingEnvironment) (*openapi.ServingEnvironment, error)

UpsertServingEnvironment creates a new serving environment if the provided serving environment's ID is nil, or updates an existing serving environment if the ID is provided.

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.