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

7 diversify_sources, 

8 filter_results, 

9 prepare_results, 

10 sort_by_relevance, 

11) 

12from lilbee.retrieval.query.formatting import ( 

13 build_context, 

14 display_source_path, 

15 format_source, 

16 strip_llm_citations, 

17) 

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

19 

20__all__ = [ 

21 "AskResult", 

22 "ChatMessage", 

23 "Searcher", 

24 "build_context", 

25 "deduplicate_sources", 

26 "display_source_path", 

27 "diversify_sources", 

28 "filter_results", 

29 "format_source", 

30 "prepare_results", 

31 "sort_by_relevance", 

32 "strip_llm_citations", 

33]