uagents-core 0.3.0__py3-none-any.whl → 0.3.2__py3-none-any.whl

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/storage.py

@@ -56,13 +56,13 @@ class ExternalStorage:
             raise RuntimeError("No identity or API token available for authentication")
 
     def upload(
-        self, asset_id: str, asset_content: str, 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()
         headers["Content-Type"] = "application/json"
         payload = {
-            "contents": base64.b64encode(asset_content.encode()).decode(),
+            "contents": base64.b64encode(content).decode(),
             "mime_type": mime_type,
         }
         response = requests.put(url, json=payload, headers=headers)
@@ -88,7 +88,7 @@ class ExternalStorage:
     def create_asset(
         self,
         name: str,
-        content: str,
+        content: bytes,
         mime_type: str = "text/plain",
         lifetime_hours: int = 24,
     ) -> str:
@@ -100,7 +100,7 @@ class ExternalStorage:
         payload = {
             "name": name,
             "mime_type": mime_type,
-            "contents": base64.b64encode(content.encode()).decode(),
+            "contents": base64.b64encode(content).decode(),
             "lifetime_hours": lifetime_hours,
         }
 

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

{uagents_core-0.3.0.dist-info → uagents_core-0.3.2.dist-info}/METADATA

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

{uagents_core-0.3.0.dist-info → uagents_core-0.3.2.dist-info}/RECORD

@@ -10,13 +10,13 @@ uagents_core/identity.py,sha256=YHI3rromJ27DGF3X102MGkuvBIZ6usAaMutlDLdVaL0,6185
 uagents_core/logger.py,sha256=5XLs3-XpKDeoSwtAE5pEM9h6x3pUPNbb0T0J7KvFmG0,877
 uagents_core/models.py,sha256=fxsFjRochkJUdgPltT1HWgBl-9K-HVQWl8sSMVgeJFY,1066
 uagents_core/protocol.py,sha256=T9jasOkltne33E16Y7VrqcB2moWVsv-Qh4XLZotyz8g,5428
-uagents_core/registration.py,sha256=h7ha58H7XbLQ__qOHMKVnHShzMlWfd2xrK9fRdgfb9w,2953
-uagents_core/storage.py,sha256=A7s2jJhdbDI0aNKAcjJpL5oxcYrb2FnHhXGwlAt3RY4,4506
+uagents_core/registration.py,sha256=HPV-bYNEfWxOwjbJqFZoMUTGu8H1CcwSLmN-vVBOTnc,3119
+uagents_core/storage.py,sha256=Nb_vvckp5FV48bcewFmYUoLFHAsWxwaiIv_-idFpGqQ,4480
 uagents_core/types.py,sha256=_W3EN1wEIRFxuhhBxyZxQH_dA_3AtoPPReIzrgcTUHc,1167
 uagents_core/utils/__init__.py,sha256=v0MaxDYCTtQlwbblEHCfLtbeTnA2hCmKKJk7mlcE20U,135
 uagents_core/utils/messages.py,sha256=5o3ahVaeGwzWMAxDs0fl2pC1duDq2VS_-czoVtZpkOc,5066
-uagents_core/utils/registration.py,sha256=HaA0J2yisx0bwVs1LbpJ4UP3-maeeAf37yziI-U07l8,9067
+uagents_core/utils/registration.py,sha256=zBJVCJwcjVnTb4PqVFfxCS1kVxeEvD9IC9-_Q4Coz_s,10022
 uagents_core/utils/resolver.py,sha256=X18oe-WPU_6pfTc6x_Oa69kZQYCMidkaSeCs68Poaik,2169
-uagents_core-0.3.0.dist-info/METADATA,sha256=SpfzcsoEMP44pjC4tRmjq0kx8UHr-KPSZ4kZGG6jp8A,1009
-uagents_core-0.3.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-uagents_core-0.3.0.dist-info/RECORD,,
+uagents_core-0.3.2.dist-info/METADATA,sha256=ijRPc3gtEpjViaklW6Wy5rs2EfNjrrlvSRx8XW5Xqs0,1009
+uagents_core-0.3.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+uagents_core-0.3.2.dist-info/RECORD,,