sql

package
v0.2.3 Latest Latest
Warning

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

Go to latest
Published: Aug 27, 2025 License: Apache-2.0 Imports: 16 Imported by: 0

Documentation

Index

Constants

View Source
const (
	IsPromptTagKey = "mlflow.prompt.is_prompt"
)

Variables

This section is empty.

Functions

func HandleResourceAlreadyExistError added in v0.2.1

func HandleResourceAlreadyExistError(name string, isExistingEntityPrompt, isNewEntityPrompt bool) *contract.Error

HandleResourceAlreadyExistError handles name conflicts in the Model Registry.

Types

type ModelRegistrySQLStore

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

func NewModelRegistrySQLStore

func NewModelRegistrySQLStore(ctx context.Context, config *config.Config) (*ModelRegistrySQLStore, error)

func (*ModelRegistrySQLStore) CreateRegisteredModel

func (m *ModelRegistrySQLStore) CreateRegisteredModel(
	ctx context.Context, name, description string, tags []*entities.RegisteredModelTag,
) (*entities.RegisteredModel, *contract.Error)

func (*ModelRegistrySQLStore) DeleteModelVersion

func (m *ModelRegistrySQLStore) DeleteModelVersion(ctx context.Context, name, version string) *contract.Error

func (*ModelRegistrySQLStore) DeleteModelVersionTag added in v0.2.1

func (m *ModelRegistrySQLStore) DeleteModelVersionTag(
	ctx context.Context, name, version, key string,
) *contract.Error

func (*ModelRegistrySQLStore) DeleteRegisteredModel

func (m *ModelRegistrySQLStore) DeleteRegisteredModel(ctx context.Context, name string) *contract.Error

func (*ModelRegistrySQLStore) DeleteRegisteredModelAlias added in v0.2.0

func (m *ModelRegistrySQLStore) DeleteRegisteredModelAlias(ctx context.Context, name, alias string) *contract.Error

func (*ModelRegistrySQLStore) DeleteRegisteredModelTag

func (m *ModelRegistrySQLStore) DeleteRegisteredModelTag(ctx context.Context, name, key string) *contract.Error

func (*ModelRegistrySQLStore) Destroy

func (m *ModelRegistrySQLStore) Destroy() error

func (*ModelRegistrySQLStore) GetLatestVersions

func (m *ModelRegistrySQLStore) GetLatestVersions(
	ctx context.Context, name string, stages []string,
) ([]*entities.ModelVersion, *contract.Error)

func (*ModelRegistrySQLStore) GetModelVersion

func (m *ModelRegistrySQLStore) GetModelVersion(
	ctx context.Context, name, version string, eager bool,
) (*entities.ModelVersion, *contract.Error)

func (*ModelRegistrySQLStore) GetModelVersionByAlias added in v0.2.0

func (m *ModelRegistrySQLStore) GetModelVersionByAlias(
	ctx context.Context, name, alias string,
) (*entities.ModelVersion, *contract.Error)

func (*ModelRegistrySQLStore) GetModelVersionDownloadURI added in v0.2.1

func (m *ModelRegistrySQLStore) GetModelVersionDownloadURI(
	ctx context.Context, name, version string,
) (string, *contract.Error)

func (*ModelRegistrySQLStore) GetRegisteredModel

func (m *ModelRegistrySQLStore) GetRegisteredModel(
	ctx context.Context, name string,
) (*entities.RegisteredModel, *contract.Error)

func (*ModelRegistrySQLStore) GetRegisteredModelByName

func (m *ModelRegistrySQLStore) GetRegisteredModelByName(
	ctx context.Context, name string,
) (*entities.RegisteredModel, *contract.Error)

func (*ModelRegistrySQLStore) RenameRegisteredModel

func (m *ModelRegistrySQLStore) RenameRegisteredModel(
	ctx context.Context, name, newName string,
) (*entities.RegisteredModel, *contract.Error)

func (*ModelRegistrySQLStore) SetModelVersionTag added in v0.2.1

func (m *ModelRegistrySQLStore) SetModelVersionTag(
	ctx context.Context, name, version, key, value string,
) *contract.Error

func (*ModelRegistrySQLStore) SetRegisteredModelAlias

func (m *ModelRegistrySQLStore) SetRegisteredModelAlias(
	ctx context.Context, name, alias, version string,
) *contract.Error

func (*ModelRegistrySQLStore) SetRegisteredModelTag

func (m *ModelRegistrySQLStore) SetRegisteredModelTag(ctx context.Context, name, key, value string) *contract.Error

func (*ModelRegistrySQLStore) TransitionModelVersionStage

func (m *ModelRegistrySQLStore) TransitionModelVersionStage(
	ctx context.Context, name, version string, stage models.ModelVersionStage, archiveExistingVersions bool,
) (*entities.ModelVersion, *contract.Error)

func (*ModelRegistrySQLStore) UpdateModelVersion

func (m *ModelRegistrySQLStore) UpdateModelVersion(
	ctx context.Context, name, version, description string,
) (*entities.ModelVersion, *contract.Error)

func (*ModelRegistrySQLStore) UpdateRegisteredModel

func (m *ModelRegistrySQLStore) UpdateRegisteredModel(
	ctx context.Context, name, description string,
) (*entities.RegisteredModel, *contract.Error)

Directories

Path Synopsis

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL