unique_toolkit 2026.24.0.dev5__py3-none-any.whl → 2026.24.0.dev6__py3-none-any.whl

unique_toolkit/__init__.py

@@ -1,23 +1,78 @@
-# Re-export commonly used classes for easier imports
-from unique_toolkit.chat import ChatService
-from unique_toolkit.content import ContentService
-from unique_toolkit.data_extraction import (
-    StructuredOutputDataExtractor,
-    StructuredOutputDataExtractorConfig,
-)
-from unique_toolkit.embedding import EmbeddingService
-from unique_toolkit.framework_utilities.openai.client import (
-    get_async_openai_client,
-    get_openai_client,
-)
-from unique_toolkit.language_model import (
-    LanguageModelMessages,
-    LanguageModelName,
-    LanguageModelService,
-    LanguageModelToolDescription,
-)
-from unique_toolkit.services.knowledge_base import KnowledgeBaseService
-from unique_toolkit.short_term_memory import ShortTermMemoryService
+from importlib import import_module
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    from unique_toolkit.chat import ChatService as ChatService
+    from unique_toolkit.content import ContentService as ContentService
+    from unique_toolkit.data_extraction import (
+        StructuredOutputDataExtractor as StructuredOutputDataExtractor,
+    )
+    from unique_toolkit.data_extraction import (
+        StructuredOutputDataExtractorConfig as StructuredOutputDataExtractorConfig,
+    )
+    from unique_toolkit.embedding import EmbeddingService as EmbeddingService
+    from unique_toolkit.framework_utilities.openai.client import (
+        get_async_openai_client as get_async_openai_client,
+    )
+    from unique_toolkit.framework_utilities.openai.client import (
+        get_openai_client as get_openai_client,
+    )
+    from unique_toolkit.language_model import (
+        LanguageModelMessages as LanguageModelMessages,
+    )
+    from unique_toolkit.language_model import LanguageModelName as LanguageModelName
+    from unique_toolkit.language_model import (
+        LanguageModelService as LanguageModelService,
+    )
+    from unique_toolkit.language_model import (
+        LanguageModelToolDescription as LanguageModelToolDescription,
+    )
+    from unique_toolkit.services.knowledge_base import (
+        KnowledgeBaseService as KnowledgeBaseService,
+    )
+    from unique_toolkit.short_term_memory import (
+        ShortTermMemoryService as ShortTermMemoryService,
+    )
+
+_EXPORTS: dict[str, tuple[str, str]] = {
+    "ChatService": ("unique_toolkit.chat", "ChatService"),
+    "ContentService": ("unique_toolkit.content", "ContentService"),
+    "EmbeddingService": ("unique_toolkit.embedding", "EmbeddingService"),
+    "get_openai_client": (
+        "unique_toolkit.framework_utilities.openai.client",
+        "get_openai_client",
+    ),
+    "get_async_openai_client": (
+        "unique_toolkit.framework_utilities.openai.client",
+        "get_async_openai_client",
+    ),
+    "KnowledgeBaseService": (
+        "unique_toolkit.services.knowledge_base",
+        "KnowledgeBaseService",
+    ),
+    "LanguageModelMessages": (
+        "unique_toolkit.language_model",
+        "LanguageModelMessages",
+    ),
+    "LanguageModelName": ("unique_toolkit.language_model", "LanguageModelName"),
+    "LanguageModelService": ("unique_toolkit.language_model", "LanguageModelService"),
+    "LanguageModelToolDescription": (
+        "unique_toolkit.language_model",
+        "LanguageModelToolDescription",
+    ),
+    "ShortTermMemoryService": (
+        "unique_toolkit.short_term_memory",
+        "ShortTermMemoryService",
+    ),
+    "StructuredOutputDataExtractor": (
+        "unique_toolkit.data_extraction",
+        "StructuredOutputDataExtractor",
+    ),
+    "StructuredOutputDataExtractorConfig": (
+        "unique_toolkit.data_extraction",
+        "StructuredOutputDataExtractorConfig",
+    ),
+}
 
 # Conditionally import config_checker (requires pydantic-settings)
 _CONFIG_CHECKER_AVAILABLE: bool = False
