unique_toolkit 0.6.8__py3-none-any.whl → 0.7.0__py3-none-any.whl

unique_toolkit/content/functions.py

@@ -130,7 +130,6 @@ async def search_content_chunks_async(
 def search_contents(
     user_id: str,
     company_id: str,
-    chat_id: str,
     where: dict,
 ):
     """
@@ -152,7 +151,6 @@ def search_contents(
         contents = unique_sdk.Content.search(
             user_id=user_id,
             company_id=company_id,
-            chatId=chat_id,
             # TODO add type parameter in SDK
             where=where,  # type: ignore
         )
@@ -165,7 +163,6 @@ def search_contents(
 async def search_contents_async(
     user_id: str,
     company_id: str,
-    chat_id: str,
     where: dict,
 ):
     """Asynchronously searches for content in the knowledge base."""
@@ -176,7 +173,6 @@ async def search_contents_async(
         contents = await unique_sdk.Content.search_async(
             user_id=user_id,
             company_id=company_id,
-            chatId=chat_id,
             where=where,  # type: ignore
         )
         return map_contents(contents)

unique_toolkit/content/service.py

@@ -38,7 +38,6 @@ class ContentService:
         company_id (str | None): The company ID.
         user_id (str | None): The user ID.
         metadata_filter (dict | None): The metadata filter.
-        chat_id (str | None): The chat ID.
     """
 
     def __init__(
@@ -48,17 +47,16 @@ class ContentService:
         user_id: str | None = None,
     ):
         self._event = event  # Changed to protected attribute
+        self.metadata_filter = None
         if event:
             self.company_id = event.company_id
             self.user_id = event.user_id
             if isinstance(event, (ChatEvent, Event)):
                 self.metadata_filter = event.payload.metadata_filter
-                self.chat_id = event.payload.chat_id
         else:
             [company_id, user_id] = validate_required_values([company_id, user_id])
             self.company_id = company_id
             self.user_id = user_id
-            self.metadata_filter = None
 
     @property
     @deprecated(
@@ -109,7 +107,7 @@ class ContentService:
             searches = search_content_chunks(
                 user_id=self.user_id,
                 company_id=self.company_id,
-                chat_id=self.chat_id,
+                chat_id="",
                 search_string=search_string,
                 search_type=search_type,
                 limit=limit,
@@ -160,7 +158,7 @@ class ContentService:
             searches = await search_content_chunks_async(
                 user_id=self.user_id,
                 company_id=self.company_id,
-                chat_id=self.chat_id,
+                chat_id="",
                 search_string=search_string,
                 search_type=search_type,
                 limit=limit,
@@ -193,7 +191,6 @@ class ContentService:
         return search_contents(
             user_id=self.user_id,
             company_id=self.company_id,
-            chat_id=self.chat_id,
             where=where,
         )
 
@@ -213,14 +210,11 @@ class ContentService:
         return await search_contents_async(
             user_id=self.user_id,
             company_id=self.company_id,
-            chat_id=self.chat_id,
             where=where,
         )
 
-    def search_content_on_chat(
-        self,
-    ) -> list[Content]:
-        where = {"ownerId": {"equals": self.chat_id}}
+    def search_content_on_chat(self, chat_id: str) -> list[Content]:
+        where = {"ownerId": {"equals": chat_id}}
 
         return self.search_contents(where)
 

unique_toolkit/language_model/infos.py

@@ -16,9 +16,10 @@ class LanguageModelName(StrEnum):
     AZURE_GPT_4o_2024_0513 = "AZURE_GPT_4o_2024_0513"
     AZURE_GPT_4o_2024_0806 = "AZURE_GPT_4o_2024_0806"
     AZURE_GPT_4o_MINI_2024_0718 = "AZURE_GPT_4o_MINI_2024_0718"
-    AZURE_GPT_o1_2024_1217 = "AZURE_GPT_o1_2024_1217"
-    AZURE_GPT_o1_MINI_2024_0912 = "AZURE_GPT_o1_MINI_2024_0912"
-    AZURE_GPT_o3_MINI_2025_0131 = "AZURE_GPT_o3_MINI_2025_0131"
+    AZURE_o1_PREVIEW_2024_0912 = "AZURE_o1_PREVIEW_2024_0912"
+    AZURE_o1_2024_1217 = "AZURE_o1_2024_1217"
+    AZURE_o1_MINI_2024_0912 = "AZURE_o1_MINI_2024_0912"
+    AZURE_o3_MINI_2025_0131 = "AZURE_o3_MINI_2025_0131"
 
 
 class EncoderName(StrEnum):
@@ -215,7 +216,7 @@ class LanguageModelInfo(BaseModel):
                     info_cutoff_at=date(2023, 10, 1),
                     published_at=date(2024, 7, 18),
                 )
-            case LanguageModelName.AZURE_GPT_o1_2024_1217:
+            case LanguageModelName.AZURE_o1_2024_1217:
                 return cls(
                     name=model_name,
                     capabilities=[
@@ -234,7 +235,10 @@ class LanguageModelInfo(BaseModel):
                     info_cutoff_at=date(2023, 10, 1),
                     published_at=date(2024, 12, 17),
                 )
-            case LanguageModelName.AZURE_GPT_o1_MINI_2024_0912:
+            case (
+                LanguageModelName.AZURE_o1_MINI_2024_0912
+                | LanguageModelName.AZURE_o1_PREVIEW_2024_0912
+            ):
                 return cls(
                     name=model_name,
                     capabilities=[
@@ -253,7 +257,7 @@ class LanguageModelInfo(BaseModel):
                     info_cutoff_at=date(2023, 10, 1),
                     published_at=date(2024, 9, 12),
                 )
-            case LanguageModelName.AZURE_GPT_o3_MINI_2025_0131:
+            case LanguageModelName.AZURE_o3_MINI_2025_0131:
                 return cls(
                     name=model_name,
                     capabilities=[

unique_toolkit/short_term_memory/service.py

@@ -136,6 +136,8 @@ class ShortTermMemoryService:
             company_id=self.company_id,
             key=key,
             value=value,
+            chat_id=self.chat_id,
+            message_id=self.message_id,
         )
 
     def create_memory(self, key: str, value: str | dict):

{unique_toolkit-0.6.8.dist-info → unique_toolkit-0.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.6.8
+Version: 0.7.0
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -111,6 +111,13 @@ 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.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
+
+## [0.6.9] - 2025-03-11
+- Add o1-preview as part of the language model info, make the name consistent across board.
+
 ## [0.6.8] - 2025-03-11
 - Add `verify_request_and_construct_event` to `verification.py`
 

{unique_toolkit-0.6.8.dist-info → unique_toolkit-0.7.0.dist-info}/RECORD

@@ -20,9 +20,9 @@ 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=yB87wrbtmHzr3jGJUHetmuhy-7RVtnqG2IQ6gqFAun8,17093
+unique_toolkit/content/functions.py,sha256=iuclQjuiupIJPk0-6ldl35EVkQ2h34_dxKxWisMrxWM,17001
 unique_toolkit/content/schemas.py,sha256=zks_Pkki2VhxICJJgHZyc-LPmRuj5dLbw3pgcUT7SW8,2362
-unique_toolkit/content/service.py,sha256=CyQNW3TJX4trMnQhm5Ec2XGE9GHIW_NSGGT7YuOZIHQ,14234
+unique_toolkit/content/service.py,sha256=xcFN-vUakuovuzMXMN8qOmdlUzt2f1ZwHE4RzOWYjnQ,14040
 unique_toolkit/content/utils.py,sha256=GUVPrkZfMoAj4MRoBs5BD_7vSuLZTZx69hyWzYFrI50,7747
 unique_toolkit/embedding/__init__.py,sha256=uUyzjonPvuDCYsvXCIt7ErQXopLggpzX-MEQd3_e2kE,250
 unique_toolkit/embedding/constants.py,sha256=Lj8-Lcy1FvuC31PM9Exq7vaFuxQV4pEI1huUMFX-J2M,52
@@ -48,7 +48,7 @@ unique_toolkit/language_model/__init__.py,sha256=jWko_vQj48wjnpTtlkg8iNdef0SMI3F
 unique_toolkit/language_model/builder.py,sha256=_lUL2shMN1Je2LLYu2PiehOvfWtn5pg1yUuYyiw3f_c,2710
 unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
 unique_toolkit/language_model/functions.py,sha256=I5jHhHsKoq7GwEQyTrM8LXB2n_6dvMAk7UklenjuHSY,7945
-unique_toolkit/language_model/infos.py,sha256=DRkF0HzVemtSsSvdtrdsajNKaQ46Xla0ZXzFhi7xMtc,16338
+unique_toolkit/language_model/infos.py,sha256=HRdQLG3oetwc-bPHtsXvDRRLsK7fSsC1q8FubvxLMsw,16459
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
 unique_toolkit/language_model/schemas.py,sha256=rrwzUgKANFOrdehCULW8Hh03uRW3tsE5dXpWqxmClfg,8618
 unique_toolkit/language_model/service.py,sha256=GupYD4uDZjy1TfVQW3jichmgQwiSgQCj350FtL4O0W4,5569
@@ -57,8 +57,8 @@ unique_toolkit/short_term_memory/__init__.py,sha256=2mI3AUrffgH7Yt-xS57EGqnHf7jn
 unique_toolkit/short_term_memory/constants.py,sha256=698CL6-wjup2MvU19RxSmQk3gX7aqW_OOpZB7sbz_Xg,34
 unique_toolkit/short_term_memory/functions.py,sha256=3WiK-xatY5nh4Dr5zlDUye1k3E6kr41RiscwtTplw5k,4484
 unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJs8FEZXcgQTNenw,1406
-unique_toolkit/short_term_memory/service.py,sha256=gdsVzoNqTXmLoBR_-p_lJlZDBo8L7Cr5EKchTNVJg1Q,5233
-unique_toolkit-0.6.8.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.6.8.dist-info/METADATA,sha256=ur8J07QmoFPjKB1oMwjXXsWhOo9XG1g_gabI3q5dpW0,20157
-unique_toolkit-0.6.8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.6.8.dist-info/RECORD,,
+unique_toolkit/short_term_memory/service.py,sha256=0wrwuo6K17Edcuiwp5LMCvxFuBSDP82HF0_5qwi0nYc,5307
+unique_toolkit-0.7.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.7.0.dist-info/METADATA,sha256=jDT49L28AksCEFj_0CXSyHJKg5Q2IWGXqNYLr0ninHY,20561
+unique_toolkit-0.7.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.7.0.dist-info/RECORD,,