uncloseai.com/languages/java/openai/uncloseai.java

137 lines
6.1 KiB
Java

import com.openai.client.OpenAIClient;
import com.openai.client.okhttp.OpenAIOkHttpClient;
import com.openai.models.chat.completions.ChatCompletion;
import com.openai.models.chat.completions.ChatCompletionCreateParams;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import org.json.JSONArray;
import org.json.JSONObject;
public class uncloseai {
public static void main(String[] args) {
System.out.println("=== uncloseai. Java Client (Official OpenAI SDK) ===\n");
// Discover endpoints from environment variables
String modelEndpoint1 = System.getenv("MODEL_ENDPOINT_1");
String modelEndpoint2 = System.getenv("MODEL_ENDPOINT_2");
String ttsEndpoint1 = System.getenv("TTS_ENDPOINT_1");
if (modelEndpoint1 == null || modelEndpoint2 == null || ttsEndpoint1 == null) {
System.err.println("ERROR: No models discovered. Set environment variables:");
System.err.println(" MODEL_ENDPOINT_1, MODEL_ENDPOINT_2, TTS_ENDPOINT_1");
System.exit(1);
}
try {
HttpClient httpClient = HttpClient.newHttpClient();
// Discover models from endpoint 1
System.out.println("Discovering models from " + modelEndpoint1 + "...");
HttpRequest req1 = HttpRequest.newBuilder()
.uri(URI.create(modelEndpoint1 + "/models"))
.GET()
.build();
HttpResponse<String> res1 = httpClient.send(req1, HttpResponse.BodyHandlers.ofString());
JSONObject json1 = new JSONObject(res1.body());
JSONArray models1 = json1.getJSONArray("data");
String model1Id = models1.getJSONObject(0).getString("id");
System.out.println("Model 1: " + model1Id + "\n");
// Discover models from endpoint 2
System.out.println("Discovering models from " + modelEndpoint2 + "...");
HttpRequest req2 = HttpRequest.newBuilder()
.uri(URI.create(modelEndpoint2 + "/models"))
.GET()
.build();
HttpResponse<String> res2 = httpClient.send(req2, HttpResponse.BodyHandlers.ofString());
JSONObject json2 = new JSONObject(res2.body());
JSONArray models2 = json2.getJSONArray("data");
String model2Id = models2.getJSONObject(0).getString("id");
System.out.println("Model 2: " + model2Id + "\n");
// Non-streaming chat with Model 1
System.out.println("=== Non-Streaming Chat (Model 1) ===");
OpenAIClient client1 = OpenAIOkHttpClient.builder()
.apiKey("dummy-key")
.baseUrl(modelEndpoint1)
.build();
ChatCompletionCreateParams params1 = ChatCompletionCreateParams.builder()
.model(model1Id)
.addUserMessage("Give a Python Fizzbuzz solution in one line of code?")
.temperature(0.5)
.maxTokens(150L)
.build();
ChatCompletion response1 = client1.chat().completions().create(params1);
System.out.println("Response: " + response1.choices().get(0).message().content().get() + "\n");
// Streaming chat with Model 1
System.out.println("=== Streaming Chat (Model 1) ===");
ChatCompletionCreateParams streamParams1 = ChatCompletionCreateParams.builder()
.model(model1Id)
.addUserMessage("Explain quantum entanglement in one sentence.")
.temperature(0.5)
.maxTokens(150L)
.build();
System.out.print("Response: ");
client1.chat().completions().createStreaming(streamParams1).stream().forEach(chunk -> {
if (!chunk.choices().isEmpty() && chunk.choices().get(0).delta().content().isPresent()) {
System.out.print(chunk.choices().get(0).delta().content().get());
}
});
System.out.println("\n");
// Non-streaming chat with Model 2
System.out.println("=== Non-Streaming Chat (Model 2) ===");
OpenAIClient client2 = OpenAIOkHttpClient.builder()
.apiKey("dummy-key")
.baseUrl(modelEndpoint2)
.build();
ChatCompletionCreateParams params2 = ChatCompletionCreateParams.builder()
.model(model2Id)
.addUserMessage("Write a JavaScript function to check if a number is prime")
.temperature(0.5)
.maxTokens(150L)
.build();
ChatCompletion response2 = client2.chat().completions().create(params2);
System.out.println("Response: " + response2.choices().get(0).message().content().get() + "\n");
// Streaming chat with Model 2
System.out.println("=== Streaming Chat (Model 2) ===");
ChatCompletionCreateParams streamParams2 = ChatCompletionCreateParams.builder()
.model(model2Id)
.addUserMessage("Give a Python Fizzbuzz solution in one line of code?")
.temperature(0.5)
.maxTokens(150L)
.build();
System.out.print("Response: ");
client2.chat().completions().createStreaming(streamParams2).stream().forEach(chunk -> {
if (!chunk.choices().isEmpty() && chunk.choices().get(0).delta().content().isPresent()) {
System.out.print(chunk.choices().get(0).delta().content().get());
}
});
System.out.println("\n");
// TTS example
System.out.println("=== TTS Speech Generation ===");
System.out.println("[SKIPPED] TTS implementation needs audio API research\n");
System.out.println("=== Examples Complete ===");
} catch (IOException | InterruptedException e) {
System.err.println("Error: " + e.getMessage());
e.printStackTrace();
}
}
}