Add auto memory distillation scheduler (APScheduler)
- scheduler.py: AsyncIOScheduler with three cron jobs
short daily 03:00 (no LLM, always fast)
mid weekly Sun 03:30 (LLM)
long monthly 1st 04:00 (LLM — off by default)
- config.py: AUTO_DISTILL, AUTO_DISTILL_SHORT/MID/LONG .env flags
- main.py: start/stop scheduler in FastAPI lifespan
- routers/distill.py: GET /distill/status — next run times + config
- requirements.txt: apscheduler>=3.10
- HELP.md: updated planned items, added /distill/status to API table
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -26,6 +26,13 @@ class Settings(BaseSettings):
|
|||||||
nextcloud_talk_bot_secret: str = "" # set in .env
|
nextcloud_talk_bot_secret: str = "" # set in .env
|
||||||
nextcloud_talk_timeout: int = 55
|
nextcloud_talk_timeout: int = 55
|
||||||
|
|
||||||
|
# Auto-distillation schedule — override in .env
|
||||||
|
# AUTO_DISTILL=false disables entirely
|
||||||
|
auto_distill: bool = True
|
||||||
|
auto_distill_short: bool = True # daily at 03:00 — rolls session logs → MEMORY_SHORT
|
||||||
|
auto_distill_mid: bool = True # weekly Sunday at 03:30 — LLM summarizes short → mid
|
||||||
|
auto_distill_long: bool = False # monthly 1st at 04:00 — off by default (manual review recommended)
|
||||||
|
|
||||||
# Memory tier token budgets — soft caps used during distillation
|
# Memory tier token budgets — soft caps used during distillation
|
||||||
# Override in .env: MEMORY_BUDGET_LONG=4000 etc.
|
# Override in .env: MEMORY_BUDGET_LONG=4000 etc.
|
||||||
memory_budget_long: int = 2000
|
memory_budget_long: int = 2000
|
||||||
|
|||||||
@@ -13,7 +13,10 @@ from routers import chat, google_chat, nextcloud_talk, files, distill
|
|||||||
|
|
||||||
@asynccontextmanager
|
@asynccontextmanager
|
||||||
async def lifespan(app: FastAPI):
|
async def lifespan(app: FastAPI):
|
||||||
|
import scheduler
|
||||||
|
scheduler.start()
|
||||||
yield
|
yield
|
||||||
|
scheduler.stop()
|
||||||
from llm_client import cleanup
|
from llm_client import cleanup
|
||||||
await cleanup()
|
await cleanup()
|
||||||
|
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
fastapi>=0.115.0
|
fastapi>=0.115.0
|
||||||
|
apscheduler>=3.10
|
||||||
uvicorn[standard]>=0.30.0
|
uvicorn[standard]>=0.30.0
|
||||||
pydantic-settings>=2.0.0
|
pydantic-settings>=2.0.0
|
||||||
python-dotenv>=1.0.0
|
python-dotenv>=1.0.0
|
||||||
|
|||||||
@@ -8,10 +8,26 @@ Manual memory distillation endpoints.
|
|||||||
"""
|
"""
|
||||||
from fastapi import APIRouter
|
from fastapi import APIRouter
|
||||||
from memory_distiller import distill_short, distill_mid, distill_long
|
from memory_distiller import distill_short, distill_mid, distill_long
|
||||||
|
import scheduler
|
||||||
|
|
||||||
router = APIRouter(prefix="/distill")
|
router = APIRouter(prefix="/distill")
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/status")
|
||||||
|
async def distill_status() -> dict:
|
||||||
|
"""Show auto-distillation schedule and next run times."""
|
||||||
|
from config import settings
|
||||||
|
return {
|
||||||
|
"enabled": settings.auto_distill,
|
||||||
|
"jobs": scheduler.status(),
|
||||||
|
"config": {
|
||||||
|
"short": settings.auto_distill_short,
|
||||||
|
"mid": settings.auto_distill_mid,
|
||||||
|
"long": settings.auto_distill_long,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
@router.post("/short")
|
@router.post("/short")
|
||||||
async def do_distill_short() -> dict:
|
async def do_distill_short() -> dict:
|
||||||
return {"ok": True, **distill_short()}
|
return {"ok": True, **distill_short()}
|
||||||
|
|||||||
99
cortex/scheduler.py
Normal file
99
cortex/scheduler.py
Normal file
@@ -0,0 +1,99 @@
|
|||||||
|
"""
|
||||||
|
Auto memory distillation scheduler.
|
||||||
|
|
||||||
|
Default schedule (all overridable via .env flags):
|
||||||
|
short — daily at 03:00 (no LLM — fast)
|
||||||
|
mid — weekly Sun at 03:30 (LLM call)
|
||||||
|
long — monthly 1st at 04:00 (LLM call — off by default)
|
||||||
|
|
||||||
|
Set AUTO_DISTILL=false to disable entirely.
|
||||||
|
Set AUTO_DISTILL_LONG=true to enable monthly long-term integration.
|
||||||
|
"""
|
||||||
|
import logging
|
||||||
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
||||||
|
from config import settings
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
_scheduler: AsyncIOScheduler | None = None
|
||||||
|
|
||||||
|
|
||||||
|
async def _run_short() -> None:
|
||||||
|
from memory_distiller import distill_short
|
||||||
|
try:
|
||||||
|
result = distill_short()
|
||||||
|
logger.info("auto distill short: %d files, %d chars", result["files_included"], result["chars_written"])
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("auto distill short failed: %s", e)
|
||||||
|
|
||||||
|
|
||||||
|
async def _run_mid() -> None:
|
||||||
|
from memory_distiller import distill_mid
|
||||||
|
try:
|
||||||
|
result = await distill_mid()
|
||||||
|
if "error" in result:
|
||||||
|
logger.warning("auto distill mid skipped: %s", result["error"])
|
||||||
|
else:
|
||||||
|
logger.info("auto distill mid: %d chars via %s", result["chars_written"], result["backend"])
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("auto distill mid failed: %s", e)
|
||||||
|
|
||||||
|
|
||||||
|
async def _run_long() -> None:
|
||||||
|
from memory_distiller import distill_long
|
||||||
|
try:
|
||||||
|
result = await distill_long()
|
||||||
|
if "error" in result:
|
||||||
|
logger.warning("auto distill long skipped: %s", result["error"])
|
||||||
|
else:
|
||||||
|
logger.info("auto distill long: %d chars via %s", result["chars_written"], result["backend"])
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("auto distill long failed: %s", e)
|
||||||
|
|
||||||
|
|
||||||
|
def start() -> None:
|
||||||
|
global _scheduler
|
||||||
|
if not settings.auto_distill:
|
||||||
|
logger.info("auto distillation disabled (AUTO_DISTILL=false)")
|
||||||
|
return
|
||||||
|
|
||||||
|
_scheduler = AsyncIOScheduler(timezone="local")
|
||||||
|
|
||||||
|
if settings.auto_distill_short:
|
||||||
|
_scheduler.add_job(_run_short, "cron", hour=3, minute=0, id="distill_short")
|
||||||
|
logger.info("scheduled: distill_short daily at 03:00")
|
||||||
|
|
||||||
|
if settings.auto_distill_mid:
|
||||||
|
_scheduler.add_job(_run_mid, "cron", day_of_week="sun", hour=3, minute=30, id="distill_mid")
|
||||||
|
logger.info("scheduled: distill_mid weekly Sun at 03:30")
|
||||||
|
|
||||||
|
if settings.auto_distill_long:
|
||||||
|
_scheduler.add_job(_run_long, "cron", day=1, hour=4, minute=0, id="distill_long")
|
||||||
|
logger.info("scheduled: distill_long monthly on 1st at 04:00")
|
||||||
|
|
||||||
|
if _scheduler.get_jobs():
|
||||||
|
_scheduler.start()
|
||||||
|
logger.info("auto distillation scheduler started (%d jobs)", len(_scheduler.get_jobs()))
|
||||||
|
else:
|
||||||
|
logger.info("auto distillation: no jobs enabled")
|
||||||
|
|
||||||
|
|
||||||
|
def stop() -> None:
|
||||||
|
global _scheduler
|
||||||
|
if _scheduler and _scheduler.running:
|
||||||
|
_scheduler.shutdown(wait=False)
|
||||||
|
logger.info("auto distillation scheduler stopped")
|
||||||
|
|
||||||
|
|
||||||
|
def status() -> list[dict]:
|
||||||
|
"""Return next-run info for all scheduled jobs."""
|
||||||
|
if not _scheduler or not _scheduler.running:
|
||||||
|
return []
|
||||||
|
jobs = []
|
||||||
|
for job in _scheduler.get_jobs():
|
||||||
|
next_run = job.next_run_time
|
||||||
|
jobs.append({
|
||||||
|
"id": job.id,
|
||||||
|
"next_run": next_run.isoformat() if next_run else None,
|
||||||
|
})
|
||||||
|
return jobs
|
||||||
@@ -177,6 +177,7 @@ For direct access or scripting:
|
|||||||
| `POST` | `/distill/mid` | Summarize short → MEMORY_MID (LLM) |
|
| `POST` | `/distill/mid` | Summarize short → MEMORY_MID (LLM) |
|
||||||
| `POST` | `/distill/long` | Integrate mid → MEMORY_LONG (LLM) |
|
| `POST` | `/distill/long` | Integrate mid → MEMORY_LONG (LLM) |
|
||||||
| `POST` | `/distill/all` | Run all three distillation steps |
|
| `POST` | `/distill/all` | Run all three distillation steps |
|
||||||
|
| `GET` | `/distill/status` | Show scheduler status and next run times |
|
||||||
| `GET` | `/health` | Health check — returns `{"status": "ok"}` |
|
| `GET` | `/health` | Health check — returns `{"status": "ok"}` |
|
||||||
|
|
||||||
Chat request body (`POST /chat`):
|
Chat request body (`POST /chat`):
|
||||||
@@ -196,7 +197,7 @@ Chat request body (`POST /chat`):
|
|||||||
|
|
||||||
## In Progress / Planned
|
## In Progress / Planned
|
||||||
|
|
||||||
- **Auto memory distillation** — currently manual trigger only; scheduled auto-run planned
|
- **Auto memory distillation (long)** — short and mid run automatically; long-term integration is off by default (set `AUTO_DISTILL_LONG=true` in `.env` to enable)
|
||||||
- **Ollama local model backend** — direct Ollama API support (no CLI wrapper)
|
- **Ollama local model backend** — direct Ollama API support (no CLI wrapper)
|
||||||
- **OAuth token auto-refresh notifications** — alert when Claude CLI token is near expiry
|
- **OAuth token auto-refresh notifications** — alert when Claude CLI token is near expiry
|
||||||
- **Multi-user support** — per-user identity/memory files; currently single-user (Scott)
|
- **Multi-user support** — per-user identity/memory files; currently single-user (Scott)
|
||||||
|
|||||||
Reference in New Issue
Block a user