unique_toolkit 0.7.38__py3-none-any.whl → 0.7.40__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/language_model/infos.py

@@ -36,7 +36,6 @@ class LanguageModelName(StrEnum):
     GEMINI_2_0_FLASH = "litellm:gemini-2-0-flash"
     GEMINI_2_5_FLASH = "litellm:gemini-2-5-flash"
     GEMINI_2_5_FLASH_LITE_PREVIEW_0617 = "litellm:gemini-2-5-flash-lite-preview-06-17"
-    GEMINI_2_5_FLASH_PREVIEW_0417 = "litellm:gemini-2-5-flash-preview-04-17"
     GEMINI_2_5_FLASH_PREVIEW_0520 = "litellm:gemini-2-5-flash-preview-05-20"
     GEMINI_2_5_PRO = "litellm:gemini-2-5-pro"
     GEMINI_2_5_PRO_EXP_0325 = "litellm:gemini-2-5-pro-exp-03-25"
@@ -548,25 +547,6 @@ class LanguageModelInfo(BaseModel):
                     info_cutoff_at=date(2025, 1, day=1),
                     published_at=date(2025, 6, 17),
                 )
-            case LanguageModelName.GEMINI_2_5_FLASH_PREVIEW_0417:
-                return cls(
-                    name=model_name,
-                    capabilities=[
-                        ModelCapabilities.FUNCTION_CALLING,
-                        ModelCapabilities.STREAMING,
-                        ModelCapabilities.VISION,
-                        ModelCapabilities.STRUCTURED_OUTPUT,
-                        ModelCapabilities.REASONING,
-                    ],
-                    provider=LanguageModelProvider.LITELLM,
-                    version="gemini-2-5-flash-preview-04-17",
-                    encoder_name=EncoderName.O200K_BASE,  # TODO:Replace with LLM tokenizer
-                    token_limits=LanguageModelTokenLimits(
-                        token_limit_input=1_048_576, token_limit_output=65_536
-                    ),
-                    info_cutoff_at=date(2025, 1, day=1),
-                    published_at=date(2025, 4, 1),
-                )
             case LanguageModelName.GEMINI_2_5_FLASH_PREVIEW_0520:
                 return cls(
                     name=model_name,

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.7.38
+Version: 0.7.40
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -113,6 +113,13 @@ 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.40] - 2025-07-30
+- Remove `GEMINI_2_5_FLASH_PREVIEW_0417` model
+
+## [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.40.dist-info}/RECORD

@@ -47,12 +47,16 @@ 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
 unique_toolkit/language_model/functions.py,sha256=WhgHbJgz4Z2aZt9TLdOpI0PGyYWA5R90tdwkwdDeT8c,11987
-unique_toolkit/language_model/infos.py,sha256=w5__BVG-IiiEYKG1FwM838wzqNbYI3eCCEDocKezc0I,34801
+unique_toolkit/language_model/infos.py,sha256=Abxcw-l7UXkBohxZsigpNz0OUUCtDf404LLr7R-ys4E,33757
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
 unique_toolkit/language_model/reference.py,sha256=nkX2VFz-IrUz8yqyc3G5jUMNwrNpxITBrMEKkbqqYoI,8583
 unique_toolkit/language_model/schemas.py,sha256=AeuDRJFblGzEYcEMyrlxpOPk12Di3J45I9rT2xZrhEU,14332
@@ -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.40.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.7.40.dist-info/METADATA,sha256=gkt3A6p7LNzXPTnDq6MnhoL-r9G_E_WahX6rAzDOXcE,25466
+unique_toolkit-0.7.40.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.7.40.dist-info/RECORD,,