Merge pull request #58 from jeddict/dev

Streaming support for AI chat interactions
jeddict · Oct 26, 2024 · 017c0fb · 017c0fb
2 parents 567640f + ef2ed2c
commit 017c0fb
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
     }