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
}
}这个代码示例展示了如何在Spring Boot应用程序中使用RestTemplate与OpenAI的API进行交互。首先，它定义了一个OpenAiService服务类，该类使用RestTemplate来发送POST请求到OpenAI的completions端点。它还展示了如何创建请求和响应对象来符合OpenAI API的要求。最后，它提供了一个getCompletion方法，该方法接受一个提示并返回一个基于OpenAI模型的响应。