unique_toolkit 0.5.10__py3-none-any.whl → 0.5.12__py3-none-any.whl

unique_toolkit/_common/_base_service.py

@@ -1,10 +1,10 @@
 import logging
 from typing import Optional
 
-from unique_toolkit.chat.state import ChatState
+from unique_toolkit.app.schemas import Event
 
 
 class BaseService:
-    def __init__(self, state: ChatState, logger: Optional[logging.Logger] = None):
-        self.state = state
+    def __init__(self, event: Event, logger: Optional[logging.Logger] = None):
+        self.event = event
         self.logger = logger or logging.getLogger(__name__)

unique_toolkit/_common/_time_utils.py

@@ -0,0 +1,5 @@
+from datetime import datetime
+
+
+def get_datetime_now(format: str = "%Y-%m-%d %H:%M:%S.%f"):
+    return datetime.now().strftime(format)

unique_toolkit/app/__init__.py

@@ -0,0 +1,25 @@
+from .init_logging import init_logging as init_logging
+from .init_sdk import get_endpoint_secret as get_endpoint_secret
+from .init_sdk import init_sdk as init_sdk
+from .performance.async_tasks import (
+    run_async_tasks_parallel as run_async_tasks_parallel,
+)
+from .performance.async_wrapper import to_async as to_async
+from .schemas import (
+    Event as Event,
+)
+from .schemas import (
+    EventAssistantMessage as EventAssistantMessage,
+)
+from .schemas import (
+    EventName as EventName,
+)
+from .schemas import (
+    EventPayload as EventPayload,
+)
+from .schemas import (
+    EventUserMessage as EventUserMessage,
+)
+from .verification import (
+    verify_signature_and_construct_event as verify_signature_and_construct_event,
+)

unique_toolkit/app/schemas.py

@@ -1,12 +1,14 @@
 from enum import StrEnum
-from typing import Any
+from typing import Any, Optional
 
 from humps import camelize
 from pydantic import BaseModel, ConfigDict
 
 # set config to convert camelCase to snake_case
 model_config = ConfigDict(
-    alias_generator=camelize, populate_by_name=True, arbitrary_types_allowed=True
+    alias_generator=camelize,
+    populate_by_name=True,
+    arbitrary_types_allowed=True,
 )
 
 
@@ -19,7 +21,9 @@ class EventUserMessage(BaseModel):
 
     id: str
     text: str
+    original_text: str
     created_at: str
+    language: str
 
 
 class EventAssistantMessage(BaseModel):
@@ -29,6 +33,12 @@ class EventAssistantMessage(BaseModel):
     created_at: str
 
 
+class EventAdditionalParameters(BaseModel):
+    model_config = model_config
+
+    translate_to_language: Optional[str] = None
+
+
 class EventPayload(BaseModel):
     model_config = model_config
 
@@ -39,7 +49,8 @@ class EventPayload(BaseModel):
     assistant_id: str
     user_message: EventUserMessage
     assistant_message: EventAssistantMessage
-    text: str | None = None
+    text: Optional[str] = None
+    additional_parameters: Optional[EventAdditionalParameters] = None
 
 
 class Event(BaseModel):
@@ -50,5 +61,5 @@ class Event(BaseModel):
     user_id: str
     company_id: str
     payload: EventPayload
-    created_at: int | None = None
-    version: str | None = None
+    created_at: Optional[int] = None
+    version: Optional[str] = None

unique_toolkit/chat/__init__.py

@@ -0,0 +1,6 @@
+from .schemas import ChatMessage as ChatMessage
+from .schemas import ChatMessageRole as ChatMessageRole
+from .service import ChatService as ChatService
+from .utils import (
+    convert_chat_history_to_injectable_string as convert_chat_history_to_injectable_string,
+)

unique_toolkit/chat/schemas.py

@@ -1,4 +1,4 @@
-from enum import Enum
+from enum import StrEnum
 
 from humps import camelize
 from pydantic import BaseModel, ConfigDict, Field, field_validator
@@ -9,7 +9,7 @@ model_config = ConfigDict(
 )
 
 
-class ChatMessageRole(str, Enum):
+class ChatMessageRole(StrEnum):
     USER = "user"
     ASSISTANT = "assistant"
 

unique_toolkit/chat/service.py

@@ -5,9 +5,10 @@ from typing import Optional
 import unique_sdk
 from unique_sdk._list_object import ListObject
 
+from unique_toolkit._common import _time_utils
 from unique_toolkit._common._base_service import BaseService
+from unique_toolkit.app.schemas import Event
 from unique_toolkit.chat.schemas import ChatMessage, ChatMessageRole
-from unique_toolkit.chat.state import ChatState
 from unique_toolkit.content.schemas import ContentReference
 from unique_toolkit.content.utils import count_tokens
 
@@ -17,12 +18,12 @@ class ChatService(BaseService):
     Provides all functionalities to manage the chat session.
 
     Attributes:
