unique_toolkit 2026.38.0.dev2__py3-none-any.whl → 2026.38.0.dev4__py3-none-any.whl

unique_toolkit/experimental/components/content_tree/functions.py

@@ -19,14 +19,21 @@ from __future__ import annotations
 import asyncio
 import json
 import logging
-from collections.abc import Awaitable, Callable
+from collections.abc import Awaitable, Callable, Mapping
 from pathlib import PurePosixPath
-from typing import Any
+from typing import Any, cast
 
 import unique_sdk
 from typing_extensions import deprecated
 
 from unique_toolkit.content.schemas import BaseFolderInfo, ContentInfo
+from unique_toolkit.content.smart_rules import (
+    AndStatement,
+    Operator,
+    OrStatement,
+    UniqueQL,
+    parse_uniqueql,
+)
 from unique_toolkit.experimental.components.content_tree.schemas import (
     FolderWalkSnapshot,
     PathTrieNode,
@@ -69,6 +76,137 @@ def _parse_content_infos_payload(payload: Any) -> tuple[list[ContentInfo], int]:
     return files, total
 
 
+_NULLISH_OPERATORS = {
+    Operator.IS_NULL.value,
+    Operator.IS_NOT_NULL.value,
+    Operator.IS_EMPTY.value,
+    Operator.IS_NOT_EMPTY.value,
+}
+
+
+def _content_uniqueql_record(info: ContentInfo) -> dict[str, Any]:
+    dumped = info.model_dump(by_alias=True, mode="json")
+    metadata = dumped.get("metadata") or {}
+    if not isinstance(metadata, dict):
+        metadata = {}
+    record = {
+        key: value
+        for key, value in dumped.items()
+        if key != "metadata" and value is not None
+    }
+    record["metadata"] = metadata
+    for key, value in metadata.items():
+        record.setdefault(key, value)
+    return record
+
+
+def _uniqueql_fill_nullish_values(node: Any) -> Any:
+    """Statement.value cannot be None; documented isNull/isEmpty UniqueQL uses null."""
+    if not isinstance(node, dict):
+        return node
+    if "and" in node:
+        return {
+            **node,
+            "and": [_uniqueql_fill_nullish_values(item) for item in node["and"]],
+        }
+    if "or" in node:
+        return {
+            **node,
+            "or": [_uniqueql_fill_nullish_values(item) for item in node["or"]],
+        }
+    filled = dict(node)
+    if filled.get("operator") in _NULLISH_OPERATORS and filled.get("value") is None:
+        filled["value"] = True
+    if isinstance(filled.get("value"), dict):
+        filled["value"] = _uniqueql_fill_nullish_values(filled["value"])
+    return filled
+
+
+def _uniqueql_matches(record: Mapping[str, Any], query: UniqueQL) -> bool:
+    if isinstance(query, AndStatement):
+        return bool(query.and_list) and all(
+            _uniqueql_matches(record, child) for child in query.and_list
+        )
+    if isinstance(query, OrStatement):
+        return bool(query.or_list) and any(
+            _uniqueql_matches(record, child) for child in query.or_list
+        )
+    actual: Any = record
+    for key in query.path:
+        if not isinstance(actual, Mapping):
+            actual = None
+            break
+        actual = actual.get(key)
+    expected = query.value
+    operator = query.operator
+    is_empty = actual is None or actual in ("", [], {})
+    match operator:
+        case Operator.EQUALS:
+            return actual == expected
+        case Operator.NOT_EQUALS:
+            return actual != expected
+        case Operator.CONTAINS:
+            try:
+                return expected in cast(Any, actual)
+            except TypeError:
+                return False
+        case Operator.NOT_CONTAINS:
+            try:
+                return expected not in cast(Any, actual)
+            except TypeError:
+                return True
+        case Operator.IN:
+            return isinstance(expected, list) and actual in expected
+        case Operator.NOT_IN:
+            return isinstance(expected, list) and actual not in expected
+        case Operator.OVERLAPS:
+            return (
+                isinstance(actual, list)
+                and isinstance(expected, list)
+                and any(item in actual for item in expected)
+            )
+        case Operator.NOT_OVERLAPS:
+            return not (
+                isinstance(actual, list)
+                and isinstance(expected, list)
+                and any(item in actual for item in expected)
+            )
+        case Operator.GREATER_THAN:
+            try:
+                return cast(Any, actual) > expected
+            except TypeError:
+                return False
+        case Operator.GREATER_THAN_OR_EQUAL:
+            try:
+                return cast(Any, actual) >= expected
+            except TypeError:
+                return False
+        case Operator.LESS_THAN:
+            try:
+                return cast(Any, actual) < expected
+            except TypeError:
+                return False
+        case Operator.LESS_THAN_OR_EQUAL:
+            try:
+                return cast(Any, actual) <= expected
+            except TypeError:
+                return False
+        case Operator.IS_NULL:
+            return actual is None
+        case Operator.IS_NOT_NULL:
+            return actual is not None
+        case Operator.IS_EMPTY:
+            return is_empty
+        case Operator.IS_NOT_EMPTY:
+            return not is_empty
+        case Operator.NESTED:
+            return (
+                isinstance(expected, (AndStatement, OrStatement))
+                and isinstance(actual, Mapping)
+                and _uniqueql_matches(actual, expected)
+            )
+
+
 def _propagate_wait_interrupt(result: object) -> None:
     """Re-raise timeout/cancellation so callers can return a partial snapshot."""
     if isinstance(result, (asyncio.CancelledError, TimeoutError)):
@@ -139,7 +277,8 @@ async def _list_direct_children_async(
 
     ``scope_id=None`` is the knowledge-base root. ``Content.get_infos`` without
     ``parentId`` lists the **entire** catalog, so root only fetches folders.
-    Files are listed per folder with ``parentId``.
+    Files are listed per folder with ``parentId``. Unique rejects ``parentId``
+    with ``metadataFilter``, so UniqueQL is applied to the returned files.
     """
     parent_params: dict[str, Any] = {}
     if scope_id:
@@ -164,23 +303,38 @@ async def _list_direct_children_async(
     if not scope_id:
         return await folders_task
 
+    parsed_filter = (
+        parse_uniqueql(_uniqueql_fill_nullish_values(metadata_filter))
+        if metadata_filter
+        else None
+    )
+
     async def _content_page(skip: int) -> Any:
-        params = dict(parent_params)
-        if metadata_filter:
-            params["metadataFilter"] = metadata_filter
         return await unique_sdk.Content.get_infos_async(
             user_id=user_id,
             company_id=company_id,
             skip=skip,
             take=step_size,
-            **params,
+            **parent_params,
         )
 
+    def _parse_content_page(payload: Any) -> tuple[list[ContentInfo], int]:
+        files, total = _parse_content_infos_payload(payload)
+        # Keep parentId listing + local UniqueQL until UniqueQL/folderId metadata
+        # listing is equivalent (soft-delete/versioning); then drop this record filter.
+        if parsed_filter is not None:
+            files = [
+                info
+                for info in files
+                if _uniqueql_matches(_content_uniqueql_record(info), parsed_filter)
+            ]
+        return files, total
+
     folder_result, file_result = await asyncio.gather(
         folders_task,
         _paginate_parent_listing(
             _content_page,
-            _parse_content_infos_payload,
+            _parse_content_page,
             step_size=step_size,
             max_concurrent_requests=max_concurrent_page_fetches,
             on_page=on_files,

unique_toolkit/experimental/components/internal_search/base/service.py

@@ -75,12 +75,26 @@ class InternalSearchBaseService(  # pyright: ignore[reportImplicitAbstractClass]
             )
         )
 
-        results = await asyncio.gather(
-            *[self._search_single_query(query=q) for q in search_queries],
-            return_exceptions=True,
+        # TaskGroup awaits cancelled children so httpcore can release the connection.
+        results: list[SearchStringResult | BaseException | None] = [None] * len(
+            search_queries
         )
 
-        found = self._collect_results(results, search_queries)
+        async def _run(i: int, query: str) -> None:
+            try:
+                results[i] = await self._search_single_query(query=query)
+            except Exception as exc:
+                # not BaseException: CancelledError must reach the group
+                results[i] = exc
+
+        async with asyncio.TaskGroup() as tg:
+            for i, query in enumerate(search_queries):
+                tg.create_task(_run(i, query))
+
+        found = self._collect_results(
+            [r for r in results if r is not None],
+            search_queries,
+        )
         return await self._finalize_run(
             search_queries,
             found,

unique_toolkit/language_model/infos.py

@@ -110,6 +110,7 @@ class LanguageModelName(StrEnum):
     GROK_4_1_FAST_NON_REASONING = "litellm:grok-4-1-fast-non-reasoning"
     GROK_4_1_FAST_REASONING = "litellm:grok-4-1-fast-reasoning"
     GROK_4_5 = "litellm:grok-4-5"
+    GROK_4_6 = "litellm:grok-4-6"
     LITELLM_OPENAI_GPT_5 = "litellm:openai-gpt-5"
     LITELLM_OPENAI_GPT_5_MINI = "litellm:openai-gpt-5-mini"
     LITELLM_OPENAI_GPT_5_NANO = "litellm:openai-gpt-5-nano"
@@ -2195,6 +2196,33 @@ class LanguageModelInfo(BaseModel):
                     published_at=date(2026, 7, 16),
                     supported_reasoning_efforts=["low", "medium", "high"],
                 )
+            case LanguageModelName.GROK_4_6:
+                # Specs from https://docs.x.ai/developers/grok-4-6
+                # Announcement date from https://x.ai/news/grok-4-6
+                return cls(
+                    name=model_name,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.REASONING,
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                        ModelCapabilities.VISION,
+                    ],
+                    provider=LanguageModelProvider.LITELLM,
+                    family=ModelFamily.XAI,
+                    version="grok-4-6",
+                    encoder_name=EncoderName.O200K_BASE,  # TODO: Update encoder with grok tokenizer
+                    token_limits=LanguageModelTokenLimits(
+                        # Context window is 500_000, we leave 20_000 tokens as buffer due to tokenizer mismatch
+                        # Assign 90% for input and 10% for output
+                        token_limit_input=int(480_000 * 0.9),
+                        token_limit_output=int(480_000 * 0.1),
+                    ),
+                    info_cutoff_at=date(2026, 1, day=1),
+                    published_at=date(2026, 8, 12),
+                    supported_reasoning_efforts=["low", "medium", "high", "xhigh"],
+                    default_options={"reasoning_effort": "high"},
+                )
             case LanguageModelName.LITELLM_OPENAI_GPT_5:
                 return cls(
                     name=model_name,

{unique_toolkit-2026.38.0.dev2.dist-info → unique_toolkit-2026.38.0.dev4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.38.0.dev2
+Version: 2026.38.0.dev4
 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.38.0.dev2.dist-info → unique_toolkit-2026.38.0.dev4.dist-info}/RECORD

@@ -304,14 +304,14 @@ unique_toolkit/experimental/_internal/streaming/pattern_replacer.py,sha256=RLLQK
 unique_toolkit/experimental/components/__init__.py,sha256=rl-s8AnHZoLEOB8ei9jQgiRIndfMWrzWJm8gEMzNcS4,2029
 unique_toolkit/experimental/components/_base.py,sha256=XZHKffTIy331ZQnsywl0m1XkJ4DCjHV2on5EBkaG-ik,4153
 unique_toolkit/experimental/components/content_tree/__init__.py,sha256=zAMzctv__M5e4OppVD_neO471V4zKBeHcbVrixZciaM,1946
-unique_toolkit/experimental/components/content_tree/functions.py,sha256=CzRnEKST7JUtoaqALqT37J2tdpHmeLPtnsiw-Xu8Zys,14231
+unique_toolkit/experimental/components/content_tree/functions.py,sha256=SwWxgwjXEhZcUh7bZJOFpJ_CwldEzLnMzMi69EEZeAQ,19465
 unique_toolkit/experimental/components/content_tree/schemas.py,sha256=8MphCYk8VCnNk5v274voeKd9HLvsOHjpk7G8D4YPuWU,9719
 unique_toolkit/experimental/components/content_tree/service.py,sha256=OKkPz-prJAr_8VzK0PLYTDZutV8LqlOaXnev57YxID0,25415
 unique_toolkit/experimental/components/internal_search/__init__.py,sha256=Jj-ujG_SS5-uqCrrNidRn_H0sQAeSvyWtt9Tw0pkbzI,3091
 unique_toolkit/experimental/components/internal_search/base/__init__.py,sha256=GB7vNBvbF6V1oLjKEf2KVvukr-BwgYpAkD5jKtGG9wU,929
 unique_toolkit/experimental/components/internal_search/base/config.py,sha256=Bf77n0YSEFXRZKAqe4efAOITnyD3hH5Gp-gEVbMw1qE,2482
 unique_toolkit/experimental/components/internal_search/base/schemas.py,sha256=N9UkSNDuKidNpec8Wimfow2-EhIWqK5G7j7c_Mn28Jw,1431
-unique_toolkit/experimental/components/internal_search/base/service.py,sha256=XHXvkyy2dL0GwM2AFZdAte2upMmcKBVyA_fllEJYxjY,6934
+unique_toolkit/experimental/components/internal_search/base/service.py,sha256=uM6PFixs0yEkpnjVfQOwXowR-aAy1bV0dGsRGcC03Vs,7478
 unique_toolkit/experimental/components/internal_search/base/utils.py,sha256=sjePAtSs5-GXIbWh0sDwC8wXbX_XWiaTx21AZ7i7sto,6431
 unique_toolkit/experimental/components/internal_search/chat/__init__.py,sha256=mnlw9kKlpHWwiYhQeECXOd0-54muym2TZj0QhAsFSa0,246
 unique_toolkit/experimental/components/internal_search/chat/config.py,sha256=2DQhY5TIMIXAqpezORDeRf3AV5_2jV1LnTKBqzrSLbk,879
@@ -394,7 +394,7 @@ unique_toolkit/language_model/builder.py,sha256=KwBzsD6-fvQnB4p2I7M-5Tk6j4tmQ973
 unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
 unique_toolkit/language_model/default_language_model.py,sha256=mcyuFs_V6tp52C8xJSHiEHO4kFGj6xed1gvozvCRD7k,1181
 unique_toolkit/language_model/functions.py,sha256=W6JC4LUh-LZ9waDgph--xVmKeQ7Ls6NRAqDx9DwQdeY,26691
-unique_toolkit/language_model/infos.py,sha256=TY4HtNbOxwokq3YW4QNiAyA8eq2ZvzzaRhGAfDkKG3M,149525
+unique_toolkit/language_model/infos.py,sha256=NWVMUxcL4lOHmgNmEXZyj7qTq-mt_gfYYWGMweCRtlE,151046
 unique_toolkit/language_model/invocation_stats.py,sha256=CQbpUz394IsVd8eNTB1gPSMt5f0UXmkN1amNsqtAW6U,5857
 unique_toolkit/language_model/model_costs.py,sha256=SMYhikdN4h_qMUKKrpHYyYO5o4MW_QlZGGEr643Hw0M,4354
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
@@ -422,6 +422,6 @@ unique_toolkit/short_term_memory/service.py,sha256=2JMBEBn6GoyK-a0xvdcWpPft-3C74
 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.38.0.dev2.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.38.0.dev2.dist-info/METADATA,sha256=GxmLZU9cThG9AWrCXho8un_5tr73a85ZeZXJPu4-NXo,7821
-unique_toolkit-2026.38.0.dev2.dist-info/RECORD,,
+unique_toolkit-2026.38.0.dev4.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.38.0.dev4.dist-info/METADATA,sha256=bLPgw9um9D6gC7lAdRZksRhdY2iSUB7paSAIS8f2ccI,7821
+unique_toolkit-2026.38.0.dev4.dist-info/RECORD,,