Spring AI 接入OpenAI大模型实现同步和流式对话
import com.openai.api.AskCompletionResponse;
import com.openai.api.Engine;
import com.openai.api.OpenAiService;
import com.openai.api.model.CompletionRequest;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
@Service
public class OpenAiConversationService {
@Autowired
private OpenAiService openAiService;
public String startConversation(String prompt) {
CompletionRequest completionRequest = CompletionRequest.builder()
.prompt(prompt)
.maxTokens(50)
.topP(1)
.frequencyPenalty(0)
.presencePenalty(0)
.engine(Engine.DAVINCI_002)
.build();
AskCompletionResponse response = openAiService.createCompletion().
ask(completionRequest);
return response.getChoices().get(0).getPlaintext();
}
}
这段代码展示了如何在Spring应用中使用自动装配的方式初始化OpenAiService,并创建一个与OpenAI大模型进行交谈的方法。它设置了一个基础的对话请求,并获取了模型的响应。这个例子简单且直接地展示了如何在Spring项目中集成OpenAI的交互式对话功能。
评论已关闭