@@ -83,3 +138,12 @@ if _CONFIG_CHECKER_AVAILABLE:
 # Add langchain-specific exports if available
 if _LANGCHAIN_AVAILABLE:
     __all__.append("get_langchain_client")
+
+
+def __getattr__(name: str) -> Any:
+    export = _EXPORTS.get(name)
+    if export is None:
+        raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+    module_name, attribute_name = export
+    return getattr(import_module(module_name), attribute_name)

unique_toolkit/app/schemas.py

@@ -4,10 +4,17 @@ import json
 from enum import StrEnum
 from logging import getLogger
 from pathlib import Path
-from typing import Any, Generic, Optional, TypeVar, override
+from typing import Annotated, Any, Generic, Optional, TypeVar, override
 
 from humps import camelize
-from pydantic import BaseModel, ConfigDict, Field, JsonValue, field_validator
+from pydantic import (
+    BaseModel,
+    BeforeValidator,
+    ConfigDict,
+    Field,
+    JsonValue,
+    field_validator,
+)
 from pydantic_settings import BaseSettings
 from typing_extensions import deprecated
 
@@ -16,6 +23,8 @@ from unique_toolkit.app.chat_event_filter_options_settings import (
     CHAT_EVENT_FILTER_OPTIONS_SETTINGS,
 )
 from unique_toolkit.app.unique_settings import UniqueChatEventFilterOptions
+from unique_toolkit.language_model.default_language_model import DEFAULT_LANGUAGE_MODEL
+from unique_toolkit.language_model.infos import LanguageModelInfo
 from unique_toolkit.smart_rules.compile import UniqueQL, parse_uniqueql
 
 FilterOptionsT = TypeVar("FilterOptionsT", bound=BaseSettings)
