adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(celery): enable periodic task scheduling for memory management

Browse Source
This commit is contained in:
Ke Sun
2026-02-10 10:44:42 +08:00
parent 100bf4fa49
commit 0011a8ce9f
1 changed files with 35 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
api/app/celery_app.py
View File

@@ -90,40 +90,41 @@ celery_app.conf.update(
celery_app.autodiscover_tasks(['app']) celery_app.autodiscover_tasks(['app'])
# Celery Beat schedule for periodic tasks # Celery Beat schedule for periodic tasks
# memory_increment_schedule = timedelta(hours=settings.MEMORY_INCREMENT_INTERVAL_HOURS) memory_increment_schedule = timedelta(hours=settings.MEMORY_INCREMENT_INTERVAL_HOURS)
# memory_cache_regeneration_schedule = timedelta(hours=settings.MEMORY_CACHE_REGENERATION_HOURS) memory_cache_regeneration_schedule = timedelta(hours=settings.MEMORY_CACHE_REGENERATION_HOURS)
# workspace_reflection_schedule = timedelta(seconds=30) # 每30秒运行一次settings.REFLECTION_INTERVAL_TIME # 这个30秒的设计不合理
# forgetting_cycle_schedule = timedelta(hours=24) # 每24小时运行一次遗忘周期 workspace_reflection_schedule = timedelta(seconds=30) # 每30秒运行一次settings.REFLECTION_INTERVAL_TIME
forgetting_cycle_schedule = timedelta(hours=24) # 每24小时运行一次遗忘周期
# 构建定时任务配置 #构建定时任务配置
# beat_schedule_config = { beat_schedule_config = {
# "run-workspace-reflection": { "run-workspace-reflection": {
# "task": "app.tasks.workspace_reflection_task", "task": "app.tasks.workspace_reflection_task",
# "schedule": workspace_reflection_schedule, "schedule": workspace_reflection_schedule,
# "args": (), "args": (),
# }, },
# "regenerate-memory-cache": { "regenerate-memory-cache": {
# "task": "app.tasks.regenerate_memory_cache", "task": "app.tasks.regenerate_memory_cache",
# "schedule": memory_cache_regeneration_schedule, "schedule": memory_cache_regeneration_schedule,
# "args": (), "args": (),
# }, },
# "run-forgetting-cycle": { "run-forgetting-cycle": {
# "task": "app.tasks.run_forgetting_cycle_task", "task": "app.tasks.run_forgetting_cycle_task",
# "schedule": forgetting_cycle_schedule, "schedule": forgetting_cycle_schedule,
# "kwargs": { "kwargs": {
# "config_id": None, # 使用默认配置，可以通过环境变量配置 "config_id": None, # 使用默认配置，可以通过环境变量配置
# }, },
# }, },
# } }
# 如果配置了默认工作空间ID，则添加记忆总量统计任务 #如果配置了默认工作空间ID，则添加记忆总量统计任务
# if settings.DEFAULT_WORKSPACE_ID: if settings.DEFAULT_WORKSPACE_ID:
# beat_schedule_config["write-total-memory"] = { beat_schedule_config["write-total-memory"] = {
# "task": "app.controllers.memory_storage_controller.search_all", "task": "app.controllers.memory_storage_controller.search_all",
# "schedule": memory_increment_schedule, "schedule": memory_increment_schedule,
# "kwargs": { "kwargs": {
# "workspace_id": settings.DEFAULT_WORKSPACE_ID, "workspace_id": settings.DEFAULT_WORKSPACE_ID,
# }, },
# } }
# celery_app.conf.beat_schedule = beat_schedule_config celery_app.conf.beat_schedule = beat_schedule_config