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

agent0_sdk/__init__.py

@@ -30,7 +30,7 @@ except ImportError:
     Agent = None
     _sdk_available = False
 
-__version__ = "0.2.2"
+__version__ = "0.3rc1"
 __all__ = [
     "SDK",
     "Agent",

agent0_sdk/core/contracts.py

@@ -477,6 +477,11 @@ DEFAULT_REGISTRIES: Dict[int, Dict[str, str]] = {
         "REPUTATION": "0x8004bd8daB57f14Ed299135749a5CB5c42d341BF",
         "VALIDATION": "0x8004C269D0A5647E51E121FeB226200ECE932d55",
     },
+    80002: {  # Polygon Amoy
+        "IDENTITY": "0x8004ad19E14B9e0654f73353e8a0B600D46C2898",
+        "REPUTATION": "0x8004B12F4C2B42d00c46479e859C92e39044C930",
+        "VALIDATION": "0x8004C11C213ff7BaD36489bcBDF947ba5eee289B",
+    },
     59141: {  # Linea Sepolia
         "IDENTITY": "0x8004aa7C931bCE1233973a0C6A667f73F66282e7",
         "REPUTATION": "0x8004bd8483b99310df121c46ED8858616b2Bba02",
@@ -487,4 +492,6 @@ DEFAULT_REGISTRIES: Dict[int, Dict[str, str]] = {
 # Default subgraph URLs for different chains
 DEFAULT_SUBGRAPH_URLS: Dict[int, str] = {
     11155111: "https://gateway.thegraph.com/api/00a452ad3cd1900273ea62c1bf283f93/subgraphs/id/6wQRC7geo9XYAhckfmfo8kbMRLeWU8KQd3XsJqFKmZLT",  # Ethereum Sepolia
+    84532: "https://gateway.thegraph.com/api/00a452ad3cd1900273ea62c1bf283f93/subgraphs/id/GjQEDgEKqoh5Yc8MUgxoQoRATEJdEiH7HbocfR1aFiHa",  # Base Sepolia
+    80002: "https://gateway.thegraph.com/api/00a452ad3cd1900273ea62c1bf283f93/subgraphs/id/2A1JB18r1mF2VNP4QBH4mmxd74kbHoM6xLXC8ABAKf7j",  # Polygon Amoy
 }

agent0_sdk/core/feedback_manager.py

@@ -703,7 +703,46 @@ class FeedbackManager:
         groupBy: Optional[List[str]] = None,
     ) -> Dict[str, Any]:
         """Get reputation summary for an agent with optional grouping."""
-        # Parse agent ID
+        # Parse chainId from agentId
+        chain_id = None
+        if ":" in agentId:
+            try:
+                chain_id = int(agentId.split(":", 1)[0])
+            except ValueError:
+                chain_id = None
+        
+        # Try subgraph first (if available and indexer supports it)
+        if self.indexer and self.subgraph_client:
+            # Get correct subgraph client for the chain
+            subgraph_client = None
+            full_agent_id = agentId
+            
+            if chain_id is not None:
+                subgraph_client = self.indexer._get_subgraph_client_for_chain(chain_id)
+            else:
+                # No chainId in agentId, use SDK's default
+                # Construct full agentId format for subgraph query
+                default_chain_id = self.web3_client.chain_id
+                token_id = agentId.split(":")[-1] if ":" in agentId else agentId
+                full_agent_id = f"{default_chain_id}:{token_id}"
+                subgraph_client = self.subgraph_client
+            
+            if subgraph_client:
+                # Use subgraph to calculate reputation
+                return self._get_reputation_summary_from_subgraph(
+                    full_agent_id, clientAddresses, tag1, tag2, groupBy
+                )
+        
+        # Fallback to blockchain (requires chain-specific web3 client)
+        # For now, only works if chain matches SDK's default
+        if chain_id is not None and chain_id != self.web3_client.chain_id:
+            raise ValueError(
+                f"Blockchain reputation summary not supported for chain {chain_id}. "
+                f"SDK is configured for chain {self.web3_client.chain_id}. "
+                f"Use subgraph-based summary instead."
+            )
+        
+        # Parse agent ID for blockchain call
         if ":" in agentId:
             tokenId = int(agentId.split(":")[-1])
         else:
@@ -765,6 +804,67 @@ class FeedbackManager:
         except Exception as e:
             raise ValueError(f"Failed to get reputation summary: {e}")
     
+    def _get_reputation_summary_from_subgraph(
+        self,
+        agentId: AgentId,
+        clientAddresses: Optional[List[Address]] = None,
+        tag1: Optional[str] = None,
+        tag2: Optional[str] = None,
+        groupBy: Optional[List[str]] = None,
+    ) -> Dict[str, Any]:
+        """Get reputation summary from subgraph."""
+        # Build tags list
+        tags = []
+        if tag1:
+            tags.append(tag1)
+        if tag2:
+            tags.append(tag2)
+        
+        # Get all feedback for the agent using indexer (which handles multi-chain)
+        # Use searchFeedback with a large limit to get all feedback
+        all_feedback = self.searchFeedback(
+            agentId=agentId,
+            clientAddresses=clientAddresses,
+            tags=tags if tags else None,
+            include_revoked=False,
+            first=1000,  # Large limit to get all feedback
+            skip=0
+        )
+        
+        # Calculate summary statistics
+        count = len(all_feedback)
+        scores = [fb.score for fb in all_feedback if fb.score is not None]
+        average_score = sum(scores) / len(scores) if scores else 0.0
+        
+        # If no grouping requested, return simple summary
+        if not groupBy:
+            return {
+                "agentId": agentId,
+                "count": count,
+                "averageScore": average_score,
+                "filters": {
+                    "clientAddresses": clientAddresses,
+                    "tag1": tag1,
+                    "tag2": tag2
+                }
+            }
+        
+        # Group feedback by requested dimensions
+        grouped_data = self._groupFeedback(all_feedback, groupBy)
+        
+        return {
+            "agentId": agentId,
+            "totalCount": count,
+            "totalAverageScore": average_score,
+            "groupedData": grouped_data,
+            "filters": {
+                "clientAddresses": clientAddresses,
+                "tag1": tag1,
+                "tag2": tag2
+            },
+            "groupBy": groupBy
+        }
+    
     def _groupFeedback(self, feedbackList: List[Feedback], groupBy: List[str]) -> Dict[str, Any]:
         """Group feedback by specified dimensions."""
         grouped = {}