unique_toolkit 0.8.33__tar.gz → 0.8.34__tar.gz

{unique_toolkit-0.8.33 → unique_toolkit-0.8.34}/CHANGELOG.md

@@ -6,6 +6,9 @@ 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).
 
 
+## [0.8.34] - 2025-09-01
+- Automatic initializations of services and event generator
+
 ## [0.8.33] - 2025-08-31
 
 fixed tool for `web_search`

{unique_toolkit-0.8.33 → unique_toolkit-0.8.34}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.8.33
+Version: 0.8.34
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -116,6 +116,9 @@ 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).
 
 
+## [0.8.34] - 2025-09-01
+- Automatic initializations of services and event generator
+
 ## [0.8.33] - 2025-08-31
 
 fixed tool for `web_search`

{unique_toolkit-0.8.33 → unique_toolkit-0.8.34}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.8.33"
+version = "0.8.34"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",

{unique_toolkit-0.8.33 → unique_toolkit-0.8.34}/unique_toolkit/app/dev_util.py

@@ -78,6 +78,30 @@ def get_event_generator(
             continue
 
 
+def get_event_stream(
+    event_type: type[T] = BaseEvent,
+    settings_config: UniqueSettings | str | None = None,
+) -> Generator[T, None, None]:
+    """
+    Get an event stream from the SSE client.
+
+    Args:
+        event_type: The type of event to get
+        settings_or_filename: The settings or filename to use to setup the Unique settings object
+    """
+
+    if isinstance(settings_config, str):
+        unique_settings = UniqueSettings.from_env_auto_with_sdk_init(
+            filename=settings_config
+        )
+    elif isinstance(settings_config, UniqueSettings):
+        unique_settings = settings_config
+    else:
+        unique_settings = UniqueSettings.from_env_auto_with_sdk_init()
+
+    return get_event_generator(unique_settings, event_type)
+
+
 def run_demo_with_sse_client(
     unique_settings: UniqueSettings,
     handler: Callable[[BaseEvent], Awaitable[None] | None],

{unique_toolkit-0.8.33 → unique_toolkit-0.8.34}/unique_toolkit/content/service.py

@@ -135,11 +135,19 @@ class ContentService:
 
     @classmethod
     def from_settings(
-        cls, settings: UniqueSettings, metadata_filter: dict | None = None
+        cls,
+        settings: UniqueSettings | str | None = None,
+        metadata_filter: dict | None = None,
     ):
         """
-        Initialize the ContentService with a settings object.
+        Initialize the ContentService with a settings object and metadata filter.
         """
+
+        if settings is None:
+            settings = UniqueSettings.from_env_auto_with_sdk_init()
+        elif isinstance(settings, str):
+            settings = UniqueSettings.from_env_auto_with_sdk_init(filename=settings)
+
         return cls(
             company_id=settings.auth.company_id.get_secret_value(),
             user_id=settings.auth.user_id.get_secret_value(),

{unique_toolkit-0.8.33 → unique_toolkit-0.8.34}/unique_toolkit/embedding/service.py

@@ -56,10 +56,16 @@ class EmbeddingService(BaseService):
         return cls(company_id=event.company_id, user_id=event.user_id)
 
     @classmethod
-    def from_settings(cls, settings: UniqueSettings):
+    def from_settings(cls, settings: UniqueSettings | str | None = None):
         """
         Initialize the EmbeddingService with a settings object.
         """
+
+        if settings is None:
+            settings = UniqueSettings.from_env_auto_with_sdk_init()
+        elif isinstance(settings, str):
+            settings = UniqueSettings.from_env_auto_with_sdk_init(filename=settings)
+
         return cls(
             company_id=settings.auth.company_id.get_secret_value(),
             user_id=settings.auth.user_id.get_secret_value(),

{unique_toolkit-0.8.33 → unique_toolkit-0.8.34}/unique_toolkit/language_model/service.py

@@ -91,10 +91,15 @@ class LanguageModelService:
         return cls(company_id=event.company_id, user_id=event.user_id)
 
     @classmethod
-    def from_settings(cls, settings: UniqueSettings):
+    def from_settings(cls, settings: UniqueSettings | str | None = None):
         """
         Initialize the LanguageModelService with a settings object.
         """
+        if settings is None:
+            settings = UniqueSettings.from_env_auto_with_sdk_init()
+        elif isinstance(settings, str):
+            settings = UniqueSettings.from_env_auto_with_sdk_init(filename=settings)
+
         return cls(
             company_id=settings.auth.company_id.get_secret_value(),
             user_id=settings.auth.user_id.get_secret_value(),