entities

type Dataset struct {
	Name       string
	Digest     string
	SourceType string
	Source     string
	Schema     string
	Profile    string
}

type DatasetInput struct {
	Tags    []*InputTag
	Dataset *Dataset
}

type Experiment struct {
	Name             string
	ExperimentID     string
	ArtifactLocation string
	LifecycleStage   string
	LastUpdateTime   int64
	CreationTime     int64
	Tags             []*ExperimentTag
}

type ExperimentTag struct {
	Key   string
	Value string
}

type InputTag struct {
	Key   string
	Value string
}

type Metric struct {
	Key           string
	Value         float64
	Timestamp     int64
	Step          int64
	IsNaN         bool
	ModelID       string
	DatasetName   string
	DatasetDigest string
}

type ModelInput struct {
	ModelID string
}

type ModelOutput struct {
	Step    int64
	ModelID string
}

type ModelVersion struct {
	Name            string
	Version         int32
	CreationTime    int64
	LastUpdatedTime int64
	Description     string
	UserID          string
	CurrentStage    string
	Source          string
	RunID           string
	Status          string
	StatusMessage   string
	RunLink         string
	StorageLocation string
	Tags            []*ModelVersionTag
	Aliases         []*RegisteredModelAlias
}

type ModelVersionTag struct {
	Key   string
	Value string
}

type Param struct {
	Key   string
	Value *string
}

type RegisteredModel struct {
	Name            string
	Tags            []*RegisteredModelTag
	Aliases         []*RegisteredModelAlias
	Versions        []*ModelVersion
	Description     *string
	CreationTime    int64
	LastUpdatedTime int64
}

type RegisteredModelAlias struct {
	Alias   string
	Version string
}

type RegisteredModelTag struct {
	Key   string
	Value string
}

type Run struct {
	Info    *RunInfo
	Data    *RunData
	Inputs  *RunInputs
	Outputs *RunOutputs
}

type RunData struct {
	Tags    []*RunTag
	Params  []*Param
	Metrics []*Metric
}

type RunInfo struct {
	RunID          string
	RunUUID        string
	RunName        string
	ExperimentID   int32
	UserID         string
	Status         string
	StartTime      int64
	EndTime        *int64
	ArtifactURI    string
	LifecycleStage string
}

type RunInputs struct {
	ModelInputs   []*ModelInput
	DatasetInputs []*DatasetInput
}

type RunOutputs struct {
	ModelOutputs []*ModelOutput
}

type RunTag struct {
	Key   string
	Value string
}

type TraceInfo struct {
	RequestID            string
	Status               string
	ExperimentID         string
	TimestampMS          int64
	ExecutionTimeMS      *int64
	Tags                 []*TraceTag
	TraceRequestMetadata []*TraceRequestMetadata
}

type TraceInfoV3 struct {
	Status               string
	RequestID            string
	ClientRequestID      *string
	ExperimentID         string
	RequestPreview       *string
	ResponsePreview      *string
	TimestampMS          int64
	ExecutionTimeMS      *int64
	Tags                 []*TraceTag
	TraceRequestMetadata []*TraceRequestMetadata
}

type TraceRequestMetadata struct {
	Key       string
	Value     string
	RequestID string
}

type TraceTag struct {
	Key       string
	Value     string
	RequestID string
}