feat: use 'agent' argument and 'agent.chat_stream()'

Author: mwbrooks

listeners/assistant/message.py

@@ -2,7 +2,7 @@
 from logging import Logger
 
 from openai.types.responses import ResponseInputParam
-from slack_bolt import BoltContext, Say, SetStatus
+from slack_bolt import BoltAgent, BoltContext, Say, SetStatus
 from slack_sdk import WebClient
 from slack_sdk.models.messages.chunk import (
     MarkdownTextChunk,
@@ -15,6 +15,7 @@
 
 
 def message(
+    agent: BoltAgent,
     client: WebClient,
     context: BoltContext,
     logger: Logger,
@@ -27,6 +28,7 @@ def message(
     Handles when users send messages or select a prompt in an assistant thread and generate AI responses:
 
     Args:
+        agent: BoltAgent for making API calls
         client: Slack WebClient for making API calls
         context: Bolt context containing channel and thread information
         logger: Logger instance for error tracking
@@ -56,13 +58,7 @@ def message(
 
             time.sleep(4)
 
-            streamer = client.chat_stream(
-                channel=channel_id,
-                recipient_team_id=team_id,
-                recipient_user_id=user_id,
-                thread_ts=thread_ts,
-                task_display_mode="plan",
-            )
+            streamer = agent.chat_stream(task_display_mode="plan")
             streamer.append(
                 chunks=[
                     MarkdownTextChunk(
@@ -140,13 +136,7 @@ def message(
                 ],
             )
 
-            streamer = client.chat_stream(
-                channel=channel_id,
-                recipient_team_id=team_id,
-                recipient_user_id=user_id,
-                thread_ts=thread_ts,
-                task_display_mode="timeline",
-            )
+            streamer = agent.chat_stream(task_display_mode="timeline")
             prompts: ResponseInputParam = [
                 {
                     "role": "user",

listeners/events/app_mentioned.py

@@ -1,19 +1,20 @@
 from logging import Logger
 
 from openai.types.responses import ResponseInputParam
-from slack_bolt import Say
+from slack_bolt import BoltAgent, Say
 from slack_sdk import WebClient
 
 from agent.llm_caller import call_llm
 from listeners.views.feedback_block import create_feedback_block
 
 
-def app_mentioned_callback(client: WebClient, event: dict, logger: Logger, say: Say):
+def app_mentioned_callback(agent: BoltAgent, client: WebClient, event: dict, logger: Logger, say: Say):
     """
     Handles the event when the app is mentioned in a Slack conversation
     and generates an AI response.
 
     Args:
+        agent: BoltAgent for making API calls
         client: Slack WebClient for making API calls
         event: Event payload containing mention details (channel, user, text, etc.)
         logger: Logger instance for error tracking
@@ -39,12 +40,7 @@ def app_mentioned_callback(client: WebClient, event: dict, logger: Logger, say:
             ],
         )
 
-        streamer = client.chat_stream(
-            channel=channel_id,
-            recipient_team_id=team_id,
-            recipient_user_id=user_id,
-            thread_ts=thread_ts,
-        )
+        streamer = agent.chat_stream()
         prompts: ResponseInputParam = [
             {
                 "role": "user",