unique_toolkit 0.8.1__py3-none-any.whl → 0.8.2__py3-none-any.whl

unique_toolkit/app/dev_util.py

@@ -0,0 +1,146 @@
+import asyncio
+import json
+from logging import getLogger
+from pathlib import Path
+from typing import (
+    Awaitable,
+    Callable,
+    Generator,
+    TypeVar,
+)
+
+from sseclient import SSEClient
+
+from unique_toolkit.app import BaseEvent, ChatEvent, EventName
+from unique_toolkit.app.init_sdk import init_unique_sdk
+from unique_toolkit.app.unique_settings import UniqueSettings
+
+T = TypeVar("T", bound=BaseEvent)
+
+LOGGER = getLogger(__name__)
+
+
+def get_event_name_from_event_class(event_class: type[T]) -> EventName | None:
+    if event_class is ChatEvent:
+        return EventName.EXTERNAL_MODULE_CHOSEN
+
+    return None
+
+
+def get_sse_client(
+    unique_settings: UniqueSettings,
+    subscriptions: list[str],
+) -> SSEClient:
+    headers = {
+        "Authorization": f"Bearer {unique_settings.app.key.get_secret_value()}",
+        "x-app-id": unique_settings.app.id.get_secret_value(),
+        "x-company-id": unique_settings.auth.company_id.get_secret_value(),
+        "x-user-id": unique_settings.auth.user_id.get_secret_value(),
+        "x-api-version": unique_settings.api.version,
+    }
+    return SSEClient(url=unique_settings.api.sse_url(subscriptions), headers=headers)
+
+
+def get_event_generator(
+    unique_settings: UniqueSettings, event_type: type[T]
+) -> Generator[T, None, None]:
+    """
+    Generator that yields only events of the specified type from an SSE stream.
+
+    Args:
+        sse_client: The SSE client to read events from
+        event_type: The event class type to filter for
+
+    Yields:
+        Events matching the specified type
+    """
+    event_name = get_event_name_from_event_class(event_type)
+    if (
+        event_name is None
+        or not issubclass(event_type, BaseEvent)
+        or event_type is BaseEvent
+    ):
+        raise ValueError(f"Event model {event_type} is not a valid event model")
+
+    subscription = event_name.value
+
+    for sse_event in get_sse_client(unique_settings, [subscription]):
+        try:
+            payload = json.loads(sse_event.data)
+            parsed_event = event_type.model_validate(payload)
+            if parsed_event is None:
+                continue
+
+            yield parsed_event
+
+        except Exception as e:
+            LOGGER.error(f"Could not parse SSE event data as JSON: {e}")
+            continue
+
+
+def run_demo_with_sse_client(
+    unique_settings: UniqueSettings,
+    handler: Callable[[BaseEvent], Awaitable[None] | None],
+    event_type: type[BaseEvent],
+) -> None:
+    """
+    Run a demo with an SSE client using sync handler.
+
+    Args:
+        unique_settings: The unique settings to use for the SSE client
+        handler: The sync handler to use for the SSE client
+        event_type: The type of event to use for the SSE client
+    """
+
+    event_name = get_event_name_from_event_class(event_type)
+    if event_name is None:
+        return
+
+    init_unique_sdk(unique_settings=unique_settings)
+    is_async_handler = asyncio.iscoroutinefunction(handler)
+
+    for event in get_event_generator(unique_settings, event_type):
+        if is_async_handler:
+            loop = asyncio.get_event_loop()
+            loop.run_until_complete(handler(event))
+        else:
+            handler(event)
+
+
+def load_event(file_path: Path, event_type: type[BaseEvent]) -> BaseEvent:
+    with file_path.open("r") as file:
+        event = json.load(file)
+
+    return event_type.model_validate(event)
+
+
+def run_demo_with_with_saved_event(
+    unique_settings: UniqueSettings,
+    handler: Callable[[BaseEvent], Awaitable[None] | None],
+    event_type: type[BaseEvent],
+    file_path: Path,
+) -> None:
+    """
+    Run a demo with an SSE client.
+
+    Note: event_type is the type of event that the handler expects.
+
+    Args:
+        unique_settings: The unique settings to use for the SSE client
+        handler: The handler to use for the SSE client
+        event_type: The type of event to use for the SSE client
+    """
+    init_unique_sdk(unique_settings=unique_settings)
+
+    event_name = get_event_name_from_event_class(event_type)
+    if event_name is None:
+        return
+
+    event = load_event(file_path, event_type)
+    if event is None:
+        raise ValueError(f"Event not found in {file_path}")
+
+    if asyncio.iscoroutinefunction(handler):
+        asyncio.run(handler(event))
+    else:
+        handler(event)

unique_toolkit/app/init_sdk.py

@@ -1,6 +1,11 @@
 import os
