PeterZhong/LinkToolAddin
1
1
Fork 0
Code Issues 3 Pull Requests Releases 1 Wiki Activity

实现连续对话

Browse Source
This commit is contained in:
PeterZhong 2025-07-29 22:18:22 +08:00
parent 93f185dc3b
commit 514d02d4d4
5 changed files with 135 additions and 241 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

326
host/Gateway.cs
View File

@ -43,175 +43,19 @@ public class Gateway
{
private static ILog log = LogManager.GetLogger(typeof(Gateway));
private static bool goOn = true;
private static DialogDockpaneView dockpaneView = null;
public static void SetDockpaneView(DialogDockpaneView view)
{
dockpaneView = view;
}
public static void StopConversation()
{
goOn = false;
}
public static async void SendMessage(string message, string model, string gdbPath, Action<MessageListItem> callback)
{
Llm bailian = new Bailian
if (dockpaneView != null)
{
api_key = "sk-db177155677e438f832860e7f4da6afc"
};
List<Message> messages = new List<Message>();
string toolInfos = await GetToolInfos(new McpServerList(),callback);
log.Info(SystemPrompt.SysPrompt(gdbPath, toolInfos));
messages.Add(new Message
{
Role = "system",
Content = SystemPrompt.SysPrompt(gdbPath, toolInfos)
});
messages.Add(new Message
{
Role = "user",
Content = message
});
bool goOn = true;
string pattern = "^<tool_use>[\\s\\S]*?<\\/tool_use>$";
string promptPattern = "^<prompt>[\\s\\S]*?<\\/prompt>$";
McpServerList mcpServerList = new McpServerList();
while (goOn)
{
string reponse = await bailian.SendChatAsync(new LlmJsonContent()
{
Model = model,
Messages = messages,
Temperature = 0.3,
TopP = 0.6,
TopK = 25,
MaxTokens = 1024,
ThinkingBudget = 1024
});
long timestamp = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
log.Info(reponse);
messages.Add(new Message
{
Role = "assistant",
Content = reponse
});
if (Regex.IsMatch(reponse, pattern))
{
//工具类型的消息
XElement toolUse = XElement.Parse(reponse);
string fullToolName = toolUse.Element("name")?.Value;
string toolArgs = toolUse.Element("arguments")?.Value;
Dictionary<string, object> toolParams = JsonConvert.DeserializeObject<Dictionary<string, object>>(toolArgs);
string serverName = fullToolName.Contains(":") ? fullToolName.Split(':')[0] : fullToolName;
string toolName = fullToolName.Contains(":") ? fullToolName.Split(':')[1] : fullToolName;
McpServer mcpServer = mcpServerList.GetServer(serverName);
if (mcpServer is SseMcpServer)
{
SseMcpServer sseMcpServer = mcpServer as SseMcpServer;
SseMcpClient client = new SseMcpClient(sseMcpServer.BaseUrl);
CallToolResponse toolResponse = await client.CallToolAsync(toolName,toolParams);
MessageListItem toolMessageItem = new ToolMessageItem
{
toolName = toolName,
toolParams = toolParams,
type = MessageType.TOOL_MESSAGE,
status = toolResponse.IsError ? "fail" : "success",
content = JsonConvert.SerializeObject(toolResponse),
id = timestamp.ToString()
};
messages.Add(new Message
{
Role = "user",
Content = toolResponse.IsError ? SystemPrompt.ErrorPrompt(JsonConvert.SerializeObject(toolResponse)) : SystemPrompt.ContinuePrompt(JsonConvert.SerializeObject(toolResponse))
});
callback?.Invoke(toolMessageItem);
}else if (mcpServer is StdioMcpServer)
{
StdioMcpServer stdioMcpServer = mcpServer as StdioMcpServer;
StdioMcpClient client = new StdioMcpClient(stdioMcpServer.Command, stdioMcpServer.Args);
CallToolResponse toolResponse = await client.CallToolAsync(toolName,toolParams);
MessageListItem toolMessageItem = new ToolMessageItem
{
toolName = toolName,
toolParams = toolParams,
type = MessageType.TOOL_MESSAGE,
status = toolResponse.IsError ? "fail" : "success",
content = JsonConvert.SerializeObject(toolResponse),
id = timestamp.ToString()
};
messages.Add(new Message
{
Role = "user",
Content = toolResponse.IsError ? SystemPrompt.ErrorPrompt(JsonConvert.SerializeObject(toolResponse)) : SystemPrompt.ContinuePrompt(JsonConvert.SerializeObject(toolResponse))
});
callback?.Invoke(toolMessageItem);
}else if (mcpServer is InnerMcpServer)
{
Type type = Type.GetType("LinkToolAddin.client.tool."+serverName);
MethodInfo method = type.GetMethod(toolName,BindingFlags.Public | BindingFlags.Static);
var task = method.Invoke(null, toolParams.Values.ToArray()) as Task<JsonRpcResultEntity>;
JsonRpcResultEntity innerResult = await task;
if (innerResult is JsonRpcErrorEntity)
{
MessageListItem toolMessageItem = new ToolMessageItem
{
toolName = toolName,
toolParams = toolParams,
type = MessageType.TOOL_MESSAGE,
status = "fail",
content = JsonConvert.SerializeObject(innerResult)
};
messages.Add(new Message
{
Role = "user",
Content = SystemPrompt.ErrorPrompt(JsonConvert.SerializeObject(innerResult))
});
callback?.Invoke(toolMessageItem);
}else if (innerResult is JsonRpcSuccessEntity)
{
MessageListItem toolMessageItem = new ToolMessageItem
{
toolName = toolName,
toolParams = toolParams,
type = MessageType.TOOL_MESSAGE,
status = "success",
content = JsonConvert.SerializeObject(innerResult)
};
messages.Add(new Message
{
Role = "user",
Content = SystemPrompt.ContinuePrompt(JsonConvert.SerializeObject(innerResult))
});
callback?.Invoke(toolMessageItem);
}
}
}
else if (Regex.IsMatch(reponse, promptPattern))
{
XElement prompt = XElement.Parse(reponse);
string fullPromptName = prompt.Element("name")?.Value;
string promptArgs = prompt.Element("arguments")?.Value;
Dictionary<string, object> promptParams = JsonConvert.DeserializeObject<Dictionary<string, object>>(promptArgs);
string serverName = fullPromptName.Contains(":") ? fullPromptName.Split(':')[0] : fullPromptName;
string promptName = fullPromptName.Contains(":") ? fullPromptName.Split(':')[1] : fullPromptName;
string promptRes = DynamicPrompt.GetPrompt(promptName, null);
messages.Add(new Message
{
Role = "user",
Content = promptRes
});
}
else
{
MessageListItem chatMessageListItem = new ChatMessageItem()
{
content = reponse,
role = "assistant",
type = MessageType.CHAT_MESSAGE,
id = timestamp.ToString()
};
callback?.Invoke(chatMessageListItem);
}
if (reponse.EndsWith("[DONE]"))
{
goOn = false;
}
dockpaneView.hideProgressBar();
}
}
@ -236,11 +80,41 @@ public class Gateway
return (matched, remaining);
}
public static async void SendMessageStream(string message, string model, string gdbPath, Action<MessageListItem> callback)
public static async void SendMessageStream(string message, string model, string gdbPath, List<Message> historyMessages, Action<MessageListItem,List< Message>> callback)
{
Llm modelObj = new Bailian();
List<string> bailianModels = ["Moonshot-Kimi-K2-Instruct","deepseek-r1","deepseek-v3","qwen3-235b-a22b","qwen3-32b","qwen-plus","qwen-turbo","abab6.5s-chat"];
List<string> dmxModels = ["grok-3-mini","o4-mini","claude-3-5-haiku-20241022","claude-sonnet-4-20250514-thinking","claude-3-7-sonnet-20250219-thinking","grok-4-ssvip","gemini-2.5-pro-ssvip","gemini-2.5-flash-ssvip","o3-pro","gpt-4.1","gpt-4o"];
List<Message> messages = null;
string toolInfos = "";
if (historyMessages.IsNullOrEmpty())
{
messages = new List<Message>();
try
{
toolInfos = await GetToolInfos(new McpServerList(),messages,callback);
messages.Add(new Message
{
Role = "system",
Content = SystemPrompt.SysPrompt(gdbPath, toolInfos)
});
}catch (Exception ex)
{
log.Error(ex);
MessageListItem endMessageListItem2 = new ChatMessageItem
{
type = MessageType.WARNING,
content = $"MCP列表读取失败{ex.Message}",
id = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().ToString(),
role = "system",
};
callback?.Invoke(endMessageListItem2,messages);
}
}
else
{
messages = historyMessages;
}
if (bailianModels.Contains(model))
{
modelObj = new Bailian
@ -263,7 +137,7 @@ public class Gateway
id = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().ToString(),
role = "system"
};
callback?.Invoke(endMessageListItem2);
callback?.Invoke(endMessageListItem2,messages);
MessageListItem endMessageListItem1 = new ChatMessageItem
{
type = MessageType.END_TAG,
@ -271,46 +145,14 @@ public class Gateway
id = (DateTimeOffset.UtcNow.ToUnixTimeMilliseconds() + 3).ToString(),
role = "assistant"
};
callback?.Invoke(endMessageListItem1);
callback?.Invoke(endMessageListItem1,messages);
return;
}
List<Message> messages = new List<Message>();
string toolInfos = "";
try
messages.Add(new Message
{
toolInfos = await GetToolInfos(new McpServerList(),callback);
messages.Add(new Message
{
Role = "system",
Content = SystemPrompt.SysPrompt(gdbPath, toolInfos)
});
messages.Add(new Message
{
Role = "user",
Content = message
});
}catch (Exception ex)
{
log.Error(ex);
MessageListItem endMessageListItem2 = new ChatMessageItem
{
type = MessageType.WARNING,
content = $"MCP列表读取失败{ex.Message}",
id = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds().ToString(),
role = "system"
};
callback?.Invoke(endMessageListItem2);
MessageListItem endMessageListItem1 = new ChatMessageItem
{
type = MessageType.END_TAG,
content = "",
id = (DateTimeOffset.UtcNow.ToUnixTimeMilliseconds() + 3).ToString(),
role = "assistant"
};
callback?.Invoke(endMessageListItem1);
// MessageBox.Show(ex.Message,"获取MCP列表失败");
}
Role = "user",
Content = message
});
goOn = true;
long accTokens = 0;
string toolPattern = "<tool_use>([\\s\\S]*?)<name>([\\s\\S]*?)<\\/name>([\\s\\S]*?)<arguments>([\\s\\S]*?)<\\/arguments>([\\s\\S]*?)<\\/tool_use>";
@ -334,7 +176,7 @@ public class Gateway
id = timestamp.ToString(),
role = "system"
};
callback?.Invoke(endMessageListItem2);
callback?.Invoke(endMessageListItem2,messages);
MessageListItem endMessageListItem1 = new ChatMessageItem
{
type = MessageType.END_TAG,
@ -342,8 +184,16 @@ public class Gateway
id = (timestamp + 3).ToString(),
role = "assistant"
};
callback?.Invoke(endMessageListItem1);
callback?.Invoke(endMessageListItem1,messages);
// MessageBox.Show("达到最大循环次数", "退出循环");
MessageListItem endAllMsgItem = new ChatMessageItem
{
type = MessageType.END_ALL,
content = "",
id = (timestamp + 4).ToString(),
role = "assistant"
};
callback?.Invoke(endAllMsgItem,messages);
break;
}
LlmJsonContent jsonContent = new LlmJsonContent()
@ -369,7 +219,15 @@ public class Gateway
id = (timestamp + 3).ToString(),
role = "assistant"
};
callback?.Invoke(endMessageListItem1);
callback?.Invoke(endMessageListItem1,messages);
MessageListItem endAllMsgItem = new ChatMessageItem
{
type = MessageType.END_ALL,
content = "",
id = (timestamp + 4).ToString(),
role = "assistant"
};
callback?.Invoke(endAllMsgItem,messages);
break;
}
@ -386,7 +244,15 @@ public class Gateway
id = (timestamp+3).ToString(),
role = "assistant"
};
callback?.Invoke(endMessageListItem2);
callback?.Invoke(endMessageListItem2,messages);
MessageListItem endAllMsgItem = new ChatMessageItem
{
type = MessageType.END_ALL,
content = "",
id = (timestamp + 4).ToString(),
role = "assistant"
};
callback?.Invoke(endAllMsgItem,messages);
break;
}
try
@ -401,7 +267,7 @@ public class Gateway
};
Application.Current.Dispatcher.Invoke(() =>
{
callback?.Invoke(reasonMessageListItem);
callback?.Invoke(reasonMessageListItem,messages);
});
messageContent = chunk;
var (matched, remaining) = ExtractMatchedPart(chunk, toolPattern);
@ -420,7 +286,7 @@ public class Gateway
};
Application.Current.Dispatcher.Invoke(() =>
{
callback?.Invoke(chatMessageListItem);
callback?.Invoke(chatMessageListItem,messages);
});
}
else
@ -433,7 +299,7 @@ public class Gateway
type = MessageType.CHAT_MESSAGE,
id = timestamp.ToString()
};
callback?.Invoke(chatMessageListItem);
callback?.Invoke(chatMessageListItem,messages);
XElement promptUse = XElement.Parse(matchedPrompt);
string promptKey = promptUse.Element("name")?.Value;
string promptArgs = promptUse.Element("arguments")?.Value;
@ -467,7 +333,7 @@ public class Gateway
};
Application.Current.Dispatcher.Invoke(() =>
{
callback?.Invoke(chatMessageListItem);
callback?.Invoke(chatMessageListItem,messages);
});
if (!executedTool)
{
@ -484,7 +350,7 @@ public class Gateway
};
Application.Current.Dispatcher.Invoke(() =>
{
callback?.Invoke(toolMessageListItem);
callback?.Invoke(toolMessageListItem,messages);
});
}
mcpToolRequests = new List<McpToolRequest>();
@ -513,7 +379,7 @@ public class Gateway
id = timestamp.ToString(),
role = "system"
};
callback?.Invoke(endMessageListItem2);
callback?.Invoke(endMessageListItem2,messages);
MessageListItem endMessageListItem1 = new ChatMessageItem
{
type = MessageType.END_TAG,
@ -521,8 +387,16 @@ public class Gateway
id = (timestamp + 3).ToString(),
role = "assistant"
};
callback?.Invoke(endMessageListItem1);
callback?.Invoke(endMessageListItem1,messages);
// MessageBox.Show(e.Message, "请求大模型出错");
MessageListItem endAllMsgItem = new ChatMessageItem
{
type = MessageType.END_ALL,
content = "",
id = (timestamp + 4).ToString(),
role = "assistant"
};
callback?.Invoke(endAllMsgItem,messages);
break;
}
if (messageContent != "")
@ -567,7 +441,7 @@ public class Gateway
});
Application.Current.Dispatcher.Invoke(() =>
{
callback?.Invoke(toolMessageItem1);
callback?.Invoke(toolMessageItem1,messages);
});
}
else if (mcpServer is StdioMcpServer)
@ -595,7 +469,7 @@ public class Gateway
});
Application.Current.Dispatcher.Invoke(() =>
{
callback?.Invoke(toolMessageItem1);
callback?.Invoke(toolMessageItem1,messages);
});
}
else if (mcpServer is InnerMcpServer)
@ -626,7 +500,7 @@ public class Gateway
});
Application.Current.Dispatcher.Invoke(() =>
{
callback?.Invoke(toolMessageItem);
callback?.Invoke(toolMessageItem,messages);
});
queriedKnowledge = true;
}
@ -651,7 +525,7 @@ public class Gateway
});
Application.Current.Dispatcher.Invoke(() =>
{
callback?.Invoke(toolMessageItem);
callback?.Invoke(toolMessageItem,messages);
});
}
continue;
@ -702,7 +576,7 @@ public class Gateway
});
Application.Current.Dispatcher.Invoke(() =>
{
callback?.Invoke(toolMessageItem1);
callback?.Invoke(toolMessageItem1,messages);
});
}
else if (innerResult is JsonRpcSuccessEntity)
@ -725,7 +599,7 @@ public class Gateway
});
Application.Current.Dispatcher.Invoke(() =>
{
callback?.Invoke(toolMessageItem1);
callback?.Invoke(toolMessageItem1,messages);
});
}
}
@ -761,7 +635,7 @@ public class Gateway
};
Application.Current.Dispatcher.Invoke(() =>
{
callback?.Invoke(toolMessageItem);
callback?.Invoke(toolMessageItem,messages);
});
}
catch (Exception e)
@ -777,13 +651,11 @@ public class Gateway
id = (timestamp+3).ToString(),
role = "assistant"
};
callback?.Invoke(endMessageListItem);
callback?.Invoke(endMessageListItem,messages);
}
MessageBox.Show("本轮回复完成", "结束提示");
}
private static async Task<string> GetToolInfos(McpServerList mcpServerList,Action<MessageListItem> callback)
private static async Task<string> GetToolInfos(McpServerList mcpServerList,List<Message> historyMessages,Action<MessageListItem,List<Message>> callback)
{
long timestamp = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
StringBuilder toolInfos = new StringBuilder();
@ -894,10 +766,10 @@ public class Gateway
{
type = MessageType.WARNING,
content = $"读取失败的MCP服务有：{JsonConvert.SerializeObject(failedMcpString)}",
id = timestamp.ToString(),
id = (timestamp-2).ToString(),
role = "system"
};
callback?.Invoke(endMessageListItem2);
callback?.Invoke(endMessageListItem2,historyMessages);
// MessageBox.Show($"读取失败的MCP服务有：{JsonConvert.SerializeObject(failedMcpString)}","MCP读取错误");
}
return toolInfos.ToString();

