15d436b3a1
- 将 A_Memorix 从旧 submodule / 插件形态迁入主线源码，主体落到 src/A_memorix - 调整主程序接入方式，使 A_Memorix 作为源码内长期记忆子系统运行 - 回收父项目插件体系中针对 A_Memorix 的特判，减少对 plugin 通用层的侵入 - 将长期记忆配置、运行时、自检、导入、调优等能力收口到 memory 路由与主线服务层 - 重做长期记忆控制台与图谱页面，按 MaiBot 现有 dashboard 风格接入 - 补充实体关系图与证据视图双视图能力，支持查看节点、关系、段落及其证据链路 - 新增长期记忆配置编辑器与 memory-api，支持主线内配置管理 - 补齐删除管理能力：删除预览、混合删除、来源批量删除、删除操作恢复 - 优化删除预览与删除操作详情的前端展示，支持分页、检索，并以实体名/关系内容/段落摘要替代单纯 hash 展示 - 修复图谱与控制台相关前端问题，包括证据视图切换、查询触发时机、删除弹层空值保护等 - 新增或更新 A_Memorix 相关测试、WebUI 路由测试、前端 vitest 测试与辅助验证脚本 - 移除旧 plugins/A_memorix、.gitmodules 及相关历史维护文档
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
回填段落时序字段。
|
|
|
|
默认策略：
|
|
1. 若段落缺失 event_time/event_time_start/event_time_end
|
|
2. 且存在 created_at
|
|
3. 写入 event_time=created_at, time_granularity=day, time_confidence=0.2
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
|
|
from _bootstrap import DEFAULT_DATA_DIR, resolve_repo_path
|
|
from A_memorix.core.storage import MetadataStore # noqa: E402
|
|
|
|
|
|
def backfill(
|
|
data_dir: Path,
|
|
dry_run: bool,
|
|
limit: int,
|
|
no_created_fallback: bool,
|
|
) -> int:
|
|
store = MetadataStore(data_dir=data_dir)
|
|
store.connect()
|
|
summary = store.backfill_temporal_metadata_from_created_at(
|
|
limit=limit,
|
|
dry_run=dry_run,
|
|
no_created_fallback=no_created_fallback,
|
|
)
|
|
store.close()
|
|
if dry_run:
|
|
print(f"[dry-run] candidates={summary['candidates']}")
|
|
return int(summary["candidates"])
|
|
if no_created_fallback:
|
|
print(f"skip update (no-created-fallback), candidates={summary['candidates']}")
|
|
return 0
|
|
print(f"updated={summary['updated']}")
|
|
return int(summary["updated"])
|
|
|
|
|
|
def main() -> int:
|
|
parser = argparse.ArgumentParser(description="Backfill temporal metadata for A_Memorix paragraphs")
|
|
parser.add_argument("--data-dir", default=str(DEFAULT_DATA_DIR), help="数据目录")
|
|
parser.add_argument("--dry-run", action="store_true", help="仅统计，不写入")
|
|
parser.add_argument("--limit", type=int, default=100000, help="最大处理条数")
|
|
parser.add_argument(
|
|
"--no-created-fallback",
|
|
action="store_true",
|
|
help="不使用 created_at 回填，仅输出候选数量",
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
backfill(
|
|
data_dir=resolve_repo_path(args.data_dir, fallback=DEFAULT_DATA_DIR),
|
|
dry_run=args.dry_run,
|
|
limit=max(1, int(args.limit)),
|
|
no_created_fallback=args.no_created_fallback,
|
|
)
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|