@@ -29,6 +38,15 @@ model_config = ConfigDict(
 _logger = getLogger(__name__)
 
 
+def _validate_language_model_info(v: Any) -> Any:
+    from unique_toolkit._common.validators import validate_and_init_language_model_info
+
+    return validate_and_init_language_model_info(v)
+
+
+LMI = Annotated[LanguageModelInfo, BeforeValidator(_validate_language_model_info)]
+
+
 class EventName(StrEnum):
     EXTERNAL_MODULE_CHOSEN = "unique.chat.external-module.chosen"
     USER_MESSAGE_CREATED = "unique.chat.user-message.created"
@@ -247,6 +265,10 @@ class ChatEventPayload(BaseModel):
     user_metadata: dict[str, Any] | None = Field(
         default_factory=dict,
     )
+    model_choice: LMI = Field(
+        default_factory=lambda: LanguageModelInfo.from_name(DEFAULT_LANGUAGE_MODEL),
+        description="The model choice the user has chosen to be used.",
+    )
     tool_choices: list[str] = Field(
         default_factory=list,
         description="A list containing the tool names the user has chosen to be activated.",
@@ -305,6 +327,10 @@ class ChatEventPayload(BaseModel):
         if value:
             return parse_uniqueql(value)
 
+    @property
+    def has_model_choice_override(self) -> bool:
+        return "model_choice" in self.model_fields_set
+
 
 @deprecated("""Use `ChatEventPayload` instead.
             This class will be removed in the next major version.""")

unique_toolkit/content/__init__.py

@@ -1,3 +1,5 @@
+from typing import TYPE_CHECKING, Any
+
 from .constants import DOMAIN_NAME as DOMAIN_NAME
 from .schemas import (
     Content as Content,
@@ -23,7 +25,6 @@ from .schemas import (
 from .schemas import (
     ContentUploadInput as ContentUploadInput,
 )
-from .service import ContentService as ContentService
 from .utils import (
     count_tokens as count_tokens,
 )
@@ -36,3 +37,31 @@ from .utils import (
 from .utils import (
     sort_content_chunks as sort_content_chunks,
 )
+
+if TYPE_CHECKING:
+    from .service import ContentService as ContentService
+
+__all__ = [
+    "Content",
+    "ContentChunk",
+    "ContentMetadata",
+    "ContentReference",
+    "ContentRerankerConfig",
+    "ContentSearchResult",
+    "ContentSearchType",
+    "ContentService",
+    "ContentUploadInput",
+    "DOMAIN_NAME",
+    "count_tokens",
+    "merge_content_chunks",
+    "pick_content_chunks_for_token_window",
+    "sort_content_chunks",
+]
+
+
+def __getattr__(name: str) -> Any:
+    if name == "ContentService":
+        from .service import ContentService
+
+        return ContentService
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

unique_toolkit/language_model/__init__.py

@@ -1,3 +1,5 @@
+from typing import TYPE_CHECKING, Any
+
 from .constants import DOMAIN_NAME
 from .default_language_model import DEFAULT_LANGUAGE_MODEL, DEFAULT_GPT_4o
 from .functions import (
@@ -28,9 +30,6 @@ from .schemas import (
     LanguageModelToolParameters,
     LanguageModelUserMessage,
 )
-from .service import (
-    LanguageModelService,
-)
 from .stream_transform import (
     NormalizationTransform,
     ReferenceInjectionTransform,
@@ -42,6 +41,19 @@ from .utils import (
     find_last_json_object,
 )
 
+if TYPE_CHECKING:
+    from .service import LanguageModelService as LanguageModelService
+
+
+def __getattr__(name: str) -> Any:
+    if name == "LanguageModelService":
+        from .service import LanguageModelService
+
+        return LanguageModelService
+
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
 __all__ = [
     "LanguageModel",
     "LanguageModelName",

{unique_toolkit-2026.24.0.dev5.dist-info → unique_toolkit-2026.24.0.dev6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.24.0.dev5
+Version: 2026.24.0.dev6
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>

{unique_toolkit-2026.24.0.dev5.dist-info → unique_toolkit-2026.24.0.dev6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-unique_toolkit/__init__.py,sha256=fuduVSYSck3dtXBDTlfgXHG0kIb9b01YzlL0RGHibAk,2714
+unique_toolkit/__init__.py,sha256=wLqk58sHN8n2-UzUqSdJKq3duNIpIBsXo453d3ueGto,5153
 unique_toolkit/_common/__init__.py,sha256=jfbOCUi5zumgGpKrIanIuTFY7f7_yfToRg7o64j6Egk,46
 unique_toolkit/_common/_base_service.py,sha256=S8H0rAebx7GsOldA7xInLp3aQJt9yEPDQdsGSFRJsGg,276
 unique_toolkit/_common/_time_utils.py,sha256=ztmTovTvr-3w71Ns2VwXC65OKUUh-sQlzbHdKTQWm-w,135
@@ -233,7 +233,7 @@ unique_toolkit/app/init_logging.py,sha256=Lv0FuEiUmxvtV2yovjD0gWNIvLdsi2xdMFeEiN
 unique_toolkit/app/init_sdk.py,sha256=5_oDoETr6akwYyBCb0ivTdMNu3SVgPSkrXcDS6ELyY8,2269
 unique_toolkit/app/performance/async_tasks.py,sha256=8Xv5kEmmFDJO7YyyBxv4e7bcNgq-3z_E2_4z1Yn9EqU,1970
 unique_toolkit/app/performance/async_wrapper.py,sha256=yVVcRDkcdyfjsxro-N29SBvi-7773wnfDplef6-y8xw,1077
-unique_toolkit/app/schemas.py,sha256=9N82MqgNMtsb-ertRfhviKs2CFEWFx-02A-Ff_PF7eo,13960
+unique_toolkit/app/schemas.py,sha256=f5h9BqCuqHvrXUAB7tjrx0PSLmuJJKlZ89othqOsuSY,14762
 unique_toolkit/app/unique_settings.py,sha256=bUrT57CIhu8fnU5pdEboZ4Ic6BPG1FetfsOh0mwlHoM,22704
 unique_toolkit/app/verification.py,sha256=GxFFwcJMy25fCA_Xe89wKW7bgqOu8PAs5y8QpHF0GSc,3861
 unique_toolkit/app/webhook.py,sha256=evpxLDnvz7P3D7-TV1XBkl8WcoldXpjOkw3tvFRxHr0,2368
@@ -248,7 +248,7 @@ unique_toolkit/chat/schemas.py,sha256=l-Epugu27RoajiX5OkFO-BrVNey-wfTmaXiEmNNmG7
 unique_toolkit/chat/service.py,sha256=6D00OL4QrGafbOhTaC5zNXaNgg7gS5W_2ePVa4LhqpE,4439
 unique_toolkit/chat/state.py,sha256=Cjgwv_2vhDFbV69xxsn7SefhaoIAEqLx3ferdVFCnOg,1445
 unique_toolkit/chat/utils.py,sha256=CsD9IER2gA1IOV0fmUOl-N0LmKvA7JP9dGrOqeOY0Gg,1173
-unique_toolkit/content/__init__.py,sha256=EdJg_A_7loEtCQf4cah3QARQreJx6pdz89Rm96YbMVg,940
+unique_toolkit/content/__init__.py,sha256=nnQDfnNTEvI8-3dgWvIojoFDYU7edwd8Kloi6lMDG0M,1586
 unique_toolkit/content/constants.py,sha256=1iy4Y67xobl5VTnJB6SxSyuoBWbdLl9244xfVMUZi5o,60
 unique_toolkit/content/functions.py,sha256=Rx3N9x23JjVt7mmUlFhOhjAjkWQCzzodtpDgoIsE2xg,33932
 unique_toolkit/content/schemas.py,sha256=z3xlTkOh9MTwHGLwcc5vfT5OmDRCiWE0F0MaiPMV4dg,10281
@@ -361,7 +361,7 @@ unique_toolkit/framework_utilities/openai/__init__.py,sha256=CrHYoC7lv2pBscitLer
 unique_toolkit/framework_utilities/openai/client.py,sha256=Ft-oLiS6BTaMiuCdObdaMLkiEPE3RYjtc9VPnqQk3W8,2449
 unique_toolkit/framework_utilities/openai/message_builder.py,sha256=RT1pZjxH42TFZlAxQ5zlqdKPvHKVTjc5t3JDUy58I7Q,6887
 unique_toolkit/framework_utilities/utils.py,sha256=JK7g2yMfEx3eMprug26769xqNpS5WJcizf8n2zWMBng,789
-unique_toolkit/language_model/__init__.py,sha256=U5AhmPWmDnkaZPU-Zzyf2KwACzJ58dRyYLVpRIRTqhQ,2290
+unique_toolkit/language_model/__init__.py,sha256=WleWs4UXFzVOVglKYGkyfmogY-cuUyOKYHvHhWmy4LY,2607
 unique_toolkit/language_model/_responses_api_utils.py,sha256=w2ZyleOQcWLpOWGJNNb2-uBYApUNlqS-ABqXoMFySlg,3084
 unique_toolkit/language_model/builder.py,sha256=KwBzsD6-fvQnB4p2I7M-5Tk6j4tmQ973MYwU1ulXFb4,4526
 unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
@@ -391,6 +391,6 @@ unique_toolkit/short_term_memory/service.py,sha256=Ncw0_0CNM6kmzJSLVJGG5GcKpBduI
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=vRe0aMJWuVIK-VsqomgxQImzke7MDmwgSxxJfYCa7zw,1386
 unique_toolkit/test_utilities/events.py,sha256=FLrhMh0A_mQutAUjoD0FdMcVQ4Gq31r35OVGzRjvLro,6760
-unique_toolkit-2026.24.0.dev5.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.24.0.dev5.dist-info/METADATA,sha256=c9BRIxMsywMfMEN91oVAv0NOEOi2KlnDULi-ohknj_c,7392
-unique_toolkit-2026.24.0.dev5.dist-info/RECORD,,
+unique_toolkit-2026.24.0.dev6.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.24.0.dev6.dist-info/METADATA,sha256=9iqeteqQZ9CeoDz6t3gI0L0YdK6AnpC376d7cqfdps4,7392
+unique_toolkit-2026.24.0.dev6.dist-info/RECORD,,