Java对接百度文心一言,Java发送POST请求,类似前端AJAX

这是项目中使用的对接百度文心一言后端代码

public class BaiduChatApi {
    private static  String CHAT_URL = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/eb-instant";

    static String inp = "{\"messages\": [\n" +
            "            {\n" +
            "                \"role\": \"%s\",\n" +
            "                \"content\": \"%s\"\n" +
            "            }\n" +
            "        ]}";

    public static String getAnswerBaiDu(String text) throws IOException{
        CHAT_URL = CHAT_URL+"?access_token=xxxx";
        URL u=new URL(CHAT_URL);
        HttpURLConnection conn=(HttpURLConnection) u.openConnection();
        conn.setConnectTimeout(10*1000);
        conn.setDoOutput(true);
        conn.setRequestMethod("POST");
        conn.setRequestProperty("Connection", "Keep-Alive");
        conn.setRequestProperty("Content-Type", "application/json");
        OutputStream out= conn.getOutputStream();
        try{
            String json=String.format(inp,"user",text);
            System.err.println(json);
            out.write(json.getBytes("UTF-8"));
            out.close();
            out=null;
            InputStream in = conn.getInputStream();
            if(conn.getResponseCode()==200){
                //流转换为二进制数组,read()是转换方法
                byte[] data = new byte[1024];
                int len = 0;
                ByteArrayOutputStream bos = new ByteArrayOutputStream();
                while ((len = in.read(data)) != -1){
                    bos.write(data,0,len);
                }
                bos.close();
                in.close();
                conn.disconnect();
                return new String(bos.toByteArray(), "UTF-8");
            }
            else in.close();
        }catch(Exception e){
            e.printStackTrace();
        }
        finally{
            if(out!=null){
                out.close();
            }
            conn.disconnect();
        }
        return null;
    }

    public static void main(String[] args) throws Exception {
        
        System.out.print("input>>");
        String answer = getAnswerBaiDu("头疼怎么办");
        System.err.println(answer);
        //使用ObjectMapper直接将String串转成对像
        ObjectMapper objectMapper = new ObjectMapper();
        BaiduChatModel chatModel = objectMapper.readValue(answer, BaiduChatModel.class);
        System.err.println(chatModel.toString());
        System.err.println(chatModel.getResult());

    }


}

版权声明:本文为博主作者:小熊火锅原创文章,版权归属原作者,如果侵权,请联系我们删除!

原文链接:https://blog.csdn.net/weixin_43091752/article/details/131953046

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2024年1月11日
下一篇 2024年1月11日

相关推荐