unique_toolkit 1.18.1__py3-none-any.whl → 1.19.0__py3-none-any.whl

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

{unique_toolkit-1.18.1.dist-info → unique_toolkit-1.19.0.dist-info}/METADATA

@@ -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.dist-info → unique_toolkit-1.19.0.dist-info}/RECORD

@@ -136,10 +136,10 @@ unique_toolkit/embedding/schemas.py,sha256=1GvKCaSk4jixzVQ2PKq8yDqwGEVY_hWclYtoA
 unique_toolkit/embedding/service.py,sha256=XViQCbjG9yznIFhWjgBkla_4s7zM2lO12Cqob1FU63k,5283
 unique_toolkit/embedding/utils.py,sha256=v86lo__bCJbxZBQ3OcLu5SuwT6NbFfWlcq8iyk6BuzQ,279
 unique_toolkit/framework_utilities/__init__.py,sha256=fvAn9y4MRL1JgoO14ufQtLVRPRHn4jP07XRqt-TItCA,68
-unique_toolkit/framework_utilities/langchain/client.py,sha256=RwVZTH5SZzwwLjP5bZJv419EQyQyVUFJNZifERcBCxs,1870
+unique_toolkit/framework_utilities/langchain/client.py,sha256=NxOLnzVMJds2JFMm5ZqTVTz1WMntrmWsLPU4-kwlzqE,2065
 unique_toolkit/framework_utilities/langchain/history.py,sha256=R9RuCeSFNaUO3OZ0G_LmIC4gmOCIANcl91MfyWLnZ1c,650
 unique_toolkit/framework_utilities/openai/__init__.py,sha256=CrHYoC7lv2pBscitLerAFweqy5jh1R671LA_jZQ4lWo,232
-unique_toolkit/framework_utilities/openai/client.py,sha256=ct1cqPcIK1wPl11G9sJV39ZnLJwKr2kDUDSra0FjvtM,2007
+unique_toolkit/framework_utilities/openai/client.py,sha256=XtueQu_4tiar7lvu0yOOgU5QSE_cIXGiiHStesW4iXs,2317
 unique_toolkit/framework_utilities/openai/message_builder.py,sha256=RT1pZjxH42TFZlAxQ5zlqdKPvHKVTjc5t3JDUy58I7Q,6887
 unique_toolkit/framework_utilities/utils.py,sha256=JK7g2yMfEx3eMprug26769xqNpS5WJcizf8n2zWMBng,789
 unique_toolkit/language_model/__init__.py,sha256=lRQyLlbwHbNFf4-0foBU13UGb09lwEeodbVsfsSgaCk,1971
@@ -165,7 +165,7 @@ unique_toolkit/short_term_memory/service.py,sha256=5PeVBu1ZCAfyDb2HLVvlmqSbyzBBu
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=Ozhh70qCn2yOzRWr9d8WmJeTo7AQurwd3tStgBMPFLA,1246
 unique_toolkit/test_utilities/events.py,sha256=_mwV2bs5iLjxS1ynDCjaIq-gjjKhXYCK-iy3dRfvO3g,6410
-unique_toolkit-1.18.1.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.18.1.dist-info/METADATA,sha256=JM7lTRErEGHbrIrH_VsVvp4NC2snqXYAzq89D6qOrF8,38686
-unique_toolkit-1.18.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.18.1.dist-info/RECORD,,
+unique_toolkit-1.19.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.19.0.dist-info/METADATA,sha256=vFbZDvCyQYheZ7FvASbBZh5M7PqjI4GoRVqDbGkcAKY,38772
+unique_toolkit-1.19.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.19.0.dist-info/RECORD,,