models

type AlembicVersion struct {
	VersionNum *string `gorm:"column:version_num;primaryKey"`
}

type Dataset struct {
	ID           string `gorm:"column:dataset_uuid;not null"`
	ExperimentID int32  `gorm:"column:experiment_id;primaryKey"`
	Name         string `gorm:"column:name;primaryKey"`
	Digest       string `gorm:"column:digest;primaryKey"`
	SourceType   string `gorm:"column:dataset_source_type;not null"`
	Source       string `gorm:"column:dataset_source;not null"`
	Schema       string `gorm:"column:dataset_schema"`
	Profile      string `gorm:"column:dataset_profile"`
}

type Experiment struct {
	ID               int32          `gorm:"column:experiment_id;primaryKey;autoIncrement:true"`
	Name             string         `gorm:"column:name;not null"`
	ArtifactLocation string         `gorm:"column:artifact_location"`
	LifecycleStage   LifecycleStage `gorm:"column:lifecycle_stage"`
	CreationTime     int64          `gorm:"column:creation_time"`
	LastUpdateTime   int64          `gorm:"column:last_update_time"`
	Tags             []ExperimentTag
	Runs             []Run
}

type ExperimentTag struct {
	Key          string `gorm:"column:key;primaryKey"`
	Value        string `gorm:"column:value"`
	ExperimentID int32  `gorm:"column:experiment_id;primaryKey"`
}

type Input struct {
	ID              string     `gorm:"column:input_uuid;not null"`
	SourceType      string     `gorm:"column:source_type;primaryKey"`
	SourceID        string     `gorm:"column:source_id;primaryKey"`
	DestinationType string     `gorm:"column:destination_type;primaryKey"`
	DestinationID   string     `gorm:"column:destination_id;primaryKey"`
	Tags            []InputTag `gorm:"foreignKey:InputID;references:ID"`
	Dataset         Dataset    `gorm:"foreignKey:ID;references:SourceID"`
}

type InputTag struct {
	Key     string `gorm:"column:name;primaryKey"`
	Value   string `gorm:"column:value;not null"`
	InputID string `gorm:"column:input_uuid;primaryKey"`
}

type LatestMetric struct {
	Key       string  `gorm:"column:key;primaryKey"`
	Value     float64 `gorm:"column:value;not null"`
	Timestamp int64   `gorm:"column:timestamp"`
	Step      int64   `gorm:"column:step;not null"`
	IsNaN     bool    `gorm:"column:is_nan;not null"`
	RunID     string  `gorm:"column:run_uuid;primaryKey"`
}

type LifecycleStage string

type LoggedModelMetric struct {
	ModelID           string  `gorm:"column:model_id"`
	MetricName        string  `gorm:"column:metric_name"`
	MetricTimestampMs int64   `gorm:"column:metric_timestamp_ms"`
	MetricStep        int64   `gorm:"column:metric_step"`
	MetricValue       float64 `gorm:"column:metric_value"`
	ExperimentID      int     `gorm:"column:experiment_id"`
	RunID             string  `gorm:"column:run_id"`
	DatasetUUID       string  `gorm:"column:dataset_uuid"`
	DatasetName       string  `gorm:"column:dataset_name"`
	DatasetDigest     string  `gorm:"column:dataset_digest"`
}

type Metric struct {
	Key       string  `gorm:"column:key;primaryKey"`
	Value     float64 `gorm:"column:value;primaryKey"`
	Timestamp int64   `gorm:"column:timestamp;primaryKey"`
	RunID     string  `gorm:"column:run_uuid;primaryKey"`
	Step      int64   `gorm:"column:step;primaryKey"`
	IsNaN     bool    `gorm:"column:is_nan;primaryKey"`
}

type Output struct {
	ID              string `gorm:"column:input_uuid;not null"`
	Step            int64  `gorm:"column:step"`
	SourceType      string `gorm:"column:source_type;primaryKey"`
	SourceID        string `gorm:"column:source_id;primaryKey"`
	DestinationType string `gorm:"column:destination_type;primaryKey"`
	DestinationID   string `gorm:"column:destination_id;primaryKey"`
}

type Param struct {
	Key   string         `gorm:"column:key;primaryKey"`
	Value sql.NullString `gorm:"column:value;not null"`
	RunID string         `gorm:"column:run_uuid;primaryKey"`
}

type Run struct {
	ID             string         `gorm:"column:run_uuid;primaryKey"`
	Name           string         `gorm:"column:name"`
	SourceType     SourceType     `gorm:"column:source_type"`
	SourceName     string         `gorm:"column:source_name"`
	EntryPointName string         `gorm:"column:entry_point_name"`
	UserID         string         `gorm:"column:user_id"`
	Status         RunStatus      `gorm:"column:status"`
	StartTime      int64          `gorm:"column:start_time"`
	EndTime        sql.NullInt64  `gorm:"column:end_time"`
	SourceVersion  string         `gorm:"column:source_version"`
	LifecycleStage LifecycleStage `gorm:"column:lifecycle_stage"`
	ArtifactURI    string         `gorm:"column:artifact_uri"`
	ExperimentID   int32          `gorm:"column:experiment_id"`
	DeletedTime    sql.NullInt64  `gorm:"column:deleted_time"`
	Params         []Param
	Tags           []Tag
	Metrics        []Metric
	LatestMetrics  []LatestMetric
	Inputs         []Input  `gorm:"foreignKey:DestinationID"`
	Outputs        []Output `gorm:"foreignKey:DestinationID"`
}

type RunStatus string

type SourceType string

type Tag struct {
	Key   string `gorm:"column:key;primaryKey"`
	Value string `gorm:"column:value"`
	RunID string `gorm:"column:run_uuid;primaryKey"`
}

type TraceInfo struct {
	RequestID            string                 `gorm:"column:request_id;primaryKey"`
	ExperimentID         string                 `gorm:"column:experiment_id"`
	ClientRequestID      sql.NullString         `gorm:"column:client_request_id"`
	RequestPreview       sql.NullString         `gorm:"column:request_preview"`
	ResponsePreview      sql.NullString         `gorm:"column:request_preview"`
	TimestampMS          int64                  `gorm:"column:timestamp_ms"`
	ExecutionTimeMS      sql.NullInt64          `gorm:"column:execution_time_ms"`
	Status               string                 `gorm:"column:status"`
	Tags                 []TraceTag             `gorm:"foreignKey:RequestID"`
	TraceRequestMetadata []TraceRequestMetadata `gorm:"foreignKey:RequestID"`
}

type TraceRequestMetadata struct {
	Key       string `gorm:"column:key;primaryKey"`
	Value     string `gorm:"column:value"`
	RequestID string `gorm:"column:request_id;primaryKey"`
}

type TraceTag struct {
	Key       string `gorm:"column:key;primaryKey"`
	Value     string `gorm:"column:value"`
	RequestID string `gorm:"column:request_id;primaryKey"`
}