From 45ef0ab3510f849df102c42cce0aa7f0fd2f4361 Mon Sep 17 00:00:00 2001
From: Andrew Charneski <acharneski@gmail.com>
Date: Wed, 17 Apr 2024 19:39:48 -0400
Subject: [PATCH] fixes

---
 .../actions/generic/GenerateRelatedFileAction.kt    | 13 ++++++++++++-
 .../aicoder/actions/generic/MultiDiffChatAction.kt  |  2 +-
 .../jopenai/exceptions/CustomException.kt           |  3 +++
 3 files changed, 16 insertions(+), 2 deletions(-)
 create mode 100644 src/main/kotlin/com/simiacryptus/jopenai/exceptions/CustomException.kt

diff --git a/src/main/kotlin/com/github/simiacryptus/aicoder/actions/generic/GenerateRelatedFileAction.kt b/src/main/kotlin/com/github/simiacryptus/aicoder/actions/generic/GenerateRelatedFileAction.kt
index 59dad47c..674cd4ee 100644
--- a/src/main/kotlin/com/github/simiacryptus/aicoder/actions/generic/GenerateRelatedFileAction.kt
+++ b/src/main/kotlin/com/github/simiacryptus/aicoder/actions/generic/GenerateRelatedFileAction.kt
@@ -3,6 +3,7 @@ package com.github.simiacryptus.aicoder.actions.generic
 import com.github.simiacryptus.aicoder.actions.FileContextAction
 import com.github.simiacryptus.aicoder.config.AppSettingsState
 import com.github.simiacryptus.aicoder.config.AppSettingsState.Companion.chatModel
+import com.github.simiacryptus.aicoder.config.Name
 import com.github.simiacryptus.aicoder.util.UITools
 import com.intellij.openapi.actionSystem.ActionUpdateThread
 import com.intellij.openapi.actionSystem.AnActionEvent
@@ -20,6 +21,7 @@ import java.io.File
 import java.io.FileInputStream
 import java.nio.file.Path
 import java.util.concurrent.TimeUnit
+import javax.swing.JTextArea
 
 class GenerateRelatedFileAction : FileContextAction<GenerateRelatedFileAction.Settings>() {
     override fun getActionUpdateThread() = ActionUpdateThread.BGT
@@ -33,7 +35,16 @@ class GenerateRelatedFileAction : FileContextAction<GenerateRelatedFileAction.Se
         val code: String = ""
     )
 
-    class SettingsUI
+    class SettingsUI {
+        @Name("Directive")
+        var directive: JTextArea = JTextArea(
+            """
+            Create test cases
+            """.trimIndent(),
+            3,
+            120
+        )
+    }
 
     class UserSettings(
         var directive: String = "",
diff --git a/src/main/kotlin/com/github/simiacryptus/aicoder/actions/generic/MultiDiffChatAction.kt b/src/main/kotlin/com/github/simiacryptus/aicoder/actions/generic/MultiDiffChatAction.kt
index 50b0478a..3cd9d1b4 100644
--- a/src/main/kotlin/com/github/simiacryptus/aicoder/actions/generic/MultiDiffChatAction.kt
+++ b/src/main/kotlin/com/github/simiacryptus/aicoder/actions/generic/MultiDiffChatAction.kt
@@ -143,7 +143,7 @@ class MultiDiffChatAction : BaseAction() {
                 path = path,
                 showMenubar = false,
             ) {
-                override val singleInput = true
+                override val singleInput = false
                 override val stickyInput = false
                 override fun newSession(user: User?, session: Session) = agents[session]!!
             }
diff --git a/src/main/kotlin/com/simiacryptus/jopenai/exceptions/CustomException.kt b/src/main/kotlin/com/simiacryptus/jopenai/exceptions/CustomException.kt
new file mode 100644
index 00000000..e77a3191
--- /dev/null
+++ b/src/main/kotlin/com/simiacryptus/jopenai/exceptions/CustomException.kt
@@ -0,0 +1,3 @@
+package com.simiacryptus.jopenai.exceptions
+
+class CustomException(message: String) : Exception(message)
\ No newline at end of file