feat: update API implementations and version

Author: tumf

inspect_ankr_types.py

@@ -1,23 +1,30 @@
 """
 Script to inspect Ankr SDK request types and their parameters
 """
+
 import inspect
+from typing import Type
+
 from ankr.types import (
-    GetNFTsByOwnerRequest,
-    GetNFTMetadataRequest,
+    GetAccountBalanceRequest,
     GetBlockchainStatsRequest,
     GetBlocksRequest,
-    GetAccountBalanceRequest,
-    GetTokenPriceRequest
+    GetNFTMetadataRequest,
+    GetNFTsByOwnerRequest,
+    GetTokenPriceRequest,
 )
 
-def print_class_params(cls):
+
+def print_class_params(cls: Type) -> None:
     """Print the parameters for a class constructor"""
     print(f"\n{cls.__name__} parameters:")
     sig = inspect.signature(cls.__init__)
     for param_name, param in sig.parameters.items():
-        if param_name != 'self':
-            print(f"  - {param_name}: {param.default if param.default is not inspect.Parameter.empty else 'REQUIRED'}")
+        if param_name != "self":
+            print(
+                f"  - {param_name}: {param.default if param.default is not inspect.Parameter.empty else 'REQUIRED'}"
+            )
+
 
 print_class_params(GetNFTsByOwnerRequest)
 print_class_params(GetNFTMetadataRequest)

src/web3_mcp/__version__.py

@@ -1,3 +1,3 @@
 """Version information."""
 
-__version__ = "0.1.0" 
+__version__ = "0.1.0"

src/web3_mcp/api/nft.py

@@ -7,13 +7,6 @@
 from ankr import AnkrWeb3
 from pydantic import BaseModel, Field
 
-from ..constants import (
-    NFT_GET_BY_OWNER,
-    NFT_GET_HOLDERS,
-    NFT_GET_METADATA,
-    NFT_GET_TRANSFERS,
-)
-
 
 class NFTCollection(BaseModel):
     blockchain: str
@@ -73,19 +66,19 @@ def __init__(self, client: AnkrWeb3):
     async def get_nfts_by_owner(self, request: NFTByOwnerRequest) -> Dict[str, Any]:
         """Get NFTs owned by a wallet address"""
         from ankr.types import GetNFTsByOwnerRequest
-        
+
         try:
             ankr_request = GetNFTsByOwnerRequest(
                 walletAddress=request.wallet_address,
-                blockchain=request.blockchain if request.blockchain else None
+                blockchain=request.blockchain if request.blockchain else None,
             )
-            
+
             if request.page_size is not None:
                 ankr_request.pageSize = request.page_size
-                
+
             if request.page_token:
                 ankr_request.pageToken = request.page_token
-            
+
             result = self.client.nft.get_nfts(ankr_request)
             assets = list(result) if result else []
             return {"assets": assets, "next_page_token": ""}
@@ -96,47 +89,48 @@ async def get_nfts_by_owner(self, request: NFTByOwnerRequest) -> Dict[str, Any]:
     async def get_nft_metadata(self, request: NFTMetadataRequest) -> Dict[str, Any]:
         """Get metadata for a specific NFT"""
         from ankr.types import GetNFTMetadataRequest
-        
+
         ankr_request = GetNFTMetadataRequest(
             blockchain=request.blockchain,
             contractAddress=request.contract_address,
             tokenId=request.token_id,
-            forceFetch=True
+            forceFetch=True,
         )
-        
+
         result = self.client.nft.get_nft_metadata(ankr_request)
         if hasattr(result, "__dict__"):
             return result.__dict__
         return {
             "name": getattr(result, "name", ""),
             "description": getattr(result, "description", ""),
             "image": getattr(result, "image", ""),
-            "attributes": getattr(result, "attributes", [])
+            "attributes": getattr(result, "attributes", []),
         }
 
     async def get_nft_holders(self, request: NFTHoldersRequest) -> Dict[str, Any]:
         """Get holders of a specific NFT collection"""
         from ankr.types import GetNFTHoldersRequest
