refactor(slack): ask user for necessary ids

Author: idiotWu

npiai/tools/slack/__test__/send_and_wait.py

@@ -1,13 +1,16 @@
 import asyncio
 
 from npiai import agent_wrapper
-from npiai.tools.slack import Slack
+from npiai.tools import Slack
+from npiai.hitl_handler import ConsoleHandler
 
 
 async def main():
     async with agent_wrapper(Slack()) as slack:
+        slack.use_hitl(ConsoleHandler())
+
         return await slack.chat(
-            'Send a direct message to Dolphin (user id: "U071QBWBVCJ") asking if he is doing well, and wait for his reply.'
+            'Send a direct message to Dolphin asking if he is doing well, and wait for his reply.'
         )
 
 

npiai/tools/slack/app.py

@@ -1,9 +1,11 @@
 import asyncio
 import json
 import os
+from typing import Literal
 
 from slack_sdk.web.async_client import AsyncWebClient, AsyncSlackResponse
 
+from npiai.context import Context
 from npiai.utils import logger
 from npiai import FunctionTool, function
 from npiai.error.auth import UnauthorizedError
@@ -68,6 +70,20 @@ def _get_messages_from_response(self, response: AsyncSlackResponse):
 
         return sorted(messages, key=lambda x: float(x['thread_id']))
 
+    @function
+    async def ask_for_id(self, ctx: Context, name: Literal['user', 'channel', 'thread']):
+        """
+        Ask the user to provide recipient's user id, channel id, or thread id.
+        Args:
+            ctx: NPi context.
+            name: The type of id to ask for.
+        """
+        return await self.hitl.input(
+            ctx,
+            self.name,
+            f"Please provide recipient's {name} ID to send messages to",
+        )
+
     @function
     async def list_channels(self):
         """Get a list of all Slack channels"""