【Unity】如何在Unity里使用文心一言AI_unity接入文心一言

想要使用文心一言，首先要登录百度智能云千帆控制台

https://cloud.baidu.com/product/wenxinworkshop?track=developer_qianfan_tanchuang

1.在控制台找到应用接入 - 然后点击创建应用

2.填写应用信息

3.创建之后，记下API Key 和 Secret Key，等会会用到

4.打开Unity，建一个简单的聊天场景

5.加入测试代码

public class ChatAI: MonoBehaviour
{
    public string token;
    //这里填写百度千帆大模型里的应用api key
    public string api_key = "xxxxxx";
    //这里填写百度千帆大模型里的应用secret key
    public string secret_key = "xxxxxxxxx";
    //发送按钮
    public Button sendBtn;
    //输入框
    public TMP_InputField info;
    //AI回应
    public TextMeshProUGUI responseText;
    // 历史对话
    private List<message> historyList = new List<message>();

    public void Awake()
    {
        //初始化文心一言,获取token
        StartCoroutine(GetToken());
        sendBtn.onClick.AddListener(OnSend);
    }
   
    public void OnSend()
    {
        OnSpeak(info.text);
    }
    //开始对话
    public void OnSpeak(string talk )
    {
        StartCoroutine(Request(talk));
    }
    private IEnumerator GetToken()
    {
        //获取token的api地址
        string _token_url = string.Format("https://aip.baidubce.com/oauth/2.0/token" + "?client_id={0}&client_secret={1}&grant_type=client_credentials" , api_key, secret_key);
        using (UnityWebRequest request = new UnityWebRequest(_token_url, "POST"))
        {
            request.downloadHandler = (DownloadHandler)new DownloadHandlerBuffer();
            yield return request.SendWebRequest();
            if (request.isDone)
            {
                string msg = request.downloadHandler.text;
                TokenInfo mTokenInfo = JsonUtility.FromJson<TokenInfo>(msg);
                //保存Token
                token = mTokenInfo.access_token;
            }
        }
    }
   
    /// <summary>
    /// 发送数据
    /// </summary> 
    /// <param name="_postWord"></param>
    /// <param name="_callback"></param>
    /// <returns></returns>
    public IEnumerator Request(string talk)
    {
        string url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/";
        string postUrl = url + GetModelType(ModelType.ERNIE_Bot) + "?access_token=" + token;
        historyList.Add(new message("user", talk));
        RequestData postData = new RequestData
        {
            messages = historyList
        };
        using (UnityWebRequest request = new UnityWebRequest(postUrl, "POST"))
        {
            string jsonData = XJsonParser.ToJson<RequestData>(postData);
            byte[] data = System.Text.Encoding.UTF8.GetBytes(jsonData);
            request.uploadHandler = (UploadHandler)new UploadHandlerRaw(data);
            request.downloadHandler = (DownloadHandler)new DownloadHandlerBuffer();
            request.SetRequestHeader("Content-Type", "application/json");
            yield return request.SendWebRequest();
            if (request.responseCode == 200)
            {
                string _msg = request.downloadHandler.text;
                ResponseData response; 
                XJsonParser.ToObject(out response, _msg);
                //加入历史数据
                historyList.Add(new message("assistant", response.result));
                responseText.text = response.result;
            }
        }
    }
 
    /// <summary>
    /// 获取资源
    /// </summary>
    /// <param name="type"></param>
    /// <returns></returns>
    private string GetModelType(ModelType type)
    {
        if (type == ModelType.ERNIE_Bot)
        {
            return "completions";
        }
        if (type == ModelType.ERNIE_Bot_turbo)
        {
            return "eb-instant";
        }
        if (type == ModelType.BLOOMZ_7B)
        {
            return "bloomz_7b1";
        }
        if (type == ModelType.Qianfan_BLOOMZ_7B_compressed)
        {
            return "qianfan_bloomz_7b_compressed";
        }
        if (type == ModelType.ChatGLM2_6B_32K)
        {
            return "chatglm2_6b_32k";
        }
        if (type == ModelType.Llama_2_7B_Chat)
        {
            return "llama_2_7b";
        }
        if (type == ModelType.Llama_2_13B_Chat)
        {
            return "llama_2_13b";
        }
        if (type == ModelType.Llama_2_70B_Chat)
        {
            return "llama_2_70b";
        }
        if (type == ModelType.Qianfan_Chinese_Llama_2_7B)
        {
            return "qianfan_chinese_llama_2_7b";
        }
        if (type == ModelType.AquilaChat_7B)
        {
            return "aquilachat_7b";
        }
        return "";
    }
 
    //发送的数据
    private class RequestData
    {
        //发送的消息
        public List<message> messages = new List<message>();
        //是否流式输出
        public bool stream = false;
        public string user_id = string.Empty;
    }
    
    private class message
    {
        //角色
        public string role =string.Empty;
        //对话内容
        public string content = string.Empty;
        public message() { }
        public message(string _role, string _content  )
        {
            if (_role != "")
            {
                role = _role;
            }
            content = _content;
        }
    }
    //接收的数据
    private class ResponseData
    {
        //本轮对话的id 
        public string id = string.Empty;
        public int created;
        //表示当前子句的序号,只有在流式接口模式下会返回该字段
        public int sentence_id;
        //表示当前子句是否是最后一句,只有在流式接口模式下会返回该字段
        public bool is_end;
        //表示当前子句是否是最后一句,只有在流式接口模式下会返回该字段
        public bool is_truncated;
        //返回的文本
        public string result = string.Empty;
        //表示输入是否存在安全
        public bool need_clear_history;
        //当need_clear_history为true时，此字段会告知第几轮对话有敏感信息，如果是当前问题，ban_round=-1
        public int ban_round;
        //token统计信息，token数 = 汉字数+单词数*1.3 
        public Usage usage = new Usage();
    }
  
    private class Usage
    {
        //问题tokens数
        public int prompt_tokens;
        //回答tokens数
        public int completion_tokens;
        //tokens总数
        public int total_tokens;
    }

   
    /// <summary>
    /// 模型名称
    /// </summary>
    public enum ModelType
    {
        ERNIE_Bot,
        ERNIE_Bot_turbo,
        BLOOMZ_7B,
        Qianfan_BLOOMZ_7B_compressed,
        ChatGLM2_6B_32K,
        Llama_2_7B_Chat,
        Llama_2_13B_Chat,
        Llama_2_70B_Chat,
        Qianfan_Chinese_Llama_2_7B,
        AquilaChat_7B,
    }

    /// <summary>
    /// 返回的token
    /// </summary>
    [System.Serializable]
    public class TokenInfo
    {
        public string access_token = string.Empty;
    }
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221

6.输入问题，然后点击发送。

等待回复会有一点点时间，实际开发的时候，可以加一些过渡动画

7.接收到来着文心一言的回答。

8.本教程只作简单的演示功能，开发者可以根据自己的需求设计出丰富的对话功能。

附官方文档
https://cloud.baidu.com/doc/WENXINWORKSHOP/s/flfmc9do2

9.另外我们平时使用文心一言是免费的，但是代码调用文心一言的API是付费的