wrap the json error with Error classes

Author: CJCrafter

build.gradle.kts

@@ -22,6 +22,7 @@ dependencies {
     implementation("com.google.code.gson:gson:2.10.1")
 
     testImplementation("io.github.cdimascio:dotenv-kotlin:6.4.1")
+    testImplementation("org.junit.jupiter:junit-jupiter:5.9.2")
 }
 
 java {
@@ -43,6 +44,10 @@ tasks {
     }
 }
 
+tasks.test {
+    useJUnitPlatform()
+}
+
 // Create javadocJar and sourcesJar tasks
 val javadocJar by tasks.registering(Jar::class) {
     archiveClassifier.set("javadoc")

src/main/kotlin/com/cjcrafter/openai/chat/ChatBot.kt

@@ -1,13 +1,11 @@
 package com.cjcrafter.openai.chat
 
+import com.cjcrafter.openai.exception.OpenAIError
 import com.google.gson.*
 import okhttp3.*
 import okhttp3.MediaType.Companion.toMediaType
-import okhttp3.OkHttpClient.Builder
 import okhttp3.RequestBody.Companion.toRequestBody
 import java.io.IOException
-import java.lang.IllegalArgumentException
-import java.util.concurrent.TimeUnit
 import java.util.function.Consumer
 
 /**
@@ -62,8 +60,6 @@ class ChatBot @JvmOverloads constructor(
      *
      * @param request The input information for ChatGPT.
      * @return The returned response.
-     * @throws IOException              If an IO Exception occurs.
-     * @throws IllegalArgumentException If the input arguments are invalid.
      */
     @Throws(IOException::class)
     fun generateResponse(request: ChatRequest): ChatResponse {
@@ -75,13 +71,14 @@ class ChatBot @JvmOverloads constructor(
         try {
             client.newCall(httpRequest).execute().use { response ->
 
-                // Servers respond to API calls with json blocks. Since raw JSON isn't
-                // very developer friendly, we wrap for easy data access.
                 rootObject = JsonParser.parseString(response.body!!.string()).asJsonObject
-                require(!rootObject!!.has("error")) { rootObject!!.get("error").asJsonObject["message"].asString }
+                if (rootObject!!.has("error"))
+                    throw OpenAIError.fromJson(rootObject!!["error"].asJsonObject)
+
                 return ChatResponse(rootObject!!)
             }
         } catch (ex: Throwable) {
+            println(rootObject)
             throw ex
         }
     }

src/main/kotlin/com/cjcrafter/openai/exception/InvalidRequestError.kt

@@ -0,0 +1,5 @@
+package com.cjcrafter.openai.exception
+
+import com.google.gson.JsonElement
+
+class InvalidRequestError(param: JsonElement?, code: String?, message: String) : OpenAIError(param, code, message)

src/main/kotlin/com/cjcrafter/openai/exception/OpenAIError.kt

@@ -0,0 +1,41 @@
+package com.cjcrafter.openai.exception
+
+import com.google.gson.JsonElement
+import com.google.gson.JsonObject
+
+// TODO docs
+abstract class OpenAIError : Exception {
+
+    val param: JsonElement?
+    val code: String?
+
+    constructor(param: JsonElement?, code: String?, message: String) : super(message) {
+        this.param = param
+        this.code = code
+    }
+    constructor(param: JsonElement?, code: String?, message: String, cause: Throwable?) : super(message, cause) {
+        this.param = param
+        this.code = code
+    }
+    constructor(param: JsonElement?, code: String?, cause: Throwable?) : super(cause) {
+        this.param = param
+        this.code = code
+    }
+
+    companion object {
+
+        @JvmStatic
+        fun fromJson(json: JsonObject) : OpenAIError {
+            val message = json["message"].asString
+            val type = json["type"].asString
+            val param = json["param"]
+            val code = json["code"].asString
+
+            // TODO add more error types
+            return when (json["type"].asString) {
+                "invalid_request_error" -> InvalidRequestError(param, code, message)
+                else -> UnknownError(param, code, message, type)
+            }
+        }
+    }
+}

src/main/kotlin/com/cjcrafter/openai/exception/OpenAIException.kt

@@ -0,0 +1,5 @@
+package com.cjcrafter.openai.exception
+
+import com.google.gson.JsonElement
+
+class UnknownError(param: JsonElement?, code: String?, message: String, val type: String?) : OpenAIError(param, code, message)

src/main/kotlin/com/cjcrafter/openai/exception/UnknownError.kt

@@ -0,0 +1,37 @@
+package com.cjcrafter.openai;
+
+import com.cjcrafter.openai.chat.*;
+import io.github.cdimascio.dotenv.Dotenv;
+import org.junit.jupiter.api.Test;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+public class ExceptionTests {
+
+    @Test
+    void test_invalidModel() {
+        String key = Dotenv.load().get("OPENAI_TOKEN");
+
+        String initialPrompt = "Just say hi";
+        List<ChatMessage> messages = new ArrayList<>(List.of(new ChatMessage(ChatUser.SYSTEM, initialPrompt)));
+        ChatRequest request = new ChatRequest("gpt-238974-invalid-model", messages);
+        ChatBot bot = new ChatBot(key);
+
+        assertThrows(Throwable.class, () -> bot.generateResponse(request));
+    }
+
+    @Test
+    void test_invalidToken() {
+        String key = "sk-Thisisaninvalidtoken";
+
+        String initialPrompt = "Just say hi";
+        List<ChatMessage> messages = new ArrayList<>(List.of(new ChatMessage(ChatUser.SYSTEM, initialPrompt)));
+        ChatRequest request = new ChatRequest("gpt-3.5-turbo", messages);
+        ChatBot bot = new ChatBot(key);
+
+        assertThrows(Throwable.class, () -> bot.generateResponse(request));
+    }
+}