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

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

2 

3Two levels: 

41. Picks. The most popular models of each parameter tier, fetched live 

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

6 

7The combined catalog puts picks first, then HF results. 

8""" 

9 

10from lilbee.catalog.download import ( 

11 DownloadConfig, 

12 disk_shortfall, 

13 download_mmproj, 

14 download_model, 

15 resolve_filename, 

16) 

17from lilbee.catalog.download_progress import ProgressCallback, make_download_callback 

18from lilbee.catalog.families import get_families 

19from lilbee.catalog.formatting import ( 

20 PARAM_COUNT_RE, 

21 QUANT_TIERS, 

22 EnrichedModel, 

23 agent_model_id, 

24 clean_display_name, 

25 display_label_for_ref, 

26 download_task_name, 

27 enrich_catalog, 

28 extract_quant, 

29 quant_tier, 

30) 

31from lilbee.catalog.models import ( 

32 CatalogModel, 

33 CatalogResult, 

34 DownloadProgress, 

35 ModelFamily, 

36 ModelVariant, 

37) 

38from lilbee.catalog.picks import ( 

39 find_pick, 

40 get_picks, 

41 picks_for, 

42 reset_picks, 

43) 

44from lilbee.catalog.query import ( 

45 build_adhoc_entry, 

46 get_catalog, 

47 is_rerank_ref, 

48 resolve_pull_target, 

49 size_bucket, 

50) 

51from lilbee.catalog.refs import DEFAULT_MMPROJ_PATTERN 

52 

53__all__ = [ 

54 "DEFAULT_MMPROJ_PATTERN", 

55 "PARAM_COUNT_RE", 

56 "QUANT_TIERS", 

57 "CatalogModel", 

58 "CatalogResult", 

59 "DownloadConfig", 

60 "DownloadProgress", 

61 "EnrichedModel", 

62 "ModelFamily", 

63 "ModelVariant", 

64 "ProgressCallback", 

65 "agent_model_id", 

66 "build_adhoc_entry", 

67 "clean_display_name", 

68 "disk_shortfall", 

69 "display_label_for_ref", 

70 "download_mmproj", 

71 "download_model", 

72 "download_task_name", 

73 "enrich_catalog", 

74 "extract_quant", 

75 "find_pick", 

76 "get_catalog", 

77 "get_families", 

78 "get_picks", 

79 "is_rerank_ref", 

80 "make_download_callback", 

81 "picks_for", 

82 "quant_tier", 

83 "reset_picks", 

84 "resolve_filename", 

85 "resolve_pull_target", 

86 "size_bucket", 

87]