unique_toolkit 0.8.33__tar.gz → 0.8.35__tar.gz

{unique_toolkit-0.8.33 → unique_toolkit-0.8.35}/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).
 
+## [0.8.35] - 2025-09-01
+- Initial toolkit documentation (WIP)
+
+## [0.8.34] - 2025-09-01
+- Automatic initializations of services and event generator
 
 ## [0.8.33] - 2025-08-31
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.8.33
+Version: 0.8.35
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -115,6 +115,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).
 
+## [0.8.35] - 2025-09-01
+- Initial toolkit documentation (WIP)
+
+## [0.8.34] - 2025-09-01
+- Automatic initializations of services and event generator
 
 ## [0.8.33] - 2025-08-31
 

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.8.33"
+version = "0.8.35"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",
@@ -45,6 +45,15 @@ pytest-asyncio = "^0.23.8"
 pytest-mock = "^3.14.0"
 unique-sdk = { path = "../unique_sdk" }
 ipykernel = "^6.29.5"
+mkdocs = "^1.6.1"
+mkdocs-material = "^9.6.11"
+mkdocs-include-markdown-plugin = "^7.1.5"
+mkdocs-callouts = "^1.16.0"
+mkdocstrings-python = "^1.16.10"
+entangled-cli = "^2.1.13"
+mkdocs-entangled-plugin = "^0.4.0"
+isort = "^6.0.1"
+mkdocs-codeinclude-plugin = "^0.2.1"
 
 [build-system]
 requires = ["poetry-core"]
@@ -52,6 +61,8 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.ruff]
 target-version = "py311"
+exclude = [".venv", "docs"]
+
 
 [tool.ruff.lint]
 extend-select = ["I"]

{unique_toolkit-0.8.33 → unique_toolkit-0.8.35}/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.35}/unique_toolkit/app/unique_settings.py

@@ -158,10 +158,19 @@ class EnvFileNotFoundError(FileNotFoundError):
 
 
 class UniqueSettings:
-    def __init__(self, auth: UniqueAuth, app: UniqueApp, api: UniqueApi):
+    def __init__(
+        self,
+        auth: UniqueAuth,
+        app: UniqueApp,
+        api: UniqueApi,
+        env_file: Path | None = None,
+    ):
         self.app = app
         self.auth = auth
         self.api = api
+        self._env_file: Path | None = (
+            env_file if (env_file and env_file.exists()) else None
+        )
 
     @classmethod
     def _find_env_file(cls, filename: str = "unique.env") -> Path:
@@ -226,7 +235,7 @@ 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)
+        return cls(auth=auth, app=app, api=api, env_file=env_file)
 
     @classmethod
     def from_env_auto(cls, filename: str = "unique.env") -> "UniqueSettings":
@@ -243,6 +252,7 @@ class UniqueSettings:
         """
         try:
             env_file = cls._find_env_file(filename)
+            logger.info(f"Environment file found at {env_file}")
             return cls.from_env(env_file=env_file)
         except EnvFileNotFoundError:
             logger.warning(

{unique_toolkit-0.8.33 → unique_toolkit-0.8.35}/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.35}/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.35}/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(),

{unique_toolkit-0.8.33 → unique_toolkit-0.8.35}/unique_toolkit/tools/a2a/service.py

@@ -2,7 +2,7 @@ from pydantic import Field, create_model
 from unique_sdk.utils.chat_in_space import send_message_and_wait_for_completion
 
 from unique_toolkit.app import ChatEvent
-from unique_toolkit.evaluators.schemas import EvaluationMetricName
+from unique_toolkit.evals.schemas import EvaluationMetricName
 from unique_toolkit.language_model import (
     LanguageModelFunction,
     LanguageModelMessage,
@@ -146,7 +146,7 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
         tool_response: ToolCallResponse,
     ) -> LanguageModelMessage:
         return ToolCallResponse(
-            id=tool_response.id,
+            id=tool_response.id,  # type: ignore
             name=tool_response.name,
-            content=tool_response["content"],
+            content=tool_response.content,
         )