unique_toolkit 1.3.0__tar.gz → 1.3.2__tar.gz

{unique_toolkit-1.3.0 → unique_toolkit-1.3.2}/CHANGELOG.md

@@ -5,6 +5,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.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
+
 ## [1.3.0] - 2025-09-28
 - Add utilitiy to enhance pydantic model with metadata
 - Add capability to collect this metadata with same hierarchy as nested pydantic models

{unique_toolkit-1.3.0 → unique_toolkit-1.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.3.0
+Version: 1.3.2
 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.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
+
 ## [1.3.0] - 2025-09-28
 - Add utilitiy to enhance pydantic model with metadata
 - Add capability to collect this metadata with same hierarchy as nested pydantic models

{unique_toolkit-1.3.0 → unique_toolkit-1.3.2}/pyproject.toml

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

{unique_toolkit-1.3.0 → unique_toolkit-1.3.2}/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.0 → unique_toolkit-1.3.2}/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

{unique_toolkit-1.3.0 → unique_toolkit-1.3.2}/unique_toolkit/content/schemas.py

@@ -26,15 +26,41 @@ class ContentMetadata(BaseModel):
 
 class ContentChunk(BaseModel):
     model_config = model_config
-    id: str
-    text: str
-    order: int
-    key: str | None = None
-    chunk_id: str | None = None
-    url: str | None = None
-    title: str | None = None
-    start_page: int | None = None
-    end_page: int | None = None
+    id: str = Field(
+        default="",
+        description="The id of the content this chunk belongs to. The id starts with 'cont_' followed by an alphanumeric string of length 24.",
+        examples=["cont_abcdefgehijklmnopqrstuvwx"],
+    )
+    text: str = Field(default="", description="The text content of the chunk.")
+    order: int = Field(
+        default=0,
+        description="The order of the chunk in the original content. Concatenating the chunks in order will give the original content.",
+    )
+    key: str | None = Field(
+        default=None,
+        description="The key of the chunk. For document chunks this is the the filename",
+    )
+    chunk_id: str | None = Field(
+        default=None,
+        description="The id of the chunk. The id starts with 'chunk_' followed by an alphanumeric string of length 24.",
+        examples=["chunk_abcdefgehijklmnopqrstuv"],
+    )
+    url: str | None = Field(
+        default=None,
+        description="For chunk retrieved from the web this is the url of the chunk.",
+    )
+    title: str | None = Field(
+        default=None,
+        description="The title of the chunk. For document chunks this is the title of the document.",
+    )
+    start_page: int | None = Field(
+        default=None,
+        description="The start page of the chunk. For document chunks this is the start page of the document.",
+    )
+    end_page: int | None = Field(
+        default=None,
+        description="The end page of the chunk. For document chunks this is the end page of the document.",
+    )
 
     object: str | None = None
     metadata: ContentMetadata | None = None
@@ -45,9 +71,19 @@ class ContentChunk(BaseModel):
 
 class Content(BaseModel):
     model_config = model_config
-    id: str
-    key: str
-    title: str | None = None
+    id: str = Field(
+        default="",
+        description="The id of the content. The id starts with 'cont_' followed by an alphanumeric string of length 24.",
+        examples=["cont_abcdefgehijklmnopqrstuvwx"],
+    )
+    key: str = Field(
+        default="",
+        description="The key of the content. For documents this is the the filename",
+    )
+    title: str | None = Field(
+        default=None,
+        description="The title of the content. For documents this is the title of the document.",
+    )
     url: str | None = None
     chunks: list[ContentChunk] = []
     write_url: str | None = None