public/liteFlow
1
0
Fork 0
mirror of https://gitee.com/dromara/liteFlow.git synced 2026-06-12 18:21:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: coze workflow test

Browse Source
This commit is contained in:
LuanY77
2025-09-08 14:15:19 +08:00
parent 29c2141990
commit 85f8401c1e
14 changed files with 128 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
liteflow-ai/liteflow-ai-engine/src/main/java/com/yomahub/liteflow/ai/engine/interact/protocol/AbstractProtocolTransformer.java
View File

@@ -23,6 +23,7 @@ import java.util.Optional;
* @since TODO
*/
@Deprecated
public abstract class AbstractProtocolTransformer implements ProtocolTransformer {
private static final String FINISHED_DATA = "finished";

3
liteflow-ai/liteflow-ai-workflow/liteflow-ai-workflow-coze/src/main/java/com/yomahub/liteflow/ai/workflow/coze/handler/CozeWorkflowRunComponentHandler.java
View File

@@ -4,6 +4,7 @@ import com.yomahub.liteflow.ai.annotation.AIComponent;
import com.yomahub.liteflow.ai.proxy.handler.WorkflowComponentHandler;
import com.yomahub.liteflow.ai.proxy.wrap.AIProxyWrapBean;
import com.yomahub.liteflow.ai.workflow.coze.annotation.CozeWorkflowRun;
import com.yomahub.liteflow.ai.workflow.coze.invocation.CozeWorkflowRunInvocationHandler;
import com.yomahub.liteflow.ai.workflow.coze.wrap.CozeWorkflowRunProxyWrapBean;
import java.lang.reflect.InvocationHandler;
@@ -29,6 +30,6 @@ public class CozeWorkflowRunComponentHandler extends WorkflowComponentHandler<Co
@Override
protected InvocationHandler getInvocationHandler(AIProxyWrapBean<CozeWorkflowRun> wrapBean) {
return null;
return new CozeWorkflowRunInvocationHandler((CozeWorkflowRunProxyWrapBean) wrapBean);
}
}

2
liteflow-ai/liteflow-ai-workflow/liteflow-ai-workflow-coze/src/main/java/com/yomahub/liteflow/ai/workflow/coze/invocation/CozeWorkflowChatInvocationHandler.java
View File

@@ -44,7 +44,7 @@ public class CozeWorkflowChatInvocationHandler extends AbstractAIInvocationHandl
.auth(new TokenAuth(property.getApiKey()))
.baseURL(
StrUtil.isBlank(property.getBaseUrl()) ?
Consts.COZE_COM_BASE_URL :
Consts.COZE_CN_BASE_URL :
property.getBaseUrl()
)
.build();

7
liteflow-ai/liteflow-ai-workflow/liteflow-ai-workflow-coze/src/main/java/com/yomahub/liteflow/ai/workflow/coze/invocation/CozeWorkflowRunInvocationHandler.java
View File

@@ -45,7 +45,7 @@ public class CozeWorkflowRunInvocationHandler extends AbstractAIInvocationHandle
.auth(new TokenAuth(property.getApiKey()))
.baseURL(
StrUtil.isBlank(property.getBaseUrl()) ?
Consts.COZE_COM_BASE_URL :
Consts.COZE_CN_BASE_URL :
property.getBaseUrl()
)
.build();
@@ -94,4 +94,9 @@ public class CozeWorkflowRunInvocationHandler extends AbstractAIInvocationHandle
return runs.create(builder.build());
}
}
@Override
protected void checkValidation(ProcessorContext<?> processorContext) {
// 空实现
}
}

2
liteflow-ai/liteflow-ai-workflow/liteflow-ai-workflow-coze/src/main/resources/META-INF/additional-spring-configuration-metadata.json
View File

@@ -11,7 +11,7 @@
"type": "java.lang.String",
"description": "apiUrl for coze service.",
"sourceType": "com.yomahub.liteflow.ai.workflow.coze.config.CozeWorkflowProperty",
"defaultValue": "https://api.coze.com"
"defaultValue": "https://api.coze.cn"
}
]
}

5
liteflow-testcase-el/liteflow-testcase-el-ai/pom.xml
View File

@@ -45,6 +45,11 @@
<artifactId>liteflow-ai-workflow-dashscope</artifactId>
<version>${revision}</version>
</dependency>
<dependency>
<groupId>com.yomahub</groupId>
<artifactId>liteflow-ai-workflow-coze</artifactId>
<version>${revision}</version>
</dependency>
<!-- model end -->
</dependencies>

31
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ai/workflow/coze/CozeWorkflowTest.java Normal file
View File

