sql

package
v0.1.0 Latest Latest
Warning

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

 Go to latest Published: Nov 14, 2024 License: Apache-2.0 Imports: 23 Imported by: 0

Documentation

Index

Constants

View Source 
const (
	LikeExpression           = "LIKE"
	ILikeExpression          = "ILIKE"
	EqualExpression          = "="
	NotEqualExpression       = "!="
	LessExpression           = "<"
	LessOrEqualExpression    = "<="
	GreaterExpression        = ">"
	GreaterOrEqualExpression = ">="
)

supported expression list.

View Source 
const (
	TraceFolderName     = "traces"
	ArtifactsFolderName = "artifacts"
)
View Source 
const (
	ArtifactFolderName  = "artifacts"
	RunNameIntegerScale = 3
	RunNameMaxLength    = 20
)
View Source 
const (
	BatchSize = 100
)
View Source 
const (
	MlflowArtifactLocation = "mlflow.artifactLocation"
)
View Source 
const RunIDMaxLength = 32

Variables

View Source 
var ErrInvalidOrderClauseInput = errors.New("input string is empty or only contains quote characters")

Functions

func GetTraceArtifactLocationTag 

func GetTraceArtifactLocationTag(
	experiment *entities.Experiment, requestID string,
) (models.TraceTag, error)

Types

type PageToken 

type PageToken struct {
	Offset int32 `json:"offset"`
}

PageToken.

type TrackingSQLStore 

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

func NewTrackingSQLStore 

func NewTrackingSQLStore(ctx context.Context, config *config.Config) (*TrackingSQLStore, error)

func (TrackingSQLStore) CreateExperiment 

func (s TrackingSQLStore) CreateExperiment(
	ctx context.Context,
	name string,
	artifactLocation string,
	tags []*entities.ExperimentTag,
) (string, *contract.Error)

func (TrackingSQLStore) CreateRun 

func (s TrackingSQLStore) CreateRun(
	ctx context.Context,
	experimentID, userID string,
	startTime int64,
	tags []*entities.RunTag,
	runName string,
) (*entities.Run, *contract.Error)

func (TrackingSQLStore) DeleteExperiment 

func (s TrackingSQLStore) DeleteExperiment(ctx context.Context, id string) *contract.Error

func (TrackingSQLStore) DeleteRun 

func (s TrackingSQLStore) DeleteRun(ctx context.Context, runID string) *contract.Error

func (TrackingSQLStore) DeleteTag 

func (s TrackingSQLStore) DeleteTag(
	ctx context.Context, runID, key string,
) *contract.Error

func (TrackingSQLStore) DeleteTraceTag 

func (s TrackingSQLStore) DeleteTraceTag(
	ctx context.Context, tag *entities.TraceTag,
) *contract.Error

func (TrackingSQLStore) DeleteTraces 

func (s TrackingSQLStore) DeleteTraces(
	ctx context.Context,
	experimentID string,
	maxTimestampMillis int64,
	maxTraces int32,
	requestIDs []string,
) (int32, *contract.Error)

func (TrackingSQLStore) Destroy 

func (s TrackingSQLStore) Destroy() error

func (TrackingSQLStore) EndTrace 

func (s TrackingSQLStore) EndTrace(
	ctx context.Context,
	reqeustID string,
	timestampMS int64,
	status string,
	metadata []*entities.TraceRequestMetadata,
	tags []*entities.TraceTag,
) (*entities.TraceInfo, error)

func (TrackingSQLStore) GetExperiment 

func (s TrackingSQLStore) GetExperiment(ctx context.Context, id string) (*entities.Experiment, *contract.Error)

func (TrackingSQLStore) GetExperimentByName 

func (s TrackingSQLStore) GetExperimentByName(
	ctx context.Context, name string,
) (*entities.Experiment, *contract.Error)

func (TrackingSQLStore) GetMetricHistory 

func (s TrackingSQLStore) GetMetricHistory(
	ctx context.Context, runID, metricKey string,
) ([]*entities.Metric, *contract.Error)

func (TrackingSQLStore) GetRun 

func (s TrackingSQLStore) GetRun(ctx context.Context, runID string) (*entities.Run, *contract.Error)

func (TrackingSQLStore) GetRunTag 

func (s TrackingSQLStore) GetRunTag(
	ctx context.Context, runID, tagKey string,
) (*entities.RunTag, *contract.Error)

func (TrackingSQLStore) GetTraceInfo 

func (s TrackingSQLStore) GetTraceInfo(ctx context.Context, reqeustID string) (*entities.TraceInfo, *contract.Error)

func (TrackingSQLStore) GetTraceTag 

func (s TrackingSQLStore) GetTraceTag(
	ctx context.Context, requestID, key string,
) (*entities.TraceTag, *contract.Error)

func (TrackingSQLStore) LogBatch 

func (s TrackingSQLStore) LogBatch(
	ctx context.Context, runID string, metrics []*entities.Metric, params []*entities.Param, tags []*entities.RunTag,
) *contract.Error

func (TrackingSQLStore) LogInputs 

func (s TrackingSQLStore) LogInputs(
	ctx context.Context, runID string, datasets []*entities.DatasetInput,
) *contract.Error

func (TrackingSQLStore) LogMetric 

func (s TrackingSQLStore) LogMetric(ctx context.Context, runID string, metric *entities.Metric) *contract.Error

func (TrackingSQLStore) LogParam 

func (s TrackingSQLStore) LogParam(
	ctx context.Context, runID string, param *entities.Param,
) *contract.Error

func (TrackingSQLStore) RenameExperiment 

func (s TrackingSQLStore) RenameExperiment(
	ctx context.Context, experimentID, name string,
) *contract.Error

func (TrackingSQLStore) RestoreExperiment 

func (s TrackingSQLStore) RestoreExperiment(ctx context.Context, id string) *contract.Error

func (TrackingSQLStore) RestoreRun 

func (s TrackingSQLStore) RestoreRun(ctx context.Context, runID string) *contract.Error

func (TrackingSQLStore) SearchExperiments 

func (s TrackingSQLStore) SearchExperiments(
	ctx context.Context,
	experimentViewType protos.ViewType,
	maxResults int64,
	filter string,
	orderBy []string,
	pageToken string,
) ([]*entities.Experiment, string, *contract.Error)

func (TrackingSQLStore) SearchRuns 

func (s TrackingSQLStore) SearchRuns(
	ctx context.Context,
	experimentIDs []string, filter string,
	runViewType protos.ViewType, maxResults int, orderBy []string, pageToken string,
) ([]*entities.Run, string, *contract.Error)

func (TrackingSQLStore) SetExperimentTag 

func (s TrackingSQLStore) SetExperimentTag(
	ctx context.Context, experimentID, key, value string,
) *contract.Error

func (TrackingSQLStore) SetTag 

func (s TrackingSQLStore) SetTag(
	ctx context.Context, runID, key, value string,
) *contract.Error

func (TrackingSQLStore) SetTrace 

func (s TrackingSQLStore) SetTrace(
	ctx context.Context,
	experimentID string,
	timestampMS int64,
	metadata []*entities.TraceRequestMetadata,
	tags []*entities.TraceTag,
) (*entities.TraceInfo, error)

func (TrackingSQLStore) SetTraceTag 

func (s TrackingSQLStore) SetTraceTag(
	ctx context.Context, requestID, key, value string,
) error

func (TrackingSQLStore) UpdateRun 

func (s TrackingSQLStore) UpdateRun(
	ctx context.Context,
	runID string,
	runStatus string,
	endTime *int64,
	runName string,
) *contract.Error

Source Files

View all Source files

Directories

Path Synopsis

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.