refactor(llm): simplify request body creation in OpenAILikeProvider #60

This commit updates the `OpenAILikeProvider` by utilizing the `toRequestBody` extension function for simpler request body creation. This change removes the need for the `create` method and improves code readability.
phodal · Aug 26, 2024 · f2b0d63 · f2b0d63
1 parent da29cf4
commit f2b0d63
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/kotlin/com/phodal/shire/llm/OpenAILikeProvider.kt b/src/main/kotlin/com/phodal/shire/llm/OpenAILikeProvider.kt
@@ -18,6 +18,7 @@ import okhttp3.MediaType.Companion.toMediaTypeOrNull
 import okhttp3.OkHttpClient
 import okhttp3.Request
 import okhttp3.RequestBody
+import okhttp3.RequestBody.Companion.toRequestBody
 import java.time.Duration
 
 class OpenAILikeProvider : CustomSSEHandler(), LlmProvider {
@@ -89,7 +90,7 @@ class OpenAILikeProvider : CustomSSEHandler(), LlmProvider {
         val customRequest = CustomRequest(messages)
         val requestContent = customRequest.updateCustomFormat(requestFormat)
 
-        val body = RequestBody.create("application/json; charset=utf-8".toMediaTypeOrNull(), requestContent)
+        val body = requestContent.toRequestBody("application/json; charset=utf-8".toMediaTypeOrNull())
 
         val builder = Request.Builder()
         if (key.isNotEmpty()) {