当前位置：首页 > news >正文

百度文心一言（ERNIE bot）API接入Android应用

news 来源：原创 2024/5/9 6:11:25

百度文心一言（ERNIE bot）API接入Android应用实践 - 拾一贰叁 - 博客园 (cnblogs.com)

Preface:

现在生成式AI越来越强大了，想在android上实现一个对话助手的功能，大概摸索了一下接入百度文心一言API的方法。

与AI助手交换信息的方式可以这么理解：

我向文心一言发送一个message：你好啊：

[{"role": "user","content": "你好啊"}
]

文心一言回答我：你好，很高兴与你交流。请问你有什么具体的问题或需要帮助吗？我会尽力回答你的问题或与你对话：

{"id":"as-n24a5sytuz","object":"chat.completion","created":1711203238,"result":"你好，请问有什么我可以帮助你的吗？如果你有任何问题或需要帮助，请随时告诉我，我会尽力回答和提供帮助。","is_truncated":false,"need_clear_history":false,"finish_reason":"normal","usage":{"prompt_tokens":1,"completion_tokens":28,"total_tokens":29}
}

接着我继续发送message：今天是几号呢？......

[{"role": "user","content": "你好啊"},{"role": "assistant","content": "你好，很高兴与你交流。请问你有什么具体的问题或需要帮助吗？我会尽力回答你的问题或与你对话。"},{"role": "user","content": "今天是几号呢"}
]

每一次发送message，都要带上之前的对话，这样才能实现连续对话的功能。

具体实现

在Android应用的AndroidManifest.xml文件中添加网络访问权限：

<uses-permission android:name="android.permission.INTERNET" />

在build.gradle中添加必要的依赖：

implementation 'com.squareup.okhttp3:okhttp:4.9.3'

接下来注册开发者账户、往里边充钱啥的，完成这些之后，在百度智能云控制台 (baidu.com)创建一个新应用，

如上图所示，我们主要需要API Key和Secret Key这俩东西

创建一个新的类以处理文心一言的API信息：WenXin.java，在Activity里需要实现文心一言的对话功能只需调用这个类就好了。

package com.example.wearespeakers;import android.view.View;
import com.google.gson.Gson;
import okhttp3.*;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;import java.io.*;/**主要用于实现对接文心一言API的功能*/
public class WenXin{public static final String APP_ID = "56****59";//这个似乎还用不到public static final String API_KEY = "oQtU**********wePzF";//填你自己应用的apikeypublic static final String SECRET_KEY = "LxfNE*************W2UW0eX";//填你自己应用的secretkeypublic JSONArray Dialogue_Content;//用来储存对话内容，当然初始是空的WenXin(){//构造函数，先初始化Dialogue_Content一下，此时里边是空的啥也没有//不过也可以预先添加对话，以实现一些希望的业务功能Dialogue_Content=new JSONArray();}static final OkHttpClient HTTP_CLIENT = new OkHttpClient().newBuilder().build();public String GetAnswer(String user_msg) throws IOException, JSONException {JSONObject jsonObject = new JSONObject();jsonObject.put("role", "user");jsonObject.put("content", user_msg);// 将JSONObject添加到JSONArray中//这里就是把用户说的话添加进对话内容里，然后发给文心一言Dialogue_Content.put(jsonObject);MediaType mediaType = MediaType.parse("application/json");//这是一行参考代码，只能进行一次对话，要想多次对话就必须动态添加历史对话的内容//RequestBody body = RequestBody.create(mediaType,  "{\"messages\":[{\"role\":\"user\",\"content\":\"你好啊\"}],\"disable_search\":false,\"enable_citation\":false}");RequestBody body = RequestBody.create(mediaType,  "{\"messages\":" +Dialogue_Content.toString() +",\"disable_search\":false,\"enable_citation\":false}");Request request = new Request.Builder().url("https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token=" +getAccessToken()).method("POST", body).addHeader("Content-Type", "application/json").build();Response response = HTTP_CLIENT.newCall(request).execute();//解析出文心一言的回答JSONObject json_feedback = new JSONObject(response.body().string());//这里在开发的时候遇到了一个问题，注意response在上一行被取出里边的内容之后就自动关闭了，不能多次传参。String re=json_feedback.getString("result");//接下来把文心一言的回答加入到Dialogue_Content中JSONObject jsontmp=new JSONObject();jsontmp.put("assistant",re);Dialogue_Content.put(jsontmp);return re;}/*** 从用户的AK，SK生成鉴权签名（Access Token）** @return 鉴权签名（Access Token）* @throws IOException IO异常*/public String getAccessToken() throws IOException, JSONException {MediaType mediaType = MediaType.parse("application/x-www-form-urlencoded");RequestBody body = RequestBody.create(mediaType, "grant_type=client_credentials&client_id=" + API_KEY+ "&client_secret=" + SECRET_KEY);Request request = new Request.Builder().url("https://aip.baidubce.com/oauth/2.0/token").method("POST", body).addHeader("Content-Type", "application/x-www-form-urlencoded").build();Response response = HTTP_CLIENT.newCall(request).execute();return new JSONObject(response.body().string()).getString("access_token");}
}

