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 display_source_path, 

15 format_source, 

16 format_sources_block, 

17 strip_llm_citations, 

18) 

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

20 

21__all__ = [ 

22 "SOURCES_BLOCK_MARKER", 

23 "AskResult", 

24 "ChatMessage", 

25 "Searcher", 

26 "build_context", 

27 "display_source_path", 

28 "diversify_sources", 

29 "filter_results", 

30 "format_source", 

31 "format_sources_block", 

32 "prepare_results", 

33 "sort_by_relevance", 

34 "strip_llm_citations", 

35]