use different table

Author: anthony-nhs

packages/cdk/resources/DatabaseTables.ts

@@ -8,6 +8,7 @@ export interface TablesProps {
 
 export class DatabaseTables extends Construct {
   public readonly slackBotStateTable: DynamoDbTable
+  public readonly feedbackTable: DynamoDbTable
 
   constructor(scope: Construct, id: string, props: TablesProps) {
     super(scope, id)
@@ -24,5 +25,19 @@ export class DatabaseTables extends Construct {
       },
       timeToLiveAttribute: "ttl"
     })
+
+    this.feedbackTable = new DynamoDbTable(this, "FeedbackTable", {
+      tableName: `${props.stackName}-Feedback`,
+      partitionKey: {
+        name: "pk",
+        type: AttributeType.STRING
+      },
+      sortKey: {
+        name: "sk",
+        type: AttributeType.STRING
+      },
+      timeToLiveAttribute: "ttl"
+    })
+
   }
 }

packages/cdk/resources/Functions.ts

@@ -34,6 +34,7 @@ export interface FunctionsProps {
   readonly slackBotTokenSecret: Secret
   readonly slackBotSigningSecret: Secret
   readonly slackBotStateTable: TableV2
+  readonly feedbackTable: TableV2
   readonly promptName: string
 }
 
@@ -75,6 +76,7 @@ export class Functions extends Construct {
         "GUARD_RAIL_ID": props.guardrailId,
         "GUARD_RAIL_VERSION": props.guardrailVersion,
         "SLACK_BOT_STATE_TABLE": props.slackBotStateTable.tableName,
+        "FEEDBACK_TABLE": props.feedbackTable.tableName,
         "QUERY_REFORMULATION_PROMPT_NAME": props.promptName,
         "QUERY_REFORMULATION_PROMPT_VERSION": QUERY_REFORMULATION_PROMPT_VERSION
       }

packages/cdk/stacks/EpsAssistMeStack.ts

@@ -130,6 +130,7 @@ export class EpsAssistMeStack extends Stack {
       slackBotTokenSecret: secrets.slackBotTokenSecret,
       slackBotSigningSecret: secrets.slackBotSigningSecret,
       slackBotStateTable: tables.slackBotStateTable.table,
+      feedbackTable: tables.feedbackTable.table,
       promptName: bedrockPromptResources.queryReformulationPrompt.promptName
     })
 

packages/slackBotFunction/app/core/config.py

@@ -15,7 +15,8 @@
 
 # DynamoDB table for deduplication and session storage
 dynamodb = boto3.resource("dynamodb")
-table = dynamodb.Table(os.environ["SLACK_BOT_STATE_TABLE"])
+slack_bot_state_table = dynamodb.Table(os.environ["SLACK_BOT_STATE_TABLE"])
+feedback_table = dynamodb.Table(os.environ["FEEDBACK_TABLE"])
 
 # get Slack credentials from Parameter Store
 bot_token_parameter = os.environ["SLACK_BOT_TOKEN_PARAMETER"]

packages/slackBotFunction/app/slack/slack_events.py

