@elizaos/python 2.0.0-alpha.27 → 2.0.0-alpha.30

package/pyproject.toml

@@ -25,7 +25,7 @@ classifiers = [
 dependencies = [
     "pydantic>=2.12.5",
     "pydantic-settings>=2.12.0",
-    "protobuf>=5.29.0",
+    "protobuf>=7.34.0",
     "uuid6>=2024.1.12",
     "aiohttp>=3.13.3",
     "cryptography>=42.0.0",
@@ -140,4 +140,3 @@ known-first-party = ["elizaos"]
 line-length = 100
 target-version = ["py311"]
 exclude = "elizaos/types/generated/"
-

package/requirements-dev.lock

@@ -1,5 +1,5 @@
 #
-# This file is autogenerated by pip-compile with Python 3.13
+# This file is autogenerated by pip-compile with Python 3.14
 # by the following command:
 #
 #    pip-compile --output-file=requirements-dev.lock requirements-dev.in
@@ -71,7 +71,7 @@ propcache==0.4.1
     # via
     #   aiohttp
     #   yarl
-protobuf==6.33.4
+protobuf==7.34.0
     # via -r requirements.in
 psutil==7.2.1
     # via pytest-xprocess

package/requirements.in

@@ -1,9 +1,8 @@
 pydantic>=2.12.5
 pydantic-settings>=2.12.0
-protobuf>=5.29.0
+protobuf>=7.34.0
 uuid6>=2024.1.12
 aiohttp>=3.13.3
 cryptography>=42.0.0
 structlog>=25.5.0
 typing-extensions>=4.9.0
-

package/requirements.lock

@@ -1,5 +1,5 @@
 #
-# This file is autogenerated by pip-compile with Python 3.13
+# This file is autogenerated by pip-compile with Python 3.14
 # by the following command:
 #
 #    pip-compile --output-file=requirements.lock requirements.in
@@ -32,7 +32,7 @@ propcache==0.4.1
     # via
     #   aiohttp
     #   yarl
-protobuf==6.33.4
+protobuf==7.34.0
     # via -r requirements.in
 pycparser==2.23
     # via cffi

package/tests/test_history_compaction.py

@@ -0,0 +1,104 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from types import SimpleNamespace
+from unittest.mock import AsyncMock
+
+import pytest
+
+from elizaos.advanced_memory.actions.reset_session import (
+    reset_session_action as advanced_reset_session_action,
+)
+from elizaos.basic_capabilities.providers.recent_messages import (
+    recent_messages_provider as basic_recent_messages_provider,
+)
+from elizaos.bootstrap.actions.reset_session import (
+    reset_session_action as bootstrap_reset_session_action,
+)
+from elizaos.bootstrap.providers.recent_messages import (
+    recent_messages_provider as bootstrap_recent_messages_provider,
+)
+from elizaos.types import Content, Memory, as_uuid
+
+
+@dataclass
+class FakeRoom:
+    id: object
+    world_id: object | None
+    metadata: dict[str, object]
+
+
+class FakeRuntime:
+    def __init__(self, room: FakeRoom, *, role: str = "OWNER") -> None:
+        self.room = room
+        self.updated_room: FakeRoom | None = None
+        self.get_room = AsyncMock(return_value=room)
+        self.update_room = AsyncMock(side_effect=self._update_room)
+        self.get_memories = AsyncMock(return_value=[])
+        self.get_entity = AsyncMock(return_value=SimpleNamespace(name="User"))
+        self.agent_id = as_uuid("91000000-0000-0000-0000-000000000001")
+        self.character = SimpleNamespace(name="CompactionAgent")
+        self.get_world = AsyncMock(
+            return_value=SimpleNamespace(
+                metadata={"roles": {"91000000-0000-0000-0000-000000000002": role}}
+            )
+        )
+
+    async def _update_room(self, room: FakeRoom) -> None:
+        self.updated_room = room
+        self.room = room
+
+
+def _message() -> Memory:
+    return Memory(
+        id=as_uuid("91000000-0000-0000-0000-000000000010"),
+        entity_id=as_uuid("91000000-0000-0000-0000-000000000002"),
+        room_id=as_uuid("91000000-0000-0000-0000-000000000003"),
+        content=Content(text="start over", source="test"),
+    )
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    "action_under_test",
+    [bootstrap_reset_session_action, advanced_reset_session_action],
+)
+async def test_reset_session_action_updates_room_metadata(action_under_test: object) -> None:
+    room = FakeRoom(
+        id=as_uuid("91000000-0000-0000-0000-000000000003"),
+        world_id=as_uuid("91000000-0000-0000-0000-000000000004"),
+        metadata={
+            "lastCompactionAt": 1000,
+            "compactionHistory": [{"timestamp": 1000, "triggeredBy": "old", "reason": "manual"}],
+        },
+    )
+    runtime = FakeRuntime(room)
+
+    result = await action_under_test.handler(runtime, _message(), None, None, None, None)
+
+    assert result.success is True
+    assert runtime.updated_room is not None
+    assert runtime.updated_room.metadata["lastCompactionAt"] >= 1000
+    assert len(runtime.updated_room.metadata["compactionHistory"]) == 2
+    assert result.values["previousCompactionAt"] == 1000
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    "provider_under_test",
+    [bootstrap_recent_messages_provider, basic_recent_messages_provider],
+)
+async def test_recent_messages_provider_uses_last_compaction_boundary(
+    provider_under_test: object,
+) -> None:
+    room = FakeRoom(
+        id=as_uuid("91000000-0000-0000-0000-000000000003"),
+        world_id=None,
+        metadata={"lastCompactionAt": 4242},
+    )
+    runtime = FakeRuntime(room)
+
+    result = await provider_under_test.get(runtime, _message(), None)
+
+    assert result.values["roomId"] == str(_message().room_id)
+    assert runtime.get_memories.await_args.kwargs["start"] == 4242

