unique_toolkit 1.11.1__tar.gz → 1.11.3__tar.gz

{unique_toolkit-1.11.1 → unique_toolkit-1.11.3}/CHANGELOG.md

@@ -4,6 +4,12 @@ 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.11.3] - 2026-10-06
+- Move smart rules to content
+- Add to documentation
+
+## [1.11.2] - 2025-10-07
+- Fix for empty metadata filter at info retrieval
 
 ## [1.11.1] - 2025-10-07
 - Fix bug where hallucination check was taking all of the chunks as input instead of only the referenced ones.
@@ -19,6 +25,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [1.9.1] - 2025-10-06
 - Switch default model used in evaluation service from `GPT-3.5-turbo (0125)` to `GPT-4o (1120)`
 
+
 ## [1.9.0] - 2026-10-04
 - Define the RequestContext and add aihttp/httpx requestors
 

{unique_toolkit-1.11.1 → unique_toolkit-1.11.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.11.1
+Version: 1.11.3
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -117,6 +117,12 @@ 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.11.3] - 2026-10-06
+- Move smart rules to content
+- Add to documentation
+
+## [1.11.2] - 2025-10-07
+- Fix for empty metadata filter at info retrieval
 
 ## [1.11.1] - 2025-10-07
 - Fix bug where hallucination check was taking all of the chunks as input instead of only the referenced ones.
@@ -132,6 +138,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [1.9.1] - 2025-10-06
 - Switch default model used in evaluation service from `GPT-3.5-turbo (0125)` to `GPT-4o (1120)`
 
+
 ## [1.9.0] - 2026-10-04
 - Define the RequestContext and add aihttp/httpx requestors
 

