Merge pull request #155 from Virtual-Protocol/feat/yang-add-back-cluster-to-acpagent

feat: add back cluster to iacp agent

Author: celesteanglm

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "virtuals-acp"
-version = "0.3.21"
+version = "0.3.22"
 description = "Agent Commerce Protocol Python SDK by Virtuals"
 authors = ["Steven Lee Soon Fatt <steven@virtuals.io>"]
 readme = "README.md"

virtuals_acp/client.py

@@ -476,6 +476,7 @@ def _hydrate_agent(self, agent_data: Dict[str, Any]) -> IACPAgent:
             wallet_address=Web3.to_checksum_address(agent_data["walletAddress"]),
             job_offerings=job_offerings,
             resources=resources,
+            cluster=agent_data.get("cluster"),
             twitter_handle=agent_data.get("twitterHandle"),
             metrics=agent_data.get("metrics"),
             contract_address=contract_address,

virtuals_acp/models.py

@@ -110,6 +110,7 @@ class IACPAgent:
     job_offerings: List["ACPJobOffering"] = field(default_factory=list)
 
     resources: List["ACPResourceOffering"] = field(default_factory=list)
+    cluster: Optional[str] = None
     twitter_handle: Optional[str] = None
     metrics: Optional[Dict[str, Any]] = None