test and prompt fixes

agentcoinorg · May 31, 2024 · 3c7c422 · 3c7c422
1 parent c1ed593
commit 3c7c422
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/autotx/agents/SwapTokensAgent.py b/autotx/agents/SwapTokensAgent.py
@@ -90,7 +90,12 @@ def swap(autotx: AutoTx, token_to_sell: str, token_to_buy: str) -> list[models.T
         )
 
     if len(sell_parts) == 2 and len(buy_parts) == 2:
-        raise InvalidInput(f"Invalid input: \"{token_to_sell} to {token_to_buy}\". Only one token amount should be provided. IMPORTANT: Take another look at the user's goal, and try again.")
+        sell_amount = Decimal(sell_parts[0])
+        buy_amount = Decimal(buy_parts[0])
+        sell_token = sell_parts[1]
+        buy_token = buy_parts[1]
+
+        raise InvalidInput(f"Invalid input: \"{token_to_sell} to {token_to_buy}\". Only one token amount should be provided. Choose between '{sell_amount} {sell_token} to {buy_token}' or '{sell_token} to {buy_amount} {buy_token}'.")
 
     if len(sell_parts) < 2 and len(buy_parts) < 2:
         raise InvalidInput(f"Invalid input: \"{token_to_sell} to {token_to_buy}\". Token amount is missing. Only one token amount should be provided.")

diff --git a/autotx/tests/agents/token/research/test_advanced.py b/autotx/tests/agents/token/research/test_advanced.py
@@ -19,7 +19,7 @@ def test_research_and_swap_many_tokens_subjective_simple(configuration, auto_tx)
     gaming_token_address = get_top_token_addresses_by_market_cap("gaming", "MAINNET", 1, auto_tx)[0]
     gaming_token_balance_in_safe = manager.balance_of(gaming_token_address)
 
-    ai_token_address = get_top_token_addresses_by_market_cap("ai-themed-coins", "MAINNET", 1, auto_tx)[0]
+    ai_token_address = get_top_token_addresses_by_market_cap("artificial-intelligence", "MAINNET", 1, auto_tx)[0]
     ai_token_balance_in_safe = manager.balance_of(ai_token_address)
 
     meme_token_address = get_top_token_addresses_by_market_cap("meme-token", "MAINNET", 1, auto_tx)[0]