前一篇介紹到如何使用application.properties設定連線，這一篇跟大家介紹pom.xml直接引用的方式~

pom.xml新增dependency

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

新增環境變數OPENAI_API_KEY

application.properties

spring.ai.openai.api-key=${OPENAI_API_KEY}

Config

這邊就不用再帶key囉！

import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.client.RestTemplate;

@Configuration
public class ChatGPTConfig {
    @Bean
    @Qualifier("openaiRestTemplate")
    RestTemplate restTemplate() {
        return new RestTemplate();
    }
}

Controller

這裡我們就可以非常簡單的直接使用ChatClient來做溝通

@RestController
@RequestMapping("api/v1")
@Slf4j
public class ChatAPIController {
    @Autowired
    private ChatClient chatClient;

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

從ChatClient可以得知Default的Model是用哪一個

可以得到相同結果哦

再試試一個AI圖片生成

改使用ImageClient，大家也可以進去看看ImageClient的程式會更了解哦

@RestController
@RequestMapping("api/v1")
@Slf4j
public class ChatAPIController {
    @Qualifier("openaiRestTemplate")
    @Autowired
    private RestTemplate restTemplate;

    @Autowired
    private ImageClient imageClient;

    @GetMapping(value = "image", produces = MediaType.IMAGE_JPEG_VALUE)
    @ResponseBody
    public byte[] getImage(@RequestParam("prompt") String prompt) throws URISyntaxException, IOException {
        Image image = imageClient.call(new ImagePrompt(prompt)).getResult().getOutput();
        URI uri = new URI(image.getUrl());
        ResponseEntity<byte[]> responseEntity = restTemplate.getForEntity(uri, byte[].class);
        return responseEntity.getBody();
    }
}

因為是生成圖片，所以改用Chrome來call api

結論

不論使用哪種方式，大家都可以去看看相關的Library，就可以用得更熟能生巧！