feat: reregister if endpoint chenages

fetchai · Apr 12, 2023 · 1d7bad4 · 1d7bad4
1 parent 828c52f
commit 1d7bad4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/uagents/agent.py b/src/uagents/agent.py
@@ -1,5 +1,6 @@
 import asyncio
 import functools
+from re import S
 from typing import Dict, List, Optional, Set, Union, Type, Tuple
 
 from cosmpy.aerial.wallet import LocalWallet, PrivateKey
@@ -375,6 +376,7 @@ def start_background_tasks(self):
             if (
                 not self._almanac_contract.is_registered(self.address)
                 or self._schedule_registration() < MIN_REGISTRATION_TIME
+                or self._endpoints != self._almanac_contract.get_endpoints(self.address)
             ):
                 self._loop.create_task(
                     _run_interval(

diff --git a/src/uagents/network.py b/src/uagents/network.py
@@ -45,6 +45,14 @@ def get_expiry(self, address: str):
 
         return (expiry - height) * BLOCK_INTERVAL
 
+    def get_endpoints(self, address: str):
+        query_msg = {"query_records": {"agent_address": address}}
+        response = self.query(query_msg)
+
+        if not response["record"]:
+            return 0
+        return response.get("record")[0]["record"]["service"]["endpoints"]
+
     def get_registration_msg(
         self,
         protocols: Dict,