当前位置：首页 > news >正文

初识SpringAI（接入硅基流动deepseek）

news 来源：原创 2025/6/7 21:26:01

①创建项目

②application.yml

spring:
  application:
    name: pgs-ai
  ai:
    openai:
      api-key: sk-vrozloxjpjgkozaggtodbmwyfmubmxqpdpbvbbxpcgleanug
      base-url: https://api.siliconflow.cn/
      chat:
        options:
          model: deepseek-ai/DeepSeek-V3

api-key：去硅基流动官网生成你的密钥

base-url：我使用的是硅基流动的api

chat：

model： deepseek-ai/DeepSeek-V3 //大模型的名

③创建客户端

package com.pgs.aidemo_2.config;

import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.openai.OpenAiChatModel;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class CommonConfiguration {
    @Bean
    public ChatClient chatClient(OpenAiChatModel model){
        return ChatClient
                .builder(model)
                .build();
    }
}

④编写接口

package com.pgs.aidemo_2.controller;

import org.springframework.ai.chat.client.ChatClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import reactor.core.publisher.Flux;


@RestController
@RequestMapping("/ai")
public class ChatController {
    @Autowired
    private ChatClient chatClient;
    @RequestMapping(value = "/chat", produces = "text/html;charset=utf8")
    public Flux<String> chat(String prompt) {
        return chatClient.prompt()
                .user(prompt)
                .stream()
                .content();
    }
}

⑤调用接口