{unique_toolkit-1.11.1 → unique_toolkit-1.11.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "1.11.1"
+version = "1.11.3"
 description = ""
 authors = [
     "Cedric Klinkert <cedric.klinkert@unique.ch>",

{unique_toolkit-1.11.1 → unique_toolkit-1.11.3}/unique_toolkit/content/functions.py

@@ -16,7 +16,8 @@ from unique_toolkit.content.schemas import (
     ContentInfo,
     ContentRerankerConfig,
     ContentSearchType,
-    PaginatedContentInfo,
+    FolderInfo,
+    PaginatedContentInfos,
 )
 from unique_toolkit.content.utils import map_contents, map_to_content_chunks
 
@@ -590,21 +591,33 @@ def get_content_info(
     file_path: str | None = None,
 ):
     """Gets the info of a content."""
+
     get_info_params = unique_sdk.Content.ContentInfoParams(
-        metadataFilter=metadata_filter,
-        skip=skip,
-        take=take,
-        filePath=file_path,
+        metadataFilter=metadata_filter or None,  # Dict cannot be empty
     )
+    if skip:
+        get_info_params["skip"] = skip
+    if take:
+        get_info_params["take"] = take
+    if file_path:
+        get_info_params["filePath"] = file_path
 
-    content_info = unique_sdk.Content.get_info(
+    content_info = unique_sdk.Content.get_infos(
         user_id=user_id, company_id=company_id, **get_info_params
     )
-    return PaginatedContentInfo.model_validate(
+    return PaginatedContentInfos.model_validate(
         content_info, by_alias=True, by_name=True
     )
 
 
+def get_folder_info(user_id: str, company_id: str, scope_id: str) -> FolderInfo:
+    info = unique_sdk.Folder.get_info(
+        user_id=user_id, company_id=company_id, scopeId=scope_id
+    )
+
+    return FolderInfo.model_validate(info, by_alias=True, by_name=True)
+
+
 def update_content(
     user_id: str,
     company_id: str,

{unique_toolkit-1.11.1 → unique_toolkit-1.11.3}/unique_toolkit/content/schemas.py

@@ -154,12 +154,13 @@ class ContentRerankerConfig(BaseModel):
 class ContentInfo(BaseModel):
     model_config = model_config
     id: str
+    object: str
     key: str
     url: str | None = None
     title: str | None = None
     metadata: dict[str, Any] | None = None
-    mime_type: str
     byte_size: int
+    mime_type: str
     owner_id: str
     created_at: datetime
     updated_at: datetime
@@ -168,7 +169,19 @@ class ContentInfo(BaseModel):
     expired_at: datetime | None = None
 
 
-class PaginatedContentInfo(BaseModel):
+class PaginatedContentInfos(BaseModel):
     model_config = model_config
-    content_info: list[ContentInfo]
+    object: str
+    content_infos: list[ContentInfo]
     total_count: int
+
+
+class FolderInfo(BaseModel):
+    model_config = model_config
+    id: str
+    name: str
+    ingestion_config: dict[str, Any]
+    createdAt: str | None
+    updatedAt: str | None
+    parentId: str | None
+    externalId: str | None

unique_toolkit-1.11.1/unique_toolkit/smart_rules/compile.py → unique_toolkit-1.11.3/unique_toolkit/content/smart_rules.py

@@ -1,7 +1,7 @@
 import re
 from datetime import datetime, timedelta, timezone
 from enum import Enum
-from typing import Any, Dict, List, Self, Union
+from typing import Any, Dict, List, Mapping, Self, Union
 
 from pydantic import AliasChoices, BaseModel, Field
 from pydantic.config import ConfigDict
@@ -30,8 +30,8 @@ class BaseStatement(BaseModel):
 
     def with_variables(
         self,
-        user_metadata: Dict[str, Union[str, int, bool]],
-        tool_parameters: Dict[str, Union[str, int, bool]],
+        user_metadata: Mapping[str, Union[str, int, bool]],
+        tool_parameters: Mapping[str, Union[str, int, bool]],
     ) -> Self:
         return self._fill_in_variables(user_metadata, tool_parameters)
 
@@ -49,8 +49,8 @@ class BaseStatement(BaseModel):
 
     def _fill_in_variables(
         self,
-        user_metadata: Dict[str, Union[str, int, bool]],
-        tool_parameters: Dict[str, Union[str, int, bool]],
+        user_metadata: Mapping[str, Union[str, int, bool]],
+        tool_parameters: Mapping[str, Union[str, int, bool]],
     ) -> Self:
         return self.model_copy()
 
@@ -62,8 +62,8 @@ class Statement(BaseStatement):
 
     def _fill_in_variables(
         self,
-        user_metadata: Dict[str, Union[str, int, bool]],
-        tool_parameters: Dict[str, Union[str, int, bool]],
+        user_metadata: Mapping[str, Union[str, int, bool]],
+        tool_parameters: Mapping[str, Union[str, int, bool]],
     ) -> Self:
         new_stmt = self.model_copy()
         new_stmt.value = eval_operator(self, user_metadata, tool_parameters)
@@ -72,13 +72,13 @@ class Statement(BaseStatement):
 
 class AndStatement(BaseStatement):
     and_list: List[Union["Statement", "AndStatement", "OrStatement"]] = Field(
-        alias="and", validation_alias=AliasChoices("and", "and_list")
+        validation_alias=AliasChoices("and", "and_list"), serialization_alias="and"
     )
 
     def _fill_in_variables(
         self,
-        user_metadata: Dict[str, Union[str, int, bool]],
-        tool_parameters: Dict[str, Union[str, int, bool]],
+        user_metadata: Mapping[str, Union[str, int, bool]],
+        tool_parameters: Mapping[str, Union[str, int, bool]],
     ) -> Self:
         new_stmt = self.model_copy()
         new_stmt.and_list = [
@@ -90,13 +90,13 @@ class AndStatement(BaseStatement):
 
 class OrStatement(BaseStatement):
     or_list: List[Union["Statement", "AndStatement", "OrStatement"]] = Field(
-        alias="or", validation_alias=AliasChoices("or", "or_list")
+        validation_alias=AliasChoices("or", "or_list"), serialization_alias="or"
     )
 
     def _fill_in_variables(
         self,
-        user_metadata: Dict[str, Union[str, int, bool]],
-        tool_parameters: Dict[str, Union[str, int, bool]],
+        user_metadata: Mapping[str, Union[str, int, bool]],
+        tool_parameters: Mapping[str, Union[str, int, bool]],
     ) -> Self:
         new_stmt = self.model_copy()
         new_stmt.or_list = [
@@ -121,8 +121,8 @@ def is_array_of_strings(value: Any) -> bool:
 
 def eval_operator(
     query: Statement,
-    user_metadata: Dict[str, Union[str, int, bool]],
-    tool_parameters: Dict[str, Union[str, int, bool]],
+    user_metadata: Mapping[str, Union[str, int, bool]],
+    tool_parameters: Mapping[str, Union[str, int, bool]],
 ) -> Any:
     if query.operator in [
         Operator.EQUALS,
@@ -149,8 +149,8 @@ def eval_operator(
 
 def eval_nested_operator(
     value: Any,
-    user_metadata: Dict[str, Union[str, int, bool]],
-    tool_parameters: Dict[str, Union[str, int, bool]],
+    user_metadata: Mapping[str, Union[str, int, bool]],
+    tool_parameters: Mapping[str, Union[str, int, bool]],
 ) -> Union[AndStatement, OrStatement]:
     if not isinstance(value, (AndStatement, OrStatement)):
         raise ValueError("Nested operator must be an AndStatement or OrStatement")
@@ -159,16 +159,16 @@ def eval_nested_operator(
 
 def binary_operator(
     value: Any,
-    user_metadata: Dict[str, Union[str, int, bool]],
-    tool_parameters: Dict[str, Union[str, int, bool]],
+    user_metadata: Mapping[str, Union[str, int, bool]],
+    tool_parameters: Mapping[str, Union[str, int, bool]],
 ) -> Any:
     return replace_variables(value, user_metadata, tool_parameters)
 
 
 def array_operator(
     value: Any,
-    user_metadata: Dict[str, Union[str, int, bool]],
-    tool_parameters: Dict[str, Union[str, int, bool]],
+    user_metadata: Mapping[str, Union[str, int, bool]],
+    tool_parameters: Mapping[str, Union[str, int, bool]],
 ) -> Any:
     if is_array_of_strings(value):
         return [
@@ -179,16 +179,16 @@ def array_operator(
 
 def null_operator(
     value: Any,
-    user_metadata: Dict[str, Union[str, int, bool]],
-    tool_parameters: Dict[str, Union[str, int, bool]],
+    user_metadata: Mapping[str, Union[str, int, bool]],
+    tool_parameters: Mapping[str, Union[str, int, bool]],
 ) -> Any:
     return value  # do nothing for now. No variables to replace
 
 
 def empty_operator(
     operator: Operator,
-    user_metadata: Dict[str, Union[str, int, bool]],
-    tool_parameters: Dict[str, Union[str, int, bool]],
+    user_metadata: Mapping[str, Union[str, int, bool]],
+    tool_parameters: Mapping[str, Union[str, int, bool]],
 ) -> Any:
     """Handle IS_EMPTY and IS_NOT_EMPTY operators."""
     if operator == Operator.IS_EMPTY:
@@ -225,8 +225,8 @@ def calculate_later_date(input_str: str) -> str:
 
 def replace_variables(
     value: Any,
-    user_metadata: Dict[str, Union[str, int, bool]],
-    tool_parameters: Dict[str, Union[str, int, bool]],
+    user_metadata: Mapping[str, Union[str, int, bool]],
+    tool_parameters: Mapping[str, Union[str, int, bool]],
 ) -> Any:
     if isinstance(value, str):
         if "||" in value:
@@ -274,8 +274,8 @@ def replace_user_metadata_patterns(
 
 def get_fallback_values(
     value: str,
-    user_metadata: Dict[str, Union[str, int, bool]],
-    tool_parameters: Dict[str, Union[str, int, bool]],
+    user_metadata: Mapping[str, Union[str, int, bool]],
+    tool_parameters: Mapping[str, Union[str, int, bool]],
 ) -> Any:
     values = value.split("||")
     for val in values:

{unique_toolkit-1.11.1 → unique_toolkit-1.11.3}/unique_toolkit/knowledge_base.py

@@ -14,6 +14,7 @@ from unique_toolkit.content.functions import (
     download_content_to_bytes,
     download_content_to_file_by_id,
     get_content_info,
+    get_folder_info,
     search_content_chunks,
     search_content_chunks_async,
     search_contents,
@@ -28,7 +29,8 @@ from unique_toolkit.content.schemas import (
     ContentInfo,
     ContentRerankerConfig,
     ContentSearchType,
-    PaginatedContentInfo,
+    FolderInfo,
+    PaginatedContentInfos,
 )
 
 _LOGGER = logging.getLogger(f"toolkit.knowledge_base.{__name__}")
@@ -483,7 +485,7 @@ class KnowledgeBaseService:
         skip: int | None = None,
         take: int | None = None,
         file_path: str | None = None,
-    ) -> PaginatedContentInfo:
+    ) -> PaginatedContentInfos:
         return get_content_info(
             user_id=self._user_id,
             company_id=self._company_id,
@@ -493,6 +495,17 @@ class KnowledgeBaseService:
             file_path=file_path,
         )
 
+    def get_folder_info(
+        self,
+        *,
+        scope_id: str,
+    ) -> FolderInfo:
+        return get_folder_info(
+            user_id=self._user_id,
+            company_id=self._company_id,
+            scope_id=scope_id,
+        )
+
     def replace_content_metadata(
         self,
         *,
@@ -506,6 +519,72 @@ class KnowledgeBaseService:
             metadata=metadata,
         )
 
+    def _resolve_visible_file_tree(self, content_infos: list[ContentInfo]) -> list[str]:
+        # collect all scope ids
+        folder_id_paths: set[str] = set()
+        known_folder_paths: set[str] = set()
+        for content_info in content_infos:
+            if (
+                content_info.metadata
+                and content_info.metadata.get(r"{FullPath}") is not None
+            ):
+                known_folder_paths.add(str(content_info.metadata.get(r"{FullPath}")))
+                continue
+
+            if (
+                content_info.metadata
+                and content_info.metadata.get("folderIdPath") is not None
+            ):
+                folder_id_paths.add(str(content_info.metadata.get("folderIdPath")))
+
+        scope_ids: set[str] = set()
+        for fp in folder_id_paths:
+            scope_ids_list = set(fp.replace("uniquepathid://", "").split("/"))
+            scope_ids.update(scope_ids_list)
+
+        scope_id_to_folder_name: dict[str, str] = {}
+        for scope_id in scope_ids:
+            folder_info = self.get_folder_info(
+                scope_id=scope_id,
+            )
+            scope_id_to_folder_name[scope_id] = folder_info.name
+
+        folder_paths: set[str] = set()
+        for folder_id_path in folder_id_paths:
+            scope_ids_list = folder_id_path.replace("uniquepathid://", "").split("/")
+
+            if all(scope_id in scope_id_to_folder_name for scope_id in scope_ids_list):
+                folder_path = [
+                    scope_id_to_folder_name[scope_id] for scope_id in scope_ids_list
+                ]
+                folder_paths.add("/".join(folder_path))
+
+        return [
+            p if p.startswith("/") else f"/{p}"
+            for p in folder_paths.union(known_folder_paths)
+        ]
+
+    def resolve_visible_file_tree(
+        self, *, metadata_filter: dict[str, Any] | None = None
+    ) -> list[str]:
+        """
+        Resolves the visible file tree for the knowledge base for the current user.
+
+        Args:
+            metadata_filter (dict[str, Any] | None): The metadata filter to use. Defaults to None.
+
+        Returns:
+            list[str]: The visible file tree.
+
+
+
+        """
+        info = self.get_paginated_content_infos(
+            metadata_filter=metadata_filter,
+        )
+
+        return self._resolve_visible_file_tree(content_infos=info.content_infos)
+
 
 if __name__ == "__main__":
     kb_service = KnowledgeBaseService.from_settings()

unique_toolkit-1.11.3/unique_toolkit/smart_rules/compile.py

@@ -0,0 +1,56 @@
+import warnings
+
+from unique_toolkit.content.smart_rules import (
+    AndStatement,
+    BaseStatement,
+    Operator,
+    OrStatement,
+    Statement,
+    UniqueQL,
+    array_operator,
+    binary_operator,
+    calculate_current_date,
+    calculate_earlier_date,
+    calculate_later_date,
+    empty_operator,
+    eval_nested_operator,
+    eval_operator,
+    get_fallback_values,
+    is_array_of_strings,
+    null_operator,
+    parse_uniqueql,
+    replace_tool_parameters_patterns,
+    replace_user_metadata_patterns,
+    replace_variables,
+)
+
+warnings.warn(
+    "unique_toolkit.smart_rules.compile is deprecated. "
+    "Please use unique_toolkit.content.smart_rules instead.",
+    DeprecationWarning,
+    stacklevel=2,
+)
+
+__all__ = [
+    "AndStatement",
+    "BaseStatement",
+    "Operator",
+    "OrStatement",
+    "Statement",
+    "UniqueQL",
+    "array_operator",
+    "binary_operator",
+    "calculate_current_date",
+    "calculate_earlier_date",
+    "calculate_later_date",
+    "empty_operator",
+    "eval_nested_operator",
+    "eval_operator",
+    "get_fallback_values",
+    "is_array_of_strings",
+    "null_operator",
+    "parse_uniqueql",
+    "replace_tool_parameters_patterns",
+    "replace_user_metadata_patterns",
+    "replace_variables",
+]