unique-toolkit 2026.24.0.dev8__tar.gz → 2026.24.0.dev9__tar.gz

{unique_toolkit-2026.24.0.dev8 → unique_toolkit-2026.24.0.dev9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.24.0.dev8
+Version: 2026.24.0.dev9
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>
@@ -16,7 +16,7 @@ Requires-Dist: sseclient>=0.0.27,<0.0.28
 Requires-Dist: openai>=1.105.0,<3
 Requires-Dist: platformdirs>=4.0.0,<5
 Requires-Dist: pillow>=12.1.1
-Requires-Dist: unique-sdk>=2026.24.0.dev3,<2026.24.0rc0
+Requires-Dist: unique-sdk>=2026.24.0.dev4,<2026.24.0rc0
 Requires-Dist: jambo>=0.1.2,<0.2
 Requires-Dist: jinja2>=3.1.6,<4
 Requires-Dist: docxtpl>=0.20.1,<0.21

{unique_toolkit-2026.24.0.dev8 → unique_toolkit-2026.24.0.dev9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.24.0.dev8"
+version = "2026.24.0.dev9"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"
@@ -28,7 +28,7 @@ dependencies = [
     "openai>=1.105.0,<3",
     "platformdirs>=4.0.0,<5",
     "pillow>=12.1.1",
-    "unique-sdk>=2026.24.0.dev3,<2026.24.0rc0",
+    "unique-sdk>=2026.24.0.dev4,<2026.24.0rc0",
     "jambo>=0.1.2,<0.2",
     "jinja2>=3.1.6,<4",
     "docxtpl>=0.20.1,<0.21",

{unique_toolkit-2026.24.0.dev8 → unique_toolkit-2026.24.0.dev9}/unique_toolkit/services/chat_service.py

@@ -2209,7 +2209,9 @@ class ChatService(ChatServiceDeprecated):
             message_id=message_id,
         )
 
-    def find_chat_memory_by_id(self, *, chat_id: str, key: str) -> ShortTermMemory:
+    def find_chat_memory_by_id(
+        self, *, chat_id: str, key: str
+    ) -> ShortTermMemory | None:
         """Finds the latest short-term memory for a specific chat synchronously.
 
         Args:
@@ -2217,7 +2219,7 @@ class ChatService(ChatServiceDeprecated):
             key (str): The memory key
 
         Returns:
-            ShortTermMemory: The latest short-term memory
+            ShortTermMemory | None: The latest short-term memory, or None if not found
 
         Raises:
             Exception: If the retrieval fails
@@ -2231,7 +2233,7 @@ class ChatService(ChatServiceDeprecated):
 
     async def find_chat_memory_by_id_async(
         self, *, chat_id: str, key: str
-    ) -> ShortTermMemory:
+    ) -> ShortTermMemory | None:
         """Finds the latest short-term memory for a specific chat asynchronously.
 
         Args:
@@ -2239,7 +2241,7 @@ class ChatService(ChatServiceDeprecated):
             key (str): The memory key
 
         Returns:
-            ShortTermMemory: The latest short-term memory
+            ShortTermMemory | None: The latest short-term memory, or None if not found
 
         Raises:
             Exception: If the retrieval fails
@@ -2253,7 +2255,7 @@ class ChatService(ChatServiceDeprecated):
 
     def find_message_memory_by_id(
         self, *, message_id: str, key: str
-    ) -> ShortTermMemory:
+    ) -> ShortTermMemory | None:
         """Finds the latest short-term memory for a specific message synchronously.
 
         Args:
@@ -2261,7 +2263,7 @@ class ChatService(ChatServiceDeprecated):
             key (str): The memory key
 
         Returns:
-            ShortTermMemory: The latest short-term memory
+            ShortTermMemory | None: The latest short-term memory, or None if not found
 
         Raises:
             Exception: If the retrieval fails
@@ -2275,7 +2277,7 @@ class ChatService(ChatServiceDeprecated):
 
     async def find_message_memory_by_id_async(
         self, *, message_id: str, key: str
-    ) -> ShortTermMemory:
+    ) -> ShortTermMemory | None:
         """Finds the latest short-term memory for a specific message asynchronously.
 
         Args:
@@ -2283,7 +2285,7 @@ class ChatService(ChatServiceDeprecated):
             key (str): The memory key
 
         Returns:
-            ShortTermMemory: The latest short-term memory
+            ShortTermMemory | None: The latest short-term memory, or None if not found
 
         Raises:
             Exception: If the retrieval fails
@@ -2416,14 +2418,14 @@ class ChatService(ChatServiceDeprecated):
             value=value,
         )
 
