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

package/elizaos/advanced_capabilities/actions/add_contact.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, cast
+from uuid import UUID as StdUUID
 
 from elizaos.bootstrap.utils.xml import parse_key_value_xml
 from elizaos.generated.spec_helpers import require_action_spec
@@ -29,7 +30,7 @@ _spec = require_action_spec("ADD_CONTACT")
 
 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 [
             [
@@ -103,7 +104,7 @@ class AddContactAction:
         notes = str(parsed.get("notes", ""))
         reason = str(parsed.get("reason", ""))
 
-        entity_id = message.entity_id
+        entity_id = StdUUID(str(message.entity_id))
         preferences = ContactPreferences(notes=notes) if notes else None
 
         await rolodex_service.add_contact(

package/elizaos/advanced_capabilities/actions/follow_room.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, cast
 
 from elizaos.generated.spec_helpers import require_action_spec
 from elizaos.types import Action, ActionExample, ActionResult, Content
@@ -15,7 +15,7 @@ _spec = require_action_spec("FOLLOW_ROOM")
 
 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 [
             [

package/elizaos/advanced_capabilities/actions/image_generation.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, cast
 
 from elizaos.bootstrap.utils.xml import parse_key_value_xml
 from elizaos.generated.spec_helpers import require_action_spec
@@ -17,7 +17,7 @@ _spec = require_action_spec("GENERATE_IMAGE")
 
 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 [
             [

package/elizaos/advanced_capabilities/actions/mute_room.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, cast
 
 from elizaos.generated.spec_helpers import require_action_spec
 from elizaos.types import Action, ActionExample, ActionResult, Content
@@ -15,7 +15,7 @@ _spec = require_action_spec("MUTE_ROOM")
 
 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 [
             [

package/elizaos/advanced_capabilities/actions/remove_contact.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, cast
 
 from elizaos.bootstrap.utils.xml import parse_key_value_xml
 from elizaos.generated.spec_helpers import require_action_spec
@@ -29,7 +29,7 @@ _spec = require_action_spec("REMOVE_CONTACT")
 
 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 [
             [

package/elizaos/advanced_capabilities/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/advanced_capabilities/actions/search_contacts.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, cast
 
 from elizaos.bootstrap.utils.xml import parse_key_value_xml
 from elizaos.generated.spec_helpers import require_action_spec
@@ -29,7 +29,7 @@ _spec = require_action_spec("SEARCH_CONTACTS")
 
 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 [
             [
@@ -109,7 +109,7 @@ class SearchContactsAction:
 
         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/advanced_capabilities/actions/send_message.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 import contextlib
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, cast
 
 from elizaos.generated.spec_helpers import require_action_spec
 from elizaos.types import Action, ActionExample, ActionResult, Content
@@ -18,7 +18,7 @@ _spec = require_action_spec("SEND_MESSAGE")
 
 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 [
             [

package/elizaos/advanced_capabilities/actions/settings.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, cast
 
 from elizaos.bootstrap.utils.xml import parse_key_value_xml
 from elizaos.generated.spec_helpers import require_action_spec
@@ -17,7 +17,7 @@ _spec = require_action_spec("UPDATE_SETTINGS")
 
 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 [
             [

package/elizaos/advanced_capabilities/actions/unfollow_room.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, cast
 
 from elizaos.generated.spec_helpers import require_action_spec
 from elizaos.types import Action, ActionExample, ActionResult, Content
@@ -15,7 +15,7 @@ _spec = require_action_spec("UNFOLLOW_ROOM")
 
 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 [
             [

package/elizaos/advanced_capabilities/actions/unmute_room.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, cast
 
 from elizaos.generated.spec_helpers import require_action_spec
 from elizaos.types import Action, ActionExample, ActionResult, Content
@@ -15,7 +15,7 @@ _spec = require_action_spec("UNMUTE_ROOM")
 
 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 [
             [

package/elizaos/advanced_capabilities/actions/update_contact.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, cast
 
 from elizaos.bootstrap.utils.xml import parse_key_value_xml
 from elizaos.generated.spec_helpers import require_action_spec
@@ -29,7 +29,7 @@ _spec = require_action_spec("UPDATE_CONTACT")
 
 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 [
             [

package/elizaos/advanced_capabilities/actions/update_entity.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, cast
 from uuid import UUID
 
 from elizaos.bootstrap.utils.xml import parse_key_value_xml
@@ -18,7 +18,7 @@ _spec = require_action_spec("UPDATE_ENTITY")
 
 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 [
             [

package/elizaos/advanced_capabilities/providers/knowledge.py

@@ -3,7 +3,6 @@ from __future__ import annotations
 from typing import TYPE_CHECKING
 
 from elizaos.types import Provider, ProviderResult
-from elizaos.types.database import MemorySearchOptions
 
 if TYPE_CHECKING:
     from elizaos.types import IAgentRuntime, Memory, State
@@ -38,14 +37,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 skipped for parity with TS/Bootstrap

package/elizaos/advanced_capabilities/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/advanced_memory/actions/reset_session.py

@@ -1,11 +1,143 @@
-from ...types import Action
-
-# TODO: Implement reset_session action
-reset_session_action: Action = {
-    "name": "RESET_SESSION",
-    "similes": [],
-    "description": "Reset the session (Implementation TODO)",
-    "handler": None,
-    "validate": None,
-    "examples": [],
-}
+from __future__ import annotations
+
+import time
+from dataclasses import dataclass, field
+from typing import TYPE_CHECKING
+
+from elizaos.types import Action, ActionExample, ActionResult, Content
+
+if TYPE_CHECKING:
+    from elizaos.types import HandlerCallback, HandlerOptions, IAgentRuntime, Memory, State
+
+
+@dataclass
+class ResetSessionAction:
+    name: str = "RESET_SESSION"
+    similes: list[str] = field(
+        default_factory=lambda: ["CLEAR_HISTORY", "NEW_SESSION", "FORGET", "START_OVER", "RESET"]
+    )
+    description: str = (
+        "Resets the conversation session by creating a compaction point. "
+        "Messages before this point will not be included in future context."
+    )
+
+    async def validate(
+        self, runtime: IAgentRuntime, message: Memory, state: State | None = None
+    ) -> bool:
+        room = None
+        if state and getattr(state, "data", None) and getattr(state.data, "room", None):
+            room = state.data.room
+        elif message.room_id:
+            room = await runtime.get_room(message.room_id)
+
+        if not room or not getattr(room, "world_id", None):
+            return True
+
+        world = await runtime.get_world(room.world_id)
+        if not world or not getattr(world, "metadata", None):
+            return False
+
+        roles = world.metadata.get("roles", {})
+        user_role = roles.get(str(message.entity_id), "NONE")
+        return user_role in ("OWNER", "ADMIN")
+
+    async def handler(
+        self,
+        runtime: IAgentRuntime,
+        message: Memory,
+        state: State | None = None,
+        options: HandlerOptions | None = None,
+        callback: HandlerCallback | None = None,
+        responses: list[Memory] | None = None,
+    ) -> ActionResult:
+        del options, responses
+
+        room = None
+        if state and getattr(state, "data", None) and getattr(state.data, "room", None):
+            room = state.data.room
+        elif message.room_id:
+            room = await runtime.get_room(message.room_id)
+
+        if room is None:
+            if callback:
+                await callback(
+                    Content(
+                        text="Unable to reset session - room not found.",
+                        actions=["RESET_SESSION_FAILED"],
+                        source=getattr(message.content, "source", None),
+                    )
+                )
+            return ActionResult(
+                text="Room not found",
+                values={"error": "room_not_found"},
+                data={"actionName": "RESET_SESSION"},
+                success=False,
+            )
+
+        metadata = dict(getattr(room, "metadata", {}) or {})
+        previous_compaction = metadata.get("lastCompactionAt")
+        compaction_history = list(metadata.get("compactionHistory", []))
+        now = int(time.time() * 1000)
+
+        compaction_history.append(
+            {
+                "timestamp": now,
+                "triggeredBy": str(message.entity_id),
+                "reason": "manual_reset",
+            }
+        )
+        metadata["lastCompactionAt"] = now
+        metadata["compactionHistory"] = compaction_history[-10:]
+
+        if (
+            hasattr(room, "metadata")
+            and hasattr(room.metadata, "clear")
+            and hasattr(room.metadata, "update")
+        ):
+            room.metadata.clear()
+            room.metadata.update(metadata)
+        else:
+            room.metadata = metadata
+
+        await runtime.update_room(room)
+
+        if callback:
+            await callback(
+                Content(
+                    text="Session has been reset. I'll start fresh from here.",
+                    actions=["RESET_SESSION"],
+                    source=getattr(message.content, "source", None),
+                )
+            )
+
+        result = ActionResult(
+            text="Session reset successfully",
+            values={
+                "success": True,
+                "compactionAt": now,
+                "roomId": str(room.id),
+            },
+            data={
+                "actionName": "RESET_SESSION",
+                "compactionAt": now,
+                "roomId": str(room.id),
+            },
+            success=True,
+        )
+        if previous_compaction is not None:
+            result.values["previousCompactionAt"] = previous_compaction
+        return result
+
+    @property
+    def examples(self) -> list[list[ActionExample]]:
+        return []
+
+
+reset_session_action = Action(
+    name=ResetSessionAction.name,
+    similes=ResetSessionAction().similes,
+    description=ResetSessionAction.description,
+    validate=ResetSessionAction().validate,
+    handler=ResetSessionAction().handler,
+    examples=ResetSessionAction().examples,
+)

package/elizaos/advanced_memory/memory_service.py

@@ -3,7 +3,8 @@ from __future__ import annotations
 import heapq
 import re
 import time
-from typing import cast
+from collections import OrderedDict
+from typing import TypeVar, cast
 from uuid import UUID, uuid4
 
 from elizaos.types.model import ModelType
@@ -27,6 +28,7 @@ from .types import (
 _TABLE_SESSION_SUMMARY = "session_summary"
 _TABLE_LONG_TERM_MEMORY = "long_term_memory"
 _GLOBAL_LONG_TERM_ROOM_ID = string_to_uuid("advanced-memory:long-term")
+_OrderedValueT = TypeVar("_OrderedValueT")
 
 
 def _parse_summary_xml(xml: str) -> SummaryResult:
@@ -79,14 +81,28 @@ def _top_k_by_confidence(items: list[LongTermMemory], limit: int) -> list[LongTe
 
 class MemoryService(Service):
     service_type = "memory"
+    _MAX_LOCAL_SESSION_SUMMARIES = 500
+    _MAX_LOCAL_EXTRACTION_CHECKPOINTS = 500
+    _MAX_LOCAL_LONG_TERM_ENTITIES = 500
+    _MAX_LOCAL_LONG_TERM_PER_ENTITY = 200
 
     def __init__(self, runtime=None) -> None:
         super().__init__(runtime=runtime)
         self._config: MemoryConfig = MemoryConfig()
         # Fallback storage for runtimes without a DB adapter (tests/benchmarks).
-        self._session_summaries: dict[str, SessionSummary] = {}
-        self._long_term: dict[str, list[LongTermMemory]] = {}
-        self._extraction_checkpoints: dict[str, int] = {}
+        self._session_summaries: OrderedDict[str, SessionSummary] = OrderedDict()
+        self._long_term: OrderedDict[str, list[LongTermMemory]] = OrderedDict()
+        self._extraction_checkpoints: OrderedDict[str, int] = OrderedDict()
+
+    @staticmethod
+    def _touch_ordered(mapping: OrderedDict[str, _OrderedValueT], key: str) -> None:
+        if key in mapping:
+            mapping.move_to_end(key)
+
+    @staticmethod
+    def _prune_ordered(mapping: OrderedDict[str, _OrderedValueT], max_entries: int) -> None:
+        while len(mapping) > max_entries:
+            mapping.popitem(last=False)
 
     @property
     def capability_description(self) -> str:
@@ -156,7 +172,11 @@ class MemoryService(Service):
                 return 0
             except Exception:
                 return 0
-        return int(self._extraction_checkpoints.get(key, 0))
+        checkpoint = self._extraction_checkpoints.get(key)
+        if checkpoint is None:
+            return 0
+        self._touch_ordered(self._extraction_checkpoints, key)
+        return int(checkpoint)
 
     async def set_last_extraction_checkpoint(
         self, entity_id: UUID, room_id: UUID, message_count: int
@@ -167,6 +187,8 @@ class MemoryService(Service):
             _ = await runtime.set_cache(key, int(message_count))
             return
         self._extraction_checkpoints[key] = int(message_count)
+        self._touch_ordered(self._extraction_checkpoints, key)
+        self._prune_ordered(self._extraction_checkpoints, self._MAX_LOCAL_EXTRACTION_CHECKPOINTS)
 
     async def should_run_extraction(
         self, entity_id: UUID, room_id: UUID, current_message_count: int
@@ -219,7 +241,11 @@ class MemoryService(Service):
                     # Best-effort; ignore corrupt rows.
                     continue
 
-        return self._session_summaries.get(str(room_id))
+        summary = self._session_summaries.get(str(room_id))
+        if summary is None:
+            return None
+        self._touch_ordered(self._session_summaries, str(room_id))
+        return summary
 
     async def store_session_summary(
         self,
@@ -271,7 +297,10 @@ class MemoryService(Service):
             )
             return s
 
-        self._session_summaries[str(room_id)] = s
+        room_key = str(room_id)
+        self._session_summaries[room_key] = s
+        self._touch_ordered(self._session_summaries, room_key)
+        self._prune_ordered(self._session_summaries, self._MAX_LOCAL_SESSION_SUMMARIES)
         return s
 
     async def update_session_summary(
@@ -322,9 +351,11 @@ class MemoryService(Service):
             )
             return
 
-        existing = self._session_summaries.get(str(room_id))
+        room_key = str(room_id)
+        existing = self._session_summaries.get(room_key)
         if not existing or existing.id != summary_id:
             return
+        self._touch_ordered(self._session_summaries, room_key)
         if "summary" in updates and isinstance(updates["summary"], str):
             existing.summary = updates["summary"]
         if "message_count" in updates and isinstance(updates["message_count"], (int, float, str)):
@@ -381,7 +412,17 @@ class MemoryService(Service):
             )
             return m
 
-        self._long_term.setdefault(str(entity_id), []).append(m)
+        entity_key = str(entity_id)
+        bucket = self._long_term.get(entity_key)
+        if bucket is None:
+            bucket = []
+            self._long_term[entity_key] = bucket
+        else:
+            self._touch_ordered(self._long_term, entity_key)
+        bucket.append(m)
+        if len(bucket) > self._MAX_LOCAL_LONG_TERM_PER_ENTITY:
+            del bucket[: len(bucket) - self._MAX_LOCAL_LONG_TERM_PER_ENTITY]
+        self._prune_ordered(self._long_term, self._MAX_LOCAL_LONG_TERM_ENTITIES)
         return m
 
     async def get_long_term_memories(
@@ -437,7 +478,10 @@ class MemoryService(Service):
                     continue
             return _top_k_by_confidence(out, limit)
 
-        local_mems = self._long_term.get(str(entity_id), [])
+        entity_key = str(entity_id)
+        local_mems = self._long_term.get(entity_key, [])
+        if entity_key in self._long_term:
+            self._touch_ordered(self._long_term, entity_key)
         if category is not None:
             local_mems = [m for m in local_mems if m.category == category]
         return _top_k_by_confidence(local_mems, limit)

package/elizaos/advanced_memory/plugin.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 from elizaos.types import Plugin
 
+from .actions.reset_session import reset_session_action
 from .evaluators import long_term_extraction_evaluator, summarization_evaluator
 from .memory_service import MemoryService
 from .providers import context_summary_provider, long_term_memory_provider
@@ -21,7 +22,7 @@ def create_advanced_memory_plugin() -> Plugin:
         init=init_plugin,
         config={},
         services=[MemoryService],
-        actions=[],
+        actions=[reset_session_action],
         providers=[long_term_memory_provider, context_summary_provider],
         evaluators=[summarization_evaluator, long_term_extraction_evaluator],
     )

package/elizaos/advanced_memory/types.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from enum import Enum
+from enum import StrEnum
 from uuid import UUID
 
 
-class LongTermMemoryCategory(str, Enum):
+class LongTermMemoryCategory(StrEnum):
     EPISODIC = "episodic"
     SEMANTIC = "semantic"
     PROCEDURAL = "procedural"

package/elizaos/advanced_planning/actions/schedule_follow_up.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from dataclasses import dataclass, field
 from datetime import datetime
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, cast
 from uuid import UUID as StdUUID
 
 from elizaos.bootstrap.utils.xml import parse_key_value_xml
@@ -32,7 +32,7 @@ _spec = require_action_spec("SCHEDULE_FOLLOW_UP")
 
 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 [
             [