Commit Graph

8 Commits

Author SHA1 Message Date
minecraft1024a
f44ece0b29 style: 统一代码风格并采用现代化类型注解
对整个代码库进行了一次全面的代码风格清理和现代化改造,主要包括:

- 移除了所有文件中多余的行尾空格。
- 将类型提示更新为 PEP 585 和 PEP 604 引入的现代语法(例如,使用 `list` 代替 `List`,使用 `|` 代替 `Optional`)。
- 清理了多个模块中未被使用的导入语句。
- 移除了不含插值变量的冗余 f-string。
- 调整了部分 `__init__.py` 文件中的 `__all__` 导出顺序,以保持一致性。

这些改动旨在提升代码的可读性和可维护性,使其与现代 Python 最佳实践保持一致,但未修改任何核心逻辑。
2025-11-19 23:37:27 +08:00
Windpicker-owo
8f668f18f7 feat:增强记忆节点的嵌入生成和日志记录- 在 MemoryBuilder 中为 SUBJECT 和 VALUE 节点类型添加了嵌入生成,确保仅为内容足够的节点创建嵌入。- 改进了 MemoryTools 的日志记录,在初始向量搜索期间提供详细见解,包括低召回情况的警告。- 调整了不同记忆类型的评分权重,以强调相似性和重要性,提高记忆检索的质量。- 将向量搜索限制从 2 倍提高到 5 倍,以改善初始召回率。- 引入了一个新脚本,用于为现有节点生成缺失的嵌入,支持批量处理并改进索引。 2025-11-19 23:37:19 +08:00
minecraft1024a
73e564a378 Revert "feat(visualizer): 引入核心图按需加载和节点扩展功能"
This reverts commit 0c41cd2a13.
2025-11-19 23:36:27 +08:00
minecraft1024a
aa0f57cb99 feat(visualizer): 引入核心图按需加载和节点扩展功能
为了解决记忆图谱数据量过大导致前端加载缓慢和渲染卡顿的问题,本次更新引入了核心图按需加载和节点扩展机制。

主要变更包括:

- **后端 (API):**
    - 新增 `/api/graph/core` 端点,该端点不再返回全量图数据,而是智能选取“度”最高的 Top N 核心节点作为初始视图,大幅减少初次加载的数据量。
    - 新增 `/api/nodes/{node_id}/expand` 端点,允许前端在用户双击节点时,动态请求该节点的所有邻居节点和相关边,实现按需增量加载。
    - 优化了数据加载逻辑,在内存中构建并缓存了节点字典和邻接表,以极高的效率支持节点扩展查询。

- **前端 (UI):**
    - 初始加载逻辑从请求 `/api/graph/full` 切换到新的 `/api/graph/core` 端点。
    - 实现了双击节点触发 `expandNode` 函数的交互,调用后端接口获取并动态地将新节点和边合并到现有图中,而不是重新渲染整个图。
    - 使用 `vis.DataSet` 来管理图数据,支持高效地动态添加和更新节点与边。
    - 节点大小现在与其“度”(连接数)相关联,使得关键节点在视觉上更加突出。
2025-11-19 23:36:26 +08:00
minecraft1024a
a0b069dfa5 fix(api): 修复记忆可视化中重复的边
在从内存管理器和文件加载图数据时,由于遍历所有记忆(memory)并收集其关联的边(edge)时没有进行去重处理,导致同一条边如果被多个记忆引用,会在最终的图谱中重复出现,造成前端可视化混乱。

为了解决这个问题,引入了一个集合(set)或字典(dict)来跟踪已经处理过的边的ID。在遍历边的过程中,只有当边的ID未被记录时,才会将其添加到最终的边列表中。这样可以确保每条边在图谱数据中只出现一次,保证了可视化结果的准确性。
2025-11-19 23:36:26 +08:00
minecraft1024a
e350b5a177 Revert "fix(api): 修复记忆可视化中重复的边"
This reverts commit f8e42f3348.
2025-11-19 23:36:26 +08:00
minecraft1024a
679c787551 fix(api): 修复记忆可视化中重复的边
在从文件或内存管理器加载图数据时,由于遍历所有记忆(memory)并将其包含的边(edge)添加到列表中,导致了边的重复。当多个记忆共享同一条边时,这条边会被多次添加到最终的边列表中,造成前端可视化图中出现冗余的连接。

此提交通过引入一个集合(set)或字典(dict)来跟踪已经处理过的边的 ID,确保每条边只被添加一次,从而解决了重复边的问题。
2025-11-19 23:36:23 +08:00
minecraft1024a
5702dd8a9f refactor(server): 将记忆可视化工具和统计功能整合进主服务
将原先独立的记忆可视化工具(Memory Visualizer)和LLM使用统计逻辑深度整合到项目主服务中。

主要变更包括:
- **移除独立的可视化工具**: 删除了 `tools/memory_visualizer` 目录下的所有独立服务器、脚本和文档,清理了项目结构。
- **API路由整合**: 在主 FastAPI 应用中注册了记忆可视化工具的路由,使其成为核心功能的一部分,可通过 `/visualizer` 访问。
- **统计逻辑重构**: 将LLM使用统计的计算逻辑从API路由层 `statistic_router.py` 中剥离,迁移到 `src/chat/utils/statistic.py` 中,实现了逻辑的解耦和复用。API路由现在直接调用重构后的统计任务。
- **依赖清理与添加**: 添加了 `jinja2` 作为模板渲染的依赖,并清除了与独立可视化工具相关的旧依赖。

此次重构简化了项目的维护和部署,将原本分散的功能统一管理,提升了代码的内聚性和可维护性。
2025-11-19 23:35:41 +08:00