unique_toolkit 0.5.4__py3-none-any.whl → 0.5.6__py3-none-any.whl

unique_toolkit/content/schemas.py

@@ -88,3 +88,9 @@ class ContentUploadInput(BaseModel):
     owner_type: Optional[str] = None
     owner_id: Optional[str] = None
     byte_size: Optional[int] = None
+
+
+class RerankerConfig(BaseModel):
+    model_config = model_config
+    deployment_name: str
+    options: dict | None = None

unique_toolkit/content/service.py

@@ -14,10 +14,13 @@ from unique_toolkit.content.schemas import (
     ContentChunk,
     ContentSearchType,
     ContentUploadInput,
+    RerankerConfig,
 )
 
 
 class ContentService:
+    DEFAULT_SEARCH_LANGUAGE = "english"
+
     def __init__(self, state: ChatState, logger: Optional[logging.Logger] = None):
         self.state = state
         self.logger = logger or logging.getLogger(__name__)
@@ -27,6 +30,8 @@ class ContentService:
         search_string: str,
         search_type: ContentSearchType,
         limit: int,
+        reranker_config: Optional[RerankerConfig] = None,
+        search_language: str = DEFAULT_SEARCH_LANGUAGE,
         scope_ids: Optional[list[str]] = None,
     ) -> list[ContentChunk]:
         """
@@ -36,16 +41,20 @@ 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.
+            reranker_config (Optional[RerankerConfig]): The reranker configuration. Defaults to None.
+            search_language (str): The language for the full-text search. Defaults to "english".
             scope_ids (Optional[list[str]]): The scope IDs. Defaults to None.
 
         Returns:
             list[ContentChunk]: The search results.
         """
         return self._trigger_search_content_chunks(
-            search_string,
-            search_type,
-            limit,
-            scope_ids,
+            search_string=search_string,
+            search_type=search_type,
+            limit=limit,
+            reranker_config=reranker_config,
+            search_language=search_language,
+            scope_ids=scope_ids,
         )
 
     @to_async
@@ -55,7 +64,9 @@ class ContentService:
         search_string: str,
         search_type: ContentSearchType,
         limit: int,
-        scope_ids: Optional[list[str]],
+        reranker_config: Optional[RerankerConfig] = None,
+        search_language: str = DEFAULT_SEARCH_LANGUAGE,
+        scope_ids: Optional[list[str]] = None,
     ):
         """
         Performs an asynchronous search for content chunks in the knowledge base.
@@ -64,16 +75,20 @@ 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.
+            reranker_config (Optional[RerankerConfig]): The reranker configuration. Defaults to None.
+            search_language (str): The language for the full-text search. Defaults to "english".
             scope_ids (Optional[list[str]]): The scope IDs. Defaults to [].
 
         Returns:
             list[ContentChunk]: The search results.
         """
         return self._trigger_search_content_chunks(
-            search_string,
-            search_type,
-            limit,
-            scope_ids,
+            search_string=search_string,
+            search_type=search_type,
+            limit=limit,
+            reranker_config=reranker_config,
+            search_language=search_language,
+            scope_ids=scope_ids,
         )
 
     def _trigger_search_content_chunks(
@@ -81,10 +96,10 @@ class ContentService:
         search_string: str,
         search_type: ContentSearchType,
         limit: int,
-        scope_ids: Optional[list[str]],
+        reranker_config: Optional[RerankerConfig] = None,
+        search_language: str = DEFAULT_SEARCH_LANGUAGE,
+        scope_ids: Optional[list[str]] = None,
     ) -> list[ContentChunk]:
-        scope_ids = scope_ids or self.state.scope_ids or []
-
         if not scope_ids:
             self.logger.warning("No scope IDs provided for search.")
 
@@ -97,6 +112,8 @@ class ContentService:
                 searchType=search_type.name,
                 scopeIds=scope_ids,
                 limit=limit,
+                reranker=reranker_config.model_dump() if reranker_config else None,
+                language=search_language,
                 chatOnly=self.state.chat_only,
             )
         except Exception as e:

unique_toolkit/language_model/service.py

@@ -92,7 +92,7 @@ class LanguageModelService:
         tools: Optional[list[LanguageModelTool]] = None,
     ) -> LanguageModelResponse:
         options = self._add_tools_to_options({}, tools)
-        messages = messages.model_dump(exclude_none=True)
+        messages = messages.model_dump(exclude_none=True, exclude={"tool_calls"})
         try:
             response = unique_sdk.ChatCompletion.create(
                 company_id=self.state.company_id,
@@ -217,7 +217,7 @@ class LanguageModelService:
             for chunk in content_chunks
         ]
 
