unique_toolkit 0.9.1__py3-none-any.whl → 1.1.0__py3-none-any.whl

unique_toolkit/agentic/__init__.py

@@ -0,0 +1 @@
+

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/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/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":

{unique_toolkit-0.9.1.dist-info → unique_toolkit-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.9.1
+Version: 1.1.0
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,6 +118,12 @@ 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
+
 ## [0.9.1] - 2025-09-17
 - update to python 3.12 due to security 
 

{unique_toolkit-0.9.1.dist-info → unique_toolkit-1.1.0.dist-info}/RECORD

@@ -19,6 +19,7 @@ unique_toolkit/_common/utils/structured_output/schema.py,sha256=Tp7kDYcmKtnUhcuR
 unique_toolkit/_common/utils/write_configuration.py,sha256=fzvr4C-XBL3OSM3Od9TbqIxeeDS9_d9CLEyTq6DDknY,1409
 unique_toolkit/_common/validate_required_values.py,sha256=Y_M1ub9gIKP9qZ45F6Zq3ZHtuIqhmOjl8Z2Vd3avg8w,588
 unique_toolkit/_common/validators.py,sha256=aZwbMho7XszN7lT5RtemaiXgC0WJ4u40oeVgsNGhF4U,2803
+unique_toolkit/agentic/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 unique_toolkit/agentic/debug_info_manager/debug_info_manager.py,sha256=8u3_oxcln7y2zOsfiGh5YOm1zYAlV5QxZ5YAsbEJG0c,584
 unique_toolkit/agentic/evaluation/config.py,sha256=ywHIrJs5SFdKr1WXfrofWuFfzb0iPQw8iZDpq5oEug4,953
 unique_toolkit/agentic/evaluation/context_relevancy/prompts.py,sha256=EdHFUOB581yVxcOL8482KUv_LzaRjuiem71EF8udYMc,1331
@@ -71,13 +72,13 @@ unique_toolkit/agentic/tools/utils/source_handling/schema.py,sha256=l4B6kA6grbL-
 unique_toolkit/agentic/tools/utils/source_handling/source_formatting.py,sha256=uZ0QXqrPWgId3ZA67dvjHQ6xrW491LK1xxx_sVJmFHg,9160
 unique_toolkit/agentic/tools/utils/source_handling/tests/test_source_formatting.py,sha256=EA8iVvb3L91OFk2XMbGcFuhe2etqm3Sx9QCYDGiOSOM,6995
 unique_toolkit/app/__init__.py,sha256=ETxYDpEizg_PKmi4JPX_P76ySq-us-xypfAIdKQ1QZU,1284
-unique_toolkit/app/dev_util.py,sha256=1C5aGUQnyO8Kot9i35PPFl4B32Oj9eK22WKTP_BrfwY,5063
+unique_toolkit/app/dev_util.py,sha256=BmrVpZ86_X3NCd9irNPvT5VFL9-7tF7muZeaCL53j4M,5185
 unique_toolkit/app/init_logging.py,sha256=Sh26SRxOj8i8dzobKhYha2lLrkrMTHfB1V4jR3h23gQ,678
 unique_toolkit/app/init_sdk.py,sha256=5_oDoETr6akwYyBCb0ivTdMNu3SVgPSkrXcDS6ELyY8,2269
 unique_toolkit/app/performance/async_tasks.py,sha256=H0l3OAcosLwNHZ8d2pd-Di4wHIXfclEvagi5kfqLFPA,1941
 unique_toolkit/app/performance/async_wrapper.py,sha256=yVVcRDkcdyfjsxro-N29SBvi-7773wnfDplef6-y8xw,1077
-unique_toolkit/app/schemas.py,sha256=xHdzMyZ_cgCzxCqzCJYwOCAYWkaQ9zIyZsRhDgSQnEA,8088
-unique_toolkit/app/unique_settings.py,sha256=vtbtfpxGrK1ZsyVUGiGulG35d2euSSFxE3qj0eiXKVA,10381
+unique_toolkit/app/schemas.py,sha256=UE1UaL_N2o48O-mLb_rF3Sp9MrVkfdDE6f_dm7WTOO4,9137
+unique_toolkit/app/unique_settings.py,sha256=JNU0LrlldVx1GUeSS0Mr1tqT67934KWU2OeVWkNUu28,11510
 unique_toolkit/app/verification.py,sha256=GxFFwcJMy25fCA_Xe89wKW7bgqOu8PAs5y8QpHF0GSc,3861
 unique_toolkit/chat/__init__.py,sha256=LRs2G-JTVuci4lbtHTkVUiNcZcSR6uqqfnAyo7af6nY,619
 unique_toolkit/chat/constants.py,sha256=05kq6zjqUVB2d6_P7s-90nbljpB3ryxwCI-CAz0r2O4,83
@@ -123,7 +124,7 @@ unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJ
 unique_toolkit/short_term_memory/service.py,sha256=5PeVBu1ZCAfyDb2HLVvlmqSbyzBBuE9sI2o9Aajqjxg,8884
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=cxWjb2dxEI2HGsakKdVCkSNi7VK9mr08w5sDcFCQyWI,9553
-unique_toolkit-0.9.1.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.9.1.dist-info/METADATA,sha256=preWvo_48KQIFiYTSJtgXfHx4ZrJMe-NipmzyL6s2nQ,32215
-unique_toolkit-0.9.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.9.1.dist-info/RECORD,,
+unique_toolkit-1.1.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.1.0.dist-info/METADATA,sha256=upTGYXRy0kIWh32G1W37OCkFFW68zXXWJoI44OqnCM8,32401
+unique_toolkit-1.1.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.1.0.dist-info/RECORD,,