首先在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();}}