uagents-core 0.3.1__tar.gz → 0.3.2__tar.gz

{uagents_core-0.3.1 → uagents_core-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uagents-core
-Version: 0.3.1
+Version: 0.3.2
 Summary: Core components for agent based systems
 License: Apache 2.0
 Author: Ed FitzGerald

{uagents_core-0.3.1 → uagents_core-0.3.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uagents-core"
-version = "0.3.1"
+version = "0.3.2"
 description = "Core components for agent based systems"
 authors = [
     { name = "Ed FitzGerald", email = "edward.fitzgerald@fetch.ai" },

{uagents_core-0.3.1 → uagents_core-0.3.2}/uagents_core/registration.py

@@ -70,7 +70,7 @@ class VerifiableModel(BaseModel):
 class AgentRegistrationAttestation(VerifiableModel):
     protocols: list[str]
     endpoints: list[AgentEndpoint]
-    metadata: dict[str, str | dict[str, str]] | None = None
+    metadata: dict[str, str | list[str] | dict[str, str]] | None = None
 
 
 # Agentverse related models
@@ -107,3 +107,9 @@ class AgentUpdates(BaseModel):
     avatar_url: str | None = Field(default=None, max_length=4000)
     short_description: str | None = Field(default=None, max_length=300)
     agent_type: AgentType | None = "custom"
+
+
+class AgentStatusUpdate(VerifiableModel):
+    is_active: bool = Field(
+        ..., description="Indicates whether the agent is currently active"
+    )

{uagents_core-0.3.1 → uagents_core-0.3.2}/uagents_core/storage.py

@@ -56,7 +56,7 @@ class ExternalStorage:
             raise RuntimeError("No identity or API token available for authentication")
 
     def upload(
-        self, asset_id: str, asset_content: bytes, mime_type: str = "text/plain"
+        self, asset_id: str, content: bytes, mime_type: str = "text/plain"
     ) -> dict:
         url = f"{self.storage_url}/assets/{asset_id}/contents/"
         headers = self._get_auth_header()

{uagents_core-0.3.1 → uagents_core-0.3.2}/uagents_core/utils/registration.py

@@ -19,6 +19,7 @@ from uagents_core.logger import get_logger
 from uagents_core.protocol import is_valid_protocol_digest
 from uagents_core.registration import (
     AgentRegistrationAttestation,
+    AgentStatusUpdate,
     AgentUpdates,
     AgentverseConnectRequest,
     ChallengeRequest,
@@ -66,7 +67,7 @@ def register_in_almanac(
     identity: Identity,
     endpoints: list[str],
     protocol_digests: list[str],
-    metadata: dict[str, str | dict[str, str]] | None = None,
+    metadata: dict[str, str | list[str] | dict[str, str]] | None = None,
     *,
     agentverse_config: AgentverseConfig | None = None,
     timeout: int = DEFAULT_REQUEST_TIMEOUT,
@@ -269,3 +270,37 @@ def register_in_agentverse(
             exc_info=e,
         )
         return False
+
+
+def update_agent_status(active: bool, identity: Identity):
+    """
+    Update the agent's active/inactive status in the Almanac API.
+
+    Args:
+        active (bool): The status of the agent.
+        identity (Identity): The identity of the agent.
+    """
+    almanac_api = AgentverseConfig().url + DEFAULT_ALMANAC_API_PATH
+
+    status_update = AgentStatusUpdate(
+        agent_identifier=identity.address, is_active=active
+    )
+    status_update.sign(identity)
+
+    logger.debug(
+        msg="Updating agent status in Almanac API",
+        extra=status_update.model_dump(),
+    )
+
+    status, _ = _send_post_request(
+        url=f"{almanac_api}/agents/{identity.address}/status",
+        data=status_update,
+    )
+
+    if status:
+        logger.info(
+            msg=f"Agent status updated to {'active' if active else 'inactive'}",
+            extra={"agent_address": identity.address},
+        )
+
+    return status