Waiting function for NFT agent (#624)

prediction_market_agent/agents/microchain_agent/nft_treasury_game/app_nft_treasury_game.py

@@ -36,6 +36,7 @@
     BroadcastPublicMessageToHumans,
     ReceiveMessage,
     SendPaidMessageToAnotherAgent,
+    Wait,
 )
 from prediction_market_agent.db.agent_communication import (
     fetch_count_unprocessed_transactions,
@@ -144,6 +145,8 @@ def customized_chat_message(
             icon = "🧠"
         case Stop.__name__:
             icon = "😴"
+        case Wait.__name__:
+            icon = "⏳"
         case ReceiveMessage.__name__:
             icon = "👤"
         case BroadcastPublicMessageToHumans.__name__:
@@ -177,6 +180,7 @@ def customized_chat_message(
             Stop.__name__,
             BroadcastPublicMessageToHumans.__name__,
             SendPaidMessageToAnotherAgent.__name__,
+            Wait.__name__,
         ):
             st.markdown(parsed_function_output_body)
 

prediction_market_agent/agents/microchain_agent/nft_treasury_game/messages_functions.py

@@ -1,3 +1,5 @@
+import time
+
 from microchain import Function
 from prediction_market_agent_tooling.config import APIKeys as APIKeys_PMAT
 from prediction_market_agent_tooling.gtypes import xdai_type
@@ -95,8 +97,23 @@ def __call__(self) -> str:
         return parse_message_for_agent(message=popped_message)
 
 
+class Wait(Function):
+    @property
+    def description(self) -> str:
+        return f"""Use {Wait.__name__} to wait for given amount of time in seconds and the reason for it. You can use this for example to wait for a while before checking for new messages."""
+
+    @property
+    def example_args(self) -> list[str]:
+        return ["10", "Waiting for responses."]
+
+    def __call__(self, wait: int, reason: str) -> str:
+        time.sleep(wait)
+        return f"Waited for {wait} seconds to {reason}."
+
+
 MESSAGES_FUNCTIONS: list[type[Function]] = [
     BroadcastPublicMessageToHumans,
     SendPaidMessageToAnotherAgent,
     ReceiveMessage,
+    Wait,
 ]