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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.24.0.dev8
+Version: 2026.24.0.dev10
 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.dev10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.24.0.dev8"
+version = "2026.24.0.dev10"
 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.dev10}/unique_toolkit/experimental/resources/feature_flags/__init__.py

@@ -31,7 +31,12 @@ Define flag name constants in your service to avoid magic strings::
 See README.md for the full evaluation-order diagram and adoption steps.
 """
 
-from .client import BoundFeatureFlagClient, FeatureFlagClient
+from .client import (
+    BoundFeatureFlagClient,
+    FeatureFlagClient,
+    get_feature_flag_client,
+    is_flag_enabled,
+)
 from .schemas import FlagEvaluation
 from .settings import FeatureFlagSettings
 
@@ -40,4 +45,6 @@ __all__ = [
     "FeatureFlagClient",
     "FlagEvaluation",
     "FeatureFlagSettings",
+    "get_feature_flag_client",
+    "is_flag_enabled",
 ]

{unique_toolkit-2026.24.0.dev8 → unique_toolkit-2026.24.0.dev10}/unique_toolkit/experimental/resources/feature_flags/client.py

@@ -114,12 +114,12 @@ class FeatureFlagClient:
             return FlagEvaluation(
                 value=value, reason="cached" if from_cache else "remote"
             )
-        except Exception:
+        except Exception as exc:
             logger.warning(
                 "FeatureFlagClient: fetch failed for '%s' (company=%s) — trying stale cache",
                 flag,
                 company_id,
-                exc_info=True,
+                exc_info=not isinstance(exc, httpx.HTTPError),
             )
             stale_value, stale_hit = self._cache.get_stale(cache_key)
             if stale_hit:
@@ -213,3 +213,37 @@ class BoundFeatureFlagClient:
 
     async def is_enabled(self, flag: str) -> bool:
         return (await self.evaluate(flag)).value
+
+
+def get_feature_flag_client() -> FeatureFlagClient:
+    """Return the process-level :class:`FeatureFlagClient` singleton.
+
+    Built lazily via :meth:`FeatureFlagClient.from_settings` which reads
+    ``CONFIGURATION_BACKEND_URL`` and ``FEATURE_FLAG_SERVICE_ID`` from env.
+    Falls back to env-var defaults on missing config or transport errors.
+    """
+    return FeatureFlagClient.from_settings()
+
+
+async def is_flag_enabled(
+    flag: str,
+    company_id: str,
+    user_id: str | None = None,
+) -> bool:
+    """Evaluate *flag* and log when disabled.
+
+    Convenience wrapper for route handlers that want a single call-site
+    without manually managing the client singleton.
+    """
+    enabled = await get_feature_flag_client().is_enabled(
+        flag, company_id=company_id, user_id=user_id
+    )
+    if not enabled:
+        user_suffix = f" user {user_id}" if user_id else ""
+        logger.info(
+            "Feature flag '%s' disabled for company %s%s.",
+            flag,
+            company_id,
+            user_suffix,
+        )
+    return enabled

{unique_toolkit-2026.24.0.dev8 → unique_toolkit-2026.24.0.dev10}/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.dev10}/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.dev10}/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,