client/prompt/DynamicPrompt.cs

@@ -4,10 +4,14 @@ namespace LinkToolAddin.client.prompt;
 
 public class DynamicPrompt
 {
-    public static string GetPrompt(string name,Dictionary<string,object> args)
+    public static string GetPrompt(string name,Dictionary<string,object> args = null)
     {
         PromptTemplates promptTemplate = new PromptTemplates();
         string template = promptTemplate.GetPrompt(name);
+        if (args == null)
+        {
+            return template;
+        }
         foreach (KeyValuePair<string,object> pair in args)
         {
             string replaceKey = "{{"+pair.Key+"}}";
@@ -15,4 +19,11 @@ public class DynamicPrompt
         }
         return template;
     }
+    
+    public static Dictionary<string, string> GetAllPrompts()
+    {
+        PromptTemplates promptTemplate = new PromptTemplates();
+        Dictionary<string, string> template = promptTemplate.GetPromptsDict();
+        return template;
+    }
 }

client/prompt/PromptTemplates.cs

@@ -17,4 +17,9 @@ public class PromptTemplates
     {
         return prompts[name];
     }
+
+    public Dictionary<string, string> GetPromptsDict()
+    {
+        return prompts;
+    }
 }

client/tool/ArcGisPro.cs

@@ -1,6 +1,11 @@
-using System.Collections.Generic;
+using System;
+using System.Collections.Generic;
 using System.ComponentModel;
 using System.Threading.Tasks;
+using ArcGIS.Core.Data;
+using ArcGIS.Core.Data.Raster;
+using ArcGIS.Core.Geometry;
+using ArcGIS.Desktop.Framework.Threading.Tasks;
 using LinkToolAddin.server;
 using ModelContextProtocol.Server;
 using Newtonsoft.Json;
@@ -9,7 +14,7 @@ namespace LinkToolAddin.client.tool;
 
 public class ArcGisPro
 {
-    [McpServerTool, Description("ArcGIS Pro Tool")]
+    [McpServerTool, Description("可以通过调用ArcGIS Pro的地理处理工具实现一些数据处理功能。")]
     public static async Task<JsonRpcResultEntity> ArcGisProTool(string toolName, List<string> toolParams)
     {
         // Call the ArcGIS Pro method and get the result
@@ -18,4 +23,55 @@ public class ArcGisPro
         // Serialize the result back to a JSON string
         return result;
     }
+
+    [McpServerTool, Description("查看指定数据的属性，包括坐标系、范围、数据类型等")]
+    public static async Task<JsonRpcResultEntity> DataProperty(string datasetPath,string dataName)
+    {
+        using Geodatabase gdb = new Geodatabase(new FileGeodatabaseConnectionPath(new Uri(datasetPath)));
+        FeatureClass featureClass = gdb.OpenDataset<FeatureClass>(dataName);
+        FeatureClassDefinition featureClassDefinition = featureClass.GetDefinition();
+        JsonRpcResultEntity result = new JsonRpcSuccessEntity()
+        {
+            Id = 1,
+            Result = JsonConvert.SerializeObject(featureClassDefinition)
+        };
+        return result;
+    }
+
+    [McpServerTool, Description("列出gdb数据库中的所有数据名称")]
+    public static async Task<JsonRpcResultEntity> ListData(string gdbPath)
+    {
+        var datasets = new List<string>();
+        await QueuedTask.Run(() =>
+        {
+            using (Geodatabase gdb = new Geodatabase(new FileGeodatabaseConnectionPath(new Uri(gdbPath))))
+            {
+                // 获取所有要素类（Feature Classes）
+                var featureClasses = gdb.GetDefinitions<FeatureClassDefinition>();
+                foreach (var fc in featureClasses)
+                    datasets.Add($"要素类: {fc.GetName()}");
+
+                // 获取所有表格（Tables）
+                var tables = gdb.GetDefinitions<TableDefinition>();
+                foreach (var table in tables)
+                    datasets.Add($"表格: {table.GetName()}");
+
+                // 获取所有要素数据集（Feature Datasets）
+                var featureDatasets = gdb.GetDefinitions<FeatureDatasetDefinition>();
+                foreach (var fd in featureDatasets)
+                    datasets.Add($"要素数据集: {fd.GetName()}");
+                
+                // 获取所有栅格数据集（Raster Datasets）
+                var rasterDatasets = gdb.GetDefinitions<RasterDatasetDefinition>();
+                foreach (var raster in rasterDatasets)
+                    datasets.Add($"栅格数据: {raster.GetName()}");
+            }
+        });
+        JsonRpcResultEntity result = new JsonRpcSuccessEntity()
+        {
+            Id = 1,
+            Result = JsonConvert.SerializeObject(datasets)
+        };
+        return result;
+    }
 }

