mezoAgent 0.2.3__tar.gz → 0.3.0__tar.gz

{mezoagent-0.2.3 → mezoagent-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mezoAgent
-Version: 0.2.3
+Version: 0.3.0
 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.2.3 → mezoagent-0.3.0}/mezoAgent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mezoAgent
-Version: 0.2.3
+Version: 0.3.0
 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.2.3 → mezoagent-0.3.0}/mezoAgent.egg-info/SOURCES.txt

@@ -8,4 +8,5 @@ mezoAgent.egg-info/top_level.txt
 mezo_agent/__init__.py
 mezo_agent/config.py
 mezo_agent/parsing.py
+mezo_agent/swap_musd_btc.py
 mezo_agent/transaction.py

mezoagent-0.3.0/mezo_agent/__init__.py

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

mezoagent-0.3.0/mezo_agent/config.py

@@ -0,0 +1,44 @@
+import os
+import json
+from dotenv import load_dotenv
+from web3 import Web3
+
+# Load environment variables
+load_dotenv()
+
+# Check for keys
+OpenAI_API_KEY = os.getenv("OPENAI_API_KEY")
+if not OpenAI_API_KEY:
+    raise ValueError("OPENAI_API_KEY not found in environment variables!")
+
+PRIVATE_KEY = os.getenv("PRIVATE_KEY")
+if not PRIVATE_KEY:
+    raise ValueError("PRIVATE_KEY not found in environment variables!")
+
+# Mezo Testnet RPC and Web3 initialization
+RPC_URL = "https://rpc.test.mezo.org"
+web3_instance = Web3(Web3.HTTPProvider(RPC_URL))
+
+# Create Account Object
+account = web3_instance.eth.account.from_key(PRIVATE_KEY)
+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"}],'
+    '"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)
+
+# 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)

{mezoagent-0.2.3 → mezoagent-0.3.0}/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)}"

{mezoagent-0.2.3 → mezoagent-0.3.0}/mezo_agent/transaction.py

@@ -31,7 +31,7 @@ def mezo_agent_transaction_btc(transaction_prompt: str) -> str:
         return f"❌ Insufficient BTC balance! You have {sender_balance_btc} BTC but need {amount} BTC."
 
     # Fetch nonce and gas price
-    nonce = web3_instance.eth.get_transaction_count(sender_address)
+    nonce = web3_instance.eth.get_transaction_count(sender_address, 'pending')
     gas_price = web3_instance.eth.gas_price
     gas_limit = web3_instance.eth.estimate_gas({"to": recipient, "value": amount_wei, "from": sender_address})
 
@@ -77,7 +77,7 @@ def mezo_agent_musd_transaction(transaction_prompt: str) -> str:
 
     try:
         # Fetch nonce and gas price for the sender
-        nonce = web3_instance.eth.get_transaction_count(sender_address)
+        nonce = web3_instance.eth.get_transaction_count(sender_address, 'pending')
         gas_price = web3_instance.eth.gas_price
 
         # Estimate gas required for the token transfer

{mezoagent-0.2.3 → mezoagent-0.3.0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="mezoAgent",
-    version="0.2.3",
+    version="0.3.0",
     packages=find_packages(),
     install_requires=[
         "web3",

mezoagent-0.2.3/mezo_agent/__init__.py

@@ -1 +0,0 @@
-from .transaction import mezo_agent_transaction_btc, mezo_agent_musd_transaction

mezoagent-0.2.3/mezo_agent/config.py

@@ -1,46 +0,0 @@
-import os
-import json
-from dotenv import load_dotenv
-from web3 import Web3
-from web3.exceptions import Web3Exception
-
-# ✅ Load environment variables
-USER_PROJECT_DIR = os.getcwd()
-USER_ENV_PATH = os.path.join(USER_PROJECT_DIR, ".env")
-
-if os.path.exists(USER_ENV_PATH):
-    load_dotenv(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")
-
-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
-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
-
-# ✅ Set up contract details (for mUSD)
-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"}]'
-)
-musd_contract = web3_instance.eth.contract(address=MUSD_ADDRESS, abi=ERC20_ABI)