@elizaos/python 2.0.0-alpha.10 → 2.0.0-alpha.26

package/elizaos/__init__.py

@@ -108,7 +108,6 @@ from elizaos.types import (
     string_to_uuid,
 )
 from elizaos.types.database import IDatabaseAdapter  # noqa: E402
-from elizaos.types.environment import ChannelType  # noqa: E402
 from elizaos.types.primitives import (  # noqa: E402
     ChannelType,
     Content,

package/elizaos/advanced_capabilities/__init__.py

@@ -1,85 +1,50 @@
 """Advanced Capabilities - Extended features for agent operation.
 
 This module provides advanced capabilities that can be enabled with
-`advanced_capabilities=True` or `enable_extended=True`:
-- Extended actions (contacts, room management, image generation, etc.)
-- Extended providers (facts, knowledge, relationships, etc.)
-- Evaluators (reflection, relationship extraction)
-- Extended services (rolodex, follow-up scheduling)
+`advanced_capabilities=True` or `enable_extended=True`.
+
+Relationship/contact extraction and social-memory features are owned by
+`plugin-rolodex` and are intentionally not auto-registered here.
 """
 
 from .actions import (
-    add_contact_action,
     advanced_actions,
     follow_room_action,
     generate_image_action,
     mute_room_action,
-    remove_contact_action,
-    schedule_follow_up_action,
-    search_contacts_action,
-    send_message_action,
     unfollow_room_action,
     unmute_room_action,
-    update_contact_action,
-    update_entity_action,
     update_role_action,
     update_settings_action,
 )
-from .evaluators import (
-    advanced_evaluators,
-    reflection_evaluator,
-    relationship_extraction_evaluator,
-)
+from .evaluators import advanced_evaluators
 from .providers import (
     advanced_providers,
     agent_settings_provider,
-    contacts_provider,
-    facts_provider,
-    follow_ups_provider,
     knowledge_provider,
-    relationships_provider,
     roles_provider,
     settings_provider,
 )
-from .services import (
-    FollowUpService,
-    RolodexService,
-    advanced_services,
-)
+from .services import advanced_services
 
 __all__ = [
     # Actions
     "advanced_actions",
-    "add_contact_action",
     "follow_room_action",
     "generate_image_action",
     "mute_room_action",
-    "remove_contact_action",
-    "schedule_follow_up_action",
-    "search_contacts_action",
-    "send_message_action",
     "unfollow_room_action",
     "unmute_room_action",
-    "update_contact_action",
-    "update_entity_action",
     "update_role_action",
     "update_settings_action",
     # Providers
     "advanced_providers",
     "agent_settings_provider",
-    "contacts_provider",
-    "facts_provider",
-    "follow_ups_provider",
     "knowledge_provider",
-    "relationships_provider",
     "roles_provider",
     "settings_provider",
     # Evaluators
     "advanced_evaluators",
-    "reflection_evaluator",
-    "relationship_extraction_evaluator",
     # Services
     "advanced_services",
-    "FollowUpService",
-    "RolodexService",
 ]

package/elizaos/advanced_capabilities/actions/__init__.py

@@ -3,52 +3,32 @@
 Extended actions that can be enabled with `advanced_capabilities=True`.
 """
 
-from .add_contact import add_contact_action
 from .follow_room import follow_room_action
 from .image_generation import generate_image_action
 from .mute_room import mute_room_action
-from .remove_contact import remove_contact_action
 from .roles import update_role_action
-from .schedule_follow_up import schedule_follow_up_action
-from .search_contacts import search_contacts_action
-from .send_message import send_message_action
 from .settings import update_settings_action
 from .unfollow_room import unfollow_room_action
 from .unmute_room import unmute_room_action
-from .update_contact import update_contact_action
-from .update_entity import update_entity_action
 
 __all__ = [
-    "add_contact_action",
     "follow_room_action",
     "generate_image_action",
     "mute_room_action",
-    "remove_contact_action",
-    "schedule_follow_up_action",
-    "search_contacts_action",
-    "send_message_action",
     "unfollow_room_action",
     "unmute_room_action",
-    "update_contact_action",
-    "update_entity_action",
     "update_role_action",
     "update_settings_action",
     "advanced_actions",
 ]
 
+# Rolodex/contact actions are provided by plugin-rolodex.
 advanced_actions = [
-    add_contact_action,
     follow_room_action,
     generate_image_action,
     mute_room_action,
-    remove_contact_action,
-    schedule_follow_up_action,
-    search_contacts_action,
-    send_message_action,
     unfollow_room_action,
     unmute_room_action,
-    update_contact_action,
-    update_entity_action,
     update_role_action,
     update_settings_action,
 ]

package/elizaos/advanced_capabilities/actions/add_contact.py

@@ -13,7 +13,6 @@ from elizaos.types import (
     Content,
     ModelType,
 )
-from elizaos.utils.spec_examples import convert_spec_examples
 
 if TYPE_CHECKING:
     from elizaos.types import (
@@ -30,7 +29,22 @@ _spec = require_action_spec("ADD_CONTACT")
 
 def _convert_spec_examples() -> list[list[ActionExample]]:
     """Convert spec examples to ActionExample format."""
-    return convert_spec_examples(_spec)
+    spec_examples = _spec.get("examples", [])
+    if spec_examples:
+        return [
+            [
+                ActionExample(
+                    name=msg.get("name", ""),
+                    content=Content(
+                        text=msg.get("content", {}).get("text", ""),
+                        actions=msg.get("content", {}).get("actions"),
+                    ),
+                )
+                for msg in example
+            ]
+            for example in spec_examples
+        ]
+    return []
 
 
 @dataclass
@@ -89,18 +103,14 @@ class AddContactAction:
         notes = str(parsed.get("notes", ""))
         reason = str(parsed.get("reason", ""))
 
-        from uuid import UUID as StdUUID
-
         entity_id = message.entity_id
-        entity_id_uuid = StdUUID(str(entity_id)) if entity_id else None
         preferences = ContactPreferences(notes=notes) if notes else None
 
-        if entity_id_uuid:
-            await rolodex_service.add_contact(
-                entity_id=entity_id_uuid,
-                categories=categories,
-                preferences=preferences,
-            )
+        await rolodex_service.add_contact(
+            entity_id=entity_id,
+            categories=categories,
+            preferences=preferences,
+        )
 
         response_text = (
             f"I've added {contact_name} to your contacts as {', '.join(categories)}. {reason}"

package/elizaos/advanced_capabilities/actions/follow_room.py

@@ -3,27 +3,41 @@ from __future__ import annotations
 from dataclasses import dataclass, field
 from typing import TYPE_CHECKING
 
+from elizaos.generated.spec_helpers import require_action_spec
 from elizaos.types import Action, ActionExample, ActionResult, Content
 
 if TYPE_CHECKING:
     from elizaos.types import HandlerCallback, HandlerOptions, IAgentRuntime, Memory, State
 
+# Get text content from centralized specs
+_spec = require_action_spec("FOLLOW_ROOM")
+
+
+def _convert_spec_examples() -> list[list[ActionExample]]:
+    """Convert spec examples to ActionExample format."""
+    spec_examples = _spec.get("examples", [])
+    if spec_examples:
+        return [
+            [
+                ActionExample(
+                    name=msg.get("name", ""),
+                    content=Content(
+                        text=msg.get("content", {}).get("text", ""),
+                        actions=msg.get("content", {}).get("actions"),
+                    ),
+                )
+                for msg in example
+            ]
+            for example in spec_examples
+        ]
+    return []
+
 
 @dataclass
 class FollowRoomAction:
-    name: str = "FOLLOW_ROOM"
-    similes: list[str] = field(
-        default_factory=lambda: [
-            "JOIN_ROOM",
-            "SUBSCRIBE_ROOM",
-            "WATCH_ROOM",
-            "ENTER_ROOM",
-        ]
-    )
-    description: str = (
-        "Follow a room to receive updates and monitor messages. "
-        "Use this when you want to actively engage with a room's content."
-    )
+    name: str = _spec["name"]
+    similes: list[str] = field(default_factory=lambda: list(_spec.get("similes", [])))
+    description: str = _spec["description"]
 
     async def validate(
         self, runtime: IAgentRuntime, message: Memory, _state: State | None = None
@@ -124,21 +138,7 @@ class FollowRoomAction:
 
     @property
     def examples(self) -> list[list[ActionExample]]:
-        return [
-            [
-                ActionExample(
-                    name="{{name1}}",
-                    content=Content(text="Can you keep an eye on this channel?"),
-                ),
-                ActionExample(
-                    name="{{name2}}",
-                    content=Content(
-                        text="I'll follow this room and monitor its activity.",
-                        actions=["FOLLOW_ROOM"],
-                    ),
-                ),
-            ],
-        ]
+        return _convert_spec_examples()
 
 
 follow_room_action = Action(

package/elizaos/advanced_capabilities/actions/image_generation.py

@@ -18,34 +18,21 @@ _spec = require_action_spec("GENERATE_IMAGE")
 def _convert_spec_examples() -> list[list[ActionExample]]:
     """Convert spec examples to ActionExample format."""
     spec_examples = _spec.get("examples", [])
-    if not isinstance(spec_examples, list):
-        return []
-    result: list[list[ActionExample]] = []
-    for example in spec_examples:
-        if not isinstance(example, list):
-            continue
-        row: list[ActionExample] = []
-        for msg in example:
-            if not isinstance(msg, dict):
-                continue
-            content = msg.get("content", {})
-            text = ""
-            actions: list[str] | None = None
-            if isinstance(content, dict):
-                text_val = content.get("text", "")
-                text = str(text_val) if text_val else ""
-                actions_val = content.get("actions")
-                if isinstance(actions_val, list) and all(isinstance(a, str) for a in actions_val):
-                    actions = list(actions_val)
-            row.append(
+    if spec_examples:
+        return [
+            [
                 ActionExample(
-                    name=str(msg.get("name", "")),
-                    content=Content(text=text, actions=actions),
+                    name=msg.get("name", ""),
+                    content=Content(
+                        text=msg.get("content", {}).get("text", ""),
+                        actions=msg.get("content", {}).get("actions"),
+                    ),
                 )
-            )
-        if row:
-            result.append(row)
-    return result
+                for msg in example
+            ]
+            for example in spec_examples
+        ]
+    return []
 
 
 @dataclass

package/elizaos/advanced_capabilities/actions/mute_room.py

@@ -16,34 +16,21 @@ _spec = require_action_spec("MUTE_ROOM")
 def _convert_spec_examples() -> list[list[ActionExample]]:
     """Convert spec examples to ActionExample format."""
     spec_examples = _spec.get("examples", [])
-    if not isinstance(spec_examples, list):
-        return []
-    result: list[list[ActionExample]] = []
-    for example in spec_examples:
-        if not isinstance(example, list):
-            continue
-        row: list[ActionExample] = []
-        for msg in example:
-            if not isinstance(msg, dict):
-                continue
-            content = msg.get("content", {})
-            text = ""
-            actions: list[str] | None = None
-            if isinstance(content, dict):
-                text_val = content.get("text", "")
-                text = str(text_val) if text_val else ""
-                actions_val = content.get("actions")
-                if isinstance(actions_val, list) and all(isinstance(a, str) for a in actions_val):
-                    actions = list(actions_val)
-            row.append(
+    if spec_examples:
+        return [
+            [
                 ActionExample(
-                    name=str(msg.get("name", "")),
-                    content=Content(text=text, actions=actions),
+                    name=msg.get("name", ""),
+                    content=Content(
+                        text=msg.get("content", {}).get("text", ""),
+                        actions=msg.get("content", {}).get("actions"),
+                    ),
                 )
-            )
-        if row:
-            result.append(row)
-    return result
+                for msg in example
+            ]
+            for example in spec_examples
+        ]
+    return []
 
 
 @dataclass

package/elizaos/advanced_capabilities/actions/remove_contact.py

@@ -13,7 +13,6 @@ from elizaos.types import (
     Content,
     ModelType,
 )
-from elizaos.utils.spec_examples import convert_spec_examples
 
 if TYPE_CHECKING:
     from elizaos.types import (
@@ -30,7 +29,22 @@ _spec = require_action_spec("REMOVE_CONTACT")
 
 def _convert_spec_examples() -> list[list[ActionExample]]:
     """Convert spec examples to ActionExample format."""
-    return convert_spec_examples(_spec)
+    spec_examples = _spec.get("examples", [])
+    if spec_examples:
+        return [
+            [
+                ActionExample(
+                    name=msg.get("name", ""),
+                    content=Content(
+                        text=msg.get("content", {}).get("text", ""),
+                        actions=msg.get("content", {}).get("actions"),
+                    ),
+                )
+                for msg in example
+            ]
+            for example in spec_examples
+        ]
+    return []
 
 
 @dataclass

package/elizaos/advanced_capabilities/actions/roles.py

@@ -7,7 +7,6 @@ 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:
@@ -20,34 +19,21 @@ _spec = require_action_spec("UPDATE_ROLE")
 def _convert_spec_examples() -> list[list[ActionExample]]:
     """Convert spec examples to ActionExample format."""
     spec_examples = _spec.get("examples", [])
-    if not isinstance(spec_examples, list):
-        return []
-    result: list[list[ActionExample]] = []
-    for example in spec_examples:
-        if not isinstance(example, list):
-            continue
-        row: list[ActionExample] = []
-        for msg in example:
-            if not isinstance(msg, dict):
-                continue
-            content = msg.get("content", {})
-            text = ""
-            actions: list[str] | None = None
-            if isinstance(content, dict):
-                text_val = content.get("text", "")
-                text = str(text_val) if text_val else ""
-                actions_val = content.get("actions")
-                if isinstance(actions_val, list) and all(isinstance(a, str) for a in actions_val):
-                    actions = list(actions_val)
-            row.append(
+    if spec_examples:
+        return [
+            [
                 ActionExample(
-                    name=str(msg.get("name", "")),
-                    content=Content(text=text, actions=actions),
+                    name=msg.get("name", ""),
+                    content=Content(
+                        text=msg.get("content", {}).get("text", ""),
+                        actions=msg.get("content", {}).get("actions"),
+                    ),
                 )
-            )
-        if row:
-            result.append(row)
-    return result
+                for msg in example
+            ]
+            for example in spec_examples
+        ]
+    return []
 
 
 class Role(str, Enum):

package/elizaos/advanced_capabilities/actions/schedule_follow_up.py

@@ -3,8 +3,10 @@ from __future__ import annotations
 from dataclasses import dataclass, field
 from datetime import datetime
 from typing import TYPE_CHECKING
+from uuid import UUID as StdUUID
 
 from elizaos.bootstrap.utils.xml import parse_key_value_xml
+from elizaos.deterministic import get_prompt_reference_datetime
 from elizaos.generated.spec_helpers import require_action_spec
 from elizaos.prompts import SCHEDULE_FOLLOW_UP_TEMPLATE
 from elizaos.types import (
@@ -34,6 +36,25 @@ def _convert_spec_examples() -> list[list[ActionExample]]:
     return convert_spec_examples(_spec)
 
 
+def _normalize_priority(raw_priority: str) -> str:
+    normalized = raw_priority.strip().lower()
+    if normalized in {"high", "medium", "low"}:
+        return normalized
+    return "medium"
+
+
+def _coerce_uuid(value: object | None) -> StdUUID | None:
+    if value is None:
+        return None
+    text = str(value).strip()
+    if not text:
+        return None
+    try:
+        return StdUUID(text)
+    except ValueError:
+        return None
+
+
 @dataclass
 class ScheduleFollowUpAction:
     name: str = _spec["name"]
@@ -79,7 +100,12 @@ class ScheduleFollowUpAction:
             )
 
         state = await runtime.compose_state(message, ["RECENT_MESSAGES", "ENTITIES"])
-        state.values["currentDateTime"] = datetime.utcnow().isoformat()
+        state.values["currentDateTime"] = get_prompt_reference_datetime(
+            runtime,
+            message,
+            state,
+            "action:schedule_follow_up",
+        ).isoformat()
 
         prompt = runtime.compose_prompt_from_state(
             state=state,
@@ -100,24 +126,53 @@ class ScheduleFollowUpAction:
         contact_name = str(parsed.get("contactName", ""))
         scheduled_at_str = str(parsed.get("scheduledAt", ""))
         reason = str(parsed.get("reason", "Follow-up"))
-        priority = str(parsed.get("priority", "medium"))
+        priority = _normalize_priority(str(parsed.get("priority", "medium")))
         follow_up_message = str(parsed.get("message", ""))
+        parsed_entity_id = _coerce_uuid(parsed.get("entityId"))
+        message_entity_id = _coerce_uuid(message.entity_id)
+
+        try:
+            scheduled_at = datetime.fromisoformat(scheduled_at_str.replace("Z", "+00:00"))
+        except ValueError:
+            return ActionResult(
+                text="Could not parse the follow-up date/time",
+                success=False,
+                values={"error": True},
+                data={"error": "Invalid follow-up datetime"},
+            )
 
-        scheduled_at = datetime.fromisoformat(scheduled_at_str.replace("Z", "+00:00"))
+        entity_id_uuid = parsed_entity_id or message_entity_id
 
-        from uuid import UUID as StdUUID
+        if entity_id_uuid is None and contact_name:
+            contacts = await rolodex_service.search_contacts(search_term=contact_name)
+            if contacts:
+                entity_id_uuid = contacts[0].entity_id
 
-        entity_id = message.entity_id
-        entity_id_uuid = StdUUID(str(entity_id)) if entity_id else None
-        if entity_id_uuid:
-            await follow_up_service.schedule_follow_up(
-                entity_id=entity_id_uuid,
-                scheduled_at=scheduled_at,
-                reason=reason,
-                priority=priority,
-                message=follow_up_message,
+        if entity_id_uuid is None:
+            return ActionResult(
+                text=f"Could not determine which contact to schedule for ({contact_name}).",
+                success=False,
+                values={"error": True},
+                data={"error": "Missing contact entity id"},
             )
 
+        contact = await rolodex_service.get_contact(entity_id_uuid)
+        if contact is None:
+            return ActionResult(
+                text=f"Contact '{contact_name}' was not found in the rolodex.",
+                success=False,
+                values={"error": True},
+                data={"error": "Contact not found"},
+            )
+
+        await follow_up_service.schedule_follow_up(
+            entity_id=entity_id_uuid,
+            scheduled_at=scheduled_at,
+            reason=reason,
+            priority=priority,
+            message=follow_up_message,
+        )
+
         response_text = f"I've scheduled a follow-up with {contact_name} for {scheduled_at.strftime('%B %d, %Y')}. Reason: {reason}"
 
         if callback:
@@ -127,11 +182,11 @@ class ScheduleFollowUpAction:
             text=response_text,
             success=True,
             values={
-                "contactId": str(entity_id),
+                "contactId": str(entity_id_uuid),
                 "scheduledAt": scheduled_at.isoformat(),
             },
             data={
-                "contactId": str(entity_id),
+                "contactId": str(entity_id_uuid),
                 "contactName": contact_name,
                 "scheduledAt": scheduled_at.isoformat(),
                 "reason": reason,

package/elizaos/advanced_capabilities/actions/search_contacts.py

@@ -13,7 +13,6 @@ from elizaos.types import (
     Content,
     ModelType,
 )
-from elizaos.utils.spec_examples import convert_spec_examples
 
 if TYPE_CHECKING:
     from elizaos.types import (
@@ -30,7 +29,22 @@ _spec = require_action_spec("SEARCH_CONTACTS")
 
 def _convert_spec_examples() -> list[list[ActionExample]]:
     """Convert spec examples to ActionExample format."""
-    return convert_spec_examples(_spec)
+    spec_examples = _spec.get("examples", [])
+    if spec_examples:
+        return [
+            [
+                ActionExample(
+                    name=msg.get("name", ""),
+                    content=Content(
+                        text=msg.get("content", {}).get("text", ""),
+                        actions=msg.get("content", {}).get("actions"),
+                    ),
+                )
+                for msg in example
+            ]
+            for example in spec_examples
+        ]
+    return []
 
 
 @dataclass
@@ -95,7 +109,7 @@ class SearchContactsAction:
 
         contact_details: list[dict[str, str]] = []
         for contact in contacts:
-            entity = await runtime.get_entity(str(contact.entity_id))
+            entity = await runtime.get_entity(contact.entity_id)
             name = entity.name if entity and entity.name else "Unknown"
             contact_details.append(
                 {