+from pathlib import Path
+from typing import overload
 
 import unique_sdk
+from typing_extensions import deprecated
+
+from unique_toolkit.app.unique_settings import UniqueSettings
 
 
 def get_env(var_name, default=None, strict=False):
@@ -24,12 +29,38 @@ def get_env(var_name, default=None, strict=False):
     return val or default
 
 
-def init_sdk(strict_all_vars=False):
+@overload
+def init_unique_sdk(*, env_file: Path | None = None): ...
+
+
+@overload
+def init_unique_sdk(*, unique_settings: UniqueSettings): ...
+
+
+def init_unique_sdk(
+    *, unique_settings: UniqueSettings | None = None, env_file: Path | None = None
+):
+    if unique_settings:
+        unique_sdk.api_key = unique_settings.app.key.get_secret_value()
+        unique_sdk.app_id = unique_settings.app.id.get_secret_value()
+        unique_sdk.api_base = unique_settings.api.sdk_url()
+    elif env_file:
+        unique_settings = UniqueSettings.from_env(env_file=env_file)
+        unique_sdk.api_key = unique_settings.app.key.get_secret_value()
+        unique_sdk.app_id = unique_settings.app.id.get_secret_value()
+        unique_sdk.api_base = unique_settings.api.sdk_url()
+
+
+@deprecated("Use init_unique_sdk instead")
+def init_sdk(
+    strict_all_vars: bool = False,
+):
     """Initialize the SDK.
 
     Args:
         strict_all_vars (bool, optional): This method raises a ValueError if strict and no value is found in the environment. Defaults to False.
     """
+
     unique_sdk.api_key = get_env("API_KEY", default="dummy", strict=strict_all_vars)
     unique_sdk.app_id = get_env("APP_ID", default="dummy", strict=strict_all_vars)
     unique_sdk.api_base = get_env("API_BASE", default=None, strict=strict_all_vars)

unique_toolkit/app/schemas.py

