forked from intitni/CopilotForXcode
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathChatProvider+Service.swift
More file actions
103 lines (90 loc) · 2.69 KB
/
ChatProvider+Service.swift
File metadata and controls
103 lines (90 loc) · 2.69 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
import ChatService
import Combine
import Foundation
import OpenAIService
import SuggestionWidget
extension ChatProvider {
convenience init(
service: ChatService,
fileURL: URL,
onCloseChat: @escaping () -> Void,
onSwitchContext: @escaping () -> Void
) {
self.init(pluginIdentifiers: service.allPluginCommands)
let cancellable = service.objectWillChange.sink { [weak self] in
guard let self else { return }
Task { @MainActor in
self.history = (await service.chatGPTService.history).map { message in
.init(
id: message.id,
isUser: message.role == .user,
text: message.summary ?? message.content
)
}
self.isReceivingMessage = service.isReceivingMessage
self.systemPrompt = service.systemPrompt
self.extraSystemPrompt = service.extraSystemPrompt
}
}
service.objectWillChange.send()
onMessageSend = { [cancellable] message in
_ = cancellable
Task {
do {
_ = try await service.send(content: message)
} catch {
PresentInWindowSuggestionPresenter().presentError(error)
}
}
}
onStop = {
Task {
await service.stopReceivingMessage()
}
}
onClear = {
Task {
await service.clearHistory()
}
}
onClose = {
Task {
await service.stopReceivingMessage()
onCloseChat()
}
}
self.onSwitchContext = {
onSwitchContext()
}
onDeleteMessage = { id in
Task {
await service.deleteMessage(id: id)
}
}
onResendMessage = { id in
Task {
do {
try await service.resendMessage(id: id)
} catch {
PresentInWindowSuggestionPresenter().presentError(error)
}
}
}
onResetPrompt = {
Task {
await service.resetPrompt()
}
}
onRunCustomCommand = { command in
Task {
let commandHandler = PseudoCommandHandler()
await commandHandler.handleCustomCommand(command)
}
}
onSetAsExtraPrompt = { id in
Task {
await service.setMessageAsExtraPrompt(id: id)
}
}
}
}