AI 大模型 API 调用

环境准备

注：JDK 必须17+ 及以上、SpringBoot 版本推荐3.2.x 及以上

 

添加OpenAI 依赖

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

 

配置文件

spring:
  application:
    name: test-openai
  ai:
    openai:
      base-url: https://api.xty.app
      api-key: sk-52YTJALWBTcbAq238XZlibp2TCwjVz54Z976W794KUSo765DyXq39WJ # API-KEY
      chat.options:
        model: gpt-3.5-turbo

注：分享个免费获取apikey的网址, 注册后可以免费获得一定数量的token

 

服务调用

import groovy.util.logging.Slf4j;
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;



@RestController
@RequestMapping("/ai")
@Slf4j
public class ChatController {

    private ChatClient chatClient;

    public ChatController(ChatClient.Builder chatClientBuilder) {
        this.chatClient = chatClientBuilder.build();
    }

    @GetMapping("/chat")
    public String chat(@RequestParam("prompt") String prompt) {
        return this.chatClient.prompt(prompt).call().content();
    }


}