vectorstore

type PoolOptions struct {
	MaxConns    int
	IdleConns   int
	IdleTimeout time.Duration
}

type PoolStats struct {
	ActiveConnections int
	MaxConnections    int
	IdleConnections   int
}

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

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)

	// 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

	// SearchByMetadata searches for chunks with matching metadata
	//
	// Parameters:
	// - ctx: Context for cancellation and timeout
	// - metadata: Metadata key-value pairs to match
	//
	// Returns:
	// - []core.Chunk: Slice of matching chunks
	// - error: Error if search fails
	SearchByMetadata(ctx context.Context, metadata map[string]string) ([]core.Chunk, error)
}