Skip to content

Commit 489fab2

Browse files
committed
Add suggestion feature provider id
1 parent 9cf8a3e commit 489fab2

File tree

2 files changed

+14
-8
lines changed

2 files changed

+14
-8
lines changed

Tool/Sources/CodeiumService/CodeiumExtension.swift

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,13 @@
1+
import BuiltinExtension
12
import CopilotForXcodeKit
23
import Foundation
3-
import Workspace
44
import Logger
5-
import BuiltinExtension
5+
import Preferences
6+
import Workspace
67

78
public final class CodeiumExtension: BuiltinExtension {
9+
public var suggestionServiceId: Preferences.BuiltInSuggestionFeatureProvider { .codeium }
10+
811
public var suggestionService: SuggestionServiceType? { _suggestionService }
912
public var chatService: ChatServiceType? { nil }
1013
public var promptToCodeService: PromptToCodeServiceType? { nil }
@@ -68,7 +71,7 @@ public final class CodeiumExtension: BuiltinExtension {
6871
let fileSize = attrs[FileAttributeKey.size] as? UInt64,
6972
fileSize > 15 * 1024 * 1024
7073
{ return }
71-
74+
7275
Task {
7376
do {
7477
let content = try String(contentsOf: documentURL, encoding: .utf8)
@@ -89,7 +92,7 @@ public final class CodeiumExtension: BuiltinExtension {
8992
}
9093
}
9194
}
92-
95+
9396
public func terminate() {
9497
for workspace in workspacePool.workspaces.values {
9598
guard let plugin = workspace.plugin(for: CodeiumWorkspacePlugin.self)

Tool/Sources/GitHubCopilotService/GitHubCopilotExtension.swift

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,13 @@
1+
import BuiltinExtension
12
import CopilotForXcodeKit
23
import Foundation
3-
import Workspace
44
import Logger
5-
import BuiltinExtension
5+
import Preferences
6+
import Workspace
67

78
public final class GitHubCopilotExtension: BuiltinExtension {
9+
public var suggestionServiceId: Preferences.BuiltInSuggestionFeatureProvider { .gitHubCopilot }
10+
811
public var suggestionService: SuggestionServiceType? { _suggestionService }
912
public var chatService: ChatServiceType? { nil }
1013
public var promptToCodeService: PromptToCodeServiceType? { nil }
@@ -75,7 +78,7 @@ public final class GitHubCopilotExtension: BuiltinExtension {
7578
let fileSize = attrs[FileAttributeKey.size] as? UInt64,
7679
fileSize > 15 * 1024 * 1024
7780
{ return }
78-
81+
7982
Task {
8083
do {
8184
let content = try String(contentsOf: documentURL, encoding: .utf8)
@@ -96,7 +99,7 @@ public final class GitHubCopilotExtension: BuiltinExtension {
9699
}
97100
}
98101
}
99-
102+
100103
public func terminate() {
101104
for workspace in workspacePool.workspaces.values {
102105
guard let plugin = workspace.plugin(for: GitHubCopilotWorkspacePlugin.self)

0 commit comments

Comments
 (0)