Fix unit tests

Author: intitni

Tool/Package.swift

@@ -215,6 +215,7 @@ let package = Package(
             name: "BuiltinExtension",
             dependencies: [
                 "SuggestionModel",
+                "SuggestionProvider",
                 "ChatBasic",
                 "Workspace",
                 "ChatTab",
@@ -369,6 +370,7 @@ let package = Package(
                 "TokenEncoder",
                 "Keychain",
                 "BuiltinExtension",
+                "ChatBasic",
                 .product(name: "JSONRPC", package: "JSONRPC"),
                 .product(name: "AsyncAlgorithms", package: "swift-async-algorithms"),
                 .product(name: "GoogleGenerativeAI", package: "generative-ai-swift"),
@@ -382,6 +384,7 @@ let package = Package(
             name: "OpenAIServiceTests",
             dependencies: [
                 "OpenAIService",
+                "ChatBasic",
                 .product(
                     name: "ComposableArchitecture",
                     package: "swift-composable-architecture"

Tool/Tests/GitHubCopilotServiceTests/FetchSuggestionsTests.swift

@@ -15,10 +15,12 @@ final class FetchSuggestionTests: XCTestCase {
     func test_process_suggestions_from_server() async throws {
         struct TestServer: GitHubCopilotLSP {
             func sendNotification(_: LanguageServerProtocol.ClientNotification) async throws {
-                throw CancellationError()
+                return
             }
 
-            func sendRequest<E>(_: E) async throws -> E.Response where E: GitHubCopilotRequestType {
+            func sendRequest<E>(_: E, timeout: TimeInterval?) async throws -> E.Response
+                where E: GitHubCopilotRequestType
+            {
                 return GitHubCopilotRequest.InlineCompletion.Response(items: [
                     .init(
                         insertText: "Hello World\n",
@@ -70,7 +72,9 @@ final class FetchSuggestionTests: XCTestCase {
                 // unimplemented
             }
 
-            func sendRequest<E>(_: E) async throws -> E.Response where E: GitHubCopilotRequestType {
+            func sendRequest<E>(_: E, timeout: TimeInterval?) async throws -> E.Response
+                where E: GitHubCopilotRequestType
+            {
                 return GitHubCopilotRequest.InlineCompletion.Response(items: [
                     .init(
                         insertText: "Hello World\n",

Tool/Tests/OpenAIServiceTests/AutoManagedChatGPTMemoryRetrievedContentTests.swift

@@ -1,3 +1,4 @@
+import ChatBasic
 import Foundation
 import XCTest
 
@@ -20,7 +21,7 @@ class AutoManagedChatGPTMemoryRetrievedContentTests: XCTestCase {
 
     func test_retrieved_content_when_the_context_window_is_large_enough() async {
         let strategy = Strategy()
-        
+
         let memory = AutoManagedChatGPTMemory(
             systemPrompt: "",
             configuration: UserPreferenceChatGPTConfiguration(),
@@ -57,7 +58,7 @@ class AutoManagedChatGPTMemoryRetrievedContentTests: XCTestCase {
         """
 
         let maxTokenCount = await strategy.countToken(.init(role: .user, content: fullContent))
-        
+
         let result = await memory.generateRetrievedContentMessage(
             maxTokenCount: maxTokenCount,
             strategy: strategy
@@ -159,7 +160,7 @@ class AutoManagedChatGPTMemoryRetrievedContentTests: XCTestCase {
         D
         """)
     }
-    
+
     func test_retrieved_content_when_the_context_window_can_take_only_one_document() async {
         let strategy = Strategy()
 
@@ -200,7 +201,7 @@ class AutoManagedChatGPTMemoryRetrievedContentTests: XCTestCase {
         A
         """)
     }
-    
+
     func test_retrieved_content_when_the_context_window_empty() async {
         let strategy = Strategy()
 

Tool/Tests/OpenAIServiceTests/ChatGPTStreamTests.swift

@@ -1,3 +1,4 @@
+import ChatBasic
 import Dependencies
 import XCTest
 @testable import OpenAIService

Tool/Tests/OpenAIServiceTests/LimitMessagesTests.swift

@@ -1,3 +1,4 @@
+import ChatBasic
 import Foundation
 import TokenEncoder
 import XCTest
@@ -103,11 +104,11 @@ class MockEncoder: TokenEncoder {
 
 struct MockStrategy: AutoManagedChatGPTMemoryStrategy {
     let encoder = MockEncoder()
-    func countToken(_ message: OpenAIService.ChatMessage) async -> Int {
+    func countToken(_ message: ChatBasic.ChatMessage) async -> Int {
         await encoder.countToken(message)
     }
 
-    func countToken<F>(_: F) async -> Int where F: OpenAIService.ChatGPTFunction {
+    func countToken<F>(_: F) async -> Int where F: ChatBasic.ChatGPTFunction {
         0
     }