Merge pull request #77 from pattern-tech/refactor/ai-v2

Fix cursor handling and rename wallet history function

Author: yasinfakhar

src/agentflow/providers/moralis_tools.py

@@ -73,7 +73,7 @@ def get_wallet_token_balances(wallet_address: str, chain: str, output_include: l
         final_results.append({item: result[item]
                               for item in result.keys() if item in output_include})
 
-    return {"cursor": api_result["cursor"],
+    return {"cursor": api_result.get(["cursor"], None),
             "results": final_results}
 
 
@@ -114,7 +114,7 @@ def get_wallet_stats(wallet_address: str, chain: str, output_include: list[str])
 
 @tool
 @handle_exceptions
-def get_wallet_history(wallet_address: str, chain: str, output_include: list[str], cursor: str = "") -> dict:
+def get_wallet_transactions(wallet_address: str, chain: str, output_include: list[str], cursor: str = "") -> dict:
     """
     Retrieve the full transaction history of a specified wallet address, including sends, receives, token and NFT transfers
     and contract interactions in a specific chain. (paginated & in descending order)
@@ -157,7 +157,7 @@ def get_wallet_history(wallet_address: str, chain: str, output_include: list[str
         final_results.append({item: result[item]
                               for item in result.keys() if item in output_include})
 
-    return {"cursor": api_result["cursor"],
+    return {"cursor": api_result.get(["cursor"], None),
             "results": final_results}
 
 
@@ -250,5 +250,5 @@ def get_token_approvals(wallet_address: str, chain: str, output_include: list[st
         final_results.append({item: result[item]
                               for item in result.keys() if item in output_include})
 
-    return {"cursor": api_result["cursor"],
+    return {"cursor": api_result.get(["cursor"], None),
             "results": final_results}