Skip to content

Commit ad03f08

Browse files
refactor: extract shared AI processing logic
1 parent c51c813 commit ad03f08

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

packages/slackBotFunction/app/slack/slack_events.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -16,14 +16,14 @@
1616
get_bot_token,
1717
get_logger,
1818
)
19-
from app.services.bedrock import query_bedrock
19+
2020
from app.services.dynamo import (
2121
delete_state_information,
2222
get_state_information,
2323
store_state_information,
2424
update_state_information,
2525
)
26-
from app.services.query_reformulator import reformulate_query
26+
2727
from app.services.slack import get_friendly_channel_name, post_error_message
2828
from app.utils.handler_utils import (
2929
conversation_key_and_root,
@@ -34,6 +34,9 @@
3434
strip_mentions,
3535
)
3636

37+
from app.services.ai_processor import process_ai_query
38+
39+
3740
logger = get_logger()
3841

3942

@@ -321,16 +324,13 @@ def process_slack_message(event: Dict[str, Any], event_id: str, client: WebClien
321324
client.chat_postMessage(**post_params)
322325
return
323326

324-
# Reformulate query for better RAG retrieval
325-
reformulated_query = reformulate_query(user_query)
326-
327-
# Check if we have an existing Bedrock conversation session
327+
# conversation continuity: reuse bedrock session across slack messages
328328
session_data = get_conversation_session_data(conversation_key)
329329
session_id = session_data.get("session_id") if session_data else None
330330

331-
# Query Bedrock Knowledge Base with conversation context
332-
kb_response = query_bedrock(reformulated_query, session_id)
333-
response_text = kb_response["output"]["text"]
331+
ai_response = process_ai_query(user_query, session_id)
332+
kb_response = ai_response["kb_response"]
333+
response_text = ai_response["text"]
334334

335335
# Post the answer (plain) to get message_ts
336336
post_params = {"channel": channel, "text": response_text}

0 commit comments

Comments
 (0)