vectorstore

type FilterCondition struct {
	Field    string         // Field to filter on
	Operator FilterOperator // Operator to use
	Value    interface{}    // Value to compare against
}

type FilterOperator string

type SearchOptions struct {
	TopK     int                    // Number of top results to return
	Filter   map[string]interface{} // Metadata filters
	MinScore float32                // Minimum similarity score
	Metadata map[string]string      // Additional metadata for search
}

type Store interface {
	// Add adds chunks and their embeddings to the vector store
	//
	// Parameters:
	// - ctx: Context for cancellation and timeout
	// - chunks: Slice of document chunks
	// - embeddings: Slice of embeddings, one for each chunk
	//
	// Returns:
	// - error: Error if storage fails
	Add(ctx context.Context, chunks []core.Chunk, embeddings [][]float32) error

	// Search searches for similar vectors to the query embedding
	//
	// Parameters:
	// - ctx: Context for cancellation and timeout
	// - query: Query embedding
	// - opts: Search options (TopK, filters, etc.)
	//
	// Returns:
	// - []core.Result: Slice of search results
	// - error: Error if search fails
	Search(ctx context.Context, query []float32, opts SearchOptions) ([]core.Result, error)

	// SearchStructured performs a structured search with filters
	//
	// Parameters:
	// - ctx: Context for cancellation and timeout
	// - query: Structured query with filters
	// - embedding: Query embedding
	//
	// Returns:
	// - []core.Result: Slice of search results
	// - error: Error if search fails
	SearchStructured(ctx context.Context, query *StructuredQuery, embedding []float32) ([]core.Result, error)

	// Delete removes vectors by their IDs
	//
	// Parameters:
	// - ctx: Context for cancellation and timeout
	// - ids: Slice of vector IDs to delete
	//
	// Returns:
	// - error: Error if deletion fails
	Delete(ctx context.Context, ids []string) error

	// GetByMetadata retrieves vectors by metadata
	//
	// Parameters:
	// - ctx: Context for cancellation and timeout
	// - metadata: Metadata filter
	//
	// Returns:
	// - []core.Result: Slice of matching results
	// - error: Error if retrieval fails
	GetByMetadata(ctx context.Context, metadata map[string]string) ([]core.Result, error)
}

type StructuredQuery struct {
	Query    string            // Natural language query
	Filters  []FilterCondition // Filter conditions
	TopK     int               // Number of top results
	MinScore float32           // Minimum similarity score
}