enhance user feedback to include current prompt and partition in stor…

…age location
aws-samples · Mar 26, 2024 · 167c830 · 167c830
1 parent 479cf30
commit 167c830
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/lib/shared/layers/python-sdk/python/genai_core/user_feedback.py b/lib/shared/layers/python-sdk/python/genai_core/user_feedback.py
@@ -22,6 +22,7 @@ def add_user_feedback(
 ):
     feedbackId = str(uuid.uuid4())
     timestamp = datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%S.%fZ")
+    prefix = datetime.utcnow().strftime("user_feedback/year=%Y/month=%m/")
 
     item = {
         "feedbackId": feedbackId,
@@ -37,7 +38,7 @@ def add_user_feedback(
 
     response = s3_client.put_object(
         Bucket=USER_FEEDBACK_BUCKET_NAME,
-        Key=feedbackId,
+        Key=f"{prefix}{feedbackId}.json",
         Body=json.dumps(item),
         ContentType="application/json",
         StorageClass='STANDARD_IA',

diff --git a/lib/user-interface/react-app/src/components/chatbot/chat.tsx b/lib/user-interface/react-app/src/components/chatbot/chat.tsx
@@ -82,7 +82,11 @@ export default function Chat(props: { sessionId?: string }) {
 
   const handleFeedback = (feedbackType: 1 | 0, idx: number, message: ChatBotHistoryItem) => {
     if (message.metadata.sessionId) {
-      const prompt = (message.metadata.prompts === null || message.metadata.prompts === undefined) ? "" : message.metadata.prompts[0][0];
+
+      let prompt = "";
+      if (Array.isArray(message.metadata.prompts) && Array.isArray(message.metadata.prompts[0])) { 
+          prompt = message.metadata.prompts[0][0];
+      }
       const completion = message.content;
       const model = message.metadata.modelId;
       const feedbackData: FeedbackData = {

diff --git a/lib/user-interface/react-app/src/components/chatbot/multi-chat.tsx b/lib/user-interface/react-app/src/components/chatbot/multi-chat.tsx
@@ -334,7 +334,10 @@ export default function MultiChat() {
     console.log("Message history: ", messageHistory);
     // metadata.prompts[0][0]
     if (message.metadata.sessionId) {
-      const prompt = (message.metadata.prompts === null || message.metadata.prompts === undefined) ? "" : message.metadata.prompts[0][0];
+      let prompt = "";
+      if (Array.isArray(message.metadata.prompts) && Array.isArray(message.metadata.prompts[0])) { 
+          prompt = message.metadata.prompts[0][0];
+      }
       const completion = message.content;
       const model = message.metadata.modelId;
       const feedbackData: FeedbackData = {