File tree 1 file changed +6
-0
lines changed
sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/red_team
1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -264,6 +264,10 @@ async def _log_redteam_results_to_mlflow(
264
264
if data_only :
265
265
f .write (json .dumps ({"conversations" : redteam_output .attack_details or []}))
266
266
elif redteam_output .scan_result :
267
+ redteam_output .scan_result ["redteaming_scorecard" ] = redteam_output .scan_result .get ("scorecard" , None )
268
+ redteam_output .scan_result ["redteaming_parameters" ] = redteam_output .scan_result .get ("parameters" , None )
269
+ redteam_output .scan_result ["redteaming_data" ] = redteam_output .scan_result .get ("attack_details" , None )
270
+
267
271
json .dump (redteam_output .scan_result , f )
268
272
269
273
# Copy all relevant files to the temp directory
@@ -276,6 +280,8 @@ async def _log_redteam_results_to_mlflow(
276
280
continue
277
281
if file .endswith ('.log' ) and not os .environ .get ('DEBUG' ):
278
282
continue
283
+ if file == artifact_name or file == eval_info_name :
284
+ continue
279
285
280
286
try :
281
287
shutil .copy (file_path , os .path .join (tmpdir , file ))
You can’t perform that action at this time.
0 commit comments