zc

Source/WulaFallenEmpire/EventSystem/AI/UI/Dialog_AIConversation.cs

@@ -7,6 +7,7 @@ using System.Threading.Tasks;
 using RimWorld;
 using UnityEngine;
 using Verse;
+using WulaFallenEmpire.EventSystem.AI;
 using WulaFallenEmpire.EventSystem.AI.Tools;
 using System.Text.RegularExpressions;
 
@@ -22,6 +23,7 @@ namespace WulaFallenEmpire.EventSystem.AI.UI
         private Vector2 _scrollPosition = Vector2.zero;
         private bool _scrollToBottom = false;
         private List<AITool> _tools = new List<AITool>();
+        private AIIntelligenceCore _core;
         private Dictionary<int, Texture2D> _portraits = new Dictionary<int, Texture2D>();
         private static readonly Regex ExpressionTagRegex = new Regex(@"\[EXPR\s*:\s*([1-6])\s*\]", RegexOptions.IgnoreCase);
         private bool _lastActionExecuted = false;
@@ -111,12 +113,12 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
             this.absorbInputAroundWindow = false;
             this.doCloseX = true;
             this.doWindowBackground = Dialog_CustomDisplay.Config.showMainWindow;
-            this.drawShadow = false; // 禁用阴影
+            this.drawShadow = Dialog_CustomDisplay.Config.showMainWindow;
             this.closeOnClickedOutside = false;
             this.draggable = true;
             this.resizeable = true;
 
-            // 关键修改：禁止Enter键自动关闭窗口
+            // 关键修改：ç¦<EFBFBD>æ­¢Enter键自动关闭窗å<EFBFBD>?
             this.closeOnAccept = false;
 
             _tools.Add(new Tool_SpawnResources());
@@ -138,11 +140,63 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
             Instance = this;
             base.PostOpen();
             LoadPortraits();
-            StartConversation();
+
+            _core = Find.World?.GetComponent<AIIntelligenceCore>();
+            if (_core != null)
+            {
+                _core.InitializeConversation(def.defName);
+                _core.OnMessageReceived += OnCoreMessageReceived;
+                _core.OnThinkingStateChanged += OnCoreThinkingStateChanged;
+                _core.OnExpressionChanged += OnCoreExpressionChanged;
+
+                _history = _core.GetHistorySnapshot();
+                _isThinking = _core.IsThinking;
+                SyncPortraitFromCore();
+            }
+            else
+            {
+                StartConversation();
+            }
+        }
+
+        private void OnCoreMessageReceived(string message)
+        {
+            if (_core == null)
+            {
+                return;
+            }
+
+            _history = _core.GetHistorySnapshot();
+            _scrollToBottom = true;
+        }
+
+        private void OnCoreThinkingStateChanged(bool isThinking)
+        {
+            _isThinking = isThinking;
+        }
+
+        private void OnCoreExpressionChanged(int id)
+        {
+            SetPortrait(id);
+        }
+
+        private void SyncPortraitFromCore()
+        {
+            if (_core == null)
+            {
+                return;
+            }
+
+            SetPortrait(_core.ExpressionId);
         }
 
         public List<(string role, string message)> GetHistorySnapshot()
         {
+            if (_core != null)
+            {
+                return _core.GetHistorySnapshot();
+            }
+
             return _history?.ToList() ?? new List<(string role, string message)>();
         }
 
@@ -795,7 +849,7 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
                     string cleaned = StripXmlTags(reply)?.Trim() ?? "";
                     if (string.IsNullOrWhiteSpace(cleaned))
                     {
-                        cleaned = "（系统）AI 返回了工具调用（XML），已被拦截。请重试或输入 /clear 清空上下文。";
+                    cleaned = "AI returned a tool-only response (XML), which was blocked. Retry or use /clear to reset context.";
                     }
                     reply = cleaned;
                 }
