Merge pull request #13 from AIDotNet/feature_kernel

add 0.1.6 抽象接口实现

Author: xuzeyu91

Directory.Build.props

@@ -1,7 +1,7 @@
 <Project>
   <!-- See https://aka.ms/dotnet/msbuild/customize for more details on customizing your build -->
   <PropertyGroup>
-    <Version>0.1.5</Version>
+    <Version>0.1.6</Version>
     <SKVersion>1.16.2</SKVersion>  
   </PropertyGroup>
 </Project>

README.md

@@ -75,6 +75,18 @@ builder.Configuration.GetSection("GraphDBConnection").Get<GraphDBConnectionOptio
 builder.Services.AddGraphRagNet();
 ```
 
+## 如果你想接入其他模型，可以参考以下代码,这里抽象了Kernel的实现，你可以自定义实现
+```
+var kernelBuild = Kernel.CreateBuilder();
+kernelBuild.Services.AddKeyedSingleton<ITextGenerationService>("mock-text", new MockTextCompletion());
+kernelBuild.Services.AddKeyedSingleton<IChatCompletionService>("mock-chat", new MockChatCompletion());
+kernelBuild.Services.AddSingleton((ITextEmbeddingGenerationService)new MockTextEmbeddingGeneratorService());
+kernelBuild.Services.AddKeyedSingleton("mock-embedding", new MockTextEmbeddingGeneratorService());
+
+builder.Services.AddGraphRagNet(kernelBuild.Build());
+```
+
+
 使用时注入 IGraphService  服务,以下为参考示例代码
 ```
 namespace GraphRag.Net.Api.Controllers

src/GraphRag.Net.Web/GraphRag.Net.Web.csproj

@@ -10,6 +10,7 @@
     <Description>商务需求联系微信xuzeyu91</Description>
     <GenerateDocumentationFile>True</GenerateDocumentationFile>
     <DocumentationFile>GraphRag.Net.Web.xml</DocumentationFile>
+    <NoWarn>CA1050,CA1707,CA2007,VSTHRD111,CS1591,RCS1110,CA5394,SKEXP0001,SKEXP0002,SKEXP0003,SKEXP0004,SKEXP0010,SKEXP0011,,SKEXP0012,SKEXP0020,SKEXP0021,SKEXP0022,SKEXP0023,SKEXP0024,SKEXP0025,SKEXP0026,SKEXP0027,SKEXP0028,SKEXP0029,SKEXP0030,SKEXP0031,SKEXP0032,SKEXP0040,SKEXP0041,SKEXP0042,SKEXP0050,SKEXP0051,SKEXP0052,SKEXP0053,SKEXP0054,SKEXP0055,SKEXP0060,SKEXP0061,SKEXP0101,SKEXP0102,KMEXP01</NoWarn>
   </PropertyGroup>
 
   <ItemGroup>

src/GraphRag.Net.Web/Mock/MockChatCompletion.cs

@@ -0,0 +1,48 @@
+using Microsoft.SemanticKernel;
+using Microsoft.SemanticKernel.ChatCompletion;
+using System.Runtime.CompilerServices;
+using System.Text;
+using System.Text.Encodings.Web;
+using System.Text.Json;
+using System.Text.Json.Serialization;
+using System.Text.Unicode;
+
+namespace GraphRag.Net.Web.Mock
+{
+    public class MockChatCompletion : IChatCompletionService
+    {
+        private readonly Dictionary<string, object?> _attributes = new();
+        private string _chatId;
+
+
+        private static readonly JsonSerializerOptions _jsonSerializerOptions = new()
+        {
+            NumberHandling = JsonNumberHandling.AllowReadingFromString,
+            Encoder = JavaScriptEncoder.Create(UnicodeRanges.All)
+        };
+
+        public IReadOnlyDictionary<string, object?> Attributes => _attributes;
+
+        public MockChatCompletion()
+        {
+
+        }
+
+        public async Task<IReadOnlyList<ChatMessageContent>> GetChatMessageContentsAsync(ChatHistory chatHistory, PromptExecutionSettings? executionSettings = null, Kernel? kernel = null, [EnumeratorCancellation] CancellationToken cancellationToken = default)
+        {
+            StringBuilder sb = new();
+            string result = $"这是一条Mock数据，便于聊天测试，你的消息是：{chatHistory.LastOrDefault().ToString()}";
+            return [new(AuthorRole.Assistant, result.ToString())];
+        }
+
+        public async IAsyncEnumerable<StreamingChatMessageContent> GetStreamingChatMessageContentsAsync(ChatHistory chatHistory, PromptExecutionSettings? executionSettings = null, Kernel? kernel = null, [EnumeratorCancellation] CancellationToken cancellationToken = default)
+        {
+            StringBuilder sb = new();
+            string result = $"这是一条Mock数据，便于聊天测试，你的消息是：{chatHistory.LastOrDefault().ToString()}";
+            foreach (var c in result)
+            {
+                yield return new StreamingChatMessageContent(AuthorRole.Assistant, c.ToString());
+            }
+        }
+    }
+}

src/GraphRag.Net.Web/Mock/MockTextCompletion.cs