-    def find_chat_memory(self, *, key: str) -> ShortTermMemory:
+    def find_chat_memory(self, *, key: str) -> ShortTermMemory | None:
         """Finds the latest short-term memory for the current chat synchronously.
 
         Args:
             key (str): The memory key
 
         Returns:
-            ShortTermMemory: The latest short-term memory
+            ShortTermMemory | None: The latest short-term memory, or None if not found
 
         Raises:
             Exception: If the retrieval fails
@@ -2433,14 +2435,14 @@ class ChatService(ChatServiceDeprecated):
             key=key,
         )
 
-    async def find_chat_memory_async(self, *, key: str) -> ShortTermMemory:
+    async def find_chat_memory_async(self, *, key: str) -> ShortTermMemory | None:
         """Finds the latest short-term memory for the current chat asynchronously.
 
         Args:
             key (str): The memory key
 
         Returns:
-            ShortTermMemory: The latest short-term memory
+            ShortTermMemory | None: The latest short-term memory, or None if not found
 
         Raises:
             Exception: If the retrieval fails
@@ -2451,21 +2453,23 @@ class ChatService(ChatServiceDeprecated):
         )
 
     @overload
-    def find_message_memory(self, *, key: str) -> ShortTermMemory: ...
+    def find_message_memory(self, *, key: str) -> ShortTermMemory | None: ...
 
     @overload
-    def find_message_memory(self, *, key: str, message_id: str) -> ShortTermMemory: ...
+    def find_message_memory(
+        self, *, key: str, message_id: str
+    ) -> ShortTermMemory | None: ...
 
     def find_message_memory(
         self, *, key: str, message_id: str | None = None
-    ) -> ShortTermMemory:
+    ) -> ShortTermMemory | None:
         """Finds the latest short-term memory for the current assistant message synchronously.
 
         Args:
             key (str): The memory key
 
         Returns:
-            ShortTermMemory: The latest short-term memory
+            ShortTermMemory | None: The latest short-term memory, or None if not found
 
         Raises:
             Exception: If the retrieval fails
@@ -2476,23 +2480,25 @@ class ChatService(ChatServiceDeprecated):
         )
 
     @overload
-    async def find_message_memory_async(self, *, key: str) -> ShortTermMemory: ...
+    async def find_message_memory_async(
+        self, *, key: str
+    ) -> ShortTermMemory | None: ...
 
     @overload
     async def find_message_memory_async(
         self, *, key: str, message_id: str
-    ) -> ShortTermMemory: ...
+    ) -> ShortTermMemory | None: ...
 
     async def find_message_memory_async(
         self, *, key: str, message_id: str | None = None
-    ) -> ShortTermMemory:
+    ) -> ShortTermMemory | None:
         """Finds the latest short-term memory for the current assistant message asynchronously.
 
         Args:
             key (str): The memory key
 
         Returns:
-            ShortTermMemory: The latest short-term memory
+            ShortTermMemory | None: The latest short-term memory, or None if not found
 
         Raises:
             Exception: If the retrieval fails

{unique_toolkit-2026.24.0.dev8 → unique_toolkit-2026.24.0.dev9}/unique_toolkit/short_term_memory/functions.py

@@ -16,7 +16,7 @@ async def find_latest_memory_async(
     key: str,
     chat_id: str | None = None,
     message_id: str | None = None,
-) -> ShortTermMemory:
+) -> ShortTermMemory | None:
     """
     Find the latest short term memory asynchronously.
 
@@ -28,7 +28,7 @@ async def find_latest_memory_async(
         message_id (str | None): The message ID.
 
     Returns:
-        ShortTermMemory: The latest short term memory.
+        ShortTermMemory | None: The latest short term memory, or None if not found.
 
     Raises:
         Exception: If an error occurs.
@@ -42,6 +42,8 @@ async def find_latest_memory_async(
             chatId=chat_id,
             messageId=message_id,
         )
+        if not stm:
+            return None
         return ShortTermMemory.model_validate(stm, by_alias=True, by_name=True)
     except Exception as e:
         logger.error(f"Error finding latest short term memory: {e}")
@@ -54,7 +56,7 @@ def find_latest_memory(
     key: str,
     chat_id: str | None = None,
     message_id: str | None = None,
-) -> ShortTermMemory:
+) -> ShortTermMemory | None:
     """
     Find the latest short term memory.
 