-        state (ChatState): The chat state.
+        event (Event): The Event object.
         logger (Optional[logging.Logger]): The logger. Defaults to None.
     """
 
-    def __init__(self, state: ChatState, logger: Optional[logging.Logger] = None):
-        super().__init__(state, logger)
+    def __init__(self, event: Event, logger: Optional[logging.Logger] = None):
+        super().__init__(event, logger)
 
     DEFAULT_PERCENT_OF_MAX_TOKENS = 0.15
     DEFAULT_MAX_MESSAGES = 4
@@ -49,17 +50,18 @@ class ChatService(BaseService):
         Raises:
             Exception: If the modification fails.
         """
-        message_id = message_id or self.state.assistant_message_id
+        message_id = message_id or self.event.payload.assistant_message.id
 
         try:
             message = unique_sdk.Message.modify(
-                user_id=self.state.user_id,
-                company_id=self.state.company_id,
-                id=message_id,  # type: ignore
-                chatId=self.state.chat_id,
+                user_id=self.event.user_id,
+                company_id=self.event.company_id,
+                id=message_id,
+                chatId=self.event.payload.chat_id,
                 text=content,
-                references=self._map_references(references),  # type: ignore
+                references=self._map_references(references),
                 debugInfo=debug_info or {},
+                completedAt=_time_utils.get_datetime_now(),  # type: ignore
             )
         except Exception as e:
             self.logger.error(f"Failed to modify assistant message: {e}")
@@ -88,17 +90,18 @@ class ChatService(BaseService):
         Raises:
             Exception: If the modification fails.
         """
-        message_id = message_id or self.state.assistant_message_id
+        message_id = message_id or self.event.payload.assistant_message.id
 
         try:
             message = await unique_sdk.Message.modify_async(
-                user_id=self.state.user_id,
-                company_id=self.state.company_id,
-                id=message_id,  # type: ignore
-                chatId=self.state.chat_id,
+                user_id=self.event.user_id,
+                company_id=self.event.company_id,
+                id=message_id,
+                chatId=self.event.payload.chat_id,
                 text=content,
-                references=self._map_references(references),  # type: ignore
+                references=self._map_references(references),
                 debugInfo=debug_info or {},
+                completedAt=_time_utils.get_datetime_now(),  # type: ignore
             )
         except Exception as e:
             self.logger.error(f"Failed to modify assistant message: {e}")
@@ -210,10 +213,10 @@ class ChatService(BaseService):
 
         try:
             message = unique_sdk.Message.create(
-                user_id=self.state.user_id,
-                company_id=self.state.company_id,
-                chatId=self.state.chat_id,
-                assistantId=self.state.assistant_id,
+                user_id=self.event.user_id,
+                company_id=self.event.company_id,
+                chatId=self.event.payload.chat_id,
+                assistantId=self.event.payload.assistant_id,
                 text=content,
                 role=ChatMessageRole.ASSISTANT.name,
                 references=self._map_references(references),  # type: ignore
@@ -246,10 +249,10 @@ class ChatService(BaseService):
         """
         try:
             message = await unique_sdk.Message.create_async(
-                user_id=self.state.user_id,
-                company_id=self.state.company_id,
-                chatId=self.state.chat_id,
-                assistantId=self.state.assistant_id,
+                user_id=self.event.user_id,
+                company_id=self.event.company_id,
+                chatId=self.event.payload.chat_id,
+                assistantId=self.event.payload.assistant_id,
                 text=content,
                 role=ChatMessageRole.ASSISTANT.name,
                 references=self._map_references(references),  # type: ignore
@@ -274,13 +277,13 @@ class ChatService(BaseService):
         ]
 
     def _get_full_history(self):
-        messages = self._trigger_list_messages(self.state.chat_id)
+        messages = self._trigger_list_messages(self.event.payload.chat_id)
         messages = self._filter_valid_messages(messages)
 
         return self._map_to_chat_messages(messages)
 
     async def _get_full_history_async(self):
-        messages = await self._trigger_list_messages_async(self.state.chat_id)
+        messages = await self._trigger_list_messages_async(self.event.payload.chat_id)
         messages = self._filter_valid_messages(messages)
 
         return self._map_to_chat_messages(messages)
@@ -305,8 +308,8 @@ class ChatService(BaseService):
     def _trigger_list_messages(self, chat_id: str):
         try:
             messages = unique_sdk.Message.list(
-                user_id=self.state.user_id,
-                company_id=self.state.company_id,
+                user_id=self.event.user_id,
+                company_id=self.event.company_id,
                 chatId=chat_id,
             )
             return messages
@@ -317,8 +320,8 @@ class ChatService(BaseService):
     async def _trigger_list_messages_async(self, chat_id: str):
         try:
             messages = await unique_sdk.Message.list_async(
-                user_id=self.state.user_id,
-                company_id=self.state.company_id,
+                user_id=self.event.user_id,
+                company_id=self.event.company_id,
                 chatId=chat_id,
             )
             return messages

unique_toolkit/chat/state.py

@@ -5,7 +5,7 @@ from unique_toolkit.app.schemas import Event
 
 
 @dataclass
