unique_toolkit 1.3.1__tar.gz → 1.3.3__tar.gz

{unique_toolkit-1.3.1 → unique_toolkit-1.3.3}/CHANGELOG.md

@@ -5,6 +5,11 @@ 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.3.3] - 2025-09-30
+- fix bug in exclusive tools not making them selectable
+
+## [1.3.2] - 2025-09-29
+- Auto-update unique settings on event arrival in SSE client
 
 ## [1.3.1] - 2025-09-29
 - More documentation on referencing

{unique_toolkit-1.3.1 → unique_toolkit-1.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.3.1
+Version: 1.3.3
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,6 +118,11 @@ 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.3.3] - 2025-09-30
+- fix bug in exclusive tools not making them selectable
+
+## [1.3.2] - 2025-09-29
+- Auto-update unique settings on event arrival in SSE client
 
 ## [1.3.1] - 2025-09-29
 - More documentation on referencing

{unique_toolkit-1.3.1 → unique_toolkit-1.3.3}/pyproject.toml

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

{unique_toolkit-1.3.1 → unique_toolkit-1.3.3}/unique_toolkit/agentic/tools/test/test_mcp_manager.py

@@ -402,8 +402,8 @@ class TestMCPManager:
         # Should only have the exclusive tool, MCP tools should be ignored
         tools = tool_manager.get_tools()
         tool_names = [tool.name for tool in tools]
-        assert "internal_search" not in tool_names
-        assert "mcp_test_tool" in tool_names
+        assert "internal_search" in tool_names
+        assert "mcp_test_tool" not in tool_names
         assert len(tools) == 1
 
     def test_init_tools_with_disabled_tool_config(

{unique_toolkit-1.3.1 → unique_toolkit-1.3.3}/unique_toolkit/agentic/tools/tool_manager.py

@@ -125,11 +125,7 @@ class ToolManager:
             if len(self._tool_choices) > 0 and t.name not in self._tool_choices:
                 continue
             # is the tool exclusive and has been choosen by the user?
-            if (
-                t.is_exclusive()
-                and len(tool_choices) > 0
-                and t.name not in tool_choices
-            ):
+            if t.is_exclusive() and len(tool_choices) > 0 and t.name in tool_choices:
                 self._tools = [t]  # override all other tools
                 break
             # if the tool is exclusive but no tool choices are given, skip it

{unique_toolkit-1.3.1 → unique_toolkit-1.3.3}/unique_toolkit/app/dev_util.py

@@ -46,7 +46,8 @@ def get_event_generator(
     event_type: type[T],
 ) -> Generator[T, None, None]:
     """
-    Generator that yields only events of the specified type from an SSE stream.
+    Generator that updates the unique settings according to the events and
+    yields only events of the specified type from an SSE stream.
 
     Args:
         sse_client: The SSE client to read events from
@@ -74,6 +75,8 @@ def get_event_generator(
             ):
                 continue
 
+            unique_settings.update_from_event(event=parsed_event)
+
             yield parsed_event
 
         except Exception as e:

{unique_toolkit-1.3.1 → unique_toolkit-1.3.3}/unique_toolkit/app/unique_settings.py

@@ -1,7 +1,7 @@
 import os
 from logging import getLogger
 from pathlib import Path
-from typing import Self, TypeVar
+from typing import TYPE_CHECKING, Self, TypeVar
 from urllib.parse import ParseResult, urlparse, urlunparse
 
 import unique_sdk
@@ -9,6 +9,10 @@ from platformdirs import user_config_dir
 from pydantic import AliasChoices, Field, SecretStr, model_validator
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
+if TYPE_CHECKING:
+    from unique_toolkit.app.schemas import BaseEvent
+
+
 logger = getLogger(__name__)
 
 T = TypeVar("T", bound=BaseSettings)
@@ -159,6 +163,13 @@ class UniqueAuth(BaseSettings):
     def _warn_about_defaults(self) -> Self:
         return warn_about_defaults(self)
 
+    @classmethod
+    def from_event(cls, event: "BaseEvent") -> Self:
+        return cls(
+            company_id=SecretStr(event.company_id),
+            user_id=SecretStr(event.user_id),
+        )
+
 
 class UniqueChatEventFilterOptions(BaseSettings):
     # Empty string evals to False
@@ -197,10 +208,10 @@ class UniqueSettings:
         chat_event_filter_options: UniqueChatEventFilterOptions | None = None,
         env_file: Path | None = None,
     ):
-        self.app = app
-        self.auth = auth
-        self.api = api
-        self.chat_event_filter_options = chat_event_filter_options
+        self._app = app
+        self._auth = auth
+        self._api = api
+        self._chat_event_filter_options = chat_event_filter_options
         self._env_file: Path | None = (
             env_file if (env_file and env_file.exists()) else None
         )
@@ -310,9 +321,9 @@ class UniqueSettings:
         This method configures the global unique_sdk module with the API key,
         app ID, and base URL from these settings.
         """
-        unique_sdk.api_key = self.app.key.get_secret_value()
-        unique_sdk.app_id = self.app.id.get_secret_value()
-        unique_sdk.api_base = self.api.sdk_url()
+        unique_sdk.api_key = self._app.key.get_secret_value()
+        unique_sdk.app_id = self._app.id.get_secret_value()
+        unique_sdk.api_base = self._api.sdk_url()
 
     @classmethod
     def from_env_auto_with_sdk_init(
@@ -331,3 +342,26 @@ class UniqueSettings:
         settings = cls.from_env_auto(filename)
         settings.init_sdk()
         return settings
+
+    def update_from_event(self, event: "BaseEvent") -> None:
+        self._auth = UniqueAuth.from_event(event)
+
+    @property
+    def api(self) -> UniqueApi:
+        return self._api
+
+    @property
+    def app(self) -> UniqueApp:
+        return self._app
+
+    @property
+    def auth(self) -> UniqueAuth:
+        return self._auth
+
+    @auth.setter
+    def auth(self, value: UniqueAuth) -> None:
+        self._auth = value
+
+    @property
+    def chat_event_filter_options(self) -> UniqueChatEventFilterOptions | None:
+        return self._chat_event_filter_options