@@ -827,7 +881,7 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
             if (matches.Count == 0)
             {
                 UpdatePhaseToolLedger(phase, false, new List<string>());
-                _history.Add(("toolcall", "<no_action/>"));
+                _history.Add(("assistant", "<no_action/>"));
                 _history.Add(("tool", $"[Tool Results]\nTool 'no_action' Result: No action taken.\n{guidance}"));
                 PersistHistory();
                 UpdateActionLedgerNote();
@@ -836,7 +890,7 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
             if (matches.Count == 1 && matches[0].Groups[1].Value.Equals("no_action", StringComparison.OrdinalIgnoreCase))
             {
                 UpdatePhaseToolLedger(phase, false, new List<string>());
-                _history.Add(("toolcall", "<no_action/>"));
+                _history.Add(("assistant", "<no_action/>"));
                 _history.Add(("tool", $"[Tool Results]\nTool 'no_action' Result: No action taken.\n{guidance}"));
                 PersistHistory();
                 UpdateActionLedgerNote();
@@ -1178,6 +1232,12 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
         {
             if (background != null) GUI.DrawTexture(inRect, background, ScaleMode.ScaleAndCrop);
 
+            if (_core != null)
+            {
+                _history = _core.GetHistorySnapshot();
+                _isThinking = _core.IsThinking;
+            }
+
             // 定义边è·<C3A8>
             float margin = 15f;
             Rect paddedRect = inRect.ContractedBy(margin);
@@ -1185,7 +1245,7 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
             float curY = paddedRect.y;
             float width = paddedRect.width;
 
-            // 立绘不需要边距，所以使用原始inRect的位置
+            // 立绘ä¸<EFBFBD>需è¦<EFBFBD>è¾¹è·<EFBFBD>，所以使用原始inRectçš„ä½<EFBFBD>ç½?
             if (portrait != null)
             {
                 Rect scaledPortraitRect = Dialog_CustomDisplay.Config.GetScaledRect(Dialog_CustomDisplay.Config.portraitSize, inRect, true);
@@ -1210,15 +1270,15 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
             string name = def.characterName ?? "The Legion";
             float nameHeight = Text.CalcHeight(name, width);
 
-            // 创建名字的矩形，使其在窗口水平居中
+            // 创建å<EFBFBD><EFBFBD>字的矩形，使其在窗å<EFBFBD>£æ°´å¹³å±…ä¸?
             Rect nameRect = new Rect(paddedRect.x, curY, width, nameHeight);
             Text.Anchor = TextAnchor.UpperCenter;  // 改为上中对é½<C3A9>
             Widgets.Label(nameRect, name);
-            Text.Anchor = TextAnchor.UpperLeft;    // 恢复左对齐
+            Text.Anchor = TextAnchor.UpperLeft;    // æ<EFBFBD>¢å¤<EFBFBD>左对é½?
 
             curY += nameHeight + 10f;
 
-            // 计算输入框高度、选项高度和聊天历史高度
+            // 计算输入框高度ã€<EFBFBD>选项高度和è<EFBFBD>Šå¤©åŽ†å<EFBFBD>²é«˜åº?
             float inputHeight = 30f;
             float optionsHeight = _options.Any() ? 100f : 0f;
             float spacing = 10f;
@@ -1247,13 +1307,13 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
 
             curY += optionsHeight + spacing;
 
-            // 输入框区域 - 使用带边距的矩形
+            // 输入框区åŸ?- 使用带边è·<C3A8>的矩形
             Rect inputRect = new Rect(paddedRect.x, curY, width, inputHeight);
 
             // ä¿<C3A4>存当å‰<C3A5>字体
             var originalFont = Text.Font;
 
-            // 设置更小的字体
+            // 设置更å°<EFBFBD>çš„å­—ä½?
             if (Text.Font == GameFont.Small)
             {
                 // 使用 Tiny 字体
@@ -1261,18 +1321,18 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
             }
             else
             {
-                // 如果当前不是 Small，降一级
+                // 如果当å‰<EFBFBD>ä¸<EFBFBD>是 Small，é™<C3A9>一çº?
                 Text.Font = GameFont.Small;
             }
 
-            // 计算输入框文本高度
+            // 计算输入框文本高�
             float textFieldHeight = Text.CalcHeight("Test", inputRect.width - 85);
             Rect textFieldRect = new Rect(inputRect.x, inputRect.y + (inputHeight - textFieldHeight) / 2, inputRect.width - 85, textFieldHeight);
 
             _inputText = Widgets.TextField(textFieldRect, _inputText);
 
-            // 发送按钮 - 使用与Dialog_CustomDisplay相同的自定义按钮样式
-            // 保存当前状态
+            // å<EFBFBD>‘é€<EFBFBD>按é’?- 使用与Dialog_CustomDisplay相å<C2B8>Œçš„自定义按钮样å¼<C3A5>
+            // ä¿<EFBFBD>存当å‰<EFBFBD>状æ€?
             var originalAnchor = Text.Anchor;
             var originalColor = GUI.color;
 
@@ -1286,7 +1346,7 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
             // 使用基类的DrawCustomButton方法绘制按钮（与Dialog_CustomDisplay一致）
             base.DrawCustomButton(sendButtonRect, "Wula_AI_Send".Translate(), isEnabled: true);
 
-            // 恢复状态
+            // æ<EFBFBD>¢å¤<EFBFBD>状æ€?
             GUI.color = originalColor;
             Text.Anchor = originalAnchor;
             Text.Font = originalFont;
@@ -1295,13 +1355,13 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
             bool sendButtonPressed = Widgets.ButtonInvisible(sendButtonRect);
 
             // 直接在DoWindowContents中处ç<E2809E>†Enter键，而ä¸<C3A4>是调用å<C2A8>•ç‹¬çš„方法
-            // 这是为了确保事件在正确的时机被处理
+            // 这是为了确ä¿<EFBFBD>事件在正确的时机被处ç<EFBFBD>?
             if (Event.current.type == EventType.KeyDown)
             {
-                // 检查是否按下了Enter键（主键盘或小键盘的Enter）
+                // 检查是å<EFBFBD>¦æŒ‰ä¸‹äº†Enter键（主键盘或å°<EFBFBD>键盘的Enterï¼?
                 if ((Event.current.keyCode == KeyCode.Return || Event.current.keyCode == KeyCode.KeypadEnter) && !string.IsNullOrEmpty(_inputText))
                 {
-                    // 如果AI正在思考，不处理Enter键
+                    // 如果AI正在æ€<EFBFBD>考，ä¸<EFBFBD>处ç<EFBFBD>†Enteré”?
                     if (!_isThinking)
                     {
                         SelectOption(_inputText);
@@ -1318,52 +1378,50 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
                 }
             }
 
-            // 处理鼠标点击发送按钮
+            // 处ç<EFBFBD>†é¼ æ ‡ç‚¹å‡»å<EFBFBD>‘é€<EFBFBD>按é’?
             if (sendButtonPressed && !string.IsNullOrEmpty(_inputText))
             {
                 SelectOption(_inputText);
                 _inputText = "";
             }
         }
-        // 气泡样式常量
-        private const float BubblePadding = 8f;
-        private const float AvatarSize = 32f;
-        private const float MessageSpacing = 8f;
-        private const float MaxBubbleWidthRatio = 0.75f;
-        private const float BubbleCornerRadius = 8f;
-
         private void DrawChatHistory(Rect rect)
         {
             var originalFont = Text.Font;
             var originalAnchor = Text.Anchor;
-            var originalColor = GUI.color;
 
             try
             {
                 float viewHeight = 0f;
                 var filteredHistory = _history.Where(e => e.role != "tool" && e.role != "system" && e.role != "toolcall").ToList();
 
-                float containerWidth = rect.width - 16f;
-                float maxBubbleWidth = containerWidth * MaxBubbleWidthRatio;
+                // 添加内边�
+                float innerPadding = 5f;
+                float contentWidth = rect.width - 16f - innerPadding * 2;
 
-                // 预计算高度
-                Text.Font = GameFont.Small;
+                // 预计算高åº?- 使用å°<C3A5>å­—ä½?
                 for (int i = 0; i < filteredHistory.Count; i++)
                 {
                     var entry = filteredHistory[i];
                     string text = entry.role == "assistant" ? ParseResponseForDisplay(entry.message) : entry.message;
-                    if (string.IsNullOrWhiteSpace(text) || (entry.role == "user" && text.StartsWith("[Tool Results]"))) continue;
+                    if (string.IsNullOrEmpty(text) || (entry.role == "user" && text.StartsWith("[Tool Results]"))) continue;
+                    bool isLastMessage = i == filteredHistory.Count - 1;
 
-                    float textWidth = maxBubbleWidth - (BubblePadding * 2);
-                    if (entry.role == "assistant") textWidth -= (AvatarSize + BubblePadding);
-                    
-                    float textHeight = Text.CalcHeight(text, textWidth);
-                    float bubbleHeight = textHeight + (BubblePadding * 2);
-                    float rowHeight = Mathf.Max(bubbleHeight, entry.role == "assistant" ? AvatarSize : 0f);
-                    viewHeight += rowHeight + MessageSpacing;
+                    // 设置更å°<C3A5>çš„å­—ä½?
+                    if (isLastMessage && entry.role == "assistant")
+                    {
+                        Text.Font = GameFont.Small; // 原æ<C5B8>¥æ˜?Medium，改ä¸?Small
+                    }
+                    else
+                    {
+                        Text.Font = GameFont.Tiny; // 原æ<C5B8>¥æ˜?Small，改ä¸?Tiny
+                    }
+                    // 增加padding
+                    float padding = (isLastMessage && entry.role == "assistant") ? 30f : 15f;
+                    viewHeight += Text.CalcHeight(text, contentWidth) + padding + 10f;
                 }
 
-                Rect viewRect = new Rect(0f, 0f, containerWidth, viewHeight);
+                Rect viewRect = new Rect(0f, 0f, rect.width - 16f, viewHeight);
                 if (_scrollToBottom)
                 {
                     _scrollPosition.y = float.MaxValue;
@@ -1378,22 +1436,36 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
                     var entry = filteredHistory[i];
                     string text = entry.role == "assistant" ? ParseResponseForDisplay(entry.message) : entry.message;
 
-                    if (string.IsNullOrWhiteSpace(text) || (entry.role == "user" && text.StartsWith("[Tool Results]"))) continue;
+                    if (string.IsNullOrEmpty(text) || (entry.role == "user" && text.StartsWith("[Tool Results]"))) continue;
+                    bool isLastMessage = i == filteredHistory.Count - 1;
 
-                    Text.Font = GameFont.Small;
-
-                    if (entry.role == "user")
+                    // 设置更å°<C3A5>çš„å­—ä½?
+                    if (isLastMessage && entry.role == "assistant")
                     {
-                        // 用户消息 - 右对齐，深蓝色气泡
-                        DrawUserMessage(new Rect(0, curY, containerWidth, 0), text, out float userMsgHeight);
-                        curY += userMsgHeight + MessageSpacing;
+                        Text.Font = GameFont.Small; // 原æ<C5B8>¥æ˜?Medium，改ä¸?Small
                     }
                     else
                     {
-                        // AI消息 - 左对齐，带头像，深红色气泡
-                        DrawAIMessage(new Rect(0, curY, containerWidth, 0), text, out float aiMsgHeight);
-                        curY += aiMsgHeight + MessageSpacing;
+                        Text.Font = GameFont.Tiny; // 原æ<C5B8>¥æ˜?Small，改ä¸?Tiny
                     }
+
+                    float padding = (isLastMessage && entry.role == "assistant") ? 30f : 15f;
+                    float height = Text.CalcHeight(text, contentWidth) + padding;
+
+                    // 添加内边�
+                    Rect labelRect = new Rect(innerPadding, curY, contentWidth, height);
+
+                    if (entry.role == "user")
+                    {
+                        Text.Anchor = TextAnchor.MiddleRight;
+                        Widgets.Label(labelRect, $"<color=#add8e6>{text}</color>");
+                    }
+                    else
+                    {
+                        Text.Anchor = TextAnchor.MiddleLeft;
+                        Widgets.Label(labelRect, $"P.I.A: {text}");
+                    }
+                    curY += height + 10f;
                 }
                 Widgets.EndScrollView();
             }
@@ -1401,141 +1473,9 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
             {
                 Text.Font = originalFont;
                 Text.Anchor = originalAnchor;
-                GUI.color = originalColor;
             }
         }
 
-        private void DrawUserMessage(Rect containerRect, string text, out float totalHeight)
-        {
-            float maxBubbleWidth = containerRect.width * MaxBubbleWidthRatio;
-            float textWidth = maxBubbleWidth - (BubblePadding * 2);
-            
-            Text.Font = GameFont.Small;
-            float textHeight = Text.CalcHeight(text, textWidth);
-            float bubbleHeight = textHeight + (BubblePadding * 2);
-            float bubbleWidth = Mathf.Min(Text.CalcSize(text).x + (BubblePadding * 2), maxBubbleWidth);
-            
-            // 右对齐
-            float bubbleX = containerRect.xMax - bubbleWidth;
-            Rect bubbleRect = new Rect(bubbleX, containerRect.y, bubbleWidth, bubbleHeight);
-            
-            // 绘制圆角气泡背景 - 用户用深蓝色 (StudentBubble)
-            DrawRoundedRect(bubbleRect, WulaLinkStyles.StudentBubbleColor, BubbleCornerRadius);
-            
-            // 绘制文字
-            GUI.color = WulaLinkStyles.StudentTextColor;
-            Text.Anchor = TextAnchor.MiddleLeft;
-            Widgets.Label(bubbleRect.ContractedBy(BubblePadding), text);
-            GUI.color = Color.white;
-            
-            totalHeight = bubbleHeight;
-        }
-
-        private void DrawAIMessage(Rect containerRect, string text, out float totalHeight)
-        {
-            float maxBubbleWidth = containerRect.width * MaxBubbleWidthRatio;
-            float textWidth = maxBubbleWidth - (BubblePadding * 2) - AvatarSize - BubblePadding;
-            
-            Text.Font = GameFont.Small;
-            float textHeight = Text.CalcHeight(text, textWidth);
-            float bubbleHeight = textHeight + (BubblePadding * 2);
-            float bubbleWidth = Mathf.Min(Text.CalcSize(text).x + (BubblePadding * 2), maxBubbleWidth - AvatarSize - BubblePadding);
-            
-            totalHeight = Mathf.Max(bubbleHeight, AvatarSize);
-            
-            // 头像区域 - 左侧
-            Rect avatarRect = new Rect(containerRect.x, containerRect.y + (totalHeight - AvatarSize) / 2f, AvatarSize, AvatarSize);
-            
-            // 绘制圆形头像
-            DrawCircularAvatar(avatarRect);
-            
-            // 气泡区域 - 头像右侧
-            float bubbleX = avatarRect.xMax + BubblePadding;
-            Rect bubbleRect = new Rect(bubbleX, containerRect.y + (totalHeight - bubbleHeight) / 2f, bubbleWidth, bubbleHeight);
-            
-            // 绘制圆角气泡背景 - AI用深红色 (SenseiBubble)
-            DrawRoundedRect(bubbleRect, WulaLinkStyles.SenseiBubbleColor, BubbleCornerRadius);
-            
-            // 绘制文字
-            GUI.color = WulaLinkStyles.SenseiTextColor;
-            Text.Anchor = TextAnchor.MiddleLeft;
-            Widgets.Label(bubbleRect.ContractedBy(BubblePadding), text);
-            GUI.color = Color.white;
-        }
-
-        private void DrawCircularAvatar(Rect rect)
-        {
-            // 获取当前头像
-            Texture2D avatarTex = portrait;
-            if (avatarTex == null && _portraits.Count > 0)
-            {
-                avatarTex = _portraits.Values.FirstOrDefault();
-            }
-            
-            // 绘制圆形背景
-            DrawRoundedRect(rect, new Color(0.3f, 0.15f, 0.15f, 1f), rect.width / 2f);
-            
-            if (avatarTex != null)
-            {
-                // 使用圆形遮罩绘制头像
-                if (WulaLinkStyles.TexCircleMask != null)
-                {
-                    // 先绘制头像
-                    GUI.DrawTexture(rect, avatarTex, ScaleMode.ScaleToFit);
-                    // 再用遮罩（如果可用）
-                    GUI.color = new Color(1f, 1f, 1f, 1f);
-                }
-                else
-                {
-                    // 无遮罩时直接绘制
-                    GUI.DrawTexture(rect, avatarTex, ScaleMode.ScaleToFit);
-                }
-            }
-            else
-            {
-                // 无头像时显示占位符
-                GUI.color = WulaLinkStyles.SenseiTextColor;
-                Text.Font = GameFont.Tiny;
-                Text.Anchor = TextAnchor.MiddleCenter;
-                Widgets.Label(rect, "P.I.A");
-            }
-            GUI.color = Color.white;
-        }
-
-        private void DrawRoundedRect(Rect rect, Color color, float radius)
-        {
-            var originalColor = GUI.color;
-            GUI.color = color;
-            
-            // RimWorld没有内置圆角矩形，使用实心矩形模拟
-            // 主体矩形
-            Widgets.DrawBoxSolid(new Rect(rect.x + radius, rect.y, rect.width - radius * 2, rect.height), color);
-            Widgets.DrawBoxSolid(new Rect(rect.x, rect.y + radius, rect.width, rect.height - radius * 2), color);
-            
-            // 四个角的圆（用小方块近似）
-            float step = radius / 3f;
-            for (float dx = 0; dx < radius; dx += step)
-            {
-                for (float dy = 0; dy < radius; dy += step)
-                {
-                    float dist = Mathf.Sqrt(dx * dx + dy * dy);
-                    if (dist <= radius)
-                    {
-                        // 左上角
-                        Widgets.DrawBoxSolid(new Rect(rect.x + radius - dx - step, rect.y + radius - dy - step, step, step), color);
-                        // 右上角
-                        Widgets.DrawBoxSolid(new Rect(rect.xMax - radius + dx, rect.y + radius - dy - step, step, step), color);
-                        // 左下角
-                        Widgets.DrawBoxSolid(new Rect(rect.x + radius - dx - step, rect.yMax - radius + dy, step, step), color);
-                        // 右下角
-                        Widgets.DrawBoxSolid(new Rect(rect.xMax - radius + dx, rect.yMax - radius + dy, step, step), color);
-                    }
-                }
-            }
-            
-            GUI.color = originalColor;
-        }
-
         private string ParseResponseForDisplay(string rawResponse)
         {
             if (string.IsNullOrEmpty(rawResponse)) return "";
@@ -1630,7 +1570,27 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
 
         private async void SelectOption(string text)
         {
-            if (!string.IsNullOrWhiteSpace(text) && string.Equals(text.Trim(), "/clear", StringComparison.OrdinalIgnoreCase))
+            if (string.IsNullOrWhiteSpace(text))
+            {
+                return;
+            }
+
+            if (_core != null)
+            {
+                if (string.Equals(text.Trim(), "/clear", StringComparison.OrdinalIgnoreCase))
+                {
+                    _isThinking = false;
+                    _options.Clear();
+                    _inputText = "";
+                }
+
+                _scrollToBottom = true;
+                _core.SendUserMessage(text);
+                _history = _core.GetHistorySnapshot();
+                return;
+            }
+
+            if (string.Equals(text.Trim(), "/clear", StringComparison.OrdinalIgnoreCase))
             {
                 _isThinking = false;
                 _options.Clear();
@@ -1647,75 +1607,35 @@ You are 'The Legion', a super AI of the Wula Empire. Your personality is authori
                     WulaLog.Debug($"[WulaAI] Failed to clear AI history: {ex}");
                 }
 
-                Messages.Message("已清除 AI 对话上下文历史。", MessageTypeDefOf.NeutralEvent);
+                Messages.Message("AI conversation history cleared.", MessageTypeDefOf.NeutralEvent);
                 return;
             }
 
-            // 构建带选中对象上下文的用户消息
-            string messageWithContext = BuildUserMessageWithContext(text);
-            _history.Add(("user", messageWithContext));
+            _history.Add(("user", text));
             PersistHistory();
             _scrollToBottom = true;
             await RunPhasedRequestAsync();
         }
 
-        private string BuildUserMessageWithContext(string userText)
-        {
-            StringBuilder sb = new StringBuilder();
-            sb.Append(userText);
-
-            // 获取当前选中的对象信息
-            if (Find.Selector != null)
-            {
-                if (Find.Selector.SingleSelectedThing != null)
-                {
-                    var selected = Find.Selector.SingleSelectedThing;
-                    sb.AppendLine();
-                    sb.AppendLine();
-                    sb.Append($"[Context: Player has selected '{selected.LabelCap}'");
-                    
-                    // 如果是Pawn，提供更多信息
-                    if (selected is Pawn pawn)
-                    {
-                        sb.Append($" ({pawn.def.label})");
-                        sb.Append($" at ({pawn.Position.x}, {pawn.Position.z})");
-                    }
-                    else
-                    {
-                        sb.Append($" at ({selected.Position.x}, {selected.Position.z})");
-                    }
-                    sb.Append("]");
-                }
-                else if (Find.Selector.SelectedObjects.Count > 1)
-                {
-                    sb.AppendLine();
-                    sb.AppendLine();
-                    sb.Append($"[Context: Player has selected {Find.Selector.SelectedObjects.Count} objects");
-                    
-                    // 列出前几个选中的对象
-                    var selectedThings = Find.Selector.SelectedObjects.OfType<Thing>().Take(5).ToList();
-                    if (selectedThings.Count > 0)
-                    {
-                        sb.Append(": ");
-                        sb.Append(string.Join(", ", selectedThings.Select(t => t.LabelCap)));
-                        if (Find.Selector.SelectedObjects.Count > 5)
-                        {
-                            sb.Append("...");
-                        }
-                    }
-                    sb.Append("]");
-                }
-            }
-
-            return sb.ToString();
-        }
-        
         public override void PostClose()
         {
+            if (_core != null)
+            {
+                _core.OnMessageReceived -= OnCoreMessageReceived;
+                _core.OnThinkingStateChanged -= OnCoreThinkingStateChanged;
+                _core.OnExpressionChanged -= OnCoreExpressionChanged;
+            }
+
             if (Instance == this) Instance = null;
-            PersistHistory();
+            if (_core == null)
+            {
+                PersistHistory();
+            }
             base.PostClose();
             HandleAction(def.dismissEffects);
         }
     }
 }
+
+
+