unique_toolkit 1.18.1__tar.gz → 1.19.0__tar.gz

{unique_toolkit-1.18.1 → unique_toolkit-1.19.0}/CHANGELOG.md

@@ -5,6 +5,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).
 
+
+## [1.19.0] - 2025-10-28
+- Enable additional headers on openai and langchain client
+
 ## [1.18.1] - 2025-10-28
 - Fix bug where sub agent references were not properly displayed in the main agent response when the sub agent response was hidden.
 

{unique_toolkit-1.18.1 → unique_toolkit-1.19.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.18.1
+Version: 1.19.0
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,6 +118,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).
 
+
+## [1.19.0] - 2025-10-28
+- Enable additional headers on openai and langchain client
+
 ## [1.18.1] - 2025-10-28
 - Fix bug where sub agent references were not properly displayed in the main agent response when the sub agent response was hidden.
 

{unique_toolkit-1.18.1 → unique_toolkit-1.19.0}/pyproject.toml

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

{unique_toolkit-1.18.1 → unique_toolkit-1.19.0}/unique_toolkit/framework_utilities/langchain/client.py

@@ -25,7 +25,10 @@ else:
 
 
 def get_langchain_client(
-    unique_settings: UniqueSettings | None = None, model: str = "AZURE_GPT_4o_2024_0806"
+    *,
+    unique_settings: UniqueSettings | None = None,
+    model: str = "AZURE_GPT_4o_2024_0806",
+    additional_headers: dict[str, str] | None = None,
 ) -> ChatOpenAI:
     """Get a Langchain ChatOpenAI client instance.
 
@@ -41,9 +44,13 @@ def get_langchain_client(
     if unique_settings is None:
         unique_settings = UniqueSettings.from_env_auto()
 
+    default_headers = get_default_headers(unique_settings.app, unique_settings.auth)
+    if additional_headers is not None:
+        default_headers.update(additional_headers)
+
     return ChatOpenAI(
         base_url=unique_settings.api.openai_proxy_url(),
-        default_headers=get_default_headers(unique_settings.app, unique_settings.auth),
+        default_headers=default_headers,
         model=model,
         api_key=unique_settings.app.key,
     )

{unique_toolkit-1.18.1 → unique_toolkit-1.19.0}/unique_toolkit/framework_utilities/openai/client.py

@@ -22,7 +22,11 @@ else:
     raise OpenAINotInstalledError()
 
 
-def get_openai_client(unique_settings: UniqueSettings | None = None) -> OpenAI:
+def get_openai_client(
+    *,
+    unique_settings: UniqueSettings | None = None,
+    additional_headers: dict[str, str] | None = None,
+) -> OpenAI:
     """Get an OpenAI client instance.
 
     Args:
@@ -38,6 +42,8 @@ def get_openai_client(unique_settings: UniqueSettings | None = None) -> OpenAI:
         unique_settings = UniqueSettings.from_env_auto()
 
     default_headers = get_default_headers(unique_settings.app, unique_settings.auth)
+    if additional_headers is not None:
+        default_headers.update(additional_headers)
 
     return OpenAI(
         api_key="dummy_key",
@@ -47,7 +53,9 @@ def get_openai_client(unique_settings: UniqueSettings | None = None) -> OpenAI:
 
 
 def get_async_openai_client(
+    *,
     unique_settings: UniqueSettings | None = None,
+    additional_headers: dict[str, str] | None = None,
 ) -> AsyncOpenAI:
     """Get an async OpenAI client instance.
 
@@ -65,6 +73,9 @@ def get_async_openai_client(
 
     default_headers = get_default_headers(unique_settings.app, unique_settings.auth)
 
+    if additional_headers is not None:
+        default_headers.update(additional_headers)
+
     return AsyncOpenAI(
         api_key="dummy_key",
         base_url=unique_settings.api.openai_proxy_url(),