index

Documentation

Index

• type DB
  • func New(dsn string, autoMigrate bool) (*DB, error)
  • func (db *DB) AutoMigrate() error
  • func (db *DB) Check(w http.ResponseWriter, _ *http.Request)
  • func (db *DB) Close() error
  • func (db *DB) ExportDatasetsToFile(ctx context.Context, path string, ids ...string) error
  • func (db *DB) ImportDatasetsFromFile(ctx context.Context, path string) error
  • func (db *DB) WithContext(ctx context.Context) *gorm.DB
• type Dataset
  • func GetDataset(db *gorm.DB, id string) (*Dataset, error)
• type Document
• type File
• type FileMetadata

Types

type DB

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

func New

func New(dsn string, autoMigrate bool) (*DB, error)

func (*DB) AutoMigrate

func (db *DB) AutoMigrate() error

func (*DB) Check

func (db *DB) Check(w http.ResponseWriter, _ *http.Request)

func (*DB) Close

func (db *DB) Close() error

func (*DB) ExportDatasetsToFile

func (db *DB) ExportDatasetsToFile(ctx context.Context, path string, ids ...string) error

func (*DB) ImportDatasetsFromFile

func (db *DB) ImportDatasetsFromFile(ctx context.Context, path string) error

func (*DB) WithContext

func (db *DB) WithContext(ctx context.Context) *gorm.DB

type Dataset

type Dataset struct {
	ID             string `gorm:"primaryKey" json:"id"`
	EmbedDimension int    `json:"embed_dim,omitempty"`
	Files          []File `gorm:"foreignKey:Dataset;references:ID;constraint:OnDelete:CASCADE;"`
}

Dataset refers to a VectorDB data space. @Description Dataset refers to a VectorDB data space.

func GetDataset

func GetDataset(db *gorm.DB, id string) (*Dataset, error)

type Document

type Document struct {
	ID      string `gorm:"primaryKey" json:"id"`
	Dataset string `gorm:"primaryKey" json:"dataset"` // Foreign key to Dataset, part of composite primary key with FileID
	FileID  string `gorm:"primaryKey" json:"file_id"` // Foreign key to File, part of composite primary key with Dataset
}

type File

type File struct {
	ID        string     `gorm:"primaryKey" json:"id"`
	Dataset   string     `gorm:"primaryKey" json:"dataset"` // Foreign key to Dataset
	Documents []Document `gorm:"foreignKey:FileID,Dataset;references:ID,Dataset;constraint:OnDelete:CASCADE;"`
	// File metadata, commonly used for deduplication
	FileMetadata `json:",inline"`
}

type FileMetadata

type FileMetadata struct {
	Name         string    `json:"name"`
	AbsolutePath string    `json:"absolute_path"`
	Size         int64     `json:"size"`
	ModifiedAt   time.Time `json:"modified_at"`
}