agent0-sdk 1.4.2__py3-none-any.whl → 1.5.1b1__py3-none-any.whl

agent0_sdk/__init__.py

@@ -16,7 +16,9 @@ from .core.models import (
     RegistrationFile,
     AgentSummary,
     Feedback,
-    SearchParams,
+    SearchFilters,
+    SearchOptions,
+    FeedbackFilters,
     SearchFeedbackParams,
 )
 
@@ -33,7 +35,7 @@ except ImportError:
     TransactionMined = None
     _sdk_available = False
 
-__version__ = "1.4.2"
+__version__ = "1.5.1b1"
 __all__ = [
     "SDK",
     "Agent",
@@ -52,6 +54,8 @@ __all__ = [
     "RegistrationFile",
     "AgentSummary",
     "Feedback",
-    "SearchParams",
+    "SearchFilters",
+    "SearchOptions",
+    "FeedbackFilters",
     "SearchFeedbackParams",
 ]

agent0_sdk/core/contracts.py

@@ -547,6 +547,7 @@ DEFAULT_REGISTRIES: Dict[int, Dict[str, str]] = {
 DEFAULT_SUBGRAPH_URLS: Dict[int, str] = {
     1: "https://gateway.thegraph.com/api/7fd2e7d89ce3ef24cd0d4590298f0b2c/subgraphs/id/FV6RR6y13rsnCxBAicKuQEwDp8ioEGiNaWaZUmvr1F8k",  # Ethereum Mainnet
     11155111: "https://gateway.thegraph.com/api/00a452ad3cd1900273ea62c1bf283f93/subgraphs/id/6wQRC7geo9XYAhckfmfo8kbMRLeWU8KQd3XsJqFKmZLT",  # Ethereum Sepolia
+    137: "https://gateway.thegraph.com/api/782d61ed390e625b8867995389699b4c/subgraphs/id/9q16PZv1JudvtnCAf44cBoxg82yK9SSsFvrjCY9xnneF",  # Polygon Mainnet
     # Other chains temporarily disabled - subgraphs to be updated
     # 84532: "https://gateway.thegraph.com/api/...",  # Base Sepolia - To be updated
     # 80002: "https://gateway.thegraph.com/api/...",  # Polygon Amoy - To be updated

agent0_sdk/core/feedback_manager.py

@@ -422,8 +422,6 @@ class FeedbackManager:
         minValue: Optional[float] = None,
         maxValue: Optional[float] = None,
         include_revoked: bool = False,
-        first: int = 100,
-        skip: int = 0,
         agents: Optional[List[AgentId]] = None,
     ) -> List[Feedback]:
         """Search feedback.
@@ -451,8 +449,6 @@ class FeedbackManager:
                 minValue,
                 maxValue,
                 include_revoked,
-                first,
-                skip,
                 agents=agents,
             )
         
@@ -469,8 +465,6 @@ class FeedbackManager:
                 minValue,
                 maxValue,
                 include_revoked,
-                first,
-                skip,
                 agents=agents,
             )
         
@@ -558,8 +552,6 @@ class FeedbackManager:
         minValue: Optional[float],
         maxValue: Optional[float],
         include_revoked: bool,
-        first: int,
-        skip: int,
         agents: Optional[List[AgentId]] = None,
     ) -> List[Feedback]:
         """Search feedback using subgraph."""
@@ -583,79 +575,84 @@ class FeedbackManager:
             includeRevoked=include_revoked
         )
         
-        # Query subgraph
-        feedbacks_data = self.subgraph_client.search_feedback(
-            params=params,
-            first=first,
-            skip=skip,
-            order_by="createdAt",
-            order_direction="desc"
-        )
-        
-        # Map to Feedback objects
         feedbacks = []
-        for fb_data in feedbacks_data:
-            feedback_file = fb_data.get('feedbackFile') or {}
-            if not isinstance(feedback_file, dict):
-                feedback_file = {}
-            
-            # Map responses
-            responses_data = fb_data.get('responses', [])
-            answers = []
-            for resp in responses_data:
-                answers.append({
-                    'responder': resp.get('responder'),
-                    'responseUri': resp.get('responseUri'),
-                    'responseHash': resp.get('responseHash'),
-                    'createdAt': resp.get('createdAt')
-                })
-            
-            # Map tags: rely on whatever the subgraph returns (may be legacy bytes/hash-like values)
-            tags_list: List[str] = []
-            tag1 = fb_data.get('tag1') or feedback_file.get('tag1')
-            tag2 = fb_data.get('tag2') or feedback_file.get('tag2')
-            if isinstance(tag1, str) and tag1:
+        batch = 1000
+        skip = 0
+        while True:
+            feedbacks_data = self.subgraph_client.search_feedback(
+                params=params,
+                first=batch,
+                skip=skip,
+                order_by="createdAt",
+                order_direction="desc",
+            )
+
+            for fb_data in feedbacks_data:
+                feedback_file = fb_data.get('feedbackFile') or {}
+                if not isinstance(feedback_file, dict):
+                    feedback_file = {}
+
+                # Map responses
+                responses_data = fb_data.get('responses', [])
+                answers = []
+                for resp in responses_data:
+                    answers.append({
+                        'responder': resp.get('responder'),
+                        'responseUri': resp.get('responseUri'),
+                        'responseHash': resp.get('responseHash'),
+                        'createdAt': resp.get('createdAt')
+                    })
+
+                # Map tags: rely on whatever the subgraph returns (may be legacy bytes/hash-like values)
+                tags_list: List[str] = []
+                tag1 = fb_data.get('tag1') or feedback_file.get('tag1')
+                tag2 = fb_data.get('tag2') or feedback_file.get('tag2')
+                if isinstance(tag1, str) and tag1:
                     tags_list.append(tag1)
-            if isinstance(tag2, str) and tag2:
+                if isinstance(tag2, str) and tag2:
                     tags_list.append(tag2)
-            
-            # Parse agentId from feedback ID
-            feedback_id = fb_data['id']
-            parts = feedback_id.split(':')
-            if len(parts) >= 2:
-                agent_id_str = f"{parts[0]}:{parts[1]}"
-                client_addr = parts[2] if len(parts) > 2 else ""
-                feedback_idx = int(parts[3]) if len(parts) > 3 else 1
-            else:
-                agent_id_str = feedback_id
-                client_addr = ""
-                feedback_idx = 1
-            
-            feedback = Feedback(
-                id=Feedback.create_id(agent_id_str, client_addr, feedback_idx),
-                agentId=agent_id_str,
-                reviewer=client_addr,
-                value=float(fb_data.get("value")) if fb_data.get("value") is not None else None,
-                tags=tags_list,
-                text=feedback_file.get('text'),
-                capability=feedback_file.get('capability'),
-                context=feedback_file.get('context'),
-                proofOfPayment={
-                    'fromAddress': feedback_file.get('proofOfPaymentFromAddress'),
-                    'toAddress': feedback_file.get('proofOfPaymentToAddress'),
-                    'chainId': feedback_file.get('proofOfPaymentChainId'),
-                    'txHash': feedback_file.get('proofOfPaymentTxHash'),
-                } if feedback_file.get('proofOfPaymentFromAddress') else None,
-                fileURI=fb_data.get('feedbackURI') or fb_data.get('feedbackUri'),  # Handle both old and new field names
-                endpoint=fb_data.get('endpoint'),
-                createdAt=fb_data.get('createdAt', int(time.time())),
-                answers=answers,
-                isRevoked=fb_data.get('isRevoked', False),
-                name=feedback_file.get('name'),
-                skill=feedback_file.get('skill'),
-                task=feedback_file.get('task'),
-            )
-            feedbacks.append(feedback)
+
+                # Parse agentId from feedback ID
+                feedback_id = fb_data['id']
+                parts = feedback_id.split(':')
+                if len(parts) >= 2:
+                    agent_id_str = f"{parts[0]}:{parts[1]}"
+                    client_addr = parts[2] if len(parts) > 2 else ""
+                    feedback_idx = int(parts[3]) if len(parts) > 3 else 1
+                else:
+                    agent_id_str = feedback_id
+                    client_addr = ""
+                    feedback_idx = 1
+
+                feedback = Feedback(
+                    id=Feedback.create_id(agent_id_str, client_addr, feedback_idx),
+                    agentId=agent_id_str,
+                    reviewer=client_addr,
+                    value=float(fb_data.get("value")) if fb_data.get("value") is not None else None,
+                    tags=tags_list,
+                    text=feedback_file.get('text'),
+                    capability=feedback_file.get('capability'),
+                    context=feedback_file.get('context'),
+                    proofOfPayment={
+                        'fromAddress': feedback_file.get('proofOfPaymentFromAddress'),
+                        'toAddress': feedback_file.get('proofOfPaymentToAddress'),
+                        'chainId': feedback_file.get('proofOfPaymentChainId'),
+                        'txHash': feedback_file.get('proofOfPaymentTxHash'),
+                    } if feedback_file.get('proofOfPaymentFromAddress') else None,
+                    fileURI=fb_data.get('feedbackURI') or fb_data.get('feedbackUri'),  # Handle both old and new field names
+                    endpoint=fb_data.get('endpoint'),
+                    createdAt=fb_data.get('createdAt', int(time.time())),
+                    answers=answers,
+                    isRevoked=fb_data.get('isRevoked', False),
+                    name=feedback_file.get('name'),
+                    skill=feedback_file.get('skill'),
+                    task=feedback_file.get('task'),
+                )
+                feedbacks.append(feedback)
+
+            if len(feedbacks_data) < batch:
+                break
+            skip += batch
         
         return feedbacks