Skip to content

Commit f21ec5c

Browse files
committed
feat(custom): add AutoDevNotifications on non-empty output #271
Add notification trigger in CustomSSEProcessor when output is non-empty.
1 parent 95a0088 commit f21ec5c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

core/src/main/kotlin/cc/unitmesh/devti/llms/custom/CustomSSEProcessor.kt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
package cc.unitmesh.devti.llms.custom
22

3+
import cc.unitmesh.devti.AutoDevNotifications
34
import cc.unitmesh.devti.coder.recording.EmptyRecording
45
import cc.unitmesh.devti.coder.recording.JsonlRecording
56
import cc.unitmesh.devti.coder.recording.Recording
@@ -145,6 +146,7 @@ open class CustomSSEProcessor(private val project: Project) {
145146

146147
if (output.isNotEmpty()) {
147148
messages += Message(ChatRole.Assistant.roleName(), output)
149+
AutoDevNotifications.notify(project, reasonerOutput)
148150
}
149151

150152
recording.write(RecordingInstruction(promptText, output))

0 commit comments

Comments
 (0)