agent0-sdk 0.31__tar.gz → 1.0.0__tar.gz

agent0_sdk-1.0.0/MANIFEST.in

@@ -0,0 +1,10 @@
+exclude .env
+global-exclude *.env
+global-exclude *.env.*
+
+# local artifacts
+prune venv
+exclude agent_registration_*.json
+
+# don't ship tests in sdist by default
+prune tests

{agent0_sdk-0.31 → agent0_sdk-1.0.0}/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.4
 Name: agent0-sdk
-Version: 0.31
+Version: 1.0.0
 Summary: Python SDK for agent portability, discovery and trust based on ERC-8004
-Author-email: Marco De Rossi <marco.derossi@consensys.net>
+Author-email: Marco De Rossi <marco@ag0.xyz>
 License: MIT License
         
         Copyright (c) 2025 Marco De Rossi
@@ -152,12 +152,15 @@ agent.setENS("myagent.eth")
 
 # Add OASF skills and domains (standardized taxonomies)
 agent.addSkill("data_engineering/data_transformation_pipeline", validate_oasf=True)
-agent.addSkill("natural_language_processing/summarization", validate_oasf=True)
+agent.addSkill("natural_language_processing/natural_language_generation/summarization", validate_oasf=True)
 agent.addDomain("finance_and_business/investment_services", validate_oasf=True)
-agent.addDomain("technology/data_science", validate_oasf=True)
+agent.addDomain("technology/data_science/data_science", validate_oasf=True)
 
 # Configure wallet and trust
-agent.setAgentWallet("0x742d35Cc6634C0532925a3b844Bc9e7595f0bEb", chainId=11155111)
+# Note: agentWallet is an on-chain verified attribute. setAgentWallet() is on-chain only.
+# EOAs: the NEW wallet must sign an EIP-712 message. If you pass new_wallet_signer, the SDK will
+# build + sign the typed data automatically.
+# If the current SDK signer address matches the new wallet, it can auto-sign without new_wallet_signer.
 agent.setTrust(reputation=True, cryptoEconomic=True)
 
 # Add metadata and set status
@@ -168,6 +171,16 @@ agent.setActive(True)
 agent.registerIPFS()
 print(f"Agent registered: {agent.agentId}")  # e.g., "11155111:123"
 print(f"Agent URI: {agent.agentURI}")  # e.g., "ipfs://Qm..."
