models

type Artifact struct {
	ModelArtifact *ModelArtifact
	DocArtifact   *DocArtifact
	DataSet       *DataSet
	Metric        *Metric
	Parameter     *Parameter
}

type ArtifactListOptions struct {
	Pagination
	Name             *string
	ExternalID       *string
	ParentResourceID *int32
	ArtifactType     *string
}

type ArtifactRepository interface {
	GetByID(id int32) (Artifact, error)
	List(listOptions ArtifactListOptions) (*ListWrapper[Artifact], error)
}

type BaseEntity[T any] struct {
	ID               *int32
	TypeID           *int32
	Attributes       *T
	Properties       *[]Properties
	CustomProperties *[]Properties
}

type DataSet interface {
	Entity[DataSetAttributes]
}

type DataSetAttributes struct {
	Name                     *string
	URI                      *string
	State                    *string
	ArtifactType             *string
	ExternalID               *string
	CreateTimeSinceEpoch     *int64
	LastUpdateTimeSinceEpoch *int64
}

type DataSetImpl = BaseEntity[DataSetAttributes]

type DataSetListOptions struct {
	Pagination
	Name             *string
	ExternalID       *string
	ParentResourceID *int32
}

type DataSetRepository interface {
	GetByID(id int32) (DataSet, error)
	List(listOptions DataSetListOptions) (*ListWrapper[DataSet], error)
	Save(dataSet DataSet, parentResourceID *int32) (DataSet, error)
}

type DocArtifact interface {
	Entity[DocArtifactAttributes]
}

type DocArtifactAttributes struct {
	Name                     *string
	URI                      *string
	State                    *string
	ArtifactType             *string
	ExternalID               *string
	CreateTimeSinceEpoch     *int64
	LastUpdateTimeSinceEpoch *int64
}

type DocArtifactImpl = BaseEntity[DocArtifactAttributes]

type DocArtifactListOptions struct {
	Pagination
	Name             *string
	ExternalID       *string
	ParentResourceID *int32
}

type DocArtifactRepository interface {
	GetByID(id int32) (DocArtifact, error)
	List(listOptions DocArtifactListOptions) (*ListWrapper[DocArtifact], error)
	Save(docArtifact DocArtifact, parentResourceID *int32) (DocArtifact, error)
}

type Entity[T any] interface {
	GetID() *int32
	SetID(int32)
	GetTypeID() *int32
	SetTypeID(int32)
	GetAttributes() *T
	GetProperties() *[]Properties
	GetCustomProperties() *[]Properties
}

type Experiment interface {
	Entity[ExperimentAttributes]
}

type ExperimentAttributes struct {
	Name                     *string
	ExternalID               *string
	CreateTimeSinceEpoch     *int64
	LastUpdateTimeSinceEpoch *int64
}

type ExperimentImpl = BaseEntity[ExperimentAttributes]

type ExperimentListOptions struct {
	Pagination
	Name       *string
	ExternalID *string
}

type ExperimentRepository interface {
	GetByID(id int32) (Experiment, error)
	List(listOptions ExperimentListOptions) (*ListWrapper[Experiment], error)
	Save(experiment Experiment) (Experiment, error)
}

type ExperimentRun interface {
	Entity[ExperimentRunAttributes]
}

type ExperimentRunAttributes struct {
	Name                     *string
	ExternalID               *string
	CreateTimeSinceEpoch     *int64
	LastUpdateTimeSinceEpoch *int64
}

type ExperimentRunImpl = BaseEntity[ExperimentRunAttributes]

type ExperimentRunListOptions struct {
	Pagination
	Name         *string
	ExternalID   *string
	ExperimentID *int32
}

type ExperimentRunRepository interface {
	GetByID(id int32) (ExperimentRun, error)
	List(listOptions ExperimentRunListOptions) (*ListWrapper[ExperimentRun], error)
	Save(experimentRun ExperimentRun, experimentID *int32) (ExperimentRun, error)
}

type InferenceService interface {
	Entity[InferenceServiceAttributes]
}

type InferenceServiceAttributes struct {
	Name                     *string
	ExternalID               *string
	CreateTimeSinceEpoch     *int64
	LastUpdateTimeSinceEpoch *int64
}

