diff --git a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/service/retrieval/impl/KnowledgeRetrievalServiceImpl.java b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/service/retrieval/impl/KnowledgeRetrievalServiceImpl.java index 5d3cb5e6..89295a69 100644 --- a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/service/retrieval/impl/KnowledgeRetrievalServiceImpl.java +++ b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/service/retrieval/impl/KnowledgeRetrievalServiceImpl.java @@ -81,7 +81,7 @@ public class KnowledgeRetrievalServiceImpl implements KnowledgeRetrievalService List finalResults = TraceNodeTemplate.withNode(traceRecordService, traceProperties, "retrieval", RagTraceNodeTypes.NODE_RETRIEVAL, KnowledgeRetrievalServiceImpl.class.getName(), "retrieve", - RagTracePayloadBuilder.retrievalInputSummary(queryVectorBo), + retrievalInputPayload, () -> retrieveUncached(queryVectorBo), RagTracePayloadBuilder::retrievalOutputSummary); cache(cacheKey, finalResults);