From 62505043be20cf8af491e30785a6ca0eeb1d276e Mon Sep 17 00:00:00 2001 From: Bob Lin Date: Sun, 3 Dec 2023 10:55:25 -0600 Subject: [PATCH] Closed #14069 (#14166) ### Description Fix #14069 ### Twitter handle [lin_bob57617](https://twitter.com/lin_bob57617) --- libs/langchain/langchain/chat_models/azure_openai.py | 4 ++-- libs/langchain/langchain/llms/openai.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/langchain/langchain/chat_models/azure_openai.py b/libs/langchain/langchain/chat_models/azure_openai.py index 8bad52f4011..2b9cb1b5f6d 100644 --- a/libs/langchain/langchain/chat_models/azure_openai.py +++ b/libs/langchain/langchain/chat_models/azure_openai.py @@ -4,7 +4,7 @@ from __future__ import annotations import logging import os import warnings -from typing import Any, Dict, Union +from typing import Any, Callable, Dict, Union from langchain_core.outputs import ChatResult from langchain_core.pydantic_v1 import BaseModel, Field, root_validator @@ -80,7 +80,7 @@ class AzureChatOpenAI(ChatOpenAI): For more: https://www.microsoft.com/en-us/security/business/identity-access/microsoft-entra-id. """ # noqa: E501 - azure_ad_token_provider: Union[str, None] = None + azure_ad_token_provider: Union[Callable[[], str], None] = None """A function that returns an Azure Active Directory token. Will be invoked on every request. diff --git a/libs/langchain/langchain/llms/openai.py b/libs/langchain/langchain/llms/openai.py index eca554f5571..9ee841f3d0b 100644 --- a/libs/langchain/langchain/llms/openai.py +++ b/libs/langchain/langchain/llms/openai.py @@ -782,7 +782,7 @@ class AzureOpenAI(BaseOpenAI): For more: https://www.microsoft.com/en-us/security/business/identity-access/microsoft-entra-id. """ # noqa: E501 - azure_ad_token_provider: Union[str, None] = None + azure_ad_token_provider: Union[Callable[[], str], None] = None """A function that returns an Azure Active Directory token. Will be invoked on every request.