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..645fb41 100644 --- a/allms/models/azure_openai.py +++ b/allms/models/azure_openai.py @@ -40,6 +40,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