Spring Cloud Alibaba AI 初体验

Spring Cloud Alibaba AI 初体验 | 极客日志

<dependencyManagement>
    <dependencies>
        <dependency>
            <groupId>com.alibaba.cloud</groupId>
            <artifactId>spring-cloud-alibaba-dependencies</artifactId>
            <version>2023.0.1.0</version>
            <type>pom</type>
            <scope>import</scope>
        </dependency>
    </dependencies>
</dependencyManagement>

<dependencies>
    <!-- Spring Cloud Alibaba AI Starter -->
    <dependency>
        <groupId>com.alibaba.cloud</groupId>
        <artifactId>spring-cloud-starter-alibaba-ai</artifactId>
    </dependency>
    
    <!-- 其他必要依赖，如 Web, Lombok 等 -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    <dependency>
        <groupId>org.projectlombok</groupId>
        <artifactId>lombok</artifactId>
        <optional>true</optional>
    </dependency>
</dependencies>

server:
  port: 8080
spring:
  application:
    name: alibaba-spring-ai-demo

  cloud:
    ai:
      tongyi:
          api-key: ${ALIYUN_TONGYI_API_KEY}
          # 可选配置，指定模型名称，默认使用 qwen-plus
          model-name: qwen-plus

@SpringBootApplication
public class MyAiApplication {
    public static void main(String[] args) {
        SpringApplication.run(MyAiApplication.class, args);
    }
}

@RestController
@RequestMapping("/ai")
@CrossOrigin
public class TongYiController {
    @Autowired
    @Qualifier("tongYiSimpleServiceImpl")
    private TongYiService tongYiSimpleService;

    @GetMapping("/simple")
    public String completion(
            @RequestParam(value = "message", defaultValue = "AI 时代下 Java 开发者该何去何从？")
            String message
    ) {
        return tongYiSimpleService.completion(message);
    }
}

public interface TongYiService {

    /**
     * 基本问答
     */
    String completion(String message);
    
    /**
     * 文生图
     */
    ImageResponse genImg(String imgPrompt);

    /**
     * 语音合成
     */
    String genAudio(String text);

}

@Service
@Slf4j
public class TongYiSimpleServiceImpl extends AbstractTongYiServiceImpl implements TongYiService {
    
    /**
     * 自动注入 ChatClient、StreamingChatClient，屏蔽模型调用细节
     */
    private final ChatClient chatClient;
    private final StreamingChatClient streamingChatClient;

    @Autowired
    public TongYiSimpleServiceImpl(ChatClient chatClient, StreamingChatClient streamingChatClient) {
        this.chatClient = chatClient;
        this.streamingChatClient = streamingChatClient;
    }

    @Override
    public String completion(String message) {
        try {
            Prompt prompt = new Prompt(new UserMessage(message));
            // 调用模型并获取结果
            ChatResponse response = chatClient.call(prompt);
            if (response != null && response.getResult() != null) {
                return response.getResult().getOutput().getContent();
            }
            return "Error: No response";
        } catch (Exception e) {
            log.error("Completion error", e);
            throw new RuntimeException(e.getMessage());
        }
    }
}

@Slf4j
@Service
public class TongYiImagesServiceImpl extends AbstractTongYiServiceImpl implements TongYiService {
    private static final Logger logger = LoggerFactory.getLogger(TongYiImagesServiceImpl.class);
    private final ImageClient imageClient;

    @Autowired
    public TongYiImagesServiceImpl(ImageClient client) {
        this.imageClient = client;
    }

    @Override
    public ImageResponse genImg(String imgPrompt) {
        try {
            var prompt = new ImagePrompt(imgPrompt);
            // 调用文生图接口
            return imageClient.call(prompt);
        } catch (Exception e) {
            logger.error("Image generation failed", e);
            throw e;
        }
    }
}

@Slf4j
@Service
public class TongYiAudioSimpleServiceImpl extends AbstractTongYiServiceImpl implements TongYiService {
    private static final Logger logger = LoggerFactory.getLogger(TongYiAudioSimpleServiceImpl.class);
    private final SpeechClient speechClient;

    @Autowired
    public TongYiAudioSimpleServiceImpl(SpeechClient client) {
        this.speechClient = client;
    }

    @Override
    public String genAudio(String text) {
        logger.info("gen audio prompt is: {}", text);
        try {
            // 调用语音合成接口，返回音频数据
            SpeechSynthesisResponse resWAV = speechClient.call(text);
            // save 的代码省略，就是将音频保存到本地而已
            return save(resWAV, SpeechSynthesisAudioFormat.WAV.getValue());
        } catch (Exception e) {
            logger.error("Audio synthesis failed", e);
            throw e;
        }
    }
}

Spring Cloud Alibaba AI 初体验

Spring Cloud Alibaba AI 初体验

什么是 Spring AI

什么是 Spring Cloud Alibaba AI

环境准备与依赖引入

配置阿里云通义千问

新建 SpringBoot 启动类

对接文本模型

文生图模型

语音合成模型

常见问题与优化建议

使用体验小结

总结

更多推荐文章

相关免费在线工具

Spring Cloud Alibaba AI 初体验

Spring Cloud Alibaba AI 初体验

什么是 Spring AI

什么是 Spring Cloud Alibaba AI

环境准备与依赖引入

配置阿里云通义千问

新建 SpringBoot 启动类

对接文本模型

文生图模型

语音合成模型

常见问题与优化建议

使用体验小结

总结

微信扫一扫，关注极客日志

更多推荐文章

相关免费在线工具