unique_toolkit 1.37.0__py3-none-any.whl → 1.38.0__py3-none-any.whl

unique_toolkit/_common/validators.py

@@ -61,17 +61,8 @@ def validate_and_init_language_model_info(
         LanguageModelInfo: The validated and initialized LanguageModelInfo object.
 
     """
-    if isinstance(v, LanguageModelName):
+    if isinstance(v, LanguageModelName | str):
         return LanguageModelInfo.from_name(v)
-    if isinstance(v, str):
-        if v in [name.value for name in LanguageModelName]:
-            return LanguageModelInfo.from_name(LanguageModelName(v))
-
-        return LanguageModelInfo(
-            name=v,
-            version="custom",
-            provider=LanguageModelProvider.CUSTOM,
-        )
 
     return v
 

unique_toolkit/language_model/infos.py

@@ -1,5 +1,9 @@
+import json
+import logging
+import os
 from datetime import date
 from enum import StrEnum
+from functools import lru_cache
 from typing import Annotated, Any, ClassVar, Optional, Self
 
 from pydantic import BaseModel, Field
@@ -9,6 +13,8 @@ from typing_extensions import deprecated
 from unique_toolkit._common.pydantic_helpers import get_configuration_dict
 from unique_toolkit.language_model.schemas import LanguageModelTokenLimits
 
+_LOGGER = logging.getLogger(__name__)
+
 
 class LanguageModelName(StrEnum):
     AZURE_GPT_35_TURBO_0125 = "AZURE_GPT_35_TURBO_0125"
@@ -202,8 +208,84 @@ class LanguageModelInfo(BaseModel):
 
     default_options: dict[str, Any] = {}
 
+    _ENV_VAR: ClassVar[str] = "LANGUAGE_MODEL_INFOS"
+    """Environment variable name for custom language model infos."""
+
+    @classmethod
+    @lru_cache(maxsize=1)
+    def _load_from_env(cls) -> dict[str, dict]:
+        """
+        Load custom language model infos from environment variable.
+
+        The environment variable should contain a JSON string with a dict of
+        LanguageModelInfo-compatible dictionaries. The key is used for model lookup.
+
+        Example:
+            LANGUAGE_MODEL_INFOS='{"AZURE_GPT_4o_CUSTOM": {"name": "AZURE_GPT_4o_2024_1120",
+            "provider": "AZURE", "version": "custom", "capabilities": ["function_calling",
+            "streaming", "vision"], "token_limits": {"token_limit_input": 3000,
+            "token_limit_output": 150}}}'
+
+        Returns:
+            A dictionary mapping model keys to their info dictionaries.
+        """
+        env_value = os.getenv(cls._ENV_VAR)
+        if not env_value:
+            return {}
+
+        try:
+            model_infos_dict = json.loads(env_value)
+            if not isinstance(model_infos_dict, dict):
+                _LOGGER.error(
+                    f"{cls._ENV_VAR} must be a JSON dict of model info objects. "
+                    f"Got {type(model_infos_dict).__name__} instead."
+                )
+                return {}
+
+            # Validate each entry in the dictionary
+            valid_model_infos: dict[str, dict] = {}
+            for model_key, model_info in model_infos_dict.items():
+                if not isinstance(model_info, dict):
+                    _LOGGER.warning(
+                        f"Skipping invalid model info entry '{model_key}' in {cls._ENV_VAR}: "
+                        f"expected dict, got {type(model_info).__name__}"
+                    )
+                    continue
+
+                valid_model_infos[model_key] = model_info
+
+            _LOGGER.debug(
+                f"Loaded {len(valid_model_infos)} custom language model infos from {cls._ENV_VAR}"
+            )
+            return valid_model_infos
+
+        except json.JSONDecodeError:
+            _LOGGER.error(
+                f"Failed to parse {cls._ENV_VAR} as JSON. "
+                "The environment variable should contain a valid JSON dict of model info objects.",
+                exc_info=True,
+            )
+            return {}
+
     @classmethod
-    def from_name(cls, model_name: LanguageModelName) -> Self:
+    def from_name(cls, model_name: LanguageModelName | str) -> Self:
+        # Check environment variable first - env definitions take precedence
+        env_model_infos = cls._load_from_env()
+        model_name_str = (
+            model_name.value
+            if isinstance(model_name, LanguageModelName)
+            else model_name
+        )
+        if model_name_str in env_model_infos.keys():
+            try:
+                return cls.model_validate(env_model_infos[model_name_str])
+            except Exception:
+                _LOGGER.warning(
+                    f"Failed to parse model info for '{model_name_str}' from "
+                    f"{cls._ENV_VAR}. Falling back to default definition.",
+                    exc_info=True,
+                )
+
         match model_name:
             case LanguageModelName.AZURE_GPT_35_TURBO_0125:
                 return cls(
@@ -1723,14 +1805,7 @@ class LanguageModel:
 
     @classmethod
     def get_model_info(cls, model_name: LanguageModelName | str) -> LanguageModelInfo:
-        if isinstance(model_name, LanguageModelName):
-            return LanguageModelInfo.from_name(model_name)
-
-        return LanguageModelInfo(
-            name=model_name,
-            version="custom",
-            provider=LanguageModelProvider.CUSTOM,
-        )
+        return LanguageModelInfo.from_name(model_name)
 
     @classmethod
     def list_models(cls) -> list[LanguageModelInfo]:

{unique_toolkit-1.37.0.dist-info → unique_toolkit-1.38.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.37.0
+Version: 1.38.0
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -121,6 +121,9 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.38.0] - 2025-12-15
+- Including capability to load LanguageModelInfos in env variable
+
 ## [1.37.0] - 2025-12-15
 - Adding a prompt appendix to enforce forced tool calls when using Qwen models
 

{unique_toolkit-1.37.0.dist-info → unique_toolkit-1.38.0.dist-info}/RECORD

@@ -39,7 +39,7 @@ unique_toolkit/_common/utils/structured_output/__init__.py,sha256=nm_orZrlCXL0FP
 unique_toolkit/_common/utils/structured_output/schema.py,sha256=Tp7kDYcmKtnUhcuRkH86TSYhylRff0ZZJYb2dLkISts,131
 unique_toolkit/_common/utils/write_configuration.py,sha256=fzvr4C-XBL3OSM3Od9TbqIxeeDS9_d9CLEyTq6DDknY,1409
 unique_toolkit/_common/validate_required_values.py,sha256=Y_M1ub9gIKP9qZ45F6Zq3ZHtuIqhmOjl8Z2Vd3avg8w,588
-unique_toolkit/_common/validators.py,sha256=LFZmAalNa886EXm1VYamFvfBuUZjYKwDdT_HOYU0BtE,2934
+unique_toolkit/_common/validators.py,sha256=ElnkMsyEY24TfzfTVHvireyT39EnZgW5N40T0P4b6gE,2638
 unique_toolkit/agentic/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 unique_toolkit/agentic/debug_info_manager/debug_info_manager.py,sha256=30ZZaw0vffjZjiu9AYdO1Sm8G9FN6XR2ehdOEUCKqh0,891
 unique_toolkit/agentic/debug_info_manager/test/test_debug_info_manager.py,sha256=_fIS6_DHA8A3AB64-LPgHgUGa1w0CFUWwtgV-ZbhkzA,10535
@@ -194,7 +194,7 @@ unique_toolkit/language_model/builder.py,sha256=4OKfwJfj3TrgO1ezc_ewIue6W7BCQ2ZY
 unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
 unique_toolkit/language_model/default_language_model.py,sha256=-_DBsJhLCsFdaU4ynAkyW0jYIl2lhrPybZm1K-GgVJs,125
 unique_toolkit/language_model/functions.py,sha256=PTBm2BBkuqISVHoyUqMIGHGXT-RMSAfz0F_Ylo2esQ8,18246
-unique_toolkit/language_model/infos.py,sha256=sZJOOij-dfReDxJWfd7ZwP3qx4KcN1LVqNchRafKmrY,79877
+unique_toolkit/language_model/infos.py,sha256=Bh6-Fs_xbqqhEPRbEiudcrLPIHYPyfHuzfRZtBVs9I0,82806
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
 unique_toolkit/language_model/reference.py,sha256=nkX2VFz-IrUz8yqyc3G5jUMNwrNpxITBrMEKkbqqYoI,8583
 unique_toolkit/language_model/schemas.py,sha256=ATiHjhfGxoubS332XuhL9PKSoFewcWvPTUVBaNGWlJo,23994
@@ -212,7 +212,7 @@ unique_toolkit/short_term_memory/service.py,sha256=5PeVBu1ZCAfyDb2HLVvlmqSbyzBBu
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=Ozhh70qCn2yOzRWr9d8WmJeTo7AQurwd3tStgBMPFLA,1246
 unique_toolkit/test_utilities/events.py,sha256=_mwV2bs5iLjxS1ynDCjaIq-gjjKhXYCK-iy3dRfvO3g,6410
-unique_toolkit-1.37.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.37.0.dist-info/METADATA,sha256=P8N25tqQqezFmyaOYAXtauCGKHdeGaxMa9kyfd-x1_Q,46284
-unique_toolkit-1.37.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.37.0.dist-info/RECORD,,
+unique_toolkit-1.38.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.38.0.dist-info/METADATA,sha256=OJdgZ3Hlti9zeds9NTOC6OOibFeUsedeFjtZzVZJMTo,46376
+unique_toolkit-1.38.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.38.0.dist-info/RECORD,,