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

package/elizaos/basic_capabilities/providers/contacts.py

@@ -30,7 +30,7 @@ async def get_contacts_context(
 
     contact_details: list[dict[str, str]] = []
     for contact in contacts:
-        entity = await runtime.get_entity(contact.entity_id)
+        entity = await runtime.get_entity(str(contact.entity_id))
         name = entity.name if entity and entity.name else "Unknown"
         contact_details.append(
             {

package/elizaos/basic_capabilities/providers/follow_ups.py

@@ -38,7 +38,7 @@ async def get_follow_ups_context(
 
     for task in upcoming:
         scheduled = datetime.fromisoformat(task.scheduled_at.replace("Z", "+00:00"))
-        entity = await runtime.get_entity(task.entity_id)
+        entity = await runtime.get_entity(str(task.entity_id))
         name = entity.name if entity and entity.name else "Unknown"
 
         item = {

package/elizaos/basic_capabilities/providers/knowledge.py

@@ -4,7 +4,6 @@ from typing import TYPE_CHECKING
 
 from elizaos.generated.spec_helpers import require_provider_spec
 from elizaos.types import Provider, ProviderResult
-from elizaos.types.database import MemorySearchOptions
 
 if TYPE_CHECKING:
     from elizaos.types import IAgentRuntime, Memory, State
@@ -42,14 +41,14 @@ async def get_knowledge_context(
     # 3. Search using the most recent embedding if available
     if embeddings:
         primary_embedding = embeddings[0]
-        params = MemorySearchOptions(
-            table_name="knowledge",
-            room_id=message.room_id,
-            embedding=primary_embedding,
-            match_threshold=0.75,
-            match_count=5,
-            unique=True,
-        )
+        params = {
+            "tableName": "knowledge",
+            "roomId": str(message.room_id),
+            "embedding": primary_embedding,
+            "matchThreshold": 0.75,
+            "matchCount": 5,
+            "unique": True,
+        }
         relevant_knowledge = await runtime.search_memories(params)
     elif query_text:
         # Fallback to search_knowledge if no embeddings found?

package/elizaos/basic_capabilities/providers/recent_messages.py

@@ -25,12 +25,17 @@ async def get_recent_messages_context(
 
     sections: list[str] = []
     message_list: list[dict[str, str | int]] = []
+    room = await runtime.get_room(room_id)
+    last_compaction_at = None
+    if room and getattr(room, "metadata", None):
+        last_compaction_at = room.metadata.get("lastCompactionAt")
 
     recent_messages = await runtime.get_memories(
         room_id=room_id,
         limit=20,
         order_by="created_at",
         order_direction="desc",
+        start=last_compaction_at,
     )
 
     recent_messages = list(reversed(recent_messages))

package/elizaos/basic_capabilities/providers/relationships.py

@@ -39,11 +39,10 @@ async def get_relationships(
         )
 
     try:
-        relationships = await runtime.get_relationships(entity_id=entity_id)
+        relationships = await runtime.get_relationships({"entityId": str(entity_id)})
     except Exception as e:
         runtime.logger.debug(
-            {"src": "provider:relationships", "agentId": runtime.agent_id, "error": str(e)},
-            "Failed to get relationships",
+            f"Failed to get relationships: src=provider:relationships agentId={runtime.agent_id} error={e}"
         )
         relationships = []
 
@@ -54,24 +53,31 @@ async def get_relationships(
             data={"relationships": []},
         )
 
-    sorted_relationships = sorted(
-        relationships,
-        key=lambda r: (r.get("metadata", {}) or {}).get("interactions", 0),
-        reverse=True,
-    )[:30]
+    def _get_interactions(relationship: object) -> int:
+        if isinstance(relationship, dict):
+            metadata = relationship.get("metadata", {})
+            if isinstance(metadata, dict):
+                value = metadata.get("interactions", 0)
+                return int(value) if isinstance(value, (int, float)) else 0
+        return 0
+
+    sorted_relationships = sorted(relationships, key=_get_interactions, reverse=True)[:30]
 
     formatted_relationships: list[str] = []
-    entity_cache: dict[object, str] = {}
+    entity_cache: dict[str, str] = {}
     for rel in sorted_relationships:
+        if not isinstance(rel, dict):
+            continue
         target_id = rel.get("targetEntityId")
         if not target_id:
             continue
 
-        target_name = entity_cache.get(target_id)
+        target_id_str = str(target_id)
+        target_name = entity_cache.get(target_id_str)
         if target_name is None:
-            target_entity = await runtime.get_entity(target_id)
-            target_name = target_entity.name if target_entity else str(target_id)[:8]
-            entity_cache[target_id] = target_name
+            target_entity = await runtime.get_entity(target_id_str)
+            target_name = target_entity.name if target_entity else target_id_str[:8]
+            entity_cache[target_id_str] = target_name
 
         formatted_relationships.append(format_relationship(rel, target_name))
 

package/elizaos/basic_capabilities/services/embedding.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+from collections import OrderedDict
 from typing import TYPE_CHECKING
 
 from elizaos.types import ModelType, Service, ServiceType
@@ -18,7 +19,7 @@ class EmbeddingService(Service):
 
     def __init__(self) -> None:
         self._runtime: IAgentRuntime | None = None
-        self._cache: dict[str, list[float]] = {}
+        self._cache: OrderedDict[str, list[float]] = OrderedDict()
         self._cache_enabled: bool = True
         self._max_cache_size: int = 1000
 
@@ -51,7 +52,9 @@ class EmbeddingService(Service):
             raise ValueError("Embedding service not started - no runtime available")
 
         if self._cache_enabled and text in self._cache:
-            return self._cache[text]
+            embedding = self._cache.pop(text)
+            self._cache[text] = embedding
+            return embedding
 
         # Truncate to stay within embedding model token limits
         embed_text = text
@@ -87,9 +90,10 @@ class EmbeddingService(Service):
         return embeddings
 
     def _add_to_cache(self, text: str, embedding: list[float]) -> None:
-        if len(self._cache) >= self._max_cache_size:
-            oldest_key = next(iter(self._cache))
-            del self._cache[oldest_key]
+        if text in self._cache:
+            self._cache.pop(text)
+        elif len(self._cache) >= self._max_cache_size:
+            self._cache.popitem(last=False)
         self._cache[text] = embedding
 
     def clear_cache(self) -> None:
@@ -105,8 +109,7 @@ class EmbeddingService(Service):
             raise ValueError("Cache size must be positive")
         self._max_cache_size = size
         while len(self._cache) > self._max_cache_size:
-            oldest_key = next(iter(self._cache))
-            del self._cache[oldest_key]
+            self._cache.popitem(last=False)
 
     async def similarity(self, text1: str, text2: str) -> float:
         embedding1 = await self.embed(text1)

package/elizaos/basic_capabilities/services/task.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from dataclasses import dataclass, field
 from datetime import UTC, datetime
-from enum import Enum
+from enum import StrEnum
 from typing import TYPE_CHECKING
 from uuid import UUID, uuid4
 
@@ -12,7 +12,7 @@ if TYPE_CHECKING:
     from elizaos.types import IAgentRuntime
 
 
-class TaskStatus(str, Enum):
+class TaskStatus(StrEnum):
     PENDING = "PENDING"
     IN_PROGRESS = "IN_PROGRESS"
     COMPLETED = "COMPLETED"
@@ -20,7 +20,7 @@ class TaskStatus(str, Enum):
     CANCELLED = "CANCELLED"
 
 
-class TaskPriority(str, Enum):
+class TaskPriority(StrEnum):
     LOW = "LOW"
     MEDIUM = "MEDIUM"
     HIGH = "HIGH"

package/elizaos/bootstrap/actions/__init__.py

@@ -7,6 +7,7 @@ from .mute_room import mute_room_action
 from .none import none_action
 from .remove_contact import remove_contact_action
 from .reply import reply_action
+from .reset_session import reset_session_action
 from .roles import update_role_action
 from .schedule_follow_up import schedule_follow_up_action
 from .search_contacts import search_contacts_action
@@ -26,6 +27,7 @@ __all__ = [
     "mute_room_action",
     "none_action",
     "remove_contact_action",
+    "reset_session_action",
     "reply_action",
     "schedule_follow_up_action",
     "search_contacts_action",
@@ -54,6 +56,7 @@ EXTENDED_ACTIONS = [
     generate_image_action,
     mute_room_action,
     remove_contact_action,
+    reset_session_action,
     schedule_follow_up_action,
     search_contacts_action,
     send_message_action,

package/elizaos/bootstrap/actions/reset_session.py

@@ -0,0 +1,3 @@
+from elizaos.advanced_memory.actions.reset_session import reset_session_action
+
+__all__ = ["reset_session_action"]

package/elizaos/bootstrap/actions/roles.py

@@ -1,12 +1,13 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from enum import Enum
-from typing import TYPE_CHECKING
+from enum import StrEnum
+from typing import TYPE_CHECKING, Any, cast
 from uuid import UUID
 
 from elizaos.bootstrap.utils.xml import parse_key_value_xml
 from elizaos.generated.spec_helpers import require_action_spec
+from elizaos.prompts import UPDATE_ROLE_TEMPLATE
 from elizaos.types import Action, ActionExample, ActionResult, Content, ModelType
 
 if TYPE_CHECKING:
@@ -18,7 +19,7 @@ _spec = require_action_spec("UPDATE_ROLE")
 
 def _convert_spec_examples() -> list[list[ActionExample]]:
     """Convert spec examples to ActionExample format."""
-    spec_examples = _spec.get("examples", [])
+    spec_examples = cast(list[list[dict[str, Any]]], _spec.get("examples", []))
     if spec_examples:
         return [
             [
@@ -36,7 +37,7 @@ def _convert_spec_examples() -> list[list[ActionExample]]:
     return []
 
 
-class Role(str, Enum):
+class Role(StrEnum):
     OWNER = "OWNER"
     ADMIN = "ADMIN"
     MEMBER = "MEMBER"

package/elizaos/bootstrap/providers/knowledge.py

@@ -4,7 +4,6 @@ from typing import TYPE_CHECKING
 
 from elizaos.generated.spec_helpers import require_provider_spec
 from elizaos.types import Provider, ProviderResult
-from elizaos.types.database import MemorySearchOptions
 
 if TYPE_CHECKING:
     from elizaos.types import IAgentRuntime, Memory, State
@@ -42,14 +41,14 @@ async def get_knowledge_context(
     # 3. Search using the most recent embedding if available
     if embeddings:
         primary_embedding = embeddings[0]
-        params = MemorySearchOptions(
-            table_name="knowledge",
-            room_id=message.room_id,
-            embedding=primary_embedding,
-            match_threshold=0.75,
-            match_count=5,
-            unique=True,
-        )
+        params = {
+            "tableName": "knowledge",
+            "roomId": str(message.room_id),
+            "embedding": primary_embedding,
+            "matchThreshold": 0.75,
+            "matchCount": 5,
+            "unique": True,
+        }
         relevant_knowledge = await runtime.search_memories(params)
     elif query_text:
         # Fallback to search_knowledge if no embeddings found?

package/elizaos/bootstrap/providers/recent_messages.py

@@ -25,12 +25,17 @@ async def get_recent_messages_context(
 
     sections: list[str] = []
     message_list: list[dict[str, str | int]] = []
+    room = await runtime.get_room(room_id)
+    last_compaction_at = None
+    if room and getattr(room, "metadata", None):
+        last_compaction_at = room.metadata.get("lastCompactionAt")
 
     recent_messages = await runtime.get_memories(
         room_id=room_id,
         limit=20,
         order_by="created_at",
         order_direction="desc",
+        start=last_compaction_at,
     )
 
     recent_messages = list(reversed(recent_messages))

package/elizaos/bootstrap/providers/relationships.py

@@ -39,11 +39,10 @@ async def get_relationships(
         )
 
     try:
-        relationships = await runtime.get_relationships(entity_id=entity_id)
+        relationships = await runtime.get_relationships({"entityId": str(entity_id)})
     except Exception as e:
         runtime.logger.debug(
-            {"src": "provider:relationships", "agentId": runtime.agent_id, "error": str(e)},
-            "Failed to get relationships",
+            f"Failed to get relationships: src=provider:relationships agentId={runtime.agent_id} error={e}"
         )
         relationships = []
 
@@ -54,24 +53,31 @@ async def get_relationships(
             data={"relationships": []},
         )
 
-    sorted_relationships = sorted(
-        relationships,
-        key=lambda r: (r.get("metadata", {}) or {}).get("interactions", 0),
-        reverse=True,
-    )[:30]
+    def _get_interactions(relationship: object) -> int:
+        if isinstance(relationship, dict):
+            metadata = relationship.get("metadata", {})
+            if isinstance(metadata, dict):
+                value = metadata.get("interactions", 0)
+                return int(value) if isinstance(value, (int, float)) else 0
+        return 0
+
+    sorted_relationships = sorted(relationships, key=_get_interactions, reverse=True)[:30]
 
     formatted_relationships: list[str] = []
-    entity_cache: dict[object, str] = {}
+    entity_cache: dict[str, str] = {}
     for rel in sorted_relationships:
+        if not isinstance(rel, dict):
+            continue
         target_id = rel.get("targetEntityId")
         if not target_id:
             continue
 
-        target_name = entity_cache.get(target_id)
+        target_id_str = str(target_id)
+        target_name = entity_cache.get(target_id_str)
         if target_name is None:
-            target_entity = await runtime.get_entity(target_id)
-            target_name = target_entity.name if target_entity else str(target_id)[:8]
-            entity_cache[target_id] = target_name
+            target_entity = await runtime.get_entity(target_id_str)
+            target_name = target_entity.name if target_entity else target_id_str[:8]
+            entity_cache[target_id_str] = target_name
 
         formatted_relationships.append(format_relationship(rel, target_name))
 

package/elizaos/bootstrap/services/embedding.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import asyncio
 import contextlib
+from collections import OrderedDict
 from typing import TYPE_CHECKING, Any
 
 from elizaos.types import ModelType, Service, ServiceType
@@ -21,10 +22,14 @@ class EmbeddingService(Service):
 
     def __init__(self) -> None:
         self._runtime: IAgentRuntime | None = None
-        self._cache: dict[str, list[float]] = {}
+        self._cache: OrderedDict[str, list[float]] = OrderedDict()
         self._cache_enabled: bool = True
         self._max_cache_size: int = 1000
-        self._queue: asyncio.Queue = asyncio.Queue()
+        self._queue_max_size: int = 1000
+        self._queue: asyncio.Queue[tuple[str | None, Any]] = asyncio.Queue(
+            maxsize=self._queue_max_size
+        )
+        self._pending_payload_keys: set[str] = set()
         self._worker_task: asyncio.Task | None = None
 
     @classmethod
@@ -60,6 +65,8 @@ class EmbeddingService(Service):
                 agentId=str(self._runtime.agent_id),
             )
         self._cache.clear()
+        self._pending_payload_keys.clear()
+        self._queue = asyncio.Queue(maxsize=self._queue_max_size)
         self._runtime = None
 
     # Max characters for embedding input (~8K tokens at ~4 chars/token)
@@ -70,7 +77,9 @@ class EmbeddingService(Service):
             raise ValueError("Embedding service not started - no runtime available")
 
         if self._cache_enabled and text in self._cache:
-            return self._cache[text]
+            embedding = self._cache.pop(text)
+            self._cache[text] = embedding
+            return embedding
 
         # Truncate to stay within embedding model token limits
         embed_text = text
@@ -106,9 +115,10 @@ class EmbeddingService(Service):
         return embeddings
 
     def _add_to_cache(self, text: str, embedding: list[float]) -> None:
-        if len(self._cache) >= self._max_cache_size:
-            oldest_key = next(iter(self._cache))
-            del self._cache[oldest_key]
+        if text in self._cache:
+            self._cache.pop(text)
+        elif len(self._cache) >= self._max_cache_size:
+            self._cache.popitem(last=False)
         self._cache[text] = embedding
 
     def clear_cache(self) -> None:
@@ -124,8 +134,7 @@ class EmbeddingService(Service):
             raise ValueError("Cache size must be positive")
         self._max_cache_size = size
         while len(self._cache) > self._max_cache_size:
-            oldest_key = next(iter(self._cache))
-            del self._cache[oldest_key]
+            self._cache.popitem(last=False)
 
     async def similarity(self, text1: str, text2: str) -> float:
         embedding1 = await self.embed(text1)
@@ -142,13 +151,45 @@ class EmbeddingService(Service):
 
     async def _handle_embedding_request(self, payload: Any) -> None:
         """Handle embedding generation request event."""
-        await self._queue.put(payload)
+        payload_key = self._get_payload_key(payload)
+        if payload_key is not None:
+            if payload_key in self._pending_payload_keys:
+                return
+            self._pending_payload_keys.add(payload_key)
+
+        try:
+            await self._queue.put((payload_key, payload))
+        except Exception:
+            if payload_key is not None:
+                self._pending_payload_keys.discard(payload_key)
+            raise
+
+    def _get_payload_key(self, payload: Any) -> str | None:
+        memory_data = getattr(payload, "memory", None)
+        if memory_data is None:
+            extra = getattr(payload, "extra", None)
+            if hasattr(extra, "__getitem__"):
+                with contextlib.suppress(Exception):
+                    if "memory" in extra:
+                        memory_data = extra["memory"]
+        if memory_data is None and isinstance(payload, dict):
+            memory_data = payload.get("memory")
+        if memory_data is None:
+            return None
+
+        if isinstance(memory_data, dict):
+            memory_id = memory_data.get("id")
+        else:
+            memory_id = getattr(memory_data, "id", None)
+        if memory_id is None:
+            return None
+        return str(memory_id)
 
     async def _worker(self) -> None:
         """Background worker for processing embedding requests."""
         while True:
             try:
-                payload = await self._queue.get()
+                payload_key, payload = await self._queue.get()
             except asyncio.CancelledError:
                 break
 
@@ -158,6 +199,8 @@ class EmbeddingService(Service):
                 if self._runtime:
                     self._runtime.logger.error(f"Error in embedding worker: {e}", exc_info=True)
             finally:
+                if payload_key is not None:
+                    self._pending_payload_keys.discard(payload_key)
                 self._queue.task_done()
 
     async def _process_embedding_request(self, payload: Any) -> None:

package/elizaos/bootstrap/services/rolodex.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from dataclasses import dataclass, field
 from datetime import datetime
-from enum import Enum
+from enum import StrEnum
 from typing import TYPE_CHECKING
 from uuid import UUID
 
@@ -12,7 +12,7 @@ if TYPE_CHECKING:
     from elizaos.types import IAgentRuntime
 
 
-class ContactCategory(str, Enum):
+class ContactCategory(StrEnum):
     FRIEND = "friend"
     FAMILY = "family"
     COLLEAGUE = "colleague"

package/elizaos/bootstrap/services/task.py

@@ -14,7 +14,7 @@ import asyncio
 import contextlib
 import time
 from dataclasses import dataclass
-from enum import Enum
+from enum import StrEnum
 from typing import TYPE_CHECKING, Any, Protocol, runtime_checkable
 from uuid import UUID, uuid4
 
@@ -27,7 +27,7 @@ if TYPE_CHECKING:
 TICK_INTERVAL_MS = 1000
 
 
-class TaskStatus(str, Enum):
+class TaskStatus(StrEnum):
     """Task status enum."""
 
     PENDING = "pending"
@@ -37,7 +37,7 @@ class TaskStatus(str, Enum):
     CANCELLED = "cancelled"
 
 
-class TaskPriority(str, Enum):
+class TaskPriority(StrEnum):
     """Task priority enum."""
 
     LOW = "low"

package/elizaos/media/mime.py

@@ -7,7 +7,7 @@ Provides robust MIME type detection from file buffers, headers, and extensions.
 from __future__ import annotations
 
 import re
-from enum import Enum
+from enum import StrEnum
 from urllib.parse import urlparse
 
 # Try to import python-magic for MIME sniffing, fallback to filetype
@@ -26,7 +26,7 @@ except ImportError:
     HAS_FILETYPE = False
 
 
-class MediaKind(str, Enum):
+class MediaKind(StrEnum):
     """Media kind categories."""
 
     IMAGE = "image"

package/elizaos/runtime.py

@@ -1680,11 +1680,11 @@ class AgentRuntime(IAgentRuntime):
         if self._adapter:
             await self._adapter.ensure_embedding_dimension(dimension)
 
-    async def get_entity(self, entity_id: UUID) -> Any | None:
+    async def get_entity(self, entity_id: UUID | str) -> Any | None:
         """Get a single entity by ID."""
         if not self._adapter:
             return None
-        entities = await self._adapter.get_entities_by_ids([entity_id])
+        entities = await self._adapter.get_entities_by_ids([str(entity_id)])
         return entities[0] if entities else None
 
     async def get_entities_by_ids(self, entity_ids: list[UUID]) -> list[Any] | None:
@@ -1744,7 +1744,7 @@ class AgentRuntime(IAgentRuntime):
         if self._adapter:
             await self._adapter.delete_component(component_id)
 
-    async def search_memories(self, params: MemorySearchOptions) -> list[Memory]:
+    async def search_memories(self, params: MemorySearchOptions | dict[str, Any]) -> list[Memory]:
         """Search memories by embedding."""
         if not self._adapter:
             raise RuntimeError("Database adapter not set")
@@ -1828,11 +1828,6 @@ class AgentRuntime(IAgentRuntime):
             return AgentRunSummaryResult(runs=[], total=0, has_more=False)
         return await self._adapter.get_agent_run_summaries(params)
 
-    async def search_memories(self, params: dict[str, Any]) -> list[Any]:
-        if not self._adapter:
-            return []
-        return await self._adapter.search_memories(params)
-
     async def create_memory(
         self,
         memory: dict[str, object] | None = None,
@@ -2412,6 +2407,12 @@ end code: {final_code}
                         step_id = CURRENT_TRAJECTORY_STEP_ID.get()
                         traj_svc = self.get_service("trajectory_logger")
                         if step_id and traj_svc is not None and hasattr(traj_svc, "log_llm_call"):
+                            max_tokens_value = params.get("maxTokens", 0)
+                            max_tokens = (
+                                int(max_tokens_value)
+                                if isinstance(max_tokens_value, (int, float, str, bytes, bytearray))
+                                else 0
+                            )
                             traj_svc.log_llm_call(  # type: ignore[call-arg]
                                 step_id=step_id,
                                 model=stream_model_type,
@@ -2419,7 +2420,7 @@ end code: {final_code}
                                 user_prompt=str(params.get("prompt", ""))[:2000],
                                 response=response_str[:2000],
                                 temperature=0.0,
-                                max_tokens=int(params.get("maxTokens", 0)),
+                                max_tokens=max_tokens,
                                 purpose="action",
                                 action_type="dynamic_prompt_exec.stream",
                                 latency_ms=0,

package/elizaos/services/hook_service.py

@@ -21,7 +21,7 @@ import uuid
 from collections.abc import Callable
 from dataclasses import dataclass, field
 from datetime import datetime
-from enum import Enum
+from enum import StrEnum
 from typing import TYPE_CHECKING, Any
 
 from elizaos.types.service import Service, ServiceType
@@ -30,7 +30,7 @@ if TYPE_CHECKING:
     from elizaos.types.runtime import IAgentRuntime
 
 
-class HookSource(str, Enum):
+class HookSource(StrEnum):
     """Source of a hook registration."""
 
     BUNDLED = "bundled"
@@ -40,7 +40,7 @@ class HookSource(str, Enum):
     RUNTIME = "runtime"
 
 
-class HookEventType(str, Enum):
+class HookEventType(StrEnum):
     """Hook-specific event types."""
 
     HOOK_COMMAND_NEW = "HOOK_COMMAND_NEW"