mezoAgent 0.5.1__py3-none-any.whl → 0.5.3__py3-none-any.whl

{mezoAgent-0.5.1.dist-info → mezoAgent-0.5.3.dist-info}/METADATA

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

{mezoAgent-0.5.1.dist-info → mezoAgent-0.5.3.dist-info}/RECORD

@@ -1,14 +1,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=LgMTntnzXrY7giGXy6JZqhPwPxxMpWWOgc2Ah6g3Y48,3017
-mezo_agent/parsing.py,sha256=6g0U4x8nQaOkwIPon1UlKlnVX6-S2Tf4LKqdY1ZGyiY,4000
+mezo_agent/config.py,sha256=6xFgk80eiBLTasKGW73Ne8hNhMtx8BbQ_Ef79YOyDlc,3233
+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=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.1.dist-info/METADATA,sha256=Kc19v_98AfSbocB62LxJTdUUbIorkE05b5uMyAc-3vs,273
-mezoAgent-0.5.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-mezoAgent-0.5.1.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
-mezoAgent-0.5.1.dist-info/RECORD,,
+mezoAgent-0.5.3.dist-info/METADATA,sha256=Q-byzl5tfJ0z1XgSz19UBLNtUPcmL-8feCiD2FZDG3s,273
+mezoAgent-0.5.3.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+mezoAgent-0.5.3.dist-info/top_level.txt,sha256=rrAci_NyTR9z6w_BrQhQrAhzMW_A0NYhVa0x2USl0nQ,11
+mezoAgent-0.5.3.dist-info/RECORD,,

mezo_agent/config.py

@@ -37,13 +37,17 @@ sender_address = account.address
 # mUSD Contract Setup using approve/allowance ABI
 MUSD_ADDRESS = "0x637e22A1EBbca50EA2d34027c238317fD10003eB"
 ERC20_ABI = json.loads(
-    '[{"constant": false, "inputs": [{"name": "recipient", "type": "address"}, {"name": "amount", "type": "uint256"}],'
+    '[{"constant": true, "inputs": [{"name": "owner", "type": "address"}], '
+    '"name": "balanceOf", "outputs": [{"name": "balance", "type": "uint256"}], '
+    '"stateMutability": "view", "type": "function"},'
+    '{"constant": false, "inputs": [{"name": "recipient", "type": "address"}, {"name": "amount", "type": "uint256"}],'
     '"name": "transfer", "outputs": [{"name": "", "type": "bool"}], "stateMutability": "nonpayable", "type": "function"},'
     '{"constant": false, "inputs": [{"name": "spender", "type": "address"}, {"name": "amount", "type": "uint256"}],'
     '"name": "approve", "outputs": [{"name": "", "type": "bool"}], "stateMutability": "nonpayable", "type": "function"},'
     '{"constant": true, "inputs": [{"name": "owner", "type": "address"}, {"name": "spender", "type": "address"}],'
     '"name": "allowance", "outputs": [{"name": "remaining", "type": "uint256"}], "stateMutability": "view", "type": "function"}]'
 )
+
 musd_contract = web3_instance.eth.contract(address=MUSD_ADDRESS, abi=ERC20_ABI)
 
 #Query graph for token info

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