Coverage for src/lilbee/data/store/__init__.py: 100%

1"""LanceDB vector store package.""" 

2 

3from __future__ import annotations 

4 

5from .core import Store 

6from .lance_helpers import ( 

7 agent_recall_predicate, 

8 ensure_table, 

9 escape_sql_string, 

10 human_recall_predicate, 

11 install_lancedb_thread_error_suppressor, 

12 local_owner_predicate, 

13 safe_delete, 

14) 

15from .ranking import cosine_sim, mmr_rerank 

16from .types import ( 

17 LOCAL_OWNER, 

18 SOURCE_STAT_UNKNOWN, 

19 ChunkType, 

20 ChunkWrite, 

21 CitationRecord, 

22 ConceptRecords, 

23 EmbeddingModelMismatchError, 

24 MemoryKind, 

25 MemoryRow, 

26 MemorySource, 

27 PageTextRecord, 

28 RemoveResult, 

29 SearchChunk, 

30 SearchScope, 

31 SourceMeta, 

32 SourceRecord, 

33 SourceStat, 

34 SourceStatBackfill, 

35 SourceType, 

36 agent_owner, 

37 is_agent_owner, 

38 scope_to_chunk_type, 

39 source_stat, 

40) 

41 

42__all__ = [ 

43 "LOCAL_OWNER", 

44 "SOURCE_STAT_UNKNOWN", 

45 "ChunkType", 

46 "ChunkWrite", 

47 "CitationRecord", 

48 "ConceptRecords", 

49 "EmbeddingModelMismatchError", 

50 "MemoryKind", 

51 "MemoryRow", 

52 "MemorySource", 

53 "PageTextRecord", 

54 "RemoveResult", 

55 "SearchChunk", 

56 "SearchScope", 

57 "SourceMeta", 

58 "SourceRecord", 

59 "SourceStat", 

60 "SourceStatBackfill", 

61 "SourceType", 

62 "Store", 

63 "agent_owner", 

64 "agent_recall_predicate", 

65 "cosine_sim", 

66 "ensure_table", 

67 "escape_sql_string", 

68 "human_recall_predicate", 

69 "install_lancedb_thread_error_suppressor", 

70 "is_agent_owner", 

71 "local_owner_predicate", 

72 "mmr_rerank", 

73 "safe_delete", 

74 "scope_to_chunk_type", 

75 "source_stat", 

76]