Implement optional discussion mode

2026-09-13 21:13:52 +02:00
parent bf1ccbe8a9
commit a39bdbe6fc
12 changed files with 291 additions and 1 deletions
@@ -17,6 +17,12 @@ final class ApplicationRuleInjectionTests: XCTestCase {
                context: AIRequestContext(userInstruction: "Generate.")
            )
        )
+        _ = try await service.discussSongProject(
+            from: SongProjectDiscussionRequest(
+                context: AIRequestContext(userInstruction: "Discuss."),
+                project: project
+            )
+        )
        _ = try await service.reviseLyrics(
            from: LyricsRevisionRequest(
                context: AIRequestContext(userInstruction: "Improve."),
@@ -37,6 +43,7 @@ final class ApplicationRuleInjectionTests: XCTestCase {
        XCTAssertEqual(
            recordedRuleContents,
            [
+                "Use private product rules.",
                "Use private product rules.",
                "Use private product rules.",
                "Use private product rules."
@@ -90,6 +97,11 @@ private actor RecordingAIService: AIService {
        )
    }

+    func discussSongProject(from request: SongProjectDiscussionRequest) async throws -> SongProjectDiscussionResult {
+        contexts.append(request.context)
+        return SongProjectDiscussionResult(questions: ["Which direction should we take?"])
+    }
+
    func reviseLyrics(from request: LyricsRevisionRequest) async throws -> LyricsRevisionResult {
        contexts.append(request.context)
        return LyricsRevisionResult(lyrics: request.sourceLyrics)