agent0-sdk 0.2.2__py3-none-any.whl → 0.3rc1__py3-none-any.whl

agent0_sdk/core/models.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import json
 from dataclasses import dataclass, field
 from enum import Enum
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional, Union, Literal
 from datetime import datetime
 
 
@@ -269,7 +269,7 @@ class Feedback:
 @dataclass
 class SearchParams:
     """Parameters for agent search."""
-    chains: Optional[List[ChainId]] = None
+    chains: Optional[Union[List[ChainId], Literal["all"]]] = None
     name: Optional[str] = None  # case-insensitive substring
     description: Optional[str] = None  # semantic; vector distance < threshold
     owners: Optional[List[Address]] = None
@@ -286,6 +286,7 @@ class SearchParams:
     mcpResources: Optional[List[str]] = None
     active: Optional[bool] = True
     x402support: Optional[bool] = None
+    deduplicate_cross_chain: bool = False  # Deduplicate same agent across chains
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dictionary, filtering out None values."""

agent0_sdk/core/sdk.py

@@ -6,10 +6,13 @@ from __future__ import annotations
 
 import asyncio
 import json
+import logging
 import time
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional, Union, Literal
 from datetime import datetime
 
+logger = logging.getLogger(__name__)
+
 from .models import (
     AgentId, ChainId, Address, URI, Timestamp, IdemKey,
     EndpointType, TrustModel, Endpoint, RegistrationFile,
@@ -102,7 +105,8 @@ class SDK:
             web3_client=self.web3_client,
             store=indexingStore,
             embeddings=embeddings,
-            subgraph_client=self.subgraph_client
+            subgraph_client=self.subgraph_client,
+            subgraph_url_overrides=self._subgraph_urls
         )
         
         # Initialize IPFS client based on configuration
@@ -438,7 +442,7 @@ class SDK:
     def searchAgents(
         self,
         params: Union[SearchParams, Dict[str, Any], None] = None,
-        sort: List[str] = None,
+        sort: Union[str, List[str], None] = None,
         page_size: int = 50,
         cursor: Optional[str] = None,
         **kwargs  # Accept search criteria as kwargs for better DX
@@ -466,7 +470,9 @@ class SDK:
         
         if sort is None:
             sort = ["updatedAt:desc"]
-        
+        elif isinstance(sort, str):
+            sort = [sort]
+
         return self.indexer.search_agents(params, sort, page_size, cursor)
 
     # Feedback methods
@@ -588,8 +594,26 @@ class SDK:
         page_size: int = 50,
         cursor: Optional[str] = None,
         sort: Optional[List[str]] = None,
+        chains: Optional[Union[List[ChainId], Literal["all"]]] = None,
     ) -> Dict[str, Any]:
         """Search agents filtered by reputation criteria."""
+        # Handle multi-chain search
+        if chains:
+            # Expand "all" if needed
+            if chains == "all":
+                chains = self.indexer._get_all_configured_chains()
+            
+            # If multiple chains or single chain different from default
+            if isinstance(chains, list) and len(chains) > 0:
+                if len(chains) > 1 or (len(chains) == 1 and chains[0] != self.chainId):
+                    return asyncio.run(
+                        self._search_agents_by_reputation_across_chains(
+                            agents, tags, reviewers, capabilities, skills, tasks, names,
+                            minAverageScore, includeRevoked, page_size, cursor, sort, chains
+                        )
+                    )
+        
+        # Single chain search (existing behavior)
         if not self.subgraph_client:
             raise ValueError("Subgraph client required for searchAgentsByReputation")
         
@@ -664,6 +688,185 @@ class SDK:
         except Exception as e:
             raise ValueError(f"Failed to search agents by reputation: {e}")
     
+    async def _search_agents_by_reputation_across_chains(
+        self,
+        agents: Optional[List[AgentId]],
+        tags: Optional[List[str]],
+        reviewers: Optional[List[Address]],
+        capabilities: Optional[List[str]],
+        skills: Optional[List[str]],
+        tasks: Optional[List[str]],
+        names: Optional[List[str]],
+        minAverageScore: Optional[int],
+        includeRevoked: bool,
+        page_size: int,
+        cursor: Optional[str],
+        sort: Optional[List[str]],
+        chains: List[ChainId],
+    ) -> Dict[str, Any]:
+        """
+        Search agents by reputation across multiple chains in parallel.
+        
+        Similar to indexer._search_agents_across_chains() but for reputation-based search.
+        """
+        import time
+        start_time = time.time()
+        
+        if sort is None:
+            sort = ["createdAt:desc"]
+        
+        order_by = "createdAt"
+        order_direction = "desc"
+        if sort and len(sort) > 0:
+            sort_field = sort[0].split(":")
+            order_by = sort_field[0] if len(sort_field) >= 1 else order_by
+            order_direction = sort_field[1] if len(sort_field) >= 2 else order_direction
+        
+        skip = 0
+        if cursor:
+            try:
+                skip = int(cursor)
+            except ValueError:
+                skip = 0
+        
+        # Define async function for querying a single chain
+        async def query_single_chain(chain_id: int) -> Dict[str, Any]:
+            """Query one chain and return its results with metadata."""
+            try:
+                # Get subgraph client for this chain
+                subgraph_client = self.indexer._get_subgraph_client_for_chain(chain_id)
+                
+                if subgraph_client is None:
+                    logger.warning(f"No subgraph client available for chain {chain_id}")
+                    return {
+                        "chainId": chain_id,
+                        "status": "unavailable",
+                        "agents": [],
+                        "error": f"No subgraph configured for chain {chain_id}"
+                    }
+                
+                # Execute reputation search query
+                try:
+                    agents_data = subgraph_client.search_agents_by_reputation(
+                        agents=agents,
+                        tags=tags,
+                        reviewers=reviewers,
+                        capabilities=capabilities,
+                        skills=skills,
+                        tasks=tasks,
+                        names=names,
+                        minAverageScore=minAverageScore,
+                        includeRevoked=includeRevoked,
+                        first=page_size * 3,  # Fetch extra to allow for filtering/sorting
+                        skip=0,  # We'll handle pagination after aggregation
+                        order_by=order_by,
+                        order_direction=order_direction
+                    )
+                    
+                    logger.info(f"Chain {chain_id}: fetched {len(agents_data)} agents by reputation")
+                except Exception as e:
+                    logger.error(f"Error in search_agents_by_reputation for chain {chain_id}: {e}", exc_info=True)
+                    agents_data = []
+                
+                return {
+                    "chainId": chain_id,
+                    "status": "success",
+                    "agents": agents_data,
+                    "count": len(agents_data),
+                }
+                
+            except Exception as e:
+                logger.error(f"Error querying chain {chain_id} for reputation search: {e}", exc_info=True)
+                return {
+                    "chainId": chain_id,
+                    "status": "error",
+                    "agents": [],
+                    "error": str(e),
+                    "count": 0
+                }
+        
+        # Execute queries in parallel
+        chain_tasks = [query_single_chain(chain_id) for chain_id in chains]
+        chain_results = await asyncio.gather(*chain_tasks)
+        
+        # Aggregate results from all chains
+        all_agents = []
+        successful_chains = []
+        failed_chains = []
+        
+        for result in chain_results:
+            chain_id = result["chainId"]
+            if result["status"] == "success":
+                successful_chains.append(chain_id)
+                agents_count = len(result.get("agents", []))
+                logger.debug(f"Chain {chain_id}: aggregating {agents_count} agents")
+                all_agents.extend(result["agents"])
+            else:
+                failed_chains.append(chain_id)
+                logger.warning(f"Chain {chain_id}: status={result.get('status')}, error={result.get('error', 'N/A')}")
+        
+        logger.debug(f"Total agents aggregated: {len(all_agents)} from {len(successful_chains)} chains")
+        
+        # Transform to AgentSummary objects
+        from .models import AgentSummary
+        results = []
+        for agent_data in all_agents:
+            reg_file = agent_data.get('registrationFile') or {}
+            if not isinstance(reg_file, dict):
+                reg_file = {}
+            
+            agent_summary = AgentSummary(
+                chainId=int(agent_data.get('chainId', 0)),
+                agentId=agent_data.get('id'),
+                name=reg_file.get('name', f"Agent {agent_data.get('id')}"),
+                image=reg_file.get('image'),
+                description=reg_file.get('description', ''),
+                owners=[agent_data.get('owner', '')],
+                operators=agent_data.get('operators', []),
+                mcp=reg_file.get('mcpEndpoint') is not None,
+                a2a=reg_file.get('a2aEndpoint') is not None,
+                ens=reg_file.get('ens'),
+                did=reg_file.get('did'),
+                walletAddress=reg_file.get('agentWallet'),
+                supportedTrusts=reg_file.get('supportedTrusts', []),
+                a2aSkills=reg_file.get('a2aSkills', []),
+                mcpTools=reg_file.get('mcpTools', []),
+                mcpPrompts=reg_file.get('mcpPrompts', []),
+                mcpResources=reg_file.get('mcpResources', []),
+                active=reg_file.get('active', True),
+                x402support=reg_file.get('x402support', False),
+                extras={'averageScore': agent_data.get('averageScore')}
+            )
+            results.append(agent_summary)
+        
+        # Sort by averageScore (descending) if available, otherwise by createdAt
+        results.sort(
+            key=lambda x: (
+                x.extras.get('averageScore') if x.extras.get('averageScore') is not None else 0,
+                x.chainId,
+                x.agentId
+            ),
+            reverse=True
+        )
+        
+        # Apply pagination
+        paginated_results = results[skip:skip + page_size]
+        next_cursor = str(skip + len(paginated_results)) if len(paginated_results) == page_size and skip + len(paginated_results) < len(results) else None
+        
+        elapsed_ms = int((time.time() - start_time) * 1000)
+        
+        return {
+            "items": paginated_results,
+            "nextCursor": next_cursor,
+            "meta": {
+                "chains": chains,
+                "successfulChains": successful_chains,
+                "failedChains": failed_chains,
+                "totalResults": len(results),
+                "timing": {"totalMs": elapsed_ms}
+            }
+        }
+    
     # Feedback methods - delegate to feedback_manager
     def signFeedbackAuth(
         self,

agent0_sdk/core/subgraph_client.py

@@ -38,7 +38,7 @@ class SubgraphClient:
                     'variables': variables or {}
                 },
                 headers={'Content-Type': 'application/json'},
-                timeout=30
+                timeout=10
             )
             response.raise_for_status()
             result = response.json()
@@ -716,12 +716,14 @@ class SubgraphClient:
                 return []
         else:
             # No feedback filters - query agents directly
-            agent_filters = []
+            # For reputation search, we want agents that have feedback
+            # Filter by totalFeedback > 0 to only get agents with feedback
+            agent_filters = ['totalFeedback_gt: 0']  # Only agents with feedback (BigInt comparison)
             if agents is not None and len(agents) > 0:
                 agent_ids = [f'"{aid}"' for aid in agents]
                 agent_filters.append(f'id_in: [{", ".join(agent_ids)}]')
             
-            agent_where = f"where: {{ {', '.join(agent_filters)} }}" if agent_filters else ""
+            agent_where = f"where: {{ {', '.join(agent_filters)} }}"
         
         # Build feedback where for agent query (to calculate scores)
         feedback_where_for_agents = f"{{ {', '.join(feedback_filters)} }}" if feedback_filters else "{}"
@@ -784,6 +786,12 @@ class SubgraphClient:
         
         try:
             result = self.query(query)
+            
+            # Check for GraphQL errors
+            if 'errors' in result:
+                logger.error(f"GraphQL errors in search_agents_by_reputation: {result['errors']}")
+                return []
+            
             agents_result = result.get('agents', [])
             
             # Calculate average scores
@@ -806,6 +814,18 @@ class SubgraphClient:
                     if agent.get('averageScore') is not None and agent['averageScore'] >= minAverageScore
                 ]
             
+            # For reputation search, filter logic:
+            # - If specific agents were requested, return them even if averageScore is None
+            #   (the user explicitly asked for these agents, so return them)
+            # - If general search (no specific agents), only return agents with reputation data
+            if agents is None or len(agents) == 0:
+                # General search - only return agents with reputation
+                agents_result = [
+                    agent for agent in agents_result
+                    if agent.get('averageScore') is not None
+                ]
+            # else: specific agents requested - return all requested agents (even if averageScore is None)
+            
             return agents_result
             
         except Exception as e:

{agent0_sdk-0.2.2.dist-info → agent0_sdk-0.3rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent0-sdk
-Version: 0.2.2
+Version: 0.3rc1
 Summary: Python SDK for agent portability, discovery and trust based on ERC-8004
 Author-email: Marco De Rossi <marco.derossi@consensys.net>
 License: MIT License
@@ -274,7 +274,6 @@ agent.register("https://example.com/agent-registration.json")
 
 - More chains (currently Ethereum Sepolia only)
 - Support for validations
-- Multi-chain agents search
 - Enhanced x402 payments
 - Semantic/Vectorial search
 - Advanced reputation aggregation

{agent0_sdk-0.2.2.dist-info → agent0_sdk-0.3rc1.dist-info}/RECORD

@@ -1,27 +1,29 @@
-agent0_sdk/__init__.py,sha256=SOJFcN81--KPhBSpHV0T85Q7Z1qZxxjLedZDCSbfkm0,919
+agent0_sdk/__init__.py,sha256=XrY2VSPzM6axhVSb1yTY-oYhJBbTTNJP5mKSFeMRPvk,920
 agent0_sdk/core/agent.py,sha256=RQND8F3Hmmpnt-PqCXBP7T8KfrjTXo0_X2nZoMJh01w,33343
-agent0_sdk/core/contracts.py,sha256=mER1pSae4-fiGBwaVTi7oqJ_QYaDnrtyIF7g6dkdE-0,19889
+agent0_sdk/core/contracts.py,sha256=LJY07PIyAb3ZSH_m2xD4OdXawsXKxaTUGhYHwGWUpmI,20435
 agent0_sdk/core/endpoint_crawler.py,sha256=QBkFc3tBSQqHj6PtSTZ5D3_HVB00KJZJdxE3uYpI9po,13611
-agent0_sdk/core/feedback_manager.py,sha256=wkV-P5wgDwOLwhaf22qOl-3S2F_PNI7oCHBBZMAxugI,36017
-agent0_sdk/core/indexer.py,sha256=bl5TqTSytt_qQyGqUsqZvrnXFlmQEuOgm6uKEE-u-BQ,41161
+agent0_sdk/core/feedback_manager.py,sha256=MV3srP-rgFaDXXELeLTJBj2aDUPdENy0tzWUsJUqrL4,39860
+agent0_sdk/core/indexer.py,sha256=erkDKrtyvAtfrblz0YHekPB2Hpw7tJm-1KJjafTMaHc,68470
 agent0_sdk/core/ipfs_client.py,sha256=fml1ai1BdBkgb95xAkf-ft8QsahV1HL30hBYRz7rQwI,13929
-agent0_sdk/core/models.py,sha256=bQARVmKETiVL5uJpHxXGN2wY2GFrdzxhPHqjzL8TBlQ,12373
-agent0_sdk/core/sdk.py,sha256=stOurnsSoedUFKaHKhrjaM1sghBzjighizV6AaZD7-U,31560
-agent0_sdk/core/subgraph_client.py,sha256=XXFVFAoHcgEfqxc2w2OksSp8vtbKMtsNIuNbNcNQOzE,27280
+agent0_sdk/core/models.py,sha256=5uXZmAtkLa5wKfnwU0-yuGkpmz00H_83Ta8zq2K7B-8,12487
+agent0_sdk/core/sdk.py,sha256=CQUpKaP7g1Lu7aokRYs-QChv78XIdkfz6Yax6nhaWqQ,40166
+agent0_sdk/core/subgraph_client.py,sha256=Iw-YEtT1-Rm9f-kY4EXZlCuwEkJQAq5-bHDpxrMVyGg,28419
 agent0_sdk/core/web3_client.py,sha256=859ntu5dAmNlcJ3YM1w_VV2gI3mpCC9QEr-GN1236zU,6850
-agent0_sdk-0.2.2.dist-info/licenses/LICENSE,sha256=rhZZbZm_Ovz4Oa9LNQ-ms8a1tA36wWh90ZkC0OR7WMw,1072
+agent0_sdk-0.3rc1.dist-info/licenses/LICENSE,sha256=rhZZbZm_Ovz4Oa9LNQ-ms8a1tA36wWh90ZkC0OR7WMw,1072
 tests/__init__.py,sha256=60ffheccPhuMCtwiiKP1X-CJJXKpxJ_Ywa0aXGHR9bY,23
 tests/config.py,sha256=1uePvkLBNubOQsvYkQSno0m007PMD1VACgm33fCYY6s,1429
 tests/conftest.py,sha256=P-HCtVVYwSvscuaJqhrgZcv39XXNnr932ekEamzIqis,589
+tests/discover_test_data.py,sha256=Fu0uQKnFk8m7qEqEp293BWbo_mT1CK5szVZcdcdlJQw,17678
 tests/test_feedback.py,sha256=7lszWYSmseJE0I4BhKzZdBiIzf2bgpPqZTZvhRrCTjY,14638
 tests/test_models.py,sha256=kCZdoPasIIcOjEw7ToPldqARdbGVK8v8byOhFwVo7OI,7115
+tests/test_multi_chain.py,sha256=6C-HD037Lsjd7_H0pwp7NDU04jLTjhardT1LHBjahFM,26711
 tests/test_real_public_servers.py,sha256=pCo4aLSCG9qv4D6T7jbyVmP1gt3r1jWxdes6z5XSNhU,3433
 tests/test_registration.py,sha256=pYanDPLAFETIfabBUvO34ZDmyD0Rbcv8vecSfgSrQ70,9542
 tests/test_registrationIpfs.py,sha256=9O3IBiN2CVMKzB19bqb-jN-nhqsN22kQINMpe9THqiI,8400
 tests/test_sdk.py,sha256=dALLFm_A6aXx0ec-CNOLGQoadaSPZ08EEeCS6Tgnm0M,9362
 tests/test_search.py,sha256=SiUio8H-M6Za8OXD_h9tUZdln0ayzkPJ3doTrkHv-Fs,18382
 tests/test_transfer.py,sha256=zRBllpoMs6NhagAmaZWmD4ckbYjSvsSUerBK4oS-HlA,9258
-agent0_sdk-0.2.2.dist-info/METADATA,sha256=mw1WxxAWC29tQ-FhDD_c7NJWPwyEGSYJ_w6atYM8CtY,11404
-agent0_sdk-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agent0_sdk-0.2.2.dist-info/top_level.txt,sha256=rgGBfOJlLi1zInQ85jBL2MpDu_ZJNbPjIGz-3Vn5rZs,17
-agent0_sdk-0.2.2.dist-info/RECORD,,
+agent0_sdk-0.3rc1.dist-info/METADATA,sha256=AHT6_T6pc63NLD3g1YS01A_TELbzmxRT8tEVN5kSpFI,11377
+agent0_sdk-0.3rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agent0_sdk-0.3rc1.dist-info/top_level.txt,sha256=rgGBfOJlLi1zInQ85jBL2MpDu_ZJNbPjIGz-3Vn5rZs,17
+agent0_sdk-0.3rc1.dist-info/RECORD,,