make sure ChatUser values are lowercase

Author: CJCrafter

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

@@ -1,9 +1,6 @@
 package com.cjcrafter.openai.chat
 
-import com.google.gson.Gson
-import com.google.gson.GsonBuilder
-import com.google.gson.JsonObject
-import com.google.gson.JsonParser
+import com.google.gson.*
 import okhttp3.MediaType
 import okhttp3.MediaType.Companion.toMediaType
 import okhttp3.OkHttpClient
@@ -25,7 +22,9 @@ class ChatBot(private val apiKey: String) {
         .connectTimeout(30, TimeUnit.SECONDS)
         .readTimeout(30, TimeUnit.SECONDS).build()
     private val mediaType: MediaType = "application/json; charset=utf-8".toMediaType()
-    private val gson: Gson = GsonBuilder().create()
+    private val gson: Gson = GsonBuilder()
+        .registerTypeAdapter(ChatUser::class.java, JsonSerializer<ChatUser> { src, _, context -> context!!.serialize(src!!.name.lowercase())!! })
+        .create()
 
     /**
      * Blocks the current thread until OpenAI responds to https request. The
@@ -43,7 +42,7 @@ class ChatBot(private val apiKey: String) {
         val json = gson.toJson(request)
         val body: RequestBody = json.toRequestBody(mediaType)
         val httpRequest: Request = Request.Builder()
-            .url("https://api.openai.com/v11/chat/completions")
+            .url("https://api.openai.com/v1/chat/completions")
             .addHeader("Content-Type", "application/json")
             .addHeader("Authorization", "Bearer $apiKey")
             .post(body).build()

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

@@ -31,12 +31,5 @@ enum class ChatUser {
     /**
      * [ASSISTANT] is the AI that generates responses.
      */
-    ASSISTANT;
-
-    /**
-     * OpenAI's API expects the user to be lowercase.
-     */
-    override fun toString(): String {
-        return name.lowercase()
-    }
+    ASSISTANT
 }

src/test/java/JavaTest.java

@@ -1,6 +1,5 @@
 import com.cjcrafter.openai.chat.*;
 import io.github.cdimascio.dotenv.Dotenv;
-import okhttp3.MediaType;
 
 import java.io.IOException;
 import java.util.ArrayList;
@@ -19,18 +18,17 @@ public static void main(String[] args) throws IOException {
         ChatRequest request = new ChatRequest("gpt-3.5-turbo", messages);
         ChatBot bot = new ChatBot(key);
 
-        MediaType.
         while (true) {
             System.out.println("Enter text below:\n\n");
             String input = scan.nextLine();
 
             // Generate a response, and print it to the user.
             messages.add(new ChatMessage(ChatUser.USER, input));
             ChatResponse response = bot.generateResponse(request);
-            System.out.println("\n" + response.getChoices().get(0).getMessage().getContent());
+            System.out.println("\n" + response.get(0).getMessage().getContent());
 
             // Save the generated message to the bot's conversational memory
-            messages.add(response.getChoices().get(0).getMessage());
+            messages.add(response.get(0).getMessage());
         }
     }
 }

src/test/kotlin/KotlinTest.kt

@@ -22,9 +22,9 @@ fun main(args: Array<String>) {
         // Generate a response, and print it to the user.
         messages.add(input.toUserMessage())
         val response = bot.generateResponse(request)
-        println("\n${response.choices[0].message.content}\n")
+        println("\n${response[0].message.content}\n")
 
         // Save the generated message to the bot's conversational memory
-        messages.add(response.choices[0].message)
+        messages.add(response[0].message)
     }
 }