Skip to content

Commit c7160bf

Browse files
authored
Merge pull request #12 from AIDotNet/feature_kernel
add custom kernel
2 parents 8b65101 + c7a5a52 commit c7160bf

File tree

2 files changed

+19
-8
lines changed

2 files changed

+19
-8
lines changed

src/GraphRag.Net.Web/GraphRag.Net.Web.xml

Lines changed: 7 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

src/GraphRag.Net/Extensions/ServiceCollectionExtensions.cs

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -59,18 +59,22 @@ public static IServiceCollection AddGraphRagNet(this IServiceCollection services
5959
/// 初始化SK
6060
/// </summary>
6161
/// <param name="services"></param>
62-
static void InitSK(IServiceCollection services)
62+
/// <param name="_kernel">可以提供自定义Kernel</param>
63+
static void InitSK(IServiceCollection services,Kernel _kernel = null)
6364
{
6465
var handler = new OpenAIHttpClientHandler();
6566
services.AddTransient<Kernel>((serviceProvider) =>
6667
{
67-
var _kernel = Kernel.CreateBuilder()
68-
.AddOpenAIChatCompletion(
69-
modelId: OpenAIOption.ChatModel,
70-
apiKey: OpenAIOption.Key,
71-
httpClient: new HttpClient(handler)
72-
)
73-
.Build();
68+
if (_kernel == null)
69+
{
70+
_kernel = Kernel.CreateBuilder()
71+
.AddOpenAIChatCompletion(
72+
modelId: OpenAIOption.ChatModel,
73+
apiKey: OpenAIOption.Key,
74+
httpClient: new HttpClient(handler)
75+
)
76+
.Build();
77+
}
7478
//导入插件
7579
_kernel.ImportPluginFromPromptDirectory(Path.Combine(RepoFiles.SamplePluginsPath(), "graph"));
7680
return _kernel;

0 commit comments

Comments
 (0)