mezoAgent 0.2.4__py3-none-any.whl → 0.3.1__py3-none-any.whl

{mezoAgent-0.2.4.dist-info → mezoAgent-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mezoAgent
-Version: 0.2.4
+Version: 0.3.1
 Summary: A LangChain based tool kit for AI Agents to send BTC and mUSD transactions on Mezo Matsnet.
 Author: Dreadwulf, Duck, Digi
 Requires-Python: >=3.7

mezoAgent-0.3.1.dist-info/RECORD

@@ -0,0 +1,9 @@
+mezo_agent/__init__.py,sha256=XOJWLvrxgH7jBkKdXFLVGLV2dmCyyRpo8VcjicWBD0I,133
+mezo_agent/config.py,sha256=WXzqmJ52kwMcRLIX5Pziq6cP4Zxj9wL6r7CIWvmmURk,2150
+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
+mezoAgent-0.3.1.dist-info/METADATA,sha256=VXHIaC450cMMhGoxON7TAcepsqilXN2OkIkSRpMzHSM,323
+mezoAgent-0.3.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+mezoAgent-0.3.1.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
+mezoAgent-0.3.1.dist-info/RECORD,,

mezo_agent/__init__.py

@@ -1 +1,2 @@
-from .transaction import mezo_agent_transaction_btc, mezo_agent_musd_transaction
+from .transaction import mezo_agent_transaction_btc, mezo_agent_musd_transaction
+from .swap_musd_btc import mezo_agent_swap_musd_btc

mezo_agent/config.py

@@ -2,9 +2,10 @@ import os
 import json
 from dotenv import load_dotenv
 from web3 import Web3
-from web3.exceptions import Web3Exception
 
-# ✅ Load environment variables
+# Load environment variables
+load_dotenv()
+
 USER_PROJECT_DIR = os.getcwd()
 USER_ENV_PATH = os.path.join(USER_PROJECT_DIR, ".env")
 
@@ -13,7 +14,6 @@ if os.path.exists(USER_ENV_PATH):
 else:
     print("⚠️ Warning: No `.env` file found in your project directory! Transactions requiring signing may fail.")
 
-# ✅ Load keys from the environment
 PRIVATE_KEY = os.getenv("PRIVATE_KEY")
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
@@ -21,26 +21,30 @@ if not PRIVATE_KEY:
     print("⚠️ Warning: PRIVATE_KEY not set. Please create a `.env` file in your project with your keys.")
     PRIVATE_KEY = None  # Allow package to be installed but prevent transactions
 
-# ✅ Define RPC URL and initialize Web3
+# Mezo Testnet RPC and Web3 initialization
 RPC_URL = "https://rpc.test.mezo.org"
 web3_instance = Web3(Web3.HTTPProvider(RPC_URL))
 
-if not web3_instance.is_connected():
-    raise ConnectionError("❌ Failed to connect to Mezo Matsnet RPC.")
-
-# ✅ Initialize account and sender address
-if PRIVATE_KEY:
-    account = web3_instance.eth.account.from_key(PRIVATE_KEY)
-    sender_address = account.address
-else:
-    account = None
-    sender_address = None
+# Create Account Object
+account = web3_instance.eth.account.from_key(PRIVATE_KEY)
+sender_address = account.address
 
-# ✅ Set up contract details (for mUSD)
+# mUSD Contract Setup using approve/allowance ABI
 MUSD_ADDRESS = "0x637e22A1EBbca50EA2d34027c238317fD10003eB"
 ERC20_ABI = json.loads(
-    '[{"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)
+musd_contract = web3_instance.eth.contract(address=MUSD_ADDRESS, abi=ERC20_ABI)
+
+# Wrapped BTC and Swap Router setup
+WRAPPED_BTC_ADDRESS = "0xA460F83cdd9584E4bD6a9838abb0baC58EAde999"
+ROUTER_ADDRESS = "0xC2E61936a542D78b9c3AA024fA141c4C632DF6c1"
+
+# Load router ABI from a JSON file packaged with mezoAgent
+router_abi_path = os.path.join(os.path.dirname(__file__), "new_router_abi.json")
+with open(router_abi_path, "r") as f:
+    router_abi = json.load(f)
+router_contract = web3_instance.eth.contract(address=ROUTER_ADDRESS, abi=router_abi)

mezo_agent/parsing.py

@@ -33,4 +33,33 @@ def extract_transaction_details(prompt: str):
     try:
         return output_parser.parse(response.content)
     except Exception as e:
-        return f"❌ Failed to extract transaction details: {str(e)}"
+        return f"❌ Failed to extract transaction details: {str(e)}"
+    
+
+swap_response_schemas = [
+    ResponseSchema(name="amount", description="The amount of mUSD to swap."),
+    ResponseSchema(name="from_currency", description="The token to swap from (should always be 'mUSD')."),
+    ResponseSchema(name="to_currency", description="The token to receive (should always be 'BTC')."),
+    ResponseSchema(name="router_address", description="The Dumpy Swap router address for executing the swap."),
+]
+swap_output_parser = StructuredOutputParser.from_response_schemas(swap_response_schemas)
+
+swap_prompt_template = PromptTemplate(
+    template=(
+        "Extract swap transaction details from this request:\n{input}\n\n"
+        "- The token to swap from should always be 'mUSD'.\n"
+        "- The token to receive should always be 'BTC'.\n"
+        "- The router address should always be '0xC2E61936a542D78b9c3AA024fA141c4C632DF6c1'.\n\n"
+        "{format_instructions}"
+    ),
+    input_variables=["input"],
+    partial_variables={"format_instructions": swap_output_parser.get_format_instructions()},
+)
+
+def extract_swap_details(prompt: str):
+    formatted_prompt = swap_prompt_template.format(input=prompt)
+    response = llm.invoke(formatted_prompt)
+    try:
+        return swap_output_parser.parse(response.content)
+    except Exception as e:
+        return f"Failed to extract swap details: {str(e)}"

mezo_agent/swap_musd_btc.py

@@ -0,0 +1,112 @@
+import time
+from langchain.tools import tool
+from .config import (
+    web3_instance,
+    sender_address,
+    account,
+    PRIVATE_KEY,
+    musd_contract,
+    ROUTER_ADDRESS,
+    router_contract,
+    MUSD_ADDRESS,
+    WRAPPED_BTC_ADDRESS
+)
+from .parsing import extract_swap_details
+
+def approve_if_needed(token_contract, amount_wei):
+    """
+    Checks if the router has enough allowance to spend tokens.
+    If not, sends an approval transaction.
+    """
+    current_allowance = token_contract.functions.allowance(sender_address, ROUTER_ADDRESS).call()
+    if current_allowance < amount_wei:
+        print(f"Current allowance ({current_allowance}) is less than required ({amount_wei}). Approving...")
+        nonce = web3_instance.eth.get_transaction_count(sender_address)
+        gas_price = web3_instance.eth.gas_price
+        approve_tx = token_contract.functions.approve(ROUTER_ADDRESS, amount_wei).build_transaction({
+            "from": sender_address,
+            "nonce": nonce,
+            "gas": 50000,
+            "gasPrice": gas_price,
+        })
+        signed_tx = web3_instance.eth.account.sign_transaction(approve_tx, PRIVATE_KEY)
+        tx_hash = web3_instance.eth.send_raw_transaction(signed_tx.raw_transaction)
+        receipt = web3_instance.eth.wait_for_transaction_receipt(tx_hash)
+        if receipt.status != 1:
+            raise Exception("Approval transaction failed.")
+        print(f"Approval successful. TX Hash: {tx_hash.hex()}")
+    else:
+        print("Sufficient allowance already set.")
+
+@tool
+def mezo_agent_swap_musd_btc(prompt: str) -> str:
+    """
+    Swaps mUSD for Wrapped BTC using the Dumpy Swap router.
+    
+    This tool extracts swap details from the prompt, approves the router if needed,
+    and executes the swap transaction.
+    
+    The prompt should specify the mUSD amount to swap.
+    """
+    swap_details = extract_swap_details(prompt)
+    if isinstance(swap_details, str):
+        return swap_details
+
+    try:
+        amount_musd = float(swap_details["amount"])
+    except KeyError as e:
+        return f"❌ Missing key in swap details: {str(e)}"
+
+    # For demonstration, we use a dummy minimum Wrapped BTC amount.
+    min_wrapped_btc = 0.000000000000001  # in BTC
+
+    # Convert amounts to Wei (assuming 18 decimals)
+    amount_musd_wei = int(amount_musd * 10**18)
+    min_wrapped_btc_wei = int(min_wrapped_btc * 10**18)
+    deadline = int(time.time()) + 600  # 10 minutes from now
+
+    # Approve the router to spend mUSD if needed
+    try:
+        approve_if_needed(musd_contract, amount_musd_wei)
+    except Exception as e:
+        return f"❌ Approval failed: {str(e)}"
+
+    # Define the swap path: mUSD -> Wrapped BTC
+    path = [MUSD_ADDRESS, WRAPPED_BTC_ADDRESS]
+
+    nonce = web3_instance.eth.get_transaction_count(sender_address)
+    gas_price = web3_instance.eth.gas_price
+
+    try:
+        swap_tx = router_contract.functions.swapExactTokensForTokens(
+            amount_musd_wei,        # mUSD amount
+            min_wrapped_btc_wei,     # Minimum Wrapped BTC to receive
+            path,                   # Swap path
+            sender_address,         # Recipient address
+            deadline                # Transaction deadline
+        ).build_transaction({
+            "from": sender_address,
+            "nonce": nonce,
+            "gasPrice": gas_price,
+        })
+    except Exception as e:
+        return f"❌ Failed to build swap transaction: {str(e)}"
+
+    # Estimate gas and add a buffer
+    try:
+        estimated_gas = web3_instance.eth.estimate_gas(swap_tx)
+        swap_tx["gas"] = estimated_gas + 10000
+    except Exception as e:
+        swap_tx["gas"] = 250000  # Fallback gas limit
+
+    try:
+        signed_swap_tx = web3_instance.eth.account.sign_transaction(swap_tx, PRIVATE_KEY)
+        tx_hash = web3_instance.eth.send_raw_transaction(signed_swap_tx.raw_transaction)
+    except Exception as e:
+        return f"❌ Swap transaction failed: {str(e)}"
+
+    receipt = web3_instance.eth.wait_for_transaction_receipt(tx_hash)
+    if receipt.status != 1:
+        return f"❌ Swap transaction failed. Receipt: {receipt}"
+    
+    return f"✅ Swap Successful! TX Hash: {tx_hash.hex()}"

mezoAgent-0.2.4.dist-info/RECORD

@@ -1,8 +0,0 @@
-mezo_agent/__init__.py,sha256=lsonH0-yY3Die0cRmiwsPSHwZQDh_UEAVIpc2uwZYIM,80
-mezo_agent/config.py,sha256=c16-JHEbK5rFBKO-QdxBRtrl4DSlYzGckD9BzbTw2dY,1708
-mezo_agent/parsing.py,sha256=dFqNHGH0LoG4_TEsHKcorj6aY_Rh4H8f5Yj7ZdfKJzE,1454
-mezo_agent/transaction.py,sha256=YpfWrkEaf0YGM_Kc4cFwlT9GmBGZkeJHWm0VGHs9Gks,4199
-mezoAgent-0.2.4.dist-info/METADATA,sha256=ACNVZPv9OaqSgU_81bYj8BDGR3-ze1IPqkH89PEcD0g,323
-mezoAgent-0.2.4.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-mezoAgent-0.2.4.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
-mezoAgent-0.2.4.dist-info/RECORD,,