-        messages = messages.model_dump(exclude_none=True)
+        messages = messages.model_dump(exclude_none=True, exclude={"tool_calls"})
 
         try:
             response = unique_sdk.Integrated.chat_stream_completion(

{unique_toolkit-0.5.4.dist-info → unique_toolkit-0.5.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.4
+Version: 0.5.6
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -18,7 +18,7 @@ Requires-Dist: regex (>=2024.5.15,<2025.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
 Requires-Dist: tiktoken (>=0.7.0,<0.8.0)
 Requires-Dist: typing-extensions (>=4.9.0,<5.0.0)
-Requires-Dist: unique-sdk (>=0.8.10,<0.9.0)
+Requires-Dist: unique-sdk (>=0.9.2,<0.10.0)
 Description-Content-Type: text/markdown
 
 # Unique Toolkit
@@ -102,6 +102,14 @@ 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.5.6] - 2024-07-30
+- Bug fix: `ContentService.search_content_chunks` and it's `async` equivalent now accept `None` as a valid parameter value for `scope_ids`.
+
+## [0.5.5] - 2024-07-30
+- Added parameters to `ContentService.search_content_chunks` and `ContentService.async_search_content_chunks`
+  - `reranker_config` to optinally rerank the search results
+  - `search_language` to specify a language for full-text-search
+
 ## [0.5.4] - 2024-07-26
 - correct ChatMessage schema
 

{unique_toolkit-0.5.4.dist-info → unique_toolkit-0.5.6.dist-info}/RECORD

@@ -9,16 +9,16 @@ unique_toolkit/chat/schemas.py,sha256=TYqrDy96ynP7XpnT_abOJlq6LkqBTdKwEft4YHeEgR
 unique_toolkit/chat/service.py,sha256=zxMoOMKlLJNwtRAO8GCpX9tGwVUsbgHWEPWztGe6htw,12300
 unique_toolkit/chat/state.py,sha256=IljPSqbzQq9TMD9cnbkII8k4eTtkyYOGzRe-DuR-Nic,1831
 unique_toolkit/chat/utils.py,sha256=ihm-wQykBWhB4liR3LnwPVPt_qGW6ETq21Mw4HY0THE,854
-unique_toolkit/content/schemas.py,sha256=u3b0R6bIEjbI4zGMoBRi-bdZy_SeRKtMRi8PKW6yMZ4,1969
-unique_toolkit/content/service.py,sha256=CLvHWu3SULOFLKbCl2LKBHgL7m11UWiRPsTPvFkhJXY,11328
+unique_toolkit/content/schemas.py,sha256=b6QiqZUBdYA14_CfJYeXEGSrPmT98n_Q8knugnP20yA,2093
+unique_toolkit/content/service.py,sha256=PQUog5lYYSnKOpSWMVfB1ITTTar09mY4Z1J1EBRjd3k,12472
 unique_toolkit/content/utils.py,sha256=x3ABo8ZCRm3YJAQwDtrr82z77DmW4Mei7KCIITjP0fk,6897
 unique_toolkit/embedding/schemas.py,sha256=1GvKCaSk4jixzVQ2PKq8yDqwGEVY_hWclYtoAr6CC2g,96
 unique_toolkit/embedding/service.py,sha256=IjBag3koSSeoQTo9836t0K5tkTbu0otvytVWnYQJHw4,2403
 unique_toolkit/language_model/infos.py,sha256=NhAkeW7PyusSIHCMvwRikLlzGG4tOXSLf_Fnq7V9rNE,8881
 unique_toolkit/language_model/schemas.py,sha256=kTGSGT3ygrH3guQELOWpxN4MTgEPuudi-CTvRu-zCcI,4377
-unique_toolkit/language_model/service.py,sha256=VHixKmi6-BP-StDQdomS-J-EWKIhHvDjNCi5zrP3mpM,9749
+unique_toolkit/language_model/service.py,sha256=OX3VCvco7K2me-2q0LgzHAdpCfPd4mHlwb1F3aZgYSk,9797
 unique_toolkit/language_model/utils.py,sha256=WBPj1XKkDgxy_-T8HCZvsfkkSzj_1w4UZzNmyvdbBLY,1081
-unique_toolkit-0.5.4.dist-info/LICENSE,sha256=bIeCWCYuoUU_MzNdg48-ubJSVm7qxakaRbzTiJ5uxrs,1065
-unique_toolkit-0.5.4.dist-info/METADATA,sha256=qB0TDQGmJpsKzyWCtjI6qxFERg1YKmc_B_ErQvLMsCY,8289
-unique_toolkit-0.5.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.5.4.dist-info/RECORD,,
+unique_toolkit-0.5.6.dist-info/LICENSE,sha256=bIeCWCYuoUU_MzNdg48-ubJSVm7qxakaRbzTiJ5uxrs,1065
+unique_toolkit-0.5.6.dist-info/METADATA,sha256=jwWhgPrtTNiTsC_Xk8KaTGYiVc1tZ9uvkkC7ePfL_XI,8715
+unique_toolkit-0.5.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.5.6.dist-info/RECORD,,