新增托盘图标

ShrlAlgoToolkit.RevitAddins/General/LLMScript/OllamaProvider.cs

@@ -0,0 +1,74 @@
+using System.IO;
+using System.Net.Http;
+using System.Text;
+
+using Newtonsoft.Json;
+using Newtonsoft.Json.Linq;
+
+namespace Szmedi.RvKits.LLMScript
+{
+    public class OllamaProvider : ILLMProvider
+    {
+        private static readonly HttpClient _client = new HttpClient();
+        private readonly string _endpoint;
+        private readonly string _model;
+
+        public OllamaProvider(string endpoint, string model)
+        {
+            _endpoint = endpoint;
+            _model = model;
+        }
+
+        public async Task GenerateStreamAsync(
+            string systemPrompt,
+            string userPrompt,
+            Action<string> onTokenReceived)
+        {
+            var payload = new
+            {
+                model = _model,
+                messages = new[]
+                {
+                new { role = "system", content = systemPrompt },
+                new { role = "user", content = userPrompt }
+            },
+                stream = true
+            };
+
+            string json = JsonConvert.SerializeObject(payload);
+
+            var request = new HttpRequestMessage(HttpMethod.Post, _endpoint);
+            request.Content = new StringContent(json, Encoding.UTF8, "application/json");
+
+            var response = await _client.SendAsync(
+                request,
+                HttpCompletionOption.ResponseHeadersRead);
+
+            response.EnsureSuccessStatusCode();
+
+            using (var stream = await response.Content.ReadAsStreamAsync())
+            using (var reader = new StreamReader(stream))
+            {
+                while (!reader.EndOfStream)
+                {
+                    string line = await reader.ReadLineAsync();
+                    if (string.IsNullOrWhiteSpace(line))
+                        continue;
+
+                    JObject obj = JObject.Parse(line);
+
+                    bool done = obj["done"]?.Value<bool>() ?? false;
+
+                    var content = obj["message"]?["content"];
+                    if (content != null)
+                    {
+                        onTokenReceived(content.ToString());
+                    }
+
+                    if (done)
+                        break;
+                }
+            }
+        }
+    }
+}