wehub-resource-sync
6c9c7fe7f3
CI / integration tests (3.13) (push) Failing after 1s
Commit lint / pull request title (push) Has been skipped
Docs / links (push) Failing after 1s
CI / unit tests (3.13) (push) Failing after 1s
CI / lint (push) Failing after 1s
CI / integration tests (push) Failing after 1s
CI / package build (push) Failing after 1s
Commit lint / commit messages (push) Failing after 1s
CI / unit tests (push) Failing after 1s
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
|