115 lines
3.2 KiB
Plaintext
115 lines
3.2 KiB
Plaintext
using Newtonsoft.Json.Linq;
|
|
using System;
|
|
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using System.Text;
|
|
using UnityEngine;
|
|
using UnityEngine.Networking;
|
|
|
|
[System.Serializable]
|
|
public struct WenXinRequestMessage
|
|
{
|
|
public string role;
|
|
public string content;
|
|
|
|
public static WenXinRequestMessage UserMessage(string message)
|
|
{
|
|
WenXinRequestMessage result;
|
|
result.role = "user";
|
|
result.content = message;
|
|
return result;
|
|
}
|
|
}
|
|
|
|
[System.Serializable]
|
|
public struct WenXinRequestStruct
|
|
{
|
|
public List<WenXinRequestMessage> messages;
|
|
public string system;
|
|
}
|
|
|
|
public class WenXinRequest
|
|
{
|
|
#region Define
|
|
|
|
#endregion
|
|
|
|
private const string API_KEY = "X8BRDILN7eCD1R2UvUMNBZsN";
|
|
private const string SECRET_KEY = "Nwerie6mJqQJm98GyssJtkrbSCwhhc41";
|
|
private const string URL = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/eb-instant?";
|
|
|
|
private static string s_accessToken;
|
|
|
|
private string m_response;
|
|
public string Response
|
|
{
|
|
get
|
|
{
|
|
return m_response;
|
|
}
|
|
}
|
|
|
|
public string GetUrl()
|
|
{
|
|
return $"{URL}access_token={s_accessToken}";
|
|
}
|
|
|
|
private IEnumerator GetAccessToken()
|
|
{
|
|
var reqParams = new Dictionary<string, string>()
|
|
{
|
|
{"grant_type", "client_credentials"},
|
|
{"client_id", API_KEY},
|
|
{"client_secret", SECRET_KEY},
|
|
};
|
|
using (var req = UnityWebRequest.Post("https://aip.baidubce.com/oauth/2.0/token", reqParams))
|
|
{
|
|
yield return req.SendWebRequest();
|
|
if (req.result == UnityWebRequest.Result.ConnectionError)
|
|
{
|
|
Debug.LogError("Winxin connection error");
|
|
yield break;
|
|
}
|
|
|
|
JObject obj = JObject.Parse(req.downloadHandler.text);
|
|
s_accessToken = obj["access_token"].ToString();
|
|
|
|
Debug.Log($"Access token got {s_accessToken}");
|
|
}
|
|
}
|
|
|
|
public IEnumerator Post(MonoBehaviour context, WenXinRequestStruct winXinRequestStruct)
|
|
{
|
|
string bodyJsonString = JsonUtility.ToJson(winXinRequestStruct);
|
|
|
|
if (string.IsNullOrEmpty(s_accessToken))
|
|
{
|
|
yield return context.StartCoroutine(GetAccessToken());
|
|
}
|
|
|
|
using (var req = new UnityWebRequest(GetUrl(), "POST"))
|
|
{
|
|
byte[] bodyRaw = Encoding.UTF8.GetBytes(bodyJsonString);
|
|
req.uploadHandler = (UploadHandler)new UploadHandlerRaw(bodyRaw);
|
|
req.downloadHandler = (DownloadHandler)new DownloadHandlerBuffer();
|
|
req.SetRequestHeader("Content-Type", "application/json");
|
|
yield return req.SendWebRequest();
|
|
if (req.result == UnityWebRequest.Result.ConnectionError)
|
|
{
|
|
Debug.LogError("Winxin connection error");
|
|
yield break;
|
|
}
|
|
|
|
JObject obj = JObject.Parse(req.downloadHandler.text);
|
|
if (obj == null)
|
|
{
|
|
Debug.LogError("Request is null, why?");
|
|
yield break;
|
|
}
|
|
|
|
m_response = obj["result"].ToString();
|
|
Debug.Log(m_response);
|
|
}
|
|
}
|
|
}
|