mezoAgent 0.7.3__py3-none-any.whl → 0.8.0__py3-none-any.whl

{mezoAgent-0.7.3.dist-info → mezoAgent-0.8.0.dist-info}/METADATA

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

{mezoAgent-0.7.3.dist-info → mezoAgent-0.8.0.dist-info}/RECORD

@@ -1,10 +1,11 @@
-mezo_agent/__init__.py,sha256=5OyYPS3Ub5e9H0Jcp5lYg0pjP39DzJjuLwX8oyoqfZk,895
+mezo_agent/__init__.py,sha256=fxRf_Z-SEitzT7dp3V0DWmi3nO8yOrb26Avl3kib4QA,795
 mezo_agent/characters.py,sha256=wub7y9BrAAdAKWwcNAdgrvhRi2zHMaZNzTr2y3qtbsw,515
 mezo_agent/chat.py,sha256=ZtNurUDV7UzJjbFyU96DNGy37hO3gl0osn19mRu7ER8,859
 mezo_agent/config.py,sha256=6xFgk80eiBLTasKGW73Ne8hNhMtx8BbQ_Ef79YOyDlc,3233
-mezo_agent/parsing.py,sha256=dnS5Qw9V8qvSA_IPgn6nCUnyEfZP9TphO8DnH8oUuiM,5080
+mezo_agent/parsing.py,sha256=PeR1R6gqzM9VXt7OdWANDoL7GbzMMRjKLzuP10uugEs,5101
+mezo_agent/safe_mode_btc_tool.py,sha256=6e8P00Mk570qfPpEiUdMmfvhN3hnHLyOAKrYHu89v4g,3410
 mezo_agent/swap_musd_btc.py,sha256=Co-XcfK73spm94VC6qzGY0VYoGwsMDKKHM8ToGh2JxU,4289
-mezo_agent/token_balance_tool.py,sha256=n5xkHWGhkJdQZILN0fDe-n4NHgXSHTSloEHJfkLMsWE,1442
+mezo_agent/token_balance_tool.py,sha256=zOjDJdfYT6LuQzPuAxxNeee1srsepevE71fqboiDSWI,1422
 mezo_agent/token_price_tool.py,sha256=vD-ukq0D_4zmjIy9bZ9qL44nGctf-hWzkQuTd-0wy2U,930
 mezo_agent/token_utils.py,sha256=3NH_Z_vbZ0uupuuGjEbmwsbUkubluQbL-0xR9kia7nU,1118
 mezo_agent/transaction.py,sha256=YpfWrkEaf0YGM_Kc4cFwlT9GmBGZkeJHWm0VGHs9Gks,4199
@@ -12,7 +13,7 @@ mezo_agent/twitter_client.py,sha256=Va8ZQaK5lZjJUq3-bhIJWnDmX4Mmt4kWRW3aKSGbLq8,
 mezo_agent/twitter_manager.py,sha256=KvxiMOhBmQbLuLoVrAQ77DpAnFIdiDSztsKbZ_3-ar4,2363
 mezo_agent/utils.py,sha256=hsGtL_iXvUVGnY5aHPQ0v6Jejxd4P4qw0yOtS1omToU,1397
 mezo_agent/data/new_router.json,sha256=A8U-NVfe1F-hDyR90_SuH8jDAxmzyyHWdJW62j9TZsc,26756
-mezoAgent-0.7.3.dist-info/METADATA,sha256=v5lV-XUshSwN69O9nuPHOfpuGYT8zZ38ugGpI2J4Yjs,273
-mezoAgent-0.7.3.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-mezoAgent-0.7.3.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
-mezoAgent-0.7.3.dist-info/RECORD,,
+mezoAgent-0.8.0.dist-info/METADATA,sha256=xRxuyZCpP3HcmYhcdGFDnWIZdzSM-zX0c8qm-A_nn_g,273
+mezoAgent-0.8.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+mezoAgent-0.8.0.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
+mezoAgent-0.8.0.dist-info/RECORD,,

mezo_agent/__init__.py

@@ -2,14 +2,10 @@ 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
 from .characters import get_character_prompt
-from .token_balance_tool import mezo_agent_token_balance_tool
-from .token_price_tool import mezo_agent_token_price_tool
-from .token_utils import get_token_address_by_symbol
-
-# ✅ Move TwitterManager import inside a function to avoid circular import
-def get_twitter_manager():
-    from .twitter_manager import TwitterManager
-    return TwitterManager
+from .tools.token_balance_tool import mezo_agent_token_balance_tool
+from .tools.token_price_tool import mezo_agent_token_price_tool
+# Import the new Safe Mode BTC Transaction Tool
+from .tools.safe_mode_btc_tool import mezo_agent_safe_mode_btc_transaction
 
 __all__ = [
     "mezo_agent_transaction_btc",
@@ -18,7 +14,6 @@ __all__ = [
     "mezo_character_chat",
     "mezo_agent_token_balance_tool",
     "mezo_agent_token_price_tool",
-    "get_token_address_by_symbol",
-    "get_twitter_manager",  
+    "mezo_agent_safe_mode_btc_transaction",  # Expose new tool
     "get_character_prompt"
-]
+]

