unique_toolkit 1.11.1__py3-none-any.whl → 1.11.2__py3-none-any.whl

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/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/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.1.dist-info → unique_toolkit-1.11.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.11.1
+Version: 1.11.2
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,6 +118,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.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.
 

{unique_toolkit-1.11.1.dist-info → unique_toolkit-1.11.2.dist-info}/RECORD

@@ -112,8 +112,8 @@ unique_toolkit/chat/state.py,sha256=Cjgwv_2vhDFbV69xxsn7SefhaoIAEqLx3ferdVFCnOg,
 unique_toolkit/chat/utils.py,sha256=ihm-wQykBWhB4liR3LnwPVPt_qGW6ETq21Mw4HY0THE,854
 unique_toolkit/content/__init__.py,sha256=EdJg_A_7loEtCQf4cah3QARQreJx6pdz89Rm96YbMVg,940
 unique_toolkit/content/constants.py,sha256=1iy4Y67xobl5VTnJB6SxSyuoBWbdLl9244xfVMUZi5o,60
-unique_toolkit/content/functions.py,sha256=culiG7WLAaxo1FnSOdpavwWLybk9Z0vHVZYlMrYLGXI,20678
-unique_toolkit/content/schemas.py,sha256=mhPMPD5VYxFJ2F2AVjaUXXJSkcDgIDj26mMt8p7L5FU,5315
+unique_toolkit/content/functions.py,sha256=6Z9fOEsngJaT1R5DgoeRl5z06RavYgCxiAAPjkw-NKI,21100
+unique_toolkit/content/schemas.py,sha256=YmZa6ddjf6_RvDFGPovamLd0YX99am9nzWYrKHo8-Hg,5579
 unique_toolkit/content/service.py,sha256=i7wN_wYjF8NBZHBcpcA5XRlMRGntuw3mlVoudAoGQRk,23293
 unique_toolkit/content/utils.py,sha256=qNVmHTuETaPNGqheg7TbgPr1_1jbNHDc09N5RrmUIyo,7901
 unique_toolkit/embedding/__init__.py,sha256=uUyzjonPvuDCYsvXCIt7ErQXopLggpzX-MEQd3_e2kE,250
@@ -129,7 +129,7 @@ unique_toolkit/framework_utilities/openai/__init__.py,sha256=CrHYoC7lv2pBscitLer
 unique_toolkit/framework_utilities/openai/client.py,sha256=ct1cqPcIK1wPl11G9sJV39ZnLJwKr2kDUDSra0FjvtM,2007
 unique_toolkit/framework_utilities/openai/message_builder.py,sha256=VU6mJm_upLcarJQKFft_t1RlLRncWDxDuLC5LIJ5lQQ,4339
 unique_toolkit/framework_utilities/utils.py,sha256=JK7g2yMfEx3eMprug26769xqNpS5WJcizf8n2zWMBng,789
-unique_toolkit/knowledge_base.py,sha256=cx_Us1-a8aXMHCd9bWuW8bkDiQ52Qvty_97SKGdMDfc,17519
+unique_toolkit/knowledge_base.py,sha256=SHAFs68zDQuHJZdrFcdU7wnkUdfNQlNzpSLNckx3Scg,20167
 unique_toolkit/language_model/__init__.py,sha256=lRQyLlbwHbNFf4-0foBU13UGb09lwEeodbVsfsSgaCk,1971
 unique_toolkit/language_model/builder.py,sha256=4OKfwJfj3TrgO1ezc_ewIue6W7BCQ2ZYQXUckWVPPTA,3369
 unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
@@ -148,7 +148,7 @@ unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJ
 unique_toolkit/short_term_memory/service.py,sha256=5PeVBu1ZCAfyDb2HLVvlmqSbyzBBuE9sI2o9Aajqjxg,8884
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=cxWjb2dxEI2HGsakKdVCkSNi7VK9mr08w5sDcFCQyWI,9553
-unique_toolkit-1.11.1.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.11.1.dist-info/METADATA,sha256=PA3FkNO1TYXrN7GCWX1PWEyRZXp6hK3zXEiTCQZV9j0,35594
-unique_toolkit-1.11.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.11.1.dist-info/RECORD,,
+unique_toolkit-1.11.2.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.11.2.dist-info/METADATA,sha256=IdERxyqJO7c7OzqAkm1CrB5sXxC2Wu96_VAXakkjvgo,35670
+unique_toolkit-1.11.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.11.2.dist-info/RECORD,,