mezoAgent 0.5.0__py3-none-any.whl → 0.5.2__py3-none-any.whl

{mezoAgent-0.5.0.dist-info → mezoAgent-0.5.2.dist-info}/METADATA

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

{mezoAgent-0.5.0.dist-info → mezoAgent-0.5.2.dist-info}/RECORD

@@ -2,13 +2,13 @@ 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=LgMTntnzXrY7giGXy6JZqhPwPxxMpWWOgc2Ah6g3Y48,3017
-mezo_agent/parsing.py,sha256=6g0U4x8nQaOkwIPon1UlKlnVX6-S2Tf4LKqdY1ZGyiY,4000
+mezo_agent/parsing.py,sha256=v5AiLfZ8Rvm46gek7QX2OmeLFO8xhvQOH_yCc6GNlLA,3850
 mezo_agent/swap_musd_btc.py,sha256=Co-XcfK73spm94VC6qzGY0VYoGwsMDKKHM8ToGh2JxU,4289
-mezo_agent/token_balance_tool.py,sha256=GQsZnM6zwIdCL5fRISd_KAdc8RNVOTa0LYHPhPIdW1g,1304
+mezo_agent/token_balance_tool.py,sha256=MxvHzMLWbcwnJ34LDiefEz3_hHTKd6asNOEhdVs8VCY,1335
 mezo_agent/transaction.py,sha256=YpfWrkEaf0YGM_Kc4cFwlT9GmBGZkeJHWm0VGHs9Gks,4199
 mezo_agent/utils.py,sha256=Kdq3YRj6fOnAmDjI6vktvOvN0vAR8U7TgoTGqcA2KcI,973
 mezo_agent/data/new_router.json,sha256=A8U-NVfe1F-hDyR90_SuH8jDAxmzyyHWdJW62j9TZsc,26756
-mezoAgent-0.5.0.dist-info/METADATA,sha256=ZfVWkIcivB8L4OlUGNFpwhg-SQXFqHp8wkFg-Ugh4W0,273
-mezoAgent-0.5.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-mezoAgent-0.5.0.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
-mezoAgent-0.5.0.dist-info/RECORD,,
+mezoAgent-0.5.2.dist-info/METADATA,sha256=LLnjBY0JDYmk47bpPph-4hGrEE78T7ZW3hR8fP3S9ZU,273
+mezoAgent-0.5.2.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+mezoAgent-0.5.2.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
+mezoAgent-0.5.2.dist-info/RECORD,,

mezo_agent/parsing.py

@@ -85,15 +85,10 @@ Your output must be a valid JSON object with no additional text.
 )
 
 def extract_balance_details(prompt: str):
-    """
-    Parses a balance query and extracts the token symbol.
-    
-    :param prompt: User input asking for a token balance.
-    :return: Dictionary containing the token symbol or an error message.
-    """
+    formatted_prompt = balance_prompt_template.format(input=prompt)
+    response = llm.invoke(formatted_prompt)
+    print("LLM raw balance response:", response.content)
     try:
-        formatted_prompt = balance_prompt_template.format(input=prompt)
-        response = balance_output_parser.parse(formatted_prompt)
-        return response
+        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)}"

mezo_agent/token_balance_tool.py

@@ -1,5 +1,6 @@
 from langchain.tools import tool
-from mezo_agent.config import web3_instance, sender_address, extract_balance_details, ERC20_ABI
+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