common/HttpRequest.cs

@@ -1,8 +1,13 @@
+using System;
+using System.Collections.Generic;
+using System.Globalization;
+using System.IO;
 using System.Net.Http;
 using System.Net.Http.Headers;
 using System.Text;
 using System.Threading.Tasks;
 using LinkToolAddin.host.llm.entity;
+using LinkToolAddin.host.llm.entity.stream;
 using Newtonsoft.Json;
 
 namespace LinkToolAddin.common;
@@ -18,4 +23,104 @@ public class HttpRequest
         var responseBody = await response.Content.ReadAsStringAsync();
         return responseBody;
     }
+    
+    public static async IAsyncEnumerable<string> SendStreamPostRequestAsync(string url, string jsonContent, string apiKey)
+    {
+        using var client = new HttpClient();
+        client.DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Bearer", apiKey);
+
+        // 发送 POST 请求并获取响应流
+        var response = await client.PostAsync(url,new StringContent(jsonContent, Encoding.UTF8, "application/json"));
+
+        // 验证响应状态
+        response.EnsureSuccessStatusCode();
+
+        // 获取响应流
+        using var stream = await response.Content.ReadAsStreamAsync();
+        using var reader = new StreamReader(stream);
+
+        // 流式读取
+        string line;
+        while ((line = await reader.ReadLineAsync()) != null)
+        {
+            string content = ProcessPartialResponse(line);
+            yield return content;
+        }
+    }
+    
+    private static string ProcessPartialResponse(string rawData)
+    {
+        try
+        {
+            var lines = rawData.Split(new[] { "data: " }, StringSplitOptions.RemoveEmptyEntries);
+            foreach (var line in lines)
+            {
+                var trimmedLine = line.Trim();
+                if (!string.IsNullOrEmpty(trimmedLine))
+                {
+                    var result = JsonConvert.DeserializeObject<LlmStreamChat>(trimmedLine);
+                    return result.Choices[0].Delta.Content;
+                }
+            }
+        }
+        catch { /* 处理解析异常 */ }
+        return null;
+    }
+    
+    public static async IAsyncEnumerable<string> PostWithStreamingResponseAsync(
+        string url,
+        string body,
+        string apiKey,
+        string contentType = "application/json",
+        Action<HttpRequestMessage> configureHeaders = null)
+    {
+        using (var client = new HttpClient())
+        {
+            // 设置超时时间为30分钟
+            client.Timeout = TimeSpan.FromMinutes(30);
+            client.DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Bearer", apiKey);
+            using (var request = new HttpRequestMessage(HttpMethod.Post, url))
+            {
+                // 设置请求头和Body
+                Console.WriteLine("开始请求...");
+                configureHeaders?.Invoke(request);
+                request.Content = new StringContent(body, Encoding.UTF8, contentType);
+
+                // 发送请求并立即开始读取响应流
+                using (var response = await client.SendAsync(
+                    request,
+                    HttpCompletionOption.ResponseHeadersRead))
+                {
+                    response.EnsureSuccessStatusCode();
+                    
+                    // 获取响应流
+                    using (var stream = await response.Content.ReadAsStreamAsync())
+                    using (var reader = new StreamReader(stream))
+                    {
+                        string line;
+                        
+                        StringBuilder incompleteJsonBuffer = new StringBuilder();
+    
+                        // 流式读取并输出到控制台
+                        while ((line = await reader.ReadLineAsync()) != null)
+                        {
+                            foreach (var chunk in line.Split(new[] { "data: " }, StringSplitOptions.RemoveEmptyEntries))
+                            {
+                                LlmStreamChat dataObj = null;
+                                try
+                                {
+                                    dataObj = JsonConvert.DeserializeObject<LlmStreamChat>(chunk);
+                                }catch{/*process exception*/}
+
+                                if (dataObj is not null)
+                                {
+                                    yield return dataObj.Choices[0].Delta.Content;
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+        }
+    }
 }

host/Gateway.cs

@@ -11,6 +11,7 @@ using System.Threading.Tasks;
 using System.Xml;
 using System.Xml.Linq;
 using ArcGIS.Desktop.Framework.Dialogs;
+using ArcGIS.Desktop.Internal.Mapping.Locate;
 using LinkToolAddin.client;
 using LinkToolAddin.client.prompt;
 using LinkToolAddin.host.llm;
@@ -68,6 +69,7 @@ public class Gateway
                 TopP = 1,
                 MaxTokens = 1000,
             });
+            long timestamp = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
             log.Info(reponse);
             messages.Add(new Message
             {
@@ -95,7 +97,8 @@ public class Gateway
                         toolParams = toolParams,
                         type = MessageType.TOOL_MESSAGE,
                         status = toolResponse.IsError ? "fail" : "success",
-                        content = toolResponse.Content.ToString()
+                        content = JsonConvert.SerializeObject(toolResponse),
+                        id = timestamp.ToString()
                     };
                     messages.Add(new Message
                     {
@@ -119,7 +122,8 @@ public class Gateway
                         toolParams = toolParams,
                         type = MessageType.TOOL_MESSAGE,
                         status = toolResponse.IsError ? "fail" : "success",
-                        content = toolResponse.Content.ToString()
+                        content = JsonConvert.SerializeObject(toolResponse),
+                        id = timestamp.ToString()
                     };
                     messages.Add(new Message
                     {
@@ -204,7 +208,8 @@ public class Gateway
                 {
                     content = reponse,
                     role = "assistant",
-                    type = MessageType.CHAT_MESSAGE
+                    type = MessageType.CHAT_MESSAGE,
+                    id = timestamp.ToString()
                 };
                 callback?.Invoke(chatMessageListItem);
             }
@@ -215,6 +220,234 @@ public class Gateway
         }
     }
     
+    public static async void SendMessageStream(string message, string model, string gdbPath, Action<MessageListItem> callback)
+    {
+        Llm bailian = new Bailian
+        {
+            api_key = "sk-db177155677e438f832860e7f4da6afc"
+        };
+        List<Message> messages = new List<Message>();
+        string toolInfos = await GetToolInfos(new McpServerList());
+        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 toolPattern = "^<tool_use>[\\s\\S]*?<\\/tool_use>$";
+        string promptPattern = "^<prompt>[\\s\\S]*?<\\/prompt>$";
+        McpServerList mcpServerList = new McpServerList();
+        while (goOn)
+        {
+            LlmJsonContent jsonContent = new LlmJsonContent()
+            {
+                Model = model,
+                Messages = messages,
+                Temperature = 0.7,
+                TopP = 1,
+                MaxTokens = 1000,
+            };
+            long timestamp = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
+            string messageContent = "";
+            await foreach(var chunk in bailian.SendChatStreamAsync(jsonContent))
+            {
+                if (chunk == "[DONE]")
+                {
+                    goOn = false;
+                }else if (chunk.StartsWith("<tool_use>"))
+                {
+                    if (Regex.IsMatch(chunk, toolPattern))
+                    {
+                        //返回工具卡片
+                        XElement toolUse = XElement.Parse(chunk);
+                        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.ErrorPromptTemplate : SystemPrompt.ContinuePromptTemplate
+                            });
+                            messages.Add(new Message
+                            {
+                                Role = "user",
+                                Content = 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.ErrorPromptTemplate : SystemPrompt.ContinuePromptTemplate
+                            });
+                            messages.Add(new Message
+                            {
+                                Role = "user",
+                                Content = 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),
+                                    id = timestamp.ToString()
+                                };
+                                messages.Add(new Message
+                                {
+                                    Role = "user",
+                                    Content = SystemPrompt.ErrorPromptTemplate
+                                });
+                                messages.Add(new Message
+                                {
+                                    Role = "user",
+                                    Content = 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),
+                                    id = timestamp.ToString()
+                                };
+                                messages.Add(new Message
+                                {
+                                    Role = "user",
+                                    Content = SystemPrompt.ContinuePromptTemplate
+                                });
+                                messages.Add(new Message
+                                {
+                                    Role = "user",
+                                    Content = JsonConvert.SerializeObject(innerResult)
+                                });
+                                callback?.Invoke(toolMessageItem);
+                            }
+                        }
+                    }
+                    else
+                    {
+                        MessageListItem toolMessageItem = new ToolMessageItem
+                        {
+                            toolName = "",
+                            toolParams = new Dictionary<string, object>(),
+                            type = MessageType.TOOL_MESSAGE,
+                            status = "loading",
+                            content = "正在生成工具调用参数",
+                            id = timestamp.ToString()
+                        };
+                        callback?.Invoke(toolMessageItem);
+                        continue;
+                    }
+                }else if (chunk.StartsWith("<prompt>"))
+                {
+                    if (Regex.IsMatch(chunk, promptPattern))
+                    {
+                        XElement promptUse = XElement.Parse(chunk);
+                        string promptKey = promptUse.Element("name")?.Value;
+                        string promptContent = DynamicPrompt.GetPrompt(promptKey,null);
+                        messages.Add(new Message
+                        {
+                            Role = "user",
+                            Content = JsonConvert.SerializeObject(promptContent)
+                        });
+                        MessageListItem toolMessageItem = new ToolMessageItem
+                        {
+                            toolName = "调用提示词",
+                            toolParams = null,
+                            type = MessageType.TOOL_MESSAGE,
+                            status = "success",
+                            content = promptKey,
+                            id = timestamp.ToString()
+                        };
+                        callback?.Invoke(toolMessageItem);
+                    }
+                    else
+                    {
+                        MessageListItem toolMessageItem = new ToolMessageItem
+                        {
+                            toolName = "调用提示词",
+                            toolParams = null,
+                            type = MessageType.TOOL_MESSAGE,
+                            status = "loading",
+                            content = "正在调用提示词",
+                            id = timestamp.ToString()
+                        };
+                        callback?.Invoke(toolMessageItem);
+                    }
+                }
+                else
+                {
+                    //普通流式消息卡片
+                    MessageListItem chatMessageListItem = new ChatMessageItem()
+                    {
+                        content = chunk,
+                        role = "assistant",
+                        type = MessageType.CHAT_MESSAGE,
+                        id = timestamp.ToString()
+                    };
+                    messageContent = chunk;
+                    callback?.Invoke(chatMessageListItem);
+                }
+            }
+            messages.Add(new Message
+            {
+                Role = "assistant",
+                Content = messageContent
+            });
+        }
+    }
+    
     private static async Task<string> GetToolInfos(McpServerList mcpServerList)
     {
         int loop = 0;
@@ -298,6 +531,21 @@ public class Gateway
                 }
             }
         }
