unique_toolkit 0.7.38__py3-none-any.whl → 0.7.39__py3-none-any.whl

unique_toolkit/framework_utilities/langchain/client.py

@@ -0,0 +1,45 @@
+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
+
+logger = logging.getLogger("toolkit.framework_utilities.langchain")
+
+
+class LangchainNotInstalledError(ImportError):
+    """Raised when langchain-openai package is not installed but functionality requiring it is accessed."""
+
+    def __init__(self):
+        super().__init__(
+            "langchain-openai package is not installed. Install it with 'poetry install --with langchain'."
+        )
+
+
+if importlib.util.find_spec("langchain_openai") is not None:
+    from langchain_openai import ChatOpenAI
+else:
+    raise LangchainNotInstalledError()
+
+
+def get_client(env_file: Path | None = None) -> ChatOpenAI:
+    """Get a Langchain ChatOpenAI client instance.
+
+    Args:
+        env_file: Optional path to environment file
+
+    Returns:
+        ChatOpenAI client instance
+
+    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,
+    )

unique_toolkit/framework_utilities/openai/client.py

@@ -0,0 +1,45 @@
+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
+
+logger = logging.getLogger("toolkit.framework_utilities.openai")
+
+
+class OpenAINotInstalledError(ImportError):
+    """Raised when OpenAI package is not installed but functionality requiring it is accessed."""
+
+    def __init__(self):
+        super().__init__(
+            "OpenAI package is not installed. Install it with 'poetry install --with openai'."
+        )
+
+
+if importlib.util.find_spec("openai") is not None:
+    from openai import OpenAI
+else:
+    raise OpenAINotInstalledError()
+
+
+def get_openai_client(env_file: Path | None = None) -> OpenAI:
+    """Get an OpenAI client instance.
+
+    Args:
+        env_file: Optional path to environment file
+
+    Returns:
+        OpenAI client instance
+
+    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)
+
+    return OpenAI(
+        api_key=settings.app.key.get_secret_value(),
+        base_url=settings.app.base_url + "/openai-proxy/",
+        default_headers=default_headers,
+    )

unique_toolkit/framework_utilities/openai/message_builder.py

@@ -0,0 +1,133 @@
+from collections.abc import Iterable
+from typing import Self
+
+from openai.types.chat.chat_completion_assistant_message_param import (
+    ChatCompletionAssistantMessageParam,
+    ContentArrayOfContentPart,
+)
+from openai.types.chat.chat_completion_content_part_param import (
+    ChatCompletionContentPartParam,
+)
+from openai.types.chat.chat_completion_content_part_text_param import (
+    ChatCompletionContentPartTextParam,
+)
+from openai.types.chat.chat_completion_developer_message_param import (
+    ChatCompletionDeveloperMessageParam,
+)
+from openai.types.chat.chat_completion_function_message_param import (
+    ChatCompletionFunctionMessageParam,
+)
+from openai.types.chat.chat_completion_message_param import ChatCompletionMessageParam
+from openai.types.chat.chat_completion_system_message_param import (
+    ChatCompletionSystemMessageParam,
+)
+from openai.types.chat.chat_completion_tool_message_param import (
+    ChatCompletionToolMessageParam,
+)
+from openai.types.chat.chat_completion_user_message_param import (
+    ChatCompletionUserMessageParam,
+)
+
+
+class OpenAIMessageBuilder:
+    def __init__(
+        self,
+        messages: list[ChatCompletionMessageParam] | None = None,
+    ) -> None:
+        self._messages: list[ChatCompletionMessageParam] = []
+        if messages:
+            self._messages = messages
+
+    @classmethod
+    def from_messages(cls, messages: list[ChatCompletionMessageParam]) -> Self:
+        builder = cls()
+        builder._messages = messages.copy()
+        return builder
+
+    def append_system_message(
+        self,
+        content: str | Iterable[ChatCompletionContentPartTextParam],
+        name: str = "user",
+    ) -> Self:
+        self._messages.append(
+            ChatCompletionSystemMessageParam(
+                content=content,
+                role="system",
+                name=name,
+            ),
+        )
+        return self
+
+    def append_user_message(
+        self,
+        content: str | Iterable[ChatCompletionContentPartParam],
+        name: str = "user",
+    ) -> Self:
+        self._messages.append(
+            ChatCompletionUserMessageParam(
+                content=content,
+                role="user",
+                name=name,
+            ),
+        )
+        return self
+
+    def append_assistant_message(
+        self,
+        content: str | Iterable[ContentArrayOfContentPart],
+        name: str = "assistant",
+    ) -> Self:
+        self._messages.append(
+            ChatCompletionAssistantMessageParam(
+                content=content,
+                role="assistant",
+                name=name,
+            ),
+        )
+        return self
+
+    def append_developper_message(
+        self,
+        content: str | Iterable[ChatCompletionContentPartTextParam],
+        name: str = "developer",
+    ) -> Self:
+        self._messages.append(
+            ChatCompletionDeveloperMessageParam(
+                content=content,
+                role="developer",
+                name=name,
+            ),
+        )
+        return self
+
+    def append_function_message(
+        self,
+        content: str | None,
+        name: str = "function",
+    ) -> Self:
+        self._messages.append(
+            ChatCompletionFunctionMessageParam(
+                content=content,
+                role="function",
+                name=name,
+            ),
+        )
+        return self
+
+    def append_tool_message(
+        self,
+        content: str | Iterable[ChatCompletionContentPartTextParam],
+        tool_call_id: str,
+    ) -> Self:
+        self._messages.append(
+            ChatCompletionToolMessageParam(
+                content=content,
+                role="tool",
+                tool_call_id=tool_call_id,
+            ),
+        )
+        return self
+
+    @property
+    def messages(self) -> list[ChatCompletionMessageParam]:
+        return self._messages

