unique_toolkit 1.0.0__tar.gz → 1.1.0__tar.gz

{unique_toolkit-1.0.0 → unique_toolkit-1.1.0}/CHANGELOG.md

@@ -5,6 +5,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.1.0] - 2025-09-18
+- Enable chat event filtering in SSE event generator via env variables
+
 ## [1.0.0] - 2025-09-18
 - Bump toolkit version to allow for both patch and minor updates
 

{unique_toolkit-1.0.0 → unique_toolkit-1.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.0.0
+Version: 1.1.0
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,6 +118,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.1.0] - 2025-09-18
+- Enable chat event filtering in SSE event generator via env variables
+
 ## [1.0.0] - 2025-09-18
 - Bump toolkit version to allow for both patch and minor updates
 

{unique_toolkit-1.0.0 → unique_toolkit-1.1.0}/pyproject.toml

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

unique_toolkit-1.1.0/unique_toolkit/agentic/__init__.py

@@ -0,0 +1 @@
+

{unique_toolkit-1.0.0 → unique_toolkit-1.1.0}/unique_toolkit/app/dev_util.py

@@ -42,7 +42,8 @@ def get_sse_client(
 
 
 def get_event_generator(
-    unique_settings: UniqueSettings, event_type: type[T]
+    unique_settings: UniqueSettings,
+    event_type: type[T],
 ) -> Generator[T, None, None]:
     """
     Generator that yields only events of the specified type from an SSE stream.
@@ -68,7 +69,9 @@ def get_event_generator(
         try:
             payload = json.loads(sse_event.data)
             parsed_event = event_type.model_validate(payload)
-            if parsed_event is None:
+            if parsed_event is None or parsed_event.filter_event(
+                filter_options=unique_settings.chat_event_filter_options
+            ):
                 continue
 
             yield parsed_event

{unique_toolkit-1.0.0 → unique_toolkit-1.1.0}/unique_toolkit/app/schemas.py

@@ -1,14 +1,18 @@
 import json
 from enum import StrEnum
 from pathlib import Path
-from typing import Any, Optional
+from typing import Any, Generic, Optional, TypeVar, override
 
 from humps import camelize
 from pydantic import BaseModel, ConfigDict, Field, field_validator
+from pydantic_settings import BaseSettings
 from typing_extensions import deprecated
 
+from unique_toolkit.app.unique_settings import UniqueChatEventFilterOptions
 from unique_toolkit.smart_rules.compile import UniqueQL, parse_uniqueql
 
+FilterOptionsT = TypeVar("FilterOptionsT", bound=BaseSettings)
+
 # set config to convert camelCase to snake_case
 model_config = ConfigDict(
     alias_generator=camelize,
@@ -30,7 +34,7 @@ class EventName(StrEnum):
     MAGIC_TABLE_UPDATE_CELL = "unique.magic-table.update-cell"
 
 
-class BaseEvent(BaseModel):
+class BaseEvent(BaseModel, Generic[FilterOptionsT]):
     model_config = model_config
 
     id: str
@@ -46,6 +50,10 @@ class BaseEvent(BaseModel):
             data = json.load(f)
         return cls.model_validate(data)
 
+    def filter_event(self, *, filter_options: FilterOptionsT | None = None) -> bool:
+        """Determine if event should be filtered out and be neglected."""
+        return False
+
 
 ###
 # MCP schemas
@@ -242,6 +250,28 @@ class ChatEvent(BaseEvent):
             "assistant": {"id": self.payload.assistant_id},
         }
 
+    @override
+    def filter_event(
+        self, *, filter_options: UniqueChatEventFilterOptions | None = None
+    ) -> bool:
+        # Empty string evals to False
+        if filter_options is None:
+            return False
+
+        if (
+            filter_options.assistant_ids
+            and self.payload.assistant_id not in filter_options.assistant_ids
+        ):
+            return True
+
+        if (
+            filter_options.references_in_code
+            and self.payload.name not in filter_options.references_in_code
+        ):
+            return True
+
+        return super().filter_event(filter_options=filter_options)
+
 
 @deprecated(
     """Use the more specific `ChatEvent` instead that has the same properties. \

{unique_toolkit-1.0.0 → unique_toolkit-1.1.0}/unique_toolkit/app/unique_settings.py

@@ -49,6 +49,7 @@ class UniqueApp(BaseSettings):
         deprecated="Use UniqueApi.base_url instead",
     )
     endpoint: str = Field(default="dummy")
+
     endpoint_secret: SecretStr = Field(default=SecretStr("dummy_secret"))
 
     @model_validator(mode="after")
@@ -159,6 +160,29 @@ class UniqueAuth(BaseSettings):
         return warn_about_defaults(self)
 
 
+class UniqueChatEventFilterOptions(BaseSettings):
+    # Empty string evals to False
+    assistant_ids: list[str] = Field(
+        default=[],
+        description="The assistant ids (space) to filter by. Default is all assistants.",
+    )
+    references_in_code: list[str] = Field(
+        default=[],
+        description="The module (reference) names in code to filter by. Default is all modules.",
+    )
+
+    model_config = SettingsConfigDict(
+        env_prefix="unique_chat_event_filter_options_",
+        env_file_encoding="utf-8",
+        case_sensitive=False,
+        extra="ignore",
+    )
+
+    @model_validator(mode="after")
+    def _warn_about_defaults(self) -> Self:
+        return warn_about_defaults(self)
+
+
 class EnvFileNotFoundError(FileNotFoundError):
     """Raised when no environment file can be found in any of the expected locations."""
 
@@ -169,11 +193,14 @@ class UniqueSettings:
         auth: UniqueAuth,
         app: UniqueApp,
         api: UniqueApi,
+        *,
+        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._env_file: Path | None = (
             env_file if (env_file and env_file.exists()) else None
         )
@@ -220,7 +247,10 @@ class UniqueSettings:
         )
 
     @classmethod
-    def from_env(cls, env_file: Path | None = None) -> "UniqueSettings":
+    def from_env(
+        cls,
+        env_file: Path | None = None,
+    ) -> "UniqueSettings":
         """Initialize settings from environment variables and/or env file.
 
         Args:
@@ -241,7 +271,14 @@ class UniqueSettings:
         auth = UniqueAuth(_env_file=env_file_str)  # type: ignore[call-arg]
         app = UniqueApp(_env_file=env_file_str)  # type: ignore[call-arg]
         api = UniqueApi(_env_file=env_file_str)  # type: ignore[call-arg]
-        return cls(auth=auth, app=app, api=api, env_file=env_file)
+        event_filter_options = UniqueChatEventFilterOptions(_env_file=env_file_str)  # type: ignore[call-arg]
+        return cls(
+            auth=auth,
+            app=app,
+            api=api,
+            chat_event_filter_options=event_filter_options,
+            env_file=env_file,
+        )
 
     @classmethod
     def from_env_auto(cls, filename: str = "unique.env") -> "UniqueSettings":