mezoAgent 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

mezoAgent-0.1.4.dist-info/METADATA

@@ -0,0 +1,11 @@
+Metadata-Version: 2.1
+Name: mezoAgent
+Version: 0.1.4
+Summary: A LangChain based tool kit for AI Agents to send BTC and mUSD transactions on Mezo Matsnet.
+Author: Dreadwulf
+Author-email: dreadwul@wtf.com
+Requires-Python: >=3.7
+Requires-Dist: web3
+Requires-Dist: langchain
+Requires-Dist: langchain-openai
+Requires-Dist: python-dotenv

mezoAgent-0.1.4.dist-info/RECORD

@@ -0,0 +1,8 @@
+mezo_agent/__init__.py,sha256=CTQD3Y_lY9lzhuTzIeQjqsNzy7sYjOPygVSSK3wBlps,51
+mezo_agent/config.py,sha256=c16-JHEbK5rFBKO-QdxBRtrl4DSlYzGckD9BzbTw2dY,1708
+mezo_agent/parsing.py,sha256=dFqNHGH0LoG4_TEsHKcorj6aY_Rh4H8f5Yj7ZdfKJzE,1454
+mezo_agent/transaction.py,sha256=75q88q94WJLg0UahMhkAfMqSxBfW-bqK3iZMGykmc-k,2069
+mezoAgent-0.1.4.dist-info/METADATA,sha256=xYqUvTErxTwqLghZILJOFYOAmhOzfx96wOuA8k9eK2Y,343
+mezoAgent-0.1.4.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+mezoAgent-0.1.4.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
+mezoAgent-0.1.4.dist-info/RECORD,,

mezoAgent-0.1.4.dist-info/top_level.txt

@@ -0,0 +1 @@
+mezo_agent

mezo_agent/__init__.py

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

mezo_agent/config.py

@@ -0,0 +1,46 @@
+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)

mezo_agent/parsing.py

@@ -0,0 +1,36 @@
+from langchain.output_parsers import StructuredOutputParser, ResponseSchema
+from langchain.prompts import PromptTemplate
+from langchain_openai import ChatOpenAI
+from .config import OPENAI_API_KEY
+
+# Define the expected response schema
+response_schemas = [
+    ResponseSchema(name="amount", description="The amount of cryptocurrency to transfer."),
+    ResponseSchema(name="currency", description="The cryptocurrency (BTC or mUSD)."),
+    ResponseSchema(name="recipient", description="The recipient's wallet address."),
+]
+
+# Initialize the structured output parser
+output_parser = StructuredOutputParser.from_response_schemas(response_schemas)
+
+# Instantiate the ChatOpenAI LLM
+llm = ChatOpenAI(temperature=0, openai_api_key=OPENAI_API_KEY)
+
+# Prepare the prompt template
+prompt_template = PromptTemplate(
+    template="Extract transaction details from this request:\n{input}\n{format_instructions}",
+    input_variables=["input"],
+    partial_variables={"format_instructions": output_parser.get_format_instructions()},
+)
+
+def extract_transaction_details(prompt: str):
+    """
+    Uses LLM to extract structured transaction details from user input.
+    """
+    formatted_prompt = prompt_template.format(input=prompt)
+    response = llm.invoke(formatted_prompt)
+
+    try:
+        return output_parser.parse(response.content)
+    except Exception as e:
+        return f"❌ Failed to extract transaction details: {str(e)}"

mezo_agent/transaction.py

@@ -0,0 +1,53 @@
+from web3.exceptions import Web3Exception
+from langchain.tools import tool
+from .config import web3_instance, sender_address, account
+from .parsing import extract_transaction_details
+
+@tool
+def mezo_agent_transaction_btc(transaction_prompt: str) -> str:
+    """
+    Sends BTC on Mezo Matsnet. Parses a transaction request and executes the transfer.
+    """
+    transaction_details = extract_transaction_details(transaction_prompt)
+
+    if isinstance(transaction_details, str):  # Handle errors
+        return transaction_details
+
+    amount = float(transaction_details["amount"])
+    currency = transaction_details["currency"].lower()
+    recipient = transaction_details["recipient"]
+
+    if currency != "btc":
+        return "❌ This function only supports BTC transactions."
+
+    # Convert amount to Wei (BTC uses 18 decimals on Mezo Matsnet)
+    amount_wei = web3_instance.to_wei(amount, "ether")
+
+    # Check sender's BTC balance
+    sender_balance = web3_instance.eth.get_balance(sender_address)
+    sender_balance_btc = web3_instance.from_wei(sender_balance, "ether")
+
+    if sender_balance < amount_wei:
+        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)
+    gas_price = web3_instance.eth.gas_price
+    gas_limit = web3_instance.eth.estimate_gas({"to": recipient, "value": amount_wei, "from": sender_address})
+
+    tx = {
+        "to": recipient,
+        "value": amount_wei,
+        "gas": gas_limit,
+        "gasPrice": gas_price,
+        "nonce": nonce,
+        "chainId": 31611,  # Mezo Testnet Chain ID
+    }
+
+    try:
+        # Sign and send the transaction
+        signed_tx = account.sign_transaction(tx)
+        tx_hash = web3_instance.eth.send_raw_transaction(signed_tx.raw_transaction)
+        return f"✅ BTC Transaction Successful! Hash: {tx_hash.hex()}"
+    except Web3Exception as e:
+        return f"❌ BTC Transaction Failed: {str(e)}"

mezoAgent-0.1.3.dist-info/METADATA

