|
16 | 16 | get_bot_token, |
17 | 17 | get_logger, |
18 | 18 | ) |
19 | | -from app.services.bedrock import query_bedrock |
| 19 | + |
20 | 20 | from app.services.dynamo import ( |
21 | 21 | delete_state_information, |
22 | 22 | get_state_information, |
23 | 23 | store_state_information, |
24 | 24 | update_state_information, |
25 | 25 | ) |
26 | | -from app.services.query_reformulator import reformulate_query |
| 26 | + |
27 | 27 | from app.services.slack import get_friendly_channel_name, post_error_message |
28 | 28 | from app.utils.handler_utils import ( |
29 | 29 | conversation_key_and_root, |
|
34 | 34 | strip_mentions, |
35 | 35 | ) |
36 | 36 |
|
| 37 | +from app.services.ai_processor import process_ai_query |
| 38 | + |
| 39 | + |
37 | 40 | logger = get_logger() |
38 | 41 |
|
39 | 42 |
|
@@ -321,16 +324,13 @@ def process_slack_message(event: Dict[str, Any], event_id: str, client: WebClien |
321 | 324 | client.chat_postMessage(**post_params) |
322 | 325 | return |
323 | 326 |
|
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 |
328 | 328 | session_data = get_conversation_session_data(conversation_key) |
329 | 329 | session_id = session_data.get("session_id") if session_data else None |
330 | 330 |
|
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"] |
334 | 334 |
|
335 | 335 | # Post the answer (plain) to get message_ts |
336 | 336 | post_params = {"channel": channel, "text": response_text} |
|
0 commit comments