copilot-sdk/java/sdk/jbang-example.java at main · github/copilot-sdk · GitHub

42 lines (36 loc) · 1.67 KB

///usr/bin/env jbang "$0" "$@" ; exit $?

//DEPS com.github:copilot-sdk-java:1.0.13-preview.1

import com.github.copilot.CopilotClient;

import com.github.copilot.generated.AssistantMessageEvent;

import com.github.copilot.generated.SessionUsageInfoEvent;

import com.github.copilot.rpc.MessageOptions;

import com.github.copilot.rpc.PermissionHandler;

import com.github.copilot.rpc.SessionConfig;

import static java.lang.System.out;

class CopilotSDK {

public static void main(String[] args) throws Exception {

// Create and start client

try (var client = new CopilotClient()) {

client.start().get();

// Create a session

var session = client.createSession(

new SessionConfig().setOnPermissionRequest(PermissionHandler.APPROVE_ALL).setModel("claude-sonnet-4.5")).get();

// Handle assistant message events

session.on(AssistantMessageEvent.class, msg -> {

out.println(msg.getData().content());

});

// Handle session usage info events

session.on(SessionUsageInfoEvent.class, usage -> {

var data = usage.getData();

out.println("\n--- Usage Metrics ---");

out.println("Current tokens: " + data.currentTokens().intValue());

out.println("Token limit: " + data.tokenLimit().intValue());

out.println("Messages count: " + data.messagesLength().intValue());

});

// Send a message

var completable = session.sendAndWait(new MessageOptions().setPrompt("What is 2+2?"));

// and wait for completion

completable.get();

}

Expand file tree