fix(CustomAgentChatProcessor): handle response actions more robustly and add logging #51

phodal · phodal · commit d49b2c0e4608 · 2024-03-07T09:48:19.000+08:00
diff --git a/src/main/kotlin/cc/unitmesh/devti/counit/CustomAgentChatProcessor.kt b/src/main/kotlin/cc/unitmesh/devti/counit/CustomAgentChatProcessor.kt
@@ -54,9 +54,14 @@ class CustomAgentChatProcessor(val project: Project) {
 
             CustomAgentResponseAction.Stream -> {
                 ui.addMessage(AutoDevBundle.message("autodev.loading"))
+                var msg = ""
                 LLMCoroutineScope.scope(project).launch {
-                    ui.updateMessage(response)
+                    msg = ui.updateMessage(response)
                 }
+
+                llmProvider.appendLocalMessage(msg, ChatRole.Assistant)
+                ui.hiddenProgressBar()
+                ui.updateUI()
             }
 
             CustomAgentResponseAction.TextChunk -> {
@@ -75,7 +80,7 @@ class CustomAgentChatProcessor(val project: Project) {
             }
 
             CustomAgentResponseAction.Flow -> {
-                TODO()
+                logger.error("will not support flow response for now")
             }
 
             CustomAgentResponseAction.WebView -> {
@@ -85,7 +90,7 @@ class CustomAgentChatProcessor(val project: Project) {
                         sb.append(it)
                     }
                 }
-                // TODO: add decode support
+
                 val content = sb.toString()
                 llmProvider.appendLocalMessage(content, ChatRole.Assistant)
 

Original file line number	Diff line number	Diff line change
`@@ -54,9 +54,14 @@ class CustomAgentChatProcessor(val project: Project) {`
`54`	`54`
`55`	`55`	`CustomAgentResponseAction.Stream -> {`
`56`	`56`	`ui.addMessage(AutoDevBundle.message("autodev.loading"))`
	`57`	`+ var msg = ""`
`57`	`58`	`LLMCoroutineScope.scope(project).launch {`
`58`		`- ui.updateMessage(response)`
	`59`	`+ msg = ui.updateMessage(response)`
`59`	`60`	`}`
	`61`	`+`
	`62`	`+ llmProvider.appendLocalMessage(msg, ChatRole.Assistant)`
	`63`	`+ ui.hiddenProgressBar()`
	`64`	`+ ui.updateUI()`
`60`	`65`	`}`
`61`	`66`
`62`	`67`	`CustomAgentResponseAction.TextChunk -> {`
`@@ -75,7 +80,7 @@ class CustomAgentChatProcessor(val project: Project) {`
`75`	`80`	`}`
`76`	`81`
`77`	`82`	`CustomAgentResponseAction.Flow -> {`
`78`		`- TODO()`
	`83`	`+ logger.error("will not support flow response for now")`
`79`	`84`	`}`
`80`	`85`
`81`	`86`	`CustomAgentResponseAction.WebView -> {`
`@@ -85,7 +90,7 @@ class CustomAgentChatProcessor(val project: Project) {`
`85`	`90`	`sb.append(it)`
`86`	`91`	`}`
`87`	`92`	`}`
`88`		`- // TODO: add decode support`
	`93`	`+`
`89`	`94`	`val content = sb.toString()`
`90`	`95`	`llmProvider.appendLocalMessage(content, ChatRole.Assistant)`
`91`	`96`