mezo_agent/parsing.py

@@ -3,7 +3,7 @@ from langchain.prompts import PromptTemplate
 from langchain_openai import ChatOpenAI
 from .config import OPENAI_API_KEY
 
-# Define the expected response schema
+# Define the expected response schema for extracting transaction details
 response_schemas = [
     ResponseSchema(name="amount", description="The amount of cryptocurrency to transfer."),
     ResponseSchema(name="currency", description="The cryptocurrency (BTC or mUSD)."),
@@ -64,7 +64,7 @@ def extract_swap_details(prompt: str):
     except Exception as e:
         return f"Failed to extract swap details: {str(e)}"
     
-#parsing for balance check tool 
+#Balance check parsing 
 
 balance_response_schema = [
     ResponseSchema(name="token_symbol", description="The token symbol to check the balance for (e.g., 'MUSD').")
@@ -93,7 +93,7 @@ def extract_balance_details(prompt: str):
     except Exception as e:
         return f"Failed to extract balance details: {str(e)}"
     
-#Price parsing tools
+#Price parsing 
 price_response_schema = [
     ResponseSchema(name="token_symbol", description="The token symbol to check the price for (e.g., 'LIMPETH').")
 ]

mezo_agent/safe_mode_btc_tool.py

@@ -0,0 +1,87 @@
+from langchain.tools import tool
+from web3.exceptions import Web3Exception
+import json
+from mezo_agent.config import web3_instance, sender_address, account
+from mezo_agent.parsing import extract_transaction_details
+from mezo_agent.utils import llm  # If you store your ChatOpenAI instance in utils.py
+
+@tool
+def mezo_agent_safe_mode_btc_transaction(prompt: str, confirm_before_sending: bool = True) -> str:
+    """
+    Sends BTC on Mezo Matsnet with a Human-in-the-Loop confirmation step.
+    1) Parses user prompt for transaction details.
+    2) Calls LLM to provide a verbose explanation of the transaction data.
+    3) Shows both raw data and LLM explanation to the user.
+    4) Prompts user for confirmation before broadcasting.
+    """
+
+    # 1) Parse transaction details
+    transaction_details = extract_transaction_details(prompt)
+    if isinstance(transaction_details, str):
+        return transaction_details  # Return parsing error if any
+
+    amount = float(transaction_details["amount"])
+    currency = transaction_details["currency"].lower()
+    recipient = transaction_details["recipient"]
+
+    if currency != "btc":
+        return "❌ This function only handles BTC transactions."
+
+    # 2) Convert to Wei (Mezo uses 18 decimals for BTC)
+    amount_wei = web3_instance.to_wei(amount, "ether")
+
+    # 3) Build transaction (do NOT send yet)
+    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_data = {
+        "to": recipient,
+        "value": amount_wei,
+        "gas": gas_limit,
+        "gasPrice": gas_price,
+        "nonce": nonce,
+        "chainId": 31611,  # Mezo Testnet Chain ID
+    }
+
+    # 4) Generate a LLM Explanation
+    analysis_prompt = f"""
+You are given the following transaction data on Mezo Matsnet:
+- Recipient: {recipient}
+- Amount (BTC): {amount}
+- Gas Price: {gas_price}
+- Gas Limit: {gas_limit}
+- Nonce: {nonce}
+- Chain ID: 31611 (Mezo Testnet)
+
+Explain these details in a user-friendly way, covering:
+1. The purpose of each field (nonce, gas, etc.).
+2. Potential fees (approximate cost) or any known risks.
+3. A short summary of what will happen if the user confirms.
+
+Use a concise and helpful tone.
+"""
+    analysis_response = llm.invoke(analysis_prompt)
+    explanation_text = analysis_response.content if analysis_response else "❌ LLM explanation unavailable."
+
+    # 5) Show user raw data & explanation, confirm
+    if confirm_before_sending:
+        print("\n--- BTC Transaction Data ---")
+        print(json.dumps(tx_data, indent=2))
+        print("----------------------------")
+
+        print("\n--- LLM Analysis ---")
+        print(explanation_text)
+        print("--------------------")
+
+        choice = input("Do you want to proceed with this BTC transaction? [y/n]: ").strip().lower()
+        if choice != 'y':
+            return "❌ Transaction aborted by user."
+
+    # 6) Sign & send
+    try:
+        signed_tx = account.sign_transaction(tx_data)
+        tx_hash = web3_instance.eth.send_raw_transaction(signed_tx.rawTransaction)
+        return f"✅ BTC transaction successful! Hash: {tx_hash.hex()}"
+    except Web3Exception as e:
+        return f"❌ Transaction failed: {str(e)}"

mezo_agent/token_balance_tool.py

@@ -1,7 +1,7 @@
 from langchain.tools import tool
 from mezo_agent.config import web3_instance, sender_address, ERC20_ABI
 from mezo_agent.parsing import extract_balance_details
-from mezo_agent.token_utils import get_token_address_by_symbol  # ✅ Correct import
+from mezo_agent.token_utils import get_token_address_by_symbol  
 
 @tool
 def mezo_agent_token_balance_tool(balance_prompt: str) -> str: