unique_toolkit 1.3.0__py3-none-any.whl → 1.3.2__py3-none-any.whl

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

{unique_toolkit-1.3.0.dist-info → unique_toolkit-1.3.2.dist-info}/METADATA

@@ -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.dist-info → unique_toolkit-1.3.2.dist-info}/RECORD

@@ -81,13 +81,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=BmrVpZ86_X3NCd9irNPvT5VFL9-7tF7muZeaCL53j4M,5185
+unique_toolkit/app/dev_util.py,sha256=5tkGEcjHZ1u4kY0yUOhTWLjhXLvAcEMutY3h8j1bBJE,5312
 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=UE1UaL_N2o48O-mLb_rF3Sp9MrVkfdDE6f_dm7WTOO4,9137
-unique_toolkit/app/unique_settings.py,sha256=JNU0LrlldVx1GUeSS0Mr1tqT67934KWU2OeVWkNUu28,11510
+unique_toolkit/app/unique_settings.py,sha256=849pbASOu8QbHu9QfCJhYTisjKRet34J-fRs0gNKZ0Q,12369
 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
@@ -99,7 +99,7 @@ unique_toolkit/chat/utils.py,sha256=ihm-wQykBWhB4liR3LnwPVPt_qGW6ETq21Mw4HY0THE,
 unique_toolkit/content/__init__.py,sha256=EdJg_A_7loEtCQf4cah3QARQreJx6pdz89Rm96YbMVg,940
 unique_toolkit/content/constants.py,sha256=1iy4Y67xobl5VTnJB6SxSyuoBWbdLl9244xfVMUZi5o,60
 unique_toolkit/content/functions.py,sha256=1zhxaJEYTvvd4qzkrbEFcrjdJxhHkfUY3dEpNfNC_hk,19052
-unique_toolkit/content/schemas.py,sha256=KJ604BOx0vBh2AwlTCZkOo55aHsI6yj8vxDAARKKqEo,2995
+unique_toolkit/content/schemas.py,sha256=WB3InkKIvfWbyg9CsKFLn8Zf4zrE-YqGpCc3a0zOk7k,4774
 unique_toolkit/content/service.py,sha256=ZUXJwfNdHsAw_F7cfRMDVgHpSKxiwG6Cn8p7c4hV8TM,24053
 unique_toolkit/content/utils.py,sha256=qNVmHTuETaPNGqheg7TbgPr1_1jbNHDc09N5RrmUIyo,7901
 unique_toolkit/embedding/__init__.py,sha256=uUyzjonPvuDCYsvXCIt7ErQXopLggpzX-MEQd3_e2kE,250
@@ -133,7 +133,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-1.3.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.3.0.dist-info/METADATA,sha256=63zIDL6zXri4f4WWuUFB4jb0m5XGi6oL4jdSUlCfTf4,33548
-unique_toolkit-1.3.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.3.0.dist-info/RECORD,,
+unique_toolkit-1.3.2.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.3.2.dist-info/METADATA,sha256=64vZcZcbk6v47satQNAuj1iuDrkR15OyO2GjqoOOX6s,33695
+unique_toolkit-1.3.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.3.2.dist-info/RECORD,,