unique-toolkit 2026.24.0.dev4__tar.gz → 2026.24.0.dev6__tar.gz

{unique_toolkit-2026.24.0.dev4 → unique_toolkit-2026.24.0.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.24.0.dev4
+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.dev4 → unique_toolkit-2026.24.0.dev6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.24.0.dev4"
+version = "2026.24.0.dev6"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"

unique_toolkit-2026.24.0.dev6/unique_toolkit/__init__.py

@@ -0,0 +1,149 @@
+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
+try:
+    from unique_toolkit._common.config_checker import (
+        ConfigDiffer,  # noqa: F401, I001
+        ConfigEntry,  # noqa: F401, I001
+        ConfigExporter,  # noqa: F401, I001
+        ConfigRegistry,  # noqa: F401, I001
+        ConfigValidator,  # noqa: F401, I001
+        DefaultChangeReport,  # noqa: F401, I001
+        ExportManifest,  # noqa: F401, I001
+        ValidationReport,  # noqa: F401, I001
+        register_config,  # noqa: F401, I001
+    )
+
+    _CONFIG_CHECKER_AVAILABLE = True  # pyright: ignore[reportConstantRedefinition]
+except ImportError:
+    pass
+
+# Conditionally import langchain utilities if langchain is installed
+_LANGCHAIN_AVAILABLE: bool = False
+try:
+    from unique_toolkit.framework_utilities.langchain.client import get_langchain_client  # noqa: F401, I001
+
+    _LANGCHAIN_AVAILABLE = True  # pyright: ignore[reportConstantRedefinition]
+except ImportError:
+    pass
+
+# You can add other classes you frequently use here as well
+
+__all__ = [
+    "ChatService",
+    "ContentService",
+    "EmbeddingService",
+    "get_openai_client",
+    "get_async_openai_client",
+    "KnowledgeBaseService",
+    "LanguageModelMessages",
+    "LanguageModelName",
+    "LanguageModelService",
+    "LanguageModelToolDescription",
+    "ShortTermMemoryService",
+    "StructuredOutputDataExtractor",
+    "StructuredOutputDataExtractorConfig",
+]
+
+# Add config_checker exports if available
+if _CONFIG_CHECKER_AVAILABLE:
+    __all__.extend(
+        [
+            "ConfigDiffer",
+            "ConfigEntry",
+            "ConfigExporter",
+            "ConfigRegistry",
+            "ConfigValidator",
+            "DefaultChangeReport",
+            "ExportManifest",
+            "ValidationReport",
+            "register_config",
+        ]
+    )
+# 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-2026.24.0.dev4 → unique_toolkit-2026.24.0.dev6}/unique_toolkit/_common/metadata_filter_scope.py

@@ -8,6 +8,8 @@ from typing import Any
 from unique_toolkit.content.smart_rules import (
     Operator,
     Statement,
+    UniqueQL,
+    uniqueql_to_dict,
 )
 
 _UQL_AND = "and"
@@ -29,16 +31,17 @@ def build_folder_id_in_clause(scope_ids: list[str]) -> dict[str, Any]:
 
 def merge_scope_clause_into_metadata_filter(
     scope_clause: Mapping[str, Any],
-    metadata_filter: dict[str, Any] | None,
+    metadata_filter: UniqueQL | Mapping[str, Any] | None,
 ) -> dict[str, Any]:
     """AND-merge *scope_clause* with *metadata_filter*, flattening a top-level ``and``."""
     scope_dict = dict(scope_clause)
-    if not metadata_filter:
+    metadata_dict = uniqueql_to_dict(metadata_filter)
+    if not metadata_dict:
         return scope_dict
-    inner_and = metadata_filter.get(_UQL_AND)
+    inner_and = metadata_dict.get(_UQL_AND)
     if isinstance(inner_and, list):
         return {_UQL_AND: [scope_dict, *inner_and]}
-    return {_UQL_AND: [scope_dict, dict(metadata_filter)]}
+    return {_UQL_AND: [scope_dict, metadata_dict]}
 
 
 __all__ = [

{unique_toolkit-2026.24.0.dev4 → unique_toolkit-2026.24.0.dev6}/unique_toolkit/_common/pydantic/rjsf_tags.py

@@ -974,7 +974,12 @@ def ui_schema_for_model(
             node["items"] = item_node
 
         # Dict -> additionalProperties (value side)
-        elif origin is dict:
+        # Skip when the node already carries a ui:widget (fully-specified
+        # widget) or an anyOf (fully-specified type branching): walking the
+        # dict value type in those cases adds a spurious "additionalProperties"
+        # node that confuses RJSF renderers.  Presentation-only meta such as
+        # ui:expandable or ui:title does NOT prevent the walk.
+        elif origin is dict and "ui:widget" not in node and "anyOf" not in node:
             _key_t, val_t = get_args(base) or (Any, Any)
             val_base, val_meta = _walk_annotated_chain(val_t)
             val_base = _unwrap_optional(val_base)

{unique_toolkit-2026.24.0.dev4 → unique_toolkit-2026.24.0.dev6}/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-2026.24.0.dev4 → unique_toolkit-2026.24.0.dev6}/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-2026.24.0.dev4 → unique_toolkit-2026.24.0.dev6}/unique_toolkit/content/smart_rules.py

@@ -1,10 +1,13 @@
+import json
 import re
+from collections.abc import Mapping
 from datetime import datetime, timedelta, timezone
 from enum import Enum
-from typing import Any, Dict, List, Mapping, Self, Union
+from typing import Annotated, Any, Dict, List, Self, Union
 
-from pydantic import AliasChoices, BaseModel, Field
+from pydantic import AliasChoices, BaseModel, BeforeValidator, Field, PlainSerializer
 from pydantic.config import ConfigDict
+from pydantic.json_schema import WithJsonSchema
 
 
 class Operator(str, Enum):
@@ -54,6 +57,10 @@ class BaseStatement(BaseModel):
     ) -> Self:
         return self.model_copy()
 