+
+        Dictionary<string, string> prompts = DynamicPrompt.GetAllPrompts();
+        foreach (KeyValuePair<string, string> prompt in prompts)
+        {
+            McpPromptDefinition promptDefinition = new McpPromptDefinition
+            {
+                Prompt = new LinkToolAddin.host.mcp.Prompt
+                {
+                    Name = prompt.Key
+                }
+            };
+            XNode node = JsonConvert.DeserializeXNode(JsonConvert.SerializeObject(promptDefinition));
+            toolInfos.AppendLine(node.ToString());
+            toolInfos.AppendLine();
+        }
         return toolInfos.ToString();
     }
     

host/McpServerList.cs

@@ -21,7 +21,7 @@ public class McpServerList
                 {"Content-Type","application/json"}
             }
         });
-        servers.Add("arcgis", new InnerMcpServer
+        servers.Add("ArcGisPro", new InnerMcpServer
         {
             Name = "ArcGisPro",
             Type = "inner",

host/llm/Bailian.cs

@@ -1,3 +1,4 @@
+using System;
 using System.Collections.Generic;
 using System.Net.Http;
 using System.Net.Http.Headers;
@@ -17,9 +18,18 @@ public class Bailian : Llm
     public string max_tokens { get; set; }
     public string app_id { get; set; }
     public string api_key { get; set; }
-    public IAsyncEnumerable<string> SendChatStreamAsync(string message)
+    public async IAsyncEnumerable<string> SendChatStreamAsync(LlmJsonContent jsonContent)
     {
-        throw new System.NotImplementedException();
+        jsonContent.Stream = true;
+        StringBuilder builder = new StringBuilder();
+        await foreach (var chunk in HttpRequest.PostWithStreamingResponseAsync(
+                           "https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions",
+                           JsonConvert.SerializeObject(jsonContent),
+                           api_key))
+        {
+            builder.Append(chunk);
+            yield return builder.ToString();
+        }
     }
 
     public IAsyncEnumerable<string> SendApplicationStreamAsync(string message)

host/llm/Llm.cs

@@ -11,7 +11,7 @@ public interface Llm
     public string top_p { get; set; }
     public string max_tokens { get; set; }
 
-    public IAsyncEnumerable<string> SendChatStreamAsync(string message);
+    public IAsyncEnumerable<string> SendChatStreamAsync(LlmJsonContent jsonContent);
     public IAsyncEnumerable<string> SendApplicationStreamAsync(string message);
     public Task<string> SendChatAsync(LlmJsonContent jsonContent);
     public Task<string> SendApplicationAsync(CommonInput commonInput);

host/llm/entity/stream/LlmStreamChat.cs

@@ -0,0 +1,54 @@
+namespace LinkToolAddin.host.llm.entity.stream
+{
+    using System;
+    using System.Collections.Generic;
+
+    using System.Globalization;
+    using Newtonsoft.Json;
+    using Newtonsoft.Json.Converters;
+
+    public partial class LlmStreamChat
+    {
+        [JsonProperty("choices")]
+        public Choice[] Choices { get; set; }
+
+        [JsonProperty("object")]
+        public string Object { get; set; }
+
+        [JsonProperty("usage")]
+        public object Usage { get; set; }
+
+        [JsonProperty("created")]
+        public long Created { get; set; }
+
+        [JsonProperty("system_fingerprint")]
+        public object SystemFingerprint { get; set; }
+
+        [JsonProperty("model")]
+        public string Model { get; set; }
+
+        [JsonProperty("id")]
+        public string Id { get; set; }
+    }
+
+    public partial class Choice
+    {
+        [JsonProperty("finish_reason")]
+        public string FinishReason { get; set; }
+
+        [JsonProperty("delta")]
+        public Delta Delta { get; set; }
+
+        [JsonProperty("index")]
+        public long Index { get; set; }
+
+        [JsonProperty("logprobs")]
+        public object Logprobs { get; set; }
+    }
+
+    public partial class Delta
+    {
+        [JsonProperty("content")]
+        public string Content { get; set; }
+    }
+}

host/mcp/McpPromptDefinition.cs

@@ -0,0 +1,27 @@
+namespace LinkToolAddin.host.mcp
+{
+    using System;
+    using System.Collections.Generic;
+
+    using System.Globalization;
+    using Newtonsoft.Json;
+    using Newtonsoft.Json.Converters;
+
+    public partial class McpPromptDefinition
+    {
+        [JsonProperty("prompt")]
+        public Prompt Prompt { get; set; }
+    }
+
+    public partial class Prompt
+    {
+        [JsonProperty("name")]
+        public string Name { get; set; }
+
+        [JsonProperty("description")]
+        public string Description { get; set; }
+
+        [JsonProperty("arguments")]
+        public string Arguments { get; set; }
+    }
+}

host/prompt/SystemPrompt.cs

@@ -2,9 +2,9 @@
 
 public class SystemPrompt
 {
-    public static string SysPromptTemplate = "现在你是一个精通ArcGIS Pro的专家，请以此身份回答用户的问题。你有以下工具可以调用{{toolInfos}}，用户的数据库路径是{{gdbPath}}。MCP工具调用的格式要求示例：<tool_use>\n  <name>search</name>\n  <arguments>{\\\"query\\\": \\\"上海 人口\\\"}</arguments>\n</tool_use>";
+    public static string SysPromptTemplate = "现在你是一个精通ArcGIS Pro的专家，请以此身份回答用户的问题。你有以下工具可以调用{{toolInfos}}，用户的数据库路径是{{gdbPath}}。MCP工具调用的格式要求示例，必须用<tool_use>标签表示工具调用：<tool_use>\n  <name>search</name>\n  <arguments>{\\\"query\\\": \\\"上海 人口\\\"}</arguments>\n</tool_use>。你每次调用请求都必须放在单独的一条消息中，不附带任何的文字说明，不带markdown格式。如需文字说明，请另外放在一次单独的消息中。\n当你认为已解决用户最初提出的问题时，请输出单独的一条消息，内容为[DONE]，不附带任何其它文字说明，程序识别到后会退出循环。\n此外，你还可以通过<prompt></prompt>调用用户提示词，从而使你更好地理解和完成用户的任务。";
 
-    public static string ContinuePromptTemplate = "上一个工具执行的结果如下，请据此继续执行";
+    public static string ContinuePromptTemplate = "工具执行的结果如下，根据以上结果决定继续执行工具或是根据结果回答问题。如果不再需要额外说明和额外的操作，请回答单独的一条内容为[DONE]的消息。工具已经成功调用，请勿重复执行上一个工具";
     
     public static string ErrorPromptTemplate = "执行上一个工具的时候出现以下错误，请根据报错信息重试";
 

ui/dockpane/TestDockpane.xaml

@@ -21,6 +21,7 @@
       <RowDefinition Height="24"/>
       <RowDefinition Height="24"/>
       <RowDefinition Height="240"/>
+      <RowDefinition Height="24"/>
     </Grid.RowDefinitions>
     <Button Grid.Row="0" Content="Test Workflow" Name="TestServer" Click="TestWorkflow_OnClick"></Button>
     <Grid Grid.Row="1">
@@ -33,5 +34,6 @@
       <Button Grid.Row="0" Grid.Column="1" Content="测试" Name="PromptTestButton" Click="PromptTestButton_OnClick"></Button>
     </Grid>
     <TextBox Grid.Row="2" ToolTip="大模型回复" Name="ReplyTextBox" TextWrapping="Wrap"></TextBox>
+    <Button Grid.Row="3" Content="Test Stream" Name="TestStream" Click="TestStream_OnClick"></Button>
   </Grid>
 </UserControl>

ui/dockpane/TestDockpane.xaml.cs

@@ -1,5 +1,6 @@
 using System.Collections.Generic;
 using System.Linq;
+using System.Text;
 using System.Windows;
 using System.Windows.Controls;
 using LinkToolAddin.client;
@@ -27,6 +28,9 @@ namespace LinkToolAddin.ui.dockpane
     {
         private static ILog log = LogManager.GetLogger(typeof(TestDockpaneView));
         
+        private List<string> idList = new List<string>();
+        private Dictionary<string,MessageListItem> messageDict = new Dictionary<string,MessageListItem>();
+        
         public TestDockpaneView()
         {
             InitLogger();
@@ -140,6 +144,35 @@ namespace LinkToolAddin.ui.dockpane
             log.Info(reponse);
         }
         
+        private async void Request_Bailian_Stream_Test()
+        {
+            LlmJsonContent jsonContent = new LlmJsonContent()
+            {
+                Model = "qwen-max",
+                Messages = new List<Message>()
+                {
+                    new Message()
+                    {
+                        Role = "user",
+                        Content = "给我写一篇1000字的高考议论文"
+                    }
+                },
+                Temperature = 0.7,
+                TopP = 1,
+                MaxTokens = 1000,
+                Stream = true
+            };
+            Llm bailian = new Bailian
+            {
+                api_key = "sk-db177155677e438f832860e7f4da6afc",
+                app_id = "6a77c5a68de64f469b79fcdcde9d5001",
+            };
+            await foreach (var chunk in bailian.SendChatStreamAsync(jsonContent))
+            {
+                log.Info(chunk);
+            }
+        }
+
         private void TestButton_OnClick(object sender, RoutedEventArgs e)
         {
             throw new System.NotImplementedException();
@@ -159,7 +192,31 @@ namespace LinkToolAddin.ui.dockpane
         private void PromptTestButton_OnClick(object sender, RoutedEventArgs e)
         {
             string userPrompt = PromptTestTextBox.Text;
-            Gateway.SendMessage(userPrompt,"qwen-max","C:/Project/test.gdb",AddReply);
+            // Gateway.SendMessage(userPrompt,"qwen-max","C:/Project/test.gdb",AddReply);
+            Gateway.SendMessageStream(userPrompt,"qwen-max","D:\\01_Project\\20250305_LinkTool\\20250420_AiDemoProject\\20250420_AiDemoProject.gdb",AddReplyStream);
+        }
+        
+        public void AddReplyStream(MessageListItem msg)
+        {
+            string id = msg.id;
+            if (idList.Contains(id))
+            {
+                messageDict[id] = msg;
+            }
+            else
+            {
+                idList.Add(id);
+                messageDict.Add(msg.id, msg);
+            }
+            ReplyTextBox.Clear();
+            StringBuilder builder = new StringBuilder();
+            foreach (KeyValuePair<string,MessageListItem> pair in messageDict)
+            {
+                MessageListItem msgItem = pair.Value;
+                builder.AppendLine(msgItem.content);
+                ReplyTextBox.Text = builder.ToString();
+                ReplyTextBox.ScrollToEnd();
+            }
         }
 
         public void AddReply(MessageListItem msg)
@@ -169,5 +226,10 @@ namespace LinkToolAddin.ui.dockpane
             string originContent = ReplyTextBox.Text;
             ReplyTextBox.Text = originContent + content;
         }
+
+        private void TestStream_OnClick(object sender, RoutedEventArgs e)
+        {
+            Request_Bailian_Stream_Test();
+        }
     }
 }