Added Image generation example (TheoKanning#87)

weifuzi0217 · Feb 13, 2023 · 32b4c98 · 32b4c98
1 parent 1a463b6
commit 32b4c98
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/example/src/main/java/example/OpenAiApiExample.java b/example/src/main/java/example/OpenAiApiExample.java
@@ -2,6 +2,7 @@
 
 import com.theokanning.openai.OpenAiService;
 import com.theokanning.openai.completion.CompletionRequest;
+import com.theokanning.openai.image.CreateImageRequest;
 
 class OpenAiApiExample {
     public static void main(String... args) {
@@ -16,5 +17,13 @@ public static void main(String... args) {
                 .user("testing")
                 .build();
         service.createCompletion(completionRequest).getChoices().forEach(System.out::println);
+
+        System.out.println("\nCreating Image...");
+        CreateImageRequest request = CreateImageRequest.builder()
+                .prompt("A cow breakdancing with a turtle")
+                .build();
+
+        System.out.println("\nImage is located at:");
+        System.out.println(service.createImage(request).getData().get(0).getUrl());
     }
 }