db

Documentation

Index

• func DeleteCollectionText(ctx context.Context, collectionName, namespace, key string) error
• func DeleteCollectionTexts(ctx context.Context, collectionName string) error
• func DeleteCollectionVector(ctx context.Context, searchMethodName string, textId int64) error
• func DeleteCollectionVectors(ctx context.Context, collectionName, searchMethodName, namespace string) error
• func GetTx(ctx context.Context) (pgx.Tx, error)
• func GetUniqueNamespaces(ctx context.Context, collectionName string) ([]string, error)
• func Initialize(ctx context.Context)
• func QueryCollectionTextsFromCheckpoint(ctx context.Context, collection, namespace string, textCheckpointId int64) ([]int64, []string, []string, [][]string, error)
• func QueryCollectionVectorsFromCheckpoint(ctx context.Context, collectionName, searchMethodName, namespace string, ...) ([]int64, []int64, []string, [][]float32, error)
• func Stop(ctx context.Context)
• func WithTx(ctx context.Context, fn func(pgx.Tx) error) error
• func WriteCollectionText(ctx context.Context, collectionName, namespace, key, text string, ...) (id int64, err error)
• func WriteCollectionTexts(ctx context.Context, collectionName, namespace string, keys, texts []string, ...) ([]int64, error)
• func WriteCollectionVector(ctx context.Context, searchMethodName string, textId int64, vector []float32) (vectorId int64, key string, err error)
• func WriteCollectionVectors(ctx context.Context, searchMethodName string, textIds []int64, ...) ([]int64, []string, error)
• func WriteInferenceHistory(ctx context.Context, model *manifest.ModelInfo, input, output any, ...)
• func WriteInferenceHistoryToDB(ctx context.Context, batch []inferenceHistory)
• func WritePluginInfo(ctx context.Context, plugin *plugins.Plugin)

Functions

func DeleteCollectionText

func DeleteCollectionText(ctx context.Context, collectionName, namespace, key string) error

func DeleteCollectionTexts

func DeleteCollectionTexts(ctx context.Context, collectionName string) error

func DeleteCollectionVector

func DeleteCollectionVector(ctx context.Context, searchMethodName string, textId int64) error

func DeleteCollectionVectors

func DeleteCollectionVectors(ctx context.Context, collectionName, searchMethodName, namespace string) error

func GetTx

func GetTx(ctx context.Context) (pgx.Tx, error)

func GetUniqueNamespaces

func GetUniqueNamespaces(ctx context.Context, collectionName string) ([]string, error)

func Initialize

func Initialize(ctx context.Context)

func QueryCollectionTextsFromCheckpoint

func QueryCollectionTextsFromCheckpoint(ctx context.Context, collection, namespace string, textCheckpointId int64) ([]int64, []string, []string, [][]string, error)

func QueryCollectionVectorsFromCheckpoint

func QueryCollectionVectorsFromCheckpoint(ctx context.Context, collectionName, searchMethodName, namespace string, vecCheckpointId int64) ([]int64, []int64, []string, [][]float32, error)

func Stop

func Stop(ctx context.Context)

func WithTx

func WithTx(ctx context.Context, fn func(pgx.Tx) error) error

func WriteCollectionText

func WriteCollectionText(ctx context.Context, collectionName, namespace, key, text string, labels []string) (id int64, err error)

func WriteCollectionTexts

func WriteCollectionTexts(ctx context.Context, collectionName, namespace string, keys, texts []string, labelsArr [][]string) ([]int64, error)

func WriteCollectionVector

func WriteCollectionVector(ctx context.Context, searchMethodName string, textId int64, vector []float32) (vectorId int64, key string, err error)

func WriteCollectionVectors

func WriteCollectionVectors(ctx context.Context, searchMethodName string, textIds []int64, vectors [][]float32) ([]int64, []string, error)

func WriteInferenceHistory

func WriteInferenceHistory(ctx context.Context, model *manifest.ModelInfo, input, output any, start, end time.Time)

func WriteInferenceHistoryToDB

func WriteInferenceHistoryToDB(ctx context.Context, batch []inferenceHistory)

func WritePluginInfo

func WritePluginInfo(ctx context.Context, plugin *plugins.Plugin)