121 lines
3.5 KiB
C#
121 lines
3.5 KiB
C#
using System;
|
||
using System.Collections;
|
||
using System.Collections.Generic;
|
||
using System.Text.RegularExpressions;
|
||
using UnityEngine;
|
||
using UnityEngine.Networking;
|
||
|
||
public class chatGPT : LLM
|
||
{
|
||
public chatGPT() {
|
||
url = "https://api.openai.com/v1/completions";
|
||
}
|
||
|
||
/// <summary>
|
||
/// api key
|
||
/// </summary>
|
||
[SerializeField] private string api_key;
|
||
//配置参数
|
||
[SerializeField] private PostData m_PostDataSetting;
|
||
/// <summary>
|
||
/// 发送消息
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
public override void PostMsg(string _msg,Action<string> _callback)
|
||
{
|
||
//提示词
|
||
string message = "当前为角色的人物设定：" + m_Prompt +
|
||
" 回答的语言：" + lan +
|
||
" 接下来是我的提问：" + _msg;
|
||
|
||
|
||
StartCoroutine(Request(message, _callback));
|
||
}
|
||
|
||
/// <summary>
|
||
/// 调用接口返回数据
|
||
/// </summary>
|
||
/// <param name="_postWord"></param>
|
||
/// <param name="_callback"></param>
|
||
/// <returns></returns>
|
||
private IEnumerator Request(string _postWord, System.Action<string> _callback)
|
||
{
|
||
using (UnityWebRequest request = new UnityWebRequest(url, "POST"))
|
||
{
|
||
PostData _postData = new PostData
|
||
{
|
||
model = m_PostDataSetting.model,
|
||
prompt = _postWord,
|
||
max_tokens = m_PostDataSetting.max_tokens,
|
||
temperature = m_PostDataSetting.temperature,
|
||
top_p = m_PostDataSetting.top_p,
|
||
frequency_penalty = m_PostDataSetting.frequency_penalty,
|
||
presence_penalty = m_PostDataSetting.presence_penalty,
|
||
stop = m_PostDataSetting.stop
|
||
};
|
||
|
||
string _jsonText = JsonUtility.ToJson(_postData);
|
||
byte[] data = System.Text.Encoding.UTF8.GetBytes(_jsonText);
|
||
request.uploadHandler = (UploadHandler)new UploadHandlerRaw(data);
|
||
request.downloadHandler = (DownloadHandler)new DownloadHandlerBuffer();
|
||
|
||
request.SetRequestHeader("Content-Type", "application/json");
|
||
request.SetRequestHeader("Authorization", string.Format("Bearer {0}", api_key));
|
||
|
||
yield return request.SendWebRequest();
|
||
|
||
if (request.responseCode == 200)
|
||
{
|
||
string _msg = request.downloadHandler.text;
|
||
TextCallback _textback = JsonUtility.FromJson<TextCallback>(_msg);
|
||
if (_textback != null && _textback.choices.Count > 0)
|
||
{
|
||
|
||
string _backMsg = Regex.Replace(_textback.choices[0].text, @"[\r\n]", "").Replace("？", "");
|
||
_callback(_backMsg);
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
|
||
}
|
||
|
||
#region 数据定义
|
||
|
||
[System.Serializable]
|
||
public class PostData
|
||
{
|
||
public string model;
|
||
public string prompt;
|
||
public int max_tokens=1024;
|
||
public float temperature=0.9f;
|
||
public int top_p;
|
||
public float frequency_penalty;
|
||
public float presence_penalty;
|
||
public string stop;
|
||
}
|
||
/// <summary>
|
||
/// 返回的信息
|
||
/// </summary>
|
||
[System.Serializable]
|
||
public class TextCallback
|
||
{
|
||
public string id;
|
||
public string created;
|
||
public string model;
|
||
public List<TextSample> choices;
|
||
|
||
[System.Serializable]
|
||
public class TextSample
|
||
{
|
||
public string text;
|
||
public string index;
|
||
public string finish_reason;
|
||
}
|
||
|
||
}
|
||
|
||
#endregion
|
||
}
|