+    def to_dict(self) -> dict[str, Any]:
+        """Convert to wire-format JSON dict (uses serialization aliases)."""
+        return self.model_dump(mode="json")
+
 
 class Statement(BaseStatement):
     operator: Operator
@@ -285,7 +292,6 @@ def get_fallback_values(
     return values
 
 
-# Example usage:
 def parse_uniqueql(json_data: Dict[str, Any]) -> UniqueQL:
     if "operator" in json_data:
         return Statement.model_validate(json_data)
@@ -299,3 +305,55 @@ def parse_uniqueql(json_data: Dict[str, Any]) -> UniqueQL:
         )
     else:
         raise ValueError("Invalid UniqueQL format")
+
+
+def parse_uniqueql_input(v: Any) -> UniqueQL | None:
+    """Parse config/API input (JSON string, dict, or model) into a UniqueQL model."""
+    if v is None:
+        return None
+    if isinstance(v, (Statement, AndStatement, OrStatement)):
+        return v
+    if isinstance(v, str):
+        if v.strip() == "":
+            return None
+        try:
+            v = json.loads(v)
+        except json.JSONDecodeError as e:
+            raise ValueError(f"Invalid JSON: {e}") from e
+    if isinstance(v, dict):
+        return parse_uniqueql(v)
+    raise ValueError(f"Expected JSON string or dict, got {type(v).__name__}")
+
+
+def uniqueql_to_dict(
+    value: UniqueQL | Mapping[str, Any] | None,
+) -> dict[str, Any] | None:
+    """Convert a UniqueQL model (or wire dict) to a JSON-serializable dict for APIs."""
+    if value is None:
+        return None
+    if isinstance(value, BaseStatement):
+        return value.to_dict()
+    return dict(value)
+
+
+def _serialize_uniqueql_field(value: UniqueQL | None) -> dict[str, Any] | None:
+    return None if value is None else value.to_dict()
+
+
+_UNIQUEQL_JSON_SCHEMA = {
+    "anyOf": [
+        {"type": "string", "title": "UniqueQL (JSON)"},
+        {"type": "null", "title": "Deactivated", "description": "None"},
+    ]
+}
+
+UniqueQLField = Annotated[
+    UniqueQL | None,
+    BeforeValidator(parse_uniqueql_input),
+    WithJsonSchema(_UNIQUEQL_JSON_SCHEMA),
+    PlainSerializer(
+        _serialize_uniqueql_field,
+        when_used="json",
+        return_type=dict[str, Any] | None,
+    ),
+]

