From 8188c25eece03778adc80ce4b6d97e904384c209 Mon Sep 17 00:00:00 2001 From: Nour Date: Wed, 29 Jan 2025 14:35:44 +0100 Subject: [PATCH] Add default_header --- .../src/autogen_ext/models/openai/config/__init__.py | 2 +- .../autogen-ext/src/autogen_ext/models/test.py | 10 ++++++++++ 2 files changed, 11 insertions(+), 1 deletion(-) create mode 100644 python/packages/autogen-ext/src/autogen_ext/models/test.py diff --git a/python/packages/autogen-ext/src/autogen_ext/models/openai/config/__init__.py b/python/packages/autogen-ext/src/autogen_ext/models/openai/config/__init__.py index a6b484780212..de27637c4663 100644 --- a/python/packages/autogen-ext/src/autogen_ext/models/openai/config/__init__.py +++ b/python/packages/autogen-ext/src/autogen_ext/models/openai/config/__init__.py @@ -74,7 +74,7 @@ class BaseOpenAIClientConfigurationConfigModel(CreateArgumentsConfigModel): max_retries: int | None = None model_capabilities: ModelCapabilities | None = None # type: ignore model_info: ModelInfo | None = None - + default_header: Dict[str, str] | None = None # See OpenAI docs for explanation of these parameters class OpenAIClientConfigurationConfigModel(BaseOpenAIClientConfigurationConfigModel): diff --git a/python/packages/autogen-ext/src/autogen_ext/models/test.py b/python/packages/autogen-ext/src/autogen_ext/models/test.py new file mode 100644 index 000000000000..c926e5925b7d --- /dev/null +++ b/python/packages/autogen-ext/src/autogen_ext/models/test.py @@ -0,0 +1,10 @@ +from src.autogen_ext.models.openai import AzureOpenAIChatCompletionClient + +model_client = AzureOpenAIChatCompletionClient( + model="gpt-4o-2024-05-13", + azure_deployment="xxxxxx", + azure_endpoint="xxxxxx", + api_version="xxxx", + api_key=os.getenv("AZURE_API_KEY"), + default_headers={"X-User-Id": os.getlogin()}, +)