package/tests/test_memory_bounds.py

@@ -0,0 +1,115 @@
+from __future__ import annotations
+
+import uuid
+from types import SimpleNamespace
+from unittest.mock import AsyncMock, MagicMock
+
+import pytest
+
+from elizaos.advanced_memory.memory_service import MemoryService
+from elizaos.advanced_memory.types import LongTermMemoryCategory
+from elizaos.basic_capabilities.services.embedding import (
+    EmbeddingService as BasicEmbeddingService,
+)
+from elizaos.bootstrap.services.embedding import (
+    EmbeddingService as BootstrapEmbeddingService,
+)
+from elizaos.types import ModelType
+
+
+def _mock_runtime() -> MagicMock:
+    runtime = MagicMock()
+    runtime.agent_id = uuid.uuid4()
+    runtime.logger = MagicMock()
+
+    async def use_model(model_type, **_kwargs):
+        if model_type == ModelType.TEXT_EMBEDDING:
+            return [0.1, 0.2, 0.3]
+        return "ok"
+
+    runtime.use_model = AsyncMock(side_effect=use_model)
+    return runtime
+
+
+@pytest.mark.asyncio
+async def test_embedding_services_use_lru_eviction() -> None:
+    for service_cls in (BasicEmbeddingService, BootstrapEmbeddingService):
+        runtime = _mock_runtime()
+        service = service_cls()
+        service._runtime = runtime
+        service.set_max_cache_size(2)
+
+        await service.embed("a")
+        await service.embed("b")
+        await service.embed("a")
+        await service.embed("c")
+
+        assert list(service._cache.keys()) == ["a", "c"]
+
+
+@pytest.mark.asyncio
+async def test_bootstrap_embedding_queue_deduplicates_and_clears_on_stop() -> None:
+    service = BootstrapEmbeddingService()
+    service._runtime = _mock_runtime()
+
+    payload = SimpleNamespace(extra={"memory": {"id": "memory-1"}})
+
+    await service._handle_embedding_request(payload)
+    await service._handle_embedding_request(payload)
+
+    assert service._queue.qsize() == 1
+    assert service._pending_payload_keys == {"memory-1"}
+
+    await service.stop()
+
+    assert service._queue.qsize() == 0
+    assert service._pending_payload_keys == set()
+
+
+@pytest.mark.asyncio
+async def test_advanced_memory_fallback_storage_is_bounded() -> None:
+    service = MemoryService(runtime=None)
+    agent_id = uuid.uuid4()
+
+    for index in range(service._MAX_LOCAL_SESSION_SUMMARIES + 25):
+        await service.store_session_summary(
+            agent_id=agent_id,
+            room_id=uuid.uuid4(),
+            summary=f"summary-{index}",
+            message_count=index,
+            last_message_offset=index,
+        )
+
+    assert len(service._session_summaries) == service._MAX_LOCAL_SESSION_SUMMARIES
+
+    for index in range(service._MAX_LOCAL_EXTRACTION_CHECKPOINTS + 25):
+        await service.set_last_extraction_checkpoint(uuid.uuid4(), uuid.uuid4(), index)
+
+    assert len(service._extraction_checkpoints) == service._MAX_LOCAL_EXTRACTION_CHECKPOINTS
+
+    for _ in range(service._MAX_LOCAL_LONG_TERM_ENTITIES + 25):
+        entity_id = uuid.uuid4()
+        await service.store_long_term_memory(
+            agent_id=agent_id,
+            entity_id=entity_id,
+            category=LongTermMemoryCategory.SEMANTIC,
+            content=f"entity-{entity_id}",
+            confidence=0.9,
+        )
+
+    assert len(service._long_term) == service._MAX_LOCAL_LONG_TERM_ENTITIES
+
+    hot_entity_id = uuid.uuid4()
+    for index in range(service._MAX_LOCAL_LONG_TERM_PER_ENTITY + 25):
+        await service.store_long_term_memory(
+            agent_id=agent_id,
+            entity_id=hot_entity_id,
+            category=LongTermMemoryCategory.SEMANTIC,
+            content=f"fact-{index}",
+            confidence=float(index),
+        )
+
+    retained = service._long_term[str(hot_entity_id)]
+    assert len(retained) == service._MAX_LOCAL_LONG_TERM_PER_ENTITY
+    assert retained[0].content == "fact-25"
+    assert retained[-1].content == f"fact-{service._MAX_LOCAL_LONG_TERM_PER_ENTITY + 24}"

