Documentation
¶
Index ¶
- func GetDB() (*sql.DB, error)
- func GetDocumentSummaryEmbedding(db *sql.DB, summaryID, embeddingModelID string) ([]float32, error)
- func GetSummaryEmbeddingByPath(db *sql.DB, path, summaryModelID, embeddingModelID string) ([]float32, error)
- func HasDocumentChunks(db *sql.DB, documentID, embeddingModelID string) (bool, error)
- func InitDB() (*sql.DB, error)
- func InsertEmbeddingModel(db *sql.DB, id string, embeddingDim int) (string, error)
- func InsertSummaryModel(db *sql.DB, id string) (string, error)
- func Migrate(db *sql.DB) error
- func Open(path string) (*sql.DB, error)
- func ReplaceDocumentChunks(db *sql.DB, documentID, embeddingModelID string, chunks []ChunkInput) error
- func UpsertDocumentSummaryEmbedding(db *sql.DB, summaryID, embeddingModelID string, embedding []float32) error
- type ChunkInput
- type Document
- type DocumentRecord
- type SearchRow
- func MergeSearchRows(limit int, groups ...[]SearchRow) []SearchRow
- func SearchDocumentChunks(db *sql.DB, embeddingModelID string, query []float32, limit int, ...) ([]SearchRow, error)
- func SearchSummaryEmbeddings(db *sql.DB, summaryModelID string, embeddingModelID string, query []float32, ...) ([]SearchRow, error)
- type SummaryRecord
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func HasDocumentChunks ¶
func InsertEmbeddingModel ¶
func ReplaceDocumentChunks ¶
func ReplaceDocumentChunks(db *sql.DB, documentID, embeddingModelID string, chunks []ChunkInput) error
Types ¶
type ChunkInput ¶
type DocumentRecord ¶
func ListDocuments ¶
type SummaryRecord ¶
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)
Click to show internal directories.
Click to hide internal directories.