File tree Expand file tree Collapse file tree 1 file changed +6
-0
lines changed
sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/red_team Expand file tree Collapse file tree 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(
264264 if data_only :
265265 f .write (json .dumps ({"conversations" : redteam_output .attack_details or []}))
266266 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+
267271 json .dump (redteam_output .scan_result , f )
268272
269273 # Copy all relevant files to the temp directory
@@ -276,6 +280,8 @@ async def _log_redteam_results_to_mlflow(
276280 continue
277281 if file .endswith ('.log' ) and not os .environ .get ('DEBUG' ):
278282 continue
283+ if file == artifact_name or file == eval_info_name :
284+ continue
279285
280286 try :
281287 shutil .copy (file_path , os .path .join (tmpdir , file ))
You can’t perform that action at this time.
0 commit comments