unique_toolkit 1.14.1__tar.gz → 1.14.3__tar.gz

{unique_toolkit-1.14.1 → unique_toolkit-1.14.3}/CHANGELOG.md

@@ -5,6 +5,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).
 
+## [1.14.3] - 2025-10-08
+- Reorganizes documentation 
+- All service interface towards a single folder
+- Add main imports to __init__
+
+## [1.14.2] - 2025-10-08
+- Add utilities for testing and fix external import issue
+
 ## [1.14.1] - 2025-10-08
 - Add utilities for testing
 

{unique_toolkit-1.14.1 → unique_toolkit-1.14.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.14.1
+Version: 1.14.3
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,6 +118,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).
 
+## [1.14.3] - 2025-10-08
+- Reorganizes documentation 
+- All service interface towards a single folder
+- Add main imports to __init__
+
+## [1.14.2] - 2025-10-08
+- Add utilities for testing and fix external import issue
+
 ## [1.14.1] - 2025-10-08
 - Add utilities for testing
 

{unique_toolkit-1.14.1 → unique_toolkit-1.14.3}/pyproject.toml

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

{unique_toolkit-1.14.1 → unique_toolkit-1.14.3}/unique_toolkit/__init__.py

@@ -2,15 +2,27 @@
 from unique_toolkit.chat import ChatService
 from unique_toolkit.content import ContentService
 from unique_toolkit.embedding import EmbeddingService
-from unique_toolkit.knowledge_base import KnowledgeBaseService
+from unique_toolkit.framework_utilities.openai.client import (
+    get_async_openai_client,
+    get_openai_client,
+)
 from unique_toolkit.language_model import (
     LanguageModelMessages,
     LanguageModelName,
     LanguageModelService,
     LanguageModelToolDescription,
 )
+from unique_toolkit.services.knowledge_base import KnowledgeBaseService
 from unique_toolkit.short_term_memory import ShortTermMemoryService
 
