zhouxy108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-07-21 15:09:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix comment

Browse Source
This commit is contained in:
Looly
2025-06-25 17:03:22 +08:00
parent 9d83869d85
commit bb607e30ae
21 changed files with 484 additions and 142 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
hutool-ai/src/main/java/cn/hutool/v7/ai/core/BaseAIService.java
View File

@@ -40,6 +40,9 @@ import java.util.function.Consumer;
*/
public class BaseAIService {
/**
* AI配置
*/
protected final AIConfig config;
/**
@@ -122,11 +125,11 @@ public class BaseAIService {
* @param paramMap 请求参数
* @param callback 流式数据回调函数
*/
protected void sendPostStream(final String endpoint, final Map<String, Object> paramMap, Consumer<String> callback) {
protected void sendPostStream(final String endpoint, final Map<String, Object> paramMap, final Consumer<String> callback) {
HttpURLConnection connection = null;
try {
// 创建连接
URL apiUrl = new URL(config.getApiUrl() + endpoint);
final URL apiUrl = new URL(config.getApiUrl() + endpoint);
connection = (HttpURLConnection) apiUrl.openConnection();
connection.setRequestMethod(Method.POST.name());
connection.setRequestProperty(HeaderName.CONTENT_TYPE.getValue(), "application/json");
@@ -137,21 +140,21 @@ public class BaseAIService {
//设置连接超时
connection.setConnectTimeout(config.getTimeout());
// 发送请求体
try (OutputStream os = connection.getOutputStream()) {
String jsonInputString = JSONUtil.toJsonStr(paramMap);
try (final OutputStream os = connection.getOutputStream()) {
final String jsonInputString = JSONUtil.toJsonStr(paramMap);
os.write(jsonInputString.getBytes());
os.flush();
}
// 读取流式响应
try (BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getInputStream()))) {
try (final BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getInputStream()))) {
String line;
while ((line = reader.readLine()) != null) {
// 调用回调函数处理每一行数据
callback.accept(line);
}
}
} catch (Exception e) {
} catch (final Exception e) {
callback.accept("{\"error\": \"" + e.getMessage() + "\"}");
} finally {
// 关闭连接

24
hutool-ai/src/main/java/cn/hutool/v7/ai/core/BaseConfig.java
View File

@@ -27,17 +27,29 @@ import java.util.concurrent.ConcurrentHashMap;
*/
public class BaseConfig implements AIConfig {
//apiKey
/**
* API Key
*/
protected volatile String apiKey;
//API请求地址
/**
* API请求地址
*/
protected volatile String apiUrl;
//具体模型
/**
* 模型名称
*/
protected volatile String model;
//动态扩展字段
/**
* 额外的配置
*/
protected final Map<String, Object> additionalConfig = new ConcurrentHashMap<>();
//连接超时时间
/**
* 请求超时
*/
protected volatile int timeout = 180000;
//读取超时时间
/**
* 读取超时
*/
protected volatile int readTimeout = 300000;
@Override

25
hutool-ai/src/main/java/cn/hutool/v7/ai/model/deepseek/DeepSeekConfig.java
View File

@@ -27,20 +27,37 @@ import cn.hutool.v7.ai.core.BaseConfig;
*/
public class DeepSeekConfig extends BaseConfig {
private final String API_URL = "https://api.deepseek.com";
/**
* 定义API的基础URL，用于后续的所有API请求
*/
public final String API_URL = "https://api.deepseek.com";
private final String DEFAULT_MODEL = Models.DeepSeek.DEEPSEEK_CHAT.getModel();
/**
* 定义默认的模型名称，用于在没有指定模型时使用
*/
public final String DEFAULT_MODEL = Models.DeepSeek.DEEPSEEK_CHAT.getModel();
/**
* 默认构造函数，用于初始化DeepSeek配置对象
* 设置API的基础URL和默认的模型名称
*/
public DeepSeekConfig() {
setApiUrl(API_URL);
setModel(DEFAULT_MODEL);
}
public DeepSeekConfig(String apiKey) {
this();
/**
* 带API密钥参数的构造函数
* 用于初始化DeepSeek配置对象，并设置API密钥
*
* @param apiKey 用户的API密钥，用于认证和授权
*/
public DeepSeekConfig(final String apiKey) {
this(); // 调用默认构造函数初始化API_URL和DEFAULT_MODEL
setApiKey(apiKey);
}
@Override
public String getModelName() {
return "deepSeek";

91
hutool-ai/src/main/java/cn/hutool/v7/ai/model/doubao/DoubaoCommon.java
View File

@@ -24,70 +24,139 @@ package cn.hutool.v7.ai.model.doubao;
*/
public class DoubaoCommon {
//doubao上下文缓存参数
/**
* doubao上下文缓存参数
*/
public enum DoubaoContext {
/**
* session
*/
SESSION("session"),
/**
* common_prefix
*/
COMMON_PREFIX("common_prefix");
private final String mode;
DoubaoContext(String mode) {
DoubaoContext(final String mode) {
this.mode = mode;
}
/**
* 获取参数
*
* @return 参数
*/
public String getMode() {
return mode;
}
}
//doubao视觉参数
/**
* doubao视觉参数
*/
public enum DoubaoVision {
/**
* 自动
*/
AUTO("auto"),
/**
* 低
*/
LOW("low"),
/**
* 高
*/
HIGH("high");
private final String detail;
DoubaoVision(String detail) {
DoubaoVision(final String detail) {
this.detail = detail;
}
/**
* 获取参数
*
* @return 参数
*/
public String getDetail() {
return detail;
}
}
//doubao视频生成参数
/**
* doubao视频生成参数
*/
public enum DoubaoVideo {
//宽高比例
/**
* 视频比例16:9，适用于横向宽屏显示，常用作标准视频比例
*/
RATIO_16_9("--rt", "16:9"),//[1280, 720]
/**
* 视频比例4:3，传统电视屏幕比例，适用于标准清晰度的视频内容
*/
RATIO_4_3("--rt", "4:3"),//[960, 720]
/**
* 视频比例1:1，正方形画面，适用于社交媒体平台上的短视频内容
*/
RATIO_1_1("--rt", "1:1"),//[720, 720]
/**
* 视频比例3:4，竖向视频比例，适用于手机端的视频播放场景
*/
RATIO_3_4("--rt", "3:4"),//[720, 960]
/**
* 视频比例9:16，常见的竖屏视频比例，广泛用于短视频应用
*/
RATIO_9_16("--rt", "9:16"),//[720, 1280]
/**
* 视频比例21:9，超宽屏幕比例，提供更广阔的视野，适合电影和游戏体验
*/
RATIO_21_9("--rt", "21:9"),//[1280, 544]
//生成视频时长
DURATION_5("--dur", 5),//文生视频，图生视频
DURATION_10("--dur", 10),//文生视频
/**
* 文生视频，图生视频
*/
DURATION_5("--dur", 5),
/**
* 文生视频
*/
DURATION_10("--dur", 10),
//帧率，即一秒时间内视频画面数量
/**
* 帧率，即一秒时间内视频画面数量
*/
FPS_5("--fps", 24),
//视频分辨率
/**
* 视频分辨率
*/
RESOLUTION_5("--rs", "720p"),
//生成视频是否包含水印
/**
* 生成视频包含水印
*/
WATERMARK_TRUE("--wm", true),
/**
* 生成视频不包含水印
*/
WATERMARK_FALSE("--wm", false);
private final String type;
private final Object value;
DoubaoVideo(String type, Object value) {
DoubaoVideo(final String type, final Object value) {
this.type = type;
this.value = value;
}

23
hutool-ai/src/main/java/cn/hutool/v7/ai/model/doubao/DoubaoConfig.java
View File

@@ -27,20 +27,33 @@ import cn.hutool.v7.ai.core.BaseConfig;
*/
public class DoubaoConfig extends BaseConfig {
private final String API_URL = "https://ark.cn-beijing.volces.com/api/v3";
// 定义API的基础URL，用于和服务器通信
private static final String API_URL = "https://ark.cn-beijing.volces.com/api/v3";
private final String DEFAULT_MODEL = Models.Doubao.DOUBAO_1_5_LITE_32K.getModel();
// 定义默认的模型配置，用于初始化配置对象时设定
private static final String DEFAULT_MODEL = Models.Doubao.DOUBAO_1_5_LITE_32K.getModel();
/**
* 无参构造函数，用于创建DoubaoConfig对象
* 初始化时会设置API_URL和DEFAULT_MODEL
*/
public DoubaoConfig() {
setApiUrl(API_URL);
setModel(DEFAULT_MODEL);
}
public DoubaoConfig(String apiKey) {
this();
setApiKey(apiKey);
/**
* 带有apiKey参数的构造函数，用于创建DoubaoConfig对象并设置API密钥
* 初始化时会设置API_URL、DEFAULT_MODEL以及传入的apiKey
*
* @param apiKey 用户的API密钥，用于验证用户身份
*/
public DoubaoConfig(final String apiKey) {
this(); // 先调用无参构造函数初始化API_URL和DEFAULT_MODEL
setApiKey(apiKey); // 设置用户的API密钥
}
@Override
public String getModelName() {
return "doubao";

144
hutool-ai/src/main/java/cn/hutool/v7/ai/model/doubao/DoubaoServiceImpl.java
View File

@@ -39,25 +39,25 @@ import java.util.function.Consumer;
public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
//对话
private final String CHAT_ENDPOINT = "/chat/completions";
private static final String CHAT_ENDPOINT = "/chat/completions";
//文本向量化
private final String EMBEDDING_TEXT = "/embeddings";
private static final String EMBEDDING_TEXT = "/embeddings";
//图文向量化
private final String EMBEDDING_VISION = "/embeddings/multimodal";
private static final String EMBEDDING_VISION = "/embeddings/multimodal";
//应用bots
private final String BOTS_CHAT = "/bots/chat/completions";
private static final String BOTS_CHAT = "/bots/chat/completions";
//分词
private final String TOKENIZATION = "/tokenization";
private static final String TOKENIZATION = "/tokenization";
//批量推理chat
private final String BATCH_CHAT = "/batch/chat/completions";
private static final String BATCH_CHAT = "/batch/chat/completions";
//创建上下文缓存
private final String CREATE_CONTEXT = "/context/create";
private static final String CREATE_CONTEXT = "/context/create";
//上下文缓存对话
private final String CHAT_CONTEXT = "/context/chat/completions";
private static final String CHAT_CONTEXT = "/context/chat/completions";
//创建视频生成任务
private final String CREATE_VIDEO = "/contents/generations/tasks";
private static final String CREATE_VIDEO = "/contents/generations/tasks";
//文生图
private final String IMAGES_GENERATIONS = "/images/generations";
private static final String IMAGES_GENERATIONS = "/images/generations";
public DoubaoServiceImpl(final AIConfig config) {
//初始化doubao客户端
@@ -66,109 +66,109 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
@Override
public String chat(final List<Message> messages) {
String paramJson = buildChatRequestBody(messages);
Response response = sendPost(CHAT_ENDPOINT, paramJson);
final String paramJson = buildChatRequestBody(messages);
final Response response = sendPost(CHAT_ENDPOINT, paramJson);
return response.bodyStr();
}
@Override
public void chat(List<Message> messages, Consumer<String> callback) {
Map<String, Object> paramMap = buildChatStreamRequestBody(messages);
public void chat(final List<Message> messages, final Consumer<String> callback) {
final Map<String, Object> paramMap = buildChatStreamRequestBody(messages);
ThreadUtil.newThread(() -> sendPostStream(CHAT_ENDPOINT, paramMap, callback::accept), "doubao-chat-sse").start();
}
@Override
public String chatVision(String prompt, final List<String> images, String detail) {
String paramJson = buildChatVisionRequestBody(prompt, images, detail);
Response response = sendPost(CHAT_ENDPOINT, paramJson);
public String chatVision(final String prompt, final List<String> images, final String detail) {
final String paramJson = buildChatVisionRequestBody(prompt, images, detail);
final Response response = sendPost(CHAT_ENDPOINT, paramJson);
return response.bodyStr();
}
@Override
public void chatVision(String prompt, List<String> images, String detail, Consumer<String> callback) {
Map<String, Object> paramMap = buildChatVisionStreamRequestBody(prompt, images, detail);
public void chatVision(final String prompt, final List<String> images, final String detail, final Consumer<String> callback) {
final Map<String, Object> paramMap = buildChatVisionStreamRequestBody(prompt, images, detail);
ThreadUtil.newThread(() -> sendPostStream(CHAT_ENDPOINT, paramMap, callback::accept), "doubao-chatVision-sse").start();
}
@Override
public String videoTasks(String text, String image, final List<DoubaoCommon.DoubaoVideo> videoParams) {
String paramJson = buildGenerationsTasksRequestBody(text, image, videoParams);
Response response = sendPost(CREATE_VIDEO, paramJson);
public String videoTasks(final String text, final String image, final List<DoubaoCommon.DoubaoVideo> videoParams) {
final String paramJson = buildGenerationsTasksRequestBody(text, image, videoParams);
final Response response = sendPost(CREATE_VIDEO, paramJson);
return response.bodyStr();
}
@Override
public String getVideoTasksInfo(String taskId) {
Response response = sendGet(CREATE_VIDEO + "/" + taskId);
public String getVideoTasksInfo(final String taskId) {
final Response response = sendGet(CREATE_VIDEO + "/" + taskId);
return response.bodyStr();
}
@Override
public String embeddingText(String[] input) {
String paramJson = buildEmbeddingTextRequestBody(input);
Response response = sendPost(EMBEDDING_TEXT, paramJson);
public String embeddingText(final String[] input) {
final String paramJson = buildEmbeddingTextRequestBody(input);
final Response response = sendPost(EMBEDDING_TEXT, paramJson);
return response.bodyStr();
}
@Override
public String embeddingVision(String text, String image) {
String paramJson = buildEmbeddingVisionRequestBody(text, image);
Response response = sendPost(EMBEDDING_VISION, paramJson);
public String embeddingVision(final String text, final String image) {
final String paramJson = buildEmbeddingVisionRequestBody(text, image);
final Response response = sendPost(EMBEDDING_VISION, paramJson);
return response.bodyStr();
}
@Override
public String botsChat(final List<Message> messages) {
String paramJson = buildBotsChatRequestBody(messages);
Response response = sendPost(BOTS_CHAT, paramJson);
final String paramJson = buildBotsChatRequestBody(messages);
final Response response = sendPost(BOTS_CHAT, paramJson);
return response.bodyStr();
}
@Override
public void botsChat(List<Message> messages, Consumer<String> callback) {
Map<String, Object> paramMap = buildBotsChatStreamRequestBody(messages);
public void botsChat(final List<Message> messages, final Consumer<String> callback) {
final Map<String, Object> paramMap = buildBotsChatStreamRequestBody(messages);
ThreadUtil.newThread(() -> sendPostStream(BOTS_CHAT, paramMap, callback::accept), "doubao-botsChat-sse").start();
}
@Override
public String tokenization(String[] text) {
String paramJson = buildTokenizationRequestBody(text);
Response response = sendPost(TOKENIZATION, paramJson);
public String tokenization(final String[] text) {
final String paramJson = buildTokenizationRequestBody(text);
final Response response = sendPost(TOKENIZATION, paramJson);
return response.bodyStr();
}
@Override
public String batchChat(final List<Message> messages) {
String paramJson = buildBatchChatRequestBody(messages);
Response response = sendPost(BATCH_CHAT, paramJson);
final String paramJson = buildBatchChatRequestBody(messages);
final Response response = sendPost(BATCH_CHAT, paramJson);
return response.bodyStr();
}
@Override
public String createContext(final List<Message> messages, String mode) {
String paramJson = buildCreateContextRequest(messages, mode);
Response response = sendPost(CREATE_CONTEXT, paramJson);
public String createContext(final List<Message> messages, final String mode) {
final String paramJson = buildCreateContextRequest(messages, mode);
final Response response = sendPost(CREATE_CONTEXT, paramJson);
return response.bodyStr();
}
@Override
public String chatContext(final List<Message> messages, String contextId) {
String paramJson = buildChatContentRequestBody(messages, contextId);
Response response = sendPost(CHAT_CONTEXT, paramJson);
public String chatContext(final List<Message> messages, final String contextId) {
final String paramJson = buildChatContentRequestBody(messages, contextId);
final Response response = sendPost(CHAT_CONTEXT, paramJson);
return response.bodyStr();
}
@Override
public void chatContext(List<Message> messages, String contextId, Consumer<String> callback) {
Map<String, Object> paramMap = buildChatContentStreamRequestBody(messages, contextId);
public void chatContext(final List<Message> messages, final String contextId, final Consumer<String> callback) {
final Map<String, Object> paramMap = buildChatContentStreamRequestBody(messages, contextId);
ThreadUtil.newThread(() -> sendPostStream(CHAT_CONTEXT, paramMap, callback::accept), "doubao-chatContext-sse").start();
}
@Override
public String imagesGenerations(String prompt) {
String paramJson = buildImagesGenerationsRequestBody(prompt);
Response response = sendPost(IMAGES_GENERATIONS, paramJson);
public String imagesGenerations(final String prompt) {
final String paramJson = buildImagesGenerationsRequestBody(prompt);
final Response response = sendPost(IMAGES_GENERATIONS, paramJson);
return response.bodyStr();
}
@@ -198,7 +198,7 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
}
//构建chatVision请求体
private String buildChatVisionRequestBody(String prompt, final List<String> images, String detail) {
private String buildChatVisionRequestBody(final String prompt, final List<String> images, final String detail) {
// 定义消息结构
final List<Message> messages = new ArrayList<>();
final List<Object> content = new ArrayList<>();
@@ -207,10 +207,10 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
contentMap.put("type", "text");
contentMap.put("text", prompt);
content.add(contentMap);
for (String img : images) {
HashMap<String, Object> imgUrlMap = new HashMap<>();
for (final String img : images) {
final HashMap<String, Object> imgUrlMap = new HashMap<>();
imgUrlMap.put("type", "image_url");
HashMap<String, String> urlMap = new HashMap<>();
final HashMap<String, String> urlMap = new HashMap<>();
urlMap.put("url", img);
urlMap.put("detail", detail);
imgUrlMap.put("image_url", urlMap);
@@ -228,7 +228,7 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
return JSONUtil.toJsonStr(paramMap);
}
private Map<String, Object> buildChatVisionStreamRequestBody(String prompt, final List<String> images, String detail) {
private Map<String, Object> buildChatVisionStreamRequestBody(final String prompt, final List<String> images, final String detail) {
// 定义消息结构
final List<Message> messages = new ArrayList<>();
final List<Object> content = new ArrayList<>();
@@ -237,10 +237,10 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
contentMap.put("type", "text");
contentMap.put("text", prompt);
content.add(contentMap);
for (String img : images) {
HashMap<String, Object> imgUrlMap = new HashMap<>();
for (final String img : images) {
final HashMap<String, Object> imgUrlMap = new HashMap<>();
imgUrlMap.put("type", "image_url");
HashMap<String, String> urlMap = new HashMap<>();
final HashMap<String, String> urlMap = new HashMap<>();
urlMap.put("url", img);
urlMap.put("detail", detail);
imgUrlMap.put("image_url", urlMap);
@@ -260,7 +260,7 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
}
//构建文本向量化请求体
private String buildEmbeddingTextRequestBody(String[] input) {
private String buildEmbeddingTextRequestBody(final String[] input) {
//使用JSON工具
final Map<String, Object> paramMap = new HashMap<>();
paramMap.put("model", config.getModel());
@@ -271,7 +271,7 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
}
//构建图文向量化请求体
private String buildEmbeddingVisionRequestBody(String text, String image) {
private String buildEmbeddingVisionRequestBody(final String text, final String image) {
//使用JSON工具
final Map<String, Object> paramMap = new HashMap<>();
paramMap.put("model", config.getModel());
@@ -311,7 +311,7 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
}
//构建分词请求体
private String buildTokenizationRequestBody(String[] text) {
private String buildTokenizationRequestBody(final String[] text) {
final Map<String, Object> paramMap = new HashMap<>();
paramMap.put("model", config.getModel());
paramMap.put("text", text);
@@ -328,7 +328,7 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
}
//构建创建上下文缓存请求体
private String buildCreateContextRequest(final List<Message> messages, String mode) {
private String buildCreateContextRequest(final List<Message> messages, final String mode) {
final Map<String, Object> paramMap = new HashMap<>();
paramMap.put("messages", messages);
paramMap.put("model", config.getModel());
@@ -340,7 +340,7 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
}
//构建上下文缓存对话请求体
private String buildChatContentRequestBody(final List<Message> messages, String contextId) {
private String buildChatContentRequestBody(final List<Message> messages, final String contextId) {
//使用JSON工具
final Map<String, Object> paramMap = new HashMap<>();
paramMap.put("model", config.getModel());
@@ -352,7 +352,7 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
return JSONUtil.toJsonStr(paramMap);
}
private Map<String, Object> buildChatContentStreamRequestBody(final List<Message> messages, String contextId) {
private Map<String, Object> buildChatContentStreamRequestBody(final List<Message> messages, final String contextId) {
//使用JSON工具
final Map<String, Object> paramMap = new HashMap<>();
paramMap.put("stream", true);
@@ -366,7 +366,7 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
}
//构建创建视频任务请求体
private String buildGenerationsTasksRequestBody(String text, String image, final List<DoubaoCommon.DoubaoVideo> videoParams) {
private String buildGenerationsTasksRequestBody(final String text, final String image, final List<DoubaoCommon.DoubaoVideo> videoParams) {
//使用JSON工具
final Map<String, Object> paramMap = new HashMap<>();
paramMap.put("model", config.getModel());
@@ -392,10 +392,10 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
//添加视频参数
if (videoParams != null && !videoParams.isEmpty()) {
//如果有文本参数就加在后面
if (textMap != null && !textMap.isEmpty()) {
int textIndex = content.indexOf(textMap);
StringBuilder textBuilder = new StringBuilder(text);
for (DoubaoCommon.DoubaoVideo videoParam : videoParams) {
if (!textMap.isEmpty()) {
final int textIndex = content.indexOf(textMap);
final StringBuilder textBuilder = new StringBuilder(text);
for (final DoubaoCommon.DoubaoVideo videoParam : videoParams) {
textBuilder.append(" ").append(videoParam.getType()).append(" ").append(videoParam.getValue());
}
textMap.put("type", "text");
@@ -408,8 +408,8 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
}
} else {
//如果没有文本参数就重新增加
StringBuilder textBuilder = new StringBuilder();
for (DoubaoCommon.DoubaoVideo videoParam : videoParams) {
final StringBuilder textBuilder = new StringBuilder();
for (final DoubaoCommon.DoubaoVideo videoParam : videoParams) {
textBuilder.append(videoParam.getType()).append(videoParam.getValue()).append(" ");
}
textMap.put("type", "text");
@@ -426,7 +426,7 @@ public class DoubaoServiceImpl extends BaseAIService implements DoubaoService {
}
//构建文生图请求体
private String buildImagesGenerationsRequestBody(String prompt) {
private String buildImagesGenerationsRequestBody(final String prompt) {
final Map<String, Object> paramMap = new HashMap<>();
paramMap.put("model", config.getModel());
paramMap.put("prompt", prompt);