model-library 0.1.0__tar.gz → 0.1.1__tar.gz

{model_library-0.1.0 → model_library-0.1.1}/.github/workflows/publish.yml

@@ -72,7 +72,7 @@ jobs:
           path: dist/
 
       - name: Sign the dists with Sigstore
-        uses: sigstore/gh-action-sigstore-python@v2.1.1
+        uses: sigstore/gh-action-sigstore-python@v3.1.0
         with:
           inputs: |
             ./dist/*.tar.gz

{model_library-0.1.0 → model_library-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: model-library
-Version: 0.1.0
+Version: 0.1.1
 Summary: Model Library for vals.ai 
 Author-email: "Vals AI, Inc." <contact@vals.ai>
 License: MIT
@@ -10,6 +10,7 @@ License-File: LICENSE
 Requires-Dist: typing-extensions<5.0,>=4.14.1
 Requires-Dist: pydantic<3.0,>=2.11.7
 Requires-Dist: pyyaml>=6.0.2
+Requires-Dist: rich
 Requires-Dist: backoff<3.0,>=2.2.1
 Requires-Dist: redis<7.0,>=6.2.0
 Requires-Dist: tiktoken==0.11.0
@@ -53,7 +54,14 @@ Open-source model library for interacting with a variety of LLM providers. Origi
 - X AI
 - ZhipuAI (zai)
 
-Run `python -m scripts.browse_models` to browse the model registry.
+Run `python -m scripts.browse_models` to browse the model registry or
+
+```python
+from model_library.registry_utils import get_model_names_by_provider, get_provider_names
+
+print(get_provider_names())
+print(get_model_names_by_provider("chosen-provider"))
+```
 
 ### Supported Input
 
@@ -70,16 +78,26 @@ Here is a basic example of how to query a model:
 
 ```python
 import asyncio
-from model_library.registry_utils import get_registry_model
+from model_library import model
 
 async def main():
     # Load a model from the registry
-    model = get_registry_model("openai/gemini-2.5-flash")
+    llm = model("anthropic/claude-opus-4-1-20250805-thinking")
+
+    # Display the LLM instance
+    llm.logger.info(llm)
+    # or print(llm)
 
     # Query the model with a simple text input
-    response = await model.query("What is QSBS? Explain your thinking in detail and make it concise.")
+    result = await llm.query(
+        "What is QSBS? Explain your thinking in detail and make it concise."
+    )
+
+    # Logger automatically logs the result
+
+    # Display only the output text
+    llm.logger.info(result.output_text)
 
-    # Logger automatically logs the response
 
 if __name__ == "__main__":
     asyncio.run(main())
@@ -88,7 +106,18 @@ if __name__ == "__main__":
 The model registry holds model attributes, ex. reasoning, file support, tool support, max tokens. You may also use models not included in the registry.
 
 ```python
-model = get_raw_model("openai/gpt-3.5-turbo", config=LLMConfig(max_tokens=1000))
+from model_library import raw_model
+from model_library.base import LLMConfig
+
+model = raw_model("grok/grok-code-fast", LLMConfig(max_tokens=10000))
+```
+
+Root logger is named "llm". To disable logging:
+
+```python
+from model_library import set_logging
+
+set_logging(enable=False)
 ```
 
 ### Environment Setup

{model_library-0.1.0 → model_library-0.1.1}/README.md

@@ -26,7 +26,14 @@ Open-source model library for interacting with a variety of LLM providers. Origi
 - X AI
 - ZhipuAI (zai)
 
-Run `python -m scripts.browse_models` to browse the model registry.
+Run `python -m scripts.browse_models` to browse the model registry or
+
+```python
+from model_library.registry_utils import get_model_names_by_provider, get_provider_names
+
+print(get_provider_names())
+print(get_model_names_by_provider("chosen-provider"))
+```
 
 ### Supported Input
 
@@ -43,16 +50,26 @@ Here is a basic example of how to query a model:
 
 ```python
 import asyncio
-from model_library.registry_utils import get_registry_model
+from model_library import model
 
 async def main():
     # Load a model from the registry
-    model = get_registry_model("openai/gemini-2.5-flash")
+    llm = model("anthropic/claude-opus-4-1-20250805-thinking")
+
+    # Display the LLM instance
+    llm.logger.info(llm)
+    # or print(llm)
 
     # Query the model with a simple text input
-    response = await model.query("What is QSBS? Explain your thinking in detail and make it concise.")
+    result = await llm.query(
+        "What is QSBS? Explain your thinking in detail and make it concise."
+    )
+
+    # Logger automatically logs the result
+
+    # Display only the output text
+    llm.logger.info(result.output_text)
 
-    # Logger automatically logs the response
 
 if __name__ == "__main__":
     asyncio.run(main())
@@ -61,7 +78,18 @@ if __name__ == "__main__":
 The model registry holds model attributes, ex. reasoning, file support, tool support, max tokens. You may also use models not included in the registry.
 
 ```python
-model = get_raw_model("openai/gpt-3.5-turbo", config=LLMConfig(max_tokens=1000))
+from model_library import raw_model
+from model_library.base import LLMConfig
+
+model = raw_model("grok/grok-code-fast", LLMConfig(max_tokens=10000))
+```
+
+Root logger is named "llm". To disable logging:
+
+```python
+from model_library import set_logging
+
+set_logging(enable=False)
 ```
 
 ### Environment Setup

{model_library-0.1.0 → model_library-0.1.1}/examples/setup.py

@@ -4,22 +4,12 @@ from typing import Any
 from rich.console import Console
 from rich.logging import RichHandler
 
-console = Console()
-
-# llm logger
-llm_logger = logging.getLogger("llm")
-llm_logger.setLevel(logging.INFO)
-llm_logger.handlers.clear()
-handler = RichHandler(console=console, markup=True, show_time=False)
-handler.setFormatter(logging.Formatter(("%(name)s - %(levelname)s - %(message)s")))
-llm_logger.addHandler(handler)
-
 # console logger (if you want to print specific colors)
 console_logger = logging.getLogger("console")
 console_logger.setLevel(logging.INFO)
 console_logger.handlers.clear()
 console_logger.addHandler(
-    RichHandler(console=console, markup=True, show_time=False, rich_tracebacks=True)
+    RichHandler(console=Console(), markup=True, show_time=False, rich_tracebacks=True)
 )
 
 # all other logs

{model_library-0.1.0 → model_library-0.1.1}/model_library/__init__.py

@@ -1,8 +1,11 @@
-from model_library.settings import ModelLibrarySettings
 from model_library.base import LLM, LLMConfig
+from model_library.logging import set_logging
+from model_library.settings import ModelLibrarySettings
 
 model_library_settings: ModelLibrarySettings = ModelLibrarySettings()
 
+set_logging()
+
 
 def model(model_str: str, override_config: LLMConfig | None = None) -> LLM:
     from model_library.registry_utils import get_registry_model
@@ -10,14 +13,15 @@ def model(model_str: str, override_config: LLMConfig | None = None) -> LLM:
     return get_registry_model(model_str, override_config)
 
 
-def raw_model(model_str: str, override_config: LLMConfig | None = None) -> LLM:
+def raw_model(model_str: str, config: LLMConfig | None = None) -> LLM:
     from model_library.registry_utils import get_raw_model
 
-    return get_raw_model(model_str, config=override_config)
+    return get_raw_model(model_str, config=config)
 
 
 __all__ = [
     "model_library_settings",
     "model",
     "raw_model",
+    "set_logging",
 ]

model_library-0.1.1/model_library/logging.py

@@ -0,0 +1,30 @@
+import logging
+
+from rich.console import Console
+from rich.logging import RichHandler
+
+_llm_logger = logging.getLogger("llm")
+
+
+def set_logging(enable: bool = True, handler: logging.Handler | None = None):
+    """
+    Sets up logging for the model library
+
+    Args:
+        enable (bool): Enable or disable logging.
+        handler (logging.Handler, optional): A custom logging handler. Defaults to RichHandler.
+    """
+    if enable:
+        _llm_logger.setLevel(logging.INFO)
+    else:
+        _llm_logger.setLevel(logging.CRITICAL)
+
+    if not enable or _llm_logger.hasHandlers():
+        return
+
+    if handler is None:
+        console = Console()
+        handler = RichHandler(console=console, markup=True, show_time=False)
+
+    handler.setFormatter(logging.Formatter("%(name)s - %(levelname)s - %(message)s"))
+    _llm_logger.addHandler(handler)

{model_library-0.1.0 → model_library-0.1.1}/model_library.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: model-library
-Version: 0.1.0
+Version: 0.1.1
 Summary: Model Library for vals.ai 
 Author-email: "Vals AI, Inc." <contact@vals.ai>
 License: MIT
@@ -10,6 +10,7 @@ License-File: LICENSE
 Requires-Dist: typing-extensions<5.0,>=4.14.1
 Requires-Dist: pydantic<3.0,>=2.11.7
 Requires-Dist: pyyaml>=6.0.2
+Requires-Dist: rich
 Requires-Dist: backoff<3.0,>=2.2.1
 Requires-Dist: redis<7.0,>=6.2.0
 Requires-Dist: tiktoken==0.11.0
@@ -53,7 +54,14 @@ Open-source model library for interacting with a variety of LLM providers. Origi
 - X AI
 - ZhipuAI (zai)
 
-Run `python -m scripts.browse_models` to browse the model registry.
+Run `python -m scripts.browse_models` to browse the model registry or
+
+```python
+from model_library.registry_utils import get_model_names_by_provider, get_provider_names
+
+print(get_provider_names())
+print(get_model_names_by_provider("chosen-provider"))
+```
 
 ### Supported Input
 
@@ -70,16 +78,26 @@ Here is a basic example of how to query a model:
 
 ```python
 import asyncio
-from model_library.registry_utils import get_registry_model
+from model_library import model
 
 async def main():
     # Load a model from the registry
-    model = get_registry_model("openai/gemini-2.5-flash")
+    llm = model("anthropic/claude-opus-4-1-20250805-thinking")
+
+    # Display the LLM instance
+    llm.logger.info(llm)
+    # or print(llm)
 
     # Query the model with a simple text input
-    response = await model.query("What is QSBS? Explain your thinking in detail and make it concise.")
+    result = await llm.query(
+        "What is QSBS? Explain your thinking in detail and make it concise."
+    )
+
+    # Logger automatically logs the result
+
+    # Display only the output text
+    llm.logger.info(result.output_text)
 
-    # Logger automatically logs the response
 
 if __name__ == "__main__":
     asyncio.run(main())
@@ -88,7 +106,18 @@ if __name__ == "__main__":
 The model registry holds model attributes, ex. reasoning, file support, tool support, max tokens. You may also use models not included in the registry.
 
 ```python
-model = get_raw_model("openai/gpt-3.5-turbo", config=LLMConfig(max_tokens=1000))
+from model_library import raw_model
+from model_library.base import LLMConfig
+
+model = raw_model("grok/grok-code-fast", LLMConfig(max_tokens=10000))
+```
+
+Root logger is named "llm". To disable logging:
+
+```python
+from model_library import set_logging
+
+set_logging(enable=False)
 ```
 
 ### Environment Setup

{model_library-0.1.0 → model_library-0.1.1}/model_library.egg-info/SOURCES.txt

@@ -29,6 +29,7 @@ model_library/__init__.py
 model_library/base.py
 model_library/exceptions.py
 model_library/file_utils.py
+model_library/logging.py
 model_library/model_utils.py
 model_library/py.typed
 model_library/register_models.py

{model_library-0.1.0 → model_library-0.1.1}/model_library.egg-info/requires.txt

@@ -1,6 +1,7 @@
 typing-extensions<5.0,>=4.14.1
 pydantic<3.0,>=2.11.7
 pyyaml>=6.0.2
+rich
 backoff<3.0,>=2.2.1
 redis<7.0,>=6.2.0
 tiktoken==0.11.0

{model_library-0.1.0 → model_library-0.1.1}/pyproject.toml

@@ -14,6 +14,7 @@ dependencies = [
   "typing-extensions>=4.14.1,<5.0",
   "pydantic>=2.11.7,<3.0",
   "pyyaml>=6.0.2",
+  "rich",
   # core
   "backoff>=2.2.1,<3.0",
   "redis>=6.2.0,<7.0",
@@ -39,8 +40,6 @@ dev = [
   # style
   "ruff",
   "basedpyright",
-  # log
-  "rich",
   # tui
   "inquirerpy",
   # env

{model_library-0.1.0 → model_library-0.1.1}/uv.lock

@@ -1,5 +1,5 @@
 version = 1
-revision = 3
+revision = 2
 requires-python = ">=3.11"
 resolution-markers = [
     "python_full_version >= '3.13'",
@@ -796,6 +796,7 @@ dependencies = [
     { name = "pydantic" },
     { name = "pyyaml" },
     { name = "redis" },
+    { name = "rich" },
     { name = "tiktoken" },
     { name = "together" },
     { name = "typing-extensions" },
@@ -810,7 +811,6 @@ dev = [
     { name = "inquirerpy" },
     { name = "pytest" },
     { name = "pytest-asyncio" },
-    { name = "rich" },
     { name = "ruff" },
 ]
 
@@ -828,6 +828,7 @@ requires-dist = [
     { name = "pydantic", specifier = ">=2.11.7,<3.0" },
     { name = "pyyaml", specifier = ">=6.0.2" },
     { name = "redis", specifier = ">=6.2.0,<7.0" },
+    { name = "rich" },
     { name = "tiktoken", specifier = "==0.11.0" },
     { name = "together", specifier = ">=1.5.25,<2.0" },
     { name = "typing-extensions", specifier = ">=4.14.1,<5.0" },
@@ -842,7 +843,6 @@ dev = [
     { name = "inquirerpy" },
     { name = "pytest" },
     { name = "pytest-asyncio" },
-    { name = "rich" },
     { name = "ruff" },
 ]