Spring Boot集成Spring AI实现快速接入openAI

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Service;
import org.springframework.web.client.RestTemplate;

@Service
public class OpenAiService {

    private final RestTemplate restTemplate;
    private final String openAiApiKey;

    @Autowired
    public OpenAiService(RestTemplate restTemplate, @Value("${openai.api-key}") String openAiApiKey) {
        this.restTemplate = restTemplate;
        this.openAiApiKey = openAiApiKey;
    }

    public String getCompletion(String prompt) {
        String url = "https://api.openai.com/v1/engines/davinci-codex/completions";
        CompletionRequest completionRequest = new CompletionRequest(prompt);
        ResponseEntity<CompletionResponse> responseEntity = restTemplate.postForEntity(url, completionRequest, CompletionResponse.class, "Authorization: Bearer " + openAiApiKey);
        if (responseEntity.getStatusCode().is2xxSuccessful()) {
            return responseEntity.getBody().getChoices()[0].getText();
        } else {
            throw new RuntimeException("Error occurred while calling OpenAI API");
        }
    }

    static class CompletionRequest {
        private String prompt;
        private String max_tokens;
        private String n;
        private String stop;

        public CompletionRequest(String prompt) {
            this.prompt = prompt;
            this.max_tokens = "1000";
            this.n = "1";
            this.stop = null;
        }
        // getters and setters
    }

    static class CompletionResponse {
        private Choice[] choices;

        public Choice[] getChoices() {
            return choices;
        }
        // getters and setters
    }

    static class Choice {
        private String text;

        public String getText() {
            return text;
        }
        // getters and setters
    }
}