type InferenceServiceImpl = BaseEntity[InferenceServiceAttributes]

type InferenceServiceListOptions struct {
	Pagination
	Name             *string
	ExternalID       *string
	ParentResourceID *int32
	Runtime          *string
}

type InferenceServiceRepository interface {
	GetByID(id int32) (InferenceService, error)
	List(listOptions InferenceServiceListOptions) (*ListWrapper[InferenceService], error)
	Save(model InferenceService) (InferenceService, error)
}

type ListWrapper[T any] struct {
	Items         []T
	NextPageToken string
	PageSize      int32
	Size          int32
}

type Metric interface {
	Entity[MetricAttributes]
}

type MetricAttributes struct {
	Name                     *string
	URI                      *string
	State                    *string
	ArtifactType             *string
	ExternalID               *string
	CreateTimeSinceEpoch     *int64
	LastUpdateTimeSinceEpoch *int64
}

type MetricHistory interface {
	Entity[MetricHistoryAttributes]
}

type MetricHistoryAttributes struct {
	Name                     *string
	URI                      *string
	State                    *string
	ArtifactType             *string
	ExternalID               *string
	CreateTimeSinceEpoch     *int64
	LastUpdateTimeSinceEpoch *int64
}

type MetricHistoryImpl = BaseEntity[MetricHistoryAttributes]

type MetricHistoryListOptions struct {
	Pagination
	Name            *string
	ExternalID      *string
	ExperimentRunID *int32
	StepIds         *string
}

type MetricHistoryRepository interface {
	GetByID(id int32) (MetricHistory, error)
	List(listOptions MetricHistoryListOptions) (*ListWrapper[MetricHistory], error)
	Save(metricHistory MetricHistory, experimentRunID *int32) (MetricHistory, error)
}

type MetricImpl = BaseEntity[MetricAttributes]

type MetricListOptions struct {
	Pagination
	Name             *string
	ExternalID       *string
	ParentResourceID *int32
}

type MetricRepository interface {
	GetByID(id int32) (Metric, error)
	List(listOptions MetricListOptions) (*ListWrapper[Metric], error)
	Save(metric Metric, parentResourceID *int32) (Metric, error)
}

type ModelArtifact interface {
	Entity[ModelArtifactAttributes]
}

type ModelArtifactAttributes struct {
	Name                     *string
	URI                      *string
	State                    *string
	ArtifactType             *string
	ExternalID               *string
	CreateTimeSinceEpoch     *int64
	LastUpdateTimeSinceEpoch *int64
}

type ModelArtifactImpl = BaseEntity[ModelArtifactAttributes]

type ModelArtifactListOptions struct {
	Pagination
	Name             *string
	ExternalID       *string
	ParentResourceID *int32
}

type ModelArtifactRepository interface {
	GetByID(id int32) (ModelArtifact, error)
	List(listOptions ModelArtifactListOptions) (*ListWrapper[ModelArtifact], error)
	Save(modelArtifact ModelArtifact, parentResourceID *int32) (ModelArtifact, error)
}

type ModelVersion interface {
	Entity[ModelVersionAttributes]
}

type ModelVersionAttributes struct {
	Name                     *string
	ExternalID               *string
	CreateTimeSinceEpoch     *int64
	LastUpdateTimeSinceEpoch *int64
}

type ModelVersionImpl = BaseEntity[ModelVersionAttributes]

type ModelVersionListOptions struct {
	Pagination
	Name             *string
	ExternalID       *string
	ParentResourceID *int32
}

type ModelVersionRepository interface {
	GetByID(id int32) (ModelVersion, error)
	List(listOptions ModelVersionListOptions) (*ListWrapper[ModelVersion], error)
	Save(model ModelVersion) (ModelVersion, error)
}

type Pagination struct {
	PageSize      *int32  `json:"pageSize,omitempty"`
	OrderBy       *string `json:"orderBy,omitempty"`
	SortOrder     *string `json:"sortOrder,omitempty"`
	NextPageToken *string `json:"nextPageToken,omitempty"`
	FilterQuery   *string `json:"filterQuery,omitempty"`
}