unique_toolkit/framework_utilities/utils.py

@@ -0,0 +1,23 @@
+from unique_toolkit.app.unique_settings import UniqueApp, UniqueAuth
+
+
+def auth_headers(auth_settings: UniqueAuth) -> dict[str, str]:
+    return {
+        "x-user-id": auth_settings.user_id.get_secret_value(),
+        "x-company-id": auth_settings.company_id.get_secret_value(),
+    }
+
+
+def app_headers(app_settings: UniqueApp) -> dict[str, str]:
+    return {
+        "x-app-id": app_settings.id.get_secret_value(),
+        "Authorization": f"Bearer {app_settings.key.get_secret_value()}",
+    }
+
+
+def get_default_headers(app_settings: UniqueApp, auth_settings: UniqueAuth):
+    default_headers = {}
+    default_headers.update(app_headers(app_settings))
+    default_headers.update(auth_headers(auth_settings))
+    default_headers["x-api-version"] = "2023-12-06"
+    return default_headers

{unique_toolkit-0.7.38.dist-info → unique_toolkit-0.7.39.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.7.38
+Version: 0.7.39
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -113,6 +113,10 @@ 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.7.39] - 2025-07-28
+- Implement utitilites to work with openai client
+- Implement utitilites to work with langchain llm 
+
 ## [0.7.38] - 2025-07-25
 - Fix issues with secret strings in settings
 

{unique_toolkit-0.7.38.dist-info → unique_toolkit-0.7.39.dist-info}/RECORD

@@ -47,7 +47,11 @@ 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/history.py,sha256=8ejA0utPUIlX4z8gtX9IkVFiooY-vOT1UcjI_MoJaAU,638
+unique_toolkit/framework_utilities/openai/client.py,sha256=itbEv098j4alz2Rftp4kbjeloFsAqaxyyBHdZTH5OYM,1297
+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
 unique_toolkit/language_model/builder.py,sha256=69WCcmkm2rMP2-YEH_EjHiEp6OzwjwCs8VbhjVJaCe0,3168
 unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
@@ -66,7 +70,7 @@ unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJ
 unique_toolkit/short_term_memory/service.py,sha256=8sW7cFJRd4vcRfotJSWb0uHZYl-e5hQWp3G1SddL5Bg,8110
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=cxWjb2dxEI2HGsakKdVCkSNi7VK9mr08w5sDcFCQyWI,9553
-unique_toolkit-0.7.38.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.7.38.dist-info/METADATA,sha256=owlyNWn4pHKXlaJae2J3Vyt7ttIgQjrDGTsfaNjOrTg,25266
-unique_toolkit-0.7.38.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.7.38.dist-info/RECORD,,
+unique_toolkit-0.7.39.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.7.39.dist-info/METADATA,sha256=4vZp5-p04GYGeJBpu6vhRg2FhCvBwZGqHl8AH1dIiPI,25393
+unique_toolkit-0.7.39.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.7.39.dist-info/RECORD,,