Coverage for src / lilbee / catalog / __init__.py: 100%

1"""Model catalog. Discovers available GGUF models from HuggingFace. 

2 

3Three levels: 

41. Featured models. Curated favorites (hardcoded, always available) 

52. HF API models. Fetched from HuggingFace API, paginated and filterable 

63. Combined catalog. Featured first, then HF results 

7""" 

8 

9from lilbee.catalog.download import ( 

10 DownloadConfig, 

11 download_model, 

12 fetch_model_file_size, 

13 find_mmproj_file, 

14 resolve_filename, 

15) 

16from lilbee.catalog.download_progress import ProgressCallback, make_download_callback 

17from lilbee.catalog.families import get_families 

18from lilbee.catalog.featured import ( 

19 FEATURED_ALL, 

20 FEATURED_CHAT, 

21 FEATURED_EMBEDDING, 

22 FEATURED_RERANK, 

23 FEATURED_VISION, 

24 VISION_MMPROJ_FILES, 

25) 

26from lilbee.catalog.formatting import ( 

27 PARAM_COUNT_RE, 

28 QUANT_TIERS, 

29 EnrichedModel, 

30 clean_display_name, 

31 display_label_for_ref, 

32 enrich_catalog, 

33 extract_quant, 

34 quant_tier, 

35) 

36from lilbee.catalog.models import ( 

37 CatalogModel, 

38 CatalogResult, 

39 DownloadProgress, 

40 ModelFamily, 

41 ModelVariant, 

42) 

43from lilbee.catalog.query import ( 

44 CatalogIndex, 

45 build_adhoc_entry, 

46 find_catalog_entry, 

47 get_catalog, 

48 is_rerank_ref, 

49 resolve_pull_target, 

50) 

51 

52__all__ = [ 

53 "FEATURED_ALL", 

54 "FEATURED_CHAT", 

55 "FEATURED_EMBEDDING", 

56 "FEATURED_RERANK", 

57 "FEATURED_VISION", 

58 "PARAM_COUNT_RE", 

59 "QUANT_TIERS", 

60 "VISION_MMPROJ_FILES", 

61 "CatalogIndex", 

62 "CatalogModel", 

63 "CatalogResult", 

64 "DownloadConfig", 

65 "DownloadProgress", 

66 "EnrichedModel", 

67 "ModelFamily", 

68 "ModelVariant", 

69 "ProgressCallback", 

70 "build_adhoc_entry", 

71 "clean_display_name", 

72 "display_label_for_ref", 

73 "download_model", 

74 "enrich_catalog", 

75 "extract_quant", 

76 "fetch_model_file_size", 

77 "find_catalog_entry", 

78 "find_mmproj_file", 

79 "get_catalog", 

80 "get_families", 

81 "is_rerank_ref", 

82 "make_download_callback", 

83 "quant_tier", 

84 "resolve_filename", 

85 "resolve_pull_target", 

86]