agent0-sdk 0.2.0__py3-none-any.whl

tests/test_sdk.py

@@ -0,0 +1,238 @@
+"""
+Tests for SDK class.
+"""
+
+import pytest
+from unittest.mock import Mock, patch
+
+from agent0_sdk.core.sdk import SDK
+from agent0_sdk.core.models import EndpointType, TrustModel
+
+
+class TestSDK:
+    """Test SDK class."""
+    
+    def test_sdk_initialization(self):
+        """Test SDK initialization."""
+        with patch('agent0_sdk.core.sdk.Web3Client') as mock_web3:
+            mock_web3.return_value.chain_id = 11155111
+            
+            sdk = SDK(
+                chain_id=11155111,
+                signer="0x1234567890abcdef",
+                rpc_url="https://eth-sepolia.g.alchemy.com/v2/test"
+            )
+            
+            assert sdk.chain_id == 11155111
+            assert sdk.rpc_url == "https://eth-sepolia.g.alchemy.com/v2/test"
+            mock_web3.assert_called_once()
+    
+    def test_create_agent(self):
+        """Test agent creation."""
+        with patch('agent0_sdk.core.sdk.Web3Client') as mock_web3:
+            mock_web3.return_value.chain_id = 11155111
+            
+            sdk = SDK(
+                chain_id=11155111,
+                signer="0x1234567890abcdef",
+                rpc_url="https://eth-sepolia.g.alchemy.com/v2/test"
+            )
+            
+            agent = sdk.createAgent(
+                name="Test Agent",
+                description="A test agent",
+                image="https://example.com/image.png"
+            )
+            
+            assert agent.name == "Test Agent"
+            assert agent.description == "A test agent"
+            assert agent.image == "https://example.com/image.png"
+            assert agent.sdk == sdk
+    
+    def test_registries(self):
+        """Test registry resolution."""
+        with patch('agent0_sdk.core.sdk.Web3Client') as mock_web3:
+            mock_web3.return_value.chain_id = 11155111
+            
+            sdk = SDK(
+                chain_id=11155111,
+                signer="0x1234567890abcdef",
+                rpc_url="https://eth-sepolia.g.alchemy.com/v2/test"
+            )
+            
+        registries = sdk.registries()
+        assert "IDENTITY" in registries
+        assert "REPUTATION" in registries
+        assert "VALIDATION" in registries
+    
+    def test_set_chain(self):
+        """Test chain switching."""
+        with patch('agent0_sdk.core.sdk.Web3Client') as mock_web3:
+            mock_web3.return_value.chain_id = 11155111
+            
+            sdk = SDK(
+                chain_id=11155111,
+                signer="0x1234567890abcdef",
+                rpc_url="https://eth-sepolia.g.alchemy.com/v2/test"
+            )
+            
+            sdk.set_chain(84532)  # Switch to Base Sepolia
+            assert sdk.chain_id == 84532
+            assert sdk._registries["IDENTITY"] is not None  # Should have Base Sepolia registry
+
+
+class TestAgent:
+    """Test Agent class."""
+    
+    def test_agent_endpoint_management(self):
+        """Test agent endpoint management."""
+        with patch('agent0_sdk.core.sdk.Web3Client') as mock_web3:
+            mock_web3.return_value.chain_id = 11155111
+            
+            sdk = SDK(
+                chain_id=11155111,
+                signer="0x1234567890abcdef",
+                rpc_url="https://eth-sepolia.g.alchemy.com/v2/test"
+            )
+            
+            agent = sdk.createAgent("Test Agent", "A test agent")
+            
+            # Add MCP endpoint
+            agent.setMCP("https://mcp.example.com/")
+            assert len(agent.endpoints) == 1
+            assert agent.endpoints[0].type == EndpointType.MCP
+            assert agent.endpoints[0].value == "https://mcp.example.com/"
+            assert agent.endpoints[0].meta["version"] == "2025-06-18"
+            
+            # Add A2A endpoint
+            agent.setA2A("https://a2a.example.com/", "1.0")
+            assert len(agent.endpoints) == 2
+            
+            # Add ENS endpoint
+            agent.setENS("test-agent.eth")
+            assert len(agent.endpoints) == 3
+            ens_endpoint = next(ep for ep in agent.endpoints if ep.type == EndpointType.ENS)
+            assert ens_endpoint.value == "test-agent.eth"
+            assert ens_endpoint.meta["version"] == "1.0"
+            
+            # Remove specific endpoint
+            agent.removeEndpoint(EndpointType.MCP, "https://mcp.example.com/")
+            assert len(agent.endpoints) == 2
+            assert agent.endpoints[0].type == EndpointType.A2A
+            
+            # Remove all endpoints
+            agent.removeEndpoints()
+            assert len(agent.endpoints) == 0
+    
+    def test_agent_trust_models(self):
+        """Test agent trust model management."""
+        with patch('agent0_sdk.core.sdk.Web3Client') as mock_web3:
+            mock_web3.return_value.chain_id = 11155111
+            
+            sdk = SDK(
+                chain_id=11155111,
+                signer="0x1234567890abcdef",
+                rpc_url="https://eth-sepolia.g.alchemy.com/v2/test"
+            )
+            
+            agent = sdk.createAgent("Test Agent", "A test agent")
+            
+            # Set trust models using new method
+            agent.setTrust(reputation=True, cryptoEconomic=True)
+            assert len(agent.trustModels) == 2
+            assert TrustModel.REPUTATION in agent.trustModels
+            assert TrustModel.CRYPTO_ECONOMIC in agent.trustModels
+            
+            # Set trust models using old method
+            agent.trustModels([TrustModel.REPUTATION, "custom_trust"])
+            assert len(agent.trustModels) == 2
+            assert TrustModel.REPUTATION in agent.trustModels
+            assert "custom_trust" in agent.trustModels
+    
+    def test_agent_metadata_management(self):
+        """Test agent metadata management."""
+        with patch('agent0_sdk.core.sdk.Web3Client') as mock_web3:
+            mock_web3.return_value.chain_id = 11155111
+            
+            sdk = SDK(
+                chain_id=11155111,
+                signer="0x1234567890abcdef",
+                rpc_url="https://eth-sepolia.g.alchemy.com/v2/test"
+            )
+            
+            agent = sdk.createAgent("Test Agent", "A test agent")
+            
+            # Set metadata
+            agent.setMetadata({"key1": "value1", "key2": "value2"})
+            assert agent.getMetadata() == {"key1": "value1", "key2": "value2"}
+            
+            # Set single key using setMetadata
+            agent.setMetadata({"key3": "value3"})
+            assert agent.getMetadata() == {"key1": "value1", "key2": "value2", "key3": "value3"}
+            
+            # Delete key
+            agent.delMetadata("key2")
+            assert agent.getMetadata() == {"key1": "value1", "key3": "value3"}
+    
+    def test_agent_info_update(self):
+        """Test agent info updates."""
+        with patch('agent0_sdk.core.sdk.Web3Client') as mock_web3:
+            mock_web3.return_value.chain_id = 11155111
+            
+            sdk = SDK(
+                chain_id=11155111,
+                signer="0x1234567890abcdef",
+                rpc_url="https://eth-sepolia.g.alchemy.com/v2/test"
+            )
+            
+            agent = sdk.createAgent("Test Agent", "A test agent")
+            
+            # Update info
+            agent.updateInfo(
+                name="Updated Agent",
+                description="An updated agent",
+                image="https://example.com/new-image.png"
+            )
+            
+            assert agent.name == "Updated Agent"
+            assert agent.description == "An updated agent"
+            assert agent.image == "https://example.com/new-image.png"
+            
+            # Set wallet address
+            agent.setAgentWallet("0x1234567890abcdef")
+            assert agent.walletAddress == "0x1234567890abcdef"
+    
+    def test_agent_json_serialization(self):
+        """Test agent JSON serialization."""
+        with patch('agent0_sdk.core.sdk.Web3Client') as mock_web3:
+            mock_web3.return_value.chain_id = 11155111
+            
+            sdk = SDK(
+                chain_id=11155111,
+                signer="0x1234567890abcdef",
+                rpc_url="https://eth-sepolia.g.alchemy.com/v2/test"
+            )
+            
+            agent = sdk.createAgent("Test Agent", "A test agent")
+            agent.setMCP("https://mcp.example.com/")
+            agent.trustModels([TrustModel.REPUTATION])
+            
+            json_data = agent.to_json()
+            assert isinstance(json_data, str)
+            assert "Test Agent" in json_data
+            assert "MCP" in json_data
+            assert "reputation" in json_data
+            
+            # Test x402 support
+            agent.setX402Support(True)
+            assert agent.x402support is True
+            
+            json_data_with_x402 = agent.to_json()
+            assert "x402support" in json_data_with_x402
+            
+            # Test active status
+            agent.setActive(True)
+            assert agent.active is True
+            
+            agent.setActive(False)
+            assert agent.active is False

tests/test_search.py

@@ -0,0 +1,271 @@
+"""
+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
+"""
+
+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("\n" + "=" * 60)
+    print("✅ Search Tests Completed!")
+    print("=" * 60)
+
+
+if __name__ == "__main__":
+    main()