mezoAgent 0.5.4__py3-none-any.whl → 0.6.1__py3-none-any.whl

{mezoAgent-0.5.4.dist-info → mezoAgent-0.6.1.dist-info}/METADATA

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

{mezoAgent-0.5.4.dist-info → mezoAgent-0.6.1.dist-info}/RECORD

@@ -2,13 +2,14 @@ mezo_agent/__init__.py,sha256=naG5ZIXKDWTRlqdx2oi1zJcRZpJBO9ugVkW0MaPkZuk,497
 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=v5AiLfZ8Rvm46gek7QX2OmeLFO8xhvQOH_yCc6GNlLA,3850
+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=OvdpEr3Vy_LU1-dg2hOt4IJR6B8Rmql_NQCuVy3cqqg,1459
+mezo_agent/token_balance_tool.py,sha256=vWWvZxJWPFX05ENIcr5mCwW6r8OVzV21iDrXRKkRBRc,1463
+mezo_agent/token_price_tool.py,sha256=MAJsRv0xJcmpIatxoXI6_hoBPVPXcJMEyzSoX-p9l-Q,916
 mezo_agent/transaction.py,sha256=YpfWrkEaf0YGM_Kc4cFwlT9GmBGZkeJHWm0VGHs9Gks,4199
-mezo_agent/utils.py,sha256=SnVRjAdrluqiiTjPNi0VhMKGGRvmYZlPXE5B4Bknq8c,1106
+mezo_agent/utils.py,sha256=XRq--PRJbWlG9kpz99QHWu7mLpYY4MYblm38U1xK4YE,2381
 mezo_agent/data/new_router.json,sha256=A8U-NVfe1F-hDyR90_SuH8jDAxmzyyHWdJW62j9TZsc,26756
-mezoAgent-0.5.4.dist-info/METADATA,sha256=djvkoN-W3_Gl04MEOu62OXfP9hxExBwA4SadY8iKkeQ,273
-mezoAgent-0.5.4.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-mezoAgent-0.5.4.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
-mezoAgent-0.5.4.dist-info/RECORD,,
+mezoAgent-0.6.1.dist-info/METADATA,sha256=g1ynqIcb6yxweH0v-kFIBtS9nFgnsUVI7jtBj_bGNec,273
+mezoAgent-0.6.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+mezoAgent-0.6.1.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
+mezoAgent-0.6.1.dist-info/RECORD,,

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)}"

mezo_agent/token_balance_tool.py

@@ -1,7 +1,6 @@
 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.utils import get_token_address_by_symbol
 
 @tool
 def mezo_agent_token_balance_tool(balance_prompt: str) -> str:
@@ -11,6 +10,7 @@ def mezo_agent_token_balance_tool(balance_prompt: str) -> str:
     :param balance_prompt: User query containing the token symbol.
     :return: Token balance for the sender's address.
     """
+    from mezo_agent.utils import get_token_address_by_symbol
     details = extract_balance_details(balance_prompt)
 
     if isinstance(details, str):  # Handle extraction errors

mezo_agent/token_price_tool.py

@@ -0,0 +1,28 @@
+from langchain.tools import tool
+from mezo_agent.config import query_graph
+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.
+    
+    """
+    from mezo_agent.utils import get_token_price
+    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)}"

mezo_agent/utils.py

@@ -1,4 +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:
     """
@@ -28,4 +29,42 @@ def get_token_address_by_symbol(symbol: str) -> str:
         return checksum_address
     else:
         available_symbols = [t["symbol"] for t in tokens]
-        raise Exception(f"Token '{symbol}' not found. Available: {', '.join(available_symbols)}")
+        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}"