unique_toolkit 0.7.7__py3-none-any.whl → 1.23.0__py3-none-any.whl

unique_toolkit/chat/service.py

@@ -1,27 +1,44 @@
-import logging
-from typing import Optional
+"""
+DEPRECATED: This module is maintained for backward compatibility only.
 
-from typing_extensions import deprecated
+Please import from `unique_toolkit.services.chat_service` instead:
+    from unique_toolkit.services.chat_service import ChatService
+
+All imports from this module will continue to work but may be removed in a future version.
+"""
+
+import warnings
 
-from unique_toolkit._common.validate_required_values import validate_required_values
-from unique_toolkit.app.schemas import ChatEvent, Event
 from unique_toolkit.chat.constants import (
     DEFAULT_MAX_MESSAGES,
     DEFAULT_PERCENT_OF_MAX_TOKENS,
     DOMAIN_NAME,
 )
+from unique_toolkit.chat.deprecated.service import ChatServiceDeprecated
 from unique_toolkit.chat.functions import (
     create_message,
     create_message_assessment,
     create_message_assessment_async,
     create_message_async,
+    create_message_execution,
+    create_message_execution_async,
+    create_message_log,
+    create_message_log_async,
     get_full_history,
     get_full_history_async,
+    get_message_execution,
+    get_message_execution_async,
     get_selection_from_history,
     modify_message,
     modify_message_assessment,
     modify_message_assessment_async,
     modify_message_async,
+    stream_complete_with_references,
+    stream_complete_with_references_async,
+    update_message_execution,
+    update_message_execution_async,
+    update_message_log,
+    update_message_log_async,
 )
 from unique_toolkit.chat.schemas import (
     ChatMessage,
@@ -30,8 +47,24 @@ from unique_toolkit.chat.schemas import (
     ChatMessageAssessmentStatus,
     ChatMessageAssessmentType,
     ChatMessageRole,
+    MessageExecution,
+    MessageExecutionType,
+    MessageExecutionUpdateStatus,
+    MessageLog,
+    MessageLogDetails,
+    MessageLogStatus,
+    MessageLogUncitedReferences,
+)
+from unique_toolkit.content.functions import (
+    download_content_to_bytes,
+    search_contents,
+    upload_content_from_bytes,
+)
+from unique_toolkit.content.schemas import (
+    Content,
+    ContentChunk,
+    ContentReference,
 )
-from unique_toolkit.content.schemas import ContentChunk, ContentReference
 from unique_toolkit.language_model.constants import (
     DEFAULT_COMPLETE_TEMPERATURE,
     DEFAULT_COMPLETE_TIMEOUT,
@@ -41,769 +74,84 @@ from unique_toolkit.language_model.infos import (
 )
 from unique_toolkit.language_model.schemas import (
     LanguageModelMessages,
+    LanguageModelResponse,
     LanguageModelStreamResponse,
     LanguageModelTool,
+    LanguageModelToolDescription,
 )
-
-from .functions import (
-    stream_complete_to_chat,
-    stream_complete_to_chat_async,
+from unique_toolkit.services.chat_service import ChatService
+
+warnings.warn(
+    "Importing from 'unique_toolkit.chat.service' is deprecated. "
+    "Please import from 'unique_toolkit.services.chat_service' instead. "
+    "This module will be removed in a future version.",
+    DeprecationWarning,
+    stacklevel=1,
 )
 
-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):
-        self._event = event
-        self.company_id = event.company_id
-        self.user_id = event.user_id
-        self.assistant_message_id = event.payload.assistant_message.id
-        self.user_message_id = event.payload.user_message.id
-        self.chat_id = event.payload.chat_id
-        self.assistant_id = event.payload.assistant_id
-        self.user_message_text = event.payload.user_message.text
-
-    @property
-    @deprecated(
-        "The event property is deprecated and will be removed in a future version."
-    )
-    def event(self) -> Event | ChatEvent | None:
-        """
-        Get the event object (deprecated).
-
-        Returns:
-            Event | BaseEvent | None: The event object.
-        """
-        return self._event
-
-    async def update_debug_info_async(self, debug_info: dict):
-        """
-        Updates the debug information for the chat session.
-
-        Args:
-            debug_info (dict): The new debug information.
-        """
-
-        return await modify_message_async(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            assistant_message_id=self.assistant_message_id,
-            chat_id=self.chat_id,
-            user_message_id=self.user_message_id,
-            user_message_text=self.user_message_text,
-            assistant=False,
-            debug_info=debug_info,
-        )
-
-    def update_debug_info(self, debug_info: dict):
-        """
-        Updates the debug information for the chat session.
-
-        Args:
-            debug_info (dict): The new debug information.
-        """
-
-        return modify_message(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            assistant_message_id=self.assistant_message_id,
-            chat_id=self.chat_id,
-            user_message_id=self.user_message_id,
-            user_message_text=self.user_message_text,
-            assistant=False,
-            debug_info=debug_info,
-        )
-
-    def modify_user_message(
-        self,
-        content: str,
-        references: list[ContentReference] = [],
-        debug_info: dict = {},
-        message_id: str | None = None,
-        set_completed_at: bool | None = False,
-    ) -> ChatMessage:
-        """
-        Modifies a user message in the chat session synchronously.
-
-        Args:
-            content (str): The new content for the message.
-            references (list[ContentReference]): list of ContentReference objects. Defaults to [].
-            debug_info (dict[str, Any]]]): Debug information. Defaults to {}.
-            message_id (str, optional): The message ID. Defaults to None, then the ChatState user message id is used.
-            set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
-
-        Returns:
-            ChatMessage: The modified message.
-
-        Raises:
-            Exception: If the modification fails.
-        """
-        return modify_message(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            assistant_message_id=self.assistant_message_id,
-            chat_id=self.chat_id,
-            user_message_id=self.user_message_id,
-            user_message_text=self.user_message_text,
-            assistant=False,
-            content=content,
-            references=references,
-            debug_info=debug_info,
-            message_id=message_id,
-            set_completed_at=set_completed_at or False,
-        )
-
-    async def modify_user_message_async(
-        self,
-        content: str,
-        references: list[ContentReference] = [],
-        debug_info: dict = {},
-        message_id: str | None = None,
-        set_completed_at: bool | None = False,
-    ) -> ChatMessage:
-        """
-        Modifies a message in the chat session asynchronously.
-
-        Args:
-            content (str): The new content for the message.
-            message_id (str, optional): The message ID. Defaults to None, then the ChatState user message id is used.
-            references (list[ContentReference]): list of ContentReference objects. Defaults to None.
-            debug_info (dict[str, Any]]]): Debug information. Defaults to {}.
-            set_completed_at (bool, optional): Whether to set the completedAt field with the current date time. Defaults to False.
-
-        Returns:
-            ChatMessage: The modified message.
-
-        Raises:
-            Exception: If the modification fails.
-        """
-        return await modify_message_async(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            assistant_message_id=self.assistant_message_id,
-            chat_id=self.chat_id,
-            user_message_id=self.user_message_id,
-            user_message_text=self.user_message_text,
-            assistant=False,
-            content=content,
-            references=references,
-            debug_info=debug_info,
-            message_id=message_id,
-            set_completed_at=set_completed_at or False,
-        )
-
-    def modify_assistant_message(
-        self,
-        content: str | None = None,
-        original_content: str | None = None,
-        references: list[ContentReference] = [],
-        debug_info: dict = {},
-        message_id: str | None = None,
-        set_completed_at: bool | None = False,
-    ) -> ChatMessage:
-        """
-        Modifies a message in the chat session synchronously.
-
-        Args:
-            content (str, optional): The new content for the message.
-            original_content (str, optional): The original content for the message.
-            references (list[ContentReference]): list of ContentReference objects. Defaults to [].
-            debug_info (dict[str, Any]]]): Debug information. Defaults to {}.
-            message_id (Optional[str]): The message ID. Defaults to None.
-            set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
-
-        Returns:
-            ChatMessage: The modified message.
-
-        Raises:
-            Exception: If the modification fails.
-        """
-        return modify_message(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            assistant_message_id=self.assistant_message_id,
-            chat_id=self.chat_id,
-            user_message_id=self.user_message_id,
-            user_message_text=self.user_message_text,
-            assistant=True,
-            content=content,
-            original_content=original_content,
-            references=references,
-            debug_info=debug_info,
-            message_id=message_id,
-            set_completed_at=set_completed_at or False,
-        )
-
-    async def modify_assistant_message_async(
-        self,
-        content: str | None = None,
-        original_content: str | None = None,
-        references: list[ContentReference] = [],
-        debug_info: dict = {},
-        message_id: str | None = None,
-        set_completed_at: bool | None = False,
-    ) -> ChatMessage:
-        """
-        Modifies a message in the chat session asynchronously.
-
-        Args:
-            content (str, optional): The new content for the message.
-            original_content (str, optional): The original content for the message.
-            message_id (str, optional): The message ID. Defaults to None, then the ChatState assistant message id is used.
-            references (list[ContentReference]): list of ContentReference objects. Defaults to None.
-            debug_info (dict[str, Any]], optional): Debug information. Defaults to None.
-            set_completed_at (bool, optional): Whether to set the completedAt field with the current date time. Defaults to False.
-
-        Returns:
-            ChatMessage: The modified message.
-
-        Raises:
-            Exception: If the modification fails.
-        """
-
-        return await modify_message_async(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            assistant_message_id=self.assistant_message_id,
-            chat_id=self.chat_id,
-            user_message_id=self.user_message_id,
-            user_message_text=self.user_message_text,
-            assistant=True,
-            content=content,
-            original_content=original_content,
-            references=references,
-            debug_info=debug_info,
-            message_id=message_id,
-            set_completed_at=set_completed_at or False,
-        )
-
-    def get_full_history(self) -> list[ChatMessage]:
-        """
-        Loads the full chat history for the chat session synchronously.
-
-        Returns:
-            list[ChatMessage]: The full chat history.
-
-        Raises:
-            Exception: If the loading fails.
-        """
-        return get_full_history(
-            event_user_id=self.user_id,
-            event_company_id=self.company_id,
-            event_payload_chat_id=self.chat_id,
-        )
-
-    async def get_full_history_async(self) -> list[ChatMessage]:
-        """
-        Loads the full chat history for the chat session asynchronously.
-
-        Returns:
-            list[ChatMessage]: The full chat history.
-
-        Raises:
-            Exception: If the loading fails.
-        """
-        return await get_full_history_async(
-            event_user_id=self.user_id,
-            event_company_id=self.company_id,
-            event_payload_chat_id=self.chat_id,
-        )
-
-    def get_full_and_selected_history(
-        self,
-        token_limit: int,
-        percent_of_max_tokens: float = DEFAULT_PERCENT_OF_MAX_TOKENS,
-        max_messages: int = DEFAULT_MAX_MESSAGES,
-    ) -> tuple[list[ChatMessage], list[ChatMessage]]:
-        """
-        Loads the chat history for the chat session synchronously.
-
-        Args:
-            token_limit (int): The maximum number of tokens to load.
-            percent_of_max_tokens (float): The percentage of the maximum tokens to load. Defaults to 0.15.
-            max_messages (int): The maximum number of messages to load. Defaults to 4.
-
-        Returns:
-            tuple[list[ChatMessage], list[ChatMessage]]: The selected and full chat history.
-
-        Raises:
-            Exception: If the loading fails.
-        """
-        full_history = get_full_history(
-            event_user_id=self.user_id,
-            event_company_id=self.company_id,
-            event_payload_chat_id=self.chat_id,
-        )
-        selected_history = get_selection_from_history(
-            full_history=full_history,
-            max_tokens=int(round(token_limit * percent_of_max_tokens)),
-            max_messages=max_messages,
-        )
-
-        return full_history, selected_history
-
-    async def get_full_and_selected_history_async(
-        self,
-        token_limit: int,
-        percent_of_max_tokens: float = DEFAULT_PERCENT_OF_MAX_TOKENS,
-        max_messages: int = DEFAULT_MAX_MESSAGES,
-    ) -> tuple[list[ChatMessage], list[ChatMessage]]:
-        """
-        Loads the chat history for the chat session asynchronously.
-
-        Args:
-            token_limit (int): The maximum number of tokens to load.
-            percent_of_max_tokens (float): The percentage of the maximum tokens to load. Defaults to 0.15.
-            max_messages (int): The maximum number of messages to load. Defaults to 4.
-
-        Returns:
-            tuple[list[ChatMessage], list[ChatMessage]]: The selected and full chat history.
-
-        Raises:
-            Exception: If the loading fails.
-        """
-        full_history = await get_full_history_async(
-            event_user_id=self.user_id,
-            event_company_id=self.company_id,
-            event_payload_chat_id=self.chat_id,
-        )
-        selected_history = get_selection_from_history(
-            full_history=full_history,
-            max_tokens=int(round(token_limit * percent_of_max_tokens)),
-            max_messages=max_messages,
-        )
-
-        return full_history, selected_history
-
-    def create_assistant_message(
-        self,
-        content: str,
-        original_content: str | None = None,
-        references: list[ContentReference] = [],
-        debug_info: dict = {},
-        set_completed_at: bool | None = False,
-    ) -> ChatMessage:
-        """
-        Creates a message in the chat session synchronously.
-
-        Args:
-            content (str): The content for the message.
-            original_content (str, optional): The original content for the message.
-            references (list[ContentReference]): list of ContentReference objects. Defaults to None.
-            debug_info (dict[str, Any]]): Debug information. Defaults to None.
-            set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
-
-        Returns:
-            ChatMessage: The created message.
-
-        Raises:
-            Exception: If the creation fails.
-        """
-        chat_message = create_message(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            chat_id=self.chat_id,
-            assistant_id=self.assistant_id,
-            role=ChatMessageRole.ASSISTANT,
-            content=content,
-            original_content=original_content,
-            references=references,
-            debug_info=debug_info,
-            set_completed_at=set_completed_at,
-        )
-        # Update the assistant message id
-        self.assistant_message_id = chat_message.id
-        return chat_message
-
-    async def create_assistant_message_async(
-        self,
-        content: str,
-        original_content: str | None = None,
-        references: list[ContentReference] = [],
-        debug_info: dict = {},
-        set_completed_at: bool | None = False,
-    ) -> ChatMessage:
-        """
-        Creates a message in the chat session asynchronously.
-
-        Args:
-            content (str): The content for the message.
-            original_content (str, optional): The original content for the message.
-            references (list[ContentReference]): list of references. Defaults to None.
-            debug_info (dict[str, Any]]): Debug information. Defaults to None.
-            set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
-
-        Returns:
-            ChatMessage: The created message.
-
-        Raises:
-            Exception: If the creation fails.
-        """
-
-        chat_message = await create_message_async(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            chat_id=self.chat_id,
-            assistant_id=self.assistant_id,
-            role=ChatMessageRole.ASSISTANT,
-            content=content,
-            original_content=original_content,
-            references=references,
-            debug_info=debug_info,
-            set_completed_at=set_completed_at,
-        )
-        # Update the assistant message id
-        self.assistant_message_id = chat_message.id
-        return chat_message
-
-    def create_user_message(
-        self,
-        content: str,
-        original_content: str | None = None,
-        references: list[ContentReference] = [],
-        debug_info: dict = {},
-        set_completed_at: bool | None = False,
-    ) -> ChatMessage:
-        """
-        Creates a user message in the chat session synchronously.
-
-        Args:
-            content (str): The content for the message.
-            original_content (str, optional): The original content for the message.
-            references (list[ContentReference]): list of ContentReference objects. Defaults to None.
-            debug_info (dict[str, Any]]): Debug information. Defaults to None.
-            set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
-
-        Returns:
-            ChatMessage: The created message.
-
-        Raises:
-            Exception: If the creation fails.
-        """
-        chat_message = create_message(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            chat_id=self.chat_id,
-            assistant_id=self.assistant_id,
-            role=ChatMessageRole.USER,
-            content=content,
-            original_content=original_content,
-            references=references,
-            debug_info=debug_info,
-            set_completed_at=set_completed_at,
-        )
-        # Update the user message id
-        self.user_message_id = chat_message.id
-        return chat_message
-
-    async def create_user_message_async(
-        self,
-        content: str,
-        original_content: str | None = None,
-        references: list[ContentReference] = [],
-        debug_info: dict = {},
-        set_completed_at: bool | None = False,
-    ) -> ChatMessage:
-        """
-        Creates a user message in the chat session asynchronously.
-
-        Args:
-            content (str): The content for the message.
-            original_content (str, optional): The original content for the message.
-            references (list[ContentReference]): list of references. Defaults to None.
-            debug_info (dict[str, Any]]): Debug information. Defaults to None.
-            set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
-
-        Returns:
-            ChatMessage: The created message.
-
-        Raises:
-            Exception: If the creation fails.
-        """
-
-        chat_message = await create_message_async(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            chat_id=self.chat_id,
-            assistant_id=self.assistant_id,
-            role=ChatMessageRole.USER,
-            content=content,
-            original_content=original_content,
-            references=references,
-            debug_info=debug_info,
-            set_completed_at=set_completed_at,
-        )
-        # Update the user message id
-        self.user_message_id = chat_message.id
-        return chat_message
-
-    def create_message_assessment(
-        self,
-        assistant_message_id: str,
-        status: ChatMessageAssessmentStatus,
-        type: ChatMessageAssessmentType,
-        title: str | None = None,
-        explanation: str | None = None,
-        label: ChatMessageAssessmentLabel | None = None,
-        is_visible: bool = True,
-    ) -> ChatMessageAssessment:
-        """
-        Creates a message assessment for an assistant message synchronously.
-
-        Args:
-            assistant_message_id (str): The ID of the assistant message to assess
-            status (MessageAssessmentStatus): The status of the assessment (e.g. "DONE")
-            type (MessageAssessmentType): The type of assessment (e.g. "HALLUCINATION")
-            title (str | None): The title of the assessment
-            explanation (str | None): Explanation of the assessment
-            label (MessageAssessmentLabel | None): The assessment label (e.g. "RED")
-            is_visible (bool): Whether the assessment is visible to users. Defaults to True.
-
-        Returns:
-            ChatMessageAssessment: The created message assessment
-
-        Raises:
-            Exception: If the creation fails
-        """
-        return create_message_assessment(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            assistant_message_id=assistant_message_id,
-            status=status,
-            type=type,
-            title=title,
-            explanation=explanation,
-            label=label,
-            is_visible=is_visible,
-        )
-
-    async def create_message_assessment_async(
-        self,
-        assistant_message_id: str,
-        status: ChatMessageAssessmentStatus,
-        type: ChatMessageAssessmentType,
-        title: str | None = None,
-        explanation: str | None = None,
-        label: ChatMessageAssessmentLabel | None = None,
-        is_visible: bool = True,
-    ) -> ChatMessageAssessment:
-        """
-        Creates a message assessment for an assistant message asynchronously.
-
-        Args:
-            assistant_message_id (str): The ID of the assistant message to assess
-            status (ChatMessageAssessmentStatus): The status of the assessment (e.g. "DONE")
-            type (ChatMessageAssessmentType): The type of assessment (e.g. "HALLUCINATION")
-            title (str | None): The title of the assessment
-            explanation (str | None): Explanation of the assessment
-            label (ChatMessageAssessmentLabel | None): The assessment label (e.g. "RED")
-            is_visible (bool): Whether the assessment is visible to users. Defaults to True.
-
-        Returns:
-            ChatMessageAssessment: The created message assessment
-
-        Raises:
-            Exception: If the creation fails
-        """
-        return await create_message_assessment_async(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            assistant_message_id=assistant_message_id,
-            status=status,
-            type=type,
-            title=title,
-            explanation=explanation,
-            label=label,
-            is_visible=is_visible,
-        )
-
-    def modify_message_assessment(
-        self,
-        assistant_message_id: str,
-        status: ChatMessageAssessmentStatus,
-        type: ChatMessageAssessmentType,
-        title: str | None = None,
-        explanation: str | None = None,
-        label: ChatMessageAssessmentLabel | None = None,
-    ) -> ChatMessageAssessment:
-        """
-        Modifies a message assessment for an assistant message synchronously.
-
-        Args:
-            assistant_message_id (str): The ID of the assistant message to assess
-            status (MessageAssessmentStatus): The status of the assessment (e.g. "DONE")
-            title (str | None): The title of the assessment
-            explanation (str | None): Explanation of the assessment
-            label (ChatMessageAssessmentLabel | None): The assessment label (e.g. "RED")
-            type (ChatMessageAssessmentType): The type of assessment (e.g. "HALLUCINATION")
-
-        Returns:
-            dict: The modified message assessment
-
-        Raises:
-            Exception: If the modification fails
-        """
-        return modify_message_assessment(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            assistant_message_id=assistant_message_id,
-            status=status,
-            type=type,
-            title=title,
-            explanation=explanation,
-            label=label,
-        )
-
-    async def modify_message_assessment_async(
-        self,
-        assistant_message_id: str,
-        type: ChatMessageAssessmentType,
-        title: str | None = None,
-        status: ChatMessageAssessmentStatus | None = None,
-        explanation: str | None = None,
-        label: ChatMessageAssessmentLabel | None = None,
-    ) -> ChatMessageAssessment:
-        """
-        Modifies a message assessment for an assistant message asynchronously.
-
-        Args:
-            assistant_message_id (str): The ID of the assistant message to assess
-            status (ChatMessageAssessmentStatus): The status of the assessment (e.g. "DONE")
-            title (str | None): The title of the assessment
-            explanation (str | None): Explanation of the assessment
-            label (ChatMessageAssessmentLabel | None): The assessment label (e.g. "RED")
-            type (ChatMessageAssessmentType): The type of assessment (e.g. "HALLUCINATION")
-
-        Returns:
-            ChatMessageAssessment: The modified message assessment
-
-        Raises:
-            Exception: If the modification fails
-        """
-        return await modify_message_assessment_async(
-            user_id=self.user_id,
-            company_id=self.company_id,
-            assistant_message_id=assistant_message_id,
-            status=status,
-            type=type,
-            title=title,
-            explanation=explanation,
-            label=label,
-        )
-
-    def stream_complete(
-        self,
-        messages: LanguageModelMessages,
-        model_name: LanguageModelName | str,
-        content_chunks: list[ContentChunk] = [],
-        debug_info: dict = {},
-        temperature: float = DEFAULT_COMPLETE_TEMPERATURE,
-        timeout: int = DEFAULT_COMPLETE_TIMEOUT,
-        tools: Optional[list[LanguageModelTool]] = None,
-        start_text: Optional[str] = None,
-        other_options: Optional[dict] = None,
-    ) -> LanguageModelStreamResponse:
-        """
-        Streams a completion in the chat session synchronously.
-        """
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            assistant_id,
-        ] = validate_required_values(
-            [
-                self.company_id,
-                self.user_id,
-                self.assistant_message_id,
-                self.user_message_id,
-                self.chat_id,
-                self.assistant_id,
-            ]
-        )
-
-        return stream_complete_to_chat(
-            company_id=company_id,
-            user_id=user_id,
-            assistant_message_id=assistant_message_id,
-            user_message_id=user_message_id,
-            chat_id=chat_id,
-            assistant_id=assistant_id,
-            messages=messages,
-            model_name=model_name,
-            content_chunks=content_chunks,
-            debug_info=debug_info,
-            temperature=temperature,
-            timeout=timeout,
-            tools=tools,
-            start_text=start_text,
-            other_options=other_options,
-        )
-
-    async def stream_complete_async(
-        self,
-        messages: LanguageModelMessages,
-        model_name: LanguageModelName | str,
-        content_chunks: list[ContentChunk] = [],
-        debug_info: dict = {},
-        temperature: float = DEFAULT_COMPLETE_TEMPERATURE,
-        timeout: int = DEFAULT_COMPLETE_TIMEOUT,
-        tools: Optional[list[LanguageModelTool]] = None,
-        start_text: Optional[str] = None,
-        other_options: Optional[dict] = None,
-    ) -> LanguageModelStreamResponse:
-        """
-        Streams a completion in the chat session asynchronously.
-        """
-
-        [
-            company_id,
-            user_id,
-            assistant_message_id,
-            user_message_id,
-            chat_id,
-            assistant_id,
-        ] = validate_required_values(
-            [
-                self.company_id,
-                self.user_id,
-                self.assistant_message_id,
-                self.user_message_id,
-                self.chat_id,
-                self.assistant_id,
-            ]
-        )
-
-        return await stream_complete_to_chat_async(
-            company_id=company_id,
-            user_id=user_id,
-            assistant_message_id=assistant_message_id,
-            user_message_id=user_message_id,
-            chat_id=chat_id,
-            assistant_id=assistant_id,
-            messages=messages,
-            model_name=model_name,
-            content_chunks=content_chunks,
-            debug_info=debug_info,
-            temperature=temperature,
-            timeout=timeout,
-            tools=tools,
-            start_text=start_text,
-            other_options=other_options,
-        )
+__all__ = [
+    # Main Service Class
+    "ChatService",
+    "ChatServiceDeprecated",
+    # Chat Functions
+    "create_message",
+    "create_message_assessment",
+    "create_message_assessment_async",
+    "create_message_async",
+    "create_message_execution",
+    "create_message_execution_async",
+    "create_message_log",
+    "create_message_log_async",
+    "get_full_history",
+    "get_full_history_async",
+    "get_message_execution",
+    "get_message_execution_async",
+    "get_selection_from_history",
+    "modify_message",
+    "modify_message_assessment",
+    "modify_message_assessment_async",
+    "modify_message_async",
+    "stream_complete_with_references",
+    "stream_complete_with_references_async",
+    "update_message_execution",
+    "update_message_execution_async",
+    "update_message_log",
+    "update_message_log_async",
+    # Chat Constants
+    "DEFAULT_MAX_MESSAGES",
+    "DEFAULT_PERCENT_OF_MAX_TOKENS",
+    "DOMAIN_NAME",
+    # Chat Schemas
+    "ChatMessage",
+    "ChatMessageAssessment",
+    "ChatMessageAssessmentLabel",
+    "ChatMessageAssessmentStatus",
+    "ChatMessageAssessmentType",
+    "ChatMessageRole",
+    "MessageExecution",
+    "MessageExecutionType",
+    "MessageExecutionUpdateStatus",
+    "MessageLog",
+    "MessageLogDetails",
+    "MessageLogStatus",
+    "MessageLogUncitedReferences",
+    # Content Functions
+    "download_content_to_bytes",
+    "search_contents",
+    "upload_content_from_bytes",
+    # Content Schemas
+    "Content",
+    "ContentChunk",
+    "ContentReference",
+    # Language Model Constants
+    "DEFAULT_COMPLETE_TEMPERATURE",
+    "DEFAULT_COMPLETE_TIMEOUT",
+    # Language Model Infos
+    "LanguageModelName",
+    # Language Model Schemas
+    "LanguageModelMessages",
+    "LanguageModelResponse",
+    "LanguageModelStreamResponse",
+    "LanguageModelTool",
+    "LanguageModelToolDescription",
+]