search

type EmbeddingService interface {
	// GenerateEmbedding generates an embedding for the given text
	GenerateEmbedding(ctx context.Context, text string, model string) ([]float32, error)
	// GenerateBatch generates embeddings for multiple texts
	GenerateBatch(ctx context.Context, texts []string, model string) ([][]float32, error)
}

type Filter map[string]any

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

type Repository interface {
	// Core repository methods - aligned with generic Repository[T] interface
	// Create stores a new search result
	Create(ctx context.Context, result *SearchResult) error
	// Get retrieves a search result by its ID
	Get(ctx context.Context, id string) (*SearchResult, error)
	// List retrieves search results matching the provided filter
	List(ctx context.Context, filter Filter) ([]*SearchResult, error)
	// Update modifies an existing search result
	Update(ctx context.Context, result *SearchResult) error
	// Delete removes a search result by its ID
	Delete(ctx context.Context, id string) error

	// Search-specific methods
	// SearchByText performs a vector search using text
	SearchByText(ctx context.Context, query string, options *SearchOptions) (*SearchResults, error)

	// SearchByVector performs a vector search using a pre-computed vector
	SearchByVector(ctx context.Context, vector []float32, options *SearchOptions) (*SearchResults, error)

	// SearchByContentID performs a "more like this" search
	SearchByContentID(ctx context.Context, contentID string, options *SearchOptions) (*SearchResults, error)

	// GetSupportedModels returns a list of models with embeddings
	GetSupportedModels(ctx context.Context) ([]string, error)

	// GetSearchStats retrieves statistics about the search index
	GetSearchStats(ctx context.Context) (map[string]any, error)
}

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

type SearchFilter struct {
	Field    string
	Operator string
	Value    any
}

type SearchOptions struct {
	Limit               int                    // Maximum number of results to return
	Offset              int                    // Number of results to skip
	MinSimilarity       float32                // Minimum similarity score (0-1)
	SimilarityThreshold float32                // Alias for MinSimilarity for backward compatibility
	MetadataFilters     map[string]interface{} // JSONB metadata filters
	HybridSearch        bool                   // Combine text + vector search
	RankingAlgorithm    string                 // "cosine", "euclidean", "dot_product"
	MaxResults          int                    // Alias for Limit for backward compatibility
	Filters             []SearchFilter         // Structured filters
	Sorts               []SearchSort           // Sort criteria
	ContentTypes        []string               // Filter by content types
	WeightFactors       map[string]float32     // Weights for hybrid search
}

type SearchResult struct {
	ID          string
	Score       float32
	Distance    float32
	Content     string
	Type        string
	Metadata    map[string]any
	ContentHash string
}

type SearchResults struct {
	Results []*SearchResult
	Total   int
	HasMore bool
}

type SearchSort struct {
	Field     string
	Direction string
}