+# Conditionally import langchain utilities if langchain is installed
+try:
+    from unique_toolkit.framework_utilities.langchain.client import get_langchain_client  # noqa: F401, I001
+
+    _LANGCHAIN_AVAILABLE = True
+except ImportError:
+    _LANGCHAIN_AVAILABLE = False
+
 # You can add other classes you frequently use here as well
 
 __all__ = [
@@ -23,4 +35,10 @@ __all__ = [
     "EmbeddingService",
     "ShortTermMemoryService",
     "KnowledgeBaseService",
+    "get_openai_client",
+    "get_async_openai_client",
 ]
+
+# Add langchain-specific exports if available
+if _LANGCHAIN_AVAILABLE:
+    __all__.append("get_langchain_client")

{unique_toolkit-1.14.1 → unique_toolkit-1.14.3}/unique_toolkit/chat/__init__.py

@@ -1,3 +1,5 @@
+import warnings
+
 from .constants import DOMAIN_NAME as DOMAIN_NAME
 from .schemas import ChatMessage as ChatMessage
 from .schemas import ChatMessageAssessment as ChatMessageAssessment
@@ -5,7 +7,12 @@ from .schemas import ChatMessageAssessmentLabel as ChatMessageAssessmentLabel
 from .schemas import ChatMessageAssessmentStatus as ChatMessageAssessmentStatus
 from .schemas import ChatMessageAssessmentType as ChatMessageAssessmentType
 from .schemas import ChatMessageRole as ChatMessageRole
-from .service import ChatService as ChatService
+
+# Import ChatService with deprecation warning suppressed for internal use
+with warnings.catch_warnings():
+    warnings.simplefilter("ignore", DeprecationWarning)
+    from .service import ChatService as ChatService
+
 from .utils import (
     convert_chat_history_to_injectable_string as convert_chat_history_to_injectable_string,
 )

unique_toolkit-1.14.3/unique_toolkit/chat/service.py

@@ -0,0 +1,157 @@
+"""
+DEPRECATED: This module is maintained for backward compatibility only.
+
+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.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,
+    ChatMessageAssessment,
+    ChatMessageAssessmentLabel,
+    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.language_model.constants import (
+    DEFAULT_COMPLETE_TEMPERATURE,
+    DEFAULT_COMPLETE_TIMEOUT,
+)
+from unique_toolkit.language_model.infos import (
+    LanguageModelName,
+)
+from unique_toolkit.language_model.schemas import (
+    LanguageModelMessages,
+    LanguageModelResponse,
+    LanguageModelStreamResponse,
+    LanguageModelTool,
+    LanguageModelToolDescription,
+)
+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,
+)
+
+__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",
+]

{unique_toolkit-1.14.1 → unique_toolkit-1.14.3}/unique_toolkit/framework_utilities/langchain/client.py

@@ -1,6 +1,8 @@
 import importlib.util
 import logging
 
+from typing_extensions import deprecated
+
 from unique_toolkit.app.unique_settings import UniqueSettings
 from unique_toolkit.framework_utilities.utils import get_default_headers
 
@@ -22,7 +24,7 @@ else:
     raise LangchainNotInstalledError()
 
 
-def get_client(
+def get_langchain_client(
     unique_settings: UniqueSettings | None = None, model: str = "AZURE_GPT_4o_2024_0806"
 ) -> ChatOpenAI:
     """Get a Langchain ChatOpenAI client instance.
@@ -45,3 +47,18 @@ def get_client(
         model=model,
         api_key=unique_settings.app.key,
     )
+
+
+@deprecated("Use get_langchain_client instead")
+def get_client(
+    unique_settings: UniqueSettings | None = None, model: str = "AZURE_GPT_4o_2024_0806"
+) -> ChatOpenAI:
+    """Get a Langchain ChatOpenAI client instance.
+
+    Args:
+        unique_settings: UniqueSettings instance
+
+    Returns:
+        ChatOpenAI client instance
+    """
+    return get_client(unique_settings, model)

unique_toolkit-1.14.1/unique_toolkit/chat/service.py → unique_toolkit-1.14.3/unique_toolkit/services/chat_service.py

@@ -11,6 +11,7 @@ from unique_toolkit.chat.constants import (
     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,
@@ -29,6 +30,8 @@ from unique_toolkit.chat.functions import (
     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,
@@ -74,12 +77,6 @@ from unique_toolkit.language_model.schemas import (
     LanguageModelToolDescription,
 )
 
-from .deprecated.service import ChatServiceDeprecated
-from .functions import (
-    stream_complete_with_references,
-    stream_complete_with_references_async,
-)
-
 logger = logging.getLogger(f"toolkit.{DOMAIN_NAME}.{__name__}")
 
 

{unique_toolkit-1.14.1 → unique_toolkit-1.14.3}/unique_toolkit/test_utilities/events.py

@@ -1,7 +1,7 @@
 import random
 import string
 from datetime import datetime
-from typing import TYPE_CHECKING
+from typing import Any
 
 from unique_toolkit.app.schemas import (
     BaseEvent,
@@ -12,11 +12,7 @@ from unique_toolkit.app.schemas import (
     ChatEventUserMessage,
     EventName,
 )
-
-if TYPE_CHECKING:
-    from typing import Any
-
-    from unique_toolkit.app.unique_settings import UniqueSettings
+from unique_toolkit.app.unique_settings import UniqueSettings
 
 
 def generated_numeric_string(length: int) -> str:
@@ -166,12 +162,18 @@ class TestEventFactory:
         user_message_original_text: str | None = None,
         assistant_message_created_at: datetime | None = None,
         configuration: dict[str, Any] | None = None,
-        chat_id: str = generated_chat_id(),
-        assistant_id: str = generated_assistant_id(),
+        chat_id: str | None = None,
+        assistant_id: str | None = None,
         user_id: str | None = None,
         company_id: str | None = None,
         version: str = "1.0",
     ) -> ChatEvent:
+        if chat_id is None:
+            chat_id = generated_chat_id()
+
+        if assistant_id is None:
+            assistant_id = generated_assistant_id()
+
         payload = self.get_chat_event_payload(
             name=name,
             description=description,