mezoAgent 0.6.4__py3-none-any.whl → 0.6.5__py3-none-any.whl

{mezoAgent-0.6.4.dist-info → mezoAgent-0.6.5.dist-info}/METADATA

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

{mezoAgent-0.6.4.dist-info → mezoAgent-0.6.5.dist-info}/RECORD

@@ -4,13 +4,13 @@ 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/swap_musd_btc.py,sha256=Co-XcfK73spm94VC6qzGY0VYoGwsMDKKHM8ToGh2JxU,4289
-mezo_agent/token_balance_tool.py,sha256=DoNuHLr_b8P-g3AGYO3_D80Z6tDd9LR-kRdxcKAPN38,1448
-mezo_agent/token_price_tool.py,sha256=9oayEJm_-lS7SXEcvoKPyJOEVY7EiUO3t_IzKrutgx8,1042
-mezo_agent/token_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mezo_agent/token_balance_tool.py,sha256=n5xkHWGhkJdQZILN0fDe-n4NHgXSHTSloEHJfkLMsWE,1442
+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/utils.py,sha256=XRq--PRJbWlG9kpz99QHWu7mLpYY4MYblm38U1xK4YE,2381
+mezo_agent/utils.py,sha256=hsGtL_iXvUVGnY5aHPQ0v6Jejxd4P4qw0yOtS1omToU,1397
 mezo_agent/data/new_router.json,sha256=A8U-NVfe1F-hDyR90_SuH8jDAxmzyyHWdJW62j9TZsc,26756
-mezoAgent-0.6.4.dist-info/METADATA,sha256=Qoep4c59KYhXpQJIO3G_DKBuQ9zgueA2RIpIdqbUses,273
-mezoAgent-0.6.4.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-mezoAgent-0.6.4.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
-mezoAgent-0.6.4.dist-info/RECORD,,
+mezoAgent-0.6.5.dist-info/METADATA,sha256=aww4rbst9dQN-a_KYXzyKDoWTeJunsS26IagAdIs1mM,273
+mezoAgent-0.6.5.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+mezoAgent-0.6.5.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
+mezoAgent-0.6.5.dist-info/RECORD,,

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  # ✅ Import from new module
+from mezo_agent.token_utils import get_token_address_by_symbol  # ✅ Correct import
 
 @tool
 def mezo_agent_token_balance_tool(balance_prompt: str) -> str:
@@ -33,3 +33,4 @@ def mezo_agent_token_balance_tool(balance_prompt: str) -> str:
         return f"✅ {token_symbol.upper()} Balance: {balance}"
     except Exception as e:
         return f"❌ Failed to fetch balance: {str(e)}"
+

mezo_agent/token_price_tool.py

@@ -1,7 +1,7 @@
 from langchain.tools import tool
 from mezo_agent.config import query_graph
 from mezo_agent.parsing import extract_price_details
-from mezo_agent.token_utils import get_token_address_by_symbol  # ✅ Import from new module
+from mezo_agent.utils import get_token_price  # ✅ Correct import
 
 @tool
 def mezo_agent_token_price_tool(price_prompt: str) -> str:
@@ -11,8 +11,6 @@ def mezo_agent_token_price_tool(price_prompt: str) -> str:
     :param price_prompt: User query containing the token symbol.
     :return: Token price details.
     """
-    from mezo_agent.utils import get_token_price  # ✅ Import moved inside function
-
     details = extract_price_details(price_prompt)
 
     if isinstance(details, str):  # Handle extraction errors
@@ -27,3 +25,4 @@ def mezo_agent_token_price_tool(price_prompt: str) -> str:
         return get_token_price(token_symbol)
     except Exception as e:
         return f"❌ Failed to fetch token price: {str(e)}"
+

mezo_agent/token_utils.py

@@ -0,0 +1,32 @@
+from mezo_agent.config import query_graph, web3_instance
+
+def get_token_address_by_symbol(symbol: str) -> str:
+    """
+    Fetches the token contract address dynamically from the Goldsky subgraph.
+    :param symbol: Token symbol (e.g., 'MUSD', 'WBTC').
+    :return: Ethereum checksum address of the token.
+    """
+    if symbol.lower() == "btc":
+        symbol = "wtbtc"
+
+    query_all = '''
+    {
+      tokens(first: 100) {
+        id
+        symbol
+      }
+    }
+    '''
+    data = query_graph(query_all)
+    tokens = data.get("data", {}).get("tokens", [])
+
+    matching_tokens = [t for t in tokens if t["symbol"].lower() == symbol.lower()]
+    
+    if matching_tokens:
+        token_address = matching_tokens[0]["id"]
+        checksum_address = web3_instance.to_checksum_address(token_address)
+        print(f"✅ Token {symbol} found at address: {checksum_address}")  # Debugging
+        return checksum_address
+    else:
+        available_symbols = [t["symbol"] for t in tokens]
+        raise Exception(f"❌ Token '{symbol}' not found. Available: {', '.join(available_symbols)}")

mezo_agent/utils.py

@@ -1,36 +1,5 @@
 from mezo_agent.config import web3_instance, query_graph
-from mezo_agent.utils import get_token_address_by_symbol
-
-def get_token_address_by_symbol(symbol: str) -> str:
-    """
-    Fetches the token contract address dynamically from the Goldsky subgraph.
-    :param symbol: Token symbol (e.g., 'MUSD', 'WBTC').
-    :return: Ethereum checksum address of the token.
-    """
-    if symbol.lower() == "btc":
-        symbol = "wtbtc"
-
-    query_all = '''
-    {
-      tokens(first: 100) {
-        id
-        symbol
-      }
-    }
-    '''
-    data = query_graph(query_all)
-    tokens = data.get("data", {}).get("tokens", [])
-    matching_tokens = [t for t in tokens if t["symbol"].lower() == symbol.lower()]
-
-    if matching_tokens:
-        token_address = matching_tokens[0]["id"]
-        checksum_address = web3_instance.to_checksum_address(token_address)
-        print(f"✅ Token {symbol} found at address: {checksum_address}")  # Debugging
-        return checksum_address
-    else:
-        available_symbols = [t["symbol"] for t in tokens]
-        raise Exception(f"Token '{symbol}' not found. Available: {', '.join(available_symbols)}")
-    
+from mezo_agent.token_utils import get_token_address_by_symbol  # ✅ Import from token_utils.py
 
 def get_token_price(token_symbol: str) -> str:
     """
@@ -43,7 +12,7 @@ def get_token_price(token_symbol: str) -> str:
         token_symbol = "wtbtc"
 
     try:
-        token_address = get_token_address_by_symbol(token_symbol)
+        token_address = get_token_address_by_symbol(token_symbol)  # ✅ Now correctly imported
     except Exception as e:
         return f"❌ Token lookup error: {e}"
 
@@ -67,4 +36,4 @@ def get_token_price(token_symbol: str) -> str:
         derivedETH = token_data.get("derivedETH", "N/A")
         return f"✅ Price of {token_symbol.upper()}: {derivedUSD} USD, {derivedETH} ETH."
     except Exception as e:
-        return f"❌ Failed to get price data: {e}"
+        return f"❌ Failed to get price data: {e}"