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

{model_library-0.1.0 → model_library-0.1.2}/.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.2}/Makefile

@@ -86,5 +86,5 @@ examples: venv_check
 	@echo "\n✅ All examples completed!"
 	
 browse_models: venv_check
-	@uv run python -m scripts.explore_models
+	@uv run python -m scripts.browse_models
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: model-library
-Version: 0.1.0
+Version: 0.1.2
 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.2}/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.2}/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.2}/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.2/model_library/base/__init__.py

@@ -0,0 +1,7 @@
+# ruff: noqa: F403,F401
+from model_library.base.base import *
+from model_library.base.batch import *
+from model_library.base.delegate_only import *
+from model_library.base.input import *
+from model_library.base.output import *
+from model_library.base.utils import *