storage

type OpenSearchConfig struct {
	Addresses    []string `toml:"addresses"`
	Username     string   `toml:"username"`
	Password     string   `toml:"password"`
	IndexName    string   `toml:"index"`
	EmbeddingDim int      `toml:"embedding_dim"`
	InsecureSSL  bool     `toml:"insecure_ssl"`
}

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

type SearchQuery struct {
	// Filters for exact match (field -> value)
	Filters map[string]any

	// TermsFilters for multi-value match (field -> []values)
	TermsFilters map[string][]string

	// RangeFilters for range queries (field -> {gte/lte/gt/lt -> value})
	RangeFilters map[string]map[string]any

	// Embedding vector for k-NN search
	Embedding []float32

	// TextQuery for full-text search (searches raw_content and content fields)
	TextQuery string

	// HybridSearch enables combining vector and text search
	// When true and both Embedding and TextQuery are provided, uses hybrid search
	HybridSearch bool

	// Score threshold for filtering results
	ScoreThreshold float64

	// Limit on results
	Limit int
}

type VectorStore interface {
	// Store stores a document with the given ID
	Store(ctx context.Context, id string, doc map[string]any) error

	// Get retrieves a document by ID
	Get(ctx context.Context, id string) (map[string]any, error)

	// Search searches for documents based on query
	Search(ctx context.Context, query SearchQuery) ([]map[string]any, error)

	// Delete deletes a document by ID
	Delete(ctx context.Context, id string) error

	// DeleteByQuery deletes documents matching the filters
	DeleteByQuery(ctx context.Context, filters map[string]any) (int, error)

	// Count counts documents matching the filters
	Count(ctx context.Context, filters map[string]any) (int, error)

	// Update updates a document (upsert)
	Update(ctx context.Context, id string, doc map[string]any) error

	// UpdateFields updates specific fields of a document
	UpdateFields(ctx context.Context, id string, fields map[string]any) error

	// Close closes the storage connection
	Close() error
}