修正

OUCC · May 1, 2024 · b414cc3 · b414cc3
1 parent 02421d7
commit b414cc3
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 28 deletions.
diff --git a/KoeBook.Core/Contracts/Services/ILlmStoryGeneratorService.cs b/KoeBook.Core/Contracts/Services/ILlmStoryGeneratorService.cs
diff --git a/KoeBook.Core/Services/ClaudeStoryGeneratorService.cs b/KoeBook.Core/Services/ClaudeStoryGeneratorService.cs
@@ -3,11 +3,11 @@
 
 namespace KoeBook.Core.Services;
 
-internal class ClaudeStoryGeneratorService(IClaudeService claudeService) : ILlmStoryGeneratorService
+public class ClaudeStoryGeneratorService(IClaudeService claudeService) : IStoryCreatorService
 {
     private readonly IClaudeService _claudeService = claudeService;
 
-    public async ValueTask<string> GenerateStoryAsync(StoryGenre storyGenre, string premise, CancellationToken cancellationToken)
+    public async ValueTask<string> CreateStoryAsync(StoryGenre storyGenre, string premise, CancellationToken cancellationToken)
     {
         if (_claudeService.Messages is null)
         {

diff --git a/KoeBook/App.xaml.cs b/KoeBook/App.xaml.cs
@@ -4,10 +4,7 @@
 using KoeBook.Contracts.Services;
 using KoeBook.Core;
 using KoeBook.Core.Contracts.Services;
-using KoeBook.Core.Services;
 using KoeBook.Core.Services.Mocks;
-using KoeBook.Epub.Contracts.Services;
-using KoeBook.Epub.Services;
 using KoeBook.Models;
 using KoeBook.Notifications;
 using KoeBook.Services;
@@ -82,7 +79,6 @@ public App()
                 services.AddSingleton<IDialogService, DialogService>();
                 services.AddSingleton<IDisplayStateChangeService, DisplayStateChangeService>();
                 services.AddSingleton<ICreateCoverFileService, CreateCoverFileService>();
-                services.AddSingleton<IStoryCreatorService, StoryCreatorService>();
 
                 // Views and ViewModels
                 services.AddTransient<SettingsViewModel>();

diff --git a/KoeBook/Services/StoryCreatorService.cs b/KoeBook/Services/StoryCreatorService.cs
diff --git a/KoeBook/Startup.cs b/KoeBook/Startup.cs
@@ -43,7 +43,8 @@ public static IHostBuilder UseCoreStartup(this IHostBuilder builder)
                 .AddSingleton<IScraperSelectorService, ScraperSelectorService>()
                 .AddSingleton<IScrapingService, ScrapingAozoraService>()
                 .AddSingleton<IScrapingService, ScrapingNaroService>()
-                .AddSingleton<AiStoryAnalyzerService>();
+                .AddSingleton<AiStoryAnalyzerService>()
+                .AddSingleton<IStoryCreatorService, ClaudeStoryGeneratorService>();
             services.AddSingleton<IEpubCreateService, EpubCreateService>();
             services.AddSingleton<ISplitBraceService, SplitBraceService>();
             services.AddSingleton<IFileExtensionService, FileExtensionService>();