518b8eca85
md-first memory extraction framework for AI agents. Markdown is the single source of truth; SQLite holds state and LanceDB provides the rebuildable vector + BM25 + scalar index. The codebase follows a single-direction DDD layering (entrypoints -> service -> memory -> infra, with component / core / config cross-cutting) enforced by import-linter. Engineering surface: - Coding conventions in .claude/rules/ (path-scoped) and workflows in .claude/skills/ (/commit, /new-branch, /pr). - GitHub Actions CI runs make lint + test + integration; pre-commit mirrors the gates locally (ruff, hygiene hooks, gitlint commit-msg). - Commit messages follow Conventional Commits, enforced by gitlint. - make lint also enforces datetime two-zone discipline and OpenAPI drift.
53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
"""Repository for ``memcell`` table — singleton bound to ``sqlite_manager``.
|
|
|
|
Pure persistence: callers build the SQLModel ``Memcell`` rows (including
|
|
``message_ids_json`` / ``sender_ids_json``) and hand them in. The pipeline
|
|
is responsible for mapping algo-side messages back to everos
|
|
``message_id`` because algo's ``Message`` does not carry per-message
|
|
identifiers.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker
|
|
|
|
from everos.core.persistence.sqlite import RepoBase, session_scope
|
|
|
|
from ..sqlite_manager import get_session_factory
|
|
from ..tables import Memcell
|
|
|
|
|
|
class _MemcellRepo(RepoBase[Memcell]):
|
|
model = Memcell
|
|
|
|
def _factory_lookup(self) -> async_sessionmaker[AsyncSession]:
|
|
return get_session_factory()
|
|
|
|
async def insert_many(self, rows: list[Memcell]) -> list[Memcell]:
|
|
"""Insert MemCell rows in one transaction; rows are constructed by caller."""
|
|
async with session_scope(self._factory) as s:
|
|
s.add_all(rows)
|
|
await s.commit()
|
|
for r in rows:
|
|
await s.refresh(r)
|
|
return rows
|
|
|
|
async def find_by_ids(self, memcell_ids: list[str]) -> list[Memcell]:
|
|
"""Bulk fetch rows by primary key list — preserves caller order.
|
|
|
|
Used by offline strategies that pull every memcell in a cluster
|
|
(membership lives in :class:`ClusterMember` and is supplied to
|
|
the strategy via :class:`everalgo.clustering.Cluster.members`).
|
|
"""
|
|
if not memcell_ids:
|
|
return []
|
|
async with session_scope(self._factory) as s:
|
|
stmt = select(Memcell).where(Memcell.memcell_id.in_(memcell_ids))
|
|
rows = list((await s.execute(stmt)).scalars().all())
|
|
by_id = {r.memcell_id: r for r in rows}
|
|
return [by_id[mid] for mid in memcell_ids if mid in by_id]
|
|
|
|
|
|
memcell_repo = _MemcellRepo()
|