unique_toolkit 1.14.2__py3-none-any.whl → 1.14.4__py3-none-any.whl

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