add retry for vybe and update deps

Author: truemagic-coder

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sakit"
-version = "12.4.1"
+version = "12.4.2"
 description = "Solana Agent Kit"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"
@@ -17,8 +17,8 @@ packages = [{ include = "sakit" }]
 python = ">=3.12,<4.0"
 httpx = "^0.28.1"
 solana-agent = ">=30.0.0"
-boto3 = "^1.38.38"
-botocore = "^1.38.38"
+boto3 = "^1.38.39"
+botocore = "^1.38.39"
 nemo-agent = "5.0.3"
 fastmcp = "^2.8.1"
 solana = "^0.36.7"

pyproject.toml

@@ -1,3 +1,4 @@
+import asyncio
 import logging
 from typing import Dict, Any, List, Optional
 from solana_agent import AutoTool, ToolRegistry
@@ -107,20 +108,29 @@ async def execute(self, user_id: str) -> Dict[str, Any]:
             "accept": "application/json",
             "X-API-KEY": self.api_key,
         }
-        try:
-            async with httpx.AsyncClient(timeout=10.0) as client:
-                resp = await client.get(url, headers=headers, timeout=15)
-                resp.raise_for_status()
-                data = resp.json()
-                summary = summarize_alphavybe_balances(data)
-                return {
-                    "status": "success",
-                    "result": summary,
-                }
-        except Exception as e:
-            logging.exception(f"Privy balance check error: {e}")
-            return {"status": "error", "message": str(e)}
-
+        max_retries = 3
+        for attempt in range(max_retries):
+            try:
+                async with httpx.AsyncClient(timeout=10.0) as client:
+                    resp = await client.get(url, headers=headers, timeout=15)
+                    if resp.status_code >= 500:
+                        raise httpx.HTTPStatusError("Server error", request=resp.request, response=resp)
+                    resp.raise_for_status()
+                    data = resp.json()
+                    summary = summarize_alphavybe_balances(data)
+                    return {
+                        "status": "success",
+                        "result": summary,
+                    }
+            except httpx.HTTPStatusError as e:
+                if resp.status_code >= 500 and attempt < max_retries - 1:
+                    await asyncio.sleep(1)
+                    continue
+                logging.exception(f"Privy balance check error: {e}")
+                return {"status": "error", "message": str(e)}
+            except Exception as e:
+                logging.exception(f"Privy balance check error: {e}")
+                return {"status": "error", "message": str(e)}
 
 class PrivyBalanceCheckerPlugin:
     def __init__(self):

sakit/privy_balance.py

@@ -80,20 +80,30 @@ async def execute(self, wallet_address: str) -> Dict[str, Any]:
             "accept": "application/json",
             "X-API-KEY": self.api_key,
         }
-        try:
-            async with httpx.AsyncClient(timeout=10.0) as client:
-                resp = await client.get(url, headers=headers, timeout=15)
-                resp.raise_for_status()
-                data = resp.json()
-                summary = summarize_alphavybe_balances(data)
-                return {
-                    "status": "success",
-                    "result": summary,
-                }
-        except Exception as e:
-            logging.exception(f"AlphaVybe balance check error: {e}")
-            return {"status": "error", "message": str(e)}
-
+        max_retries = 3
+        for attempt in range(max_retries):
+            try:
+                async with httpx.AsyncClient(timeout=10.0) as client:
+                    resp = await client.get(url, headers=headers, timeout=15)
+                    if resp.status_code >= 500:
+                        raise httpx.HTTPStatusError("Server error", request=resp.request, response=resp)
+                    resp.raise_for_status()
+                    data = resp.json()
+                    summary = summarize_alphavybe_balances(data)
+                    return {
+                        "status": "success",
+                        "result": summary,
+                    }
+            except httpx.HTTPStatusError as e:
+                if resp.status_code >= 500 and attempt < max_retries - 1:
+                    import asyncio
+                    await asyncio.sleep(1)
+                    continue
+                logging.exception(f"AlphaVybe balance check error: {e}")
+                return {"status": "error", "message": str(e)}
+            except Exception as e:
+                logging.exception(f"AlphaVybe balance check error: {e}")
+                return {"status": "error", "message": str(e)}
 
 class AlphaVybeBalanceCheckerPlugin:
     def __init__(self):