// OpenAI配置builder.Configuration.GetSection("OpenAI").Get<OpenAIOption>();// 文档切片配置builder.Configuration.GetSection("TextChunker").Get<TextChunkerOption>();// 配置数据库连接builder.Configuration.GetSection("GraphDBConnection").Get<GraphDBConnectionOption>();
// 注意，需要先注入配置文件，然后再注入GraphRag.Netbuilder.Services.AddGraphRagNet();

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());

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());}}}

 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; } } }

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(){
}}