+
+# (Optional) Change the agent wallet after registration
+# - On mint/registration, `agentWallet` defaults to the current owner address.
+# - Call this only if you want a DIFFERENT wallet (or after a transfer, since the wallet resets to zero).
+# - Transaction is sent by the SDK signer (agent owner), but the signature must be produced by the NEW wallet.
+agent.setAgentWallet(
+    "0x742d35Cc6634C0532925a3b844Bc9e7595f0bEb",
+    chainId=11155111,
+    new_wallet_signer=os.getenv("NEW_WALLET_PRIVATE_KEY"),
+)
 ```
 
 ### 3. Load and Edit Agent
@@ -213,7 +226,8 @@ agent_summary = sdk.getAgent("11155111:123")
 feedback_file = sdk.prepareFeedback(
     agentId="11155111:123",
     score=85,  # 0-100 (mandatory)
-    tags=["data_analyst", "finance"],  # Optional
+    tags=["data_analyst", "finance"],  # Optional: tags are now strings (not bytes32)
+    endpoint="https://example.com/endpoint",  # Optional: endpoint URI associated with feedback
     capability="tools",  # Optional: MCP capability
     name="code_generation",  # Optional: MCP tool name
     skill="python"  # Optional: A2A skill
@@ -307,7 +321,7 @@ OASF skills and domains appear in your agent's registration file:
       ],
       "domains": [
         "finance_and_business/investment_services",
-        "technology/data_science"
+        "technology/data_science/data_science"
       ]
     }
   ]

{agent0_sdk-0.31 → agent0_sdk-1.0.0}/README.md

@@ -82,12 +82,15 @@ agent.setENS("myagent.eth")
 
 # Add OASF skills and domains (standardized taxonomies)
 agent.addSkill("data_engineering/data_transformation_pipeline", validate_oasf=True)
-agent.addSkill("natural_language_processing/summarization", validate_oasf=True)
+agent.addSkill("natural_language_processing/natural_language_generation/summarization", validate_oasf=True)
 agent.addDomain("finance_and_business/investment_services", validate_oasf=True)
-agent.addDomain("technology/data_science", validate_oasf=True)
+agent.addDomain("technology/data_science/data_science", validate_oasf=True)
 
 # Configure wallet and trust
-agent.setAgentWallet("0x742d35Cc6634C0532925a3b844Bc9e7595f0bEb", chainId=11155111)
+# Note: agentWallet is an on-chain verified attribute. setAgentWallet() is on-chain only.
+# EOAs: the NEW wallet must sign an EIP-712 message. If you pass new_wallet_signer, the SDK will
+# build + sign the typed data automatically.
+# If the current SDK signer address matches the new wallet, it can auto-sign without new_wallet_signer.
 agent.setTrust(reputation=True, cryptoEconomic=True)
 
 # Add metadata and set status
@@ -98,6 +101,16 @@ agent.setActive(True)
 agent.registerIPFS()
 print(f"Agent registered: {agent.agentId}")  # e.g., "11155111:123"
 print(f"Agent URI: {agent.agentURI}")  # e.g., "ipfs://Qm..."
+
+# (Optional) Change the agent wallet after registration
+# - On mint/registration, `agentWallet` defaults to the current owner address.
+# - Call this only if you want a DIFFERENT wallet (or after a transfer, since the wallet resets to zero).
+# - Transaction is sent by the SDK signer (agent owner), but the signature must be produced by the NEW wallet.
+agent.setAgentWallet(
+    "0x742d35Cc6634C0532925a3b844Bc9e7595f0bEb",
+    chainId=11155111,
+    new_wallet_signer=os.getenv("NEW_WALLET_PRIVATE_KEY"),
+)
 ```
 
 ### 3. Load and Edit Agent
@@ -143,7 +156,8 @@ agent_summary = sdk.getAgent("11155111:123")
 feedback_file = sdk.prepareFeedback(
     agentId="11155111:123",
     score=85,  # 0-100 (mandatory)
-    tags=["data_analyst", "finance"],  # Optional
+    tags=["data_analyst", "finance"],  # Optional: tags are now strings (not bytes32)
+    endpoint="https://example.com/endpoint",  # Optional: endpoint URI associated with feedback
     capability="tools",  # Optional: MCP capability
     name="code_generation",  # Optional: MCP tool name
     skill="python"  # Optional: A2A skill
@@ -237,7 +251,7 @@ OASF skills and domains appear in your agent's registration file:
       ],
       "domains": [
         "finance_and_business/investment_services",
-        "technology/data_science"
+        "technology/data_science/data_science"
       ]
     }
   ]

{agent0_sdk-0.31 → agent0_sdk-1.0.0}/agent0_sdk/__init__.py

@@ -30,7 +30,7 @@ except ImportError:
     Agent = None
     _sdk_available = False
 