@@ -19,6 +19,15 @@ model_config = ConfigDict(
 
 class EventName(StrEnum):
     EXTERNAL_MODULE_CHOSEN = "unique.chat.external-module.chosen"
+    USER_MESSAGE_CREATED = "unique.chat.user-message.created"
+    INGESTION_CONTENT_UPLOADED = "unique.ingestion.content.uploaded"
+    INGESTION_CONTENT_FINISHED = "unique.ingestion.content.finished"
+    MAGIC_TABLE_IMPORT_COLUMNS = "unique.magic-table.import-columns"
+    MAGIC_TABLE_ADD_META_DATA = "unique.magic-table.add-meta-data"
+    MAGIC_TABLE_ADD_DOCUMENT = "unique.magic-table.add-document"
+    MAGIC_TABLE_DELETE_ROW = "unique.magic-table.delete-row"
+    MAGIC_TABLE_DELETE_COLUMN = "unique.magic-table.delete-column"
+    MAGIC_TABLE_UPDATE_CELL = "unique.magic-table.update-cell"
 
 
 class BaseEvent(BaseModel):

unique_toolkit/app/unique_settings.py

@@ -1,15 +1,40 @@
+from logging import getLogger
 from pathlib import Path
+from typing import Self, TypeVar
+from urllib.parse import urlparse, urlunparse
 
-from pydantic import SecretStr
+from pydantic import Field, SecretStr, model_validator
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
+logger = getLogger(__name__)
+
+T = TypeVar("T", bound=BaseSettings)
+
+
+def warn_about_defaults(instance: T) -> T:
+    """Log warnings for fields that are using default values."""
+    for field_name, model_field in instance.model_fields.items():
+        field_value = getattr(instance, field_name)
+        if field_value == model_field.default:
+            logger.warning(
+                f"Using default value for '{field_name}': {model_field.default}"
+            )
+    return instance
+
 
 class UniqueApp(BaseSettings):
-    id: SecretStr
-    key: SecretStr
-    base_url: str
-    endpoint: str
-    endpoint_secret: SecretStr
+    id: SecretStr = Field(default=SecretStr("dummy_id"))
+    key: SecretStr = Field(default=SecretStr("dummy_key"))
+    base_url: str = Field(
+        default="http://localhost:8092/",
+        deprecated="Use UniqueApi.base_url instead",
+    )
+    endpoint: str = Field(default="dummy")
+    endpoint_secret: SecretStr = Field(default=SecretStr("dummy_secret"))
+
+    @model_validator(mode="after")
+    def _warn_about_defaults(self) -> Self:
+        return warn_about_defaults(self)
 
     model_config = SettingsConfigDict(
         env_prefix="unique_app_",
@@ -19,9 +44,52 @@ class UniqueApp(BaseSettings):
     )
 
 
+class UniqueApi(BaseSettings):
+    base_url: str = Field(
+        default="http://localhost:8092/",
+        description="The base URL of the Unique API. Ask your admin to provide you with the correct URL.",
+    )
+    version: str = Field(default="2023-12-06")
+
+    model_config = SettingsConfigDict(
+        env_prefix="unique_api_",
+        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)
+
+    def sse_url(self, subscriptions: list[str]) -> str:
+        parsed = urlparse(self.base_url)
+        return urlunparse(
+            parsed._replace(
+                path="/public/event-socket/events/stream",
+                query=f"subscriptions={','.join(subscriptions)}",
+                fragment=None,
+            )
+        )
+
+    def sdk_url(self) -> str:
+        parsed = urlparse(self.base_url)
+
+        path = "/public/chat"
+        if parsed.hostname and "qa.unique" in parsed.hostname:
+            path = "/public/chat-gen2"
+        return urlunparse(parsed._replace(path=path, query=None, fragment=None))
+
+    def openai_proxy_url(self) -> str:
+        parsed = urlparse(self.base_url)
+        return urlunparse(
+            parsed._replace(path="/public/openai-proxy/", query=None, fragment=None)
+        )
+
+
 class UniqueAuth(BaseSettings):
-    company_id: SecretStr
-    user_id: SecretStr
+    company_id: SecretStr = Field(default=SecretStr("dummy_company_id"))
+    user_id: SecretStr = Field(default=SecretStr("dummy_user_id"))
 
     model_config = SettingsConfigDict(
         env_prefix="unique_auth_",
@@ -30,11 +98,16 @@ class UniqueAuth(BaseSettings):
         extra="ignore",
     )
 
+    @model_validator(mode="after")
+    def _warn_about_defaults(self) -> Self:
+        return warn_about_defaults(self)
+
 
 class UniqueSettings:
-    def __init__(self, auth: UniqueAuth, app: UniqueApp):
+    def __init__(self, auth: UniqueAuth, app: UniqueApp, api: UniqueApi):
         self.app = app
         self.auth = auth
+        self.api = api
 
     @classmethod
     def from_env(cls, env_file: Path | None = None) -> "UniqueSettings":
@@ -55,7 +128,7 @@ class UniqueSettings:
 
         # Initialize settings with environment file if provided
         env_file_str = str(env_file) if env_file else None
-        auth = UniqueAuth(_env_file=env_file_str, _env_file_encoding="utf-8")
-        app = UniqueApp(_env_file=env_file_str, _env_file_encoding="utf-8")
-
-        return cls(auth=auth, app=app)
+        auth = UniqueAuth(_env_file=env_file_str)
+        app = UniqueApp(_env_file=env_file_str)
+        api = UniqueApi(_env_file=env_file_str)
+        return cls(auth=auth, app=app, api=api)

unique_toolkit/chat/service.py

@@ -57,15 +57,6 @@ logger = logging.getLogger(f"toolkit.{DOMAIN_NAME}.{__name__}")
 class ChatService:
     """
     Provides all functionalities to manage the chat session.
-
-    Attributes:
-        company_id (str | None): The company ID.
-        user_id (str | None): The user ID.
-        assistant_message_id (str | None): The assistant message ID.
-        user_message_id (str | None): The user message ID.
-        chat_id (str | None): The chat ID.
-        assistant_id (str | None): The assistant ID.
-        user_message_text (str | None): The user message text.
     """
 
     def __init__(self, event: ChatEvent | Event):
@@ -88,6 +79,7 @@ class ChatService:
 
         Returns:
             Event | BaseEvent | None: The event object.
+
         """
         return self._event
 
@@ -101,6 +93,7 @@ class ChatService:
 
         Returns:
             str | None: The company identifier.
+
         """
         return self._company_id
 

unique_toolkit/content/service.py

@@ -1,6 +1,6 @@
 import logging
 from pathlib import Path
-from typing import Any
+from typing import Any, overload
 
 import unique_sdk
 from requests import Response
@@ -35,13 +35,30 @@ logger = logging.getLogger(f"toolkit.{DOMAIN_NAME}.{__name__}")
 class ContentService:
     """
     Provides methods for searching, downloading and uploading content in the knowledge base.
+    """
+
+    @deprecated(
+        "Use __init__ with company_id, user_id and chat_id instead or use the classmethod `from_event`"
+    )
+    @overload
+    def __init__(self, event: Event | ChatEvent | BaseEvent): ...
+
+    """
+        Initialize the ContentService with an event (deprecated)
+    """
+
+    @overload
+    def __init__(
+        self,
+        *,
+        company_id: str,
+        user_id: str,
+        chat_id: str | None,
+        metadata_filter: dict | None = None,
+    ): ...
 
-    Attributes:
-        event: BaseEvent | Event, this can be None ONLY if company_id and user_id are provided.
-        company_id (str): The company ID.
-        user_id (str): The user ID.
-        chat_id (str): The chat ID. Defaults to None
-        metadata_filter (dict | None): is only initialised from an Event(Deprecated) or ChatEvent.
+    """
+        Initialize the ContentService with a company_id, user_id and chat_id and metadata_filter.
     """
 
     def __init__(
@@ -50,7 +67,12 @@ class ContentService:
         company_id: str | None = None,
         user_id: str | None = None,
         chat_id: str | None = None,
+        metadata_filter: dict | None = None,
     ):
+        """
+        Initialize the ContentService with a company_id, user_id and chat_id.
+        """
+
         self._event = event  # Changed to protected attribute
         self._metadata_filter = None
         if event:
@@ -64,6 +86,26 @@ class ContentService:
             self._company_id: str = company_id
             self._user_id: str = user_id
             self._chat_id: str | None = chat_id
+            self._metadata_filter = metadata_filter
+
+    @classmethod
+    def from_event(cls, event: Event | ChatEvent | BaseEvent):
+        """
+        Initialize the ContentService with an event.
+        """
+        chat_id = None
+        metadata_filter = None
+
+        if isinstance(event, (ChatEvent | Event)):
+            chat_id = event.payload.chat_id
+            metadata_filter = event.payload.metadata_filter
+
+        return cls(
+            company_id=event.company_id,
+            user_id=event.user_id,
+            chat_id=chat_id,
+            metadata_filter=metadata_filter,
+        )
 
     @property
     @deprecated(

unique_toolkit/embedding/service.py

@@ -1,3 +1,5 @@
+from typing import overload
+
 from typing_extensions import deprecated
 
 from unique_toolkit._common._base_service import BaseService
@@ -11,10 +13,23 @@ from unique_toolkit.embedding.schemas import Embeddings
 class EmbeddingService(BaseService):
     """
     Provides methods to interact with the Embedding service.
+    """
 
-    Attributes:
-        company_id (str | None): The company ID.
-        user_id (str | None): The user ID.
+    @deprecated(
+        "Use __init__ with company_id and user_id instead or use the classmethod `from_event`"
+    )
+    @overload
+    def __init__(self, event: Event | BaseEvent): ...
+
+    """
+        Initialize the EmbeddingService with an event (deprecated)
+    """
+
+    @overload
+    def __init__(self, *, company_id: str, user_id: str): ...
+
+    """
+        Initialize the EmbeddingService with a company_id and user_id.
     """
 
     def __init__(
@@ -32,6 +47,13 @@ class EmbeddingService(BaseService):
             self._company_id: str = company_id
             self._user_id: str = user_id
 
+    @classmethod
+    def from_event(cls, event: Event | BaseEvent):
+        """
+        Initialize the EmbeddingService with an event.
+        """
+        return cls(company_id=event.company_id, user_id=event.user_id)
+
     @property
     @deprecated(
         "The event property is deprecated and will be removed in a future version."

unique_toolkit/framework_utilities/langchain/client.py

@@ -1,6 +1,5 @@
 import importlib.util
 import logging
-from pathlib import Path
 
 from unique_toolkit.app.unique_settings import UniqueSettings
 from unique_toolkit.framework_utilities.utils import get_default_headers
@@ -23,11 +22,13 @@ else:
     raise LangchainNotInstalledError()
 
 
-def get_client(env_file: Path | None = None) -> ChatOpenAI:
+def get_client(
+    unique_settings: UniqueSettings, model: str = "AZURE_GPT_4o_2024_0806"
+) -> ChatOpenAI:
     """Get a Langchain ChatOpenAI client instance.
 
     Args:
-        env_file: Optional path to environment file
+        unique_settings: UniqueSettings instance
 
     Returns:
         ChatOpenAI client instance
@@ -35,11 +36,10 @@ def get_client(env_file: Path | None = None) -> ChatOpenAI:
     Raises:
         LangchainNotInstalledError: If langchain-openai package is not installed
     """
-    settings = UniqueSettings.from_env(env_file=env_file)
 
     return ChatOpenAI(
-        base_url=settings.app.base_url + "/openai-proxy/",
-        default_headers=get_default_headers(settings.app, settings.auth),
-        model="AZURE_GPT_4o_2024_0806",
-        api_key=settings.app.key,
+        base_url=unique_settings.api.openai_proxy_url(),
+        default_headers=get_default_headers(unique_settings.app, unique_settings.auth),
+        model=model,
+        api_key=unique_settings.app.key,
     )

unique_toolkit/framework_utilities/openai/client.py

@@ -1,6 +1,5 @@
 import importlib.util
 import logging
-from pathlib import Path
 
 from unique_toolkit.app.unique_settings import UniqueSettings
 from unique_toolkit.framework_utilities.utils import get_default_headers
@@ -23,7 +22,7 @@ else:
     raise OpenAINotInstalledError()
 
 
-def get_openai_client(env_file: Path | None = None) -> OpenAI:
+def get_openai_client(unique_settings: UniqueSettings) -> OpenAI:
     """Get an OpenAI client instance.
 
     Args:
@@ -35,11 +34,10 @@ def get_openai_client(env_file: Path | None = None) -> OpenAI:
     Raises:
         OpenAINotInstalledError: If OpenAI package is not installed
     """
-    settings = UniqueSettings.from_env(env_file=env_file)
-    default_headers = get_default_headers(settings.app, settings.auth)
+    default_headers = get_default_headers(unique_settings.app, unique_settings.auth)
 
     return OpenAI(
-        api_key=settings.app.key.get_secret_value(),
-        base_url=settings.app.base_url + "/openai-proxy/",
+        api_key=unique_settings.app.key.get_secret_value(),
+        base_url=unique_settings.api.openai_proxy_url(),
         default_headers=default_headers,
     )

unique_toolkit/language_model/service.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Any, Optional, Type
+from typing import Any, Optional, Type, overload
 
 from pydantic import BaseModel
 from typing_extensions import deprecated
@@ -33,36 +33,61 @@ logger = logging.getLogger(f"toolkit.{DOMAIN_NAME}.{__name__}")
 class LanguageModelService:
     """
     Provides methods to interact with the Language Model by generating responses.
+    """
+
+    @deprecated(
+        "Use __init__ with company_id and user_id instead or use the classmethod `from_event`"
+    )
+    @overload
+    def __init__(self, event: Event | ChatEvent | BaseEvent): ...
+
+    """
+        Initialize the LanguageModelService with an event (deprecated)
+    """
+
+    @overload
+    def __init__(self, *, company_id: str, user_id: str): ...
 
-    Args:
-        company_id (str | None, optional): The company identifier. Defaults to None.
-        user_id (str | None, optional): The user identifier. Defaults to None.
-        chat_id (str | None, optional): The chat identifier. Defaults to None.
-        assistant_id (str | None, optional): The assistant identifier. Defaults to None.
+    """
+        Initialize the LanguageModelService with a company_id and user_id.
     """
 
     def __init__(
         self,
-        event: Event | BaseEvent | None = None,
+        event: Event | ChatEvent | BaseEvent | None = None,
         company_id: str | None = None,
         user_id: str | None = None,
-        chat_id: str | None = None,
-        assistant_id: str | None = None,
+        **kwargs: dict[str, Any],  # only here for backward compatibility
     ):
-        self._event = event
-        self._chat_id: str | None = chat_id
-        self._assistant_id: str | None = assistant_id
-
-        if event:
+        if isinstance(event, (ChatEvent, Event)):
+            self._event = event
+            self._chat_id: str | None = event.payload.chat_id
+            self._assistant_id: str | None = event.payload.assistant_id
             self._company_id = event.company_id
             self._user_id = event.user_id
             if isinstance(event, (ChatEvent, Event)):
                 self._chat_id = event.payload.chat_id
                 self._assistant_id = event.payload.assistant_id
+        elif isinstance(event, BaseEvent):
+            self._event = event
+            self._company_id = event.company_id
+            self._user_id = event.user_id
+            self._chat_id: str | None = None
+            self._assistant_id: str | None = None
         else:
             [company_id, user_id] = validate_required_values([company_id, user_id])
+            self._event = None
             self._company_id: str = company_id
             self._user_id: str = user_id
+            self._chat_id: str | None = None
+            self._assistant_id: str | None = None
+
+    @classmethod
+    def from_event(cls, event: BaseEvent):
+        """
+        Initialize the LanguageModelService with an event.
+        """
+        return cls(company_id=event.company_id, user_id=event.user_id)
 
     @property
     @deprecated(

unique_toolkit/short_term_memory/service.py

@@ -1,3 +1,5 @@
+from typing import overload
+
 from typing_extensions import deprecated
 
 from unique_toolkit._common.validate_required_values import validate_required_values
@@ -16,17 +18,35 @@ from .schemas import ShortTermMemory
 class ShortTermMemoryService:
     """
     Provides methods to manage short term memory.
+    """
+
+    @deprecated(
+        "Use __init__ with company_id and user_id instead or use the classmethod `from_event`"
+    )
+    @overload
+    def __init__(self, event: Event | ChatEvent | BaseEvent): ...
+
+    """
+        Initialize the ShortTermMemoryService with an event (deprecated)
+    """
+
+    @overload
+    def __init__(
+        self,
+        *,
+        company_id: str,
+        user_id: str,
+        chat_id: str | None,
+        message_id: str | None,
+    ): ...
 
-    Attributes:
-        user_id (str | None): The user ID.
-        company_id (str | None): The company ID.
-        chat_id (str | None): The chat ID.
-        message_id (str | None): The message ID.
+    """
+        Initialize the ShortTermMemoryService with a company_id, user_id, chat_id and message_id.
     """
 
     def __init__(
         self,
-        event: Event | BaseEvent | None = None,
+        event: Event | ChatEvent | BaseEvent | None = None,
         user_id: str | None = None,
         company_id: str | None = None,
         chat_id: str | None = None,
@@ -50,6 +70,18 @@ class ShortTermMemoryService:
             self._chat_id: str | None = chat_id
             self._message_id: str | None = message_id
 
+    @classmethod
+    def from_event(cls, event: ChatEvent):
+        """
+        Initialize the ShortTermMemoryService with a chat event.
+        """
+        return cls(
+            company_id=event.company_id,
+            user_id=event.user_id,
+            chat_id=event.payload.chat_id,
+            message_id=event.payload.user_message.id,
+        )
+
     @property
     @deprecated(
         "The event property is deprecated and will be removed in a future version."

{unique_toolkit-0.8.1.dist-info → unique_toolkit-0.8.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.8.1
+Version: 0.8.2
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -113,6 +113,14 @@ 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.2] - 2025-08-05
+- Implement overloads for services for clearer dev experience
+- Proper typing for SSE event handling
+- Enhanced unique settings. Expose usage of default values in logs
+- SDK Initialization from unique settings
+- Add utilities for to run llm/agent flows for devs
+
 ## [0.8.1] - 2025-08-05
 - Bump SDK version to support the latest features.
 

{unique_toolkit-0.8.1.dist-info → unique_toolkit-0.8.2.dist-info}/RECORD

@@ -5,33 +5,32 @@ unique_toolkit/_common/exception.py,sha256=caQIE1btsQnpKCHqL2cgWUSbHup06enQu_Pt7
 unique_toolkit/_common/validate_required_values.py,sha256=Y_M1ub9gIKP9qZ45F6Zq3ZHtuIqhmOjl8Z2Vd3avg8w,588
 unique_toolkit/_common/validators.py,sha256=l7-hWyRTZ3aF_e73oTQFZdz93s06VhNWVpkERbg2a64,1569
 unique_toolkit/app/__init__.py,sha256=Vq3RXzrWGZNC0EU9SoheKbUvr8g7Wnc5DLBVazaZBNM,1284
-unique_toolkit/app/event_util.py,sha256=lfI64OcOXJZhh2GcVTmvE8AVc888jPWhi7l1vEzDsgs,535
+unique_toolkit/app/dev_util.py,sha256=rN-xSg4OGfmwjaToy8m_hQroehcLYyk9-GCmQJ-f6uY,4302
 unique_toolkit/app/init_logging.py,sha256=Sh26SRxOj8i8dzobKhYha2lLrkrMTHfB1V4jR3h23gQ,678
-unique_toolkit/app/init_sdk.py,sha256=Nv4Now4pMfM0AgRhbtatLpm_39rKxn0WmRLwmPhRl-8,1285
+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=fQJWKZVTzIFXDpwSqY0joPX2WFKBuK1HchuMp6VeGLQ,6837
-unique_toolkit/app/sse_client.py,sha256=jtOhB2g_oE-vJBqtVuWnWyYUJYv3oOhN2U8j8wuHJ5Y,668
-unique_toolkit/app/unique_settings.py,sha256=fi3V9Dru1G1YK7Pxju_KBGiNJHkHdx7JzpDglcvMZro,1820
+unique_toolkit/app/schemas.py,sha256=JdC2rNVPRrr6QhGMZweE0ID760onbRY2oq9m1LVFego,7429
+unique_toolkit/app/unique_settings.py,sha256=YwKXKKSrLSoNk5kow5ChC0uOqii37gd3ttFL4rSAVF4,4443
 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
 unique_toolkit/chat/functions.py,sha256=QsJVhBXgK6jDWRYpEAt-22jy5NKGsYs4fmHrOEdHAyc,29865
 unique_toolkit/chat/schemas.py,sha256=abPPeDtUIeEyKDnLhIqgIyqTsFANxh3j44EYrITBlHw,2786
-unique_toolkit/chat/service.py,sha256=dRydaABJspU1KlFUQ_Q-rXl9lDkhgeh99SxAguY4s_M,37432
+unique_toolkit/chat/service.py,sha256=CKJfCrW0mTDP0SePlJtEoDUQ8-WB5xI8GslQx1_lnBg,37038
 unique_toolkit/chat/state.py,sha256=Cjgwv_2vhDFbV69xxsn7SefhaoIAEqLx3ferdVFCnOg,1445
 unique_toolkit/chat/utils.py,sha256=ihm-wQykBWhB4liR3LnwPVPt_qGW6ETq21Mw4HY0THE,854
 unique_toolkit/content/__init__.py,sha256=EdJg_A_7loEtCQf4cah3QARQreJx6pdz89Rm96YbMVg,940
 unique_toolkit/content/constants.py,sha256=1iy4Y67xobl5VTnJB6SxSyuoBWbdLl9244xfVMUZi5o,60
 unique_toolkit/content/functions.py,sha256=0ELepm3_sl0SD_SYzvQVQ-jTdrcUqK5mVJZv0nQBuAw,18367
 unique_toolkit/content/schemas.py,sha256=KJ604BOx0vBh2AwlTCZkOo55aHsI6yj8vxDAARKKqEo,2995
-unique_toolkit/content/service.py,sha256=jGRTIt0JQKL6qRyZ9-_njWJejKKvIvDOzEQrf8CdMaU,19491
+unique_toolkit/content/service.py,sha256=9Re_UK7pkv-zNROlKRmfoomfhUQcuzFQvQ05aUl8q6Y,20502
 unique_toolkit/content/utils.py,sha256=qNVmHTuETaPNGqheg7TbgPr1_1jbNHDc09N5RrmUIyo,7901
 unique_toolkit/embedding/__init__.py,sha256=uUyzjonPvuDCYsvXCIt7ErQXopLggpzX-MEQd3_e2kE,250
 unique_toolkit/embedding/constants.py,sha256=Lj8-Lcy1FvuC31PM9Exq7vaFuxQV4pEI1huUMFX-J2M,52
 unique_toolkit/embedding/functions.py,sha256=3qp-BfuMAbnp8YB04rh3xH8vsJuCBPizoy-JeaBFtoQ,1944
 unique_toolkit/embedding/schemas.py,sha256=1GvKCaSk4jixzVQ2PKq8yDqwGEVY_hWclYtoAr6CC2g,96
-unique_toolkit/embedding/service.py,sha256=xdR28rqxCicXN9qGPm0oaw88NAbRXJNIBwTZSARLn2g,4082
+unique_toolkit/embedding/service.py,sha256=z1MwwaeEloLvgT8rwUAydlt28TnKZ9TkcPgdYIyqZvs,4657
 unique_toolkit/embedding/utils.py,sha256=v86lo__bCJbxZBQ3OcLu5SuwT6NbFfWlcq8iyk6BuzQ,279
 unique_toolkit/evaluators/__init__.py,sha256=3Rfpnowm7MUXHWmeU4UV4s_3Hk-sw3V20oBwQCYlejQ,50
 unique_toolkit/evaluators/config.py,sha256=_DIXToJ-hGNpDAdWa7Q6GMjAsxiC_DquLF-SS5s9rTE,717
@@ -47,9 +46,9 @@ unique_toolkit/evaluators/hallucination/service.py,sha256=k8qro5Lw4Ak58m4HYp3G4H
 unique_toolkit/evaluators/hallucination/utils.py,sha256=gO2AOzDQwVTev2_5vDKgJ9A6A9e0himJyAta_wglVG8,8326
 unique_toolkit/evaluators/output_parser.py,sha256=eI72qkzK1dZyUvnfP2SOAQCGBj_-PwX5wy_aLPMsJMY,883
 unique_toolkit/evaluators/schemas.py,sha256=Jaue6Uhx75X1CyHKWj8sT3RE1JZXTqoLtfLt2xQNCX8,2507
-unique_toolkit/framework_utilities/langchain/client.py,sha256=GWzL2GSxErj52dpYC15w8ABmDB5S2nr2QmV8rKmeEFM,1367
+unique_toolkit/framework_utilities/langchain/client.py,sha256=Msfmr7uezwqagyRJ2zjWbQRFqzDExWYK0y5KLEnDNqM,1329
 unique_toolkit/framework_utilities/langchain/history.py,sha256=8ejA0utPUIlX4z8gtX9IkVFiooY-vOT1UcjI_MoJaAU,638
-unique_toolkit/framework_utilities/openai/client.py,sha256=itbEv098j4alz2Rftp4kbjeloFsAqaxyyBHdZTH5OYM,1297
+unique_toolkit/framework_utilities/openai/client.py,sha256=Zs5zxFbRfa0XrlbZvhGf-GpHomKLBJx0D28iowk1Eek,1236
 unique_toolkit/framework_utilities/openai/message_builder.py,sha256=meJVGPaSQ6xEafNZtACujcD0KXdiSYGTNiv02FSJULE,3834
 unique_toolkit/framework_utilities/utils.py,sha256=JK7g2yMfEx3eMprug26769xqNpS5WJcizf8n2zWMBng,789
 unique_toolkit/language_model/__init__.py,sha256=lRQyLlbwHbNFf4-0foBU13UGb09lwEeodbVsfsSgaCk,1971
@@ -60,21 +59,21 @@ unique_toolkit/language_model/infos.py,sha256=Abxcw-l7UXkBohxZsigpNz0OUUCtDf404L
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
 unique_toolkit/language_model/reference.py,sha256=nkX2VFz-IrUz8yqyc3G5jUMNwrNpxITBrMEKkbqqYoI,8583
 unique_toolkit/language_model/schemas.py,sha256=AeuDRJFblGzEYcEMyrlxpOPk12Di3J45I9rT2xZrhEU,14332
-unique_toolkit/language_model/service.py,sha256=VRkUk2XbijqGlnTTvqU7uCue6qtT7lpLd_Y8f3bWv1I,10486
+unique_toolkit/language_model/service.py,sha256=zlvC_t9T1wixwcGDPRxl6yYniaKl2725NxWrbW51jUs,11290
 unique_toolkit/language_model/utils.py,sha256=bPQ4l6_YO71w-zaIPanUUmtbXC1_hCvLK0tAFc3VCRc,1902
 unique_toolkit/protocols/support.py,sha256=V15WEIFKVMyF1QCnR8vIi4GrJy4dfTCB6d6JlqPZ58o,2341
 unique_toolkit/short_term_memory/__init__.py,sha256=2mI3AUrffgH7Yt-xS57EGqnHf7jnn6xquoKEhJqk3Wg,185
 unique_toolkit/short_term_memory/constants.py,sha256=698CL6-wjup2MvU19RxSmQk3gX7aqW_OOpZB7sbz_Xg,34
 unique_toolkit/short_term_memory/functions.py,sha256=3WiK-xatY5nh4Dr5zlDUye1k3E6kr41RiscwtTplw5k,4484
 unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJs8FEZXcgQTNenw,1406
-unique_toolkit/short_term_memory/service.py,sha256=8sW7cFJRd4vcRfotJSWb0uHZYl-e5hQWp3G1SddL5Bg,8110
+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/tools/tool_definitions.py,sha256=YYu53vXMJBeJtuSU1L_FJBsiN52LSA5LIDt9O-1HBgE,4500
 unique_toolkit/tools/tool_definitionsV2.py,sha256=yjLmP85pFGd1QtIVMC3oLQPSQ2NckBj9hIihjIr2FZg,5728
 unique_toolkit/tools/tool_factory.py,sha256=ux11jd7Oobb-6eBeS51T-tviH14k6HKqsKmljA7h6qA,879
 unique_toolkit/tools/tool_progress_reporter.py,sha256=AyPdgxpd48qotJyPB8qJ7h7ghiv2w2EK8nlyqQVFRt4,8048
-unique_toolkit-0.8.1.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.8.1.dist-info/METADATA,sha256=mCMbEXB_uaZEQfGplkJMktiMDO45l5QZ2ZfiSlSYY9g,25737
-unique_toolkit-0.8.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.8.1.dist-info/RECORD,,
+unique_toolkit-0.8.2.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.8.2.dist-info/METADATA,sha256=nfNsDNjxM9xxW1FdMV3UVLEIXry-6kQU5Uu7RGUPrsc,26025
+unique_toolkit-0.8.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.8.2.dist-info/RECORD,,

unique_toolkit/app/event_util.py

@@ -1,24 +0,0 @@
-import json
-from logging import getLogger
-from typing import Literal, overload
-
-from unique_toolkit.app import ChatEvent, EventName
-
-LOGGER = getLogger(__name__)
-
-
-@overload
-def load_and_filter_event(
-    event: dict,
-    event_type: Literal[EventName.EXTERNAL_MODULE_CHOSEN],
-) -> type[ChatEvent] | None: ...
-
-
-def load_and_filter_event(event: dict, event_type: EventName):
-    event = json.loads(event.data)
-
-    match event_type:
-        case EventName.EXTERNAL_MODULE_CHOSEN:
-            return ChatEvent(**event)
-
-    return None

unique_toolkit/app/sse_client.py

@@ -1,20 +0,0 @@
-from logging import getLogger
-
-from sseclient import SSEClient
-
-from unique_toolkit.app.unique_settings import UniqueSettings
-
-LOGGER = getLogger(__name__)
-
-
-def get_sse_client(
-    unique_settings: UniqueSettings,
-    subscriptions: list[str],
-) -> SSEClient:
-    url = f"{unique_settings.app.base_url}/public/event-socket/events/stream?subscriptions={','.join(subscriptions)}"
-    headers = {
-        "Authorization": f"Bearer {unique_settings.app.key.get_secret_value()}",
-        "x-app-id": unique_settings.app.id.get_secret_value(),
-        "x-company-id": unique_settings.auth.company_id.get_secret_value(),
-    }
-    return SSEClient(url=url, headers=headers)