This commit is contained in:
team 1
2026-04-24 18:54:25 +02:00
parent 372a6797fa
commit c439fb99d6
12 changed files with 1126 additions and 336 deletions

View File

@@ -16,6 +16,7 @@ final readonly class RetriexEffectiveConfigProvider
private ModelGenerationConfigProvider $modelProvider,
private IndexConfigurationProvider $indexProvider,
private PromptBuilderConfig $promptConfig,
private NdjsonHybridRetrieverConfig $retrieverConfig,
) {
}
@@ -144,30 +145,8 @@ final readonly class RetriexEffectiveConfigProvider
private function retrievalConfig(): array
{
return [
'hard_max_chunks' => NdjsonHybridRetrieverConfig::HARD_MAX_CHUNKS,
'hard_max_vectork' => NdjsonHybridRetrieverConfig::HARD_MAX_VECTORK,
'hard_max_keywordk' => NdjsonHybridRetrieverConfig::HARD_MAX_KEYWORDK,
'vector_score_threshold' => NdjsonHybridRetrieverConfig::VECTOR_SCORE_THRESHOLD,
'threshold_floor' => NdjsonHybridRetrieverConfig::THRESHOLD_FLOOR,
'threshold_ceil' => NdjsonHybridRetrieverConfig::THRESHOLD_CEIL,
'list_bonus' => NdjsonHybridRetrieverConfig::LIST_BONUS,
'rrf_k' => NdjsonHybridRetrieverConfig::RRF_K,
'keyword_topk_multiplier' => NdjsonHybridRetrieverConfig::KEYWORD_TOPK_MULTIPLIER,
'keyword_score_threshold' => NdjsonHybridRetrieverConfig::KEYWORD_SCORE_THRESHOLD,
'keyword_rrf_weight' => NdjsonHybridRetrieverConfig::KEYWORD_RRF_WEIGHT,
'scoped_vector_rrf_weight' => NdjsonHybridRetrieverConfig::SCOPED_VECTOR_RRF_WEIGHT,
'scoped_keyword_rrf_weight' => NdjsonHybridRetrieverConfig::SCOPED_KEYWORD_RRF_WEIGHT,
'empty_rrf_fallback_topn' => NdjsonHybridRetrieverConfig::EMPTY_RRF_FALLBACK_TOPN,
'max_chunks_per_doc' => NdjsonHybridRetrieverConfig::MAX_CHUNKS_PER_DOC,
'min_chunk_distance' => NdjsonHybridRetrieverConfig::MIN_CHUNK_DISTANCE,
'dominant_doc_window' => NdjsonHybridRetrieverConfig::DOMINANT_DOC_WINDOW,
'dominant_doc_min_hits' => NdjsonHybridRetrieverConfig::DOMINANT_DOC_MIN_HITS,
'dominant_doc_max_chunks' => NdjsonHybridRetrieverConfig::DOMINANT_DOC_MAX_CHUNKS,
'exact_document_max_chunks' => NdjsonHybridRetrieverConfig::EXACT_DOCUMENT_MAX_CHUNKS,
'focused_product_window' => NdjsonHybridRetrieverConfig::FOCUSED_PRODUCT_WINDOW,
'focused_product_min_score' => NdjsonHybridRetrieverConfig::FOCUSED_PRODUCT_MIN_SCORE,
'focused_product_min_gap' => NdjsonHybridRetrieverConfig::FOCUSED_PRODUCT_MIN_GAP,
'focused_product_max_chunks' => NdjsonHybridRetrieverConfig::FOCUSED_PRODUCT_MAX_CHUNKS,
...$this->retrieverConfig->toArray(),
'vocabulary' => $this->retrieverConfig->vocabularyToArray(),
'inventory_parameter' => $this->param('retriex.retrieval.inventory', []),
];
}