Update test examples to use new API

Author: CJCrafter

src/test/java/JavaChatStreamTest.java

@@ -10,27 +10,31 @@ public class JavaChatStreamTest {
 
     public static void main(String[] args) {
         Scanner scan = new Scanner(System.in);
-        String key = Dotenv.load().get("OPENAI_TOKEN");
 
-        // Create the initial prompt, we will reuse it later.
-        String initialPrompt = "You are a customer support chat-bot. Write brief summaries of the user's questions so that agents can easily find the answer in a database.";
-        List<ChatMessage> messages = new ArrayList<>(List.of(new ChatMessage(ChatUser.SYSTEM, initialPrompt)));
-        ChatRequest request = new ChatRequest("gpt-3.5-turbo", messages);
+        // Prepare the ChatRequest
+        ChatMessage prompt = ChatMessage.toSystemMessage("Be as unhelpful as possible");
+        List<ChatMessage> messages = new ArrayList<>(List.of(prompt));
+        ChatRequest request = ChatRequest.builder()
+                .model("gpt-3.5-turbo")
+                .messages(messages).build();
+
+        // Load TOKEN from .env file
+        String key = Dotenv.load().get("OPENAI_TOKEN");
         OpenAI openai = new OpenAI(key);
 
-        while (true) {
-            System.out.println("Enter text below:\n\n");
-            String input = scan.nextLine();
+        // Ask the user for input
+        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));
-            openai.streamChatCompletion(request, message -> {
-                System.out.print(message.get(0).getDelta());
+        // Stream the response. Print out each 'delta' (new tokens)
+        messages.add(new ChatMessage(ChatUser.USER, input));
+        openai.streamChatCompletion(request, message -> {
+            System.out.print(message.get(0).getDelta());
 
-                if (message.get(0).getFinishReason() != null) {
-                    messages.add(message.get(0).getMessage());
-                }
-            });
-        }
+            // Once the message is complete, we should save the message to our
+            // conversation (In case you want to generate more responses).
+            if (message.get(0).getFinishReason() != null)
+                messages.add(message.get(0).getMessage());
+        });
     }
 }

src/test/java/JavaChatTest.java

@@ -11,24 +11,44 @@ public class JavaChatTest {
 
     public static void main(String[] args) throws IOException {
         Scanner scan = new Scanner(System.in);
-        String key = Dotenv.load().get("OPENAI_TOKEN");
 
-        // Create the initial prompt, we will reuse it later.
-        String initialPrompt = "You are a customer support chat-bot. Write brief summaries of the user's questions so that agents can easily find the answer in a database.";
-        List<ChatMessage> messages = new ArrayList<>(List.of(new ChatMessage(ChatUser.SYSTEM, initialPrompt)));
-        ChatRequest request = new ChatRequest("gpt-3.5-turbo", messages);
+        // This is the prompt that the bot will refer back to for every message.
+        ChatMessage prompt = ChatMessage.toSystemMessage("You are a customer support chat-bot. Write brief summaries of the user's questions so that agents can easily find the answer in a database.");
+
+        // Use a mutable (modifiable) list! Always! You should be reusing the
+        // ChatRequest variable, so in order for a conversation to continue
+        // you need to be able to modify the list.
+        List<ChatMessage> messages = new ArrayList<>(List.of(prompt));
+
+        // ChatRequest is the request we send to OpenAI API. You can modify the
+        // model, temperature, maxTokens, etc. This should be saved, so you can
+        // reuse it for a conversation.
+        ChatRequest request = ChatRequest.builder()
+                .model("gpt-3.5-turbo")
+                .messages(messages).build();
+
+        // Loads the API key from the .env file in the root directory.
+        String key = Dotenv.load().get("OPENAI_TOKEN");
         OpenAI openai = new OpenAI(key);
 
+        // The conversation lasts until the user quits the program
         while (true) {
+
+            // Prompt the user to enter a response
             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));
+            // Add the newest user message to the conversation
+            messages.add(ChatMessage.toUserMessage(input));
+
+            // Use the OpenAI API to generate a response to the current
+            // conversation. Print the resulting message.
             ChatResponse response = openai.createChatCompletion(request);
             System.out.println("\n" + response.get(0).getMessage().getContent());
 
-            // Save the generated message to the bot's conversational memory
+            // Save the generated message to the conversational memory. It is
+            // crucial to save this message, otherwise future requests will be
+            // confused that there was no response.
             messages.add(response.get(0).getMessage());
         }
     }

