Update the tests

Author: kean

Modules/Sources/WordPressIntelligence/IntelligenceService.swift

@@ -48,8 +48,8 @@ public actor IntelligenceService {
 
     /// Summarizes a WordPress post.
     @available(iOS 26, *)
-    public func summarizePost(content: String) -> LanguageModelSession.ResponseStream<String> {
-        PostSummary.execute(content: content)
+    public func summarizePost(content: String) async throws -> String {
+        try await PostSummary.execute(content: content)
     }
 
     /// Summarizes a support ticket to a short title.
@@ -67,7 +67,7 @@ public actor IntelligenceService {
 
     /// Extracts relevant text from post content, removing HTML and limiting size.
     public nonisolated static func extractRelevantText(from post: String, ratio: CGFloat = 0.6) -> String {
-        let extract = try? IntelligenceUtilities.extractRelevantText(from: post)
+        let extract = try? ContentExtractor.extractRelevantText(from: post)
         let postSizeLimit = Double(IntelligenceService.contextSizeLimit) * ratio
         return String((extract ?? post).prefix(Int(postSizeLimit)))
     }

Modules/Sources/WordPressIntelligence/UseCases/ExcerptGeneration.swift

@@ -1,24 +1,37 @@
 import Foundation
 import FoundationModels
 
-    /// Excerpt generation for WordPress posts.
-    ///
-    /// Generates multiple excerpt variations for blog posts with customizable
-    /// length and writing style. Supports session-based usage (for UI with continuity)
-    /// and one-shot generation (for tests and background tasks).
-    @available(iOS 26, *)
+/// Excerpt generation for WordPress posts.
+///
+/// Generates multiple excerpt variations for blog posts with customizable
+/// length and writing style. Supports session-based usage (for UI with continuity)
+/// and one-shot generation (for tests and background tasks).
+@available(iOS 26, *)
 public struct ExcerptGeneration {
-    public let length: ContentLength
-    public let style: WritingStyle
+    public var length: ContentLength
+    public var style: WritingStyle
     public var options: GenerationOptions
 
-    public init(length: ContentLength, style: WritingStyle, options: GenerationOptions = GenerationOptions(temperature: 0.7)) {
+    public init(
+        length: ContentLength,
+        style: WritingStyle,
+        options: GenerationOptions = GenerationOptions(temperature: 0.7)
+    ) {
         self.length = length
         self.style = style
         self.options = options
     }
 
-    // MARK: - Instance Methods
+    /// Generates excerpts with this configuration.
+    public func generate(for content: String) async throws -> [String] {
+        let content = IntelligenceService.extractRelevantText(from: content)
+        let response = try await makeSession().respond(
+            to: makePrompt(content: content),
+            generating: Result.self,
+            options: options
+        )
+        return response.content.excerpts
+    }
 
     /// Creates a language model session configured for excerpt generation.
     public func makeSession() -> LanguageModelSession {
@@ -28,63 +41,48 @@ public struct ExcerptGeneration {
         )
     }
 
+    /// Instructions for the language model session.
+    public static var instructions: String {
+        """
+        Generate exactly 3 excerpts for the blog post and follow the instructions from the prompt regarding the length and the style.
+        
+        Generate excerpts in the same language as the POST_CONTENT.
+        
+        **Paramters**
+        - POST_CONTENT: contents of the post (HTML or plain text)
+        - GENERATED_CONTENT_LENGTH: the length of the generated content
+        - GENERATION_STYLE: the writing style to follow
+        
+        **Requirements**
+        - Each excerpt must follow the provided GENERATED_CONTENT_LENGTH and use GENERATION_STYLE
+        
+        **Excerpt best practices**
+        - Follow the best practices for post excerpts esteblished in the WordPress ecosystem
+        - Include the post's main value proposition
+        - Use active voice (avoid "is", "are", "was", "were" when possible)
+        - End with implicit promise of more information
+        - Do not use ellipsis (...) at the end
+        - Focus on value, not summary
+        - Include strategic keywords naturally
+        - Write independently from the introduction – excerpt shouldn't just duplicate your opening paragraph. While your introduction eases readers into the topic, your excerpt needs to work as standalone copy that makes sense out of context—whether it appears in search results, social media cards, or email newsletters.
+        """
+    }
+
+
     /// Creates a prompt for this excerpt configuration.
     public func makePrompt(content: String) -> String {
         """
         Generate three different excerpts for the given post and parameters
-
+        
         GENERATED_CONTENT_LENGTH: \(length.promptModifier)
-
+        
         GENERATION_STYLE: \(style.promptModifier)
-
+        
         POST_CONTENT: '''
         \(content)
         """
     }
 
-    /// Generates excerpts with this configuration.
-    public func generate(content: String) async throws -> [String] {
-        let extractedContent = IntelligenceService.extractRelevantText(from: content)
-        let session = makeSession()
-
-        let response = try await session.respond(
-            to: makePrompt(content: extractedContent),
-            generating: Result.self,
-            options: options
-        )
-
-        return response.content.excerpts
-    }
-
-    // MARK: - Building Blocks (for UI with session continuity)
-
-    /// Instructions for the language model session.
-    public static var instructions: String {
-            """
-            Generate exactly 3 excerpts for the blog post and follow the instructions from the prompt regarding the length and the style.
-            
-            Generate excerpts in the same language as the POST_CONTENT.
-            
-            **Paramters**
-            - POST_CONTENT: contents of the post (HTML or plain text)
-            - GENERATED_CONTENT_LENGTH: the length of the generated content
-            - GENERATION_STYLE: the writing style to follow
-            
-            **Requirements**
-            - Each excerpt must follow the provided GENERATED_CONTENT_LENGTH and use GENERATION_STYLE
-            
-            **Excerpt best practices**
-            - Follow the best practices for post excerpts esteblished in the WordPress ecosystem
-            - Include the post's main value proposition
-            - Use active voice (avoid "is", "are", "was", "were" when possible)
-            - End with implicit promise of more information
-            - Do not use ellipsis (...) at the end
-            - Focus on value, not summary
-            - Include strategic keywords naturally
-            - Write independently from the introduction – excerpt shouldn't just duplicate your opening paragraph. While your introduction eases readers into the topic, your excerpt needs to work as standalone copy that makes sense out of context—whether it appears in search results, social media cards, or email newsletters.
-            """
-    }
-
     /// Prompt for generating additional excerpt options.
     public static var loadMorePrompt: String {
         "Generate additional three options"

Modules/Sources/WordPressIntelligence/UseCases/PostSummary.swift

@@ -8,7 +8,7 @@ extension IntelligenceService {
     /// Generates concise summaries that capture the main points and key information
     /// from WordPress post content in the same language as the source.
     public enum PostSummary {
-        static func execute(content: String) -> LanguageModelSession.ResponseStream<String> {
+        static func execute(content: String) async throws -> String {
             let content = IntelligenceService.extractRelevantText(from: content, ratio: 0.8)
 
             let instructions = """
@@ -31,7 +31,7 @@ extension IntelligenceService {
             \(content)
             """
 
-            return session.streamResponse(to: prompt)
+            return try await session.respond(to: prompt).content
         }
     }
 }

Modules/Sources/WordPressIntelligence/IntelligenceUtilities.swift renamed to Modules/Sources/WordPressIntelligence/Utilities/ContentExtractor.swift

@@ -1,7 +1,7 @@
 import Foundation
 import SwiftSoup
 
-public struct IntelligenceUtilities {
+public enum ContentExtractor {
     /// Extracts semantically meaningful content from HTML for LLM processing.
    ///
    /// Optimized for language models by:

Modules/Tests/WordPressIntelligenceTests/IntelligenceUtilitiesTests.swift renamed to Modules/Tests/WordPressIntelligenceTests/ContentExtractorTests.swift

@@ -1,9 +1,9 @@
 import Testing
 @testable import WordPressIntelligence
 
-struct IntelligenceUtilitiesTests {
+struct ContentExtractorTests {
     @Test func extractRelevantText() throws {
-        let text = try IntelligenceUtilities.extractRelevantText(from: IntelligenceUtilities.post)
+        let text = try ContentExtractor.extractRelevantText(from: ContentExtractor.post)
 
         #expect(text == """
         <h1>The Art of Making Perfect Sourdough Bread at Home</h1>
@@ -52,7 +52,7 @@ struct IntelligenceUtilitiesTests {
 
     /// Blockquote contain nested block and the implementation should account for that.
     @Test func blockquotes() throws {
-        let text = try IntelligenceUtilities.extractRelevantText(from: """
+        let text = try ContentExtractor.extractRelevantText(from: """
         <!-- wp:paragraph -->
         <p>Welcome to <strong><em>WordPress</em></strong>! This is your first post. Edit or delete it to take the first step in your blogging journey.</p>
         <!-- /wp:paragraph -->
@@ -71,13 +71,13 @@ struct IntelligenceUtilitiesTests {
     }
 
     @Test func extractRelevantTextFromPlainText() throws {
-        let text = try IntelligenceUtilities.extractRelevantText(from: "This is a plain text post")
+        let text = try ContentExtractor.extractRelevantText(from: "This is a plain text post")
 
         #expect(text == "This is a plain text post")
     }
 }
 
-extension IntelligenceUtilities {
+extension ContentExtractor {
     static let post = """
     <!-- wp:heading {"level":1} -->
     <h1>The Art of Making Perfect Sourdough Bread at Home</h1>