@@ -474,11 +474,11 @@ void inputFilteringStrict() {
474474 .isEqualTo (
475475 "400 - Filtering Module - Input Filter: Prompt filtered due to safety violations. Please modify the prompt and try again." );
476476
477- assertThat (e .getAzureContentSafety ()).isNotNull ();
478- assertThat (e .getAzureContentSafety ().getHate ()).isEqualTo (NUMBER_6 );
479- assertThat (e .getAzureContentSafety ().getSelfHarm ()).isEqualTo (NUMBER_0 );
480- assertThat (e .getAzureContentSafety ().getSexual ()).isEqualTo (NUMBER_0 );
481- assertThat (e .getAzureContentSafety ().getViolence ()).isEqualTo (NUMBER_6 );
477+ assertThat (e .getAzureContentSafetyInput ()).isNotNull ();
478+ assertThat (e .getAzureContentSafetyInput ().getHate ()).isEqualTo (NUMBER_6 );
479+ assertThat (e .getAzureContentSafetyInput ().getSelfHarm ()).isEqualTo (NUMBER_0 );
480+ assertThat (e .getAzureContentSafetyInput ().getSexual ()).isEqualTo (NUMBER_0 );
481+ assertThat (e .getAzureContentSafetyInput ().getViolence ()).isEqualTo (NUMBER_6 );
482482
483483 assertThat (e .getLlamaGuard38b ()).isNotNull ();
484484 assertThat (e .getLlamaGuard38b ().isViolentCrimes ()).isTrue ();
@@ -520,11 +520,11 @@ void outputFilteringStrict() {
520520 Map .of ("violent_crimes" , true )));
521521 assertThat (e .getErrorResponse ()).isNull ();
522522
523- assertThat (e .getAzureContentSafety ()).isNotNull ();
524- assertThat (e .getAzureContentSafety ().getHate ()).isEqualTo (NUMBER_6 );
525- assertThat (e .getAzureContentSafety ().getSelfHarm ()).isEqualTo (NUMBER_0 );
526- assertThat (e .getAzureContentSafety ().getSexual ()).isEqualTo (NUMBER_0 );
527- assertThat (e .getAzureContentSafety ().getViolence ()).isEqualTo (NUMBER_6 );
523+ assertThat (e .getAzureContentSafetyOutput ()).isNotNull ();
524+ assertThat (e .getAzureContentSafetyOutput ().getHate ()).isEqualTo (NUMBER_6 );
525+ assertThat (e .getAzureContentSafetyOutput ().getSelfHarm ()).isEqualTo (NUMBER_0 );
526+ assertThat (e .getAzureContentSafetyOutput ().getSexual ()).isEqualTo (NUMBER_0 );
527+ assertThat (e .getAzureContentSafetyOutput ().getViolence ()).isEqualTo (NUMBER_6 );
528528
529529 assertThat (e .getLlamaGuard38b ()).isNotNull ();
530530 assertThat (e .getLlamaGuard38b ().isViolentCrimes ()).isTrue ();
@@ -796,11 +796,11 @@ void streamChatCompletionOutputFilterErrorHandling() throws IOException {
796796 "azure_content_safety" ,
797797 Map .of ("Hate" , 0 , "SelfHarm" , 0 , "Sexual" , 0 , "Violence" , 4 )));
798798
799- assertThat (e .getAzureContentSafety ()).isNotNull ();
800- assertThat (e .getAzureContentSafety ().getHate ()).isEqualTo (NUMBER_0 );
801- assertThat (e .getAzureContentSafety ().getSelfHarm ()).isEqualTo (NUMBER_0 );
802- assertThat (e .getAzureContentSafety ().getSexual ()).isEqualTo (NUMBER_0 );
803- assertThat (e .getAzureContentSafety ().getViolence ()).isEqualTo (NUMBER_4 );
799+ assertThat (e .getAzureContentSafetyOutput ()).isNotNull ();
800+ assertThat (e .getAzureContentSafetyOutput ().getHate ()).isEqualTo (NUMBER_0 );
801+ assertThat (e .getAzureContentSafetyOutput ().getSelfHarm ()).isEqualTo (NUMBER_0 );
802+ assertThat (e .getAzureContentSafetyOutput ().getSexual ()).isEqualTo (NUMBER_0 );
803+ assertThat (e .getAzureContentSafetyOutput ().getViolence ()).isEqualTo (NUMBER_4 );
804804 });
805805 }
806806
0 commit comments