Spring AI

Spring AI简介

Spring AI官方文档

Spring AI 是一个面向人工智能工程的应用框架，它类似于一个客户端，可以实现向不同厂商对应的AI模型服务发送请求获取响应

SpringAI目前能够支持的AI模型

Spring AI实践

Open ai Api Key第三方平台

在该平台上注册可以免费获取一定的使用额度

参考SpringAI openAI官方文档，快速搭建一个应用

引入pom依赖

dependency>
        <groupId>org.springframework.ai</groupId>
        <artifactId>spring-ai-openai-spring-boot-starter</artifactId>
    </dependency>

进行相关配置

#服务地址
spring.ai.openai.base-url=https://api.xty.app
#openai key
spring.ai.openai.api-key=sk-yJ86bNiRwQJVyczO122aF307B7554905A0F53eF6B3B27132
#使用模型名称
spring.ai.openai.chat.options.model=gpt-3.5-turbo
#随机性
spring.ai.openai.chat.options.temperature=0.7

编写一个简单controller

@RestController
public class ChatController {

    private final OpenAiChatClient chatClient;

    @Autowired
    public ChatController(OpenAiChatClient chatClient) {
        this.chatClient = chatClient;
    }
	//整体响应
    @GetMapping("/ai/generate")
    public Map generate(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {
        return Map.of("generation", chatClient.call(message));
    }
	//流式响应
    @GetMapping("/ai/generateStream")
	public Flux<ChatResponse> generateStream(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {
        Prompt prompt = new Prompt(new UserMessage(message));
        return chatClient.stream(prompt);
    }
}

测试

构造请求:http://localhost:8080/ai/generate?message=%E5%AD%A6%E4%B9%A0Java%E8%B7%AF%E7%BA%BF

得到的响应:

其他应用

调用openAI相关模型生成图片

官方文档参考

controller代码

//注入openAiImageClien
@Autowired
OpenAiImageClient openAiImageClient;
@GetMapping("/ai/image")
public String generateImage(String prompt){
    ImageResponse response = openAiImageClient.call(new ImagePrompt(prompt));
    //可以两种形式返回生成的图片
    //图片的url:response.getResult().getOutput().getUrl()
    //图片的base64编码:response.getResult().getOutput().getB64Json()
    return response.getResult().getOutput().getUrl();
}