Coverage for src / qdrant_loader_mcp_server / search / enhanced / cdi / extractors / recommendation.py: 100%

1from __future__ import annotations 

2 

3from ....components.models import HybridSearchResult 

4from ..interfaces import Recommender 

5from ..models import ComplementaryContent 

6 

7 

8class DefaultRecommender(Recommender): 

9 """Adapter to legacy ComplementaryContentFinder for behavior parity.""" 

10 

11 def __init__(self, similarity_calculator, knowledge_graph=None): 

12 from ...cross_document_intelligence import ( 

13 ComplementaryContentFinder as LegacyComplementaryFinder, # type: ignore 

14 ) 

15 

16 self._legacy = LegacyComplementaryFinder(similarity_calculator, knowledge_graph) 

17 

18 def recommend( 

19 self, 

20 target: HybridSearchResult, 

21 pool: list[HybridSearchResult], 

22 ) -> ComplementaryContent: # type: ignore[override] 

23 return self._legacy.find_complementary_content(target, pool)