-        
+
         ankr_request = GetNFTHoldersRequest(
             blockchain=request.blockchain,
             contractAddress=request.contract_address,
             pageToken=request.page_token,
-            pageSize=request.page_size
+            pageSize=request.page_size,
         )
-        
+
         result = self.client.nft.get_nft_holders(ankr_request)
         if hasattr(result, "__dict__"):
             return result.__dict__
-        
+
         if hasattr(result, "__iter__"):
             holders = list(result) if result else []
-            return {"holders": holders}
+            return {"holders": holders, "next_page_token": ""}
+        return {"holders": [], "next_page_token": ""}
 
     async def get_nft_transfers(self, request: NFTTransfersRequest) -> Dict[str, Any]:
         """Get transfer history for NFTs"""
         from ankr.types import GetNFTTransfersRequest
-        
+
         ankr_request = GetNFTTransfersRequest(
             blockchain=request.blockchain,
             contractAddress=request.contract_address,
@@ -145,13 +139,14 @@ async def get_nft_transfers(self, request: NFTTransfersRequest) -> Dict[str, Any
             fromBlock=request.from_block,
             toBlock=request.to_block,
             pageToken=request.page_token,
-            pageSize=request.page_size
+            pageSize=request.page_size,
         )
-        
+
         result = self.client.nft.get_nft_transfers(ankr_request)
         if hasattr(result, "__dict__"):
             return result.__dict__
-        
+
         if hasattr(result, "__iter__"):
             transfers = list(result) if result else []
-            return {"transfers": transfers}
+            return {"transfers": transfers, "next_page_token": ""}
+        return {"transfers": [], "next_page_token": ""}

src/web3_mcp/api/query.py

@@ -7,15 +7,6 @@
 from ankr import AnkrWeb3
 from pydantic import BaseModel
 
-from ..constants import (
-    QUERY_GET_BLOCKCHAIN_STATS,
-    QUERY_GET_BLOCKS,
-    QUERY_GET_INTERACTIONS,
-    QUERY_GET_LOGS,
-    QUERY_GET_TRANSACTIONS_BY_ADDRESS,
-    QUERY_GET_TRANSACTIONS_BY_HASH,
-)
-
 
 class BlockchainStatsRequest(BaseModel):
     blockchain: str
@@ -76,51 +67,49 @@ def __init__(self, client: AnkrWeb3):
     async def get_blockchain_stats(self, request: BlockchainStatsRequest) -> Dict[str, Any]:
         """Get blockchain statistics"""
         from ankr.types import GetBlockchainStatsRequest
-        
-        ankr_request = GetBlockchainStatsRequest(
-            blockchain=request.blockchain
-        )
-        
+
+        ankr_request = GetBlockchainStatsRequest(blockchain=request.blockchain)
+
         result = self.client.query.get_blockchain_stats(ankr_request)
         if hasattr(result, "__dict__"):
-            return result.__dict__
-        return {
-            "last_block_number": getattr(result, "lastBlockNumber", 0),
+            return {"stats": result.__dict__}
+
+        stats = {
+            "lastBlockNumber": getattr(result, "lastBlockNumber", 0),
             "transactions": getattr(result, "transactions", 0),
-            "tps": getattr(result, "tps", 0)
+            "tps": getattr(result, "tps", 0),
         }
+        return {"stats": stats}
 
     async def get_blocks(self, request: BlocksRequest) -> Dict[str, Any]:
         """Get blocks information"""
         from ankr.types import GetBlocksRequest
-        
-        params = {
-            "blockchain": request.blockchain
-        }
-        
+
+        params = {"blockchain": request.blockchain}
+
         if request.from_block is not None:
-            params["fromBlock"] = request.from_block
-            
+            params["fromBlock"] = str(request.from_block)
+
         if request.to_block is not None:
-            params["toBlock"] = request.to_block
-            
+            params["toBlock"] = str(request.to_block)
+
         if request.descending_order is not None:
-            params["descOrder"] = request.descending_order
-            
-        
+            params["descOrder"] = str(request.descending_order).lower()
+
         ankr_request = GetBlocksRequest(**params)
-        
+
         result = self.client.query.get_blocks(ankr_request)
         if hasattr(result, "__dict__"):
             return result.__dict__
         if hasattr(result, "__iter__"):
             blocks = list(result) if result else []
             return {"blocks": blocks, "next_page_token": ""}
+        return {"blocks": [], "next_page_token": ""}
 
     async def get_logs(self, request: LogsRequest) -> Dict[str, Any]:
         """Get blockchain logs"""
         from ankr.types import GetLogsRequest
-        
+
         ankr_request = GetLogsRequest(
             blockchain=request.blockchain,
             fromBlock=request.from_block,
@@ -129,62 +118,63 @@ async def get_logs(self, request: LogsRequest) -> Dict[str, Any]:
             topics=request.topics,
             descOrder=request.descending_order,
             pageToken=request.page_token,
-            pageSize=request.page_size
+            pageSize=request.page_size,
         )
-        
+
         result = self.client.query.get_logs(ankr_request)
         if hasattr(result, "__dict__"):
             return result.__dict__
         if hasattr(result, "__iter__"):
             logs = list(result) if result else []
             return {"logs": logs, "next_page_token": ""}
+        return {"logs": [], "next_page_token": ""}
 
     async def get_transactions_by_hash(self, request: TransactionsByHashRequest) -> Dict[str, Any]:
         """Get transactions by hash"""
         from ankr.types import GetTransactionByHashRequest
-        
+
         ankr_request = GetTransactionByHashRequest(
-            blockchain=request.blockchain,
-            transaction_hash=request.transaction_hash
+            blockchain=request.blockchain, transaction_hash=request.transaction_hash
         )
-        
+
         result = self.client.query.get_transaction_by_hash(ankr_request)
         if hasattr(result, "__dict__"):
             return result.__dict__
         return {
             "hash": getattr(result, "hash", ""),
             "from": getattr(result, "from", ""),
             "to": getattr(result, "to", ""),
-            "value": getattr(result, "value", "")
+            "value": getattr(result, "value", ""),
         }
 
     async def get_transactions_by_address(
         self, request: TransactionsByAddressRequest
     ) -> Dict[str, Any]:
         """Get transactions by address"""
         from ankr.types import GetTransactionsByAddressRequest
-        
+
         ankr_request = GetTransactionsByAddressRequest(
             blockchain=request.blockchain,
             walletAddress=request.wallet_address,
             fromBlock=request.from_block,
             toBlock=request.to_block,
             descOrder=request.descending_order,
             pageToken=request.page_token,
-            pageSize=request.page_size
+            pageSize=request.page_size,
         )
-        
+
         result = self.client.query.get_transactions_by_address(ankr_request)
         if hasattr(result, "__dict__"):
             return result.__dict__
         if hasattr(result, "__iter__"):
             transactions = list(result) if result else []
             return {"transactions": transactions, "next_page_token": ""}
+        return {"transactions": [], "next_page_token": ""}
 
     async def get_interactions(self, request: InteractionsRequest) -> Dict[str, Any]:
         """Get wallet interactions with contracts"""
         from ankr.types import GetInteractionsRequest
-        
+
         ankr_request = GetInteractionsRequest(
             blockchain=request.blockchain,
             walletAddress=request.wallet_address,
@@ -193,12 +183,13 @@ async def get_interactions(self, request: InteractionsRequest) -> Dict[str, Any]
             contractAddress=request.contract_address,
             descOrder=request.descending_order,
             pageToken=request.page_token,
-            pageSize=request.page_size
+            pageSize=request.page_size,
         )
-        
+
         result = self.client.query.get_interactions(ankr_request)
         if hasattr(result, "__dict__"):
             return result.__dict__
         if hasattr(result, "__iter__"):
             interactions = list(result) if result else []
             return {"interactions": interactions, "next_page_token": ""}
+        return {"interactions": [], "next_page_token": ""}