xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-09-13 08:25:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(retrieval): 适配 KnowledgeRetrievalServiceImpl 新增 trace 依赖

Browse Source 
构造测试实例时补上 TraceRecordService 与 TraceProperties 两个新参数,
修复编译/构造失败。

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
ageerle
2026-07-28 11:22:26 +08:00
parent b80dc6e23f
commit 77846601ee
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ruoyi-modules/ruoyi-chat/src/test/java/org/ruoyi/service/retrieval/impl/KnowledgeRetrievalServiceRegressionTest.java
View File

@@ -2,6 +2,8 @@ package org.ruoyi.service.retrieval.impl;
import org.junit.jupiter.api.Tag; import org.junit.jupiter.api.Tag;
import org.junit.jupiter.api.Test; 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.bo.vector.QueryVectorBo;
import org.ruoyi.domain.vo.knowledge.KnowledgeRetrievalVo; import org.ruoyi.domain.vo.knowledge.KnowledgeRetrievalVo;
import org.ruoyi.factory.RerankModelFactory; import org.ruoyi.factory.RerankModelFactory;
@@ -21,7 +23,8 @@ class KnowledgeRetrievalServiceRegressionTest {
@Test @Test
void rrfMergesSameStableIdAndKeepsDistinctItems() throws Exception { void rrfMergesSameStableIdAndKeepsDistinctItems() throws Exception {
KnowledgeRetrievalServiceImpl service = new KnowledgeRetrievalServiceImpl( 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 vectorA = item("fid-a", "A", 0.9);
KnowledgeRetrievalVo vectorB = item("fid-b", "B", 0.8); KnowledgeRetrievalVo vectorB = item("fid-b", "B", 0.8);
KnowledgeRetrievalVo keywordA = item("fid-a", "A", 10.0); KnowledgeRetrievalVo keywordA = item("fid-a", "A", 10.0);
@@ -44,7 +47,8 @@ class KnowledgeRetrievalServiceRegressionTest {
VectorStoreService vectorStore = mock(VectorStoreService.class); VectorStoreService vectorStore = mock(VectorStoreService.class);
when(vectorStore.search(org.mockito.ArgumentMatchers.any())).thenReturn(List.of(item("fid-a", "A", 0.75))); when(vectorStore.search(org.mockito.ArgumentMatchers.any())).thenReturn(List.of(item("fid-a", "A", 0.75)));
KnowledgeRetrievalServiceImpl service = new KnowledgeRetrievalServiceImpl( 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(); QueryVectorBo query = new QueryVectorBo();
query.setKid("1"); query.setKid("1");
query.setQuery("test"); query.setQuery("test");