unique_toolkit 1.42.8__py3-none-any.whl → 1.42.9__py3-none-any.whl

unique_toolkit/agentic/feature_flags/__init__.py

@@ -0,0 +1,6 @@
+from unique_toolkit.agentic.feature_flags.feature_flags import (
+    FeatureFlags,
+    feature_flags,
+)
+
+__all__ = ["FeatureFlags", "feature_flags"]

unique_toolkit/agentic/feature_flags/feature_flags.py

@@ -0,0 +1,32 @@
+from pydantic import Field
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class FeatureFlags(BaseSettings):
+    """Feature flags loaded from environment variables.
+
+    Environment variables are automatically loaded based on field names.
+    For example, `feature_flag_enable_new_answers_ui_un_14411` will be loaded from
+    `FEATURE_FLAG_ENABLE_NEW_ANSWERS_UI_UN_14411`.
+    """
+
+    feature_flag_enable_new_answers_ui_un_14411: str = Field(
+        default="",
+        description="Enable new answers UI (UN-14411). Can be 'true' or comma-separated company IDs.",
+    )
+
+    model_config = SettingsConfigDict(
+        extra="ignore",
+        case_sensitive=False,
+    )
+
+    def is_new_answers_ui_enabled(self, company_id: str | None = None) -> bool:
+        """Check if new answers UI is enabled for the given company."""
+        value = self.feature_flag_enable_new_answers_ui_un_14411
+        return value.lower() == "true" or bool(
+            company_id and company_id in [id.strip() for id in value.split(",")]
+        )
+
+
+# Initialize once at module load - import this where needed
+feature_flags = FeatureFlags()

unique_toolkit/agentic/message_log_manager/service.py

@@ -8,6 +8,7 @@ from collections import defaultdict
 from logging import getLogger
 
 from unique_toolkit.chat.schemas import (
+    MessageLog,
     MessageLogDetails,
     MessageLogStatus,
     MessageLogUncitedReferences,
@@ -61,29 +62,23 @@ class MessageStepLogger:
         self,
         *,
         text: str,
+        status: MessageLogStatus = MessageLogStatus.COMPLETED,
         details: MessageLogDetails | None = None,
-        references: list[ContentReference],
-    ) -> None:
+        references: list[ContentReference] = [],
+    ) -> MessageLog | None:
         """
         Create a full message log entry with question, hits, and references.
-
-        Args:
-            text: The prepared string for the message log entry
-            details: Some formal details about the message log entry
-            references: List of ContentReference objects to reference
+        The created message log entry is returned. If the message log entry is not created, None is returned.
         """
-
-        # Creating a new message log entry with the found hits.
         if not self._chat_service._assistant_message_id:
             _LOGGER.warning(
                 "Assistant message id is not set. Skipping message log entry creation."
             )
             return
