chore: Add JVM sample

Author: osugikoji

sample/jvm/README.md

@@ -56,4 +56,19 @@ This endpoint generates images based on the provided prompt.
 
 ##### Example:
 
-`GET http://localhost:8080/api/ychat/generations?prompt="ocean"
+`GET http://localhost:8080/api/ychat/generations?prompt="ocean"
+
+### Edits Endpoint
+
+This endpoint edits the prompt based on the provided instruction.
+
+##### Endpoint: http://localhost:[port_number]/api/ychat/edits
+
+##### Parameters:
+
+- `input`: The input text to use as a starting point for the edit.
+- `instruction`: The instruction that tells the model how to edit the prompt.
+
+##### Example:
+
+`GET http://localhost:8080/api/ychat/edits?input=What day of the wek is it?&instruction=Fix the spelling mistakes

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

@@ -40,6 +40,15 @@ public ResponseEntity<String> imageGenerations(
         return ResponseEntity.ok(result);
     }
 
+    @GetMapping("edits")
+    public ResponseEntity<String> edits(
+            @RequestParam(value = "input") String input,
+            @RequestParam(value = "instruction") String instruction
+    ) throws Exception {
+        String result = YChatService.getEditsAnswer(input, instruction);
+        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";

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

@@ -42,6 +42,14 @@ public String getImageGenerationsAnswer(String prompt) throws Exception {
         return future.get().get(0);
     }
 
+    public String getEditsAnswer(String input, String instruction) throws Exception {
+        final CompletableFuture<List<String>> future = new CompletableFuture<>();
+        ychat.edits()
+                .setInput(input)
+                .execute(instruction, new CompletionCallbackResult<>(future));
+        return future.get().get(0);
+    }
+
     private static class CompletionCallbackResult<T> implements YChat.Callback<T> {
 
         private final CompletableFuture<T> future;

ychat/src/commonTest/kotlin/infrastructure/MockStorage.kt

@@ -18,6 +18,6 @@ object MockStorage {
         "{\"created\":1678805561,\"data\":[{\"url\":\"$text\"}]}"
 
     fun editsSuccessResult(text: String) = "{\"object\":\"edit\",\"created\":1679072839," +
-            "\"choices\":[{\"text\":\"$text\",\"index\":0}]," +
-            "\"usage\":{\"prompt_tokens\":25,\"completion_tokens\":28,\"total_tokens\":53}}"
+        "\"choices\":[{\"text\":\"$text\",\"index\":0}]," +
+        "\"usage\":{\"prompt_tokens\":25,\"completion_tokens\":28,\"total_tokens\":53}}"
 }