Coverage for src/lilbee/retrieval/query/__init__.py: 100%

1"""RAG query pipeline. Embed question, search, generate answer with citations.""" 

2 

3from __future__ import annotations 

4 

5from lilbee.retrieval.query.dedup import ( 

6 diversify_sources, 

7 filter_results, 

8 prepare_results, 

9 sort_by_relevance, 

10) 

11from lilbee.retrieval.query.formatting import ( 

12 SOURCES_BLOCK_MARKER, 

13 build_context, 

14 format_sources_block, 

15 strip_llm_citations, 

16) 

17from lilbee.retrieval.query.searcher import AskResult, ChatMessage, Searcher 

18 

19__all__ = [ 

20 "SOURCES_BLOCK_MARKER", 

21 "AskResult", 

22 "ChatMessage", 

23 "Searcher", 

24 "build_context", 

25 "diversify_sources", 

26 "filter_results", 

27 "format_sources_block", 

28 "prepare_results", 

29 "sort_by_relevance", 

30 "strip_llm_citations", 

31]