docs: Add an example in the JVM sample

Author: osugikoji

sample/jvm/README.md

@@ -23,8 +23,23 @@ This endpoint generates text based on the provided prompt.
 
 ##### Parameters:
 
-- input: The prompt for generating text.
+- `input`: The prompt for generating text.
 
 ##### Example:
 
 `GET http://localhost:8080/api/ychat/completion?input="What is 1 + 1?"`
+
+### Chat Completions Endpoint
+
+This endpoint generates text based on the provided prompt and a specified topic. The generated text will be related to the topic provided.
+
+##### Endpoint: http://localhost:[port_number]/api/ychat/chat-completions
+
+##### Parameters:
+
+- `input`: The prompt for generating text.
+- `topic`: The topic to limit the response to.
+
+##### Example:
+
+`GET http://localhost:8080/api/ychat/chat-completions?input="Tell me an exercise plan"&topic=fitness`

sample/jvm/src/main/java/co/yml/ychat/jvm/controller/YChatController.java

@@ -6,24 +6,35 @@
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
-import co.yml.ychat.jvm.services.CompletionService;
+import co.yml.ychat.jvm.services.YChatService;
 
 @RestController
 @RequestMapping("api/ychat")
 public class YChatController {
 
     @Autowired
-    private CompletionService completionService;
+    private YChatService YChatService;
 
     @GetMapping("completion")
     public ResponseEntity<String> completion(
             @RequestParam(value = "input", defaultValue = Defaults.COMPLETION_INPUT) String input
     ) throws Exception {
-        String result = completionService.getCompletionAnswer(input);
+        String result = YChatService.getCompletionAnswer(input);
+        return ResponseEntity.ok(result);
+    }
+
+    @GetMapping("chat-completions")
+    public ResponseEntity<String> chatCompletions(
+            @RequestParam(value = "input", defaultValue = Defaults.CHAT_COMPLETION_INPUT) String input,
+            @RequestParam(value = "topic", defaultValue = Defaults.CHAT_COMPLETION_TOPIC) String topic
+    ) throws Exception {
+        String result = YChatService.getChatCompletionsAnswer(input, topic);
         return ResponseEntity.ok(result);
     }
 
     private static class Defaults {
         static final String COMPLETION_INPUT = "Say this is a test.";
+        static final String CHAT_COMPLETION_INPUT = "Tell me one strength exercise";
+        static final String CHAT_COMPLETION_TOPIC = "fitness";
     }
 }

sample/jvm/src/main/java/co/yml/ychat/jvm/services/CompletionService.java

@@ -0,0 +1,56 @@
+package co.yml.ychat.jvm.services;
+
+import co.yml.ychat.domain.model.ChatMessage;
+import java.util.List;
+import org.jetbrains.annotations.NotNull;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+import java.util.concurrent.CompletableFuture;
+import co.yml.ychat.YChat;
+
+@Service
+public class YChatService {
+
+    @Autowired
+    private YChat ychat;
+
+    private static final int MAX_TOKENS = 512;
+
+    public String getCompletionAnswer(String input) throws Exception {
+        final CompletableFuture<String> future = new CompletableFuture<>();
+        ychat.completion()
+                .setMaxTokens(MAX_TOKENS)
+                .setInput(input)
+                .execute(new CompletionCallbackResult<>(future));
+        return future.get();
+    }
+
+    public String getChatCompletionsAnswer(String input, String topic) throws Exception {
+        final CompletableFuture<List<ChatMessage>> future = new CompletableFuture<>();
+        String content = "You are a helpful assistant the only answer questions related to " + topic;
+        ychat.chatCompletions()
+                .setMaxTokens(MAX_TOKENS)
+                .addMessage("system", content)
+                .execute(input, new CompletionCallbackResult<>(future));
+        return future.get().get(0).getContent();
+    }
+
+    private static class CompletionCallbackResult<T> implements YChat.Callback<T> {
+
+        private final CompletableFuture<T> future;
+
+        CompletionCallbackResult(CompletableFuture<T> future) {
+            this.future = future;
+        }
+
+        @Override
+        public void onSuccess(T result) {
+            future.complete(result);
+        }
+
+        @Override
+        public void onError(@NotNull Throwable throwable) {
+            future.completeExceptionally(throwable);
+        }
+    }
+}