slhaf/Partner
1
0
Fork 0
mirror of https://github.com/slhaf/Partner.git synced 2026-05-12 16:53:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(memory): remove legacy dialogMap in MemoryRuntime and related modules

Browse Source
This commit is contained in:
slhafzjw
2026-03-29 18:54:23 +08:00
parent d4a5c5a0ed
commit cb09b86b23
4 changed files with 3 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
Partner-Core/src/main/java/work/slhaf/partner/module/memory/runtime/MemoryRuntime.java
View File

@@ -25,7 +25,6 @@ import java.nio.file.Path;
import java.nio.file.Paths;
import java.time.Instant;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.ZoneId;
import java.util.*;
import java.util.concurrent.CopyOnWriteArrayList;
@@ -47,7 +46,6 @@ public class MemoryRuntime extends AbstractAgentModule.Standalone {
private final ReentrantLock runtimeLock = new ReentrantLock();
private Map<String, CopyOnWriteArrayList<SliceRef>> topicSlices = new HashMap<>();
private Map<LocalDate, CopyOnWriteArrayList<SliceRef>> dateIndex = new HashMap<>();
private HashMap<LocalDateTime, String> dialogMap = new HashMap<>();
@Init
public void init() {
@@ -79,7 +77,7 @@ public class MemoryRuntime extends AbstractAgentModule.Standalone {
}
}
public void recordMemory(MemoryUnit memoryUnit, String topicPath, List<String> relatedTopicPaths, String dialogSummary) {
public void recordMemory(MemoryUnit memoryUnit, String topicPath, List<String> relatedTopicPaths) {
memoryCapability.saveMemoryUnit(memoryUnit);
MemorySlice memorySlice = memoryUnit.getSlices().getLast();
SliceRef sliceRef = new SliceRef(memoryUnit.getId(), memorySlice.getId());
@@ -90,7 +88,6 @@ public class MemoryRuntime extends AbstractAgentModule.Standalone {
bindTopic(relatedTopicPath, sliceRef);
}
}
updateDialogMap(LocalDateTime.now(), dialogSummary);
}
private void indexMemoryUnit(MemoryUnit memoryUnit) {
@@ -139,40 +136,6 @@ public class MemoryRuntime extends AbstractAgentModule.Standalone {
return buildActivatedMemorySlices(findByDate(date));
}
private void updateDialogMap(LocalDateTime dateTime, String newDialogCache) {
runtimeLock.lock();
try {
List<LocalDateTime> keysToRemove = new ArrayList<>();
dialogMap.forEach((k, v) -> {
if (dateTime.minusDays(2).isAfter(k)) {
keysToRemove.add(k);
}
});
for (LocalDateTime temp : keysToRemove) {
dialogMap.remove(temp);
}
dialogMap.put(dateTime, newDialogCache);
saveState();
} finally {
runtimeLock.unlock();
}
}
public String getDialogMapStr() {
StringBuilder str = new StringBuilder();
dialogMap.entrySet().stream()
.sorted(Map.Entry.comparingByKey())
.forEach(entry -> str.append("\n\n[")
.append(entry.getKey())
.append("]\n")
.append(entry.getValue()));
return str.toString();
}
public boolean containsDialogSummary(String summary) {
return dialogMap.containsValue(summary);
}
public String getTopicTree() {
TopicTreeNode root = new TopicTreeNode();
for (Map.Entry<String, CopyOnWriteArrayList<SliceRef>> entry : topicSlices.entrySet()) {
@@ -267,12 +230,10 @@ public class MemoryRuntime extends AbstractAgentModule.Standalone {
RuntimeState state = (RuntimeState) ois.readObject();
topicSlices = state.topicSlices;
dateIndex = state.dateIndex;
dialogMap = state.dialogMap;
} catch (Exception e) {
log.error("[MemoryRuntime] 加载运行态失败", e);
topicSlices = new HashMap<>();
dateIndex = new HashMap<>();
dialogMap = new HashMap<>();
}
}
@@ -295,7 +256,6 @@ public class MemoryRuntime extends AbstractAgentModule.Standalone {
RuntimeState state = new RuntimeState();
state.topicSlices = new HashMap<>(topicSlices);
state.dateIndex = new HashMap<>(dateIndex);
state.dialogMap = new HashMap<>(dialogMap);
oos.writeObject(state);
}
Files.move(tempPath, filePath, java.nio.file.StandardCopyOption.REPLACE_EXISTING);
@@ -325,6 +285,5 @@ public class MemoryRuntime extends AbstractAgentModule.Standalone {
private Map<String, CopyOnWriteArrayList<SliceRef>> topicSlices = new HashMap<>();
private Map<LocalDate, CopyOnWriteArrayList<SliceRef>> dateIndex = new HashMap<>();
private HashMap<LocalDateTime, String> dialogMap = new HashMap<>();
}
}

5
Partner-Core/src/main/java/work/slhaf/partner/module/memory/selector/MemorySelector.java
View File

@@ -210,7 +210,6 @@ public class MemorySelector extends AbstractAgentModule.Running<PartnerRunningFl
log.debug("[MemorySelector] 触发记忆回溯...");
LinkedHashMap<String, ActivatedMemorySlice> candidates = new LinkedHashMap<>();
setMemoryCandidates(candidates, extractorResult.getMatches());
removeDuplicateSlice(candidates.values());
EvaluatorInput evaluatorInput = EvaluatorInput.builder()
.inputs(snapshotInputs)
.memorySlices(new ArrayList<>(candidates.values()))
@@ -239,10 +238,6 @@ public class MemorySelector extends AbstractAgentModule.Running<PartnerRunningFl
}
}
private void removeDuplicateSlice(Collection<ActivatedMemorySlice> candidates) {
candidates.removeIf(m -> memoryRuntime.containsDialogSummary(m.getSummary()));
}
@Override
public int order() {
return 2;

3
Partner-Core/src/main/java/work/slhaf/partner/module/memory/updater/MemoryUpdater.java
View File

@@ -172,8 +172,7 @@ public class MemoryUpdater extends AbstractAgentModule.Running<PartnerRunningFlo
memoryRuntime.recordMemory(
memoryUnit,
summarizeResult.getTopicPath(),
summarizeResult.getRelatedTopicPath(),
summarizeResult.getSummary()
summarizeResult.getRelatedTopicPath()
);
log.debug("[MemoryUpdater] 记忆更新流程结束...");
MemorySlice newSlice = memoryUnit.getSlices().getLast();