1.3.2 (#138)

* 1.3.2

* wip

* wip

* fixes

* Update StaticAppSettingsConfigurable.kt

* wip

Author: acharneski

README.md

@@ -8,23 +8,28 @@
 
 # **AI Coding Assistant Plugin for IntelliJ**
 
-**Fully open source plugin for IntelliJ that integrates with OpenAI's GPT-4 API**
-
-* This requires an OpenAI access token, which you can get by signing up for a developer account at https://platform.openai.com/
-* No membership fees! API access is paid but is billed by usage, with no base fee. The plugin is free and open source.
-* Fully customizable actions in Groovy, including the ability to create your own actions.
+**Fully open source plugin for IntelliJ that integrates with a variety of LLM APIs**
+
+* No membership fees! API access is typically paid but is billed by usage, with no base fee. 
+* The plugin is free and open source, with a focus on utility. No hype here! I'm sharing a tool I use, and I hope you find it useful too.
+* A variety of APIs are supported! Get your api keys at: 
+  * https://platform.openai.com/
+  * https://www.perplexity.ai/
+  * https://console.groq.com/
+  * https://modelslab.com/dashboard/
+  * https://console.aws.amazon.com/bedrock/
+* Fully customizable actions, including the ability to create your own actions using AI
 * Toolbar UI for quick configuration of temperature/model and display of current token count
 * Ability to intercept, edit, and log API requests
-* Now with added support for models from https://www.perplexity.ai/, https://console.groq.com/, and https://modelslab.com/dashboard/, expanding the plugin's capabilities in code generation and analysis.
 
 **NOTE**: This project is not affiliated with OpenAI, JetBrains, or any other corporation or organization. 
-It is provided free of charge, as-is, with no warranty or guarantee of any kind.
+It is provided free of charge, as-is, with no warranty or guarantee of any kind. 
+It is the work of a sole developer working as a hobby project.
 
 ## **Installation & Configuration**
 
-To begin with AI Coding Assistant, you will need an [OpenAI access token](https://platform.openai.com/account/api-keys). 
-After obtaining your OpenAI token, input it into the appropriate field in the plugin's settings panel.
- Now, you can also configure the plugin to use models from Perplexity AI, Groq Console, and Models Lab for enhanced coding assistance.
+To begin with AI Coding Assistant, you will need one or more api tokens, 
+which needs to be input it into the appropriate field in the plugin's settings panel.
 
 ## **Usage Overview**
 
@@ -37,14 +42,15 @@ over when and how to use them. You can access these actions via the context menu
 * **DiffChatAction**: Engage in a chat session to generate and apply code diffs directly within the IDE, streamlining the code review and modification process.
 * **MultiDiffChatAction**: Facilitates collaborative code review and diff generation across multiple files, enhancing team productivity.
 * **AutoDevAction**: Translates user directives into actionable development tasks and code modifications, automating parts of the development workflow.
+* **mermaid.js**: Generate diagrams using the mermaid.js library, providing a visual representation of your code and ideas.
 
 ## **Action Customization**
 
-In our latest version, we provide the capability to tailor actions to your coding habits and project requirements.
-Within the settings UI, you can view, edit, clone, or delete actions, enabling you to fine-tune existing tools or create
-new ones from scratch.
+Tailor actions to your coding habits and project requirements!
+Within the settings UI, you can view, edit, clone, or delete actions, 
+enabling you to fine-tune existing tools or create new ones from scratch.
 
-These custom actions leverage Groovy scripting, and can be as complex or simple as required, even having access to the
+These custom actions use dynamically compiled Kotlin, and can be as complex or simple as required, even having access to the
 entire IntelliJ API. This powerful feature allows for numerous possibilities, from refining prompts to adding intricate
 logic to better support your preferred coding language.
 

build.gradle.kts

@@ -26,7 +26,7 @@ repositories {
 val kotlin_version = "1.9.21"
 val jetty_version = "11.0.18"
 val slf4j_version = "2.0.9"
-val skyenet_version = "1.0.50"
+val skyenet_version = "1.0.52"
 val remoterobot_version = "0.11.21"
 dependencies {
 
@@ -37,7 +37,7 @@ dependencies {
         exclude(group = "org.jetbrains.kotlin", module = "")
     }
 
-    implementation(group = "com.simiacryptus", name = "jo-penai", version = "1.0.46")
+    implementation(group = "com.simiacryptus", name = "jo-penai", version = "1.0.48")
     {
         exclude(group = "org.jetbrains.kotlin", module = "")
     }

gradle.properties

@@ -1,6 +1,6 @@
 pluginName=intellij-aicoder
 pluginRepositoryUrl=https://github.com/SimiaCryptus/intellij-aicoder
-pluginVersion=1.3.1
+pluginVersion=1.3.2
 
 jvmArgs=-Xmx8g
 org.gradle.jvmargs=-Xmx8g

src/main/kotlin/com/github/simiacryptus/aicoder/actions/generic/AutoDevAction.kt

@@ -6,6 +6,7 @@ import com.github.simiacryptus.aicoder.actions.dev.AppServer
 import com.github.simiacryptus.aicoder.util.UITools
 import com.github.simiacryptus.aicoder.util.addApplyDiffLinks
 import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.actionSystem.PlatformDataKeys
 import com.simiacryptus.jopenai.API
 import com.simiacryptus.jopenai.describe.Description
 import com.simiacryptus.jopenai.models.ChatModels
@@ -103,9 +104,9 @@ class AutoDevAction : BaseAction() {
     val ui: ApplicationInterface,
     val model: ChatModels,
     val tools: List<String> = emptyList(),
-    val actorMap: Map<ActorTypes, BaseActor<*, *>> = mapOf(
+    private val actorMap: Map<ActorTypes, BaseActor<*, *>> = mapOf(
       ActorTypes.DesignActor to ParsedActor(
-        parserClass = TaskListParser::class.java,
+        resultClass = TaskList::class.java,
         prompt = """
             Translate the user directive into an action plan for the project.
             Break the user's request into a list of simple tasks to be performed.
@@ -139,19 +140,25 @@ class AutoDevAction : BaseAction() {
       ),
     ),
     val event: AnActionEvent,
-  ) : ActorSystem<AutoDevAgent.ActorTypes>(actorMap, dataStorage, user, session) {
+  ) : ActorSystem<AutoDevAgent.ActorTypes>(actorMap.map { it.key.name to it.value.javaClass }.toMap(), dataStorage, user, session) {
     enum class ActorTypes {
       DesignActor,
       TaskCodingActor,
     }
 
-    val designActor by lazy { getActor(ActorTypes.DesignActor) as ParsedActor<TaskList> }
-    val taskActor by lazy { getActor(ActorTypes.TaskCodingActor) as SimpleActor }
+    private val designActor by lazy { getActor(ActorTypes.DesignActor) as ParsedActor<TaskList> }
+    private val taskActor by lazy { getActor(ActorTypes.TaskCodingActor) as SimpleActor }
 
     fun start(
       userMessage: String,
     ) {
       val codeFiles = mutableMapOf<String, String>()
+      val root = PlatformDataKeys.VIRTUAL_FILE_ARRAY.getData(event.dataContext)?.map { it.toFile.toPath() }?.toTypedArray()?.commonRoot()!!
+      PlatformDataKeys.VIRTUAL_FILE_ARRAY.getData(event.dataContext)?.forEach { file ->
+        val code = file.inputStream.bufferedReader().use { it.readText() }
+        codeFiles[root.relativize(file.toNioPath()).toString()] = code
+      }
+      require(codeFiles.isNotEmpty()) { "No files selected" }
       fun codeSummary() = codeFiles.entries.joinToString("\n\n") { (path, code) ->
         "# $path\n```${
           path.split('.').last()
@@ -177,15 +184,18 @@ class AutoDevAction : BaseAction() {
             val task = ui.newTask()
             task.header("Task: $description")
             AgentPatterns.retryable(ui,task) {
+              val filter = codeFiles.filter { (path, _) -> paths?.find { path.contains(it) }?.isNotEmpty() == true }
+              require(filter.isNotEmpty()) { "No files found for $paths" }
               renderMarkdown(
                 ui.socketManager.addApplyDiffLinks(
-                  codeFiles.filter { (path, _) -> paths?.contains(path) == true },
-                  taskActor.answer(listOf(
+                  code = codeFiles,
+                  response = taskActor.answer(listOf(
+                    codeSummary(),
                     userMessage,
-                    architectureResponse.text,
-                    codeFiles.filter { (path, _) -> paths?.contains(path) == true }.entries.joinToString("\n\n") {
+                    filter.entries.joinToString("\n\n") {
                       "# ${it.key}\n```${it.key.split('.').last()}\n${it.value}\n```"
                     },
+                    architectureResponse.text,
                     "Provide a change for ${paths?.joinToString(",") { it } ?: ""} ($description)"
                   ), api)
                 ) { newCodeMap ->
@@ -229,11 +239,6 @@ class AutoDevAction : BaseAction() {
       return socketServer
     }
 
-    interface TaskListParser : java.util.function.Function<String, TaskList> {
-      @Description("Parse out a list of tasks to be performed in this project")
-      override fun apply(text: String): TaskList
-    }
-
     data class TaskList(
       @Description("List of tasks to be performed in this project")
       val tasks: List<Task> = emptyList()

src/main/kotlin/com/github/simiacryptus/aicoder/actions/generic/CreateFileAction.kt

@@ -12,15 +12,6 @@ class CreateFileAction : FileContextAction<CreateFileAction.Settings>(false, tru
 
   class ProjectFile(var path: String = "", var code: String = "")
 
-  class SettingsUI {
-    @Name("Directive")
-    var directive: JTextArea = JTextArea(
-      """
-                Create a default log4j configuration file
-            """.trimIndent(), 3, 120
-    )
-  }
-
   class Settings(var directive: String = "")
 
   override fun processSelection(

src/main/kotlin/com/github/simiacryptus/aicoder/actions/generic/DocumentationCompilerAction.kt

@@ -31,19 +31,6 @@ class DocumentationCompilerAction : FileContextAction<DocumentationCompilerActio
   }
 
   class SettingsUI {
-    @Name("Transformation Message")
-    var transformationMessage: JTextArea = JTextArea(
-      "Create user documentation",
-      3,
-      120
-    )
-
-    @Name("Output Filename")
-    var outputFilename: JTextArea = JTextArea(
-      "compiled_documentation.md",
-      1,
-      120
-    )
 
     @Name("Files to Process")
     var filesToProcessScrollPane: JBScrollPane = JBScrollPane()
@@ -82,7 +69,7 @@ class DocumentationCompilerAction : FileContextAction<DocumentationCompilerActio
       "Compile Documentation"
     ) { }
     settings.filesToProcess = files.filter { path -> filesToProcess.isItemSelected(path) }.toList()
-      //.map { path -> return@map root?.resolve(path) }.filterNotNull()
+    //.map { path -> return@map root?.resolve(path) }.filterNotNull()
     return Settings(settings, project)
   }
 
@@ -117,7 +104,14 @@ class DocumentationCompilerAction : FileContextAction<DocumentationCompilerActio
             markdownContent.append("\n\n")
             path
           }
-        }.toTypedArray().map { future -> future.get() ?: return@map null }.filterNotNull()
+        }.toTypedArray().map { future ->
+          try {
+            future.get() ?: return@map null
+          } catch (e: Exception) {
+            log.warn("Error processing file", e)
+            return@map null
+          }
+        }.filterNotNull()
       Files.write(outputPath, markdownContent.toString().toByteArray())
       open(config?.project!!, outputPath)
       return arrayOf(outputPath.toFile())