Skip to content

Commit d3e28d0

Browse files
authored
[evaluation] refactor: Make AzureOpenAIGrader._get_token_provider private (#43218)
1 parent 8d03e35 commit d3e28d0

File tree

1 file changed

+2
-2
lines changed
  • sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_aoai

1 file changed

+2
-2
lines changed

sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_aoai/aoai_grader.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def get_client(self) -> Any:
118118
api_key=api_key, # Default-style access to appease linters.
119119
api_version=DEFAULT_AOAI_API_VERSION, # Force a known working version
120120
azure_deployment=model_config.get("azure_deployment", ""),
121-
azure_ad_token_provider=self.get_token_provider(self._credential) if not api_key else None,
121+
azure_ad_token_provider=self._get_token_provider(self._credential) if not api_key else None,
122122
default_headers=default_headers,
123123
)
124124
from openai import OpenAI
@@ -132,7 +132,7 @@ def get_client(self) -> Any:
132132
)
133133

134134
@staticmethod
135-
def get_token_provider(cred: TokenCredential) -> "AzureADTokenProvider":
135+
def _get_token_provider(cred: TokenCredential) -> "AzureADTokenProvider":
136136
"""Get the token provider the AzureOpenAI client.
137137
138138
:param TokenCredential cred: The Azure authentication credential.

0 commit comments

Comments
 (0)