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

33 enrich_catalog, 

34 extract_quant, 

35 quant_tier, 

36) 

37from lilbee.catalog.models import ( 

38 CatalogModel, 

39 CatalogResult, 

40 DownloadProgress, 

41 ModelFamily, 

42 ModelVariant, 

43) 

44from lilbee.catalog.query import ( 

45 CatalogIndex, 

46 build_adhoc_entry, 

47 find_catalog_entry, 

48 get_catalog, 

49 is_rerank_ref, 

50 resolve_pull_target, 

51) 

52 

53__all__ = [ 

54 "FEATURED_ALL", 

55 "FEATURED_CHAT", 

56 "FEATURED_EMBEDDING", 

57 "FEATURED_RERANK", 

58 "FEATURED_VISION", 

59 "PARAM_COUNT_RE", 

60 "QUANT_TIERS", 

61 "VISION_MMPROJ_FILES", 

62 "CatalogIndex", 

63 "CatalogModel", 

64 "CatalogResult", 

65 "DownloadConfig", 

66 "DownloadProgress", 

67 "EnrichedModel", 

68 "ModelFamily", 

69 "ModelVariant", 

70 "ProgressCallback", 

71 "build_adhoc_entry", 

72 "clean_display_name", 

73 "display_label_for_ref", 

74 "download_model", 

75 "download_task_name", 

76 "enrich_catalog", 

77 "extract_quant", 

78 "fetch_model_file_size", 

79 "find_catalog_entry", 

80 "find_mmproj_file", 

81 "get_catalog", 

82 "get_families", 

83 "is_rerank_ref", 

84 "make_download_callback", 

85 "quant_tier", 

86 "resolve_filename", 

87 "resolve_pull_target", 

88]