From 801da3929ea314f97cc1487fc102233ed9e9194c Mon Sep 17 00:00:00 2001 From: Piotr C Zielinski Date: Thu, 4 Jul 2024 13:28:08 +0200 Subject: [PATCH 1/2] Add possibility to pass azure_ad_token --- allms/domain/configuration.py | 3 ++- allms/models/azure_openai.py | 2 ++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/allms/domain/configuration.py b/allms/domain/configuration.py index b2ccf42..55ab712 100644 --- a/allms/domain/configuration.py +++ b/allms/domain/configuration.py @@ -12,7 +12,8 @@ class AzureOpenAIConfiguration: deployment: str model_name: str api_version: str - api_key: str + api_key: Optional[str] = None + azure_ad_token: Optional[str] = None @dataclass diff --git a/allms/models/azure_openai.py b/allms/models/azure_openai.py index 2007b02..3e3a037 100644 --- a/allms/models/azure_openai.py +++ b/allms/models/azure_openai.py @@ -33,6 +33,7 @@ def __init__( event_loop=event_loop ) + from langchain_community.chat_models import AzureChatOpenAI def _create_llm(self) -> AzureChatOpenAI: return AzureChatOpenAI( deployment_name=self._config.deployment, @@ -40,6 +41,7 @@ def _create_llm(self) -> AzureChatOpenAI: model_name=self._config.model_name, base_url=self._config.base_url, api_key=self._config.api_key, + azure_ad_token=self._config.azure_ad_token, temperature=self._temperature, max_tokens=self._max_output_tokens, request_timeout=self._request_timeout_s From cd44d24741a9299eb53800915f805b42393fbafd Mon Sep 17 00:00:00 2001 From: Piotr C Zielinski Date: Thu, 4 Jul 2024 13:31:10 +0200 Subject: [PATCH 2/2] Remove unnecessary line --- allms/models/azure_openai.py | 1 - 1 file changed, 1 deletion(-) diff --git a/allms/models/azure_openai.py b/allms/models/azure_openai.py index 3e3a037..645fb41 100644 --- a/allms/models/azure_openai.py +++ b/allms/models/azure_openai.py @@ -33,7 +33,6 @@ def __init__( event_loop=event_loop ) - from langchain_community.chat_models import AzureChatOpenAI def _create_llm(self) -> AzureChatOpenAI: return AzureChatOpenAI( deployment_name=self._config.deployment,