8
ui/dockpane/DialogDockpane.xaml
View File

@ -23,7 +23,7 @@
<RowDefinition Height="*"/>
<RowDefinition Height="24"/>
<RowDefinition Height="Auto"/>
<RowDefinition Height="24"/>
<RowDefinition Height="16"/>
</Grid.RowDefinitions>
<Grid Grid.Row="0" HorizontalAlignment="Center">
<Grid.ColumnDefinitions>
@ -66,5 +66,11 @@
<TextBox Grid.Column="0" Name="QuestionTextbox" Text="" TextWrapping="Wrap" AcceptsReturn="True" VerticalScrollBarVisibility="Auto" PreviewKeyDown="QuestionTextbox_OnPreviewKeyDown"></TextBox>
<Button Grid.Column="1" Name="SendButton" Content="发送" Click="SendButton_OnClick"></Button>
</Grid>
<ProgressBar x:Name="IndeterminateProgressBar"
Grid.Row="4"
Margin="3"
Height="16"
IsIndeterminate="True"
Visibility="Hidden"/>
</Grid>
</UserControl>

28
ui/dockpane/DialogDockpane.xaml.cs
View File

@ -54,12 +54,14 @@ namespace LinkToolAddin.ui.dockpane
private List<string> idList = new List<string>();
private ConcurrentDictionary<string,MessageListItem> messageDict = new ConcurrentDictionary<string,MessageListItem>();
private ConcurrentDictionary<string,Border> borderItemsDict = new ConcurrentDictionary<string,Border>();
private List<Message> historyMessages = new List<Message>();
public DialogDockpaneView()
{
InitLogger();
InitializeComponent();
DataContext = this;
Gateway.SetDockpaneView(this);
}
private async void TestServer_OnClick(object sender, RoutedEventArgs e)
@ -105,6 +107,7 @@ namespace LinkToolAddin.ui.dockpane
private void SendButton_OnClick(object sender, RoutedEventArgs e)
{
StatusTextBlock.Text = "正在读取用户输入和工具列表";
IndeterminateProgressBar.Visibility = Visibility.Visible;
string question = QuestionTextbox.Text;
string defaultGdbPath = Project.Current.DefaultGeodatabasePath;
string gdbPath = @"";
@ -128,7 +131,12 @@ namespace LinkToolAddin.ui.dockpane
model = ShowInputBox("自定义模型", "请输入模型名称:", "");
}
ScrollViewer.ScrollToBottom();
Gateway.SendMessageStream(question,model,defaultGdbPath,NewMessage_Recall);
Gateway.SendMessageStream(question,model,defaultGdbPath,historyMessages,NewMessage_Recall);
}
public void hideProgressBar()
{
IndeterminateProgressBar.Visibility = Visibility.Hidden;
}
private string ShowInputBox(string title, string message, string defaultValue = "")
@ -171,7 +179,13 @@ namespace LinkToolAddin.ui.dockpane
return textBox.Text;
}
public void NewMessage_Recall(MessageListItem msg)
public void NewMessage_Recall(MessageListItem msg, List<Message> historyMessages)
{
this.historyMessages = historyMessages;
NewMessage_Recall(msg);
}
private void NewMessage_Recall(MessageListItem msg)
{
string msgId = msg.id;
log.Info(msg.content);
@ -187,6 +201,10 @@ namespace LinkToolAddin.ui.dockpane
if (msg.type == MessageType.END_TAG)
{
StatusTextBlock.Text = "";
}else if (msg.type == MessageType.END_ALL)
{
StatusTextBlock.Text = "";
IndeterminateProgressBar.Visibility = Visibility.Hidden;
}
return;
}
@ -250,6 +268,10 @@ namespace LinkToolAddin.ui.dockpane
if (msg.type == MessageType.END_TAG)
{
StatusTextBlock.Text = "";
}else if (msg.type == MessageType.END_ALL)
{
StatusTextBlock.Text = "";
IndeterminateProgressBar.Visibility = Visibility.Hidden;
}
ChatHistoryStackPanel.Children.Remove(borderItemsDict[msgId]);
borderItemsDict.TryRemove(msgId, out Border border);
@ -309,6 +331,7 @@ namespace LinkToolAddin.ui.dockpane
Grid grid = borderItem.Child as Grid;
TextBox textBox = grid.Children[1] as TextBox;
textBox.Text = msg.content;
StatusTextBlock.Text = "出现错误信息";
}
}
}
@ -647,6 +670,7 @@ namespace LinkToolAddin.ui.dockpane
borderItemsDict.Clear();
ChatHistoryStackPanel.Children.Clear();
QuestionTextbox.Clear();
historyMessages.Clear();
StatusTextBlock.Text = "";
}

11
ui/dockpane/TestDockpane.xaml.cs
View File

@ -201,16 +201,7 @@ namespace LinkToolAddin.ui.dockpane
private async void PromptTestButton_OnClick(object sender, RoutedEventArgs e)
{
string userPrompt = PromptTestTextBox.Text;
// model可选值：qwen3-235b-a22b,qwen-max,deepseek-r1
try
{
await Task.Run(() => Gateway.SendMessageStream(userPrompt,"qwen3-235b-a22b", "F:\\secondsemester\\linktool\\test\\LinkTool0604\\LinkTool0604.gdb", AddReplyStream));
}
catch (Exception exception)
{
log.Error(exception.Message);
}
throw new NotImplementedException();
}
public async void AddReplyStream(MessageListItem msg)

3
ui/message/MessageListItem.cs
View File

@ -7,7 +7,8 @@ public enum MessageType
REASON_MESSAGE,
END_TAG,
WARNING,
ERROR
ERROR,
END_ALL
}
public interface MessageListItem