Added test and exposed list_models.

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "llm"  # The pip install <name>.
-version = "0.4.1"
+version = "0.4.2"
 description = "Library for easy use of LLMs."
 readme = "README.md"
 authors = [

src/llm/__init__.py

@@ -3,28 +3,35 @@
 
 
 __all__ = (
-    "model",
-    "embedding")
+    'model',
+    'list_models',
+    'embedding')
 
 
 if TYPE_CHECKING:
     from .models.selection import model
+    from .models.selection import list_models
     from .models.embed import EmbeddingModel
 
 
 def __getattr__(name: str) -> Any:
 
-    if name == "model":
+    if name == 'model':
         from .models.selection import model
         globals()[name] = model
         return model
+
+    if name == 'list_models':
+        from .models.selection import list_models
+        globals()[name] = list_models
+        return list_models
 
-    if name == "embedding":
+    if name == 'embedding':
         from .models.embed import EmbeddingModel
         globals()[name] = EmbeddingModel
         return EmbeddingModel
 
-    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+    raise AttributeError(f'module {__name__!r} has no attribute {name!r}')
 
 
 def __dir__() -> list[str]:

tests/test_list_models.py

@@ -0,0 +1,24 @@
+"""
+Tests for llm.list_models.
+"""
+
+import pytest
+
+import llm
+
+
+def test_list_models_llm():
+    result = llm.list_models(kind='llm')
+    assert isinstance(result, list)
+    assert len(result) > 0
+
+
+def test_list_models_embed():
+    result = llm.list_models(kind='embed')
+    assert isinstance(result, list)
+    assert len(result) > 0
+
+
+def test_list_models_invalid_kind():
+    with pytest.raises(ValueError):
+        llm.list_models(kind='invalid')