Skip to content

refactor: use say_stream helper instead of client.chat_stream #82

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
zimeg wants to merge 1 commit into
base: main
Choose a base branch
from
Open

refactor: use say_stream helper instead of client.chat_stream #82

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 7 additions & 23 deletions listeners/assistant/message.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@
from logging import Logger

from openai.types.responses import ResponseInputParam
from slack_bolt import BoltContext, Say, SetStatus
from slack_sdk import WebClient
from slack_bolt import Say, SetStatus
from slack_bolt.context.say_stream import SayStream
from slack_sdk.models.messages.chunk import (
MarkdownTextChunk,
PlanUpdateChunk,
Expand All @@ -15,31 +15,23 @@


def message(
client: WebClient,
context: BoltContext,
logger: Logger,
message: dict,
payload: dict,
say: Say,
say_stream: SayStream,
set_status: SetStatus,
):
"""
Handles when users send messages or select a prompt in an assistant thread and generate AI responses:

Args:
client: Slack WebClient for making API calls
context: Bolt context containing channel and thread information
logger: Logger instance for error tracking
payload: Event payload with message details (channel, user, text, etc.)
message: The message event payload
say: Function to send messages to the thread
say_stream: Function to start a chat stream in the thread
set_status: Function to update the assistant's status
"""
try:
channel_id = payload["channel"]
team_id = context.team_id
thread_ts = payload["thread_ts"]
user_id = context.user_id

# The first example shows a message with thinking steps that has different
# chunks to construct and update a plan alongside text outputs.
if message["text"] == "Wonder a few deep thoughts.":
Expand All @@ -56,11 +48,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,
streamer = say_stream(
task_display_mode="plan",
)
streamer.append(
Expand Down Expand Up @@ -140,11 +128,7 @@ def message(
],
)

streamer = client.chat_stream(
channel=channel_id,
recipient_team_id=team_id,
recipient_user_id=user_id,
thread_ts=thread_ts,
streamer = say_stream(
task_display_mode="timeline",
)
prompts: ResponseInputParam = [
Expand Down