Adding the credential to jailbreak adv sim init to make sure it can b…

…e used at call state
microsoft · Jul 9, 2024 · 490af3c · 490af3c
1 parent f2f56d3
commit 490af3c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/promptflow-evals/promptflow/evals/synthetic/jailbreak_adversarial_simulator.py b/src/promptflow-evals/promptflow/evals/synthetic/jailbreak_adversarial_simulator.py
@@ -66,6 +66,7 @@ def __init__(self, *, azure_ai_project: Dict[str, Any], credential=None):
             credential = DefaultAzureCredential()
         elif "credential" in azure_ai_project:
             credential = azure_ai_project["credential"]
+        self.credential = credential
         self.azure_ai_project = azure_ai_project
         self.token_manager = ManagedIdentityAPITokenManager(
             token_scope=TokenScope.DEFAULT_AZURE_MANAGEMENT,