refactor: replace assistant_threads_setStatus method with assistant class set_status (#12)

Co-authored-by: Ale Mercado <maria.mercado@slack-corp.com>
Co-authored-by: Michael Brooks <mbrooks@slack-corp.com>

Author: 3 people

listeners/assistant/assistant.py

@@ -1,9 +1,10 @@
 import logging
-from typing import List, Dict
-from slack_bolt import Assistant, BoltContext, Say, SetSuggestedPrompts
+from typing import Dict, List
+
+from slack_bolt import Assistant, BoltContext, Say, SetStatus, SetSuggestedPrompts
 from slack_bolt.context.get_thread_context import GetThreadContext
 from slack_sdk import WebClient
-from slack_sdk.models.blocks import Block, ContextActionsBlock, FeedbackButtonsElement, FeedbackButtonObject
+from slack_sdk.models.blocks import Block, ContextActionsBlock, FeedbackButtonObject, FeedbackButtonsElement
 
 from ..llm_caller import call_llm
 
@@ -83,23 +84,27 @@ def start_assistant_thread(
 # This listener is invoked when the human user sends a reply in the assistant thread
 @assistant.user_message
 def respond_in_assistant_thread(
-    payload: dict,
-    logger: logging.Logger,
-    context: BoltContext,
     client: WebClient,
+    context: BoltContext,
+    logger: logging.Logger,
+    payload: dict,
     say: Say,
+    set_status: SetStatus,
 ):
     try:
         channel_id = payload["channel"]
         thread_ts = payload["thread_ts"]
 
-        loading_messages = [
-            "Teaching the hamsters to type faster…",
-            "Untangling the internet cables…",
-            "Consulting the office goldfish…",
-            "Polishing up the response just for you…",
-            "Convincing the AI to stop overthinking…",
-        ]
+        set_status(
+            status="Drafting...",
+            loading_messages=[
+                "Teaching the hamsters to type faster…",
+                "Untangling the internet cables…",
+                "Consulting the office goldfish…",
+                "Polishing up the response just for you…",
+                "Convincing the AI to stop overthinking…",
+            ],
+        )
 
         replies = client.conversations_replies(
             channel=context.channel_id,
@@ -114,10 +119,6 @@ def respond_in_assistant_thread(
 
         returned_message = call_llm(messages_in_thread)
 
-        client.assistant_threads_setStatus(
-            channel_id=channel_id, thread_ts=thread_ts, status="Bolt is typing", loading_messages=loading_messages
-        )
-
         stream_response = client.chat_startStream(
             channel=channel_id,
             thread_ts=thread_ts,

requirements.txt

@@ -1,5 +1,5 @@
-slack-sdk==3.36.0.dev3
-slack-bolt>=1.21,<2
+slack-sdk==3.36.0.dev5
+slack-bolt==1.26.0.dev2
 
 # If you use a different LLM vendor, replace this dependency
 openai