fix(core): agent registration endpoint mismatch (#466)

python/docs/api/uagents/agent.md

@@ -128,7 +128,7 @@ An agent that interacts within a communication environment.
 - `_resolver` _Resolver_ - The resolver for agent communication.
 - `_loop` _asyncio.AbstractEventLoop_ - The asyncio event loop used by the agent.
 - `_logger` - The logger instance for logging agent activities.
-- `_endpoints` _List[dict]_ - List of endpoints at which the agent is reachable.
+- `_endpoints` _List[AgentEndpoint]_ - List of endpoints at which the agent is reachable.
 - `_use_mailbox` _bool_ - Indicates if the agent uses a mailbox for communication.
 - `_agentverse` _dict_ - Agentverse configuration settings.
 - `_mailbox_client` _MailboxClient_ - The client for interacting with the agentverse mailbox.
@@ -473,14 +473,14 @@ Sign the registration data for Almanac contract.
 #### update`_`endpoints
 
 ```python
-def update_endpoints(endpoints: List[Dict[str, Any]])
+def update_endpoints(endpoints: List[AgentEndpoint])
 ```
 
 Update the list of endpoints.
 
 **Arguments**:
 
-- `endpoints` _List[Dict[str, Any]]_ - List of endpoint dictionaries.
+- `endpoints` _List[AgentEndpoint]_ - List of endpoint dictionaries.
 
 <a id="src.uagents.agent.Agent.update_loop"></a>
 

python/docs/api/uagents/config.md

@@ -9,7 +9,7 @@
 ```python
 def parse_endpoint_config(
     endpoint: Optional[Union[str, List[str], Dict[str, dict]]]
-) -> Optional[List[Dict[str, Any]]]
+) -> List[AgentEndpoint]
 ```
 
 Parse the user-provided endpoint configuration.

python/docs/api/uagents/envelope.md

@@ -21,7 +21,7 @@ Represents an envelope for message communication between agents.
 - `target` _str_ - The target's address.
 - `session` _UUID4_ - The session UUID that persists for back-and-forth
   dialogues between agents.
-- `schema_digest` _str_ - The schema digest for the enclosed message (alias for protocol).
+- `schema_digest` _str_ - The schema digest for the enclosed message.
 - `protocol_digest` _Optional[str]_ - The digest of the protocol associated with the message
   (optional).
 - `payload` _Optional[str]_ - The encoded message payload of the envelope (optional).

python/docs/api/uagents/network.md

@@ -161,7 +161,7 @@ Get the expiry height of an agent's registration.
 #### get`_`endpoints
 
 ```python
-def get_endpoints(address: str)
+def get_endpoints(address: str) -> List[AgentEndpoint]
 ```
 
 Get the endpoints associated with an agent's registration.
@@ -201,7 +201,7 @@ Get the protocols associated with an agent's registration.
 ```python
 async def register(ledger: LedgerClient, wallet: LocalWallet,
                    agent_address: str, protocols: List[str],
-                   endpoints: List[Dict[str, Any]], signature: str)
+                   endpoints: List[AgentEndpoint], signature: str)
 ```
 
 Register an agent with the Almanac contract.

python/src/uagents/agent.py

@@ -32,6 +32,7 @@
     REGISTRATION_RETRY_INTERVAL_SECONDS,
     REGISTRATION_UPDATE_INTERVAL_SECONDS,
     TESTNET_PREFIX,
+    AgentEndpoint,
     parse_agentverse_config,
     parse_endpoint_config,
 )
@@ -197,7 +198,7 @@ class Agent(Sink):
         _resolver (Resolver): The resolver for agent communication.
         _loop (asyncio.AbstractEventLoop): The asyncio event loop used by the agent.
         _logger: The logger instance for logging agent activities.
-        _endpoints (List[dict]): List of endpoints at which the agent is reachable.
+        _endpoints (List[AgentEndpoint]): List of endpoints at which the agent is reachable.
         _use_mailbox (bool): Indicates if the agent uses a mailbox for communication.
         _agentverse (dict): Agentverse configuration settings.
         _mailbox_client (MailboxClient): The client for interacting with the agentverse mailbox.
@@ -316,10 +317,10 @@ def __init__(
             self._mailbox_client = MailboxClient(self, self._logger)
             # if mailbox is provided, override endpoints with mailbox endpoint
             self._endpoints = [
-                {
-                    "url": f"{self.mailbox['http_prefix']}://{self.mailbox['base_url']}/v1/submit",
-                    "weight": 1,
-                }
+                AgentEndpoint(
+                    url=f"{self.mailbox['http_prefix']}://{self.mailbox['base_url']}/v1/submit",
+                    weight=1,
+                ),
             ]
         else:
             self._mailbox_client = None
@@ -613,12 +614,12 @@ def sign_registration(self) -> str:
             self._almanac_contract.get_sequence(self.address),
         )
 
-    def update_endpoints(self, endpoints: List[Dict[str, Any]]):
+    def update_endpoints(self, endpoints: List[AgentEndpoint]):
         """
         Update the list of endpoints.
 
         Args:
-            endpoints (List[Dict[str, Any]]): List of endpoint dictionaries.
+            endpoints (List[AgentEndpoint]): List of endpoint dictionaries.
 
         """
 
@@ -669,7 +670,7 @@ async def register(self):
                     "I do not have enough funds to register on Almanac contract"
                 )
                 if self._test:
-                    add_testnet_funds(str(self.wallet.address()))
+                    add_testnet_funds(self.wallet.address().data)
                     self._logger.info(
                         f"Adding testnet funds to {self.wallet.address()}"
                     )

python/src/uagents/asgi.py

@@ -158,7 +158,10 @@ async def serve(self):
         self._logger.info(
             f"Starting server on http://{HOST}:{self._port} (Press CTRL+C to quit)"
         )
-        await self._server.serve()
+        try:
+            await self._server.serve()
+        except KeyboardInterrupt:
+            self._logger.info("Shutting down server")
 
     async def __call__(self, scope, receive, send):  #  pylint: disable=too-many-branches
         """

python/src/uagents/network.py

@@ -190,7 +190,7 @@ def get_expiry(self, address: str) -> int:
 
         return (expiry - height) * AVERAGE_BLOCK_INTERVAL
 
-    def get_endpoints(self, address: str) -> Optional[List[AgentEndpoint]]:
+    def get_endpoints(self, address: str) -> List[AgentEndpoint]:
         """
         Get the endpoints associated with an agent's registration.
 
@@ -204,7 +204,7 @@ def get_endpoints(self, address: str) -> Optional[List[AgentEndpoint]]:
         response = self.query(query_msg)
 
         if not response["record"]:
-            return None
+            return []
 
         endpoints = []
         for endpoint in response.get("record")[0]["record"]["service"]["endpoints"]: