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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mezoAgent
-Version: 0.7.2
+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.2.dist-info → mezoAgent-0.8.0.dist-info}/RECORD

@@ -1,18 +1,19 @@
-mezo_agent/__init__.py,sha256=1UatqstSlOF8pU5-fu675ONGteB6INqyE74xMi8S1o4,752
+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
 mezo_agent/twitter_client.py,sha256=Va8ZQaK5lZjJUq3-bhIJWnDmX4Mmt4kWRW3aKSGbLq8,3248
-mezo_agent/twitter_manager.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+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.2.dist-info/METADATA,sha256=UmaWJ9SG3FkEPtKYQ2AYlAu28Dm8vqyyENlVLanIsYM,273
-mezoAgent-0.7.2.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-mezoAgent-0.7.2.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
-mezoAgent-0.7.2.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,10 +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
-from .twitter_manager import 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",
@@ -14,7 +14,6 @@ __all__ = [
     "mezo_character_chat",
     "mezo_agent_token_balance_tool",
     "mezo_agent_token_price_tool",
-    "get_token_address_by_symbol",
-    "TwitterManager",
+    "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:

mezo_agent/twitter_manager.py

@@ -0,0 +1,63 @@
+import os
+import json
+
+class TwitterManager:
+    """
+    Manages Twitter clients for MezoAgent characters.
+    """
+
+    def __init__(self, config_file="twitter_config.json"):
+        """
+        Initializes the Twitter Manager.
+
+        :param config_file: The JSON file that stores character Twitter credentials.
+        """
+        self.config_file = config_file
+        self.characters = {}
+        self.load_characters()
+
+    def load_characters(self):
+        """
+        Loads characters and their Twitter credentials from a JSON config file.
+        """
+        if os.path.exists(self.config_file):
+            with open(self.config_file, "r") as f:
+                self.characters = json.load(f)
+
+    def save_characters(self):
+        """
+        Saves character Twitter credentials to a JSON config file.
+        """
+        with open(self.config_file, "w") as f:
+            json.dump(self.characters, f, indent=4)
+
+    def add_character(self, character_name: str, api_key: str, api_secret: str, access_token: str, access_secret: str, personality: str):
+        """
+        Registers a character with Twitter API credentials and starts their Twitter client.
+
+        :param character_name: Name of the character.
+        :param api_key: Twitter API Key.
+        :param api_secret: Twitter API Secret Key.
+        :param access_token: Twitter Access Token.
+        :param access_secret: Twitter Access Token Secret.
+        :param personality: Custom personality description for AI-generated tweets.
+        """
+        from mezo_agent.twitter_client import TwitterClient  # ✅ Move import inside the function
+
+        if character_name in self.characters:
+            print(f"❌ Character '{character_name}' already has a Twitter client.")
+            return
+
+        self.characters[character_name] = {
+            "api_key": api_key,
+            "api_secret": api_secret,
+            "access_token": access_token,
+            "access_secret": access_secret,
+            "personality": personality
+        }
+        self.save_characters()
+        print(f"✅ Character '{character_name}' registered for Twitter with personality!")
+
+        # Start the Twitter client with personality
+        TwitterClient(character_name, api_key, api_secret, access_token, access_secret, personality)
+