-__version__ = "0.31"
+__version__ = "1.0.0"
 __all__ = [
     "SDK",
     "Agent",

{agent0_sdk-0.31 → agent0_sdk-1.0.0}/agent0_sdk/core/agent.py

@@ -176,16 +176,14 @@ class Agent:
         return self.registration_file
 
     def _collectMetadataForRegistration(self) -> List[Dict[str, Any]]:
-        """Collect all metadata entries for registration."""
+        """Collect all metadata entries for registration.
+        
+        Note: agentWallet is now a reserved metadata key and cannot be set via setMetadata().
+        It must be set separately using setAgentWallet() with EIP-712 signature verification.
+        """
         metadata_entries = []
         
-        # Add wallet address metadata
-        if self.walletAddress:
-            addr_bytes = bytes.fromhex(self.walletAddress[2:])  # Remove '0x' prefix
-            metadata_entries.append({
-                "key": "agentWallet",
-                "value": addr_bytes
-            })
+        # Note: agentWallet is no longer set via metadata - it's now reserved and managed via setAgentWallet()
         
         # Add ENS name metadata
         if self.ensEndpoint:
@@ -339,7 +337,7 @@ class Agent:
         Add a skill to the OASF endpoint.
         
         Args:
-            slug: The skill slug to add (e.g., "natural_language_processing/summarization")
+            slug: The skill slug to add (e.g., "natural_language_processing/natural_language_generation/summarization")
             validate_oasf: If True, validate the slug against the OASF taxonomy (default: False)
         
         Returns:
@@ -492,20 +490,55 @@ class Agent:
         self.registration_file.updatedAt = int(time.time())
         return self
 
-    def setAgentWallet(self, addr: Optional[Address], chainId: Optional[int] = None) -> 'Agent':
-        """Set agent wallet address in registration file (will be saved on-chain during next register call)."""
-        # Validate address format if provided
-        if addr:
-            if not addr.startswith("0x") or len(addr) != 42:
-                raise ValueError(f"Invalid Ethereum address format: {addr}. Must be 42 characters starting with '0x'")
-            
-            # Validate hexadecimal characters
-            try:
-                int(addr[2:], 16)
-            except ValueError:
-                raise ValueError(f"Invalid hexadecimal characters in address: {addr}")
+    def setAgentWallet(
+        self,
+        new_wallet: Address,
+        chainId: Optional[int] = None,
+        *,
+        new_wallet_signer: Optional[Union[str, Any]] = None,
+        deadline: Optional[int] = None,
+        signature: Optional[bytes] = None,
+    ) -> 'Agent':
+        """Set agent wallet address on-chain (ERC-8004 agentWallet).
+
+        This method is **on-chain only**. The `agentWallet` is a verified attribute and must be set via
+        the IdentityRegistry `setAgentWallet` function.
+
+        EOAs: provide `new_wallet_signer` (private key string or eth-account account) OR ensure the SDK
+        signer address matches `new_wallet` so the SDK can auto-sign.\n
+        Contract wallets (ERC-1271): provide `signature` bytes produced by the wallet’s signing mechanism.
+        The SDK will build the correct EIP-712 typed data internally, but cannot produce the wallet signature.
+
+        Args:
+            new_wallet: New wallet address (must be controlled by the signer that produces the signature)
+            chainId: Optional local bookkeeping for registration file (walletChainId). Defaults to agent chain.
+            new_wallet_signer: EOA signer used to sign the EIP-712 message (private key string or eth-account account)
+            deadline: Signature deadline timestamp. Defaults to now+60s (must be <= now+5min per contract).
+            signature: Raw signature bytes (intended for ERC-1271 / external signing only)
+        """
+        # Breaking/clean: this API is only meaningful for already-registered agents.
+        if not self.agentId:
+            raise ValueError(
+                "Cannot set agent wallet before the agent is registered on-chain. "
+                "Call agent.register(...) / agent.registerIPFS() first to obtain agentId."
+            )
+
+        addr = new_wallet
+
+        if not addr:
+            raise ValueError("Wallet address cannot be empty. Use a non-zero address.")
         
-        # Determine chain ID to use
+        # Validate address format
+        if not addr.startswith("0x") or len(addr) != 42:
+            raise ValueError(f"Invalid Ethereum address format: {addr}. Must be 42 characters starting with '0x'")
+        
+        # Validate hexadecimal characters
+        try:
+            int(addr[2:], 16)
+        except ValueError:
+            raise ValueError(f"Invalid hexadecimal characters in address: {addr}")
+        
+        # Determine chain ID to use (local bookkeeping)
         if chainId is None:
             # Extract chain ID from agentId if available, otherwise use SDK's chain ID
             if self.agentId and ":" in self.agentId:
@@ -516,14 +549,104 @@ class Agent:
             else:
                 chainId = self.sdk.chainId  # Use SDK's chain ID as fallback
         
-        # Check if wallet changed
-        if addr != self._last_registered_wallet:
-            self._dirty_metadata.add("agentWallet")
+        # Parse agent ID
+        agent_id_int = int(self.agentId.split(":")[-1]) if ":" in self.agentId else int(self.agentId)
+
+        # Check if wallet is already set to this address (skip if same)
+        try:
+            current_wallet = self.sdk.web3_client.call_contract(
+                self.sdk.identity_registry,
+                "getAgentWallet",
+                agent_id_int
+            )
+            if current_wallet and current_wallet.lower() == addr.lower():
+                logger.debug(f"Agent wallet is already set to {addr}, skipping on-chain update")
+                # Still update local registration file
+                self.registration_file.walletAddress = addr
+                self.registration_file.walletChainId = chainId
+                self.registration_file.updatedAt = int(time.time())
+                return self
+        except Exception as e:
+            logger.debug(f"Could not check current agent wallet: {e}, proceeding with update")
+        
+        # Set deadline (default to 60 seconds from now; contract max is now+5min)
+        if deadline is None:
+            deadline = int(time.time()) + 60
+        
+        # Resolve typed data + signature
+        identity_registry_address = self.sdk.identity_registry.address
+        owner_address = self.sdk.web3_client.call_contract(self.sdk.identity_registry, "ownerOf", agent_id_int)
+
+        full_message = self.sdk.web3_client.build_agent_wallet_set_typed_data(
+            agent_id=agent_id_int,
+            new_wallet=addr,
+            owner=owner_address,
+            deadline=deadline,
+            verifying_contract=identity_registry_address,
+            chain_id=self.sdk.web3_client.chain_id,
+        )
+
+        if signature is None:
+            # EOA signing paths
+            if new_wallet_signer is not None:
+                # Validate signer address matches addr (fail fast)
+                try:
+                    from eth_account import Account as _Account
+                    if isinstance(new_wallet_signer, str):
+                        signer_addr = _Account.from_key(new_wallet_signer).address
+                    else:
+                        signer_addr = getattr(new_wallet_signer, "address", None)
+                except Exception:
+                    signer_addr = getattr(new_wallet_signer, "address", None)
+
+                if not signer_addr or signer_addr.lower() != addr.lower():
+                    raise ValueError(
+                        f"new_wallet_signer address ({signer_addr}) does not match new_wallet ({addr})."
+                    )
+
+                signature = self.sdk.web3_client.sign_typed_data(full_message, new_wallet_signer)  # type: ignore[arg-type]
+            else:
+                # Auto-sign only if SDK signer == new wallet
+                current_address = self.sdk.web3_client.account.address if self.sdk.web3_client.account else None
+                if current_address and current_address.lower() == addr.lower():
+                    signature = self.sdk.web3_client.sign_typed_data(full_message, self.sdk.web3_client.account)
+                else:
+                    raise ValueError(
+                        f"New wallet must sign. Provide new_wallet_signer (EOA) or signature (ERC-1271/external). "
+                        f"SDK signer is {current_address}, new_wallet is {addr}."
+                    )
+
+            # Optional: verify recover matches addr for EOA signatures
+            recovered = self.sdk.web3_client.w3.eth.account.recover_message(
+                __import__("eth_account.messages").messages.encode_typed_data(full_message=full_message),
+                signature=signature,
+            )
+            if recovered.lower() != addr.lower():
+                raise ValueError(f"Signature verification failed: recovered {recovered} but expected {addr}")
+        
+        # Call setAgentWallet on the contract
+        try:
+            txHash = self.sdk.web3_client.transact_contract(
+                self.sdk.identity_registry,
+                "setAgentWallet",
+                agent_id_int,
+                addr,
+                deadline,
+                signature
+            )
+            
+            # Wait for transaction
+            receipt = self.sdk.web3_client.wait_for_transaction(txHash)
+            logger.debug(f"Agent wallet set on-chain: {txHash}")
+            
+        except Exception as e:
+            raise ValueError(f"Failed to set agent wallet on-chain: {e}")
         
         # Update local registration file
         self.registration_file.walletAddress = addr
         self.registration_file.walletChainId = chainId
         self.registration_file.updatedAt = int(time.time())
+        self._last_registered_wallet = addr
         
         return self
 
@@ -636,7 +759,7 @@ class Agent:
             agentId = int(self.agentId.split(":")[-1])
             txHash = self.sdk.web3_client.transact_contract(
                 self.sdk.identity_registry,
-                "setAgentUri",
+                "setAgentURI",
                 agentId,
                 f"ipfs://{ipfsCid}"
             )
@@ -673,7 +796,7 @@ class Agent:
             agentId = int(self.agentId.split(":")[-1])
             txHash = self.sdk.web3_client.transact_contract(
                 self.sdk.identity_registry,
-                "setAgentUri",
+                "setAgentURI",
                 agentId,
                 f"ipfs://{ipfsCid}"
             )
@@ -715,7 +838,7 @@ class Agent:
         txHash = self.sdk.web3_client.transact_contract(
             self.sdk.identity_registry,
             "register",
-            "",  # Empty tokenUri for now
+            "",  # Empty agentURI for now
             metadata_entries
         )
         
@@ -820,7 +943,7 @@ class Agent:
             agentId = int(self.registration_file.agentId.split(":")[-1])
             txHash = self.sdk.web3_client.transact_contract(
                 self.sdk.identity_registry,
-                "setAgentUri",
+                "setAgentURI",
                 agentId,
                 agentURI
             )
@@ -846,7 +969,12 @@ class Agent:
         approve_operator: bool = False,
         idem: Optional[IdemKey] = None,
     ) -> Dict[str, Any]:
