model-library 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

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/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.dist-info → model_library-0.1.1.dist-info}/METADATA

@@ -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.dist-info → model_library-0.1.1.dist-info}/RECORD

@@ -1,7 +1,8 @@
-model_library/__init__.py,sha256=9EDDz6UKJG21p7mrak6Rxzr-DUMOwCgE8-yAWP1-W_s,652
+model_library/__init__.py,sha256=AKc_15aklOf-LbcS9z1Xer_moRWNpG6Dh3kqvSQ0nOI,714
 model_library/base.py,sha256=JRzDZkYhzlEarknC0gX0sRplfxoFaqSb47gYhQy57sA,23834
 model_library/exceptions.py,sha256=FnEQXTeC1GpcMEpwukGK7Uwu1_Bnvl87MUs1zOqVm0o,8750
 model_library/file_utils.py,sha256=vLxYWI0-kwp67UONcFdZw2qDTV38P7IZLBaXFJDNtO4,3666
+model_library/logging.py,sha256=McyaPHUk7RkB38-LrfnudrrU1B62ta8wAbbIBwLRmj0,853
 model_library/model_utils.py,sha256=l8oCltGeimMGtnne_3Q1EguVtzCj61UMsLsma-1czwg,753
 model_library/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 model_library/register_models.py,sha256=JOSt_VdtsllIshRGKVzKWG9DT7cFHd9JsjZLysi7cUw,13267
@@ -46,8 +47,8 @@ model_library/providers/zai.py,sha256=4Ui-2bPQ7m0NwQLJAaMezbLUNZOqBfWzq1JCR8f65N
 model_library/providers/google/__init__.py,sha256=ypuLVL_QJEQ7C3S47FhC9y4wyawYOdGikAViJmACI0U,115
 model_library/providers/google/batch.py,sha256=4TE90Uo1adi54dVtGcGyUAxw11YExJq-Y4KmkQ-cyHA,9978
 model_library/providers/google/google.py,sha256=nsfiA6x0Mo6e6Q307ogA99fAOUqiEqWoMjmbLvFUikQ,16347
-model_library-0.1.0.dist-info/licenses/LICENSE,sha256=x6mf4o7U_wHaaqcfxoU-0R6uYJLbqL_TNuoULP3asaA,1070
-model_library-0.1.0.dist-info/METADATA,sha256=mwsJnEXNJ6A9mXXoFkVeSyOdgYV9yosX6WoVMoiIAW4,6490
-model_library-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-model_library-0.1.0.dist-info/top_level.txt,sha256=HtQYxA_7RP8UT35I6VcUw20L6edI0Zf2t5Ys1uDGVjs,14
-model_library-0.1.0.dist-info/RECORD,,
+model_library-0.1.1.dist-info/licenses/LICENSE,sha256=x6mf4o7U_wHaaqcfxoU-0R6uYJLbqL_TNuoULP3asaA,1070
+model_library-0.1.1.dist-info/METADATA,sha256=IjFE4RqayFyEoejuGc680tYVZurKlFSuHq7X949p_Qc,7034
+model_library-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+model_library-0.1.1.dist-info/top_level.txt,sha256=HtQYxA_7RP8UT35I6VcUw20L6edI0Zf2t5Ys1uDGVjs,14
+model_library-0.1.1.dist-info/RECORD,,