From 77846601ee4c208fd13c52cc13a5401e0a81ff75 Mon Sep 17 00:00:00 2001 From: ageerle Date: Tue, 28 Jul 2026 11:22:26 +0800 Subject: [PATCH] =?UTF-8?q?test(retrieval):=20=E9=80=82=E9=85=8D=20Knowled?= =?UTF-8?q?geRetrievalServiceImpl=20=E6=96=B0=E5=A2=9E=20trace=20=E4=BE=9D?= =?UTF-8?q?=E8=B5=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 构造测试实例时补上 TraceRecordService 与 TraceProperties 两个新参数, 修复编译/构造失败。 Co-Authored-By: Claude --- .../impl/KnowledgeRetrievalServiceRegressionTest.java | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/ruoyi-modules/ruoyi-chat/src/test/java/org/ruoyi/service/retrieval/impl/KnowledgeRetrievalServiceRegressionTest.java b/ruoyi-modules/ruoyi-chat/src/test/java/org/ruoyi/service/retrieval/impl/KnowledgeRetrievalServiceRegressionTest.java index 0841af78..4c21ded9 100644 --- a/ruoyi-modules/ruoyi-chat/src/test/java/org/ruoyi/service/retrieval/impl/KnowledgeRetrievalServiceRegressionTest.java +++ b/ruoyi-modules/ruoyi-chat/src/test/java/org/ruoyi/service/retrieval/impl/KnowledgeRetrievalServiceRegressionTest.java @@ -2,6 +2,8 @@ package org.ruoyi.service.retrieval.impl; import org.junit.jupiter.api.Tag; import org.junit.jupiter.api.Test; +import org.ruoyi.common.trace.config.TraceProperties; +import org.ruoyi.common.trace.service.TraceRecordService; import org.ruoyi.domain.bo.vector.QueryVectorBo; import org.ruoyi.domain.vo.knowledge.KnowledgeRetrievalVo; import org.ruoyi.factory.RerankModelFactory; @@ -21,7 +23,8 @@ class KnowledgeRetrievalServiceRegressionTest { @Test void rrfMergesSameStableIdAndKeepsDistinctItems() throws Exception { KnowledgeRetrievalServiceImpl service = new KnowledgeRetrievalServiceImpl( - mock(VectorStoreService.class), mock(RerankModelFactory.class), mock(KnowledgeFragmentMapper.class)); + mock(VectorStoreService.class), mock(RerankModelFactory.class), mock(KnowledgeFragmentMapper.class), + mock(TraceRecordService.class), new TraceProperties()); KnowledgeRetrievalVo vectorA = item("fid-a", "A", 0.9); KnowledgeRetrievalVo vectorB = item("fid-b", "B", 0.8); KnowledgeRetrievalVo keywordA = item("fid-a", "A", 10.0); @@ -44,7 +47,8 @@ class KnowledgeRetrievalServiceRegressionTest { VectorStoreService vectorStore = mock(VectorStoreService.class); when(vectorStore.search(org.mockito.ArgumentMatchers.any())).thenReturn(List.of(item("fid-a", "A", 0.75))); KnowledgeRetrievalServiceImpl service = new KnowledgeRetrievalServiceImpl( - vectorStore, mock(RerankModelFactory.class), mock(KnowledgeFragmentMapper.class)); + vectorStore, mock(RerankModelFactory.class), mock(KnowledgeFragmentMapper.class), + mock(TraceRecordService.class), new TraceProperties()); QueryVectorBo query = new QueryVectorBo(); query.setKid("1"); query.setQuery("test");