连接 OpenAI 模型:基础操作

时间:2025-01-22 15:20:51
package com.example.springaidemo; import com.theokanning.openai.completion.CompletionRequest; import com.theokanning.openai.completion.CompletionResult; import com.theokanning.openai.service.OpenAiService; import org.springframework.beans.factory.annotation.Value; import org.springframework.stereotype.Service; @Service public class OpenAiService { @Value("${openai.api.key}") private String apiKey; public String getResponse(String prompt) { // 创建 OpenAI 服务对象 OpenAiService service = new OpenAiService(apiKey); // 创建请求对象 CompletionRequest completionRequest = CompletionRequest.builder() .prompt(prompt) .model("text-davinci-003") // 选择模型,例如 text-davinci-003 .maxTokens(150) .temperature(0.7) .build(); // 发送请求并获取返回结果 CompletionResult result = service.createCompletion(completionRequest); // 返回模型的回答 return result.getChoices().get(0).getText(); } }