store

package
v1.0.0 Latest Latest
Warning

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

Go to latest
Published: Mar 22, 2026 License: MIT Imports: 9 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func GetDB

func GetDB() (*sql.DB, error)

func GetDocumentSummaryEmbedding

func GetDocumentSummaryEmbedding(db *sql.DB, summaryID, embeddingModelID string) ([]float32, error)

func GetSummaryEmbeddingByPath

func GetSummaryEmbeddingByPath(db *sql.DB, path, summaryModelID, embeddingModelID string) ([]float32, error)

func HasDocumentChunks

func HasDocumentChunks(db *sql.DB, documentID, embeddingModelID string) (bool, error)

func InitDB

func InitDB() (*sql.DB, error)

func InsertEmbeddingModel

func InsertEmbeddingModel(db *sql.DB, id string, embeddingDim int) (string, error)

func InsertSummaryModel

func InsertSummaryModel(db *sql.DB, id string) (string, error)

func Migrate

func Migrate(db *sql.DB) error

func Open

func Open(path string) (*sql.DB, error)

func ReplaceDocumentChunks

func ReplaceDocumentChunks(db *sql.DB, documentID, embeddingModelID string, chunks []ChunkInput) error

func UpsertDocumentSummaryEmbedding

func UpsertDocumentSummaryEmbedding(db *sql.DB, summaryID, embeddingModelID string, embedding []float32) error

Types

type ChunkInput

type ChunkInput struct {
	ChunkIndex int
	StartLine  int
	EndLine    int
	Content    string
	Embedding  []float32
}

type Document

type Document struct {
	ID          string
	Path        string
	ContentHash string
}

func GetDocumentByPath

func GetDocumentByPath(db *sql.DB, path string) (*Document, error)

func UpsertDocument

func UpsertDocument(db *sql.DB, path, contentHash string) (Document, bool, error)

type DocumentRecord

type DocumentRecord struct {
	ID     string
	Path   string
	Models []string
}

func ListDocuments

func ListDocuments(db *sql.DB, modelFilter string, limit int) ([]DocumentRecord, error)

type SearchRow

type SearchRow struct {
	DocumentID string
	Path       string
	Distance   float64
}

func MergeSearchRows

func MergeSearchRows(limit int, groups ...[]SearchRow) []SearchRow

func SearchDocumentChunks

func SearchDocumentChunks(
	db *sql.DB,
	embeddingModelID string,
	query []float32,
	limit int,
	excludePath string,
) ([]SearchRow, error)

func SearchSummaryEmbeddings

func SearchSummaryEmbeddings(
	db *sql.DB,
	summaryModelID string,
	embeddingModelID string,
	query []float32,
	limit int,
	excludePath string,
) ([]SearchRow, error)

type SummaryRecord

type SummaryRecord struct {
	ID             string
	DocumentID     string
	SummaryModelID string
	Content        string
}

func GetDocumentSummary

func GetDocumentSummary(db *sql.DB, documentID, summaryModelID string) (*SummaryRecord, error)

func UpsertDocumentSummary

func UpsertDocumentSummary(db *sql.DB, documentID, summaryModelID, content string) (SummaryRecord, error)

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL