agent0-sdk 0.2.2__py3-none-any.whl → 0.5__py3-none-any.whl

tests/test_search.py

@@ -1,415 +0,0 @@
-"""
-Test for Agent Search and Discovery using Subgraph
-Tests various search and filtering capabilities for discovering agents and their reputation.
-
-Flow:
-1. Get a specific agent by ID
-2. Search agents by name (partial match)
-3. Search agents by capabilities (MCP tools)
-4. Search agents by skills (A2A)
-5. Search agents by ENS domain
-6. Search agents by active status
-7. Combine multiple filters (capabilities + skills)
-8. Search agents by reputation with minimum average score
-9. Search agents by reputation with specific tags
-10. Search agents by reputation with capability filtering
-11. Advanced: Find top-rated agents with specific skills
-12. Advanced: Find agents with multiple requirements
-13. Pagination test
-14. Sort by activity test
-15. Search agents by single owner address
-16. Search agents by multiple owner addresses
-17. Search agents by operator addresses
-18. Combined search: owner + active status
-19. Combined search: owner + name filter
-"""
-
-import logging
-import time
-import random
-import sys
-
-# Configure logging: root logger at WARNING to suppress noisy dependencies
-logging.basicConfig(
-    level=logging.WARNING,
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
-    datefmt='%Y-%m-%d %H:%M:%S',
-    handlers=[logging.StreamHandler(sys.stdout)]
-)
-
-# Set debug level ONLY for agent0_sdk
-logging.getLogger('agent0_sdk').setLevel(logging.DEBUG)
-logging.getLogger('agent0_sdk.core').setLevel(logging.DEBUG)
-
-from agent0_sdk import SDK, SearchParams
-from config import CHAIN_ID, RPC_URL, AGENT_PRIVATE_KEY, SUBGRAPH_URL, AGENT_ID, print_config
-
-
-def main():
-    print("🔍 Testing Agent Search and Discovery")
-    print_config()
-    print("=" * 60)
-    
-    # Initialize SDK without signer (read-only operations)
-    sdk = SDK(
-        chainId=CHAIN_ID,
-        rpcUrl=RPC_URL
-    )
-    
-    print(f"\n📍 Step 1: Get Agent by ID")
-    print("-" * 60)
-    try:
-        agent = sdk.getAgent(AGENT_ID)
-        print(f"✅ Agent found: {agent.name}")
-        print(f"   Description: {agent.description[:80] if agent.description else 'N/A'}...")
-        print(f"   Chain ID: {agent.chainId}")
-        print(f"   Active: {agent.active}")
-        print(f"   ENS: {agent.ens}")
-        print(f"   Agent ID: {agent.agentId}")
-        # MCP and A2A support determined by endpoint presence
-        if hasattr(agent, 'mcpEndpoint') and agent.mcpEndpoint:
-            print(f"   MCP: {agent.mcpEndpoint}")
-        if hasattr(agent, 'a2aEndpoint') and agent.a2aEndpoint:
-            print(f"   A2A: {agent.a2aEndpoint}")
-    except Exception as e:
-        print(f"❌ Failed to get agent: {e}")
-    
-    print(f"\n📍 Step 2: Search Agents by Name (Partial Match)")
-    print("-" * 60)
-    try:
-        results = sdk.searchAgents(name="Test")
-        agents = results.get('items', [])
-        print(f"✅ Found {len(agents)} agent(s) with name matching 'Test'")
-        for i, agent in enumerate(agents[:3], 1):
-            agent_id = agent.get('agentId', agent.get('id', 'N/A'))
-            print(f"   {i}. {agent['name']} (ID: {agent_id})")
-            if agent.get('description'):
-                print(f"      {agent['description'][:60]}...")
-    except Exception as e:
-        print(f"❌ Failed to search by name: {e}")
-    
-    print(f"\n📍 Step 3: Search Agents by MCP Tools (Capabilities)")
-    print("-" * 60)
-    try:
-        # Using capabilities from test_feedback.py: data_analysis, code_generation, natural_language_understanding, problem_solving, communication
-        results = sdk.searchAgents(mcpTools=["data_analysis"])
-        agents = results.get('items', [])
-        print(f"✅ Found {len(agents)} agent(s) with 'data_analysis' capability")
-        for i, agent in enumerate(agents[:3], 1):
-            print(f"   {i}. {agent['name']}")
-            if agent.get('mcpTools'):
-                print(f"      Tools: {', '.join(agent['mcpTools'][:3])}...")
-    except Exception as e:
-        print(f"❌ Failed to search by capabilities: {e}")
-    
-    print(f"\n📍 Step 4: Search Agents by A2A Skills")
-    print("-" * 60)
-    try:
-        # Using skills from test_feedback.py: python, javascript, machine_learning, web_development, cloud_computing
-        results = sdk.searchAgents(a2aSkills=["javascript"])
-        agents = results.get('items', [])
-        print(f"✅ Found {len(agents)} agent(s) with 'javascript' skill")
-        for i, agent in enumerate(agents[:3], 1):
-            print(f"   {i}. {agent['name']}")
-            if agent.get('a2aSkills'):
-                print(f"      Skills: {', '.join(agent['a2aSkills'][:3])}...")
-    except Exception as e:
-        print(f"❌ Failed to search by skills: {e}")
-    
-    print(f"\n📍 Step 5: Search Agents by ENS Domain")
-    print("-" * 60)
-    try:
-        results = sdk.searchAgents(ens="test")
-        agents = results.get('items', [])
-        print(f"✅ Found {len(agents)} agent(s) with ENS matching 'test'")
-        for i, agent in enumerate(agents[:3], 1):
-            print(f"   {i}. {agent['name']}")
-            if agent.get('ens'):
-                print(f"      ENS: {agent['ens']}")
-    except Exception as e:
-        print(f"❌ Failed to search by ENS: {e}")
-    
-    print(f"\n📍 Step 6: Search Only Active Agents")
-    print("-" * 60)
-    try:
-        results = sdk.searchAgents(active=True, page_size=10)
-        agents = results.get('items', [])
-        print(f"✅ Found {len(agents)} active agent(s)")
-        for i, agent in enumerate(agents[:3], 1):
-            print(f"   {i}. {agent['name']} - {agent.get('totalFeedback', 0)} feedback")
-    except Exception as e:
-        print(f"❌ Failed to search active agents: {e}")
-    
-    print(f"\n📍 Step 7: Search Agents with Multiple Filters (Capabilities + Skills)")
-    print("-" * 60)
-    try:
-        # Using capabilities and skills from test_feedback.py:
-        # Capabilities: data_analysis, code_generation, natural_language_understanding, problem_solving, communication
-        # Skills: python, javascript, machine_learning, web_development, cloud_computing
-        results = sdk.searchAgents(
-            mcpTools=["communication"],
-            a2aSkills=["python"]
-        )
-        agents = results.get('items', [])
-        print(f"✅ Found {len(agents)} agent(s) with 'communication' capability AND 'python' skill")
-        for i, agent in enumerate(agents[:3], 1):
-            print(f"   {i}. {agent['name']}")
-    except Exception as e:
-        print(f"❌ Failed to search with multiple filters: {e}")
-    
-    print(f"\n📍 Step 8: Search Agents by Reputation (Minimum Average Score)")
-    print("-" * 60)
-    try:
-        results = sdk.searchAgentsByReputation(minAverageScore=80)
-        agents = results.get('items', [])
-        print(f"✅ Found {len(agents)} agent(s) with average score >= 80")
-        for i, agent in enumerate(agents[:3], 1):
-            # AgentSummary object - use attributes, not dict.get()
-            avg_score = agent.extras.get('averageScore', 'N/A') if agent.extras else 'N/A'
-            print(f"   {i}. {agent.name}")
-            print(f"      Average Score: {avg_score}")
-            print(f"      Agent ID: {agent.agentId}")
-    except Exception as e:
-        print(f"❌ Failed to search by reputation score: {e}")
-    
-    print(f"\n📍 Step 9: Search Agents by Reputation with Specific Tags")
-    print("-" * 60)
-    try:
-        # Using tags that actually exist in feedback: "enterprise" appears frequently
-        # Note: GraphQL query has known issue with mixing filters, so using includeRevoked=True as workaround
-        # which may affect results, but at least the query will execute
-        results = sdk.searchAgentsByReputation(
-            tags=["enterprise"],
-            minAverageScore=0,  # No threshold to see any results
-            includeRevoked=True  # Workaround for GraphQL query builder issue
-        )
-        agents = results.get('items', [])
-        print(f"✅ Found {len(agents)} agent(s) with 'enterprise' tag")
-        for i, agent in enumerate(agents[:3], 1):
-            # AgentSummary object - use attributes
-            avg_score = agent.extras.get('averageScore', 'N/A') if agent.extras else 'N/A'
-            print(f"   {i}. {agent.name} - Avg: {avg_score}")
-    except Exception as e:
-        print(f"❌ Failed to search by reputation tags: {e}")
-    
-    print(f"\n📍 Step 10: Search Agents by Reputation with Capability Filtering")
-    print("-" * 60)
-    try:
-        # Using capabilities that actually exist in feedback: code_generation, problem_solving, data_analysis
-        results = sdk.searchAgentsByReputation(
-            capabilities=["code_generation"],
-            minAverageScore=0  # No threshold to see any results
-        )
-        agents = results.get('items', [])
-        print(f"✅ Found {len(agents)} agent(s) with 'code_generation' capability")
-        for i, agent in enumerate(agents[:3], 1):
-            avg_score = agent.extras.get('averageScore', 'N/A') if hasattr(agent, 'extras') and agent.extras else 'N/A'
-            print(f"   {i}. {agent.name}")
-            print(f"      Avg Score: {avg_score}")
-    except Exception as e:
-        print(f"❌ Failed to search by reputation with capabilities: {e}")
-    
-    print(f"\n📍 Step 11: Advanced - Find Top-Rated Agents with Specific Skills")
-    print("-" * 60)
-    try:
-        # Using skills that actually exist in feedback: python, machine_learning, cloud_computing, web_development
-        results = sdk.searchAgentsByReputation(
-            skills=["python"],
-            minAverageScore=0  # No threshold to see any results
-        )
-        agents = results.get('items', [])
-        print(f"✅ Found {len(agents)} agent(s) with 'python' skill")
-        for i, agent in enumerate(agents[:3], 1):
-            # AgentSummary object - use attributes
-            avg_score = agent.extras.get('averageScore', 'N/A') if agent.extras else 'N/A'
-            print(f"   {i}. {agent.name}")
-            print(f"      Average Score: {avg_score}")
-            if agent.a2aSkills:
-                print(f"      Skills: {', '.join(agent.a2aSkills[:3])}")
-    except Exception as e:
-        print(f"❌ Failed advanced skill search: {e}")
-    
-    print(f"\n📍 Step 12: Advanced - Complex Multi-Criteria Search")
-    print("-" * 60)
-    try:
-        results = sdk.searchAgents(name="Test", active=True, page_size=10)
-        agents = results.get('items', [])
-        print(f"✅ Found {len(agents)} agent(s) matching multiple criteria")
-        for i, agent in enumerate(agents[:3], 1):
-            print(f"   {i}. {agent['name']}")
-            print(f"      Active: {agent.get('active', False)}, Feedback: {agent.get('totalFeedback', 0)}")
-            if agent.get('ens'):
-                print(f"      ENS: {agent['ens']}")
-    except Exception as e:
-        print(f"❌ Failed complex search: {e}")
-    
-    print(f"\n📍 Step 13: Pagination Test (First 5 Agents)")
-    print("-" * 60)
-    try:
-        results = sdk.searchAgents(page_size=5)
-        agents = results.get('items', [])
-        print(f"✅ Retrieved first page: {len(agents)} agent(s)")
-        for i, agent in enumerate(agents, 1):
-            agent_id = agent.get('agentId', agent.get('id', 'N/A'))
-            print(f"   {i}. {agent['name']} (ID: {agent_id})")
-    except Exception as e:
-        print(f"❌ Failed pagination test: {e}")
-    
-    print(f"\n📍 Step 14: Sort by Activity (Most Recent)")
-    print("-" * 60)
-    try:
-        results = sdk.searchAgents(page_size=10, sort=["updatedAt:desc"])
-        agents = results.get('items', [])
-        if agents:
-            print(f"✅ Retrieved {len(agents)} agents")
-            print(f"   Most recent activity IDs:")
-            for i, agent in enumerate(agents[:5], 1):
-                agent_id = agent.get('agentId', agent.get('id', 'N/A'))
-                print(f"   {i}. ID {agent_id}: {agent['name']}")
-    except Exception as e:
-        print(f"❌ Failed activity sort: {e}")
-
-    print(f"\n📍 Step 15: Search Agents by Single Owner Address")
-    print("-" * 60)
-    try:
-        # First, get an agent to extract its owner address for testing
-        test_agent = sdk.getAgent(AGENT_ID)
-        if hasattr(test_agent, 'owner') and test_agent.owner:
-            owner_address = test_agent.owner
-            print(f"   Testing with owner address: {owner_address}")
-
-            results = sdk.searchAgents(owners=[owner_address])
-            agents = results.get('items', [])
-            print(f"✅ Found {len(agents)} agent(s) owned by {owner_address[:10]}...{owner_address[-8:]}")
-
-            for i, agent in enumerate(agents[:3], 1):
-                agent_id = agent.get('agentId', agent.get('id', 'N/A'))
-                agent_owner = agent.get('owner', 'N/A')
-                print(f"   {i}. {agent['name']} (ID: {agent_id})")
-                print(f"      Owner: {agent_owner[:10]}...{agent_owner[-8:] if len(agent_owner) > 18 else agent_owner}")
-        else:
-            print("⚠️  Test agent doesn't have owner information")
-    except Exception as e:
-        print(f"❌ Failed to search by single owner: {e}")
-
-    print(f"\n📍 Step 16: Search Agents by Multiple Owner Addresses")
-    print("-" * 60)
-    try:
-        # Get multiple agents to collect different owner addresses
-        results = sdk.searchAgents(page_size=5)
-        agents = results.get('items', [])
-        owner_addresses = []
-
-        for agent in agents:
-            if agent.get('owner') and agent['owner'] not in owner_addresses:
-                owner_addresses.append(agent['owner'])
-                if len(owner_addresses) >= 2:
-                    break
-
-        if len(owner_addresses) >= 2:
-            print(f"   Testing with {len(owner_addresses)} owner addresses")
-            results = sdk.searchAgents(owners=owner_addresses)
-            agents = results.get('items', [])
-            print(f"✅ Found {len(agents)} agent(s) owned by multiple addresses")
-
-            for i, agent in enumerate(agents[:5], 1):
-                agent_id = agent.get('agentId', agent.get('id', 'N/A'))
-                agent_owner = agent.get('owner', 'N/A')
-                print(f"   {i}. {agent['name']} (Owner: {agent_owner[:10]}...{agent_owner[-8:] if len(agent_owner) > 18 else agent_owner})")
-        else:
-            print("⚠️  Not enough different owners found for multi-owner test")
-    except Exception as e:
-        print(f"❌ Failed to search by multiple owners: {e}")
-
-    print(f"\n📍 Step 17: Search Agents by Operator Addresses")
-    print("-" * 60)
-    try:
-        # First check if any agents have operators defined
-        results = sdk.searchAgents(page_size=10)
-        agents = results.get('items', [])
-        test_operators = []
-
-        for agent in agents:
-            if agent.get('operators') and len(agent['operators']) > 0:
-                test_operators = agent['operators'][:1]  # Use first operator
-                print(f"   Testing with operator: {test_operators[0][:10]}...{test_operators[0][-8:]}")
-                break
-
-        if test_operators:
-            results = sdk.searchAgents(operators=test_operators)
-            found_agents = results.get('items', [])
-            print(f"✅ Found {len(found_agents)} agent(s) with specified operator")
-
-            for i, agent in enumerate(found_agents[:3], 1):
-                agent_id = agent.get('agentId', agent.get('id', 'N/A'))
-                agent_operators = agent.get('operators', [])
-                print(f"   {i}. {agent['name']} (ID: {agent_id})")
-                if agent_operators:
-                    print(f"      Operators: {len(agent_operators)} total")
-        else:
-            print("⚠️  No agents with operators found for testing")
-    except Exception as e:
-        print(f"❌ Failed to search by operators: {e}")
-
-    print(f"\n📍 Step 18: Combined Search - Owner + Active Status")
-    print("-" * 60)
-    try:
-        # Get an owner address from an active agent
-        results = sdk.searchAgents(active=True, page_size=5)
-        agents = results.get('items', [])
-
-        if agents and agents[0].get('owner'):
-            test_owner = agents[0]['owner']
-            print(f"   Testing owner {test_owner[:10]}...{test_owner[-8:]} + active=True")
-
-            results = sdk.searchAgents(owners=[test_owner], active=True)
-            found_agents = results.get('items', [])
-            print(f"✅ Found {len(found_agents)} active agent(s) owned by specified address")
-
-            for i, agent in enumerate(found_agents[:3], 1):
-                agent_id = agent.get('agentId', agent.get('id', 'N/A'))
-                print(f"   {i}. {agent['name']} (ID: {agent_id})")
-                print(f"      Active: {agent.get('active', False)}, Owner: {agent.get('owner', 'N/A')[:10]}...")
-        else:
-            print("⚠️  No active agents with owner information found")
-    except Exception as e:
-        print(f"❌ Failed combined owner + active search: {e}")
-
-    print(f"\n📍 Step 19: Combined Search - Owner + Name Filter")
-    print("-" * 60)
-    try:
-        # Get agents and find one with both name and owner
-        results = sdk.searchAgents(page_size=10)
-        agents = results.get('items', [])
-
-        test_agent = None
-        for agent in agents:
-            if agent.get('owner') and agent.get('name'):
-                test_agent = agent
-                break
-
-        if test_agent:
-            test_owner = test_agent['owner']
-            # Use partial name for search
-            name_part = test_agent['name'][:4] if len(test_agent['name']) > 4 else test_agent['name']
-            print(f"   Testing owner filter + name contains '{name_part}'")
-
-            results = sdk.searchAgents(owners=[test_owner], name=name_part)
-            found_agents = results.get('items', [])
-            print(f"✅ Found {len(found_agents)} agent(s) matching both criteria")
-
-            for i, agent in enumerate(found_agents[:3], 1):
-                agent_id = agent.get('agentId', agent.get('id', 'N/A'))
-                print(f"   {i}. {agent['name']} (ID: {agent_id})")
-        else:
-            print("⚠️  No suitable test agent found")
-    except Exception as e:
-        print(f"❌ Failed combined owner + name search: {e}")
-
-    print("\n" + "=" * 60)
-    print("✅ Search Tests Completed!")
-    print("=" * 60)
-
-
-if __name__ == "__main__":
-    main()

