Merge pull request #83 from devoxx/issue-82

Fix #82: Set max width for streaming editorPane
devoxx · May 28, 2024 · 13222bd · 13222bd
2 parents c6c45ea + fd23ba6
commit 13222bd
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 2 deletions.
diff --git a/build.gradle.kts b/build.gradle.kts
@@ -5,7 +5,7 @@ plugins {
 }
 
 group = "com.devoxx.genie"
-version = "0.1.14"
+version = "0.1.15"
 
 repositories {
     mavenCentral()

diff --git a/src/main/java/com/devoxx/genie/ui/panel/ChatStreamingResponsePanel.java b/src/main/java/com/devoxx/genie/ui/panel/ChatStreamingResponsePanel.java
@@ -8,6 +8,7 @@
 import org.jetbrains.annotations.NotNull;
 
 import javax.swing.*;
+import java.awt.*;
 import java.util.concurrent.atomic.AtomicReference;
 
 import static com.devoxx.genie.ui.util.DevoxxGenieFonts.SourceCodeProFontPlan14;
@@ -32,8 +33,14 @@ public ChatStreamingResponsePanel(@NotNull ChatMessageContext chatMessageContext
         add(new ResponseHeaderPanel(chatMessageContext));
         add(editorPane);
 
+        setMaxWidth();
+
         parser = Parser.builder().build();
-        renderer = HtmlRenderer
+        renderer = createHTMLRenderer(chatMessageContext);
+    }
+
+    private static HtmlRenderer createHTMLRenderer(@NotNull ChatMessageContext chatMessageContext) {
+        return HtmlRenderer
             .builder()
             .nodeRendererFactory(context -> {
                 AtomicReference<CodeBlockNodeRenderer> codeBlockRenderer = new AtomicReference<>();
@@ -46,6 +53,12 @@ public ChatStreamingResponsePanel(@NotNull ChatMessageContext chatMessageContext
             .build();
     }
 
+    private void setMaxWidth() {
+        Dimension maximumSize = new Dimension(Integer.MAX_VALUE, Integer.MAX_VALUE);
+        editorPane.setMaximumSize(maximumSize);
+        editorPane.setMinimumSize(new Dimension(editorPane.getPreferredSize().width, editorPane.getPreferredSize().height));
+    }
+
     /**
      * Insert token into document stream
      * @param token the LLM string token

diff --git a/src/main/resources/META-INF/plugin.xml b/src/main/resources/META-INF/plugin.xml
@@ -32,6 +32,10 @@
     ]]></description>
 
     <change-notes><![CDATA[
+        <h2>v0.1.15</h2>
+        <UL>
+            <LI>Fix #82: Wrap text to new line for streaming output.</LI>
+        </UL>
         <h2>v0.1.14</h2>
         <UL>
             <LI>Feat #78: Set chat memory size in Settings page.</LI>