mezoAgent 0.4.2__py3-none-any.whl → 0.5.0__py3-none-any.whl

{mezoAgent-0.4.2.dist-info → mezoAgent-0.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mezoAgent
-Version: 0.4.2
+Version: 0.5.0
 Summary: A Python package for Mezo Agent transactions with LangChain tools
 Author: Dreadwulf, Duck, Digi
 Requires-Dist: python-dotenv

mezoAgent-0.5.0.dist-info/RECORD

@@ -0,0 +1,14 @@
+mezo_agent/__init__.py,sha256=naG5ZIXKDWTRlqdx2oi1zJcRZpJBO9ugVkW0MaPkZuk,497
+mezo_agent/characters.py,sha256=wub7y9BrAAdAKWwcNAdgrvhRi2zHMaZNzTr2y3qtbsw,515
+mezo_agent/chat.py,sha256=ZtNurUDV7UzJjbFyU96DNGy37hO3gl0osn19mRu7ER8,859
+mezo_agent/config.py,sha256=LgMTntnzXrY7giGXy6JZqhPwPxxMpWWOgc2Ah6g3Y48,3017
+mezo_agent/parsing.py,sha256=6g0U4x8nQaOkwIPon1UlKlnVX6-S2Tf4LKqdY1ZGyiY,4000
+mezo_agent/swap_musd_btc.py,sha256=Co-XcfK73spm94VC6qzGY0VYoGwsMDKKHM8ToGh2JxU,4289
+mezo_agent/token_balance_tool.py,sha256=GQsZnM6zwIdCL5fRISd_KAdc8RNVOTa0LYHPhPIdW1g,1304
+mezo_agent/transaction.py,sha256=YpfWrkEaf0YGM_Kc4cFwlT9GmBGZkeJHWm0VGHs9Gks,4199
+mezo_agent/utils.py,sha256=Kdq3YRj6fOnAmDjI6vktvOvN0vAR8U7TgoTGqcA2KcI,973
+mezo_agent/data/new_router.json,sha256=A8U-NVfe1F-hDyR90_SuH8jDAxmzyyHWdJW62j9TZsc,26756
+mezoAgent-0.5.0.dist-info/METADATA,sha256=ZfVWkIcivB8L4OlUGNFpwhg-SQXFqHp8wkFg-Ugh4W0,273
+mezoAgent-0.5.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+mezoAgent-0.5.0.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
+mezoAgent-0.5.0.dist-info/RECORD,,

mezo_agent/__init__.py

@@ -1,12 +1,14 @@
 from .transaction import mezo_agent_transaction_btc, mezo_agent_musd_transaction
 from .swap_musd_btc import mezo_agent_swap_musd_btc
-from .chat import mezo_character_chat  # Corrected function name
+from .chat import mezo_character_chat
 from .characters import get_character_prompt
+from .token_balance_tool import mezo_agent_token_balance_tool
 
 __all__ = [
     "mezo_agent_transaction_btc",
     "mezo_agent_musd_transaction",
     "mezo_agent_swap_musd_btc",
-    "mezo_character_chat",  # Corrected function name
+    "mezo_character_chat",
+    "mezo_agent_token_balance_tool",
     "get_character_prompt"
 ]

mezo_agent/config.py

@@ -3,6 +3,8 @@ import json
 from dotenv import load_dotenv
 from web3 import Web3
 import importlib.resources as pkg_resources
+import requests
+
 # Load environment variables
 load_dotenv()
 
@@ -25,6 +27,9 @@ if not PRIVATE_KEY:
 RPC_URL = "https://rpc.test.mezo.org"
 web3_instance = Web3(Web3.HTTPProvider(RPC_URL))
 
+#Graph endpoint
+GRAPH_URL = "https://api.goldsky.com/api/public/project_cm48lsrzo0axx01tna6rb1ee9/subgraphs/exchange-v2-mezo/1.0.0/gn"
+
 # Create Account Object
 account = web3_instance.eth.account.from_key(PRIVATE_KEY)
 sender_address = account.address
@@ -32,13 +37,28 @@ sender_address = account.address
 # mUSD Contract Setup using approve/allowance ABI
 MUSD_ADDRESS = "0x637e22A1EBbca50EA2d34027c238317fD10003eB"
 ERC20_ABI = json.loads(
-    '[{"constant": false, "inputs": [{"name": "spender", "type": "address"}, {"name": "amount", "type": "uint256"}],'
+    '[{"constant": false, "inputs": [{"name": "recipient", "type": "address"}, {"name": "amount", "type": "uint256"}],'
+    '"name": "transfer", "outputs": [{"name": "", "type": "bool"}], "stateMutability": "nonpayable", "type": "function"},'
+    '{"constant": false, "inputs": [{"name": "spender", "type": "address"}, {"name": "amount", "type": "uint256"}],'
     '"name": "approve", "outputs": [{"name": "", "type": "bool"}], "stateMutability": "nonpayable", "type": "function"},'
     '{"constant": true, "inputs": [{"name": "owner", "type": "address"}, {"name": "spender", "type": "address"}],'
     '"name": "allowance", "outputs": [{"name": "remaining", "type": "uint256"}], "stateMutability": "view", "type": "function"}]'
 )
 musd_contract = web3_instance.eth.contract(address=MUSD_ADDRESS, abi=ERC20_ABI)
 
+#Query graph for token info
+def query_graph(query: str) -> dict:
+    """
+    Queries the Goldsky GraphQL API for blockchain data.
+    :param query: GraphQL query string.
+    :return: JSON response as a dictionary.
+    """
+    response = requests.post(GRAPH_URL, json={"query": query})
+    if response.status_code == 200:
+        return response.json()
+    else:
+        raise Exception(f"GraphQL query failed with status code {response.status_code}")
+
 # Wrapped BTC and Swap Router setup
 WRAPPED_BTC_ADDRESS = "0xA460F83cdd9584E4bD6a9838abb0baC58EAde999"
 ROUTER_ADDRESS = "0xC2E61936a542D78b9c3AA024fA141c4C632DF6c1"
@@ -47,4 +67,4 @@ ROUTER_ADDRESS = "0xC2E61936a542D78b9c3AA024fA141c4C632DF6c1"
 with pkg_resources.open_text("mezo_agent.data", "new_router.json") as f:
     router_abi = json.load(f)
 
-router_contract = web3_instance.eth.contract(address=ROUTER_ADDRESS, abi=router_abi)
+router_contract = web3_instance.eth.contract(address=ROUTER_ADDRESS, abi=router_abi)

mezo_agent/parsing.py

@@ -62,4 +62,38 @@ def extract_swap_details(prompt: str):
     try:
         return swap_output_parser.parse(response.content)
     except Exception as e:
-        return f"Failed to extract swap details: {str(e)}"
+        return f"Failed to extract swap details: {str(e)}"
+    
+#parsing for balance check tool 
+
+balance_response_schema = [
+    ResponseSchema(name="token_symbol", description="The token symbol to check the balance for (e.g., 'MUSD').")
+]
+
+balance_output_parser = StructuredOutputParser.from_response_schemas(balance_response_schema)
+
+balance_prompt_template = PromptTemplate(
+    template="""Output a JSON object with the following key:
+- token_symbol: (string) The token symbol for which to check the balance (e.g., "MUSD").
+
+Extract this detail from the following query:
+{input}
+
+Your output must be a valid JSON object with no additional text.
+""",
+    input_variables=["input"],
+)
+
+def extract_balance_details(prompt: str):
+    """
+    Parses a balance query and extracts the token symbol.
+    
+    :param prompt: User input asking for a token balance.
+    :return: Dictionary containing the token symbol or an error message.
+    """
+    try:
+        formatted_prompt = balance_prompt_template.format(input=prompt)
+        response = balance_output_parser.parse(formatted_prompt)
+        return response
+    except Exception as e:
+        return f"❌ Failed to extract balance details: {str(e)}"

mezo_agent/token_balance_tool.py

@@ -0,0 +1,32 @@
+from langchain.tools import tool
+from mezo_agent.config import web3_instance, sender_address, extract_balance_details, ERC20_ABI
+from mezo_agent.utils import get_token_address_by_symbol
+
+@tool
+def mezo_agent_token_balance_tool(balance_prompt: str) -> str:
+    """
+    Queries the wallet balance for a specified token on Mezo Matsnet.
+
+    :param balance_prompt: User query containing the token symbol.
+    :return: Token balance for the sender's address.
+    """
+    details = extract_balance_details(balance_prompt)
+
+    if isinstance(details, str):  # Handle extraction errors
+        return details
+
+    token_symbol = details.get("token_symbol")
+    if not token_symbol:
+        return "❌ Could not extract token symbol for balance query."
+
+    try:
+        # Get the contract address dynamically
+        token_address = get_token_address_by_symbol(token_symbol)
+        token_contract = web3_instance.eth.contract(address=token_address, abi=ERC20_ABI)
+
+        # Fetch balance
+        balance_wei = token_contract.functions.balanceOf(sender_address).call()
+        balance = web3_instance.from_wei(balance_wei, "ether")
+        return f"✅ {token_symbol.upper()} Balance: {balance}"
+    except Exception as e:
+        return f"❌ Failed to fetch balance: {str(e)}"

mezo_agent/utils.py

@@ -0,0 +1,29 @@
+from mezo_agent.config import web3_instance, query_graph
+
+def get_token_address_by_symbol(symbol: str) -> str:
+    """
+    Fetches the token contract address dynamically from the Goldsky subgraph.
+    :param symbol: Token symbol (e.g., 'MUSD', 'WBTC').
+    :return: Ethereum checksum address of the token.
+    """
+    if symbol.lower() == "btc":
+        symbol = "wtbtc"
+
+    query_all = '''
+    {
+      tokens(first: 100) {
+        id
+        symbol
+      }
+    }
+    '''
+    data = query_graph(query_all)
+    tokens = data.get("data", {}).get("tokens", [])
+    matching_tokens = [t for t in tokens if t["symbol"].lower() == symbol.lower()]
+
+    if matching_tokens:
+        token_address = matching_tokens[0]["id"]
+        return web3_instance.to_checksum_address(token_address)
+    else:
+        available_symbols = [t["symbol"] for t in tokens]
+        raise Exception(f"Token '{symbol}' not found. Available: {', '.join(available_symbols)}")

mezoAgent-0.4.2.dist-info/RECORD

@@ -1,12 +0,0 @@
-mezo_agent/__init__.py,sha256=PRFZfcq8B68pQI2toci1ACM_SPcu7suy9D3OD8IvZeI,450
-mezo_agent/characters.py,sha256=wub7y9BrAAdAKWwcNAdgrvhRi2zHMaZNzTr2y3qtbsw,515
-mezo_agent/chat.py,sha256=ZtNurUDV7UzJjbFyU96DNGy37hO3gl0osn19mRu7ER8,859
-mezo_agent/config.py,sha256=1PWeaHdQPKqKPraE-tM81CwRKeBiCbz5C_8iZTEiCgo,2148
-mezo_agent/parsing.py,sha256=rolPVE8r_RS6BqKe25r1nyZQ47rvTPG2Hv09Kxxfv4c,2821
-mezo_agent/swap_musd_btc.py,sha256=Co-XcfK73spm94VC6qzGY0VYoGwsMDKKHM8ToGh2JxU,4289
-mezo_agent/transaction.py,sha256=YpfWrkEaf0YGM_Kc4cFwlT9GmBGZkeJHWm0VGHs9Gks,4199
-mezo_agent/data/new_router.json,sha256=A8U-NVfe1F-hDyR90_SuH8jDAxmzyyHWdJW62j9TZsc,26756
-mezoAgent-0.4.2.dist-info/METADATA,sha256=YEbcKgTvvW13zDdNJKRz47odDmCOX73-Bl_KTa3Qn_k,273
-mezoAgent-0.4.2.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-mezoAgent-0.4.2.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
-mezoAgent-0.4.2.dist-info/RECORD,,