137 lines
6.1 KiB
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();
|
|
}
|
|
}
|
|
}
|