unique_toolkit 0.7.0__tar.gz → 0.7.1__tar.gz

{unique_toolkit-0.7.0 → unique_toolkit-0.7.1}/CHANGELOG.md

@@ -5,9 +5,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).
 
+## [0.7.1] - 2025-03-11
+- Fix Breaking change: `ContentService.search_content_chunks` `ContentService.search_content_chunks` now accepts`chat_id` for the specific to handle chat_only instances
+
 ## [0.7.0] - 2025-03-11
 - Fix the issue with `ShortTermMemoryService.create_memory_async` adding `self.chat_id` and `self.message_id` as part of the parameter.
-- Breaking chang: `ContentService.search_content_on_chat` now requires you pass in a `chat_id` for the specific chat instance
+- Breaking change: `ContentService.search_content_on_chat` now requires you pass in a `chat_id` for the specific chat instance
 
 ## [0.6.9] - 2025-03-11
 - Add o1-preview as part of the language model info, make the name consistent across board.

{unique_toolkit-0.7.0 → unique_toolkit-0.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.7.0
+Version: 0.7.1
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -111,9 +111,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).
 
+## [0.7.1] - 2025-03-11
+- Fix Breaking change: `ContentService.search_content_chunks` `ContentService.search_content_chunks` now accepts`chat_id` for the specific to handle chat_only instances
+
 ## [0.7.0] - 2025-03-11
 - Fix the issue with `ShortTermMemoryService.create_memory_async` adding `self.chat_id` and `self.message_id` as part of the parameter.
-- Breaking chang: `ContentService.search_content_on_chat` now requires you pass in a `chat_id` for the specific chat instance
+- Breaking change: `ContentService.search_content_on_chat` now requires you pass in a `chat_id` for the specific chat instance
 
 ## [0.6.9] - 2025-03-11
 - Add o1-preview as part of the language model info, make the name consistent across board.

{unique_toolkit-0.7.0 → unique_toolkit-0.7.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.7.0"
+version = "0.7.1"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",

{unique_toolkit-0.7.0 → unique_toolkit-0.7.1}/unique_toolkit/content/service.py

@@ -35,9 +35,10 @@ class ContentService:
     Provides methods for searching, downloading and uploading content in the knowledge base.
 
     Attributes:
-        company_id (str | None): The company ID.
-        user_id (str | None): The user ID.
-        metadata_filter (dict | None): The metadata filter.
+        event: BaseEvent | Event, this can be None ONLY if company_id and user_id are provided.
+        company_id (str): The company ID.
+        user_id (str): The user ID.
+        metadata_filter (dict | None): is only initialised from an Event(Deprecated) or ChatEvent.
     """
 
     def __init__(
@@ -77,6 +78,7 @@ class ContentService:
         search_type: ContentSearchType,
         limit: int,
         search_language: str = DEFAULT_SEARCH_LANGUAGE,
+        chat_id: str = "",
         reranker_config: ContentRerankerConfig | None = None,
         scope_ids: list[str] | None = None,
         chat_only: bool | None = None,
@@ -90,24 +92,32 @@ class ContentService:
             search_string (str): The search string.
             search_type (ContentSearchType): The type of search to perform.
             limit (int): The maximum number of results to return.
-            search_language (str): The language for the full-text search. Defaults to "english".
-            reranker_config (ContentRerankerConfig | None): The reranker configuration. Defaults to None.
-            scope_ids (list[str] | None): The scope IDs. Defaults to None.
-            chat_only (bool | None): Whether to search only in the current chat. Defaults to None.
-            metadata_filter (dict | None): UniqueQL metadata filter. If unspecified/None, it tries to use the metadata filter from the event. Defaults to None.
-            content_ids (list[str] | None): The content IDs to search. Defaults to None.
+            search_language (str, optional): The language for the full-text search. Defaults to "english".
+            chat_id (str, optional): The chat ID for context. Defaults to empty string.
+            reranker_config (ContentRerankerConfig | None, optional): The reranker configuration. Defaults to None.
+            scope_ids (list[str] | None, optional): The scope IDs to filter by. Defaults to None.
+            chat_only (bool | None, optional): Whether to search only in the current chat. Defaults to None.
+            metadata_filter (dict | None, optional): UniqueQL metadata filter. If unspecified/None, it tries to use the metadata filter from the event. Defaults to None.
+            content_ids (list[str] | None, optional): The content IDs to search within. Defaults to None.
+
         Returns:
             list[ContentChunk]: The search results.
+
+        Raises:
+            Exception: If there's an error during the search operation.
         """
 
         if metadata_filter is None:
             metadata_filter = self.metadata_filter
 
+        if chat_only and not chat_id:
+            raise ValueError("Please provide chat_id when limiting with chat_only")
+
         try:
             searches = search_content_chunks(
                 user_id=self.user_id,
                 company_id=self.company_id,
-                chat_id="",
+                chat_id=chat_id,
                 search_string=search_string,
                 search_type=search_type,
                 limit=limit,
@@ -129,6 +139,7 @@ class ContentService:
         search_type: ContentSearchType,
         limit: int,
         search_language: str = DEFAULT_SEARCH_LANGUAGE,
+        chat_id: str = "",
         reranker_config: ContentRerankerConfig | None = None,
         scope_ids: list[str] | None = None,
         chat_only: bool | None = None,
@@ -142,23 +153,31 @@ class ContentService:
             search_string (str): The search string.
             search_type (ContentSearchType): The type of search to perform.
             limit (int): The maximum number of results to return.
-            search_language (str): The language for the full-text search. Defaults to "english".
-            reranker_config (ContentRerankerConfig | None): The reranker configuration. Defaults to None.
-            scope_ids (list[str] | None): The scope IDs. Defaults to None.
-            chat_only (bool | None): Whether to search only in the current chat. Defaults to None.
-            metadata_filter (dict | None): UniqueQL metadata filter. If unspecified/None, it tries to use the metadata filter from the event. Defaults to None.
-            content_ids (list[str] | None): The content IDs to search. Defaults to None.
+            search_language (str, optional): The language for the full-text search. Defaults to "english".
+            chat_id (str, optional): The chat ID for context. Defaults to empty string.
+            reranker_config (ContentRerankerConfig | None, optional): The reranker configuration. Defaults to None.
+            scope_ids (list[str] | None, optional): The scope IDs to filter by. Defaults to None.
+            chat_only (bool | None, optional): Whether to search only in the current chat. Defaults to None.
+            metadata_filter (dict | None, optional): UniqueQL metadata filter. If unspecified/None, it tries to use the metadata filter from the event. Defaults to None.
+            content_ids (list[str] | None, optional): The content IDs to search within. Defaults to None.
+
         Returns:
             list[ContentChunk]: The search results.
+
+        Raises:
+            Exception: If there's an error during the search operation.
         """
         if metadata_filter is None:
             metadata_filter = self.metadata_filter
 
+        if chat_only and not chat_id:
+            raise ValueError("Please provide chat_id when limiting with chat_only.")
+
         try:
             searches = await search_content_chunks_async(
                 user_id=self.user_id,
                 company_id=self.company_id,
-                chat_id="",
+                chat_id=chat_id,
                 search_string=search_string,
                 search_type=search_type,
                 limit=limit,