2727
2828import com .github .copilot .sdk .CopilotClient ;
2929import com .github .copilot .sdk .CopilotSession ;
30- import com .github .copilot .sdk .events . AbstractSessionEvent ;
31- import com .github .copilot .sdk .events . AssistantMessageDeltaEvent ;
32- import com .github .copilot .sdk .events . AssistantMessageEvent ;
33- import com .github .copilot .sdk .events . SessionErrorEvent ;
30+ import com .github .copilot .sdk .generated . AssistantMessageDeltaEvent ;
31+ import com .github .copilot .sdk .generated . AssistantMessageEvent ;
32+ import com .github .copilot .sdk .generated . SessionErrorEvent ;
33+ import com .github .copilot .sdk .generated . SessionEvent ;
3434import com .github .copilot .sdk .json .MessageOptions ;
3535import com .github .copilot .sdk .json .SessionConfig ;
3636
@@ -116,6 +116,7 @@ private CompletableFuture<CopilotSession> createSession() {
116116 SessionConfig config = new SessionConfig ()
117117 .setStreaming (true )
118118 .setModel (model )
119+ .setOnPermissionRequest (com .github .copilot .sdk .json .PermissionHandler .APPROVE_ALL )
119120 .setSystemMessage (new com .github .copilot .sdk .json .SystemMessageConfig ()
120121 .setMode (com .github .copilot .sdk .SystemMessageMode .APPEND )
121122 .setContent (JMETER_SYSTEM_PROMPT ));
@@ -166,15 +167,15 @@ private void subscribeToEvents() {
166167 }
167168 }
168169
169- private void handleEvent (AbstractSessionEvent event ) {
170+ private void handleEvent (SessionEvent event ) {
170171 try {
171172 if (event instanceof AssistantMessageDeltaEvent deltaEvent ) {
172- String delta = deltaEvent .getData ().getDeltaContent ();
173+ String delta = deltaEvent .getData ().deltaContent ();
173174 if (delta != null && streamingHandler != null ) {
174175 streamingHandler .accept (delta );
175176 }
176177 } else if (event instanceof AssistantMessageEvent messageEvent ) {
177- String content = messageEvent .getData ().getContent ();
178+ String content = messageEvent .getData ().content ();
178179 if (content != null && !content .isBlank ()) {
179180 ChatMessage message = new ChatMessage (ChatMessage .Role .ASSISTANT , content );
180181 conversationHistory .addMessage (message );
@@ -184,7 +185,7 @@ private void handleEvent(AbstractSessionEvent event) {
184185 }
185186 } else if (event instanceof SessionErrorEvent errorEvent ) {
186187 LOG .log (Level .WARNING , "Session error: {0}" ,
187- errorEvent .getData () != null ? errorEvent .getData ().getMessage () : "Unknown error" );
188+ errorEvent .getData () != null ? errorEvent .getData ().message () : "Unknown error" );
188189 }
189190 } catch (Exception e ) {
190191 LOG .log (Level .SEVERE , "Error handling event" , e );
@@ -228,7 +229,7 @@ public CompletableFuture<String> sendMessageAndWait(String prompt) {
228229 return session .sendAndWait (new MessageOptions ().setPrompt (prompt ))
229230 .thenApply (response -> {
230231 if (response != null && response .getData () != null ) {
231- return response .getData ().getContent ();
232+ return response .getData ().content ();
232233 }
233234 return null ;
234235 });
0 commit comments