spring快速搭建聊天AI

java灰灰

已于 2024-04-11 21:59:16 修改

阅读量925

点赞数 9

文章标签： spring java 后端

于 2024-04-11 21:57:48 首次发布

本文链接：https://blog.csdn.net/Peng_423/article/details/137654303

版权

官网url: https://spring.io/projects/spring-ai

本文演示的是open AI

1创建java项目

2.拿到AI的key（没有的话可以去淘宝花几块钱买一个）

//YOUR_API_KEY写你自己的open AI的key

spring.ai.openai.api-key=YOUR_API_KEY
spring.ai.openai.chat.options.model=gpt-3.5-turbo
spring.ai.openai.chat.options.temperature=0.7

3.导入依赖

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

4.编写接口

@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 = "帮我写一个for循环") 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);
    }
}