refactor(counit): refactor code to use new QAExample DTO and remove unnecessary dependencies on CoUnitApi #51

Author: phodal

src/main/kotlin/cc/unitmesh/devti/counit/CoUnitPreProcessor.kt

@@ -1,8 +1,6 @@
 package cc.unitmesh.devti.counit
 
 import cc.unitmesh.devti.util.LLMCoroutineScope
-import cc.unitmesh.devti.counit.dto.ExplainQuery
-import cc.unitmesh.devti.counit.dto.QueryResult
 import cc.unitmesh.devti.gui.chat.ChatCodingPanel
 import cc.unitmesh.devti.gui.chat.ChatContext
 import cc.unitmesh.devti.gui.chat.ChatRole
@@ -15,7 +13,6 @@ import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.project.Project
 import kotlinx.coroutines.launch
 // keep this import
-import kotlinx.serialization.decodeFromString
 import kotlinx.serialization.json.Json
 
 const val CO_UNIT = "/counit"
@@ -56,20 +53,11 @@ class CoUnitPreProcessor(val project: Project) {
 
             llmProvider.appendLocalMessage(result, ChatRole.Assistant)
 
-            val explain = try {
-                json.decodeFromString<ExplainQuery>(extractJsonResponse(result))
-            } catch (e: Exception) {
-                LOG.error("parse result error: $e")
-                return@launch
-            }
-
             val searchTip = "search API by query and hypothetical document"
             llmProvider.appendLocalMessage(searchTip, ChatRole.User)
             ui.addMessage(searchTip, true, searchTip)
 
-            val queryResult = coUnitPromptGenerator.semanticQuery(explain)
-
-            val related = buildDocAsContext(queryResult)
+            val related = coUnitPromptGenerator.semanticQuery("") ?: ""
             if (related.isEmpty()) {
                 val noResultTip = "no related API found"
                 llmProvider.appendLocalMessage(noResultTip, ChatRole.Assistant)
@@ -85,32 +73,6 @@ class CoUnitPreProcessor(val project: Project) {
         }
     }
 
-    private fun buildDocAsContext(queryResult: QueryResult): String {
-        val sb = StringBuilder()
-        val normalDoc = queryResult.englishQuery
-        if (normalDoc.isNotEmpty()) {
-            sb.append("here is related API to origin query's result: \n```markdown\n")
-            sb.append(normalDoc[0].displayText)
-            sb.append("\n```\n")
-        }
-
-        val nature = queryResult.naturalLangQuery
-        if (nature.isNotEmpty()) {
-            sb.append("here is natural language query's result: \n```markdown\n")
-            sb.append(nature[0].displayText)
-            sb.append("\n```\n")
-        }
-
-        val hyde = queryResult.hypotheticalDocument
-        if (hyde.isNotEmpty()) {
-            sb.append("here is hypothetical document's result: \n```markdown\n")
-            sb.append(hyde[0].displayText)
-            sb.append("\n```\n")
-        }
-
-        return sb.toString()
-    }
-
     /**
      * This method is used to extract JSON response from a given string.
      * It removes the leading and trailing ````json` tags from the string,

src/main/kotlin/cc/unitmesh/devti/counit/CoUnitPromptGenerator.kt

@@ -1,44 +1,15 @@
 package cc.unitmesh.devti.counit
 
-import cc.unitmesh.devti.counit.client.CoUnitApi
-import cc.unitmesh.devti.counit.dto.ExplainQuery
-import cc.unitmesh.devti.counit.dto.PayloadType
-import cc.unitmesh.devti.counit.dto.QueryResponse
-import cc.unitmesh.devti.counit.dto.QueryResult
-import cc.unitmesh.devti.counit.configurable.customRagSettings
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.project.Project
-import retrofit2.Retrofit
-import retrofit2.converter.gson.GsonConverterFactory
 
 @Service(Service.Level.PROJECT)
 class CoUnitPromptGenerator(val project: Project) {
     fun findIntention(input: String): String? {
-        val service: CoUnitApi = coUnitApi()
-        val body = service.explainQuery(input).execute().body()
-        return body?.prompt
+        return null
     }
 
-    fun semanticQuery(query: ExplainQuery): QueryResult {
-        val service: CoUnitApi = coUnitApi()
-        val englishQuery: QueryResponse? = service.query(query.query, PayloadType.OpenApi).execute().body()
-        val hydeDoc: QueryResponse? = service.query(query.hypotheticalDocument, PayloadType.OpenApi).execute().body()
-        val naturalLangQuery: QueryResponse? = service.query(query.natureLangQuery, PayloadType.OpenApi).execute().body()
-
-        return QueryResult(
-            englishQuery?.data ?: emptyList(),
-            naturalLangQuery?.data ?: emptyList(),
-            hydeDoc?.data ?: emptyList()
-        )
-    }
-
-    private fun coUnitApi(): CoUnitApi {
-        val retrofit = Retrofit.Builder()
-            .baseUrl(project.customRagSettings.serverAddress)
-            .addConverterFactory(GsonConverterFactory.create())
-            .build()
-
-        val service: CoUnitApi = retrofit.create(CoUnitApi::class.java)
-        return service
+    fun semanticQuery(query: String): String? {
+        return null
     }
 }

src/main/kotlin/cc/unitmesh/devti/counit/client/CoUnitApi.kt

@@ -0,0 +1,9 @@
+package cc.unitmesh.devti.counit.dto
+
+import kotlinx.serialization.Serializable
+
+@Serializable
+data class QAExample(
+    val question: String,
+    val answer: String,
+)