fix(custom-agent): fix CustomAgentExecutor requestFormat issue #51

- Remove new line character at the end of the requestFormat string in CustomAgentExecutor.
- Set the responseFormat to the value from CustomAgentConfig if available.

Author: phodal

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

@@ -20,12 +20,12 @@ class CustomAgentExecutor(val project: Project) : CustomSSEProcessor() {
     private var client = OkHttpClient()
     private val logger = logger<CustomAgentExecutor>()
 
-    override var requestFormat: String = "{ \"messageKeys\": {\"role\": \"role\", \"content\": \"content\"} }\n"
+    override var requestFormat: String = "{ \"messageKeys\": {\"role\": \"role\", \"content\": \"content\"} }"
     override var responseFormat: String = "\$.choices[0].delta.content"
 
     fun execute(input: String, agent: CustomAgentConfig): Flow<String>? {
-        this.responseFormat = agent.connector?.responseFormat ?: this.responseFormat
         this.requestFormat = agent.connector?.requestFormat ?: this.requestFormat
+        this.responseFormat = agent.connector?.responseFormat ?: this.responseFormat
 
         val customRequest = CustomRequest(listOf(Message("user", input)))
         val request = customRequest.updateCustomFormat(requestFormat)

src/main/kotlin/cc/unitmesh/devti/llms/custom/JSONBodyResponseCallback.kt

@@ -17,6 +17,14 @@ class JSONBodyResponseCallback(private val responseFormat: String,private val ca
 
     override fun onResponse(call: Call, response: Response) {
         val responseBody: String? = response.body?.string()
+        if (responseFormat.isEmpty()) {
+            runBlocking {
+                callback(responseBody ?: "")
+            }
+
+            return
+        }
+
         val responseContent: String = JsonPath.parse(responseBody)?.read(responseFormat) ?: ""
 
         runBlocking() {