Streaming support for AI chat interactions

Integrate a streaming feature into the AI assistant, facilitating continuous interaction and dynamic responses. This enhancement allows users to engage with the assistant more fluidly during coding tasks, improving overall efficiency and satisfaction with the tool.
jeddict · Oct 26, 2024 · ef2ed2c · ef2ed2c
1 parent 0eb54c2
commit ef2ed2c
Show file tree

Hide file tree

Showing 21 changed files with 501 additions and 292 deletions.
diff --git a/pom.xml b/pom.xml
@@ -147,6 +147,7 @@
                     </moduleDependencies>
                     <publicPackages>
                         <publicPackage>io.github.jeddict.ai</publicPackage>
+                        <publicPackage>io.github.jeddict.ai.lang</publicPackage>
                     </publicPackages>
                     <useOSGiDependencies>false</useOSGiDependencies>
                 </configuration>

diff --git a/src/main/java/io/github/jeddict/ai/completion/JeddictCompletionProvider.java b/src/main/java/io/github/jeddict/ai/completion/JeddictCompletionProvider.java
@@ -48,7 +48,7 @@
 import com.sun.source.tree.Tree;
 import com.sun.source.tree.VariableTree;
 import com.sun.source.util.DocTrees;
-import io.github.jeddict.ai.Snippet;
+import io.github.jeddict.ai.lang.Snippet;
 import static io.github.jeddict.ai.scanner.ProjectClassScanner.getFileObjectFromEditor;
 import io.github.jeddict.ai.scanner.ClassData;
 import java.util.stream.Collectors;

diff --git a/src/main/java/io/github/jeddict/ai/components/AssistantTopComponent.java b/src/main/java/io/github/jeddict/ai/components/AssistantTopComponent.java
@@ -22,6 +22,7 @@
 import static io.github.jeddict.ai.util.EditorUtil.isSuitableForWebAppDirectory;
 import static io.github.jeddict.ai.util.StringUtil.convertToCapitalized;
 import java.awt.BorderLayout;
+import java.awt.Color;
 import java.awt.Desktop;
 import java.awt.event.MouseAdapter;
 import java.awt.event.MouseEvent;
@@ -114,6 +115,14 @@ public JEditorPane createHtmlPane(String content) {
         parentPanel.add(editorPane);
         return editorPane;
     }
