knowledge

type Chunker interface {
	Split(document Document) ([]Document, error)
}

type Citation struct {
	ID         string `json:"id"`
	ParentID   string `json:"parent_id,omitempty"`
	Source     string `json:"source,omitempty"`
	Title      string `json:"title,omitempty"`
	URL        string `json:"url,omitempty"`
	ChunkIndex string `json:"chunk_index,omitempty"`
	ChunkStart string `json:"chunk_start,omitempty"`
	ChunkEnd   string `json:"chunk_end,omitempty"`
}

type DeleteRequest struct {
	Namespace string `json:"namespace,omitempty"`
	ID        string `json:"id"`
}

type Document struct {
	ID        string            `json:"id"`
	Content   string            `json:"content"`
	Metadata  map[string]string `json:"metadata,omitempty"`
	Namespace string            `json:"namespace,omitempty"`
}

type DocumentEmbedding struct {
	Document Document  `json:"document"`
	Vector   []float32 `json:"vector"`
}

type HybridSearcher interface {
	HybridQuery(ctx context.Context, query Query) ([]SearchResult, error)
}

type IndexResult struct {
	Documents int `json:"documents"`
	Chunks    int `json:"chunks"`
}

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

type IndexerConfig struct {
	Embedder  llm.Embedder
	Store     VectorStore
	Profile   string
	Namespace string
	BatchSize int
	Chunker   Chunker
}

type Loader interface {
	Load(ctx context.Context) ([]Document, error)
}

type Query struct {
	Namespace    string            `json:"namespace,omitempty"`
	Text         string            `json:"text,omitempty"`
	Mode         SearchMode        `json:"mode,omitempty"`
	Vector       []float32         `json:"vector"`
	Limit        int               `json:"limit,omitempty"`
	Filter       map[string]string `json:"filter,omitempty"`
	VectorWeight float64           `json:"vector_weight,omitempty"`
	TextWeight   float64           `json:"text_weight,omitempty"`
}

type Reranker interface {
	Rerank(ctx context.Context, query string, results []SearchResult) ([]SearchResult, error)
}

type SearchMode string

type SearchResult struct {
	Document Document `json:"document"`
	Score    float64  `json:"score"`
}

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

type TextSplitterConfig struct {
	MaxRunes     int
	OverlapRunes int
}

type VectorStore interface {
	Upsert(ctx context.Context, documents []DocumentEmbedding) error
	Query(ctx context.Context, query Query) ([]SearchResult, error)
	Delete(ctx context.Context, req DeleteRequest) error
}