Armor Crypto MCP

Instructions

Get the balance for a list of wallet/token pairs.

Expects a WalletTokenPairsContainer, returns a list of WalletTokenBalance.

Implementation Reference

• armor_crypto_mcp/armor_mcp.py:112-126 (handler)

  Handler function for the MCP tool 'get_wallet_token_balance'. Registered via @mcp.tool() decorator. Delegates to armor_client.get_wallet_token_balance.

  @mcp.tool()
  async def get_wallet_token_balance(wallet_token_pairs: WalletTokenPairsContainer) -> List[WalletTokenBalance]:
      """
      Get the balance for a list of wallet/token pairs.
      
      Expects a WalletTokenPairsContainer, returns a list of WalletTokenBalance.
      """
      if not armor_client:
          return [{"error": "Not logged in"}]
      try:
          result: List[WalletTokenBalance] = await armor_client.get_wallet_token_balance(wallet_token_pairs)
          return result
      except Exception as e:
          return [{"error": str(e)}]

• armor_crypto_mcp/armor_client.py:549-554 (helper)

  Core helper method in ArmorWalletAPIClient that performs the actual API call to fetch wallet token balances.

  async def get_wallet_token_balance(self, data: WalletTokenPairsContainer) -> List[WalletTokenBalance]:
      """Get balances from a list of wallet and token pairs."""
      # payload = [v.model_dump() for v in data.wallet_token_pairs]
      payload = data.model_dump(exclude_none=True)['wallet_token_pairs']
      return await self._api_call("POST", "tokens/wallet-token-balance/", payload)

• armor_crypto_mcp/armor_client.py:21-30 (schema)

  Pydantic models for input (WalletTokenPairs) and output (WalletTokenBalance) data structures.

  class WalletTokenPairs(BaseModel):
      wallet: str = Field(description="The name of wallet. To get wallet names use `get_user_wallets_and_groups_list`")
      token: str = Field(description="public address of token. To get the address from a token symbol use `get_token_details`")
  
  
  class WalletTokenBalance(BaseModel):
      wallet: str = Field(description="name of wallet")
      token: str = Field(description="public address of token")
      balance: float = Field(description="balance of token")

• armor_crypto_mcp/armor_client.py:417-419 (schema)

  Container model for input list of wallet-token pairs used in the tool.

  class WalletTokenPairsContainer(BaseModel):
      wallet_token_pairs: List[WalletTokenPairs]

• armor_crypto_mcp/armor_mcp.py:10-112 (registration)

  Imports the necessary models and client from armor_client.py, enabling the handler.

  from .armor_client import (
      ArmorWalletAPIClient,
      calculate,
      WalletTokenBalance,
      ConversionResponse,
      SwapQuoteResponse,
      SwapTransactionResponse,
      Wallet,
      TokenDetailsResponseContainer,
      GroupInfo,
      SingleGroupInfo,
      WalletInfo,
      WalletArchiveOrUnarchiveResponse,
      CreateGroupResponse,
      AddWalletToGroupResponse,
      GroupArchiveOrUnarchiveResponse,
      RemoveWalletFromGroupResponse,
      TransferTokenResponse,
      DCAOrderResponse,
      CancelDCAOrderResponse,
      ListSingleGroupRequest,
      TopTrendingTokensRequest,
      CandleStickRequest,
      StakeBalanceResponse,
      ListWalletsRequest,
      ListDCAOrderRequest,
      ListOrderRequest,
      PrivateKeyRequest,
      WalletTokenPairsContainer,
      ConversionRequestContainer,
      SwapQuoteRequestContainer,
      SwapTransactionRequestContainer,
      TokenDetailsRequestContainer,
      TokenSearchRequest,
      TokenSearchResponseContainer,
      TransferTokensRequestContainer,
      DCAOrderRequestContainer,
      CancelDCAOrderRequestContainer,
      CreateWalletRequestContainer,
      ArchiveWalletsRequestContainer,
      UnarchiveWalletRequestContainer,
      CreateGroupsRequestContainer,
      AddWalletToGroupRequestContainer,
      ArchiveWalletGroupRequestContainer,
      UnarchiveWalletGroupRequestContainer,
      RemoveWalletsFromGroupRequestContainer,
      CreateOrderRequestContainer,
      CancelOrderRequestContainer,
      CreateOrderResponseContainer,
      CancelOrderResponseContainer,
      StakeQuoteRequestContainer,
      UnstakeQuoteRequestContainer,
      StakeTransactionRequestContainer,
      UnstakeTransactionRequestContainer,
      RenameWalletRequestContainer,
      ListDCAOrderResponseContainer,
      ListOrderResponseContainer,
  )
  
  # Load environment variables (e.g. BASE_API_URL, etc.)
  load_dotenv()
  
  # Create an MCP server instance with FastMCP
  mcp = FastMCP("Armor Crypto MCP")
  
  # Global variable to hold the authenticated Armor API client
  ACCESS_TOKEN = os.getenv('ARMOR_API_KEY') or os.getenv('ARMOR_ACCESS_TOKEN')
  BASE_API_URL = os.getenv('ARMOR_API_URL') or 'https://app.armorwallet.ai/api/v1'
  
  armor_client = ArmorWalletAPIClient(ACCESS_TOKEN, base_api_url=BASE_API_URL) #, log_path='armor_client.log')
  
  # Include version endpoint
  from armor_crypto_mcp import __version__
  @mcp.tool()
  async def get_armor_mcp_version():
      """Get the current Armor Wallet version"""
      return {'armor_version': __version__}
  
  @mcp.tool()
  async def wait_a_moment(seconds:float):
      """Wait for some short amount of time, no more than 10 seconds"""
      await asyncio.sleep(seconds)
      return {"waited": seconds}
  
  from datetime import datetime, timezone
  @mcp.tool()
  async def get_current_time() -> Dict:
      """Gets the current time and date"""
      return {"current_time": datetime.now(timezone.utc).strftime("%Y-%m-%d %H:%M:%S")}
  
  @mcp.tool()
  async def calculator(expression:str, variables:dict[str, Any]):
      """
      Safely evaluates a mathematical or statistical expression string using Python syntax.
  
      Supports arithmetic operations (+, -, *, /, **, %, //), list expressions, and a range of math and statistics functions: 
      abs, round, min, max, len, sum, mean, median, stdev, variance, sin, cos, tan, sqrt, log, exp, floor, ceil, etc.
  
      Custom variables can be passed via the 'variables' dict, including lists for time series data.
      """
      return {'result': calculate(expression, variables)}
  
  @mcp.tool()