From 805293e80f9005007da42fa1295fd044e3bef13a Mon Sep 17 00:00:00 2001 From: PeterZhong Date: Mon, 16 Jun 2025 00:14:56 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84=E8=AD=A6=E5=91=8A=E5=92=8C?= =?UTF-8?q?=E9=94=99=E8=AF=AF=E4=BF=A1=E6=81=AF=E6=8F=90=E7=A4=BA=E9=80=BB?= =?UTF-8?q?=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- host/Gateway.cs | 30 ++++++++-- host/McpServerList.cs | 94 +++++++++++++++--------------- ui/dockpane/DialogDockpane.xaml.cs | 90 +++++++++++++++++++++++++++- 3 files changed, 160 insertions(+), 54 deletions(-) diff --git a/host/Gateway.cs b/host/Gateway.cs index ad85e57..8ef2dc7 100644 --- a/host/Gateway.cs +++ b/host/Gateway.cs @@ -56,7 +56,7 @@ public class Gateway api_key = "sk-db177155677e438f832860e7f4da6afc" }; List messages = new List(); - string toolInfos = await GetToolInfos(new McpServerList()); + string toolInfos = await GetToolInfos(new McpServerList(),callback); log.Info(SystemPrompt.SysPrompt(gdbPath, toolInfos)); messages.Add(new Message { @@ -246,7 +246,7 @@ public class Gateway string toolInfos = ""; try { - toolInfos = await GetToolInfos(new McpServerList()); + toolInfos = await GetToolInfos(new McpServerList(),callback); messages.Add(new Message { Role = "system", @@ -441,7 +441,16 @@ public class Gateway }catch (Exception e) { log.Error(e.Message); - MessageBox.Show(e.Message, "请求大模型出错"); + MessageListItem endMessageListItem2 = new ChatMessageItem + { + type = MessageType.ERROR, + content = $"请求大模型出错:{e.Message}", + id = timestamp.ToString(), + role = "system" + }; + callback?.Invoke(endMessageListItem2); + // MessageBox.Show(e.Message, "请求大模型出错"); + break; } if (messageContent != "") { @@ -697,10 +706,13 @@ public class Gateway }; callback?.Invoke(endMessageListItem); } + + MessageBox.Show("本轮回复完成", "结束提示"); } - private static async Task GetToolInfos(McpServerList mcpServerList) + private static async Task GetToolInfos(McpServerList mcpServerList,Action callback) { + long timestamp = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds(); StringBuilder toolInfos = new StringBuilder(); int i = 0; List failedMcp = new List(); @@ -805,7 +817,15 @@ public class Gateway if (!failedMcp.IsNullOrEmpty()) { - MessageBox.Show($"读取失败的MCP服务有：{JsonConvert.SerializeObject(failedMcpString)}","MCP读取错误"); + MessageListItem endMessageListItem2 = new ChatMessageItem + { + type = MessageType.WARNING, + content = $"读取失败的MCP服务有：{JsonConvert.SerializeObject(failedMcpString)}", + id = timestamp.ToString(), + role = "system" + }; + callback?.Invoke(endMessageListItem2); + // MessageBox.Show($"读取失败的MCP服务有：{JsonConvert.SerializeObject(failedMcpString)}","MCP读取错误"); } return toolInfos.ToString(); } diff --git a/host/McpServerList.cs b/host/McpServerList.cs index 3393414..8a7ceb7 100644 --- a/host/McpServerList.cs +++ b/host/McpServerList.cs @@ -36,53 +36,53 @@ public class McpServerList // "有地理信息的相关案例步骤参考以及Arcgis Pro的工具详细信息", IsActive = true }); - //servers.Add("filesystem", new StdioMcpServer() - //{ - // Name = "filesystem", - // Type = "stdio", - // Command = "npx", - // Args = new List() - // { - // "-y", - // "@modelcontextprotocol/server-filesystem", - // "F:\\secondsemester\\linktool\\test\\LinkTool0607\\LinkTool0607.gdb" - // } - //}); - //servers.Add("fetch", new StdioMcpServer() - //{ - // Name = "fetch", - // Type = "stdio", - // Command = "uvx", - // Args = new List() - // { - // "mcp-server-fetch" - // } - //}); - //servers.Add("bing-search", new StdioMcpServer() - //{ - // Name = "bing-search", - // Type = "stdio", - // Command = "npx", - // Args = new List() - // { - // "bing-cn-mcp" - // } - //}); - //servers.Add("mcp-python-interpreter", new StdioMcpServer() - //{ - // Name = "mcp-python-interpreter", - // Type = "stdio", - // Command = "uvx", - // Args = new List() - // { - // "--native-tls", - // "mcp-python-interpreter", - // "--dir", - // "F:\\secondsemester\\linktool\\test\\LinkTool0607\\LinkTool0607.gdb", - // "--python-path", - // "C:\\Program Files\\ArcGIS\\Pro\\bin\\Python\\envs\\custom\\python.exe" - // } - //}); + servers.Add("filesystem", new StdioMcpServer() + { + Name = "filesystem", + Type = "stdio", + Command = "npx", + Args = new List() + { + "-y", + "@modelcontextprotocol/server-filesystem", + "F:\\secondsemester\\linktool\\test\\LinkTool0607\\LinkTool0607.gdb" + } + }); + servers.Add("fetch", new StdioMcpServer() + { + Name = "fetch", + Type = "stdio", + Command = "uvx", + Args = new List() + { + "mcp-server-fetch" + } + }); + servers.Add("bing-search", new StdioMcpServer() + { + Name = "bing-search", + Type = "stdio", + Command = "npx", + Args = new List() + { + "bing-cn-mcp" + } + }); + servers.Add("mcp-python-interpreter", new StdioMcpServer() + { + Name = "mcp-python-interpreter", + Type = "stdio", + Command = "uvx", + Args = new List() + { + "--native-tls", + "mcp-python-interpreter", + "--dir", + "F:\\secondsemester\\linktool\\test\\LinkTool0607\\LinkTool0607.gdb", + "--python-path", + "C:\\Program Files\\ArcGIS\\Pro\\bin\\Python\\envs\\custom\\python.exe" + } + }); } public McpServer GetServer(string name) diff --git a/ui/dockpane/DialogDockpane.xaml.cs b/ui/dockpane/DialogDockpane.xaml.cs index 68b55b0..6d535d6 100644 --- a/ui/dockpane/DialogDockpane.xaml.cs +++ b/ui/dockpane/DialogDockpane.xaml.cs @@ -100,6 +100,7 @@ namespace LinkToolAddin.ui.dockpane private void SendButton_OnClick(object sender, RoutedEventArgs e) { + StatusTextBlock.Text = "正在读取用户输入和工具列表"; string question = QuestionTextbox.Text; string defaultGdbPath = Project.Current.DefaultGeodatabasePath; string gdbPath = @""; @@ -223,7 +224,16 @@ namespace LinkToolAddin.ui.dockpane { if (msg.type == MessageType.WARNING) { - + Border border = GetWarningChatBorder(msg); + borderItemsDict[msgId] = border; + ChatHistoryStackPanel.Children.Add(border); + StatusTextBlock.Text = "出现警告信息"; + }else if (msg.type == MessageType.ERROR) + { + Border border = GetWarningChatBorder(msg); + borderItemsDict[msgId] = border; + ChatHistoryStackPanel.Children.Add(border); + StatusTextBlock.Text = "出现错误信息"; } } } @@ -263,7 +273,7 @@ namespace LinkToolAddin.ui.dockpane StatusTextBlock.Text = "正在读取用户输入"; } } - else + else if(msg.role == "assistant") { if (msg.type == MessageType.REASON_MESSAGE) { @@ -280,6 +290,22 @@ namespace LinkToolAddin.ui.dockpane textBox.Text = msg.content; StatusTextBlock.Text = "回答生成中"; } + }else if (msg.role == "system") + { + if (msg.type == MessageType.WARNING) + { + Border borderItem = borderItemsDict[msgId]; + Grid grid = borderItem.Child as Grid; + TextBox textBox = grid.Children[1] as TextBox; + textBox.Text = msg.content; + StatusTextBlock.Text = "出现警告信息"; + }else if (msg.type == MessageType.ERROR) + { + Border borderItem = borderItemsDict[msgId]; + Grid grid = borderItem.Child as Grid; + TextBox textBox = grid.Children[1] as TextBox; + textBox.Text = msg.content; + } } } if (Math.Abs(verticalOffset + viewportHeight - contentHeight) < tolerance) @@ -446,6 +472,66 @@ namespace LinkToolAddin.ui.dockpane return border; } + private Border GetWarningChatBorder(MessageListItem msg) + { + Border border = new Border(); + border.Margin = new Thickness(24); + border.Padding = new Thickness(8); + border.BorderThickness = new Thickness(1); + border.BorderBrush = Brushes.DarkGoldenrod; + border.CornerRadius = new CornerRadius(3); + border.Background = Brushes.Moccasin; + Grid grid = new Grid(); + // Image icon = new Image() + // { + // Source = LocalResource.ReadImageByResource("LinkToolAddin.resource.img.tool.png"), + // Stretch = Stretch.Fill, + // HorizontalAlignment = HorizontalAlignment.Center, VerticalAlignment = VerticalAlignment.Center + // }; + // icon.Margin = new Thickness(0, 0, 8, 0); + // grid.ColumnDefinitions.Add(new ColumnDefinition(){Width = new GridLength(24, GridUnitType.Pixel)}); + grid.ColumnDefinitions.Add(new ColumnDefinition(){Width = new GridLength(1, GridUnitType.Star)}); + TextBlock textBlock = new TextBlock(); + textBlock.Text = (msg as ChatMessageItem).content; + textBlock.TextWrapping = TextWrapping.Wrap; + // grid.Children.Add(icon); + grid.Children.Add(textBlock); + // Grid.SetColumn(icon, 0); + Grid.SetColumn(textBlock, 1); + border.Child = grid; + return border; + } + + private Border GetErrorChatBorder(MessageListItem msg) + { + Border border = new Border(); + border.Margin = new Thickness(24); + border.Padding = new Thickness(8); + border.BorderThickness = new Thickness(1); + border.BorderBrush = Brushes.Crimson; + border.CornerRadius = new CornerRadius(3); + border.Background = Brushes.LightPink; + Grid grid = new Grid(); + // Image icon = new Image() + // { + // Source = LocalResource.ReadImageByResource("LinkToolAddin.resource.img.tool.png"), + // Stretch = Stretch.Fill, + // HorizontalAlignment = HorizontalAlignment.Center, VerticalAlignment = VerticalAlignment.Center + // }; + // icon.Margin = new Thickness(0, 0, 8, 0); + grid.ColumnDefinitions.Add(new ColumnDefinition(){Width = new GridLength(24, GridUnitType.Pixel)}); + grid.ColumnDefinitions.Add(new ColumnDefinition(){Width = new GridLength(1, GridUnitType.Star)}); + TextBlock textBlock = new TextBlock(); + textBlock.Text = (msg as ChatMessageItem).content; + textBlock.TextWrapping = TextWrapping.Wrap; + // grid.Children.Add(icon); + grid.Children.Add(textBlock); + // Grid.SetColumn(icon, 0); + Grid.SetColumn(textBlock, 1); + border.Child = grid; + return border; + } + private Border GetToolChatBorder(MessageListItem msg) { Border border = new Border();