@@ -1,15 +0,0 @@
-Metadata-Version: 2.1
-Name: mezoAgent
-Version: 0.1.3
-Summary: A LangChain tool for sending BTC and mUSD transactions on Mezo Matsnet.
-Home-page: https://github.com/yourusername/mezoTransactionTool
-Author: Dreadwulf
-Author-email: dreadwul@wtf.com
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.7
-Requires-Dist: web3
-Requires-Dist: langchain
-Requires-Dist: langchain-openai
-Requires-Dist: python-dotenv

mezoAgent-0.1.3.dist-info/RECORD

@@ -1,6 +0,0 @@
-mezoTransactionTool/__init__.py,sha256=4izlG8P3PTGXZb_RvEQ3HlPMxQ1jzSINnF9nSlxnEXE,161
-mezoTransactionTool/transaction_tools.py,sha256=0VxSaiMSQW067K6qw2SJR9dm-CH7obwMF0cagbEEX1g,5013
-mezoAgent-0.1.3.dist-info/METADATA,sha256=nMCSfEH-FTHSDb8iEP1W2sGNtNu7ahB-ZzEGy5CZFBo,536
-mezoAgent-0.1.3.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-mezoAgent-0.1.3.dist-info/top_level.txt,sha256=Czoa5A8CU_I_oPKDa5OXa5H7RM4Jk94PF9mIpz2V9t4,20
-mezoAgent-0.1.3.dist-info/RECORD,,

mezoAgent-0.1.3.dist-info/top_level.txt

@@ -1 +0,0 @@
-mezoTransactionTool

mezoTransactionTool/__init__.py

@@ -1,3 +0,0 @@
-from .transaction_tools import mezo_agent_transaction_btc, mezo_agent_transaction_musd
-
-__all__ = ["mezo_agent_transaction_btc", "mezo_agent_transaction_musd"]

mezoTransactionTool/transaction_tools.py

@@ -1,126 +0,0 @@
-import os
-import json
-from dotenv import load_dotenv
-from web3 import Web3
-from web3.exceptions import Web3Exception
-from langchain.tools import tool
-from langchain_openai import ChatOpenAI
-from langchain.output_parsers import StructuredOutputParser, ResponseSchema
-from langchain.prompts import PromptTemplate
-
-# ✅ 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 Private Key from User's `.env`
-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 before initializing Web3
-RPC_URL = "https://rpc.test.mezo.org"
-web3_instance = Web3(Web3.HTTPProvider(RPC_URL))
-
-# ✅ Check Web3 connection
-if not web3_instance.is_connected():
-    raise ConnectionError("❌ Failed to connect to Mezo Matsnet RPC.")
-
-# ✅ Initialize account object using `web3_instance`
-if PRIVATE_KEY:
-    account = web3_instance.eth.account.from_key(PRIVATE_KEY)
-    sender_address = account.address
-else:
-    account = None
-    sender_address = None
-
-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)
-
-# ✅ Define Structured Output Parser for Transaction Details
-response_schemas = [
-    ResponseSchema(name="amount", description="The amount of cryptocurrency to transfer."),
-    ResponseSchema(name="currency", description="The cryptocurrency (BTC or mUSD)."),
-    ResponseSchema(name="recipient", description="The recipient's wallet address."),
-]
-
-output_parser = StructuredOutputParser.from_response_schemas(response_schemas)
-
-# ✅ Define LLM Prompt for Parsing Transactions
-llm = ChatOpenAI(temperature=0, openai_api_key=OPENAI_API_KEY)
-
-prompt_template = PromptTemplate(
-    template="Extract transaction details from this request:\n{input}\n{format_instructions}",
-    input_variables=["input"],
-    partial_variables={"format_instructions": output_parser.get_format_instructions()},
-)
-
-def extract_transaction_details(prompt: str):
-    """
-    Uses LLM to extract structured transaction details from user input.
-    """
-    formatted_prompt = prompt_template.format(input=prompt)
-    response = llm.invoke(formatted_prompt)
-
-    try:
-        return output_parser.parse(response.content)
-    except Exception as e:
-        return f"❌ Failed to extract transaction details: {str(e)}"
-
-@tool
-def mezo_agent_transaction_btc(transaction_prompt: str) -> str:
-    """
-    Sends BTC on Mezo Matsnet. Parses a transaction request and executes the transfer.
-    """
-    transaction_details = extract_transaction_details(transaction_prompt)
-
-    if isinstance(transaction_details, str):  # Handle errors
-        return transaction_details
-
-    amount = float(transaction_details["amount"])
-    currency = transaction_details["currency"].lower()
-    recipient = transaction_details["recipient"]
-
-    if currency != "btc":
-        return "❌ This function only supports BTC transactions."
-
-    # ✅ Convert amount to Wei (BTC uses 18 decimals on Mezo Matsnet)
-    amount_wei = web3_instance.to_wei(amount, "ether")
-
-    # ✅ Check sender's BTC balance
-    sender_balance = web3_instance.eth.get_balance(sender_address)
-    sender_balance_btc = web3_instance.from_wei(sender_balance, "ether")
-
-    if sender_balance < amount_wei:
-        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)
-    gas_price = web3_instance.eth.gas_price
-    gas_limit = web3_instance.eth.estimate_gas({"to": recipient, "value": amount_wei, "from": sender_address})
-
-    tx = {
-        "to": recipient,
-        "value": amount_wei,
-        "gas": gas_limit,
-        "gasPrice": gas_price,
-        "nonce": nonce,
-        "chainId": 31611,  # Mezo Testnet Chain ID
-    }
-
-    try:
-        # ✅ Sign and send the transaction
-        signed_tx = account.sign_transaction(tx)
-        tx_hash = web3_instance.eth.send_raw_transaction(signed_tx.raw_transaction)
-        return f"✅ BTC Transaction Successful! Hash: {tx_hash.hex()}"
-
-    except Web3Exception as e:
-        return f"❌ BTC Transaction Failed: {str(e)}"