mezoAgent 0.5.4__tar.gz → 0.6.0__tar.gz

{mezoagent-0.5.4 → mezoagent-0.6.0}/PKG-INFO

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

{mezoagent-0.5.4 → mezoagent-0.6.0}/mezoAgent.egg-info/PKG-INFO

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

{mezoagent-0.5.4 → mezoagent-0.6.0}/mezoAgent.egg-info/SOURCES.txt

@@ -13,6 +13,7 @@ mezo_agent/config.py
 mezo_agent/parsing.py
 mezo_agent/swap_musd_btc.py
 mezo_agent/token_balance_tool.py
+mezo_agent/token_price_tool.py
 mezo_agent/transaction.py
 mezo_agent/utils.py
 mezo_agent/data/new_router.json

{mezoagent-0.5.4 → mezoagent-0.6.0}/mezo_agent/parsing.py

@@ -91,4 +91,37 @@ def extract_balance_details(prompt: str):
     try:
         return balance_output_parser.parse(response.content)
     except Exception as e:
-        return f"Failed to extract balance details: {str(e)}"
+        return f"Failed to extract balance details: {str(e)}"
+    
+#Price parsing tools
+price_response_schema = [
+    ResponseSchema(name="token_symbol", description="The token symbol to check the price for (e.g., 'LIMPETH').")
+]
+price_output_parser = StructuredOutputParser.from_response_schemas(price_response_schema)
+
+price_prompt_template = PromptTemplate(
+    template="""Output a JSON object with the following key:
+- token_symbol: (string) The token symbol for which to check the price (e.g., "LIMPETH").
+
+Extract this detail from the following request:
+{input}
+
+Your output must be a valid JSON object with no additional text.
+""",
+    input_variables=["input"],
+)
+
+def extract_price_details(prompt: str):
+    """
+    Parses a price query and extracts the token symbol.
+    
+    :param prompt: User input asking for a token price.
+    :return: Dictionary containing the token symbol or an error message.
+    """
+    formatted_prompt = price_prompt_template.format(input=prompt)
+    response = llm.invoke(formatted_prompt)
+    print("LLM raw price response:", response.content)  # Debugging
+    try:
+        return price_output_parser.parse(response.content)
+    except Exception as e:
+        return f"❌ Failed to extract price details: {str(e)}"

mezoagent-0.6.0/mezo_agent/token_price_tool.py

@@ -0,0 +1,27 @@
+from langchain.tools import tool
+from mezo_agent.config import query_graph
+from mezo_agent.utils import get_token_address_by_symbol, get_token_price
+from mezo_agent.parsing import extract_price_details
+
+@tool
+def mezo_agent_token_price_tool(price_prompt: str) -> str:
+    """
+    Queries the price of a specified token in USD and ETH on Mezo Matsnet.
+
+    :param price_prompt: User query containing the token symbol.
+    :return: Token price details.
+    """
+    details = extract_price_details(price_prompt)
+
+    if isinstance(details, str):  # Handle extraction errors
+        return details
+
+    token_symbol = details.get("token_symbol")
+    if not token_symbol:
+        return "❌ Could not extract token symbol for price query."
+
+    try:
+        # Get the token price
+        return get_token_price(token_symbol)
+    except Exception as e:
+        return f"❌ Failed to fetch token price: {str(e)}"

mezoagent-0.6.0/mezo_agent/utils.py

@@ -0,0 +1,70 @@
+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}"

{mezoagent-0.5.4 → mezoagent-0.6.0}/setup.py

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

mezoagent-0.5.4/mezo_agent/utils.py

@@ -1,31 +0,0 @@
-from mezo_agent.config import web3_instance, query_graph
-
-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)}")