feat: handle exceptions on context send

fetchai · Aug 30, 2023 · 4ac5c1a · 4ac5c1a
1 parent 266a4f7
commit 4ac5c1a
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/python/src/uagents/context.py b/python/src/uagents/context.py
@@ -242,11 +242,18 @@ async def send_raw(
         env.encode_payload(json_message)
         env.sign(self._identity)
 
-        async with aiohttp.ClientSession() as session:
-            async with session.post(
-                endpoint, headers={"content-type": "application/json"}, data=env.json()
-            ) as resp:
-                success = resp.status == 200
+        try:
+            async with aiohttp.ClientSession() as session:
+                async with session.post(
+                    endpoint,
+                    headers={"content-type": "application/json"},
+                    data=env.json(),
+                ) as resp:
+                    success = resp.status == 200
+        except aiohttp.ClientConnectorError as ex:
+            self._logger.exception(f"Failed to connect to {endpoint}: {ex}")
+        except Exception as ex:
+            self._logger.exception(f"Failed to send message to {destination}: {ex}")
 
         if not success:
             self._logger.exception(