百度千帆大模型文心一言api调用

shy_snow

已于 2023-09-18 10:26:25 修改

阅读量3.7k

点赞数 5

分类专栏：代码示例大模型文章标签：百度

于 2023-09-11 15:07:13 首次发布

本文链接：https://blog.csdn.net/shy_snow/article/details/132759686

版权

代码示例同时被 2 个专栏收录

22 篇文章

订阅专栏

大模型

6 篇文章

订阅专栏

demo工程(csdn上传总是报错461, 只好使用百度网盘了)
链接：https://pan.baidu.com/s/1EXbQDBMMNh1pyMIKwCmnow?pwd=7891
提取码：7891

注册百度智能云账号并申请文心千帆大模型资格

https://login.bce.baidu.com/
https://cloud.baidu.com/product/wenxinworkshop

创建应用用于获取access_token

在这里插入图片描述

创建应用成功后,可以获取到API Key和Secret Key

获取access_token

curl 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=【API Key】&client_secret=【Secret Key】'

在这里插入图片描述

开通付费服务

在这里插入图片描述
api调用是按token(字数)收费的,不开通收费无法使用。

发送对话请求

curl -XPOST 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token=[步骤一调用接口获取的access_token]' -d '{
   "messages": [
    {"role":"user","content":"介绍一下你自己"}
   ]
}'

在这里插入图片描述

Java发送http方式调用

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.gson.Gson;

import okhttp3.MediaType;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.RequestBody;
import okhttp3.Response;

/**
 * api文档 https://cloud.baidu.com/doc/WENXINWORKSHOP/s/jlil56u11
 *
 */
public class BaiduWenXinChat {

	static final OkHttpClient HTTP_CLIENT = new OkHttpClient().newBuilder().connectTimeout(5, TimeUnit.SECONDS) // 连接超时设置为20秒
			.writeTimeout(15, TimeUnit.SECONDS) // 写入超时设置为30秒
			.readTimeout(20, TimeUnit.SECONDS) // 读取超时设置为30秒
			.build();
	static Gson gson = new Gson();

	// 历史对话信息
	Map<String, List<ChatMsg>> mapChatList = new HashMap<String, List<ChatMsg>>();

	public static void main(String[] args) throws Exception {

		/*  获取acessToken:
		curl 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=【API Key】&client_secret=【Secret Key】'
		*/
		String accessToken = "24.621fe77e232121321213213213213213c6b";
		String url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token="
				+ accessToken;
		String msg = "介绍下你自己";

		// 结合prompt增强的当前待发送信息
		ChatMsg chatMsg = new ChatMsg();
		chatMsg.setRole("user");
		chatMsg.setContent(msg);

		// 当前发送消息数组
		List<ChatMsg> messages = new ArrayList<ChatMsg>();
		messages.add(chatMsg);
		String messagesJson = gson.toJson(messages);
		String content = "{\"messages\":" + messagesJson + "}";

		long start = System.currentTimeMillis();

		MediaType mediaType = MediaType.parse("application/json");
		RequestBody body = RequestBody.create(mediaType, content);

		System.out.println(content);
		Request request = new Request.Builder().url(url).method("POST", body)
				.addHeader("Content-Type", "application/json").build();

		Response response = HTTP_CLIENT.newCall(request).execute();
		String responseText = response.body().string();
		System.out.println("response返回: \n" + responseText);

		long end = System.currentTimeMillis();
		System.out.println("该回答花费时间为：" + (end - start) / 1000.0 + "秒");

		ObjectMapper objectMapper = new ObjectMapper();
		JsonNode rootNode = objectMapper.readTree(responseText);
		String answer = rootNode.get("result").asText();
		System.out.println(answer);
	}

}

class ChatMsg {

	private String role;
	private String content;

	public String getRole() {
		return role;
	}

	public void setRole(String role) {
		this.role = role;
	}

	public String getContent() {
		return content;
	}

	public void setContent(String content) {
		this.content = content;
	}

}