一、获取文心一言密钥
进入百度智能云获取API Key和Secret Key,地址百度智能云
二、创建一个springboot项目
首先导入依赖
<dependency>
<groupId>group.springframework.ai</groupId>
<artifactId>spring-ai-qianfan-spring-boot-starter</artifactId>
<version>1.1.0</version>
</dependency>
注意!要前往中央仓库下载
<repository>
<id>central</id>
<url>https://repo.maven.apache.org/maven2</url>
</repository>
创建配置文件application.properties
填写好申请的API Key和Secret Key
spring.ai.qianfan.api-key=
spring.ai.qianfan.secret-key=
spring.ai.qianfan.chat.options.model=ernie_speed
spring.ai.qianfan.chat.options.temperature=0.7
创建一个ChatController类
源代码可以在spring官网查看:spring官网源代码
@RestController
public class ChatController {
private final QianFanChatModel chatClient;
@Autowired
public ChatController(QianFanChatModel chatClient) {
this.chatClient = chatClient;
}
@GetMapping("/ai/generate")
public Map generate(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {
return Map.of("generation", this.chatClient.call(message));
}
@GetMapping("/ai/generateStream")
public Flux<ChatResponse> generateStream(@RequestParam(value = "message", defaultValue = "Tell me a joke") String message) {
var prompt = new Prompt(new UserMessage(message));
return this.chatClient.stream(prompt);
}
三、展示效果
启动项目,输入”你好“
localhost:8080/ai/generate?message="你好"
返回结果