76 lines
3.1 KiB
Java
76 lines
3.1 KiB
Java
package tech.easyflow.manuagent.agent;
|
||
|
||
import static org.assertj.core.api.Assertions.assertThat;
|
||
import static org.mockito.ArgumentMatchers.any;
|
||
import static org.mockito.ArgumentMatchers.eq;
|
||
import static org.mockito.Mockito.mock;
|
||
import static org.mockito.Mockito.verify;
|
||
import static org.mockito.Mockito.when;
|
||
|
||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||
import io.agentscope.core.agui.adapter.AguiAgentAdapter;
|
||
import java.util.UUID;
|
||
import org.junit.jupiter.api.Test;
|
||
import reactor.core.publisher.Flux;
|
||
import reactor.core.publisher.Mono;
|
||
import tech.easyflow.manuagent.project.ProjectFileService;
|
||
import tech.easyflow.manuagent.project.ProjectService;
|
||
import tech.easyflow.manuagent.skill.SkillService;
|
||
|
||
/**
|
||
* 验证 Agent 事件持久化的精简规则。
|
||
*/
|
||
class AgentExecutionServiceTest {
|
||
|
||
/**
|
||
* Run 在创建时已经固化模型配置 ID；执行和模型重连都必须沿用该 ID，
|
||
* 不能在工厂内部重新读取可能已经变化的全局默认模型。
|
||
*/
|
||
@Test
|
||
void shouldCreateAgentWithModelBoundToRun() {
|
||
UUID projectId = UUID.randomUUID();
|
||
UUID runId = UUID.randomUUID();
|
||
UUID modelId = UUID.randomUUID();
|
||
AgentFactory factory = mock(AgentFactory.class);
|
||
AgentFactory.AgentHandle handle = mock(AgentFactory.AgentHandle.class);
|
||
AguiAgentAdapter adapter = mock(AguiAgentAdapter.class);
|
||
SkillService skillService = mock(SkillService.class);
|
||
when(handle.adapter()).thenReturn(adapter);
|
||
when(adapter.run(any())).thenReturn(Flux.empty());
|
||
when(skillService.enabledNames()).thenReturn(new String[] {"document"});
|
||
when(factory.create(eq(projectId), eq(modelId), any(String[].class))).thenReturn(handle);
|
||
|
||
AgentExecutionService service = new AgentExecutionService(
|
||
new ObjectMapper(),
|
||
factory,
|
||
mock(AgentEventService.class),
|
||
mock(ProjectFileService.class),
|
||
skillService);
|
||
ProjectService.ProjectView project = mock(ProjectService.ProjectView.class);
|
||
when(project.id()).thenReturn(projectId);
|
||
when(project.threadId()).thenReturn("project-" + projectId);
|
||
AgentRunService.RunView run = new AgentRunService.RunView(
|
||
runId, projectId, modelId, "INITIAL", "RUNNING", null, null, null, null);
|
||
|
||
service.execute(project, run, "执行测试", Mono.never(), () -> { }, () -> false);
|
||
|
||
verify(factory).create(eq(projectId), eq(modelId), any(String[].class));
|
||
}
|
||
|
||
/**
|
||
* 验证文档视觉结果保留图片路径元数据并丢弃 Base64 正文。
|
||
*/
|
||
@Test
|
||
void shouldStripInlineImageDataFromPersistedToolResult() {
|
||
String content = """
|
||
document_view_result={"images":[{"path":"work/tmp/document-view/a/render-1.png"}]}
|
||
{"type":"image","source":{"media_type":"image/png","data":"very-large-base64"}}
|
||
""";
|
||
|
||
String result = AgentExecutionService.stripInlineImageData(content);
|
||
|
||
assertThat(result).contains("render-1.png");
|
||
assertThat(result).doesNotContain("very-large-base64");
|
||
}
|
||
}
|