@@ -0,0 +1,49 @@
+using Microsoft.SemanticKernel;
+using Microsoft.SemanticKernel.Services;
+using Microsoft.SemanticKernel.TextGeneration;
+using System.Text;
+using System.Text.Encodings.Web;
+using System.Text.Json;
+using System.Text.Json.Serialization;
+using System.Text.Unicode;
+
+namespace GraphRag.Net.Web.Mock
+{
+    public class MockTextCompletion : ITextGenerationService, IAIService
+    {
+        private readonly Dictionary<string, object?> _attributes = new();
+        private string _chatId;
+
+        private static readonly JsonSerializerOptions _jsonSerializerOptions = new()
+        {
+            NumberHandling = JsonNumberHandling.AllowReadingFromString,
+            Encoder = JavaScriptEncoder.Create(UnicodeRanges.All)
+        };
+
+        public IReadOnlyDictionary<string, object?> Attributes => _attributes;
+
+        public MockTextCompletion()
+        {
+
+        }
+
+        public async Task<IReadOnlyList<TextContent>> GetTextContentsAsync(string prompt, PromptExecutionSettings? executionSettings = null, Kernel? kernel = null, CancellationToken cancellationToken = default)
+        {
+            StringBuilder sb = new();
+            string result = $"这是一条Mock数据，便于聊天测试，你的消息是：{prompt}";
+            return [new(result.ToString())];
+        }
+
+        public async IAsyncEnumerable<StreamingTextContent> GetStreamingTextContentsAsync(string prompt, PromptExecutionSettings? executionSettings = null, Kernel? kernel = null, CancellationToken cancellationToken = default)
+        {
+            StringBuilder sb = new();
+            string result = $"这是一条Mock数据，便于聊天测试，你的消息是：{prompt}";
+            foreach (var c in result)
+            {
+                var streamingTextContent = new StreamingTextContent(c.ToString(), modelId: "mock");
+
+                yield return streamingTextContent;
+            }
+        }
+    }
+}

src/GraphRag.Net.Web/Mock/MockTextEmbeddingGeneratorService.cs

@@ -0,0 +1,39 @@
+using Microsoft.SemanticKernel;
+using Microsoft.SemanticKernel.Embeddings;
+using Microsoft.SemanticKernel.Services;
+
+namespace GraphRag.Net.Web.Mock
+{
+    public sealed class MockTextEmbeddingGeneratorService : ITextEmbeddingGenerationService
+    {
+        private Dictionary<string, object?> AttributesInternal { get; } = [];
+        public IReadOnlyDictionary<string, object?> Attributes => this.AttributesInternal;
+        public MockTextEmbeddingGeneratorService()
+        {
+
+        }
+        public async Task<IList<ReadOnlyMemory<float>>> GenerateEmbeddingsAsync(
+          IList<string> data,
+          Kernel? kernel = null,
+          CancellationToken cancellationToken = default)
+        {
+            IList<ReadOnlyMemory<float>> results = new List<ReadOnlyMemory<float>>();
+
+            float[] array1 = { 1.0f, 2.0f, 3.0f };
+            float[] array2 = { 4.0f, 5.0f, 6.0f };
+            float[] array3 = { 7.0f, 8.0f, 9.0f };
+
+            // 将数组包装为ReadOnlyMemory<float>并添加到列表中
+            results.Add(new ReadOnlyMemory<float>(array1));
+            results.Add(new ReadOnlyMemory<float>(array2));
+            results.Add(new ReadOnlyMemory<float>(array3));
+
+            return results;
+        }
+
+        public void Dispose()
+        {
+
+        }
+    }
+}

src/GraphRag.Net.Web/Program.cs

@@ -2,7 +2,13 @@
 using GraphRag.Net;
 using GraphRag.Net.Common.Options;
 using GraphRag.Net.Options;
+using GraphRag.Net.Web.Mock;
 using Microsoft.AspNetCore.Components;
+using Microsoft.Extensions.Options;
+using Microsoft.SemanticKernel;
+using Microsoft.SemanticKernel.ChatCompletion;
+using Microsoft.SemanticKernel.Embeddings;
+using Microsoft.SemanticKernel.TextGeneration;
 using System.Reflection;
 
 var builder = WebApplication.CreateBuilder(args);
@@ -35,6 +41,16 @@
 
 builder.Services.AddGraphRagNet();
 
+////自定义Kernel 可以实现其他模型的对接实现
+//var kernelBuild = Kernel.CreateBuilder();
+//kernelBuild.Services.AddKeyedSingleton<ITextGenerationService>("mock-text", new MockTextCompletion());
+//kernelBuild.Services.AddKeyedSingleton<IChatCompletionService>("mock-chat", new MockChatCompletion());
+//kernelBuild.Services.AddSingleton((ITextEmbeddingGenerationService)new MockTextEmbeddingGeneratorService());
+//kernelBuild.Services.AddKeyedSingleton("mock-embedding", new MockTextEmbeddingGeneratorService());
+
+//builder.Services.AddGraphRagNet(kernelBuild.Build());
+
+
 var app = builder.Build();
 
 // Configure the HTTP request pipeline.

src/GraphRag.Net/Extensions/ServiceCollectionExtensions.cs

@@ -18,7 +18,7 @@ public static class ServiceCollectionExtensions
         /// </summary>
         /// <param name="services">容器</param>
         /// <returns></returns>
-        public static IServiceCollection AddGraphRagNet(this IServiceCollection services)
+        public static IServiceCollection AddGraphRagNet(this IServiceCollection services, Kernel _kernel = null)
         {
             Type attributeType = typeof(ServiceDescriptionAttribute);
             //var refAssembyNames = Assembly.GetExecutingAssembly().GetReferencedAssemblies();
@@ -50,7 +50,7 @@ public static IServiceCollection AddGraphRagNet(this IServiceCollection services
             }
 
             CodeFirst();
-            InitSK(services);
+            InitSK(services, _kernel);
 
             return services;
         }