public/liteFlow
1
0
Fork 0
mirror of https://gitee.com/dromara/liteFlow.git synced 2026-06-11 10:36:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor: 将配置参数移动到 AIComponent 中，类型节点关注节点特有参数

Browse Source
This commit is contained in:
LuanY77
2025-07-23 13:54:55 +08:00
parent d7b6b20f23
commit 215458eade
3 changed files with 104 additions and 205 deletions
Download Patch File Download Diff File Expand all files Collapse all files

102
liteflow-ai/liteflow-ai-core/src/main/java/com/yomahub/liteflow/ai/annotation/AIChat.java
View File

@@ -5,9 +5,6 @@ import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import com.yomahub.liteflow.ai.util.KeyValue;
import com.yomahub.liteflow.ai.util.TriState;
/**
* AI Chat 节点注解
*
@@ -39,103 +36,4 @@ public @interface AIChat {
*/
boolean streaming() default false;
// --- 连接、鉴权参数 ---
/**
* API URL
*/
String baseUrl() default "";
/**
* 模型名
*/
String modelName() default "";
/**
* API KEY
*/
String apiKey() default "";
/**
* 版本
*/
String version() default "";
// --- 核心参数 ---
/**
* 温度参数
*/
double temperature() default -1.0;
/**
* Top P 参数
*/
double topP() default -1.0;
/**
* Top K 参数
*/
int topK() default -1;
/**
* 最大 Token 数量
* 统一：maxTokens, maxOutputTokens, numPredict
*/
int maxTokens() default -1;
/**
* 停止序列
* 统一: stop, stopSequences
*/
String[] stop() default {};
/**
* seed
*/
int seed() default -1;
// --- 惩罚参数 ---
/**
* 重复惩罚
*/
double repeatPenalty() default -1.0;
/**
* 存在惩罚
*/
double presencePenalty() default -1.0;
/**
* 频率惩罚
*/
double frequencyPenalty() default -1.0;
// --- Format ----
/**
* 响应格式
*/
String responseFormat() default "";
/**
* 是否严格遵循 JSON Schema
*/
TriState strictJsonSchema() default TriState.UNSET;
// --- Tool Calling ----
/**
* 是否并行 ToolCall
*/
TriState parallelToolCalls() default TriState.UNSET;
// --- 网络和日志参数 ---
/**
* 超时时间
*/
String timeout() default "";
/**
* 最大重试次数
*/
int maxRetries() default -1;
/**
* 是否记录请求日志
*/
TriState logRequests() default TriState.UNSET;
/**
* 是否记录响应日志
*/
TriState logResponses() default TriState.UNSET;
// --- 其他参数 ---
/**
* 自定义请求头
*/
KeyValue[] customHeaders() default {};
}

103
liteflow-ai/liteflow-ai-core/src/main/java/com/yomahub/liteflow/ai/annotation/AIClassify.java
View File

@@ -5,9 +5,6 @@ import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import com.yomahub.liteflow.ai.util.KeyValue;
import com.yomahub.liteflow.ai.util.TriState;
/**
* AI 分类节点注解
*
@@ -42,104 +39,4 @@ public @interface AIClassify {
* 是否多标签分类，默认 false
*/
boolean multiLabel() default false;
// --- 连接、鉴权参数 ---
/**
* API URL
*/
String baseUrl() default "";
/**
* 模型名
*/
String modelName() default "";
/**
* API KEY
*/
String apiKey() default "";
/**
* 版本
*/
String version() default "";
// --- 核心参数 ---
/**
* 温度参数
*/
double temperature() default -1.0;
/**
* Top P 参数
*/
double topP() default -1.0;
/**
* Top K 参数
*/
int topK() default -1;
/**
* 最大 Token 数量
* 统一：maxTokens, maxOutputTokens, numPredict
*/
int maxTokens() default -1;
/**
* 停止序列
* 统一: stop, stopSequences
*/
String[] stop() default {};
/**
* seed
*/
int seed() default -1;
// --- 惩罚参数 ---
/**
* 重复惩罚
*/
double repeatPenalty() default -1.0;
/**
* 存在惩罚
*/
double presencePenalty() default -1.0;
/**
* 频率惩罚
*/
double frequencyPenalty() default -1.0;
// --- Format ----
/**
* 响应格式
*/
String responseFormat() default "";
/**
* 是否严格遵循 JSON Schema
*/
TriState strictJsonSchema() default TriState.UNSET;
// --- Tool Calling ----
/**
* 是否并行 ToolCall
*/
TriState parallelToolCalls() default TriState.UNSET;
// --- 网络和日志参数 ---
/**
* 超时时间
*/
String timeout() default "";
/**
* 最大重试次数
*/
int maxRetries() default -1;
/**
* 是否记录请求日志
*/
TriState logRequests() default TriState.UNSET;
/**
* 是否记录响应日志
*/
TriState logResponses() default TriState.UNSET;
// --- 其他参数 ---
/**
* 自定义请求头
*/
KeyValue[] customHeaders() default {};
}

104
liteflow-ai/liteflow-ai-core/src/main/java/com/yomahub/liteflow/ai/annotation/AIComponent.java
View File

@@ -1,5 +1,8 @@
package com.yomahub.liteflow.ai.annotation;
import com.yomahub.liteflow.ai.util.KeyValue;
import com.yomahub.liteflow.ai.util.TriState;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
@@ -19,4 +22,105 @@ public @interface AIComponent {
String nodeId() default "";
String nodeName() default "";
// --- 连接、鉴权参数 ---
/**
* API URL
*/
String baseUrl() default "";
/**
* 模型名
*/
String model() default "";
/**
* API KEY
*/
String apiKey() default "";
/**
* 版本
*/
String version() default "";
// --- 核心参数 ---
/**
* 温度参数
*/
double temperature() default -1.0;
/**
* Top P 参数
*/
double topP() default -1.0;
/**
* Top K 参数
*/
int topK() default -1;
/**
* 最大 Token 数量
* 统一：maxTokens, maxOutputTokens, numPredict
*/
int maxTokens() default -1;
/**
* 停止序列
* 统一: stop, stopSequences
*/
String[] stop() default {};
/**
* seed
*/
int seed() default -1;
// --- 惩罚参数 ---
/**
* 重复惩罚
*/
double repeatPenalty() default -1.0;
/**
* 存在惩罚
*/
double presencePenalty() default -1.0;
/**
* 频率惩罚
*/
double frequencyPenalty() default -1.0;
// --- Format ----
/**
* 响应格式
*/
String responseFormat() default "";
/**
* 是否严格遵循 JSON Schema
*/
TriState strictJsonSchema() default TriState.UNSET;
// --- Tool Calling ----
/**
* 是否并行 ToolCall
*/
TriState parallelToolCalls() default TriState.UNSET;
// --- 网络和日志参数 ---
/**
* 超时时间
*/
String timeout() default "";
/**
* 最大重试次数
*/
int maxRetries() default -1;
/**
* 是否记录请求日志
*/
TriState logRequests() default TriState.UNSET;
/**
* 是否记录响应日志
*/
TriState logResponses() default TriState.UNSET;
// --- 其他参数 ---
/**
* 自定义请求头
*/
KeyValue[] customHeaders() default {};
}