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

38 page_window, 

39) 

40from lilbee.catalog.picks import ( 

41 find_pick, 

42 get_picks, 

43 picks_for, 

44 reset_picks, 

45) 

46from lilbee.catalog.query import ( 

47 build_adhoc_entry, 

48 get_catalog, 

49 is_rerank_ref, 

50 resolve_pull_target, 

51 size_bucket, 

52) 

53from lilbee.catalog.refs import DEFAULT_MMPROJ_PATTERN 

54 

55__all__ = [ 

56 "DEFAULT_MMPROJ_PATTERN", 

57 "PARAM_COUNT_RE", 

58 "QUANT_TIERS", 

59 "CatalogModel", 

60 "CatalogResult", 

61 "DownloadConfig", 

62 "DownloadProgress", 

63 "EnrichedModel", 

64 "ModelFamily", 

65 "ModelVariant", 

66 "PageWindow", 

67 "ProgressCallback", 

68 "agent_model_id", 

69 "build_adhoc_entry", 

70 "clean_display_name", 

71 "disk_shortfall", 

72 "display_label_for_ref", 

73 "download_mmproj", 

74 "download_model", 

75 "download_task_name", 

76 "enrich_catalog", 

77 "extract_quant", 

78 "find_pick", 

79 "get_catalog", 

80 "get_families", 

81 "get_picks", 

82 "is_rerank_ref", 

83 "make_download_callback", 

84 "page_window", 

85 "picks_for", 

86 "quant_tier", 

87 "reset_picks", 

88 "resolve_filename", 

89 "resolve_pull_target", 

90 "size_bucket", 

91]