首先在maven里面引入官方依賴
<dependency><groupId>com.volcengine</groupId><artifactId>volcengine-java-sdk-ark-runtime</artifactId><version>LATEST</version></dependency>
然后我們編寫測試類
package com.volcengine.ark.runtime;import com.volcengine.ark.runtime.model.completion.chat.ChatCompletionRequest;
import com.volcengine.ark.runtime.model.completion.chat.ChatMessage;
import com.volcengine.ark.runtime.model.completion.chat.ChatMessageRole;
import com.volcengine.ark.runtime.service.ArkService;import java.util.ArrayList;
import java.util.List;public class ChatCompletionsExample {public static void main(String[] args) {String apiKey = System.getenv("ARK_API_KEY");ArkService service = ArkService.builder().apiKey(apiKey).build();System.out.println("\n----- standard request -----");final List<ChatMessage> messages = new ArrayList<>();final ChatMessage systemMessage = ChatMessage.builder().role(ChatMessageRole.SYSTEM).content("你是豆包,是由字節跳動開發的 AI 人工智能助手").build();final ChatMessage userMessage = ChatMessage.builder().role(ChatMessageRole.USER).content("常見的十字花科植物有哪些?").build();messages.add(systemMessage);messages.add(userMessage);ChatCompletionRequest chatCompletionRequest = ChatCompletionRequest.builder().model("<Model>").messages(messages).build();service.createChatCompletion(chatCompletionRequest).getChoices().forEach(choice -> System.out.println(choice.getMessage().getContent()));// shutdown serviceservice.shutdownExecutor();}}
多輪請求請換成這個
package com.volcengine.ark.runtime;import com.volcengine.ark.runtime.model.completion.chat.ChatCompletionRequest;
import com.volcengine.ark.runtime.model.completion.chat.ChatMessage;
import com.volcengine.ark.runtime.model.completion.chat.ChatMessageRole;
import com.volcengine.ark.runtime.service.ArkService;import java.util.Arrays;
import java.util.List;public class ChatCompletionsExample {public static void main(String[] args) {String apiKey = System.getenv("ARK_API_KEY");ArkService service = ArkService.builder().apiKey(apiKey).build();System.out.println("\n----- multiple rounds request -----");final List<ChatMessage> messages = Arrays.asList(ChatMessage.builder().role(ChatMessageRole.SYSTEM).content("你是豆包,是由字節跳動開發的 AI 人工智能助手").build(),ChatMessage.builder().role(ChatMessageRole.USER).content("花椰菜是什么?").build(),ChatMessage.builder().role(ChatMessageRole.ASSISTANT).content("花椰菜又稱菜花、花菜,是一種常見的蔬菜。").build(),ChatMessage.builder().role(ChatMessageRole.USER).content("再詳細點").build());ChatCompletionRequest chatCompletionRequest = ChatCompletionRequest.builder().model("<Model>").messages(messages).build();service.createChatCompletion(chatCompletionRequest).getChoices().forEach(choice -> System.out.println(choice.getMessage().getContent()));// shutdown serviceservice.shutdownExecutor();}}