@@ -66,7 +68,7 @@ def find_latest_memory(
         message_id (str | None): The message ID.
 
     Returns:
-        ShortTermMemory: The latest short term memory.
+        ShortTermMemory | None: The latest short term memory, or None if not found.
 
     Raises:
         Exception: If an error occurs.
@@ -80,6 +82,8 @@ def find_latest_memory(
             chatId=chat_id,
             messageId=message_id,
         )
+        if not stm:
+            return None
         return ShortTermMemory.model_validate(stm, by_alias=True, by_name=True)
     except Exception as e:
         logger.error(f"Error finding latest short term memory: {e}")
@@ -183,7 +187,7 @@ def find_last_chat_memory(
     company_id: str,
     key: str,
     chat_id: str,
-) -> ShortTermMemory:
+) -> ShortTermMemory | None:
     """
     Find the last chat short term memory.
     """
@@ -193,6 +197,8 @@ def find_last_chat_memory(
         memoryName=key,
         chatId=chat_id,
     )
+    if not stm:
+        return None
     return ShortTermMemory.model_validate(stm, by_alias=True, by_name=True)
 
 
@@ -202,7 +208,7 @@ async def find_last_chat_memory_async(
     company_id: str,
     key: str,
     chat_id: str,
-) -> ShortTermMemory:
+) -> ShortTermMemory | None:
     """
     Find the last chat short term memory.
     """
@@ -212,6 +218,8 @@ async def find_last_chat_memory_async(
         memoryName=key,
         chatId=chat_id,
     )
+    if not stm:
+        return None
     return ShortTermMemory.model_validate(stm, by_alias=True, by_name=True)
 
 
@@ -221,7 +229,7 @@ def find_last_message_memory(
     company_id: str,
     key: str,
     message_id: str,
-) -> ShortTermMemory:
+) -> ShortTermMemory | None:
     """
     Find the last message short term memory.
     """
@@ -232,6 +240,8 @@ def find_last_message_memory(
         memoryName=key,
         messageId=message_id,
     )
+    if not stm:
+        return None
     return ShortTermMemory.model_validate(stm, by_alias=True, by_name=True)
 
 
@@ -241,7 +251,7 @@ async def find_last_message_memory_async(
     company_id: str,
     key: str,
     message_id: str,
-) -> ShortTermMemory:
+) -> ShortTermMemory | None:
     """
     Find the last message short term memory.
     """
@@ -252,6 +262,8 @@ async def find_last_message_memory_async(
         memoryName=key,
         messageId=message_id,
     )
+    if not stm:
+        return None
     return ShortTermMemory.model_validate(stm, by_alias=True, by_name=True)
 
 

{unique_toolkit-2026.24.0.dev8 → unique_toolkit-2026.24.0.dev9}/unique_toolkit/short_term_memory/service.py

@@ -209,7 +209,7 @@ class ShortTermMemoryService:
             message_id=chat_event.payload.user_message.id,
         )
 
-    async def find_latest_memory_async(self, key: str) -> ShortTermMemory:
+    async def find_latest_memory_async(self, key: str) -> ShortTermMemory | None:
         """
         Find the latest short term memory.
 
@@ -217,7 +217,7 @@ class ShortTermMemoryService:
             key (str): The key.
 
         Returns:
-            ShortTermMemory: The latest short term memory.
+            ShortTermMemory | None: The latest short term memory, or None if not found.
 
         Raises:
             Exception: If an error occurs.
@@ -231,7 +231,7 @@ class ShortTermMemoryService:
             message_id=self._message_id,
         )
 
-    def find_latest_memory(self, key: str) -> ShortTermMemory:
+    def find_latest_memory(self, key: str) -> ShortTermMemory | None:
         """
         Find the latest short term memory.
 
@@ -239,7 +239,7 @@ class ShortTermMemoryService:
             key (str): The key.
 
         Returns:
-            ShortTermMemory: The latest short term memory.
+            ShortTermMemory | None: The latest short term memory, or None if not found.
 
         Raises:
             Exception: If an error occurs.
@@ -312,7 +312,7 @@ class ShortTermMemoryService:
         )
 
     @deprecated("Use find_latest_memory_async instead")
-    async def get(self, key: str) -> ShortTermMemory:
+    async def get(self, key: str) -> ShortTermMemory | None:
         return await find_latest_memory_async(
             user_id=self._user_id,
             company_id=self._company_id,