uagents-core 0.2.2__tar.gz → 0.3.0__tar.gz

{uagents_core-0.2.2 → uagents_core-0.3.0}/PKG-INFO

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

{uagents_core-0.2.2 → uagents_core-0.3.0}/pyproject.toml

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

{uagents_core-0.2.2 → uagents_core-0.3.0}/uagents_core/config.py

@@ -6,6 +6,7 @@ DEFAULT_REGISTRATION_PATH = "/v1/agents"
 DEFAULT_CHALLENGE_PATH = "/v1/auth/challenge"
 DEFAULT_MAILBOX_PATH = "/v1/submit"
 DEFAULT_PROXY_PATH = "/v1/proxy/submit"
+DEFAULT_STORAGE_PATH = "/v1/storage"
 
 DEFAULT_MAX_ENDPOINTS = 10
 
@@ -30,3 +31,7 @@ class AgentverseConfig(BaseModel):
     @property
     def proxy_endpoint(self) -> str:
         return f"{self.url}{DEFAULT_PROXY_PATH}"
+
+    @property
+    def storage_endpoint(self) -> str:
+        return f"{self.url}{DEFAULT_STORAGE_PATH}"

uagents_core-0.3.0/uagents_core/storage.py

@@ -0,0 +1,135 @@
+import base64
+import struct
+from datetime import datetime
+from secrets import token_bytes
+from typing import Optional
+
+import requests
+
+from uagents_core.config import AgentverseConfig
+from uagents_core.identity import Identity
+
+
+def compute_attestation(
+    identity: Identity, validity_start: datetime, validity_secs: int, nonce: bytes
+) -> str:
+    """
+    Compute a valid agent attestation token for authentication.
+    """
+    assert len(nonce) == 32, "Nonce is of invalid length"
+
+    valid_from = int(validity_start.timestamp())
+    valid_to = valid_from + validity_secs
+
+    public_key = bytes.fromhex(identity.pub_key)
+
+    payload = public_key + struct.pack(">QQ", valid_from, valid_to) + nonce
+    assert len(payload) == 81, "attestation payload is incorrect"
+
+    signature = identity.sign(payload)
+    attestation = f"attr:{base64.b64encode(payload).decode()}:{signature}"
+    return attestation
+
+
+class ExternalStorage:
+    def __init__(
+        self,
+        identity: Optional[Identity] = None,
+        storage_url: Optional[str] = None,
+        api_token: Optional[str] = None,
+    ):
+        self.identity = identity
+        self.api_token = api_token
+        self.storage_url = storage_url or AgentverseConfig().storage_endpoint
+
+    def _make_attestation(self) -> str:
+        nonce = token_bytes(32)
+        now = datetime.now()
+        return compute_attestation(self.identity, now, 3600, nonce)
+
+    def _get_auth_header(self) -> dict:
+        if self.api_token:
+            return {"Authorization": f"Bearer {self.api_token}"}
+        elif self.identity:
+            return {"Authorization": f"Agent {self._make_attestation()}"}
+        else:
+            raise RuntimeError("No identity or API token available for authentication")
+
+    def upload(
+        self, asset_id: str, asset_content: str, 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(),
+            "mime_type": mime_type,
+        }
+        response = requests.put(url, json=payload, headers=headers)
+        if response.status_code != 200:
+            raise RuntimeError(
+                f"Upload failed: {response.status_code}, {response.text}"
+            )
+
+        return response.json()
+
+    def download(self, asset_id: str) -> dict:
+        url = f"{self.storage_url}/assets/{asset_id}/contents/"
+        headers = self._get_auth_header()
+
+        response = requests.get(url, headers=headers)
+        if response.status_code != 200:
+            raise RuntimeError(
+                f"Download failed: {response.status_code}, {response.text}"
+            )
+
+        return response.json()
+
+    def create_asset(
+        self,
+        name: str,
+        content: str,
+        mime_type: str = "text/plain",
+        lifetime_hours: int = 24,
+    ) -> str:
+        if not self.api_token:
+            raise RuntimeError("API token required to create assets")
+        url = f"{self.storage_url}/assets/"
+        headers = self._get_auth_header()
+        headers["Content-Type"] = "application/json"
+        payload = {
+            "name": name,
+            "mime_type": mime_type,
+            "contents": base64.b64encode(content.encode()).decode(),
+            "lifetime_hours": lifetime_hours,
+        }
+
+        response = requests.post(url, json=payload, headers=headers)
+        if response.status_code != 201:
+            raise RuntimeError(
+                f"Asset creation failed: {response.status_code}, {response.text}"
+            )
+
+        return response.json()["asset_id"]
+
+    def set_permissions(
+        self, asset_id: str, agent_address: str, read: bool = True, write: bool = True
+    ):
+        if not self.api_token:
+            raise RuntimeError("API token required to set permissions")
+        url = f"{self.storage_url}/assets/{asset_id}/permissions/"
+        headers = self._get_auth_header()
+        headers["Content-Type"] = "application/json"
+        payload = {
+            "agent_address": agent_address,
+            "read": read,
+            "write": write,
+        }
+
+        response = requests.put(url, json=payload, headers=headers)
+        if response.status_code != 200:
+            raise RuntimeError(
+                f"Set permissions failed: {response.status_code}, {response.text}"
+            )
+
+        return response.json()

{uagents_core-0.2.2 → uagents_core-0.3.0}/uagents_core/utils/messages.py

@@ -144,7 +144,7 @@ def send_message_to_agent(
             result.append(
                 MsgStatus(
                     status=DeliveryStatus.FAILED,
-                    detail=response.text,
+                    detail=str(e),
                     destination=destination,
                     endpoint=endpoint,
                     session=env.session,

{uagents_core-0.2.2 → uagents_core-0.3.0}/uagents_core/utils/registration.py

@@ -51,8 +51,11 @@ def _send_post_request(
         response.raise_for_status()
         return True, response
     except requests.RequestException as e:
+        error_detail = getattr(e, "response", None)
+        if error_detail is not None:
+            error_detail = error_detail.text
         logger.error(
-            msg="Error submitting request",
+            msg=f"Error submitting request: {error_detail}",
             extra={"url": url, "data": data.model_dump_json()},
             exc_info=e,
         )
@@ -63,6 +66,7 @@ def register_in_almanac(
     identity: Identity,
     endpoints: list[str],
     protocol_digests: list[str],
+    metadata: dict[str, str | dict[str, str]] | None = None,
     *,
     agentverse_config: AgentverseConfig | None = None,
     timeout: int = DEFAULT_REQUEST_TIMEOUT,
@@ -115,7 +119,7 @@ def register_in_almanac(
         agent_identifier=agent_address,
         protocols=protocol_digests,
         endpoints=agent_endpoints,
-        metadata=None,
+        metadata=metadata,
     )
 
     logger.info(msg="Registering with Almanac API", extra=attestation.model_dump())
@@ -220,11 +224,6 @@ def register_in_agentverse(
                 extra=registration_metadata,
             )
             return False
-        if response.status_code == 409:
-            logger.info(
-                msg="Agent already registered with Agentverse",
-                extra=registration_metadata,
-            )
         else:
             registration_response = RegistrationResponse.model_validate_json(
                 response.text