第一步:注册并登录百度智能云,创建应用并获取自己的APIKey与SecretKey,参考网址:
第二步:引入千帆的pom依赖
xml
<dependency>
<groupId>com.baidubce</groupId>
<artifactId>qianfan</artifactId>
<version>0.0.9</version>
</dependency>
第三步:创建前端需要的controller
java
import com.baidubce.qianfan.core.auth.Auth;
import com.lx.vue.common.resp.ResultData;
import com.lx.vue.common.resp.ReturnCodeEnum;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;
import com.baidubce.qianfan.Qianfan;
import com.baidubce.qianfan.core.builder.ChatBuilder;
import com.baidubce.qianfan.model.chat.ChatResponse;
@RestController
public class QianFanController {
private static final String APIKey = "你的APIKey";
private static final String SecretKey = "你的SecretKey";
private static Qianfan qianfan = new Qianfan(Auth.TYPE_OAUTH,APIKey, SecretKey);
@PostMapping("/ai/sendMsg")
public ResultData sendMsg(@RequestBody String problem) {
String result = null;
try {
result = chat(problem);
} catch (Exception e) {
e.printStackTrace();
return new ResultData(ReturnCodeEnum.RC500.getCode(),"服务暂不可用",null);
}
return new ResultData(ReturnCodeEnum.RC200.getCode(),ReturnCodeEnum.RC200.getMessage(),result);
}
private static String chat(String problem) {
ChatBuilder bulder = qianfan.chatCompletion()
.model("ERNIE-Speed-8K");//你要使用的大模型款式,最好和我一样,其他的很有可能是收费的
bulder.addMessage("user",problem);//你的问题
ChatResponse response = bulder.execute();
return response.getResult();
}
}