SpringBootコース

SpringBootへ生成AIの組み込み

ChatGPTのAPI呼び出しを組み込もう

####  pom.xml
~~~xml
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-webflux</artifactId>
</dependency>
~~~

~~~java
package com.example.demo;

import org.springframework.stereotype.Service;
import org.springframework.web.reactive.function.client.WebClient;

import reactor.core.publisher.Mono;

/**
 * ChatGPT APIと通信を行うサービスクラスです。
 * WebClientを使用して、OpenAIのAPIと非同期通信を行います。
 */
@Service
public class ChatGPTService {
	private static final String API_URL = "https://api.openai.com/v1/chat/completions";
	private static final String API_KEY = System.getenv("OPENAI_API_KEY");

	private final WebClient webClient;

	/**
	 * ChatGPTServiceのコンストラクタです。
	 * WebClientを初期化し、必要なヘッダー（Authorization、Content-Type）を設定します。
	 */
	public ChatGPTService() {
		this.webClient = WebClient.builder()
				.baseUrl(API_URL)
				.defaultHeader("Authorization", "Bearer " + API_KEY)
				.defaultHeader("Content-Type", "application/json")
				.build();
	}

	/**
	 * ChatGPT APIにプロンプトを送信して、レスポンスを非同期に受け取ります。
	 * 
	 * @param prompt ユーザーからの入力を含む文字列。ChatGPTに送信されます。
	 * @return 非同期に結果を返すMono。レスポンスのボディがStringとして返されます。
	 */
	public Mono<String> chat(String prompt) {
		String requestBody = String.format(
				"{\"model\": \"gpt-4o-mini\", \"messages\": [{\"role\": \"user\", \"content\": \"%s\"}], \"temperature\": 0.7}",
				prompt);

		return webClient.post()
				.bodyValue(requestBody)
				.retrieve()
				.bodyToMono(String.class);
	}

	/**
	 * このクラス単体でChatGPT APIを試すためのメインメソッドです。
	 * @param args コマンドライン引数（現在は使用されません）
	 */
	public static void main(String[] args) {
		ChatGPTService service = new ChatGPTService();
		System.out.println("ChatGPT API KEY: " + API_KEY);
		service.chat(
				"Just simply give me three random English words separated by commas. All lower cases without spaces")
				.doOnNext(System.out::println)
				.block();
	}
}

~~~

## 確認方法

左サイドのエクスプローラから、作成したクラスを右クリック→ [実行] →[Java アプリケーション] 
![image](https://prod-code-lesson-api.s3.ap-northeast-1.amazonaws.com/cms/f970408f-2fef-46cd-948c-b087d9edf850.png)

![image](https://prod-code-lesson-api.s3.ap-northeast-1.amazonaws.com/cms/b0265351-812b-46aa-8f2d-d125bdab59e6.png)

![image](https://prod-code-lesson-api.s3.ap-northeast-1.amazonaws.com/cms/3210f7ac-e0cf-4a49-a414-c875e24cbaa9.png)