Java接入火山引擎快速体验
文章目录
- Java接入火山引擎快速体验
- 准备
- 代码
Java接入火山引擎快速体验
火山官网:
https://www.volcengine.com/activity/enterprise?utm_source=5&utm_medium=sem_bing&utm_term=sem_bing_pinzhuan_bitaoti_qyqyzx_1&utm_campaign=logo&utm_content=qyqyzx_gzq
准备
- 申请apikey
https://console.volcengine.com/ark/region:ark+cn-beijing/apiKey?apikey=%7B%7D
2. 开通模型
火山里的模型需要先开通才能用,新用户有免费额度可以使用。
https://console.volcengine.com/ark/region:ark+cn-beijing/openManagement?LLM=%7B%22PageSize%22%3A10%2C%22PageNumber%22%3A1%2C%22Filter%22%3A%7B%7D%7D&OpenModelVisible=false&OpenTokenDrawer=false
选一个所需模型开通即可。
代码
- 导入依赖:
<dependency><groupId>com.volcengine</groupId><artifactId>volcengine-java-sdk-ark-runtime</artifactId><version>LATEST</version></dependency>
- demo简单使用
@RestController
@RequestMapping("/ark")
public class ArkController {@Value("${spring.ai.volcano.api-key}")private String apiKey;private ArkService arkService;@PostConstructpublic void init () {// 初始化实例arkService = ArkService.builder().apiKey(apiKey).baseUrl("https://ark.cn-beijing.volces.com/api/v3").build();}@RequestMapping("/hello")public void hello(@RequestParam(value = "query", defaultValue = "hello") String query) {// 初始化消息列表List<ChatMessage> chatMessages = new ArrayList<>();// 创建用户消息ChatMessage userMessage = ChatMessage.builder().role(ChatMessageRole.USER) // 设置消息角色为用户.content(query) // 设置消息内容.build();// 将用户消息添加到消息列表chatMessages.add(userMessage);// 创建聊天完成请求ChatCompletionRequest chatCompletionRequest = ChatCompletionRequest.builder().model("doubao-seed-1-6-250615")// 按需替换Model ID.messages(chatMessages) // 设置消息列表.build();// 发送聊天完成请求并打印响应try {// 获取响应并打印每个选择的消息内容List<ChatCompletionChoice> choices = arkService.createChatCompletion(chatCompletionRequest).getChoices();System.out.println(choices.get(0).getMessage().getContent());} catch (Exception e) {System.out.println("请求失败: " + e.getMessage());} finally {// 关闭服务执行器arkService.shutdownExecutor();}}
}
将代码中api-key和开通的模型ID换成你刚申请的。
输入:
你是谁?
返回:
成功返回结果。
还有很多请求参数和返回参数等,可以自行查看官方api文档进行查看和学习。
https://www.volcengine.com/docs/82379/1494384