Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.litongjava.openai.client.OpenAiClient Maven / Gradle / Ivy
package com.litongjava.openai.client;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import com.litongjava.openai.chat.ChatMessage;
import com.litongjava.openai.chat.ChatResponseVo;
import com.litongjava.openai.chat.OpenAiChatRequestVo;
import com.litongjava.openai.constants.OpenAiConstatns;
import com.litongjava.openai.constants.OpenAiModels;
import com.litongjava.openai.embedding.EmbeddingRequestVo;
import com.litongjava.openai.embedding.EmbeddingResponseVo;
import com.litongjava.tio.utils.environment.EnvUtils;
import com.litongjava.tio.utils.http.OkHttpClientPool;
import com.litongjava.tio.utils.json.JsonUtils;
import okhttp3.Callback;
import okhttp3.Headers;
import okhttp3.MediaType;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.RequestBody;
import okhttp3.Response;
public class OpenAiClient {
public static Response chatCompletions(String authorization, String bodyString) {
Map header = new HashMap<>();
header.put("authorization", "Bearer " + authorization);
return chatCompletions(header, bodyString);
}
public static Response chatCompletions(Map header, String bodyString) {
String serverUrl = EnvUtils.get("OPENAI_API_URL");
if (serverUrl == null) {
serverUrl = OpenAiConstatns.server_url;
}
serverUrl += "/v1";
return chatCompletions(serverUrl, header, bodyString);
}
public static Response chatCompletions(String bodyString) {
String authorization = EnvUtils.get("OPENAI_API_KEY");
return chatCompletions(authorization, bodyString);
}
public static void chatCompletions(String bodyString, Callback callback) {
Map header = new HashMap<>(1);
String authorization = EnvUtils.get("OPENAI_API_KEY");
header.put("authorization", "Bearer " + authorization);
chatCompletions(header, bodyString, callback);
}
public static void chatCompletions(Map header, String bodyString, Callback callback) {
String serverUrl = EnvUtils.get("OPENAI_API_URL");
if (serverUrl == null) {
serverUrl = OpenAiConstatns.server_url;
}
serverUrl += "/v1";
chatCompletions(serverUrl, header, bodyString, callback);
}
public static ChatResponseVo chatCompletionsWithRole(String role, String prompt) {
ChatMessage chatMessage = new ChatMessage(role, prompt);
return chatCompletions(OpenAiModels.gpt_4o_mini, chatMessage);
}
public static ChatResponseVo chatCompletionsByModel(String model, String role, String prompt) {
ChatMessage chatMessage = new ChatMessage(role, prompt);
return chatCompletions(model, chatMessage);
}
public static ChatResponseVo chatCompletions(String model, ChatMessage chatMessage) {
OpenAiChatRequestVo chatRequestVo = new OpenAiChatRequestVo();
chatRequestVo.setModel(model);
chatRequestVo.setStream(false);
List messages = new ArrayList<>();
messages.add(chatMessage);
chatRequestVo.setMessages(messages);
return chatCompletions(chatRequestVo);
}
public static ChatResponseVo chatCompletions(OpenAiChatRequestVo chatRequestVo) {
String json = JsonUtils.toJson(chatRequestVo);
ChatResponseVo respVo = null;
try (Response response = chatCompletions(json)) {
String bodyString = response.body().string();
if (response.isSuccessful()) {
respVo = JsonUtils.parse(bodyString, ChatResponseVo.class);
} else {
throw new RuntimeException(bodyString);
}
} catch (IOException e) {
throw new RuntimeException(e);
}
return respVo;
}
public static void chatCompletions(OpenAiChatRequestVo chatRequestVo, Callback callback) {
String json = JsonUtils.toJson(chatRequestVo);
chatCompletions(json, callback);
}
public static ChatResponseVo chatCompletions(String apiPerfixUrl, String apiKey, OpenAiChatRequestVo chatRequestVo) {
String json = JsonUtils.toJson(chatRequestVo);
ChatResponseVo respVo = null;
try (Response response = chatCompletions(apiPerfixUrl, apiKey, json)) {
String bodyString = response.body().string();
if (response.isSuccessful()) {
respVo = JsonUtils.parse(bodyString, ChatResponseVo.class);
} else {
throw new RuntimeException(bodyString);
}
} catch (IOException e) {
throw new RuntimeException(e);
}
return respVo;
}
public static Response chatCompletions(String apiPerfixUrl, String authorization, String bodyString) {
Map header = new HashMap<>(1);
header.put("authorization", "Bearer " + authorization);
return chatCompletions(apiPerfixUrl, header, bodyString);
}
public static void chatCompletions(String apiPerfixUrl, String authorization, String bodyString, Callback callback) {
Map header = new HashMap<>(1);
header.put("authorization", "Bearer " + authorization);
chatCompletions(apiPerfixUrl, header, bodyString, callback);
}
public static Response chatCompletions(String uri, Map requestHeaders, String bodyString) {
OkHttpClient httpClient = OkHttpClientPool.getHttpClient();
MediaType mediaType = MediaType.parse("application/json");
RequestBody body = RequestBody.create(mediaType, bodyString);
Headers headers = Headers.of(requestHeaders);
String url = uri + "/chat/completions";
Request request = new Request.Builder() //
.url(url) //
.method("POST", body).headers(headers) //
.build();
try {
return httpClient.newCall(request).execute();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
public static void chatCompletions(String serverUrl, String apiKey, OpenAiChatRequestVo chatRequestVo,
Callback callback) {
chatCompletions(serverUrl, apiKey, JsonUtils.toJson(chatRequestVo), callback);
}
public static void chatCompletions(String apiPrefixUrl, Map requestHeaders, String bodyString,
Callback callback) {
OkHttpClient httpClient = OkHttpClientPool.getHttpClient();
MediaType mediaType = MediaType.parse("application/json");
RequestBody body = RequestBody.create(mediaType, bodyString);
Headers headers = Headers.of(requestHeaders);
String url = apiPrefixUrl + "/chat/completions";
Request request = new Request.Builder() //
.url(url) //
.method("POST", body).headers(headers) //
.build();
httpClient.newCall(request).enqueue(callback);
}
public static Response embeddings(String bodyString) {
Map header = new HashMap<>(1);
String authorization = EnvUtils.get("OPENAI_API_KEY");
header.put("authorization", "Bearer " + authorization);
return embeddings(header, bodyString);
}
public static Response embeddings(Map requestHeaders, String bodyString) {
String serverUrl = EnvUtils.get("OPENAI_API_URL");
return embeddings(serverUrl, requestHeaders, bodyString);
}
public static Response embeddings(String serverUrl, Map requestHeaders, String bodyString) {
if (serverUrl == null) {
serverUrl = OpenAiConstatns.server_url;
}
OkHttpClient httpClient = OkHttpClientPool.getHttpClient();
MediaType mediaType = MediaType.parse("application/json");
RequestBody body = RequestBody.create(mediaType, bodyString);
Headers headers = Headers.of(requestHeaders);
Request request = new Request.Builder() //
.url(serverUrl + "/v1/embeddings") //
.method("POST", body).headers(headers) //
.build();
try {
return httpClient.newCall(request).execute();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
public static float[] embeddingArray(String input, String model) {
EmbeddingRequestVo embeddingRequestVo = new EmbeddingRequestVo(input, model);
String serverUrl = EnvUtils.get("OPENAI_API_URL");
return embeddings(serverUrl, embeddingRequestVo).getData().get(0).getEmbedding();
}
public static float[] embeddingArray(String serverUrl, String input, String model) {
EmbeddingRequestVo embeddingRequestVo = new EmbeddingRequestVo(input, model);
return embeddings(serverUrl, embeddingRequestVo).getData().get(0).getEmbedding();
}
public static float[] embeddingArray(String input) {
return embeddingArray(input, OpenAiModels.text_embedding_3_small);
}
public static float[] embeddingArrayByLargeModel(String input) {
return embeddingArray(input, OpenAiModels.text_embedding_3_large);
}
public static EmbeddingResponseVo embeddings(EmbeddingRequestVo embeddingRequestVo) {
String serverUrl = EnvUtils.get("OPENAI_API_URL");
return embeddings(serverUrl, embeddingRequestVo);
}
public static EmbeddingResponseVo embeddings(String serverUrl, Map header, EmbeddingRequestVo reoVo) {
EmbeddingResponseVo respVo = null;
try (Response response = embeddings(serverUrl, header, JsonUtils.toJson(reoVo))) {
String bodyString = response.body().string();
if (response.isSuccessful()) {
respVo = JsonUtils.parse(bodyString, EmbeddingResponseVo.class);
} else {
throw new RuntimeException("status:" + response.code() + ",sbody:" + bodyString);
}
} catch (IOException e) {
throw new RuntimeException(e);
}
return respVo;
}
public static EmbeddingResponseVo embeddings(String serverUrl, EmbeddingRequestVo reoVo) {
EmbeddingResponseVo respVo = null;
try (Response response = embeddings(serverUrl, JsonUtils.toJson(reoVo))) {
String bodyString = response.body().string();
if (response.isSuccessful()) {
respVo = JsonUtils.parse(bodyString, EmbeddingResponseVo.class);
} else {
throw new RuntimeException("status:" + response.code() + ",body:" + bodyString);
}
} catch (IOException e) {
throw new RuntimeException(e);
}
return respVo;
}
public static Response embeddings(String serverUrl, String json) {
Map header = new HashMap<>(1);
String authorization = EnvUtils.get("OPENAI_API_KEY");
header.put("authorization", "Bearer " + authorization);
return embeddings(serverUrl, header, json);
}
public static float[] embeddingArray(String serverUrl, Map header, String input, String model) {
EmbeddingRequestVo reqVo = new EmbeddingRequestVo(input, model);
return embeddingArray(serverUrl, header, reqVo);
}
public static float[] embeddingArray(String serverUrl, Map header, EmbeddingRequestVo reqVo) {
EmbeddingResponseVo embeddings = embeddings(serverUrl, header, reqVo);
return embeddings.getData().get(0).getEmbedding();
}
}
