Code block rendering error?

[Ethan77-d]

HI,
I found a problem.when AI-generated code and its corresponding explanations are provided, the explanation information is sometimes mixed in after the code wrapped in markdown.
Just like this：

 public static void main(String[] args) {
        System.out.println("Hello World!");

    }
    # This is a sample code.

"# This is a sample code." should not be in the code block

[damiano1996]

Thanks for reporting. I have a unit test for this. Maybe there are uncovered cases: src/test/java/com/github/damiano1996/jetbrains/incoder/tool/window/chat/body/messages/markdown/MarkdownChatMessageTest.java

If you want you can contribute

[Ethan77-d]

Thanks for reporting. I have a unit test for this. Maybe there are uncovered cases: src/test/java/com/github/damiano1996/jetbrains/incoder/tool/window/chat/body/messages/markdown/MarkdownChatMessageTest.java

If you want you can contribute

Thank you,I will try

[Ethan77-d]

The root cause is that ChatBody.write() determines the write target based on the "last element" of the messages list, while the switching of the Markdown block only synchronously updates MarkdownChatMessage.currentMarkdownBlock, and the new block is added to the messages list asynchronously via addChatMessage(). The timing between the two is inconsistent, resulting in tokens continuing to be written to the old block. this is fix in 'chatbody':

    @Nullable
    private ChatMessage getLastStreamTargetMessage() {
        if (messages.isEmpty()) {
            return null;
        }

        // Prefer the last MarkdownChatMessage as the streaming target, because it
        // internally
        // manages the current markdown block (code/text) synchronously.
        // Note: This relies on addChatMessage() synchronously updating the messages
        // list,
        // ensuring that block transitions (code->text, text->code) are immediately
        // visible
        // to streaming tokens, avoiding race conditions in the event queue.
        for (int i = messages.size() - 1; i >= 0; i--) {
            ChatMessage message = messages.get(i);
            if (message instanceof MarkdownChatMessage) {
                return message;
            }
        }

        // Fallback to the last message if no markdown message is present
        return messages.get(messages.size() - 1);
    }
    ...
   

Replace all ‘messages.get(messages.size() - 1)’ with getLastStreamTargetMessage()