package/tests/test_validation.py

@@ -128,7 +128,7 @@ def test_validate_action_regex():
 
     # Unicode characters
     msg = create_mock_memory("Transfer 100 €")
-    assert not validate_action_regex(msg, [], r"transfer \d+ €") # case sensitive
+    assert not validate_action_regex(msg, [], r"transfer \d+ €")  # case sensitive
     assert validate_action_regex(msg, [], r"(?i)transfer \d+ €")
 
     # Special characters

package/uv.lock

@@ -485,7 +485,7 @@ requires-dist = [
     { name = "isort", marker = "extra == 'dev'", specifier = ">=5.13.0" },
     { name = "mypy", marker = "extra == 'dev'", specifier = ">=1.19.1" },
     { name = "pip-tools", marker = "extra == 'dev'", specifier = ">=7.4.1" },
-    { name = "protobuf", specifier = ">=5.29.0" },
+    { name = "protobuf", specifier = ">=7.34.0" },
     { name = "pydantic", specifier = ">=2.12.5" },
     { name = "pydantic-settings", specifier = ">=2.12.0" },
     { name = "pytest", marker = "extra == 'dev'", specifier = ">=9.0.2" },
@@ -1022,17 +1022,17 @@ wheels = [
 
 [[package]]
 name = "protobuf"
-version = "6.33.5"
+version = "7.34.0"
 source = { registry = "https://pypi.org/simple" }
-sdist = { url = "https://files.pythonhosted.org/packages/ba/25/7c72c307aafc96fa87062aa6291d9f7c94836e43214d43722e86037aac02/protobuf-6.33.5.tar.gz", hash = "sha256:6ddcac2a081f8b7b9642c09406bc6a4290128fce5f471cddd165960bb9119e5c", size = 444465, upload-time = "2026-01-29T21:51:33.494Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/f2/00/04a2ab36b70a52d0356852979e08b44edde0435f2115dc66e25f2100f3ab/protobuf-7.34.0.tar.gz", hash = "sha256:3871a3df67c710aaf7bb8d214cc997342e63ceebd940c8c7fc65c9b3d697591a", size = 454726, upload-time = "2026-02-27T00:30:25.421Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/b1/79/af92d0a8369732b027e6d6084251dd8e782c685c72da161bd4a2e00fbabb/protobuf-6.33.5-cp310-abi3-win32.whl", hash = "sha256:d71b040839446bac0f4d162e758bea99c8251161dae9d0983a3b88dee345153b", size = 425769, upload-time = "2026-01-29T21:51:21.751Z" },
-    { url = "https://files.pythonhosted.org/packages/55/75/bb9bc917d10e9ee13dee8607eb9ab963b7cf8be607c46e7862c748aa2af7/protobuf-6.33.5-cp310-abi3-win_amd64.whl", hash = "sha256:3093804752167bcab3998bec9f1048baae6e29505adaf1afd14a37bddede533c", size = 437118, upload-time = "2026-01-29T21:51:24.022Z" },
-    { url = "https://files.pythonhosted.org/packages/a2/6b/e48dfc1191bc5b52950246275bf4089773e91cb5ba3592621723cdddca62/protobuf-6.33.5-cp39-abi3-macosx_10_9_universal2.whl", hash = "sha256:a5cb85982d95d906df1e2210e58f8e4f1e3cdc088e52c921a041f9c9a0386de5", size = 427766, upload-time = "2026-01-29T21:51:25.413Z" },
-    { url = "https://files.pythonhosted.org/packages/4e/b1/c79468184310de09d75095ed1314b839eb2f72df71097db9d1404a1b2717/protobuf-6.33.5-cp39-abi3-manylinux2014_aarch64.whl", hash = "sha256:9b71e0281f36f179d00cbcb119cb19dec4d14a81393e5ea220f64b286173e190", size = 324638, upload-time = "2026-01-29T21:51:26.423Z" },
-    { url = "https://files.pythonhosted.org/packages/c5/f5/65d838092fd01c44d16037953fd4c2cc851e783de9b8f02b27ec4ffd906f/protobuf-6.33.5-cp39-abi3-manylinux2014_s390x.whl", hash = "sha256:8afa18e1d6d20af15b417e728e9f60f3aa108ee76f23c3b2c07a2c3b546d3afd", size = 339411, upload-time = "2026-01-29T21:51:27.446Z" },
-    { url = "https://files.pythonhosted.org/packages/9b/53/a9443aa3ca9ba8724fdfa02dd1887c1bcd8e89556b715cfbacca6b63dbec/protobuf-6.33.5-cp39-abi3-manylinux2014_x86_64.whl", hash = "sha256:cbf16ba3350fb7b889fca858fb215967792dc125b35c7976ca4818bee3521cf0", size = 323465, upload-time = "2026-01-29T21:51:28.925Z" },
-    { url = "https://files.pythonhosted.org/packages/57/bf/2086963c69bdac3d7cff1cc7ff79b8ce5ea0bec6797a017e1be338a46248/protobuf-6.33.5-py3-none-any.whl", hash = "sha256:69915a973dd0f60f31a08b8318b73eab2bd6a392c79184b3612226b0a3f8ec02", size = 170687, upload-time = "2026-01-29T21:51:32.557Z" },
+    { url = "https://files.pythonhosted.org/packages/13/c4/6322ab5c8f279c4c358bc14eb8aefc0550b97222a39f04eb3c1af7a830fa/protobuf-7.34.0-cp310-abi3-macosx_10_9_universal2.whl", hash = "sha256:8e329966799f2c271d5e05e236459fe1cbfdb8755aaa3b0914fa60947ddea408", size = 429248, upload-time = "2026-02-27T00:30:14.924Z" },
+    { url = "https://files.pythonhosted.org/packages/45/99/b029bbbc61e8937545da5b79aa405ab2d9cf307a728f8c9459ad60d7a481/protobuf-7.34.0-cp310-abi3-manylinux2014_aarch64.whl", hash = "sha256:9d7a5005fb96f3c1e64f397f91500b0eb371b28da81296ae73a6b08a5b76cdd6", size = 325753, upload-time = "2026-02-27T00:30:17.247Z" },
+    { url = "https://files.pythonhosted.org/packages/cc/79/09f02671eb75b251c5550a1c48e7b3d4b0623efd7c95a15a50f6f9fc1e2e/protobuf-7.34.0-cp310-abi3-manylinux2014_s390x.whl", hash = "sha256:4a72a8ec94e7a9f7ef7fe818ed26d073305f347f8b3b5ba31e22f81fd85fca02", size = 340200, upload-time = "2026-02-27T00:30:18.672Z" },
+    { url = "https://files.pythonhosted.org/packages/b5/57/89727baef7578897af5ed166735ceb315819f1c184da8c3441271dbcfde7/protobuf-7.34.0-cp310-abi3-manylinux2014_x86_64.whl", hash = "sha256:964cf977e07f479c0697964e83deda72bcbc75c3badab506fb061b352d991b01", size = 324268, upload-time = "2026-02-27T00:30:20.088Z" },
+    { url = "https://files.pythonhosted.org/packages/1f/3e/38ff2ddee5cc946f575c9d8cc822e34bde205cf61acf8099ad88ef19d7d2/protobuf-7.34.0-cp310-abi3-win32.whl", hash = "sha256:f791ec509707a1d91bd02e07df157e75e4fb9fbdad12a81b7396201ec244e2e3", size = 426628, upload-time = "2026-02-27T00:30:21.555Z" },
+    { url = "https://files.pythonhosted.org/packages/cb/71/7c32eaf34a61a1bae1b62a2ac4ffe09b8d1bb0cf93ad505f42040023db89/protobuf-7.34.0-cp310-abi3-win_amd64.whl", hash = "sha256:9f9079f1dde4e32342ecbd1c118d76367090d4aaa19da78230c38101c5b3dd40", size = 437901, upload-time = "2026-02-27T00:30:22.836Z" },
+    { url = "https://files.pythonhosted.org/packages/a4/e7/14dc9366696dcb53a413449881743426ed289d687bcf3d5aee4726c32ebb/protobuf-7.34.0-py3-none-any.whl", hash = "sha256:e3b914dd77fa33fa06ab2baa97937746ab25695f389869afdf03e81f34e45dc7", size = 170716, upload-time = "2026-02-27T00:30:23.994Z" },
 ]
 
 [[package]]