阅读量:1
第一步:注册并登录百度智能云,创建应用并获取自己的APIKey与SecretKey,参考网址:
第二步:引入千帆的pom依赖
<dependency> <groupId>com.baidubce</groupId> <artifactId>qianfan</artifactId> <version>0.0.9</version> </dependency>
第三步:创建前端需要的controller
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(); } }