连接 OpenAI 模型:基础操作
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();
}
}