1
0

reindex_milvus.py 544 B

123456789101112131415161718192021222324
  1. """
  2. 将 SQLite 历史文本记忆回填到 Milvus。
  3. 运行:
  4. cd backend && .venv/bin/python scripts/reindex_milvus.py
  5. """
  6. from __future__ import annotations
  7. from memory.store import list_all_user_ids
  8. from rag.indexers import reindex_user
  9. def main() -> None:
  10. users = list_all_user_ids(limit=5000)
  11. total = 0
  12. for uid in users:
  13. n = reindex_user(uid, limit=500)
  14. total += n
  15. print(f"user={uid} indexed={n}")
  16. print(f"done users={len(users)} total_chunks={total}")
  17. if __name__ == "__main__":
  18. main()