generated from JetBrains/intellij-platform-plugin-template
-
Notifications
You must be signed in to change notification settings - Fork 316
/
CustomAgentChatProcessor.kt
140 lines (116 loc) · 4.94 KB
/
CustomAgentChatProcessor.kt
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
package cc.unitmesh.devti.agent
import cc.unitmesh.devti.AutoDevBundle
import cc.unitmesh.devti.agent.model.CustomAgentConfig
import cc.unitmesh.devti.agent.model.CustomAgentState
import cc.unitmesh.devti.agent.model.CustomAgentResponseAction
import cc.unitmesh.devti.gui.chat.ChatCodingPanel
import cc.unitmesh.devti.gui.chat.ChatRole
import cc.unitmesh.devti.llms.LLMProvider
import cc.unitmesh.devti.provider.ContextPrompter
import cc.unitmesh.devti.provider.devins.LanguagePromptProcessor
import cc.unitmesh.devti.provider.devins.CustomAgentContext
import cc.unitmesh.devti.util.LLMCoroutineScope
import cc.unitmesh.devti.util.parser.Code
import com.intellij.openapi.components.Service
import com.intellij.openapi.components.service
import com.intellij.openapi.diagnostic.logger
import com.intellij.openapi.project.Project
import kotlinx.coroutines.flow.*
import kotlinx.coroutines.launch
import kotlinx.coroutines.runBlocking
@Service(Service.Level.PROJECT)
class CustomAgentChatProcessor(val project: Project) {
private val customAgentExecutor = project.service<CustomAgentExecutor>()
private val logger = logger<CustomAgentChatProcessor>()
fun handleChat(prompter: ContextPrompter, ui: ChatCodingPanel, llmProvider: LLMProvider) {
val originPrompt = prompter.requestPrompt()
ui.addMessage(originPrompt, true, originPrompt)
val request = originPrompt.trim()
val selectedAgent: CustomAgentConfig = ui.getSelectedCustomAgent()
selectedAgent.state = CustomAgentState.HANDLING
val response: Flow<String>? = customAgentExecutor.execute(request, selectedAgent)
if (response == null) {
logger.error("error for custom agent: $selectedAgent with request: $request")
return
}
selectedAgent.state = CustomAgentState.FINISHED
var devInCode: String? = ""
when (selectedAgent.responseAction) {
CustomAgentResponseAction.Direct -> {
val message = ui.addMessage("loading", false, "")
val sb = StringBuilder()
runBlocking {
val result = ui.updateMessage(response)
sb.append(result)
}
val content = sb.toString().removeSurrounding("\"")
llmProvider.appendLocalMessage(content, ChatRole.Assistant)
message.reRenderAssistantOutput()
val code = Code.parse(content)
if (code.language.displayName == "DevIn") {
devInCode = code.text
}
ui.hiddenProgressBar()
ui.updateUI()
}
CustomAgentResponseAction.Stream -> {
ui.addMessage(AutoDevBundle.message("autodev.loading"))
var msg = ""
LLMCoroutineScope.scope(project).launch {
msg = ui.updateMessage(response)
}
llmProvider.appendLocalMessage(msg, ChatRole.Assistant)
ui.hiddenProgressBar()
ui.updateUI()
val code = Code.parse(msg)
if (code.language.displayName == "DevIn") {
devInCode = code.text
}
}
CustomAgentResponseAction.TextChunk -> {
val sb = StringBuilder()
runBlocking {
response.collect {
sb.append(it)
}
}
val content = sb.toString()
llmProvider.appendLocalMessage(content, ChatRole.Assistant)
ui.removeLastMessage()
ui.moveCursorToStart()
ui.setInput(content)
ui.hiddenProgressBar()
}
CustomAgentResponseAction.Flow -> {
logger.error("will not support flow response for now")
}
CustomAgentResponseAction.WebView -> {
val sb = StringBuilder()
runBlocking {
response.collect {
sb.append(it)
}
}
val content = sb.toString()
llmProvider.appendLocalMessage(content, ChatRole.Assistant)
ui.appendWebView(content, project)
ui.hiddenProgressBar()
}
CustomAgentResponseAction.DevIns -> {
ui.addMessage(AutoDevBundle.message("autodev.loading"))
val msg: String = runBlocking {
ui.updateMessage(response)
}
llmProvider.appendLocalMessage(msg, ChatRole.Assistant)
ui.hiddenProgressBar()
ui.updateUI()
devInCode = msg
}
}
if (!devInCode.isNullOrEmpty()) {
LanguagePromptProcessor.instance("DevIn").forEach {
it.execute(project, CustomAgentContext(selectedAgent, devInCode))
}
}
}
}