type Parameter interface {
	Entity[ParameterAttributes]
}

type ParameterAttributes struct {
	Name                     *string
	URI                      *string
	State                    *string
	ArtifactType             *string
	ExternalID               *string
	CreateTimeSinceEpoch     *int64
	LastUpdateTimeSinceEpoch *int64
}

type ParameterImpl = BaseEntity[ParameterAttributes]

type ParameterListOptions struct {
	Pagination
	Name             *string
	ExternalID       *string
	ParentResourceID *int32
}

type ParameterRepository interface {
	GetByID(id int32) (Parameter, error)
	List(listOptions ParameterListOptions) (*ListWrapper[Parameter], error)
	Save(parameter Parameter, parentResourceID *int32) (Parameter, error)
}

type Properties struct {
	Name             string
	IsCustomProperty bool
	IntValue         *int32
	DoubleValue      *float64
	StringValue      *string
	BoolValue        *bool
	ByteValue        *[]byte
	ProtoValue       *[]byte
}

type RegisteredModel interface {
	Entity[RegisteredModelAttributes]
}

type RegisteredModelAttributes struct {
	Name                     *string
	ExternalID               *string
	CreateTimeSinceEpoch     *int64
	LastUpdateTimeSinceEpoch *int64
}

type RegisteredModelImpl = BaseEntity[RegisteredModelAttributes]

type RegisteredModelListOptions struct {
	Pagination
	Name       *string
	ExternalID *string
}

type RegisteredModelRepository interface {
	GetByID(id int32) (RegisteredModel, error)
	List(listOptions RegisteredModelListOptions) (*ListWrapper[RegisteredModel], error)
	Save(model RegisteredModel) (RegisteredModel, error)
}

type ServeModel interface {
	Entity[ServeModelAttributes]
}

type ServeModelAttributes struct {
	Name                     *string
	ExternalID               *string
	LastKnownState           *string
	CreateTimeSinceEpoch     *int64
	LastUpdateTimeSinceEpoch *int64
}

type ServeModelImpl = BaseEntity[ServeModelAttributes]

type ServeModelListOptions struct {
	Pagination
	Name               *string
	ExternalID         *string
	InferenceServiceID *int32
}

type ServeModelRepository interface {
	GetByID(id int32) (ServeModel, error)
	List(listOptions ServeModelListOptions) (*ListWrapper[ServeModel], error)
	Save(serveModel ServeModel, inferenceServiceID *int32) (ServeModel, error)
}

type ServingEnvironment interface {
	Entity[ServingEnvironmentAttributes]
}

type ServingEnvironmentAttributes struct {
	Name                     *string
	ExternalID               *string
	CreateTimeSinceEpoch     *int64
	LastUpdateTimeSinceEpoch *int64
}

type ServingEnvironmentImpl = BaseEntity[ServingEnvironmentAttributes]

type ServingEnvironmentListOptions struct {
	Pagination
	Name       *string
	ExternalID *string
}

type ServingEnvironmentRepository interface {
	GetByID(id int32) (ServingEnvironment, error)
	List(listOptions ServingEnvironmentListOptions) (*ListWrapper[ServingEnvironment], error)
	Save(model ServingEnvironment) (ServingEnvironment, error)
}

type Type interface {
	GetID() *int32
	GetAttributes() *TypeAttributes
}

type TypeAttributes struct {
	Name        *string
	Version     *string
	TypeKind    *int32
	Description *string
	InputType   *string
	OutputType  *string
	ExternalID  *string
}

type TypeImpl struct {
	ID         *int32
	Attributes *TypeAttributes
}

type TypeProperty interface {
	GetTypeID() int32
	GetName() string
	GetDataType() *int32
}

type TypePropertyImpl struct {
	TypeID   int32
	Name     string
	DataType *int32
}

type TypePropertyRepository interface {
	// Save stores a type property if it doesn't exist.
	Save(tp TypeProperty) (TypeProperty, error)
}

type TypeRepository interface {
	// GetAll returns every registered type.
	GetAll() ([]Type, error)

	// Save updates a type, if the definition differs from what's stored.
	Save(t Type) (Type, error)
}