最简单的springAi+文心一言模型，实现人工智能对话

qq_74929891

1654人浏览 · 2025-01-02 20:49:19

qq_74929891 · 2025-01-02 20:49:19 发布

一、获取文心一言密钥

进入百度智能云获取API Key和Secret Key，地址百度智能云

二、创建一个springboot项目

首先导入依赖

        <dependency>
            <groupId>group.springframework.ai</groupId>
            <artifactId>spring-ai-qianfan-spring-boot-starter</artifactId>
            <version>1.1.0</version>
        </dependency>

注意！要前往中央仓库下载

       <repository>
            <id>central</id>
            <url>https://repo.maven.apache.org/maven2</url>
        </repository>

创建配置文件application.properties

填写好申请的API Key和Secret Key


spring.ai.qianfan.api-key=
spring.ai.qianfan.secret-key=
spring.ai.qianfan.chat.options.model=ernie_speed
spring.ai.qianfan.chat.options.temperature=0.7

创建一个ChatController类

源代码可以在spring官网查看：spring官网源代码

@RestController
public class ChatController {
    private final QianFanChatModel chatClient;

    @Autowired
    public ChatController(QianFanChatModel chatClient) {
        this.chatClient = chatClient;
    }

    @GetMapping("/ai/generate")
    public Map generate(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {
        return Map.of("generation", this.chatClient.call(message));
    }

    @GetMapping("/ai/generateStream")
    public Flux<ChatResponse> generateStream(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {
        var prompt = new Prompt(new UserMessage(message));
        return this.chatClient.stream(prompt);
    }