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

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/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.3.dist-info → uagents_core-0.3.0.dist-info}/METADATA

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

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

@@ -1,5 +1,5 @@
 uagents_core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-uagents_core/config.py,sha256=mvCXtvt_efDbmoTcw0JNHymWZGOtXxIYXlrooWfbY1Q,798
+uagents_core/config.py,sha256=Cz2c2qhS_0uqUPxG7mWiqQSXWI_rPrfC7VoTSwBkdWk,940
 uagents_core/contrib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uagents_core/contrib/protocols/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uagents_core/contrib/protocols/chat/__init__.py,sha256=EAt2uQeaZtrSEdt2SiFyqFcC4nxKtyoJRaHhjKtwngw,3032
@@ -11,11 +11,12 @@ 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/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/resolver.py,sha256=X18oe-WPU_6pfTc6x_Oa69kZQYCMidkaSeCs68Poaik,2169
-uagents_core-0.2.3.dist-info/METADATA,sha256=9_LsBarAzyOA-ttbyu_4Y5h8kLaVUTZiB3EyjkKKK8o,1009
-uagents_core-0.2.3.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-uagents_core-0.2.3.dist-info/RECORD,,
+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.2.3.dist-info → uagents_core-0.3.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.2
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any