unique_toolkit 1.42.7__tar.gz → 1.42.9__tar.gz

{unique_toolkit-1.42.7 → unique_toolkit-1.42.9}/CHANGELOG.md

@@ -5,6 +5,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).
 
+## [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
+
 ## [1.42.7] - 2026-01-08
 - Add aliases for endpoint secret env var.
 

{unique_toolkit-1.42.7 → unique_toolkit-1.42.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.42.7
+Version: 1.42.9
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -124,6 +124,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).
 
+## [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
+
 ## [1.42.7] - 2026-01-08
 - Add aliases for endpoint secret env var.
 

{unique_toolkit-1.42.7 → unique_toolkit-1.42.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "1.42.7"
+version = "1.42.9"
 description = ""
 authors = [
     "Cedric Klinkert <cedric.klinkert@unique.ch>",

unique_toolkit-1.42.9/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-1.42.9/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-1.42.9/unique_toolkit/agentic/message_log_manager/service.py

@@ -0,0 +1,169 @@
+"""
+Logger Service for Unique Agentic tools.
+
+Target of the method is to extend the step tracking on all levels of the tool.
+"""
+
+from collections import defaultdict
+from logging import getLogger
+
+from unique_toolkit.chat.schemas import (
+    MessageLog,
+    MessageLogDetails,
+    MessageLogStatus,
+    MessageLogUncitedReferences,
+)
+from unique_toolkit.chat.service import ChatService
+from unique_toolkit.content.schemas import ContentReference
+
+_LOGGER = getLogger(__name__)
+
+# Per-request counters for message log ordering - keyed by message_id
+# This is a mandatory global variable since we have in the system a bug which makes it impossible to use it as a proper class variable.
+_request_counters = defaultdict(int)
+
+
+class MessageStepLogger:
+    """
+    Logger class for tracking message steps in agentic tools.
+
+    This class provides utilities for creating message log entries with proper
+    ordering, references, and details.
+    """
+
+    def __init__(self, chat_service: ChatService) -> None:
+        """
+        Initialize the MessageStepLogger.
+
+        Args:
+            chat_service: ChatService instance for logging
+        """
+        self._chat_service: ChatService = chat_service
+
+    def _get_next_message_order(self, *, message_id: str) -> int:
+        """
+        Get the next message log order number for a specific request.
+
+        Each message_id (request) gets its own counter: 1, 2, 3, 4, ...
+        This ensures proper ordering within each request while avoiding
+        race conditions between concurrent requests.
+
+        Args:
+            message_id: The message ID to get the next order for
+
+        Returns:
+            Next order number for this specific request
+        """
+
+        _request_counters[message_id] += 1
+        return _request_counters[message_id]
+
+    def create_message_log_entry(
+        self,
+        *,
+        text: str,
+        status: MessageLogStatus = MessageLogStatus.COMPLETED,
+        details: MessageLogDetails | None = None,
+        references: list[ContentReference] = [],
+    ) -> MessageLog | None:
+        """
+        Create a full message log entry with question, hits, and references.
+        The created message log entry is returned. If the message log entry is not created, None is returned.
+        """
+        if not self._chat_service._assistant_message_id:
+            _LOGGER.warning(
+                "Assistant message id is not set. Skipping message log entry creation."
+            )
+            return
+        return self._chat_service.create_message_log(
+            message_id=self._chat_service._assistant_message_id,
+            text=text,
+            status=status,
+            order=self._get_next_message_order(
+                message_id=self._chat_service._assistant_message_id
+            ),
+            details=details or MessageLogDetails(data=[]),
+            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.7 → unique_toolkit-1.42.9}/unique_toolkit/agentic_table/schemas.py

@@ -53,6 +53,13 @@ class BaseMetadata(BaseModel):
         default_factory=dict, description="Additional information for the sheet"
     )
 
+    @field_validator("additional_sheet_information", mode="before")
+    @classmethod
+    def normalize_additional_sheet_information(cls, v):
+        if v is None:
+            return {}
+        return v
+
 
 class RowMetadataEntry(BaseModel):
     model_config = get_configuration_dict()

unique_toolkit-1.42.7/unique_toolkit/agentic/message_log_manager/service.py

@@ -1,93 +0,0 @@
-"""
-Logger Service for Unique Agentic tools.
-
-Target of the method is to extend the step tracking on all levels of the tool.
-"""
-
-from collections import defaultdict
-from logging import getLogger
-
-from unique_toolkit.chat.schemas import (
-    MessageLogDetails,
-    MessageLogStatus,
-    MessageLogUncitedReferences,
-)
-from unique_toolkit.chat.service import ChatService
-from unique_toolkit.content.schemas import ContentReference
-
-_LOGGER = getLogger(__name__)
-
-# Per-request counters for message log ordering - keyed by message_id
-# This is a mandatory global variable since we have in the system a bug which makes it impossible to use it as a proper class variable.
-_request_counters = defaultdict(int)
-
-
-class MessageStepLogger:
-    """
-    Logger class for tracking message steps in agentic tools.
-
-    This class provides utilities for creating message log entries with proper
-    ordering, references, and details.
-    """
-
-    def __init__(self, chat_service: ChatService) -> None:
-        """
-        Initialize the MessageStepLogger.
-
-        Args:
-            chat_service: ChatService instance for logging
-        """
-        self._chat_service: ChatService = chat_service
-
-    def _get_next_message_order(self, *, message_id: str) -> int:
-        """
-        Get the next message log order number for a specific request.
-
-        Each message_id (request) gets its own counter: 1, 2, 3, 4, ...
-        This ensures proper ordering within each request while avoiding
-        race conditions between concurrent requests.
-
-        Args:
-            message_id: The message ID to get the next order for
-
-        Returns:
-            Next order number for this specific request
-        """
-
-        _request_counters[message_id] += 1
-        return _request_counters[message_id]
-
-    def create_message_log_entry(
-        self,
-        *,
-        text: str,
-        details: MessageLogDetails | None = None,
-        references: list[ContentReference],
-    ) -> 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
-        """
-
-        # 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(
-            message_id=self._chat_service._assistant_message_id,
-            text=text,
-            status=MessageLogStatus.COMPLETED,
-            order=self._get_next_message_order(
-                message_id=self._chat_service._assistant_message_id
-            ),
-            details=details or MessageLogDetails(data=[]),
-            uncited_references=MessageLogUncitedReferences(data=references),
-            references=[],
-        )