retriever

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

type CohereRerankerConfig struct {
	// Model is the Cohere rerank model to use
	// Options: "rerank-v3.5", "rerank-english-v3.0", "rerank-multilingual-v3.0"
	Model string
	// TopK is the number of documents to return
	TopK int
	// APIBase is the custom API base URL (optional)
	APIBase string
	// Timeout is the HTTP request timeout
	Timeout time.Duration
}

type CombinedDocumentScore struct {
	Document       rag.Document
	TotalScore     float64
	RetrieverCount int
	Sources        []string
	Metadata       map[string]any
}

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

type CrossEncoderRerankerConfig struct {
	// ModelName is the name of the cross-encoder model
	// Common models:
	// - "cross-encoder/ms-marco-MiniLM-L-6-v2"
	// - "cross-encoder/ms-marco-MiniLM-L-12-v2"
	// - "cross-encoder/mmarco-mMiniLMv2-L12-H384-v1"
	ModelName string
	// TopK is the number of documents to return
	TopK int
	// APIBase is the URL of the cross-encoder service
	// This can be a local service (e.g., http://localhost:8000/rerank)
	// or a remote service
	APIBase string
	// Timeout is the HTTP request timeout
	Timeout time.Duration
}

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

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

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

type JinaRerankerConfig struct {
	// Model is the Jina rerank model to use
	// Options: "jina-reranker-v1-base-en", "jina-reranker-v2-base-multilingual"
	Model string
	// TopK is the number of documents to return
	TopK int
	// APIBase is the custom API base URL (optional)
	APIBase string
	// Timeout is the HTTP request timeout
	Timeout time.Duration
}

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

type LLMRerankerConfig struct {
	// TopK is the number of documents to return
	TopK int
	// ScoreThreshold is the minimum relevance score (0-1)
	ScoreThreshold float64
	// SystemPrompt is a custom system prompt for scoring
	SystemPrompt string
	// BatchSize is the number of documents to score in a single request (for efficiency)
	BatchSize int
}

type SimpleReranker struct {
}

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

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