Skip to content

Commit 2f2eff8

Browse files
authored
use token manager credential for RAI service calls (#41200)
* use token manager credential for RAI service calls * upper bound azure-ai-projects for CI
1 parent 4a3aba2 commit 2f2eff8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/red_team/_red_team.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ def __init__(
243243
self.scan_id = None
244244
self.scan_output_dir = None
245245

246-
self.generated_rai_client = GeneratedRAIClient(azure_ai_project=self.azure_ai_project, token_manager=self.token_manager.get_aad_credential()) #type: ignore
246+
self.generated_rai_client = GeneratedRAIClient(azure_ai_project=self.azure_ai_project, token_manager=self.token_manager.credential) #type: ignore
247247

248248
# Initialize a cache for attack objectives by risk category and strategy
249249
self.attack_objectives = {}

sdk/evaluation/azure-ai-evaluation/dev_requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,6 @@ pytest-cov
77
pytest-mock
88
pytest-xdist
99
azure-ai-inference>=1.0.0b4
10-
azure-ai-projects
10+
azure-ai-projects<=1.0.0b10
1111
aiohttp
1212
-e ../azure-ai-evaluation

0 commit comments

Comments
 (0)