src/test/kotlin/KotlinChatStreamTest.kt

@@ -7,26 +7,28 @@ import java.util.*
 
 fun main(args: Array<String>) {
     val scan = Scanner(System.`in`)
-    val key = dotenv()["OPENAI_TOKEN"]
 
-    // Create the initial prompt, we will reuse it later.
-    val initialPrompt = "Follow the users instructions"
-    val messages = mutableListOf(initialPrompt.toSystemMessage())
-    val request = ChatRequest("gpt-3.5-turbo", messages)
+    // Prepare the ChatRequest
+    val prompt = "Be as unhelpful as possible"
+    val messages = mutableListOf(prompt.toSystemMessage())
+    val request = ChatRequest(model="gpt-3.5-turbo", messages=messages)
+
+    // Loads the API key from the .env file in the root directory.
+    val key = dotenv()["OPENAI_TOKEN"]
     val openai = OpenAI(key)
 
-    while (true) {
-        println("Enter text below:\n")
-        val input = scan.nextLine()
+    // Ask the user for input
+    println("Enter text below:\n")
+    val input = scan.nextLine()
 
-        // Generate a response, and print it to the user.
-        messages.add(input.toUserMessage())
-        openai.streamChatCompletionKotlin(request) {
-            print(choices[0].delta)
+    // Generate a response, and print it to the user.
+    messages.add(input.toUserMessage())
+    openai.streamChatCompletionKotlin(request) {
+        print(choices[0].delta)
 
-            if (choices[0].finishReason != null) {
-                messages.add(choices[0].message)
-            }
-        }
+        // Once the message is complete, we should save the message to our
+        // conversation (In case you want to generate more responses).
+        if (choices[0].finishReason != null)
+            messages.add(choices[0].message)
     }
 }

src/test/kotlin/KotlinChatTest.kt

@@ -7,24 +7,42 @@ import java.util.*
 
 fun main(args: Array<String>) {
     val scan = Scanner(System.`in`)
-    val key = dotenv()["OPENAI_TOKEN"]
 
-    // Create the initial prompt, we will reuse it later.
-    val initialPrompt = "You are a customer support chat-bot. Write brief summaries of the user's questions so that agents can easily find the answer in a database."
-    val messages = mutableListOf(initialPrompt.toSystemMessage())
-    val request = ChatRequest("gpt-3.5-turbo", messages)
+    // This is the prompt that the bot will refer back to for every message.
+    val prompt = "You are a customer support chat-bot. Write brief summaries of the user's questions so that agents can easily find the answer in a database."
+
+    // Use a mutable (modifiable) list! Always! You should be reusing the
+    // ChatRequest variable, so in order for a conversation to continue you
+    // need to be able to modify the list.
+    val messages = mutableListOf(prompt.toSystemMessage())
+
+    // ChatRequest is the request we send to OpenAI API. You can modify the
+    // model, temperature, maxTokens, etc. This should be saved, so you can
+    // reuse it for a conversation.
+    val request = ChatRequest(model="gpt-3.5-turbo", messages=messages)
+
+    // Loads the API key from the .env file in the root directory.
+    val key = dotenv()["OPENAI_TOKEN"]
     val openai = OpenAI(key)
 
+    // The conversation lasts until the user quits the program
     while (true) {
+
+        // Prompt the user to enter a response
         println("Enter text below:\n")
         val input = scan.nextLine()
 
-        // Generate a response, and print it to the user.
+        // Add the newest user message to the conversation
         messages.add(input.toUserMessage())
+
+        // Use the OpenAI API to generate a response to the current
+        // conversation. Print the resulting message.
         val response = openai.createChatCompletion(request)
         println("\n${response[0].message.content}\n")
 
-        // Save the generated message to the bot's conversational memory
+        // Save the generated message to the conversational memory. It is
+        // crucial to save this message, otherwise future requests will be
+        // confused that there was no response.
         messages.add(response[0].message)
     }
 }