tests/test_transfer.py

@@ -1,255 +0,0 @@
-#!/usr/bin/env python3
-"""
-Test agent transfer functionality.
-
-This script demonstrates:
-1. Creating and registering an agent with owner A
-2. Transferring agent from owner A to owner B
-3. Verifying ownership changed on-chain
-4. Attempting to transfer from non-owner (should fail)
-5. Verifying agent metadata remains unchanged after transfer
-"""
-
-import logging
-import time
-import sys
-
-# Configure logging: root logger at WARNING to suppress noisy dependencies
-logging.basicConfig(
-    level=logging.WARNING,
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
-    datefmt='%Y-%m-%d %H:%M:%S',
-    handlers=[logging.StreamHandler(sys.stdout)]
-)
-
-# Set debug level ONLY for agent0_sdk
-logging.getLogger('agent0_sdk').setLevel(logging.DEBUG)
-logging.getLogger('agent0_sdk.core').setLevel(logging.DEBUG)
-
-from config import (
-    CHAIN_ID, RPC_URL, AGENT_PRIVATE_KEY, PINATA_JWT, SUBGRAPH_URL,
-    print_config
-)
-from agent0_sdk import SDK, EndpointType, TrustModel
-
-def main():
-    print("=" * 60)
-    print("🔄 AGENT TRANSFER TEST")
-    print("=" * 60)
-    
-    print_config()
-    
-    # Configuration for SDK
-    sdkConfig_pinata = {
-        "chainId": CHAIN_ID,
-        "rpcUrl": RPC_URL,
-        "signer": AGENT_PRIVATE_KEY,
-        "ipfs": "pinata",
-        "pinataJwt": PINATA_JWT
-        # Subgraph URL auto-defaults from DEFAULT_SUBGRAPH_URLS
-    }
-    
-    # Initialize SDK with Pinata
-    print("\n📡 Initializing SDK with Pinata...")
-    agentSdk = SDK(**sdkConfig_pinata)
-    
-    # Create a second private key for testing transfer
-    # In a real scenario, this would be a different wallet
-    # For testing, we'll use a different address derived from the same key
-    print("\n🔑 Setting up test accounts...")
-    ownerA_address = agentSdk.web3_client.account.address
-    print(f"Owner A address: {ownerA_address}")
-    
-    # For testing purposes, we'll use a different address
-    # In practice, this would be a completely different private key
-    ownerB_address = "0x742d35Cc6634C0532925a3b8D4C9db96C4b4d8b6"  # Example address
-    print(f"Owner B address: {ownerB_address}")
-    
-    print("\n" + "=" * 60)
-    print("STEP 1: CREATE AND REGISTER AGENT WITH OWNER A")
-    print("=" * 60)
-    
-    # Create agent
-    agent = agentSdk.createAgent(
-        name="Transfer Test Agent",
-        description="An agent for testing transfer functionality",
-        image="https://example.com/transfer-test-agent.png"
-    )
-    
-    # Configure agent details
-    agent.setAgentWallet(ownerA_address, CHAIN_ID)
-    agent.setENS("transfer-test-agent.eth")
-    agent.setMetadata({
-        "version": "1.0",
-        "category": "test",
-        "transfer_test": True
-    })
-    
-    # Add endpoints
-    agent.setMCP("https://mcp.example.com/transfer-test", auto_fetch=False)
-    agent.setA2A("https://a2a.example.com/transfer-test-agent.json", auto_fetch=False)
-    
-    print(f"✅ Agent created with ID: {agent.agentId}")
-    print(f"📄 Registration file prepared")
-    
-    # Register agent on-chain
-    print("\n📝 Registering agent on-chain...")
-    registration_result = agent.registerIPFS()
-    
-    print(f"✅ Agent registered successfully!")
-    print(f"   Agent ID: {registration_result.agentId}")
-    print(f"   IPFS URI: {registration_result.agentURI}")
-    
-    # Verify initial ownership using utility function
-    print("\n🔍 Verifying initial ownership...")
-    current_owner = agentSdk.getAgentOwner(registration_result.agentId)
-    print(f"✅ Current owner: {current_owner}")
-    print(f"✅ Matches Owner A: {current_owner.lower() == ownerA_address.lower()}")
-    
-    print("\n" + "=" * 60)
-    print("STEP 2: TRANSFER AGENT TO OWNER B")
-    print("=" * 60)
-    
-    # Transfer agent using Agent.transfer() method
-    print(f"🔄 Transferring agent {registration_result.agentId} to {ownerB_address}...")
-    
-    try:
-        transfer_result = agent.transfer(ownerB_address)
-        print(f"✅ Transfer successful!")
-        print(f"   Transaction: {transfer_result['txHash']}")
-        print(f"   From: {transfer_result['from']}")
-        print(f"   To: {transfer_result['to']}")
-        print(f"   Agent ID: {transfer_result['agentId']}")
-    except Exception as e:
-        print(f"❌ Transfer failed: {e}")
-        return
-    
-    print("\n" + "=" * 60)
-    print("STEP 3: VERIFY OWNERSHIP CHANGE")
-    print("=" * 60)
-    
-    # Verify ownership changed using utility function
-    print("🔍 Verifying ownership change...")
-    new_owner = agentSdk.getAgentOwner(registration_result.agentId)
-    print(f"✅ New owner: {new_owner}")
-    print(f"✅ Matches Owner B: {new_owner.lower() == ownerB_address.lower()}")
-    
-    # Verify agent metadata remains unchanged
-    print("\n🔍 Verifying agent metadata remains unchanged...")
-    # Parse agentId to extract tokenId for contract call
-    agent_id_str = str(registration_result.agentId)
-    if ":" in agent_id_str:
-        token_id = int(agent_id_str.split(":")[-1])
-    else:
-        token_id = int(agent_id_str)
-    
-    agent_uri = agentSdk.web3_client.call_contract(
-        agentSdk.identity_registry,
-        "tokenURI",
-        token_id
-    )
-    print(f"✅ Agent URI unchanged: {agent_uri}")
-    # Note: registration_result.agentURI may be None if not set in RegistrationFile
-    if registration_result.agentURI:
-        print(f"✅ Matches original: {agent_uri == registration_result.agentURI}")
-    else:
-        print(f"✅ Agent URI retrieved from blockchain: {agent_uri}")
-    
-    print("\n" + "=" * 60)
-    print("STEP 4: TEST SDK.transferAgent() METHOD")
-    print("=" * 60)
-    
-    # Test SDK-level transfer method (this will fail since we're not Owner B)
-    print("🔄 Testing SDK.transferAgent() method...")
-    print("   (This should fail since we're not the current owner)")
-    
-    try:
-        # Try to transfer back to Owner A (should fail)
-        sdk_transfer_result = agentSdk.transferAgent(
-            registration_result.agentId,
-            ownerA_address
-        )
-        print(f"❌ Unexpected success: {sdk_transfer_result}")
-    except Exception as e:
-        print(f"✅ Expected failure: {e}")
-    
-    print("\n" + "=" * 60)
-    print("STEP 5: TEST INVALID TRANSFER ATTEMPTS")
-    print("=" * 60)
-    
-    # Test invalid transfer attempts
-    print("🔄 Testing invalid transfer attempts...")
-    
-    # Test zero address
-    try:
-        agent.transfer("0x0000000000000000000000000000000000000000")
-        print("❌ Zero address transfer should have failed")
-    except ValueError as e:
-        print(f"✅ Zero address correctly rejected: {e}")
-    
-    # Test self-transfer
-    try:
-        agent.transfer(ownerB_address)  # Try to transfer to same address
-        print("❌ Self-transfer should have failed")
-    except ValueError as e:
-        print(f"✅ Self-transfer correctly rejected: {e}")
-    
-    # Test invalid address format
-    try:
-        agent.transfer("invalid_address")
-        print("❌ Invalid address should have failed")
-    except ValueError as e:
-        print(f"✅ Invalid address correctly rejected: {e}")
-    
-    print("\n" + "=" * 60)
-    print("STEP 6: VERIFY AGENT DATA INTEGRITY")
-    print("=" * 60)
-    
-    # Load agent and verify all data is intact
-    print("🔍 Loading agent and verifying data integrity...")
-    
-    try:
-        loaded_agent = agentSdk.loadAgent(registration_result.agentId)
-        print(f"✅ Agent loaded successfully")
-        print(f"   Name: {loaded_agent.name}")
-        print(f"   Description: {loaded_agent.description}")
-        print(f"   Image: {loaded_agent.image}")
-        print(f"   Wallet Address: {loaded_agent.walletAddress}")
-        print(f"   ENS: {loaded_agent.ensEndpoint}")
-        print(f"   Metadata: {loaded_agent.metadata}")
-        print(f"   MCP Endpoint: {loaded_agent.mcpEndpoint}")
-        print(f"   A2A Endpoint: {loaded_agent.a2aEndpoint}")
-        print(f"   Active: {loaded_agent.active}")
-        
-        # Verify ownership is correctly reflected
-        print(f"\n🔍 Ownership verification:")
-        print(f"   Current owner (on-chain): {new_owner}")
-        print(f"   Expected owner: {ownerB_address}")
-        print(f"   Match: {new_owner.lower() == ownerB_address.lower()}")
-        
-    except Exception as e:
-        print(f"❌ Failed to load agent: {e}")
-        import traceback
-        traceback.print_exc()
-    
-    print("\n" + "=" * 60)
-    print("🎉 TRANSFER TEST COMPLETED SUCCESSFULLY!")
-    print("=" * 60)
-    
-    print("\n📋 Summary:")
-    print(f"   ✅ Agent created and registered with Owner A")
-    print(f"   ✅ Agent successfully transferred to Owner B")
-    print(f"   ✅ Ownership change verified on-chain")
-    print(f"   ✅ Invalid transfer attempts properly rejected")
-    print(f"   ✅ Agent data integrity maintained")
-    print(f"   ✅ Both Agent.transfer() and SDK.transferAgent() methods tested")
-    
-    print(f"\n🔗 Agent Details:")
-    print(f"   Agent ID: {registration_result.agentId}")
-    print(f"   Current Owner: {new_owner}")
-    print(f"   Agent URI: {agent_uri}")
-    if transfer_result and 'txHash' in transfer_result:
-        print(f"   Transfer Transaction: {transfer_result['txHash']}")
-
-if __name__ == "__main__":
-    main()