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.
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import pydantic
|
|
import pytest
|
|
from everalgo.types import ChatMessage, MemCell
|
|
|
|
from everos.memory.events import AgentPipelineStarted, UserPipelineStarted
|
|
|
|
|
|
def _sample_memcell() -> MemCell:
|
|
return MemCell(
|
|
items=[
|
|
ChatMessage(
|
|
id="m1",
|
|
role="user",
|
|
content="hello",
|
|
timestamp=1_700_000_000_000,
|
|
sender_id="u1",
|
|
),
|
|
ChatMessage(
|
|
id="m2",
|
|
role="assistant",
|
|
content="hi back",
|
|
timestamp=1_700_000_001_000,
|
|
sender_id="agent",
|
|
),
|
|
],
|
|
timestamp=1_700_000_001_000,
|
|
)
|
|
|
|
|
|
def test_user_pipeline_started_topic_is_module_qualified() -> None:
|
|
assert UserPipelineStarted.topic() == "everos.memory.events:UserPipelineStarted"
|
|
|
|
|
|
def test_agent_pipeline_started_topic_is_module_qualified() -> None:
|
|
assert AgentPipelineStarted.topic() == "everos.memory.events:AgentPipelineStarted"
|
|
|
|
|
|
def test_user_pipeline_started_roundtrip_json() -> None:
|
|
event = UserPipelineStarted(
|
|
memcell_id="mc_a", session_id="s1", memcell=_sample_memcell()
|
|
)
|
|
restored = UserPipelineStarted.model_validate_json(event.model_dump_json())
|
|
assert restored.memcell_id == "mc_a"
|
|
assert restored.session_id == "s1"
|
|
|
|
|
|
def test_user_pipeline_started_is_frozen_and_extra_forbid() -> None:
|
|
event = UserPipelineStarted(
|
|
memcell_id="mc_a",
|
|
session_id="s1",
|
|
memcell=_sample_memcell(),
|
|
)
|
|
with pytest.raises(pydantic.ValidationError):
|
|
UserPipelineStarted( # type: ignore[call-arg]
|
|
memcell_id="mc_a",
|
|
session_id="s1",
|
|
memcell=_sample_memcell(),
|
|
extra_field=1,
|
|
)
|
|
with pytest.raises(pydantic.ValidationError):
|
|
event.memcell_id = "mc_b" # type: ignore[misc]
|
|
|
|
|
|
def test_user_pipeline_started_carries_memcell() -> None:
|
|
event = UserPipelineStarted(
|
|
memcell_id="mc_a",
|
|
session_id="s1",
|
|
memcell=_sample_memcell(),
|
|
)
|
|
assert event.memcell.items[0].content == "hello"
|
|
assert event.memcell.items[1].sender_id == "agent"
|
|
|
|
|
|
def test_user_pipeline_started_nested_roundtrip_json() -> None:
|
|
event = UserPipelineStarted(
|
|
memcell_id="mc_a",
|
|
session_id="s1",
|
|
memcell=_sample_memcell(),
|
|
)
|
|
restored = UserPipelineStarted.model_validate_json(event.model_dump_json())
|
|
assert restored.memcell.items[0].id == "m1"
|
|
assert restored.memcell.items[1].content == "hi back"
|
|
assert restored.memcell.timestamp == 1_700_000_001_000
|