mezoAgent 0.6.4__tar.gz → 0.6.5__tar.gz

{mezoagent-0.6.4 → mezoagent-0.6.5}/PKG-INFO

@@ -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 → mezoagent-0.6.5}/mezoAgent.egg-info/PKG-INFO

@@ -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 → mezoagent-0.6.5}/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)}"
+

{mezoagent-0.6.4 → mezoagent-0.6.5}/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)}"
+

mezoagent-0.6.5/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)}")

mezoagent-0.6.5/mezo_agent/utils.py

@@ -0,0 +1,39 @@
+from mezo_agent.config import web3_instance, query_graph
+from mezo_agent.token_utils import get_token_address_by_symbol  # ✅ Import from token_utils.py
+
+def get_token_price(token_symbol: str) -> str:
+    """
+    Fetches the price of a given token in USD and ETH from the Goldsky subgraph.
+    
+    :param token_symbol: The token symbol (e.g., 'MUSD', 'WBTC').
+    :return: A formatted string with the token price in USD and ETH.
+    """
+    if token_symbol.lower() == "btc":
+        token_symbol = "wtbtc"
+
+    try:
+        token_address = get_token_address_by_symbol(token_symbol)  # ✅ Now correctly imported
+    except Exception as e:
+        return f"❌ Token lookup error: {e}"
+
+    token_id = token_address.lower()
+    query = f'''
+    {{
+      token(id: "{token_id}") {{
+        id
+        decimals
+        derivedUSD
+        derivedETH
+      }}
+    }}
+    '''
+    try:
+        result = query_graph(query)
+        token_data = result["data"]["token"]
+        if token_data is None:
+            return f"❌ Price data for token {token_symbol.upper()} not found."
+        derivedUSD = token_data.get("derivedUSD", "N/A")
+        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}"

{mezoagent-0.6.4 → mezoagent-0.6.5}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="mezoAgent",
-    version="0.6.4",
+    version="0.6.5",
     packages=find_packages(),
     include_package_data=True,
     package_data={

mezoagent-0.6.4/mezo_agent/utils.py

@@ -1,70 +0,0 @@
-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)}")
-    
-
-def get_token_price(token_symbol: str) -> str:
-    """
-    Fetches the price of a given token in USD and ETH from the Goldsky subgraph.
-    
-    :param token_symbol: The token symbol (e.g., 'MUSD', 'WBTC').
-    :return: A formatted string with the token price in USD and ETH.
-    """
-    if token_symbol.lower() == "btc":
-        token_symbol = "wtbtc"
-
-    try:
-        token_address = get_token_address_by_symbol(token_symbol)
-    except Exception as e:
-        return f"❌ Token lookup error: {e}"
-
-    token_id = token_address.lower()
-    query = f'''
-    {{
-      token(id: "{token_id}") {{
-        id
-        decimals
-        derivedUSD
-        derivedETH
-      }}
-    }}
-    '''
-    try:
-        result = query_graph(query)
-        token_data = result["data"]["token"]
-        if token_data is None:
-            return f"❌ Price data for token {token_symbol.upper()} not found."
-        derivedUSD = token_data.get("derivedUSD", "N/A")
-        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}"