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 .crawl_pages import CrawlPage, CrawlPages 

7from .lance_helpers import ( 

8 agent_recall_predicate, 

9 ensure_table, 

10 escape_sql_string, 

11 human_recall_predicate, 

12 install_lancedb_thread_error_suppressor, 

13 local_owner_predicate, 

14 safe_delete, 

15) 

16from .ranking import cosine_sim, mmr_rerank 

17from .types import ( 

18 DOCUMENTS_PAGE_DEFAULT, 

19 DOCUMENTS_PAGE_MAX, 

20 LOCAL_OWNER, 

21 MEMORY_CONTENT_TYPE, 

22 MEMORY_SOURCE_PREFIX, 

23 REMOVED_REASON, 

24 SOURCE_STAT_UNKNOWN, 

25 ChunkType, 

26 ChunkWrite, 

27 CitationRecord, 

28 ConceptRecords, 

29 EmbeddingModelMismatchError, 

30 HeldOut, 

31 HeldRecord, 

32 IndexMismatch, 

33 MemoryKind, 

34 MemoryRow, 

35 MemorySource, 

36 PageTextRecord, 

37 RemoveResult, 

38 SearchChunk, 

39 SearchScope, 

40 SourceHead, 

41 SourceMeta, 

42 SourceMove, 

43 SourceRecord, 

44 SourceRelocation, 

45 SourcesCheckpoint, 

46 SourceStat, 

47 SourceStatBackfill, 

48 SourceSummary, 

49 SourceType, 

50 agent_owner, 

51 is_agent_owner, 

52 is_memory_source, 

53 memory_source, 

54 scope_to_chunk_type, 

55 source_stat, 

56) 

57 

58__all__ = [ 

59 "DOCUMENTS_PAGE_DEFAULT", 

60 "DOCUMENTS_PAGE_MAX", 

61 "LOCAL_OWNER", 

62 "MEMORY_CONTENT_TYPE", 

63 "MEMORY_SOURCE_PREFIX", 

64 "REMOVED_REASON", 

65 "SOURCE_STAT_UNKNOWN", 

66 "ChunkType", 

67 "ChunkWrite", 

68 "CitationRecord", 

69 "ConceptRecords", 

70 "CrawlPage", 

71 "CrawlPages", 

72 "EmbeddingModelMismatchError", 

73 "HeldOut", 

74 "HeldRecord", 

75 "IndexMismatch", 

76 "MemoryKind", 

77 "MemoryRow", 

78 "MemorySource", 

79 "PageTextRecord", 

80 "RemoveResult", 

81 "SearchChunk", 

82 "SearchScope", 

83 "SourceHead", 

84 "SourceMeta", 

85 "SourceMove", 

86 "SourceRecord", 

87 "SourceRelocation", 

88 "SourceStat", 

89 "SourceStatBackfill", 

90 "SourceSummary", 

91 "SourceType", 

92 "SourcesCheckpoint", 

93 "Store", 

94 "agent_owner", 

95 "agent_recall_predicate", 

96 "cosine_sim", 

97 "ensure_table", 

98 "escape_sql_string", 

99 "human_recall_predicate", 

100 "install_lancedb_thread_error_suppressor", 

101 "is_agent_owner", 

102 "is_memory_source", 

103 "local_owner_predicate", 

104 "memory_source", 

105 "mmr_rerank", 

106 "safe_delete", 

107 "scope_to_chunk_type", 

108 "source_stat", 

109]