{unique_toolkit-2026.24.0.dev4 → unique_toolkit-2026.24.0.dev6}/unique_toolkit/experimental/components/internal_search/knowledge_base/config.py

@@ -6,8 +6,10 @@ from typing import Annotated, Self
 from pydantic import Field, model_validator
 
 from unique_toolkit._common.config_checker import register_config
+from unique_toolkit._common.pydantic.rjsf_tags import RJSFMetaTag
 from unique_toolkit._common.pydantic_helpers import DeactivatedNone
 from unique_toolkit.content.schemas import ContentRerankerConfig
+from unique_toolkit.content.smart_rules import UniqueQLField
 from unique_toolkit.experimental.components.internal_search.base.config import (
     InternalSearchConfig,
 )
@@ -15,9 +17,11 @@ from unique_toolkit.experimental.components.internal_search.base.config import (
 
 @register_config()
 class KnowledgeBaseInternalSearchConfig(InternalSearchConfig):
-    scope_ids: list[str] | DeactivatedNone = Field(
+    scope_ids: Annotated[
+        Annotated[list[str], Field(title="Scope IDs")] | DeactivatedNone,
+        RJSFMetaTag({"anyOf": [{"items": {"ui:title": "Scope ID"}}, {}]}),
+    ] = Field(
         default=None,
-        title="Active",
         deprecated=True,
         description=(
             "Not accepted for new configs. Use ``metadata_filter`` with "
@@ -26,10 +30,29 @@ class KnowledgeBaseInternalSearchConfig(InternalSearchConfig):
             "a ``folderId in [scope_ids]`` metadata filter at search time."
         ),
     )
-    metadata_filter: dict[str, object] | None = Field(
+    metadata_filter: Annotated[
+        UniqueQLField,
+        # Put textarea attrs on the string branch (index 0) only, not at field level.
+        # If set at field level, RJSF's MultiSchemaField applies ui:widget to every
+        # anyOf branch during option iteration, which crashes on non-string branches.
+        RJSFMetaTag(
+            {
+                "ui:options": {"customValidation": "uniqueql"},
+                "anyOf": [
+                    {
+                        "ui:widget": "textarea",
+                        "ui:placeholder": '{"operator": "equals", "value": "...", "path": ["fieldName"]}',
+                        "ui:emptyValue": "",
+                    },
+                    {},
+                ],
+            }
+        ),
+    ] = Field(
         default=None,
         description=(
-            "Static metadata filter applied to every KB search. "
+            "Static UniqueQL metadata filter applied to every KB search. "
+            "Pass as a JSON string or a plain dict. "
             "Overridden by chat context filter or per-invocation state override."
         ),
     )

{unique_toolkit-2026.24.0.dev4 → unique_toolkit-2026.24.0.dev6}/unique_toolkit/experimental/components/internal_search/knowledge_base/service.py

@@ -105,7 +105,11 @@ class KnowledgeBaseInternalSearchService(
             return cast("dict[str, Any] | None", self._state.metadata_filter_override)
         if self._context.chat is not None:
             return self._context.chat.metadata_filter
-        return self._config.metadata_filter
+        return (
+            self._config.metadata_filter.to_dict()
+            if self._config.metadata_filter
+            else None
+        )
 
     @property
     def _effective_scope_ids(self) -> list[str] | None:

{unique_toolkit-2026.24.0.dev4 → unique_toolkit-2026.24.0.dev6}/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.dev4 → unique_toolkit-2026.24.0.dev6}/unique_toolkit/smart_rules/compile.py

@@ -7,6 +7,7 @@ from unique_toolkit.content.smart_rules import (
     OrStatement,
     Statement,
     UniqueQL,
+    UniqueQLField,
     array_operator,
     binary_operator,
     calculate_current_date,
@@ -19,9 +20,11 @@ from unique_toolkit.content.smart_rules import (
     is_array_of_strings,
     null_operator,
     parse_uniqueql,
+    parse_uniqueql_input,
     replace_tool_parameters_patterns,
     replace_user_metadata_patterns,
     replace_variables,
+    uniqueql_to_dict,
 )
 
 warnings.warn(
@@ -38,6 +41,7 @@ __all__ = [
     "OrStatement",
     "Statement",
     "UniqueQL",
+    "UniqueQLField",
     "array_operator",
     "binary_operator",
     "calculate_current_date",
@@ -50,7 +54,9 @@ __all__ = [
     "is_array_of_strings",
     "null_operator",
     "parse_uniqueql",
+    "parse_uniqueql_input",
     "replace_tool_parameters_patterns",
     "replace_user_metadata_patterns",
     "replace_variables",
+    "uniqueql_to_dict",
 ]

unique_toolkit-2026.24.0.dev4/unique_toolkit/__init__.py

@@ -1,85 +0,0 @@
-# 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
-
-# Conditionally import config_checker (requires pydantic-settings)
-_CONFIG_CHECKER_AVAILABLE: bool = False
-try:
-    from unique_toolkit._common.config_checker import (
-        ConfigDiffer,  # noqa: F401, I001
-        ConfigEntry,  # noqa: F401, I001
-        ConfigExporter,  # noqa: F401, I001
-        ConfigRegistry,  # noqa: F401, I001
-        ConfigValidator,  # noqa: F401, I001
-        DefaultChangeReport,  # noqa: F401, I001
-        ExportManifest,  # noqa: F401, I001
-        ValidationReport,  # noqa: F401, I001
-        register_config,  # noqa: F401, I001
-    )
-
-    _CONFIG_CHECKER_AVAILABLE = True  # pyright: ignore[reportConstantRedefinition]
-except ImportError:
-    pass
-
-# Conditionally import langchain utilities if langchain is installed
-_LANGCHAIN_AVAILABLE: bool = False
-try:
-    from unique_toolkit.framework_utilities.langchain.client import get_langchain_client  # noqa: F401, I001
-
-    _LANGCHAIN_AVAILABLE = True  # pyright: ignore[reportConstantRedefinition]
-except ImportError:
-    pass
-
-# You can add other classes you frequently use here as well
-
-__all__ = [
-    "ChatService",
-    "ContentService",
-    "EmbeddingService",
-    "get_openai_client",
-    "get_async_openai_client",
-    "KnowledgeBaseService",
-    "LanguageModelMessages",
-    "LanguageModelName",
-    "LanguageModelService",
-    "LanguageModelToolDescription",
-    "ShortTermMemoryService",
-    "StructuredOutputDataExtractor",
-    "StructuredOutputDataExtractorConfig",
-]
-
-# Add config_checker exports if available
-if _CONFIG_CHECKER_AVAILABLE:
-    __all__.extend(
-        [
-            "ConfigDiffer",
-            "ConfigEntry",
-            "ConfigExporter",
-            "ConfigRegistry",
-            "ConfigValidator",
-            "DefaultChangeReport",
-            "ExportManifest",
-            "ValidationReport",
-            "register_config",
-        ]
-    )
-# Add langchain-specific exports if available
-if _LANGCHAIN_AVAILABLE:
-    __all__.append("get_langchain_client")