Repositories and snapshots

import asyncio

from core.state.canonical.state import CanonicalState
from core.state.repository.memory import InMemoryStateRepository


async def main() -> None:
    repository = InMemoryStateRepository[CanonicalState]()

    state = CanonicalState()
    state.set_field("status", "draft")

    await repository.save("registration-17", state)
    restored = await repository.get("registration-17")

    assert restored is state
    assert await repository.exists("registration-17")
    assert await repository.delete("registration-17")


asyncio.run(main())

import asyncio

from core.state.interpretive.schema import ValueConfidence
from core.state.interpretive.state import InterpretiveState
from core.state.snapshot.memory import InMemorySnapshotStore


async def main() -> None:
    store = InMemorySnapshotStore[InterpretiveState](max_snapshots=2)
    state = InterpretiveState()

    state.add_value("status", ValueConfidence(value="draft", confidence=0.7))
    await store.record_snapshot("extractor-1", state)

    state.add_value("status", ValueConfidence(value="confirmed", confidence=0.9))
    await store.record_snapshot("extractor-1", state)

    latest = await store.get_latest()
    assert latest is not None
    assert latest.index == 1
    assert (await store.count()) == 2


asyncio.run(main())