@@ -10,7 +10,8 @@
 from botocore.exceptions import ClientError
 from slack_sdk import WebClient
 from app.core.config import (
-    table,
+    slack_bot_state_table,
+    feedback_table,
     logger,
     KNOWLEDGEBASE_ID,
     RAG_MODEL_ID,
@@ -47,7 +48,7 @@ def cleanup_previous_unfeedback_qa(conversation_key, current_message_ts, session
             return
 
         # Atomically delete Q&A only if no feedback received
-        table.delete_item(
+        slack_bot_state_table.delete_item(
             Key={"pk": f"qa#{conversation_key}#{previous_message_ts}", "sk": "turn"},
             ConditionExpression="attribute_not_exists(feedback_received)",
         )
@@ -67,7 +68,7 @@ def store_qa_pair(conversation_key, user_query, bot_response, message_ts, sessio
     Store Q&A pair for feedback correlation
     """
     try:
-        table.put_item(
+        slack_bot_state_table.put_item(
             Item={
                 "pk": f"qa#{conversation_key}#{message_ts}",
                 "sk": "turn",
@@ -90,7 +91,7 @@ def _mark_qa_feedback_received(conversation_key, message_ts):
     Mark Q&A record as having received feedback to prevent deletion
     """
     try:
-        table.update_item(
+        slack_bot_state_table.update_item(
             Key={"pk": f"qa#{conversation_key}#{message_ts}", "sk": "turn"},
             UpdateExpression="SET feedback_received = :val",
             ExpressionAttributeValues={":val": True},
@@ -322,9 +323,9 @@ def store_feedback(
             feedback_item["feedback_text"] = feedback_text[:4000]
 
         if condition:
-            table.put_item(Item=feedback_item, ConditionExpression=condition)
+            feedback_table.put_item(Item=feedback_item, ConditionExpression=condition)
         else:
-            table.put_item(Item=feedback_item)
+            feedback_table.put_item(Item=feedback_item)
 
         # Mark Q&A as having received feedback to prevent deletion
         if message_ts:
@@ -365,7 +366,7 @@ def get_conversation_session_data(conversation_key):
     Get full session data for this conversation
     """
     try:
-        response = table.get_item(Key={"pk": conversation_key, "sk": SESSION_SK})
+        response = slack_bot_state_table.get_item(Key={"pk": conversation_key, "sk": SESSION_SK})
         if "Item" in response:
             logger.info("Found existing session", extra={"conversation_key": conversation_key})
             return response["Item"]
@@ -380,7 +381,7 @@ def get_latest_message_ts(conversation_key):
     Get latest message timestamp from session
     """
     try:
-        response = table.get_item(Key={"pk": conversation_key, "sk": SESSION_SK})
+        response = slack_bot_state_table.get_item(Key={"pk": conversation_key, "sk": SESSION_SK})
         if "Item" in response:
             return response["Item"].get("latest_message_ts")
         return None
@@ -413,7 +414,7 @@ def store_conversation_session(
         if latest_message_ts:
             item["latest_message_ts"] = latest_message_ts
 
-        table.put_item(Item=item)
+        slack_bot_state_table.put_item(Item=item)
         logger.info("Stored session", extra={"session_id": session_id, "conversation_key": conversation_key})
     except Exception as e:
         logger.error("Error storing session", extra={"error": str(e)})
@@ -424,7 +425,7 @@ def update_session_latest_message(conversation_key, message_ts):
     Update session with latest message timestamp
     """
     try:
-        table.update_item(
+        slack_bot_state_table.update_item(
             Key={"pk": conversation_key, "sk": SESSION_SK},
             UpdateExpression="SET latest_message_ts = :ts",
             ExpressionAttributeValues={":ts": message_ts},

packages/slackBotFunction/app/slack/slack_handlers.py

@@ -15,7 +15,7 @@
 from botocore.exceptions import ClientError
 
 from app.core.config import (
-    table,
+    slack_bot_state_table,
     bot_token,
     logger,
     BOT_MESSAGES,
@@ -198,7 +198,7 @@ def thread_message_handler(event, event_id, client):
 
     conversation_key = f"{THREAD_PREFIX}{channel_id}#{thread_root}"
     try:
-        resp = table.get_item(Key={"pk": conversation_key, "sk": SESSION_SK})
+        resp = slack_bot_state_table.get_item(Key={"pk": conversation_key, "sk": SESSION_SK})
         if "Item" not in resp:
             logger.info(f"No session found for thread: {conversation_key}")
             return  # not a bot-owned thread; ignore
@@ -335,7 +335,7 @@ def _is_duplicate_event(event_id):
     """
     try:
         ttl = int(time.time()) + TTL_EVENT_DEDUP
-        table.put_item(
+        slack_bot_state_table.put_item(
             Item={"pk": f"{EVENT_PREFIX}{event_id}", "sk": DEDUP_SK, "ttl": ttl, "timestamp": int(time.time())},
             ConditionExpression="attribute_not_exists(pk)",
         )
@@ -350,7 +350,7 @@ def _is_duplicate_event(event_id):
 def _is_latest_message(conversation_key, message_ts):
     """Check if message_ts is the latest bot message using session data"""
     try:
-        response = table.get_item(Key={"pk": conversation_key, "sk": SESSION_SK})
+        response = slack_bot_state_table.get_item(Key={"pk": conversation_key, "sk": SESSION_SK})
         if "Item" in response:
             latest_message_ts = response["Item"].get("latest_message_ts")
             return latest_message_ts == message_ts