Update README with new coding example

Author: CJCrafter

README.md

@@ -23,40 +23,55 @@ See the [maven repository](https://central.sonatype.com/artifact/com.cjcrafter/o
 
 
 # Working Example
+Check the wiki for more examples.
 ```java
-import java.io.IOException;
-import java.util.List;
-import java.util.Scanner;
+public class JavaChatTest {
 
-public class Main {
-    public static void main(String[] args) throws IOException {
+    public static void main(String[] args) throws OpenAIError {
         Scanner scan = new Scanner(System.in);
-        String key = "sk-YOUR KEY HERE";  // TODO Add your open ai key here
- 
-        // 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<ChatBot.ChatMessage> messages = List.of(new ChatBot.ChatMessage("system", initialPrompt));
-        ChatBot.ChatCompletionRequest request = new ChatBot.ChatCompletionRequest("gpt-3.5-turbo", messages);
-        ChatBot bot = new ChatBot(key);
-
-        // ChatCompletionRequest copies the list, so let's modify the request's
-        // copy of the list.
-        messages = request.getMessages();
 
+        // 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 ChatBot.ChatMessage("user", input));
-            ChatBot.ChatCompletionResponse response = bot.generateResponse(request);
-            System.out.println("\n" + response.getChoices().get(0).getMessage().getContent());
+            // 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
-            messages.add(response.getChoices().get(0).getMessage());
+            // 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());
         }
     }
 }
+
 ```
 
 # Support