feat(model): introduce CustomRagApp and ResponseAction enum, refactor CodePayload to use text instead of payload_type, update AutoDevInputSection to use send icon, add examples to Tooling class as QAExample objects. #51

This commit introduces a new `CustomRagApp` data class and an `ResponseAction` enum to handle response actions in the counit model. Additionally, the `CodePayload` class has been refactored to use `text` instead of `payload_type`. The `AutoDevInputSection` has been updated to use the send icon for the presentation, and the `Tooling` class now includes `QAExample` objects for its examples.

Author: phodal

src/main/kotlin/cc/unitmesh/devti/counit/dto/PayloadType.kt

@@ -1,23 +1,13 @@
 package cc.unitmesh.devti.counit.model
 
-import cc.unitmesh.devti.counit.dto.PayloadType
-import com.google.gson.annotations.SerializedName
 import kotlinx.serialization.Serializable
 
 typealias Embedding = List<Float>
 
 @Serializable
 data class CodePayload(
-    val lang: String,
-    @SerializedName("repo_name") val repoName: String,
-    @SerializedName("repo_ref") val repoRef: String,
-    @SerializedName("payload_type") val payloadType: PayloadType,
-    @SerializedName("relative_path") val relativePath: String,
-    @SerializedName("content_hash") val contentHash: String,
-    @SerializedName("display_text") val displayText: String,
-    @SerializedName("origin_text") val originText: String,
-    @SerializedName("branches") val branches: List<String>,
     val id: String?,
+    val text: String?,
     val embedding: Embedding?,
     val score: Float?,
 )

src/main/kotlin/cc/unitmesh/devti/counit/dto/QAExample.kt

@@ -0,0 +1,45 @@
+package cc.unitmesh.devti.counit.model
+
+/**
+ * Enumeration of possible response actions.
+ *
+ * @property Direct Direct display result
+ * @property TextChunk Text splitting result
+ * @property Flow Will be handled by the client
+ */
+enum class ResponseAction {
+    /**
+     * Direct display result
+     */
+    Direct,
+
+    /**
+     * Text splitting result
+     */
+    TextChunk,
+
+    /**
+     * will be handled by the client
+     */
+    Flow
+}
+
+data class CustomFlowTransition(
+    /**
+     * will be JsonPath
+     */
+    val source: String,
+    /**
+     * will be JsonPath too
+     */
+    val target: String,
+)
+
+data class CustomRagApp(
+    val name: String,
+    val description: String = "",
+    val url: String = "",
+    val icon: String = "",
+    val responseAction: ResponseAction = ResponseAction.Direct,
+    val customFlowTransition: List<CustomFlowTransition> = emptyList(),
+)

src/main/kotlin/cc/unitmesh/devti/counit/model/CodePayload.kt

@@ -1,5 +1,7 @@
 package cc.unitmesh.devti.counit.model
 
+import kotlinx.serialization.Serializable
+
 /**
  * `Tooling` class represents a tooling configuration.
  *
@@ -12,5 +14,14 @@ data class Tooling(
     val name: String,
     val description: String,
     val schema: String,
-    val examples: List<String>
+    val examples: List<QAExample>,
+)
+
+/**
+ * `QAExample` is a data class that represents a question-answer pair.
+ */
+@Serializable
+data class QAExample(
+    val question: String,
+    val answer: String,
 )

src/main/kotlin/cc/unitmesh/devti/counit/model/CustomRagApp.kt

@@ -61,7 +61,6 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
             input.text = text
         }
 
-
     init {
         val presentation = Presentation(AutoDevBundle.message("chat.panel.send"))
         presentation.setIcon(AutoDevIcons.Send)