287 lines
9.1 KiB
C#
287 lines
9.1 KiB
C#
|
using Newtonsoft.Json;
|
|||
|
|
using System;
|
|||
|
|
using System.Collections;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
using System.Text;
|
|||
|
|
using System.Text.RegularExpressions;
|
|||
|
|
using TMPro;
|
|||
|
|
using UnityEngine;
|
|||
|
|
using UnityEngine.Networking;
|
|||
|
|
using UnityEngine.UI;
|
|||
|
|
|
|||
|
|
public class DeepSeekReasonerStreamManager : MonoBehaviour
|
|||
|
|
{
|
|||
|
|
// API 配置
|
|||
|
|
[Header("API Settings")]
|
|||
|
|
[SerializeField] private string apiKey = "sk-c89fd3342dda4a8eb53cb533607e2e89"; // 替换成你的 API Key
|
|||
|
|
[SerializeField] private string modelName = "deepseek-reasoner";
|
|||
|
|
[SerializeField] private string apiUrl = "https://api.deepseek.com/v1/chat/completions";
|
|||
|
|
|
|||
|
|
// 推理参数
|
|||
|
|
[Header("Inference Settings")]
|
|||
|
|
[Range(0, 2)] public float temperature = 0.7f;
|
|||
|
|
[Range(1, 1000)] public int maxTokens = 150;
|
|||
|
|
|
|||
|
|
// UI 控件（请在 Inspector 中指定，只需一个 Text 节点）
|
|||
|
|
[Header("UI Elements")]
|
|||
|
|
public TextMeshProUGUI dialogueText; // 显示推理过程及最终回复
|
|||
|
|
|
|||
|
|
public InfoShow infoShow;
|
|||
|
|
|
|||
|
|
// 内部缓冲区
|
|||
|
|
private StringBuilder reasoningBuffer = new StringBuilder();
|
|||
|
|
private StringBuilder contentBuffer = new StringBuilder();
|
|||
|
|
private bool streamingCompleted = false;
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 开始流式推理请求，传入对话消息列表（例如第一轮对话）
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="messages">对话消息列表，格式参见 DeepSeekMessage</param>
|
|||
|
|
public void SendStreamReasonerRequest(List<DeepSeekMessage> messages)
|
|||
|
|
{
|
|||
|
|
// 清空之前的缓冲区及 UI 显示
|
|||
|
|
reasoningBuffer.Clear();
|
|||
|
|
contentBuffer.Clear();
|
|||
|
|
dialogueText.text = "";
|
|||
|
|
streamingCompleted = false;
|
|||
|
|
|
|||
|
|
StartCoroutine(ProcessStreamReasonerRequest(messages));
|
|||
|
|
// 开启打字机效果协程，先显示推理过程
|
|||
|
|
StartCoroutine(AnimateReasoningText());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 流式请求的协程：发送请求并通过自定义下载处理器实时处理返回块
|
|||
|
|
/// </summary>
|
|||
|
|
private IEnumerator ProcessStreamReasonerRequest(List<DeepSeekMessage> messages)
|
|||
|
|
{
|
|||
|
|
ChatStreamRequest requestBody = new ChatStreamRequest
|
|||
|
|
{
|
|||
|
|
model = modelName,
|
|||
|
|
messages = messages,
|
|||
|
|
stream = true
|
|||
|
|
};
|
|||
|
|
GameManager.Instance.infoTGo.text = "智慧正在酝酿中，请稍候…";
|
|||
|
|
string jsonBody = JsonUtility.ToJson(requestBody);
|
|||
|
|
Debug.Log("Sending streaming JSON: " + jsonBody);
|
|||
|
|
|
|||
|
|
// 使用自定义 StreamingDownloadHandler 逐块处理响应
|
|||
|
|
StreamingDownloadHandler downloadHandler = new StreamingDownloadHandler();
|
|||
|
|
downloadHandler.onChunkReceived = ProcessChunk;
|
|||
|
|
|
|||
|
|
using (UnityWebRequest request = new UnityWebRequest(apiUrl, "POST"))
|
|||
|
|
{
|
|||
|
|
byte[] bodyRaw = Encoding.UTF8.GetBytes(jsonBody);
|
|||
|
|
request.uploadHandler = new UploadHandlerRaw(bodyRaw);
|
|||
|
|
request.downloadHandler = downloadHandler;
|
|||
|
|
request.SetRequestHeader("Content-Type", "application/json");
|
|||
|
|
request.SetRequestHeader("Authorization", $"Bearer {apiKey}");
|
|||
|
|
request.SetRequestHeader("Accept", "application/json");
|
|||
|
|
|
|||
|
|
yield return request.SendWebRequest();
|
|||
|
|
|
|||
|
|
if (IsRequestError(request))
|
|||
|
|
{
|
|||
|
|
Debug.LogError($"API Error: {request.responseCode}\n{request.downloadHandler.text}");
|
|||
|
|
yield break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
streamingCompleted = true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 对每个返回块进行解析，如果存在推理内容则追加到 reasoningBuffer，
|
|||
|
|
/// 否则将最终结果追加到 contentBuffer（不直接更新 UI）。
|
|||
|
|
/// </summary>
|
|||
|
|
private void ProcessChunk(string chunk)
|
|||
|
|
{
|
|||
|
|
chunk = chunk.Trim();
|
|||
|
|
if (string.IsNullOrEmpty(chunk))
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
if (chunk.Contains("keep-alive"))
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
if (chunk.StartsWith("data:"))
|
|||
|
|
{
|
|||
|
|
chunk = chunk.Substring("data:".Length).Trim();
|
|||
|
|
if (chunk == "[DONE]")
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
ChatStreamResponseChunk chunkData = JsonUtility.FromJson<ChatStreamResponseChunk>(chunk);
|
|||
|
|
if (chunkData != null && chunkData.choices != null && chunkData.choices.Length > 0)
|
|||
|
|
{
|
|||
|
|
StreamingDelta delta = chunkData.choices[0].delta;
|
|||
|
|
if (!string.IsNullOrEmpty(delta.reasoning_content))
|
|||
|
|
{
|
|||
|
|
reasoningBuffer.Append(delta.reasoning_content);
|
|||
|
|
}
|
|||
|
|
else if (!string.IsNullOrEmpty(delta.content))
|
|||
|
|
{
|
|||
|
|
contentBuffer.Append(delta.content);
|
|||
|
|
// 此处不直接更新 dialogueText，待流式完成后统一显示最终结果
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
catch (Exception e)
|
|||
|
|
{
|
|||
|
|
Debug.LogError("解析返回块失败: " + e.Message + "\nChunk content: " + chunk);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 实现打字机效果：不断检查 reasoningBuffer 中的新字符，并逐个显示到 dialogueText，
|
|||
|
|
/// 流式加载结束后，替换显示最终回复内容。
|
|||
|
|
/// </summary>
|
|||
|
|
private IEnumerator AnimateReasoningText()
|
|||
|
|
{
|
|||
|
|
string displayed = "";
|
|||
|
|
while (!streamingCompleted || displayed.Length < reasoningBuffer.Length)
|
|||
|
|
{
|
|||
|
|
if (displayed.Length < reasoningBuffer.Length)
|
|||
|
|
{
|
|||
|
|
displayed = reasoningBuffer.ToString().Substring(0, displayed.Length + 1);
|
|||
|
|
dialogueText.text = displayed;
|
|||
|
|
}
|
|||
|
|
yield return new WaitForSeconds(0.05f);
|
|||
|
|
}
|
|||
|
|
// 推理过程已显示完毕，等待片刻后替换为最终结果
|
|||
|
|
yield return new WaitForSeconds(0.5f);
|
|||
|
|
|
|||
|
|
|
|||
|
|
//dialogueText.text = contentBuffer.ToString();
|
|||
|
|
linkidResponse(contentBuffer.ToString());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void linkidResponse(string response)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
Dictionary<string, string> dictionary = JsonConvert.DeserializeObject<Dictionary<string, string>>(response);
|
|||
|
|
infoShow.ShowInfo(dictionary);
|
|||
|
|
}
|
|||
|
|
catch (System.Exception)
|
|||
|
|
{
|
|||
|
|
infoShow.ShowInfo(
|
|||
|
|
new Dictionary<string, string>
|
|||
|
|
{
|
|||
|
|
{"什么是潜艇?","一种潜水用具"},
|
|||
|
|
{"什么是航母?","一种船"}
|
|||
|
|
}
|
|||
|
|
);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
// 解析 [] 内的内容，转换成 <link> 形式
|
|||
|
|
public string ParseClickableText(string rawText)
|
|||
|
|
{
|
|||
|
|
return Regex.Replace(rawText, @"\[(.*?)\]", "<link=$1><color=yellow><u>$1</u></color></link>");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 检查请求是否出现错误
|
|||
|
|
/// </summary>
|
|||
|
|
private bool IsRequestError(UnityWebRequest request)
|
|||
|
|
{
|
|||
|
|
return request.result == UnityWebRequest.Result.ConnectionError ||
|
|||
|
|
request.result == UnityWebRequest.Result.ProtocolError ||
|
|||
|
|
request.result == UnityWebRequest.Result.DataProcessingError;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 可序列化的请求和响应数据结构
|
|||
|
|
|
|||
|
|
[Serializable]
|
|||
|
|
public class DeepSeekMessage
|
|||
|
|
{
|
|||
|
|
public string role; // "user", "assistant", "system" 等
|
|||
|
|
public string content; // 消息内容
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
[Serializable]
|
|||
|
|
private class ChatStreamRequest
|
|||
|
|
{
|
|||
|
|
public string model;
|
|||
|
|
public List<DeepSeekMessage> messages;
|
|||
|
|
public bool stream;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
[Serializable]
|
|||
|
|
private class StreamingDelta
|
|||
|
|
{
|
|||
|
|
public string reasoning_content;
|
|||
|
|
public string content;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
[Serializable]
|
|||
|
|
private class StreamingChoice
|
|||
|
|
{
|
|||
|
|
public StreamingDelta delta;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
[Serializable]
|
|||
|
|
private class ChatStreamResponseChunk
|
|||
|
|
{
|
|||
|
|
public StreamingChoice[] choices;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 自定义流式下载处理器：将接收到的数据按行分割，并调用回调处理每一行 JSON 数据
|
|||
|
|
/// </summary>
|
|||
|
|
public class StreamingDownloadHandler : DownloadHandlerScript
|
|||
|
|
{
|
|||
|
|
public Action<string> onChunkReceived;
|
|||
|
|
private StringBuilder buffer = new StringBuilder();
|
|||
|
|
|
|||
|
|
public StreamingDownloadHandler() : base(new byte[1024])
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
protected override bool ReceiveData(byte[] data, int dataLength)
|
|||
|
|
{
|
|||
|
|
if (data == null || dataLength <= 0)
|
|||
|
|
return false;
|
|||
|
|
|
|||
|
|
string chunk = Encoding.UTF8.GetString(data, 0, dataLength);
|
|||
|
|
buffer.Append(chunk);
|
|||
|
|
|
|||
|
|
// 假设每个 JSON 块以换行符结尾
|
|||
|
|
string fullText = buffer.ToString();
|
|||
|
|
string[] lines = fullText.Split(new[] { "\n" }, StringSplitOptions.None);
|
|||
|
|
// 处理除最后一行外的所有完整行
|
|||
|
|
for (int i = 0; i < lines.Length - 1; i++)
|
|||
|
|
{
|
|||
|
|
string line = lines[i].Trim();
|
|||
|
|
if (!string.IsNullOrEmpty(line))
|
|||
|
|
{
|
|||
|
|
onChunkReceived?.Invoke(line);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 保留最后可能不完整的那一行
|
|||
|
|
buffer.Clear();
|
|||
|
|
buffer.Append(lines[lines.Length - 1]);
|
|||
|
|
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
protected override void CompleteContent()
|
|||
|
|
{
|
|||
|
|
string remaining = buffer.ToString().Trim();
|
|||
|
|
if (!string.IsNullOrEmpty(remaining))
|
|||
|
|
{
|
|||
|
|
onChunkReceived?.Invoke(remaining);
|
|||
|
|
}
|
|||
|
|
base.CompleteContent();
|
|||
|
|
}
|
|||
|
|
}
|