-class ChatState:
+class ChatStateTemplate:
     """
     Represents the state of the chat session.
 
@@ -36,12 +36,12 @@ class ChatState:
             event (Event): The Event object.
 
         Returns:
-            ChatManager: The ChatManager instance.
+            ChatState: The ChatState instance.
         """
         return cls(
             user_id=event.user_id,
-            chat_id=event.payload.chat_id,
             company_id=event.company_id,
+            chat_id=event.payload.chat_id,
             assistant_id=event.payload.assistant_id,
             user_message_text=event.payload.user_message.text,
             user_message_id=event.payload.user_message.id,

unique_toolkit/content/__init__.py

@@ -0,0 +1,37 @@
+from .schemas import (
+    Content as Content,
+)
+from .schemas import (
+    ContentChunk as ContentChunk,
+)
+from .schemas import (
+    ContentMetadata as ContentMetadata,
+)
+from .schemas import (
+    ContentReference as ContentReference,
+)
+from .schemas import (
+    ContentRerankerConfig as ContentRerankerConfig,
+)
+from .schemas import (
+    ContentSearchResult as ContentSearchResult,
+)
+from .schemas import (
+    ContentSearchType as ContentSearchType,
+)
+from .schemas import (
+    ContentUploadInput as ContentUploadInput,
+)
+from .service import ContentService as ContentService
+from .utils import (
+    count_tokens as count_tokens,
+)
+from .utils import (
+    merge_content_chunks as merge_content_chunks,
+)
+from .utils import (
+    pick_content_chunks_for_token_window as pick_content_chunks_for_token_window,
+)
+from .utils import (
+    sort_content_chunks as sort_content_chunks,
+)

unique_toolkit/content/schemas.py

@@ -92,7 +92,7 @@ class ContentUploadInput(BaseModel):
     byte_size: Optional[int] = None
 
 
-class RerankerConfig(BaseModel):
+class ContentRerankerConfig(BaseModel):
     model_config = model_config
     deployment_name: str = Field(serialization_alias="deploymentName")
     options: dict | None = None

unique_toolkit/content/service.py

@@ -8,13 +8,13 @@ import requests
 import unique_sdk
 
 from unique_toolkit._common._base_service import BaseService
-from unique_toolkit.chat.state import ChatState
+from unique_toolkit.app.schemas import Event
 from unique_toolkit.content.schemas import (
     Content,
     ContentChunk,
+    ContentRerankerConfig,
     ContentSearchType,
     ContentUploadInput,
-    RerankerConfig,
 )
 
 
@@ -23,12 +23,12 @@ class ContentService(BaseService):
     Provides methods for searching, downloading and uploading content in the knowledge base.
 
     Attributes:
-        state (ChatState): The chat state.
+        event (Event): The Event object.
         logger (Optional[logging.Logger]): The logger. Defaults to None.
     """
 
-    def __init__(self, state: ChatState, logger: Optional[logging.Logger] = None):
-        super().__init__(state, logger)
+    def __init__(self, event: Event, logger: Optional[logging.Logger] = None):
+        super().__init__(event, logger)
 
     DEFAULT_SEARCH_LANGUAGE = "english"
 
@@ -38,7 +38,7 @@ class ContentService(BaseService):
         search_type: ContentSearchType,
         limit: int,
         search_language: str = DEFAULT_SEARCH_LANGUAGE,
-        reranker_config: Optional[RerankerConfig] = None,
+        reranker_config: Optional[ContentRerankerConfig] = None,
         scope_ids: Optional[list[str]] = None,
         chat_only: Optional[bool] = None,
     ) -> list[ContentChunk]:
@@ -50,7 +50,7 @@ class ContentService(BaseService):
             search_type (ContentSearchType): The type of search to perform.
             limit (int): The maximum number of results to return.
             search_language (str): The language for the full-text search. Defaults to "english".
-            reranker_config (Optional[RerankerConfig]): The reranker configuration. Defaults to None.
+            reranker_config (Optional[ContentRerankerConfig]): The reranker configuration. Defaults to None.
             scope_ids (Optional[list[str]]): The scope IDs. Defaults to None.
             chat_only (Optional[bool]): Whether to search only in the current chat. Defaults to None.
 
@@ -62,9 +62,9 @@ class ContentService(BaseService):
 
         try:
             searches = unique_sdk.Search.create(
-                user_id=self.state.user_id,
-                company_id=self.state.company_id,
-                chatId=self.state.chat_id,
+                user_id=self.event.user_id,
+                company_id=self.event.company_id,
+                chatId=self.event.payload.chat_id,
                 searchString=search_string,
                 searchType=search_type.name,
                 scopeIds=scope_ids,
@@ -92,7 +92,7 @@ class ContentService(BaseService):
         search_type: ContentSearchType,
         limit: int,
         search_language: str = DEFAULT_SEARCH_LANGUAGE,
-        reranker_config: Optional[RerankerConfig] = None,
+        reranker_config: Optional[ContentRerankerConfig] = None,
         scope_ids: Optional[list[str]] = None,
         chat_only: Optional[bool] = None,
     ):
@@ -104,7 +104,7 @@ class ContentService(BaseService):
             search_type (ContentSearchType): The type of search to perform.
             limit (int): The maximum number of results to return.
             search_language (str): The language for the full-text search. Defaults to "english".
-            reranker_config (Optional[RerankerConfig]): The reranker configuration. Defaults to None.
+            reranker_config (Optional[ContentRerankerConfig]): The reranker configuration. Defaults to None.
             scope_ids (Optional[list[str]]): The scope IDs. Defaults to None.
             chat_only (Optional[bool]): Whether to search only in the current chat. Defaults to None.
 
@@ -116,9 +116,9 @@ class ContentService(BaseService):
 
         try:
             searches = await unique_sdk.Search.create_async(
-                user_id=self.state.user_id,
-                company_id=self.state.company_id,
-                chatId=self.state.chat_id,
+                user_id=self.event.user_id,
+                company_id=self.event.company_id,
+                chatId=self.event.payload.chat_id,
                 searchString=search_string,
                 searchType=search_type.name,
                 scopeIds=scope_ids,
@@ -156,9 +156,9 @@ class ContentService(BaseService):
         """
         try:
             contents = unique_sdk.Content.search(
-                user_id=self.state.user_id,
-                company_id=self.state.company_id,
-                chatId=self.state.chat_id,
+                user_id=self.event.user_id,
+                company_id=self.event.company_id,
+                chatId=self.event.payload.chat_id,
                 # TODO add type parameter
                 where=where,  # type: ignore
             )
@@ -183,9 +183,9 @@ class ContentService(BaseService):
         """
         try:
             contents = await unique_sdk.Content.search_async(
-                user_id=self.state.user_id,
-                company_id=self.state.company_id,
-                chatId=self.state.chat_id,
+                user_id=self.event.user_id,
+                company_id=self.event.company_id,
+                chatId=self.event.payload.chat_id,
                 # TODO add type parameter
                 where=where,  # type: ignore
             )
@@ -325,8 +325,8 @@ class ContentService(BaseService):
                     "mimeType": input.mime_type,
                 }
             content = unique_sdk.Content.upsert(
-                user_id=self.state.user_id,
-                company_id=self.state.company_id,
+                user_id=self.event.user_id,
+                company_id=self.event.company_id,
                 input=input_json,  # type: ignore
                 fileUrl=content_url,
                 scopeId=scope_id,
@@ -374,8 +374,8 @@ class ContentService(BaseService):
         headers = {
             "x-api-version": unique_sdk.api_version,
             "x-app-id": unique_sdk.app_id,
-            "x-user-id": self.state.user_id,
-            "x-company-id": self.state.company_id,
+            "x-user-id": self.event.user_id,
+            "x-company-id": self.event.company_id,
             "Authorization": "Bearer %s" % (unique_sdk.api_key,),
         }
 

unique_toolkit/content/utils.py

@@ -122,7 +122,7 @@ def _generate_pages_postfix(chunks: list[ContentChunk]) -> str:
         return list(range(start, end + 1))
 
     page_numbers_array = [
-        gen_all_numbers_in_between(s.start_page, s.end_page) for s in chunks
+        gen_all_numbers_in_between(c.start_page, c.end_page) for c in chunks
     ]
     page_numbers = [number for sublist in page_numbers_array for number in sublist]
     page_numbers = [p for p in page_numbers if p > 0]
@@ -171,7 +171,7 @@ def pick_content_chunks_for_token_window(
     return picked_chunks
 
 
-def count_tokens(text, encoding_model="cl100k_base") -> int:
+def count_tokens(text: str, encoding_model="cl100k_base") -> int:
     """
     Counts the number of tokens in the provided text.
 

unique_toolkit/embedding/__init__.py

@@ -0,0 +1,5 @@
+from .schemas import Embeddings as Embeddings
+from .service import EmbeddingService as EmbeddingService
+from .utils import (
+    calculate_cosine_similarity as calculate_cosine_similarity,
+)

unique_toolkit/embedding/service.py

@@ -1,11 +1,10 @@
 import logging
 from typing import Optional
 
-import numpy as np
 import unique_sdk
 
 from unique_toolkit._common._base_service import BaseService
-from unique_toolkit.chat.state import ChatState
+from unique_toolkit.app.schemas import Event
 from unique_toolkit.embedding.schemas import Embeddings
 
 
@@ -14,12 +13,12 @@ class EmbeddingService(BaseService):
     Provides methods to interact with the Embedding service.
 
     Attributes:
-        state (ChatState): The ChatState object.
+        event (Event): The Event object.
         logger (Optional[logging.Logger]): The logger object. Defaults t∏o None.
     """
 
-    def __init__(self, state: ChatState, logger: Optional[logging.Logger] = None):
-        super().__init__(state, logger)
+    def __init__(self, event: Event, logger: Optional[logging.Logger] = None):
+        super().__init__(event, logger)
 
     DEFAULT_TIMEOUT = 600_000
 
@@ -77,18 +76,8 @@ class EmbeddingService(BaseService):
 
     def _get_request_obj(self, texts: list[str], timeout: int) -> dict:
         return {
-            "user_id": self.state.user_id,
-            "company_id": self.state.company_id,
+            "user_id": self.event.user_id,
+            "company_id": self.event.company_id,
             "texts": texts,
             "timeout": timeout,
         }
-
-    def get_cosine_similarity(
-        self,
-        embedding_1: list[float],
-        embedding_2: list[float],
-    ) -> float:
-        """Get cosine similarity."""
-        return np.dot(embedding_1, embedding_2) / (
-            np.linalg.norm(embedding_1) * np.linalg.norm(embedding_2)
-        )

unique_toolkit/embedding/utils.py

@@ -0,0 +1,11 @@
+import numpy as np
+
+
+def calculate_cosine_similarity(
+    embedding_1: list[float],
+    embedding_2: list[float],
+) -> float:
+    """Get cosine similarity."""
+    return np.dot(embedding_1, embedding_2) / (
+        np.linalg.norm(embedding_1) * np.linalg.norm(embedding_2)
+    )

unique_toolkit/language_model/__init__.py

@@ -0,0 +1,59 @@
+from .infos import LanguageModel as LanguageModel
+from .infos import LanguageModelName as LanguageModelName
+from .schemas import (
+    LanguageModelAssistantMessage as LanguageModelAssistantMessage,
+)
+from .schemas import (
+    LanguageModelCompletionChoice as LanguageModelCompletionChoice,
+)
+from .schemas import (
+    LanguageModelFunction as LanguageModelFunction,
+)
+from .schemas import (
+    LanguageModelFunctionCall as LanguageModelFunctionCall,
+)
+from .schemas import (
+    LanguageModelMessage as LanguageModelMessage,
+)
+from .schemas import (
+    LanguageModelMessageRole as LanguageModelMessageRole,
+)
+from .schemas import (
+    LanguageModelMessages as LanguageModelMessages,
+)
+from .schemas import (
+    LanguageModelResponse as LanguageModelResponse,
+)
+from .schemas import (
+    LanguageModelStreamResponse as LanguageModelStreamResponse,
+)
+from .schemas import (
+    LanguageModelStreamResponseMessage as LanguageModelStreamResponseMessage,
+)
+from .schemas import (
+    LanguageModelSystemMessage as LanguageModelSystemMessage,
+)
+from .schemas import (
+    LanguageModelTokenLimits as LanguageModelTokenLimits,
+)
+from .schemas import (
+    LanguageModelTool as LanguageModelTool,
+)
+from .schemas import (
+    LanguageModelToolParameterProperty as LanguageModelToolParameterProperty,
+)
+from .schemas import (
+    LanguageModelToolParameters as LanguageModelToolParameters,
+)
+from .schemas import (
+    LanguageModelUserMessage as LanguageModelUserMessage,
+)
+from .service import (
+    LanguageModelService as LanguageModelService,
+)
+from .utils import (
+    convert_string_to_json as convert_string_to_json,
+)
+from .utils import (
+    find_last_json_object as find_last_json_object,
+)

unique_toolkit/language_model/infos.py

@@ -1,4 +1,3 @@
-import warnings
 from datetime import date
 from enum import StrEnum
 from typing import ClassVar, Optional, Type, cast
@@ -149,10 +148,11 @@ class LanguageModel:
     def get_model_info(cls, model_name: LanguageModelName) -> LanguageModelInfo:
         for subclass in cls.__subclasses__():
             if hasattr(subclass, "info") and subclass._info.name == model_name:
-                if subclass._info.retirement_at:
-                    warning_text = f"WARNING: {subclass._info.name} will be retired on {subclass._info.retirement_at.isoformat()} and from then on not accessible anymore. {subclass._info.retirement_text}"
-                    print(warning_text)
-                    warnings.warn(warning_text, DeprecationWarning, stacklevel=2)
+                # TODO find alternative solution for warning
+                # if subclass._info.retirement_at:
+                #     warning_text = f"WARNING: {subclass._info.name} will be retired on {subclass._info.retirement_at.isoformat()} and from then on not accessible anymore. {subclass._info.retirement_text}"
+                #     print(warning_text)
+                #     warnings.warn(warning_text, DeprecationWarning, stacklevel=2)
                 return subclass._info
         raise ValueError(f"Model {model_name} not found.")
 

unique_toolkit/language_model/schemas.py

@@ -24,7 +24,7 @@ class LanguageModelFunction(BaseModel):
 
     id: Optional[str] = None
     name: str
-    arguments: Optional[dict[str, any]] = None  # type: ignore
+    arguments: Optional[dict[str, list | dict | str | int | float | bool]] = None  # type: ignore
 
     @field_validator("arguments", mode="before")
     def set_arguments(cls, value):
@@ -104,7 +104,7 @@ class LanguageModelStreamResponseMessage(BaseModel):
     role: LanguageModelMessageRole
     text: str
     original_text: Optional[str] = None
-    references: list[dict[str, any]] = []  # type: ignore
+    references: list[dict[str, list | dict | str | int | float | bool]] = []  # type: ignore
 
     # TODO make sdk return role in lowercase
     # Currently needed as sdk returns role in uppercase

unique_toolkit/language_model/service.py

@@ -4,7 +4,7 @@ from typing import Optional, cast
 import unique_sdk
 
 from unique_toolkit._common._base_service import BaseService
-from unique_toolkit.chat.state import ChatState
+from unique_toolkit.app.schemas import Event
 from unique_toolkit.content.schemas import ContentChunk
 from unique_toolkit.language_model.infos import LanguageModelName
 from unique_toolkit.language_model.schemas import (
@@ -20,12 +20,12 @@ class LanguageModelService(BaseService):
     Provides methods to interact with the Language Model by generating responses.
 
     Attributes:
-        state (ChatState): The ChatState object.
+        event (Event): The Event object.
         logger (Optional[logging.Logger]): The logger object. Defaults to None.
     """
 
-    def __init__(self, state: ChatState, logger: Optional[logging.Logger] = None):
-        super().__init__(state, logger)
+    def __init__(self, event: Event, logger: Optional[logging.Logger] = None):
+        super().__init__(event, logger)
 
     DEFAULT_COMPLETE_TIMEOUT = 240_000
     DEFAULT_COMPLETE_TEMPERATURE = 0.0
@@ -55,9 +55,9 @@ class LanguageModelService(BaseService):
         messages = messages.model_dump(exclude_none=True)
         try:
             response = unique_sdk.ChatCompletion.create(
-                company_id=self.state.company_id,
+                company_id=self.event.company_id,
                 # TODO change or extend types in unique_sdk
-                model=model_name.name,  # type: ignore
+                model=model_name.name,
                 messages=cast(
                     list[unique_sdk.Integrated.ChatCompletionRequestMessage],
                     messages,
@@ -96,9 +96,9 @@ class LanguageModelService(BaseService):
         messages = messages.model_dump(exclude_none=True, exclude={"tool_calls"})
         try:
             response = await unique_sdk.ChatCompletion.create_async(
-                company_id=self.state.company_id,
+                company_id=self.event.company_id,
                 # TODO change or extend types in unique_sdk
-                model=model_name.name,  # type: ignore
+                model=model_name.name,
                 messages=cast(
                     list[unique_sdk.Integrated.ChatCompletionRequestMessage],
                     messages,
@@ -145,21 +145,21 @@ class LanguageModelService(BaseService):
 
         try:
             response = unique_sdk.Integrated.chat_stream_completion(
-                user_id=self.state.user_id,
-                company_id=self.state.company_id,
-                assistantMessageId=self.state.assistant_message_id,  # type: ignore
-                userMessageId=self.state.user_message_id,  # type: ignore
+                user_id=self.event.user_id,
+                company_id=self.event.company_id,
+                assistantMessageId=self.event.payload.assistant_message.id,
+                userMessageId=self.event.payload.user_message.id,
                 messages=cast(
                     list[unique_sdk.Integrated.ChatCompletionRequestMessage],
                     messages,
                 ),
-                chatId=self.state.chat_id,
+                chatId=self.event.payload.chat_id,
                 searchContext=search_context,
                 # TODO change or extend types in unique_sdk
-                model=model_name.name,  # type: ignore
+                model=model_name.name,
                 timeout=timeout,
                 temperature=temperature,
-                assistantId=self.state.assistant_id,
+                assistantId=self.event.payload.assistant_id,
                 debugInfo=debug_info,
                 options=options,  # type: ignore
                 startText=start_text,
@@ -203,22 +203,22 @@ class LanguageModelService(BaseService):
 
         try:
             response = await unique_sdk.Integrated.chat_stream_completion_async(
-                user_id=self.state.user_id,
-                company_id=self.state.company_id,
-                assistantMessageId=self.state.assistant_message_id,  # type: ignore
-                userMessageId=self.state.user_message_id,  # type: ignore
+                user_id=self.event.user_id,
+                company_id=self.event.company_id,
+                assistantMessageId=self.event.payload.assistant_message.id,
+                userMessageId=self.event.payload.user_message.id,
                 messages=cast(
                     list[unique_sdk.Integrated.ChatCompletionRequestMessage],
                     messages,
                 ),
-                chatId=self.state.chat_id,
+                chatId=self.event.payload.chat_id,
                 searchContext=search_context,
-                # TODO change or extend types in unique_sdk
-                model=model_name.name,  # type: ignore
+                model=model_name.name,
                 timeout=timeout,
                 temperature=temperature,
-                assistantId=self.state.assistant_id,
+                assistantId=self.event.payload.assistant_id,
                 debugInfo=debug_info,
+                # TODO change or extend types in unique_sdk
                 options=options,  # type: ignore
                 startText=start_text,
             )

{unique_toolkit-0.5.10.dist-info → unique_toolkit-0.5.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.10
+Version: 0.5.12
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -13,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: numpy (>=1.26.4,<2.0.0)
 Requires-Dist: pydantic (>=2.8.2,<3.0.0)
 Requires-Dist: pyhumps (>=3.8.0,<4.0.0)
+Requires-Dist: pytest-mock (>=3.14.0,<4.0.0)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: regex (>=2024.5.15,<2025.0.0)
 Requires-Dist: tiktoken (>=0.7.0,<0.8.0)
@@ -55,7 +56,6 @@ The `unique_toolkit.app` module encompasses functions for initializing and secur
 
 The `unique_toolkit.chat` module encompasses all chat related functionality.
 
-- `state.py` comprises the ChatState which is used to store the current state of the chat interaction and the user information.
 - `service.py` comprises the ChatService and provides an interface to manage and load the chat history and interact with the chat ui, e.g., creating a new assistant message.
 - `schemas.py` comprises all relevant schemas, e.g., ChatMessage, used in the ChatService.
 - `utils.py` comprises utility functions to use and convert ChatMessage objects in assistants, e.g., convert_chat_history_to_injectable_string converts the chat history to a string that can be injected into a prompt. 
@@ -101,6 +101,19 @@ 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.5.12] - 2024-08-7
+- added `completedAt` datetime to `unique_sdk.Message.modify` and `unique_sdk.Message.modify_async`
+- added `original_text` and `language` to `EventUserMessage`
+
+## [0.5.11] - 2024-08-6
+- made all domain specific functions and classes directly importable from `unique_toolkit.[DOMAIN_NAME]`
+- renamed `RerankerConfig` to `ContentRerankerConfig`
+- renamed `get_cosine_similarity` to `calculate_cosine_similarity` and moved it to `unique_toolkit.embedding.utils`
+- moved `calculate_tokens` from `unique_toolkit.content.utils` to `unique_toolkit.embedding.utils`
+- disabled deprecation warning in `LanguageModel`
+- added `additional_parameters` to event
+- removed `ChatState` and use `Event` instead
+
 ## [0.5.10] - 2024-08-6
 - fix content schema
 

unique_toolkit-0.5.12.dist-info/RECORD

@@ -0,0 +1,32 @@
+unique_toolkit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+unique_toolkit/_common/_base_service.py,sha256=S8H0rAebx7GsOldA7xInLp3aQJt9yEPDQdsGSFRJsGg,276
+unique_toolkit/_common/_time_utils.py,sha256=ztmTovTvr-3w71Ns2VwXC65OKUUh-sQlzbHdKTQWm-w,135
+unique_toolkit/app/__init__.py,sha256=sZyGrz74jBlAjv6OcHgcp6VtP6-AKKpaVYjakr1Xk60,735
+unique_toolkit/app/init_logging.py,sha256=Sh26SRxOj8i8dzobKhYha2lLrkrMTHfB1V4jR3h23gQ,678
+unique_toolkit/app/init_sdk.py,sha256=Nv4Now4pMfM0AgRhbtatLpm_39rKxn0WmRLwmPhRl-8,1285
+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=_PIROOUtdKvZFZdvkCORhj-MVWvfkopIQW7VIFmguRg,1364
+unique_toolkit/app/verification.py,sha256=UZqTHg3PX_QxMjeLH_BVBYoMVqMnMpeMoqvyTBKDqj8,1996
+unique_toolkit/chat/__init__.py,sha256=1prdTVfLOf6NgU-Aa1VIO-XiR6OYuRm51LaVRfKDCqc,267
+unique_toolkit/chat/schemas.py,sha256=ff4M-XMagF0Evn3FcKHHP5xzDEyufZgq9Dmit3i8r_E,802
+unique_toolkit/chat/service.py,sha256=mjzj0GCpIbGMX_k1aei8BY-132DCPv53FI7TqbXfaKk,13544
+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=MSH2sxjQyKD2Sef92fzE5Dt9SihdzivB6yliSwJfTmQ,890
+unique_toolkit/content/schemas.py,sha256=UlC5nBIaFkq9TD31LR6ioG9JRZ1ScmtABi0l06HZR70,2231
+unique_toolkit/content/service.py,sha256=4CHqZDE3C-VQpG6RAlenarvBmBgI28Ec6T9HZf9TB84,13565
+unique_toolkit/content/utils.py,sha256=Lake671plRsqNvO3pN_rmyVcpwbdED_KQpLcCnc4lv4,6902
+unique_toolkit/embedding/__init__.py,sha256=dr8M9jvslQTxPpxgaGwzxY0FildiWf-DidN_cahPAWw,191
+unique_toolkit/embedding/schemas.py,sha256=1GvKCaSk4jixzVQ2PKq8yDqwGEVY_hWclYtoAr6CC2g,96
+unique_toolkit/embedding/service.py,sha256=Iiw-sbdkjuWlWMfLM9qyC4GNTJOotQAaVjkYvh5Su4Y,2370
+unique_toolkit/embedding/utils.py,sha256=v86lo__bCJbxZBQ3OcLu5SuwT6NbFfWlcq8iyk6BuzQ,279
+unique_toolkit/language_model/__init__.py,sha256=QgU_uwpVh1URQyVs6l-6Am4UwmEEhuGXNic3dUZ0FCc,1701
+unique_toolkit/language_model/infos.py,sha256=W74PiBkAIEdTdgygdVoq_EWlziVOOfhSuETq4dKX7LM,8934
+unique_toolkit/language_model/schemas.py,sha256=6PySftHNX1e00ncieRn4h35T4VneOO8j-mfpkC8Cvk8,4447
+unique_toolkit/language_model/service.py,sha256=JjsOOcGDcR7db3yF3_oDXclEGfxqmwWpL5jor7Q42cU,10470
+unique_toolkit/language_model/utils.py,sha256=WBPj1XKkDgxy_-T8HCZvsfkkSzj_1w4UZzNmyvdbBLY,1081
+unique_toolkit-0.5.12.dist-info/LICENSE,sha256=bIeCWCYuoUU_MzNdg48-ubJSVm7qxakaRbzTiJ5uxrs,1065
+unique_toolkit-0.5.12.dist-info/METADATA,sha256=ttyZ2AUYRAvvYkaV_B8_UC3TVER9tdHQXI2rXgrhXBQ,9812
+unique_toolkit-0.5.12.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.5.12.dist-info/RECORD,,

unique_toolkit-0.5.10.dist-info/RECORD

@@ -1,25 +0,0 @@
-unique_toolkit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-unique_toolkit/_common/_base_service.py,sha256=SDhO69RA6E1ydUuV9N99RKreHZbQmMJj6SCdrYiakhU,283
-unique_toolkit/app/init_logging.py,sha256=Sh26SRxOj8i8dzobKhYha2lLrkrMTHfB1V4jR3h23gQ,678
-unique_toolkit/app/init_sdk.py,sha256=Nv4Now4pMfM0AgRhbtatLpm_39rKxn0WmRLwmPhRl-8,1285
-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=UAbIkrgdqwWEwKqnJwB4TNQWuJgEsFzDxqJbEC8xcOc,1098
-unique_toolkit/app/verification.py,sha256=UZqTHg3PX_QxMjeLH_BVBYoMVqMnMpeMoqvyTBKDqj8,1996
-unique_toolkit/chat/schemas.py,sha256=TYqrDy96ynP7XpnT_abOJlq6LkqBTdKwEft4YHeEgR8,801
-unique_toolkit/chat/service.py,sha256=jTxFchdjcGp3IQZxWR4x3XTdATxodP0xD4LGJZYltVw,13338
-unique_toolkit/chat/state.py,sha256=qrXkxmYlTM6PRpHX6j_vhJNWkJX4r-xgJ3NvqM80rAY,1441
-unique_toolkit/chat/utils.py,sha256=ihm-wQykBWhB4liR3LnwPVPt_qGW6ETq21Mw4HY0THE,854
-unique_toolkit/content/schemas.py,sha256=5Tkn6fulop6Ya2iCZvth7I5XPwi8QPqPUgDFV7ZAQQg,2224
-unique_toolkit/content/service.py,sha256=_o2mCgi0cN3YIvlFFJnwBZw8YHzR37DM2Mqsk157IRc,13507
-unique_toolkit/content/utils.py,sha256=x3ABo8ZCRm3YJAQwDtrr82z77DmW4Mei7KCIITjP0fk,6897
-unique_toolkit/embedding/schemas.py,sha256=1GvKCaSk4jixzVQ2PKq8yDqwGEVY_hWclYtoAr6CC2g,96
-unique_toolkit/embedding/service.py,sha256=xcU84ANMIebBui_AWBhfN1rqz7evXYRgg_Z-132veNE,2703
-unique_toolkit/language_model/infos.py,sha256=NhAkeW7PyusSIHCMvwRikLlzGG4tOXSLf_Fnq7V9rNE,8881
-unique_toolkit/language_model/schemas.py,sha256=kTGSGT3ygrH3guQELOWpxN4MTgEPuudi-CTvRu-zCcI,4377
-unique_toolkit/language_model/service.py,sha256=0g0nAGj9Te16YHFhqO9pkszhEwtuZ6bSqor6hi49e6c,10549
-unique_toolkit/language_model/utils.py,sha256=WBPj1XKkDgxy_-T8HCZvsfkkSzj_1w4UZzNmyvdbBLY,1081
-unique_toolkit-0.5.10.dist-info/LICENSE,sha256=bIeCWCYuoUU_MzNdg48-ubJSVm7qxakaRbzTiJ5uxrs,1065
-unique_toolkit-0.5.10.dist-info/METADATA,sha256=wZsvcVpS4cZhkXfa9PRrycLHMoOnKW0c8tDK6V0FY5Y,9173
-unique_toolkit-0.5.10.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.5.10.dist-info/RECORD,,