在Activity中是这样写的（Activity里的RecyclerView对话界面参考Android RecyclerView的使用（以实现一个简单的动态聊天界面为例），下边大多数都是无关的代码，主要就那几行）：

package com.example.wearespeakers;
import android.app.Activity;
import android.content.Intent;
import android.os.Handler;
import android.os.Looper;
import android.os.Message;
import android.view.View;
import android.widget.Button;
import android.widget.EditText;
import androidx.appcompat.app.AppCompatActivity;
import android.os.Bundle;
import androidx.recyclerview.widget.LinearLayoutManager;
import androidx.recyclerview.widget.RecyclerView;
import org.json.JSONException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import static java.security.AccessController.getContext;//此activity主要用来实现聊天界面
public class ChatActivity extends Activity {private EditText et_chat;private Button btn_send,btn_chat_return;private ChatlistAdapter chatAdapter;private List<Chatlist> mDatas;private RecyclerView rc_chatlist;final int MESSAGE_UPDATE_VIEW = 1;@Overrideprotected void onCreate(Bundle savedInstanceState) {super.onCreate(savedInstanceState);setContentView(R.layout.activity_chat);init();//聊天信息mDatas = new ArrayList<Chatlist>();Chatlist C1;C1=new Chatlist("ABC：","Hello,world!");mDatas.add(C1);Chatlist C2;C2=new Chatlist("DEF：","This is a new app.");mDatas.add(C2);//可以通过数据库插入数据chatAdapter=new ChatlistAdapter(this,mDatas);LinearLayoutManager layoutManager = new LinearLayoutManager(this );rc_chatlist.setLayoutManager(layoutManager);//如果可以确定每个item的高度是固定的，设置这个选项可以提高性能rc_chatlist.setHasFixedSize(true);//创建并设置Adapterrc_chatlist.setAdapter(chatAdapter);//点击btn_send发送聊天信息btn_send.setOnClickListener(new View.OnClickListener() {@Overridepublic void onClick(View v) {//用户的提问String user_ask=et_chat.getText().toString();//获取输入框里的信息Chatlist C3;C3=new Chatlist("User：",user_ask);mDatas.add(C3);chatAdapter.ResetChatlistAdapter(mDatas);rc_chatlist.setAdapter(chatAdapter);//文心一言的回答（以下才是用到WenXin.java的地方）new Thread(new Runnable(){@Overridepublic void run() {//请求详情Chatlist C4;try {WenXin wx=new WenXin();C4=new Chatlist("WenXin：",wx.GetAnswer(user_ask));} catch (IOException | JSONException e) {throw new RuntimeException(e);} finally {}mDatas.add(C4);chatAdapter.ResetChatlistAdapter(mDatas);Message msg = new Message();msg.what = MESSAGE_UPDATE_VIEW;ChatActivity.this.gHandler.sendMessage(msg);}}).start();/*为什么要弄new Thread...这样呢？因为像这种网络请求往往有延迟，需要新开一个进程去处理，与下面的gHandler相对应当app收到来自文心一言的回答后，就去通知gHandler更新界面，把回答的段落显示出来*/}});//点击返回,返回mainActivitybtn_chat_return.setOnClickListener(new View.OnClickListener() {@Overridepublic void onClick(View v) {Intent intent=new Intent(ChatActivity.this,MainActivity.class);startActivity(intent);ChatActivity.this.finish();}});}private void init(){//执行一些初始化操作btn_send=findViewById(R.id.btn_send);et_chat=findViewById(R.id.et_chat);btn_chat_return=findViewById(R.id.btn_chat_return);rc_chatlist=(RecyclerView) findViewById(R.id.rc_chatlist);}public Handler gHandler = new Handler(Looper.getMainLooper()) {@Overridepublic void handleMessage(Message msg) {if (msg.what == MESSAGE_UPDATE_VIEW) {rc_chatlist.setAdapter(chatAdapter);//更新对话界面}}};}

其实只需要关注new Thread和gHandler部分的代码即可。

效果：