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

19 MEMORY_SOURCE_PREFIX, 

20 SOURCE_STAT_UNKNOWN, 

21 ChunkType, 

22 ChunkWrite, 

23 CitationRecord, 

24 ConceptRecords, 

25 EmbeddingModelMismatchError, 

26 IndexMismatch, 

27 MemoryKind, 

28 MemoryRow, 

29 MemorySource, 

30 PageTextRecord, 

31 RemoveResult, 

32 SearchChunk, 

33 SearchScope, 

34 SourceMeta, 

35 SourceRecord, 

36 SourceStat, 

37 SourceStatBackfill, 

38 SourceType, 

39 agent_owner, 

40 is_agent_owner, 

41 is_memory_source, 

42 memory_source, 

43 scope_to_chunk_type, 

44 source_stat, 

45) 

46 

47__all__ = [ 

48 "LOCAL_OWNER", 

49 "MEMORY_CONTENT_TYPE", 

50 "MEMORY_SOURCE_PREFIX", 

51 "SOURCE_STAT_UNKNOWN", 

52 "ChunkType", 

53 "ChunkWrite", 

54 "CitationRecord", 

55 "ConceptRecords", 

56 "EmbeddingModelMismatchError", 

57 "IndexMismatch", 

58 "MemoryKind", 

59 "MemoryRow", 

60 "MemorySource", 

61 "PageTextRecord", 

62 "RemoveResult", 

63 "SearchChunk", 

64 "SearchScope", 

65 "SourceMeta", 

66 "SourceRecord", 

67 "SourceStat", 

68 "SourceStatBackfill", 

69 "SourceType", 

70 "Store", 

71 "agent_owner", 

72 "agent_recall_predicate", 

73 "cosine_sim", 

74 "ensure_table", 

75 "escape_sql_string", 

76 "human_recall_predicate", 

77 "install_lancedb_thread_error_suppressor", 

78 "is_agent_owner", 

79 "is_memory_source", 

80 "local_owner_predicate", 

81 "memory_source", 

82 "mmr_rerank", 

83 "safe_delete", 

84 "scope_to_chunk_type", 

85 "source_stat", 

86]