@@ -59,8 +59,8 @@ class SafeGuardAdvisorTests {
5959
6060 @ BeforeEach
6161 void setUp () {
62- safeRequest = new ChatClientRequest (Prompt .builder ().content ("hello world" ).build (), Map .of ());
63- unsafeRequest = new ChatClientRequest (Prompt .builder ().content ("this contains secret" ).build (), Map .of ());
62+ this . safeRequest = new ChatClientRequest (Prompt .builder ().content ("hello world" ).build (), Map .of ());
63+ this . unsafeRequest = new ChatClientRequest (Prompt .builder ().content ("this contains secret" ).build (), Map .of ());
6464 }
6565
6666 @ Test
@@ -82,7 +82,7 @@ void adviseCallInterceptsWhenContainsSensitiveWord() {
8282
8383 CallAdvisorChain mockChain = mock (CallAdvisorChain .class );
8484
85- ChatClientResponse response = advisor .adviseCall (unsafeRequest , mockChain );
85+ ChatClientResponse response = advisor .adviseCall (this . unsafeRequest , mockChain );
8686
8787 assertThat (response .chatResponse ().getResult ().getOutput ().getText ()).contains ("I'm unable to respond" );
8888 verify (mockChain , never ()).nextCall (any ());
@@ -98,12 +98,12 @@ void adviseCallPassesThroughWhenNoSensitiveWord() {
9898 .context (Map .of ())
9999 .build ();
100100
101- when (mockChain .nextCall (safeRequest )).thenReturn (expected );
101+ when (mockChain .nextCall (this . safeRequest )).thenReturn (expected );
102102
103- ChatClientResponse response = advisor .adviseCall (safeRequest , mockChain );
103+ ChatClientResponse response = advisor .adviseCall (this . safeRequest , mockChain );
104104
105105 assertThat (response ).isSameAs (expected );
106- verify (mockChain ).nextCall (safeRequest );
106+ verify (mockChain ).nextCall (this . safeRequest );
107107 }
108108
109109 @ Test
@@ -112,7 +112,7 @@ void adviseStreamInterceptsWhenContainsSensitiveWord() {
112112
113113 StreamAdvisorChain mockChain = mock (StreamAdvisorChain .class );
114114
115- Flux <ChatClientResponse > flux = advisor .adviseStream (unsafeRequest , mockChain );
115+ Flux <ChatClientResponse > flux = advisor .adviseStream (this . unsafeRequest , mockChain );
116116
117117 StepVerifier .create (flux )
118118 .assertNext (r -> assertThat (r .chatResponse ().getResult ().getOutput ().getText ())
@@ -132,13 +132,13 @@ void adviseStreamPassesThroughWhenNoSensitiveWord() {
132132 .context (Map .of ())
133133 .build ();
134134
135- when (mockChain .nextStream (safeRequest )).thenReturn (Flux .just (expected ));
135+ when (mockChain .nextStream (this . safeRequest )).thenReturn (Flux .just (expected ));
136136
137- Flux <ChatClientResponse > flux = advisor .adviseStream (safeRequest , mockChain );
137+ Flux <ChatClientResponse > flux = advisor .adviseStream (this . safeRequest , mockChain );
138138
139139 StepVerifier .create (flux ).expectNext (expected ).verifyComplete ();
140140
141- verify (mockChain ).nextStream (safeRequest );
141+ verify (mockChain ).nextStream (this . safeRequest );
142142 }
143143
144144 @ Test
0 commit comments