mezoAgent 0.1.1__tar.gz → 0.1.3__tar.gz

{mezoagent-0.1.1 → mezoagent-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mezoAgent
-Version: 0.1.1
+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

{mezoagent-0.1.1 → mezoagent-0.1.3}/mezoAgent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mezoAgent
-Version: 0.1.1
+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

{mezoagent-0.1.1 → mezoagent-0.1.3}/mezoTransactionTool/transaction_tools.py

@@ -9,7 +9,7 @@ from langchain.output_parsers import StructuredOutputParser, ResponseSchema
 from langchain.prompts import PromptTemplate
 
 # ✅ Load environment variables
-USER_PROJECT_DIR = os.getcwd()  # Get the directory where the script is run
+USER_PROJECT_DIR = os.getcwd()
 USER_ENV_PATH = os.path.join(USER_PROJECT_DIR, ".env")
 
 if os.path.exists(USER_ENV_PATH):
@@ -19,17 +19,31 @@ else:
 
 # ✅ 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
 
-# ✅ Create Account Object
-account = Web3.eth.account.from_key(PRIVATE_KEY)
-sender_address = account.address
+# ✅ 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"  
+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.eth.contract(address=MUSD_ADDRESS, abi=ERC20_ABI)
+musd_contract = web3_instance.eth.contract(address=MUSD_ADDRESS, abi=ERC20_ABI)
 
 # ✅ Define Structured Output Parser for Transaction Details
 response_schemas = [
@@ -41,7 +55,7 @@ response_schemas = [
 output_parser = StructuredOutputParser.from_response_schemas(response_schemas)
 
 # ✅ Define LLM Prompt for Parsing Transactions
-llm = ChatOpenAI(temperature=0, openai_api_key=os.getenv("OPENAI_API_KEY"))
+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}",
@@ -65,14 +79,6 @@ def extract_transaction_details(prompt: str):
 def mezo_agent_transaction_btc(transaction_prompt: str) -> str:
     """
     Sends BTC on Mezo Matsnet. Parses a transaction request and executes the transfer.
-
-    Example usage:
-    ```
-    mezo_agent_transaction_btc("Send 0.05 BTC to 0x93a1Eadb069A791d23aAeDF3C272E2905bb63240")
-    ```
-
-    :param transaction_prompt: Natural language transaction request.
-    :return: Transaction hash or failure message.
     """
     transaction_details = extract_transaction_details(transaction_prompt)
 
@@ -87,19 +93,19 @@ def mezo_agent_transaction_btc(transaction_prompt: str) -> str:
         return "❌ This function only supports BTC transactions."
 
     # ✅ Convert amount to Wei (BTC uses 18 decimals on Mezo Matsnet)
-    amount_wei = Web3.to_wei(amount, "ether")
+    amount_wei = web3_instance.to_wei(amount, "ether")
 
     # ✅ Check sender's BTC balance
-    sender_balance = Web3.eth.get_balance(sender_address)
-    sender_balance_btc = Web3.from_wei(sender_balance, "ether")
+    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.eth.get_transaction_count(sender_address)
-    gas_price = Web3.eth.gas_price
-    gas_limit = Web3.eth.estimate_gas({"to": recipient, "value": amount_wei, "from": sender_address})
+    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,
@@ -113,55 +119,8 @@ def mezo_agent_transaction_btc(transaction_prompt: str) -> str:
     try:
         # ✅ Sign and send the transaction
         signed_tx = account.sign_transaction(tx)
-        tx_hash = Web3.eth.send_raw_transaction(signed_tx.raw_transaction)
+        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)}"
-
-@tool
-def mezo_agent_transaction_musd(transaction_prompt: str) -> str:
-    """
-    Sends mUSD on Mezo Matsnet. Parses a transaction request and executes the transfer.
-
-    Example usage:
-    ```
-    mezo_agent_transaction_musd("Transfer 100 mUSD to 0xABC123...")
-    ```
-
-    :param transaction_prompt: Natural language transaction request.
-    :return: Transaction hash or failure message.
-    """
-    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 != "musd":
-        return "❌ This function only supports mUSD transactions."
-
-    # ✅ Convert amount to Wei (mUSD uses 18 decimals)
-    amount_musd_wei = int(amount * 10**18)
-    nonce = Web3.eth.get_transaction_count(sender_address)
-    gas_price = Web3.eth.gas_price
-
-    try:
-        # ✅ Build the transaction for mUSD transfer
-        txn = musd_contract.functions.transfer(recipient, amount_musd_wei).build_transaction({
-            "from": sender_address,
-            "nonce": nonce,
-            "gas": 50000,
-            "gasPrice": gas_price,
-        })
-
-        signed_txn = Web3.eth.account.sign_transaction(txn, PRIVATE_KEY)
-        tx_hash = Web3.eth.send_raw_transaction(signed_txn.raw_transaction)
-
-        return f"✅ mUSD Transaction Successful! Hash: {tx_hash.hex()}"
-
-    except Web3Exception as e:
-        return f"❌ mUSD Transaction Failed: {str(e)}"
+        return f"❌ BTC Transaction Failed: {str(e)}"

{mezoagent-0.1.1 → mezoagent-0.1.3}/setup.py

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