-
-        _ = self._chat_service.create_message_log(
+        return self._chat_service.create_message_log(
             message_id=self._chat_service._assistant_message_id,
             text=text,
-            status=MessageLogStatus.COMPLETED,
+            status=status,
             order=self._get_next_message_order(
                 message_id=self._chat_service._assistant_message_id
             ),
@@ -91,3 +86,84 @@ class MessageStepLogger:
             uncited_references=MessageLogUncitedReferences(data=references),
             references=[],
         )
+
+    def update_message_log_entry(
+        self,
+        *,
+        message_log: MessageLog,
+        text: str | None = None,
+        status: MessageLogStatus,
+        details: MessageLogDetails | None = None,
+        references: list[ContentReference] = [],
+    ) -> MessageLog:
+        """
+        Update a message log entry with a new status.
+        The updated message log entry is returned. If the message log entry is not updated, None is returned.
+        """
+        if message_log.message_log_id is None:
+            _LOGGER.warning(
+                "Message log id is not set. Skipping message log entry update."
+            )
+            return message_log
+        return self._chat_service.update_message_log(
+            message_log_id=message_log.message_log_id,
+            order=message_log.order,
+            text=text,
+            status=status,
+            details=details or MessageLogDetails(data=[]),
+            uncited_references=MessageLogUncitedReferences(data=references),
+            references=[],
+        )
+
+    def create_or_update_message_log(
+        self,
+        *,
+        active_message_log: MessageLog | None,
+        header: str,
+        progress_message: str | None = None,
+        status: MessageLogStatus = MessageLogStatus.RUNNING,
+        details: MessageLogDetails | None = None,
+        references: list[ContentReference] | None = [],
+    ) -> MessageLog | None:
+        """
+        Create a new message log entry or update an existing one.
+
+        This is a convenience method that handles the common pattern of:
+        - Creating a new message log if active_message_log is None
+        - Updating the existing message log otherwise
+
+        Args:
+            active_message_log: The current active message log, or None if none exists
+            header: The header to show in bold (e.g., "Internal Search", "Web Search")
+            progress_message: Optional progress message to append after the display name
+            status: The status of the message log
+            details: Optional message log details
+            references: Optional list of content references
+
+        Returns:
+            The created or updated MessageLog, or None if the operation failed
+        """
+        text = (
+            f"**{header}**\n{progress_message}"
+            if progress_message is not None
+            else f"**{header}**"
+        )
+
+        if references is None:
+            references = []
+
+        if active_message_log is None:
+            return self.create_message_log_entry(
+                text=text,
+                details=details,
+                references=references,
+                status=status,
+            )
+        else:
+            return self.update_message_log_entry(
+                message_log=active_message_log,
+                text=text,
+                status=status,
+                details=details,
+                references=references,
+            )

{unique_toolkit-1.42.8.dist-info → unique_toolkit-1.42.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.42.8
+Version: 1.42.9
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -124,6 +124,9 @@ 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).
 
+## [1.42.9] - 2026-01-11
+- Include feature flag to have message logs compatible with new ChatUI
+
 ## [1.42.8] - 2026-01-08
 - Add validator to `BaseMetadata` in case `additional_sheet_information` is empty
 - Add more code snippets to create references and pull file metadata

{unique_toolkit-1.42.8.dist-info → unique_toolkit-1.42.9.dist-info}/RECORD

@@ -59,6 +59,8 @@ unique_toolkit/agentic/evaluation/output_parser.py,sha256=0FDo8YY_Dc4qlTNeYyQkzn
 unique_toolkit/agentic/evaluation/schemas.py,sha256=m9JMCUmeqP8KhsJOVEzsz6dRXUe1uKw-bxRDtn5qwvM,3156
 unique_toolkit/agentic/evaluation/tests/test_context_relevancy_service.py,sha256=4tDxHTApbaTMxN1sNS8WCqj2BweRk6YqZ5_zHP45jto,7977
 unique_toolkit/agentic/evaluation/tests/test_output_parser.py,sha256=RN_HcBbU6qy_e_PoYyUFcjWnp3ymJ6-gLj6TgEOupAI,3107
+unique_toolkit/agentic/feature_flags/__init__.py,sha256=LhE2cHoa9AYBOR7TjiIToOn46sttm9paKcrzE7gnDPM,149
+unique_toolkit/agentic/feature_flags/feature_flags.py,sha256=4jPH0GGGt5-tQ6PJWNpMBIlYzNrQIIqBLx8W02lwxD0,1140
 unique_toolkit/agentic/history_manager/history_construction_with_contents.py,sha256=kzxpVzTtQqL8TjdIvOy7gkRVxD4BsOMyimECryg7vdc,9060
 unique_toolkit/agentic/history_manager/history_manager.py,sha256=7V7_173XkAjc8otBACF0G3dbqRs34FSlURbBPrE95Wk,9537
 unique_toolkit/agentic/history_manager/loop_token_reducer.py,sha256=3c-uonDovtanEJUpAO4zlA4-n9MS_Ws_V0Yb6G7hPM0,20172
@@ -77,7 +79,7 @@ unique_toolkit/agentic/loop_runner/runners/qwen/__init__.py,sha256=b2zAQIjtPy_BB
 unique_toolkit/agentic/loop_runner/runners/qwen/helpers.py,sha256=JBnxeYKu8HiUq3VHjnb8XdHCe1c3cJ3OwagckF4UvnU,1763
 unique_toolkit/agentic/loop_runner/runners/qwen/qwen_runner.py,sha256=7tYfTkNKwxYVn6C1Htya5QEIK2BfWkxCPNkdMoaZGq0,4920
 unique_toolkit/agentic/message_log_manager/__init__.py,sha256=3-KY_sGkPbNoSnrzwPY0FQIJNnsz4NHXvocXgGRUeuE,169
-unique_toolkit/agentic/message_log_manager/service.py,sha256=AiuIq2dKQg9Y8bEYgGcve1X8-WRRdqPZXaZXXLJxfFM,3057
+unique_toolkit/agentic/message_log_manager/service.py,sha256=DeixRsgmyDdWsL-4e12flB9d-Uy7s0X52upPIbTn6vA,5895
 unique_toolkit/agentic/postprocessor/postprocessor_manager.py,sha256=CoKzVFeLIr1eRP3ZLnmUJ8KNsFLyvK5iuvUilbcGAm0,7662
 unique_toolkit/agentic/reference_manager/reference_manager.py,sha256=x51CT0D8HHu2LzgXdHGy0leOYpjnsxVbPZ2nc28G9mA,4005
 unique_toolkit/agentic/responses_api/__init__.py,sha256=9WTO-ef7fGE9Y1QtZJFm8Q_jkwK8Srtl-HWvpAD2Wxs,668
@@ -215,7 +217,7 @@ unique_toolkit/short_term_memory/service.py,sha256=5PeVBu1ZCAfyDb2HLVvlmqSbyzBBu
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=Ozhh70qCn2yOzRWr9d8WmJeTo7AQurwd3tStgBMPFLA,1246
 unique_toolkit/test_utilities/events.py,sha256=_mwV2bs5iLjxS1ynDCjaIq-gjjKhXYCK-iy3dRfvO3g,6410
-unique_toolkit-1.42.8.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.42.8.dist-info/METADATA,sha256=su-BJN4NWsxuKBdWmeWGF2tGzPDwh0Da_ea8stKGzWE,47356
-unique_toolkit-1.42.8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.42.8.dist-info/RECORD,,
+unique_toolkit-1.42.9.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.42.9.dist-info/METADATA,sha256=UlLw100oSlKFi6ycPSH6swuqJZwbzMQebOdbfbookk0,47453
+unique_toolkit-1.42.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.42.9.dist-info/RECORD,,