Java对接文心一言

注册千帆大模型

首先就是注册百度的千帆大模型平台，第一次注册会送一张20元的优惠卷，可以用这个先免费用一段时间。

创建应用

目的：获得APIKey和SecretKey 注册以后进入到百度智能云控制台应用接入（百度智能云控制台 (baidu.com)）这里。先在应用接入这里创建应用，接下来会用到你创建这个应用的APIKey和SecretKey。这个是可以重复访问的。

开通服务

目的：开通服务，获得请求地址接下来进入到在线服务。要先开通服务后才能使用他的API接口来访问文心一言。可以先到百度提供的API列表看看你要用哪个服务（API列表 - 千帆大模型平台 | 百度智能云文档 (baidu.com)）。API鉴权及调用可以看服务介绍，平台计费可以看价格。选择一个开通付费就行。选择以后，在API鉴权及调用那里点击你选择的服务后面的“创建chat”。我以“ERNIE-3.5-8K”这个为例。点击进去后这个API采用两种鉴权方式，我用的是访问凭证access_token鉴权。这个要先获得access_token，然后再通过access_token调用API获得回复。待会需要用到文档里面的请求说明信息。不同服务的请求地址不同。

Java对接文心一言代码实现

获得token

使用文心一言要先获得access_token，然后通过这个access_token再来调用文心一言。百度有提供获得access_token的示例代码(百度智能云控制台 (baidu.com))。下面这个是我自己写的，都一样。这个就要用到刚才创建应用时的APIKey和SecretKey。

/**
 * 需要添加依赖
 * <!-- https://mvnrepository.com/artifact/com.squareup.okhttp3/okhttp -->
 * <dependency>
 *     <groupId>org.apache.httpcomponents</groupId>
 *     <artifactId>httpclient</artifactId>
 *     <version>4.5.14</version>
 * </dependency>
 */

public String getWenxinToken() throws IOException {
        CloseableHttpClient httpClient = HttpClientBuilder.create().build();

        String apiKey = "";
        String secretKey = "";

        HttpPost post = new HttpPost("https://aip.baidubce.com/oauth/2.0/token?client_id="
                + apiKey + "&client_secret=" + secretKey + "&grant_type=client_credentials");
        post.addHeader("Content-Type", "application/json");
        post.addHeader("Accept", "application/json");
        CloseableHttpResponse response = httpClient.execute(post);

        String s = EntityUtils.toString(response.getEntity());
        JSONObject objects = JSONArray.parseObject(s);
        String token = objects.getString("access_token");
        System.out.println("token:" + token);
        return token;
    }

获得文心一言回复

这个要用到请求说明里基本信息的请求地址。

public void wenxinyiyanApi() throws IOException {
        CloseableHttpClient httpClient = HttpClientBuilder.create().build();

        // 1、获取token，就是调用上面的方法。
        String access_token = new ApiTest().getWenxinToken();
        // 2、看你开通的服务请求地址是什么，不一样在这里改。
        HttpPost post = new HttpPost("https://aip.baidubce.com/rpc/2.0/" +
                "ai_custom/v1/wenxinworkshop/chat/completions?access_token=" + access_token); 
        post.addHeader("Content-Type", "application/json");
        post.addHeader("Accept", "application/json");
        // 3、问题内容。
        String paramJson = "{\"messages\": [{\"role\": \"user\", " +
                "\"content\": \"帮我写一个冒泡排序\"}]\n}";
        StringEntity stringEntity = new StringEntity(paramJson, 
                ContentType.create("application/json", "UTF-8"));
        post.setEntity(stringEntity);
        CloseableHttpResponse response = httpClient.execute(post);
        String s = EntityUtils.toString(response.getEntity());
        System.out.println("输出:" + s);
    }