-        """Transfer agent ownership."""
+        """Transfer agent ownership.
+        
+        Note: When an agent is transferred, the agentWallet is automatically reset
+        to the zero address on-chain. The new owner must call setAgentWallet() to
+        set a new wallet address with EIP-712 signature verification.
+        """
         if not self.registration_file.agentId:
             raise ValueError("Agent must be registered before transferring")
         
@@ -863,6 +991,11 @@ class Agent:
         
         receipt = self.sdk.web3_client.wait_for_transaction(txHash)
         
+        # Note: agentWallet will be reset to zero address by the contract
+        # Update local state to reflect this
+        self.registration_file.walletAddress = None
+        self._last_registered_wallet = None
+        
         return {
             "txHash": txHash,
             "agentId": self.registration_file.agentId,
@@ -907,6 +1040,10 @@ class Agent:
         
         Only the current owner can transfer the agent.
         
+        Note: When an agent is transferred, the agentWallet is automatically reset
+        to the zero address on-chain. The new owner must call setAgentWallet() to
+        set a new wallet address with EIP-712 signature verification.
+        
         Args:
             newOwnerAddress: Ethereum address of the new owner
             
@@ -964,6 +1101,11 @@ class Agent:
         
         logger.debug(f"Agent {self.registration_file.agentId} successfully transferred to {checksum_address}")
         
+        # Note: agentWallet will be reset to zero address by the contract
+        # Update local state to reflect this
+        self.registration_file.walletAddress = None
+        self._last_registered_wallet = None
+        
         return {"txHash": txHash, "from": currentOwner, "to": checksum_address, "agentId": self.registration_file.agentId}
 
     def activate(self, idem: Optional[IdemKey] = None) -> RegistrationFile: