docs: async example updates

Author: TomerAberbach

openai-java-example/src/main/java/com/openai/example/AssistantAsyncExample.java

@@ -0,0 +1,113 @@
+package com.openai.example;
+
+import com.openai.client.OpenAIClientAsync;
+import com.openai.client.okhttp.OpenAIOkHttpClientAsync;
+import com.openai.models.*;
+import java.util.concurrent.CompletableFuture;
+
+public final class AssistantAsyncExample {
+    private AssistantAsyncExample() {}
+
+    public static void main(String[] args) {
+        // Configures using one of:
+        // - The `OPENAI_API_KEY` environment variable
+        // - The `AZURE_OPENAI_ENDPOINT` and `AZURE_OPENAI_KEY` environment variables
+        OpenAIClientAsync client = OpenAIOkHttpClientAsync.fromEnv();
+
+        CompletableFuture<Assistant> assistantFuture = client.beta()
+                .assistants()
+                .create(BetaAssistantCreateParams.builder()
+                        .name("Math Tutor")
+                        .instructions("You are a personal math tutor. Write and run code to answer math questions.")
+                        .addTool(CodeInterpreterTool.builder().build())
+                        .model(ChatModel.GPT_4O_MINI)
+                        .build());
+        CompletableFuture<String> threadIdFuture = client.beta()
+                .threads()
+                .create(BetaThreadCreateParams.builder().build())
+                .thenComposeAsync(thread -> client.beta()
+                        .threads()
+                        .messages()
+                        .create(BetaThreadMessageCreateParams.builder()
+                                .threadId(thread.id())
+                                .role(BetaThreadMessageCreateParams.Role.USER)
+                                .content("I need to solve the equation `3x + 11 = 14`. Can you help me?")
+                                .build()))
+                .thenApply(Message::threadId);
+
+        CompletableFuture<Run> runFuture = CompletableFuture.allOf(assistantFuture, threadIdFuture)
+                .thenComposeAsync(unused -> client.beta()
+                        .threads()
+                        .runs()
+                        .create(BetaThreadRunCreateParams.builder()
+                                .threadId(threadIdFuture.join())
+                                .assistantId(assistantFuture.join().id())
+                                .instructions("Please address the user as Jane Doe. The user has a premium account.")
+                                .build()));
+        CompletableFuture<Run> polledRunFuture = runFuture.thenComposeAsync(run -> pollRun(client, run));
+
+        polledRunFuture
+                .thenComposeAsync(run -> {
+                    if (!run.status().equals(RunStatus.COMPLETED)) {
+                        return CompletableFuture.completedFuture(null);
+                    }
+
+                    return listThreadMessages(client, run.threadId())
+                            .thenComposeAsync(unused -> client.beta()
+                                    .assistants()
+                                    .delete(BetaAssistantDeleteParams.builder()
+                                            .assistantId(assistantFuture.join().id())
+                                            .build()))
+                            .thenAccept(assistantDeleted ->
+                                    System.out.println("Assistant deleted: " + assistantDeleted.deleted()));
+                })
+                .join();
+    }
+
+    private static CompletableFuture<Run> pollRun(OpenAIClientAsync client, Run run) {
+        if (!run.status().equals(RunStatus.QUEUED) && !run.status().equals(RunStatus.IN_PROGRESS)) {
+            System.out.println("Run completed with status: " + run.status() + "\n");
+            return CompletableFuture.completedFuture(run);
+        }
+
+        System.out.println("Polling run...");
+        try {
+            java.lang.Thread.sleep(500);
+        } catch (InterruptedException e) {
+            throw new RuntimeException(e);
+        }
+
+        return client.beta()
+                .threads()
+                .runs()
+                .retrieve(BetaThreadRunRetrieveParams.builder()
+                        .threadId(run.threadId())
+                        .runId(run.id())
+                        .build())
+                .thenComposeAsync(newRun -> pollRun(client, newRun));
+    }
+
+    private static CompletableFuture<Void> listThreadMessages(OpenAIClientAsync client, String threadId) {
+        CompletableFuture<BetaThreadMessageListPageAsync> pageFuture = client.beta()
+                .threads()
+                .messages()
+                .list(BetaThreadMessageListParams.builder()
+                        .threadId(threadId)
+                        .order(BetaThreadMessageListParams.Order.ASC)
+                        .build());
+        return pageFuture.thenComposeAsync(page -> page.autoPager()
+                .forEach(
+                        currentMessage -> {
+                            System.out.println(currentMessage.role().toString().toUpperCase());
+                            currentMessage.content().stream()
+                                    .flatMap(content -> content.text().stream())
+                                    .forEach(textBlock ->
+                                            System.out.println(textBlock.text().value()));
+                            System.out.println();
+
+                            // Keep iterating
+                            return true;
+                        },
+                        pageFuture.defaultExecutor()));
+    }
+}

openai-java-example/src/main/java/com/openai/example/CompletionsConversationAsyncExample.java

@@ -7,8 +7,6 @@
 import com.openai.models.*;
 import java.util.List;
 import java.util.concurrent.CompletableFuture;
-import java.util.concurrent.ExecutorService;
-import java.util.concurrent.Executors;
 
 public final class CompletionsConversationAsyncExample {
     private CompletionsConversationAsyncExample() {}
@@ -31,13 +29,11 @@ public static void main(String[] args) {
                         .content("Tell me a story about building the best SDK!")
                         .build());
 
-        ExecutorService executor = Executors.newCachedThreadPool();
-
         CompletableFuture<Void> future = CompletableFuture.completedFuture(null);
         for (int i = 0; i < 4; i++) {
             final int index = i;
             future = future.thenComposeAsync(
-                            unused -> client.chat().completions().create(createParamsBuilder.build()), executor)
+                            unused -> client.chat().completions().create(createParamsBuilder.build()))
                     .thenAccept(completion -> {
                         List<ChatCompletionMessage> messages = completion.choices().stream()
                                 .map(ChatCompletion.Choice::message)
@@ -61,6 +57,5 @@ public static void main(String[] args) {
         }
 
         future.join();
-        executor.shutdown();
     }
 }