22from logging import Logger
33
44from agents import Runner
5- from slack_bolt import Say
5+ from slack_bolt import BoltAgent , Say
66from slack_sdk import WebClient
77
88from agent import CaseyDeps , casey_agent
2121CONTEXTUAL_EMOJIS = ["+1" , "raised_hands" , "rocket" , "tada" , "bulb" , "fire" ]
2222
2323
24- def handle_message (client : WebClient , event : dict , logger : Logger , say : Say ):
24+ def handle_message (client : WebClient , event : dict , agent : BoltAgent , logger : Logger , say : Say ):
2525 """Handle direct messages sent to Casey."""
2626 # Skip bot messages and message subtypes (edits, deletes, etc.)
2727 if event .get ("bot_id" ) or event .get ("subtype" ):
@@ -33,7 +33,6 @@ def handle_message(client: WebClient, event: dict, logger: Logger, say: Say):
3333
3434 try :
3535 channel_id = event ["channel" ]
36- team_id = event .get ("team" )
3736 text = event .get ("text" , "" )
3837 thread_ts = event .get ("thread_ts" ) or event ["ts" ]
3938 user_id = event ["user" ]
@@ -79,12 +78,7 @@ def handle_message(client: WebClient, event: dict, logger: Logger, say: Say):
7978 result = Runner .run_sync (casey_agent , input = input_items , context = deps )
8079
8180 # Stream response in thread with feedback buttons
82- streamer = client .chat_stream (
83- channel = channel_id ,
84- recipient_team_id = team_id ,
85- recipient_user_id = user_id ,
86- thread_ts = thread_ts ,
87- )
81+ streamer = agent .chat_stream ()
8882 streamer .append (markdown_text = result .final_output )
8983 feedback_blocks = create_feedback_block ()
9084 streamer .stop (blocks = feedback_blocks )
0 commit comments