+
+    public JEditorPane createPane() {
+        JEditorPane editorPane = new JEditorPane();
+        editorPane.setEditable(false);
+        editorPane.setBackground(Color.WHITE);
+        parentPanel.add(editorPane);
+        return editorPane;
+    }
 
     public JEditorPane createCodePane(String content) {
         JEditorPane editorPane = new JEditorPane();

diff --git a/src/main/java/io/github/jeddict/ai/hints/ExpressionFix.java b/src/main/java/io/github/jeddict/ai/hints/ExpressionFix.java
@@ -22,8 +22,9 @@
 import com.sun.source.tree.ExpressionTree;
 import com.sun.source.tree.Tree;
 import com.sun.source.util.TreePath;
+import io.github.jeddict.ai.JeddictUpdateManager;
 import io.github.jeddict.ai.completion.Action;
-import io.github.jeddict.ai.JeddictChatModel;
+import io.github.jeddict.ai.lang.JeddictChatModel;
 import io.github.jeddict.ai.util.StringUtil;
 import org.netbeans.api.java.source.JavaSource;
 import org.netbeans.api.java.source.TreePathHandle;
@@ -48,7 +49,7 @@ public ExpressionFix(TreePathHandle tpHandle, Action action, TreePath treePath)
 
     @Override
     protected String getText() {
-        return NbBundle.getMessage(JeddictChatModel.class, "HINT_ENHANCE_EXPRESSION");
+        return NbBundle.getMessage(JeddictUpdateManager.class, "HINT_ENHANCE_EXPRESSION");
     }
 
     @Override

diff --git a/src/main/java/io/github/jeddict/ai/hints/JavaDocFixImpl.java b/src/main/java/io/github/jeddict/ai/hints/JavaDocFixImpl.java
@@ -18,9 +18,7 @@
  */
 package io.github.jeddict.ai.hints;
 
-import io.github.jeddict.ai.util.StringUtil;
 import com.sun.source.doctree.DocCommentTree;
-import com.sun.source.doctree.DocTree;
 import com.sun.source.tree.CompilationUnitTree;
 import com.sun.source.tree.MethodTree;
 import com.sun.source.tree.Tree;
@@ -31,21 +29,18 @@
 import com.sun.source.tree.VariableTree;
 import com.sun.source.util.DocTrees;
 import com.sun.source.util.TreePath;
+import io.github.jeddict.ai.JeddictUpdateManager;
 import io.github.jeddict.ai.completion.Action;
-import io.github.jeddict.ai.JeddictChatModel;
 import static io.github.jeddict.ai.completion.Action.ENHANCE;
+import io.github.jeddict.ai.lang.JeddictChatModel;
 import static io.github.jeddict.ai.util.SourceUtil.geIndentaion;
+import io.github.jeddict.ai.util.StringUtil;
 import static io.github.jeddict.ai.util.StringUtil.removeCodeBlockMarkers;
-import static io.github.jeddict.ai.util.StringUtil.trimLeadingSpaces;
-import java.util.LinkedList;
-import java.util.List;
 import javax.lang.model.element.Element;
-import javax.lang.model.element.ElementKind;
 import javax.swing.text.BadLocationException;
 import javax.swing.text.Document;
 import org.netbeans.api.java.source.ElementHandle;
 import org.netbeans.api.java.source.JavaSource;
-import org.netbeans.api.java.source.TreeMaker;
 import org.netbeans.api.java.source.TreePathHandle;
 import org.netbeans.api.java.source.WorkingCopy;
 import org.netbeans.spi.java.hints.JavaFix;
@@ -71,9 +66,9 @@ public JavaDocFixImpl(TreePathHandle tpHandle, Action type,
     protected String getText() {
         switch (action) {
             case ENHANCE:
-                return NbBundle.getMessage(JeddictChatModel.class, "HINT_JAVADOC_GENERATED", StringUtil.convertToCapitalized(classType.getKind().toString()));//NOI18N
+                return NbBundle.getMessage(JeddictUpdateManager.class, "HINT_JAVADOC_GENERATED", StringUtil.convertToCapitalized(classType.getKind().toString()));//NOI18N
             default:
-                return NbBundle.getMessage(JeddictChatModel.class, "HINT_JAVADOC", StringUtil.convertToCapitalized(classType.getKind().toString()));//NOI18N
+                return NbBundle.getMessage(JeddictUpdateManager.class, "HINT_JAVADOC", StringUtil.convertToCapitalized(classType.getKind().toString()));//NOI18N
             }
     }
 

diff --git a/src/main/java/io/github/jeddict/ai/hints/JeddictHint.java b/src/main/java/io/github/jeddict/ai/hints/JeddictHint.java
@@ -31,7 +31,7 @@
 import com.sun.source.util.SourcePositions;
 import com.sun.source.util.TreePath;
 import com.sun.source.util.Trees;
-import io.github.jeddict.ai.JeddictChatModel;
+import io.github.jeddict.ai.JeddictUpdateManager;
 import io.github.jeddict.ai.completion.Action;
 import io.github.jeddict.ai.settings.PreferencesManager;
 import java.util.ArrayList;
@@ -189,7 +189,7 @@ public static ErrorDescription run(HintContext ctx) {
                 return null;
         }
         Fix[] fixesArray = fixes.toArray(new Fix[0]); // Convert to array
-        String desc = NbBundle.getMessage(JeddictChatModel.class, "ERR_HINT"); //NOI18N
+        String desc = NbBundle.getMessage(JeddictUpdateManager.class, "ERR_HINT"); //NOI18N
         return ErrorDescriptionFactory.forTree(ctx, ctx.getPath(), desc, fixesArray); //NOI18N
     }