Springboot快速接入豆包大模型

发布于:2025-02-27 ⋅ 阅读:(16) ⋅ 点赞:(0)

背景

       突然接到上面的通知,想要在系统里面接入各大模型的能力,我这边随机选了个豆包,然后快速对接了一下,很顺利,一把过,
	现在文档的快速入门还是很ok的,在此记录一下过程,给宝子们参考!

接入流程

参考官方快速入门文档

总结了一下,通过代码调用模型服务,大致四个步骤,如下:

  1. 走一下注册流程,创建一下appkey
  2. 在模型广场选一个模型,并开通,拿到对应的模型id
  3. 添加maven依赖
  4. 通过appkey和模型id就可以进行对应的AI能力了

相关页面
创建API key
在这里插入图片描述
模型选择
在这里插入图片描述

效果

我这边提问:你好,请帮我生成一份关于java策略模式的代码样例
响应结果:

在这里插入图片描述

代码样例

pom.xml
    <dependency>
        <groupId>com.volcengine</groupId>
        <artifactId>volcengine-java-sdk-ark-runtime</artifactId>
        <version>LATEST</version>
    </dependency>```
**代码demo**


```java
package com.ideal.jason.sdk.ai;


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;

/**
 * 豆包
 *
 * @author: jason
 * @Date: 26 2月 2025
 *
 * 这是一个示例类,展示了如何使用ArkService来完成聊天功能。
 */
public class DoubaoChatCompletionsExample {
    public static void main(String[] args) {
        
        //String apiKey = System.getenv("ARK_API_KEY");
        String apiKey = "xxxxxxxxx";// 从环境变量中获取API密钥,或直接写死创建的appkey

        // 创建ArkService实例
        ArkService arkService = ArkService.builder().apiKey(apiKey).build();

        // 初始化消息列表
        List<ChatMessage> chatMessages = new ArrayList<>();

        // 创建用户消息
        ChatMessage userMessage = ChatMessage.builder()
            .role(ChatMessageRole.USER) // 设置消息角色为用户
            .content("你好,请帮我生成一份关于java策略模式的代码样例") // 设置消息内容
            .build();

        // 将用户消息添加到消息列表
        chatMessages.add(userMessage);

        // 创建聊天完成请求
        ChatCompletionRequest chatCompletionRequest = ChatCompletionRequest.builder()
            .model("doubao-1.5-pro-32k-250115")// 需要替换为Model ID
            .messages(chatMessages) // 设置消息列表
            .build();

        // 发送聊天完成请求并打印响应
        try {
            // 获取响应并打印每个选择的消息内容
            arkService.createChatCompletion(chatCompletionRequest)
                .getChoices()
                .forEach(choice -> System.out.println(choice.getMessage().getContent()));
        } catch (Exception e) {
            System.out.println("请求失败: " + e.getMessage());
        } finally {
            // 关闭服务执行器
            arkService.shutdownExecutor();
        }
    }
}