Kodit

package search

// Result represents a generic search result.
type Result struct {
	snippetID string
	score     float64
}

// NewResult creates a new Result.
func NewResult(snippetID string, score float64) Result {
	return Result{
		snippetID: snippetID,
		score:     score,
	}
}

// SnippetID returns the snippet ID.
func (r Result) SnippetID() string { return r.snippetID }

// Score returns the search score.
func (r Result) Score() float64 { return r.score }

// FusionRequest represents a fusion request input.
type FusionRequest struct {
	id    string
	score float64
}

// NewFusionRequest creates a new FusionRequest.
func NewFusionRequest(id string, score float64) FusionRequest {
	return FusionRequest{
		id:    id,
		score: score,
	}
}

// ID returns the document ID.
func (f FusionRequest) ID() string { return f.id }

// Score returns the score.
func (f FusionRequest) Score() float64 { return f.score }

// FusionResult represents a fusion result.
type FusionResult struct {
	id             string
	score          float64
	originalScores []float64
}

// NewFusionResult creates a new FusionResult.
func NewFusionResult(id string, score float64, originalScores []float64) FusionResult {
	scores := make([]float64, len(originalScores))
	copy(scores, originalScores)
	return FusionResult{
		id:             id,
		score:          score,
		originalScores: scores,
	}
}

// ID returns the document ID.
func (f FusionResult) ID() string { return f.id }

// Score returns the fused score.
func (f FusionResult) Score() float64 { return f.score }

// OriginalScores returns the original scores from each search method.
func (f FusionResult) OriginalScores() []float64 {
	scores := make([]float64, len(f.originalScores))
	copy(scores, f.originalScores)
	return scores
}

// Embedding represents a snippet with its pre-computed embedding vector.
type Embedding struct {
	snippetID string
	vector    []float64
}

// NewEmbedding creates a new Embedding value object.
func NewEmbedding(snippetID string, vector []float64) Embedding {
	cp := make([]float64, len(vector))
	copy(cp, vector)
	return Embedding{
		snippetID: snippetID,
		vector:    cp,
	}
}

// SnippetID returns the snippet identifier.
func (e Embedding) SnippetID() string { return e.snippetID }

// Vector returns a defensive copy of the embedding vector.
func (e Embedding) Vector() []float64 {
	cp := make([]float64, len(e.vector))
	copy(cp, e.vector)
	return cp
}

// Document represents a generic document for indexing.
type Document struct {
	snippetID string
	text      string
}

// NewDocument creates a new Document.
func NewDocument(snippetID, text string) Document {
	return Document{
		snippetID: snippetID,
		text:      text,
	}
}

// SnippetID returns the snippet ID.
func (d Document) SnippetID() string { return d.snippetID }

// Text returns the document text.
func (d Document) Text() string { return d.text }

// IndexRequest represents a generic indexing request.
type IndexRequest struct {
	documents []Document
}

// NewIndexRequest creates a new IndexRequest.
func NewIndexRequest(documents []Document) IndexRequest {
	docs := make([]Document, len(documents))
	copy(docs, documents)
	return IndexRequest{documents: docs}
}

// Documents returns the documents to index.
func (i IndexRequest) Documents() []Document {
	docs := make([]Document, len(i.documents))
	copy(docs, i.documents)
	return docs
}