@@ -0,0 +1,31 @@
package com.yomahub.liteflow.test.ai.workflow.coze;
import com.yomahub.liteflow.ai.context.ChatContext;
import com.yomahub.liteflow.ai.util.SpringUtil;
import com.yomahub.liteflow.core.FlowExecutor;
import com.yomahub.liteflow.flow.LiteflowResponse;
import com.yomahub.liteflow.test.ai.workflow.coze.cmp.WorkFlowContext;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.context.annotation.ComponentScan;
import org.springframework.test.context.TestPropertySource;
import javax.annotation.Resource;
@TestPropertySource(properties = {"spring.config.location=classpath:workflow/coze/application.yaml"})
@SpringBootTest(classes = {CozeWorkflowTest.class, SpringUtil.class})
@EnableAutoConfiguration
@ComponentScan({"com.yomahub.liteflow.test.ai.workflow.coze.cmp"})
public class CozeWorkflowTest {
@Resource
private FlowExecutor flowExecutor;
@Test
public void testCozeWorkflow() {
LiteflowResponse response = flowExecutor.execute2Resp("chain1", null, ChatContext.class, WorkFlowContext.class);
Assertions.assertTrue(response.isSuccess());
}
}

15
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ai/workflow/coze/cmp/ACmp.java Normal file
View File

@@ -0,0 +1,15 @@
package com.yomahub.liteflow.test.ai.workflow.coze.cmp;
import com.yomahub.liteflow.core.NodeComponent;
import org.springframework.stereotype.Component;
@Component("a")
public class ACmp extends NodeComponent {
@Override
public void process() throws Exception {
WorkFlowContext context = this.getContextBean(WorkFlowContext.class);
context.setInput("What is the capital of France?");
System.out.println("ACmp executed!");
}
}

14
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ai/workflow/coze/cmp/BCmp.java Normal file
View File

@@ -0,0 +1,14 @@
package com.yomahub.liteflow.test.ai.workflow.coze.cmp;
import com.yomahub.liteflow.annotation.LiteflowComponent;
import com.yomahub.liteflow.core.NodeComponent;
@LiteflowComponent("b")
public class BCmp extends NodeComponent {
@Override
public void process() throws Exception {
WorkFlowContext context = this.getContextBean(WorkFlowContext.class);
System.out.println("BCmp executed! The result is: " + context.getResult().getData());
}
}

10
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ai/workflow/coze/cmp/WorkFlowContext.java Normal file
View File

@@ -0,0 +1,10 @@
package com.yomahub.liteflow.test.ai.workflow.coze.cmp;
import com.coze.openapi.client.workflows.run.RunWorkflowResp;
import lombok.Data;
@Data
public class WorkFlowContext {
private String input;
private RunWorkflowResp result;
}

23
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ai/workflow/coze/cmp/WorkflowCmp.java Normal file
View File

@@ -0,0 +1,23 @@
package com.yomahub.liteflow.test.ai.workflow.coze.cmp;
import com.yomahub.liteflow.ai.annotation.AIComponent;
import com.yomahub.liteflow.ai.annotation.model.io.AIOutput;
import com.yomahub.liteflow.ai.util.KeyValue;
import com.yomahub.liteflow.ai.workflow.coze.annotation.CozeWorkflowRun;
@AIComponent(
nodeId = "cozeWorkflowNode",
nodeName = "Coze-Workflow"
)
@CozeWorkflowRun(
workflowId = "7547589243724939314",
parameters = {
@KeyValue(key = "input", value = "{{input}}")
},
appId = "7547538064139485230"
)
@AIOutput(
methodExpress = "setResult"
)
public interface WorkflowCmp {
}

4
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ai/workflow/dashscope/WorkflowTest.java → liteflow-testcase-el/liteflow-testcase-el-ai/src/test/java/com/yomahub/liteflow/test/ai/workflow/dashscope/DashScopeWorkflowTest.java
View File

@@ -15,10 +15,10 @@ import org.springframework.test.context.TestPropertySource;
import javax.annotation.Resource;
@TestPropertySource(properties = {"spring.config.location=classpath:workflow/dashscope/application.yaml"})
@SpringBootTest(classes = {WorkflowTest.class, SpringUtil.class})
@SpringBootTest(classes = {DashScopeWorkflowTest.class, SpringUtil.class})
@EnableAutoConfiguration
@ComponentScan({"com.yomahub.liteflow.test.ai.workflow.dashscope.cmp"})
public class WorkflowTest {
public class DashScopeWorkflowTest {
@Resource
private FlowExecutor flowExecutor;

9
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/resources/workflow/coze/application.yaml Normal file
View File

@@ -0,0 +1,9 @@
liteflow:
rule-source: workflow/coze/flow.el.xml
ai:
base-packages:
- com.yomahub.liteflow.test.ai.workflow.coze.cmp
enable: true
workflow:
coze:
apiKey:

8
liteflow-testcase-el/liteflow-testcase-el-ai/src/test/resources/workflow/coze/flow.el.xml Normal file
View File

@@ -0,0 +1,8 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE flow PUBLIC "liteflow" "liteflow.dtd">
<flow>
<chain name="chain1">
THEN(a, cozeWorkflowNode, b);
</chain>
</flow>