Spring AI -使用Spring快速开发ChatGPT应用
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import ai.spring.api.ChatApi;
import ai.spring.model.ChatCompletionResponse;
import ai.spring.model.ChatCompletionRequest;
@Service
public class ChatGPTService {
@Autowired
private ChatApi chatApi;
public String generateResponse(String message) {
ChatCompletionRequest request = new ChatCompletionRequest()
.prompt(message)
.maxTokens(50)
.model("text-davinci-003");
ChatCompletionResponse response = chatApi.createChatCompletion(request);
return response.getChoices().get(0).getText();
}
}
这段代码展示了如何使用Spring框架和Spring AI SDK来创建一个简单的服务,用于与ChatGPT交流。ChatGPTService
类中注入了ChatApi
,这是Spring AI SDK提供的用于与Spring Chat API交互的客户端。generateResponse
方法接收一个消息,创建一个ChatCompletionRequest
,并发送给Chat API以生成响应。然后它返回了API返回的第一个可能的答案。
评论已关闭