MyBook/Assets/script/DeepSeekDialogueManager.cs

167 lines
6.0 KiB
C#
Raw Normal View History

2025-03-17 17:27:32 +08:00
using UnityEngine;
using UnityEngine.Networking;
using System.Collections;
using System.Collections.Generic;
public class DeepSeekDialogueManager : MonoBehaviour
{
// API配置
[Header("API Settings")]
[SerializeField] private string apiKey = "sk-c89fd3342dda4a8eb53cb533607e2e89"; // DeepSeek API密钥
[SerializeField] private string modelName = "deepseek-chat"; // 使用的模型名称
[SerializeField] private string apiUrl = "https://api.deepseek.com/v1/chat/completions"; // API请求地址
// 对话参数
[Header("Dialogue Settings")]
[Range(0, 2)] public float temperature = 0.7f; // 控制生成文本的随机性0-2值越高越随机
[Range(1, 1000)] public int maxTokens = 150; // 生成的最大令牌数(控制回复长度)
// 角色设定
[System.Serializable]
public class NPCCharacter
{
public string name;
[TextArea(3, 10)]
public string personalityPrompt = "你是妤如晨曦,是个性格活泼,聪明可爱的女生。喜欢吃小蛋糕。最喜欢的主人是舒荣森"; // 角色设定提示词
}
[SerializeField] public NPCCharacter npcCharacter;
// 回调委托用于异步处理API响应
public delegate void DialogueCallback(string response, bool isSuccess);
/// <summary>
/// 发送对话请求
/// </summary>
/// <param name="userMessage">玩家的输入内容</param>
/// <param name="callback">回调函数用于处理API响应</param>
public void SendDialogueRequest(string userMessage, DialogueCallback callback)
{
StartCoroutine(ProcessDialogueRequest(userMessage, callback));
}
/// <summary>
/// 处理对话请求的协程
/// </summary>
/// <param name="userInput">玩家的输入内容</param>
/// <param name="callback">回调函数用于处理API响应</param>
private IEnumerator ProcessDialogueRequest(string userInput, DialogueCallback callback)
{
// 构建消息列表,包含系统提示和用户输入
List<Message> messages = new List<Message>
{
new Message { role = "system", content = npcCharacter.personalityPrompt }, // 系统角色设定
new Message { role = "user", content = userInput } // 用户输入
};
// 构建请求体
ChatRequest requestBody = new ChatRequest
{
model = modelName, // 模型名称
messages = messages, // 消息列表
temperature = temperature, // 温度参数
max_tokens = maxTokens // 最大令牌数
};
string jsonBody = JsonUtility.ToJson(requestBody);
Debug.Log("Sending JSON: " + jsonBody); // 调试用打印发送的JSON数据
UnityWebRequest request = CreateWebRequest(jsonBody);
yield return request.SendWebRequest();
if (IsRequestError(request))
{
Debug.LogError($"API Error: {request.responseCode}\n{request.downloadHandler.text}");
callback?.Invoke(null, false);
yield break;
}
DeepSeekResponse response = ParseResponse(request.downloadHandler.text);
if (response != null && response.choices.Length > 0)
{
string npcReply = response.choices[0].message.content;
callback?.Invoke(npcReply, true);
}
else
{
callback?.Invoke(npcCharacter.name + "(陷入沉默)", false);
}
}
/// <summary>
/// 创建UnityWebRequest对象
/// </summary>
/// <param name="jsonBody">请求体的JSON字符串</param>
/// <returns>配置好的UnityWebRequest对象</returns>
private UnityWebRequest CreateWebRequest(string jsonBody)
{
byte[] bodyRaw = System.Text.Encoding.UTF8.GetBytes(jsonBody);
var request = new UnityWebRequest(apiUrl, "POST");
request.uploadHandler = new UploadHandlerRaw(bodyRaw); // 设置上传处理器
request.downloadHandler = new DownloadHandlerBuffer(); // 设置下载处理器
request.SetRequestHeader("Content-Type", "application/json"); // 设置请求头
request.SetRequestHeader("Authorization", $"Bearer {apiKey}"); // 设置认证头
request.SetRequestHeader("Accept", "application/json"); // 设置接受类型
return request;
}
/// <summary>
/// 检查请求是否出错
/// </summary>
/// <param name="request">UnityWebRequest对象</param>
/// <returns>如果请求出错返回true否则返回false</returns>
private bool IsRequestError(UnityWebRequest request)
{
return request.result == UnityWebRequest.Result.ConnectionError ||
request.result == UnityWebRequest.Result.ProtocolError ||
request.result == UnityWebRequest.Result.DataProcessingError;
}
/// <summary>
/// 解析API响应
/// </summary>
/// <param name="jsonResponse">API响应的JSON字符串</param>
/// <returns>解析后的DeepSeekResponse对象</returns>
private DeepSeekResponse ParseResponse(string jsonResponse)
{
try
{
return JsonUtility.FromJson<DeepSeekResponse>(jsonResponse);
}
catch (System.Exception e)
{
Debug.LogError($"JSON解析失败: {e.Message}\n响应内容{jsonResponse}");
return null;
}
}
// 可序列化数据结构
[System.Serializable]
private class ChatRequest
{
public string model; // 模型名称
public List<Message> messages; // 消息列表
public float temperature; // 温度参数
public int max_tokens; // 最大令牌数
}
[System.Serializable]
public class Message
{
public string role; // 角色system/user/assistant
public string content; // 消息内容
}
[System.Serializable]
private class DeepSeekResponse
{
public Choice[] choices; // 生成的选择列表
}
[System.Serializable]
private class Choice
{
public Message message; // 生成的消息
}
}