Merge pull request #49 from ModusCreateOrg/ADE-166

[ADE-166] - Enhance Bedrock service and ChatPage error handling

Author: jkestering-modus

frontend/src/common/services/ai/bedrock.service.ts

@@ -1,6 +1,12 @@
 import { BedrockRuntimeClient, InvokeModelCommand } from '@aws-sdk/client-bedrock-runtime';
 import { fetchAuthSession } from '@aws-amplify/auth';
 import { REGION } from '../../config/aws-config';
+import i18n from '../../utils/i18n';
+
+// This function gets the translated message from i18n
+const getContentFilteredMessage = (): string => {
+  return i18n.t('ai.content_filtered', { ns: 'errors' });
+};
 
 export interface ChatMessage {
   role: 'user' | 'assistant' | 'system';
@@ -14,11 +20,24 @@ export interface ChatSession {
   updatedAt: Date;
 }
 
+// Interfaces for Bedrock API responses
+interface BedrockResult {
+  tokenCount: number;
+  outputText: string;
+  completionReason: 'CONTENT_FILTERED' | 'COMPLETE' | 'LENGTH' | 'STOP_SEQUENCE' | string;
+}
+
+interface BedrockResponse {
+  inputTextTokenCount: number;
+  results: BedrockResult[];
+}
+
 class BedrockService {
   private client: BedrockRuntimeClient | null = null;
   private readonly MODEL_ID = 'amazon.titan-text-lite-v1';
   private sessions: Map<string, ChatSession> = new Map();
   private isTestEnvironment: boolean;
+  private contentFilteredCount: number = 0; // Track number of filtered responses
 
   constructor() {
     // Check if we're in a test environment (Node.js environment with no window)
@@ -67,6 +86,27 @@ class BedrockService {
     }
   }
 
+  private handleBedrockResponse(parsedResponse: BedrockResponse): string {
+    // Check if we have results
+    if (!parsedResponse.results || !parsedResponse.results.length) {
+      throw new Error('Invalid response structure: missing results');
+    }
+    
+    const result = parsedResponse.results[0];
+    
+    // Check for content filtering
+    if (result.completionReason === "CONTENT_FILTERED") {
+      // Increment counter for analytics
+      this.contentFilteredCount++;
+      
+      // Return the translated message
+      return getContentFilteredMessage();
+    }
+
+    
+    return result.outputText;
+  }
+
   private async invokeModel(prompt: string): Promise<string> {
     // In test environment, return a mock response
     if (this.isTestEnvironment || !this.client) {
@@ -92,8 +132,9 @@ class BedrockService {
       const command = new InvokeModelCommand(input);
       const response = await this.client.send(command);
       const responseBody = new TextDecoder().decode(response.body);
-      const parsedResponse = JSON.parse(responseBody);
-      return parsedResponse.results[0].outputText;
+      const parsedResponse = JSON.parse(responseBody) as BedrockResponse;
+      
+      return this.handleBedrockResponse(parsedResponse);
     } catch (error) {
       console.error('Error invoking Bedrock model:', error);
       throw error;
@@ -162,6 +203,11 @@ class BedrockService {
   public getAllSessions(): ChatSession[] {
     return Array.from(this.sessions.values());
   }
+
+  // Add a method to get stats
+  public getContentFilteredStats(): { count: number } {
+    return { count: this.contentFilteredCount };
+  }
 }
 
 export const bedrockService = new BedrockService(); 

frontend/src/common/utils/i18n/index.ts

@@ -19,7 +19,7 @@ i18n
     // languages, namespaces, and resources
     supportedLngs: ['en', 'es', 'fr'],
     fallbackLng: 'en',
-    ns: ['account', 'auth', 'common', 'home', 'user'],
+    ns: ['account', 'auth', 'common', 'errors', 'home', 'user'],
     defaultNS: 'common',
     resources: { en, es, fr },
 

frontend/src/common/utils/i18n/resources/en/errors.json

@@ -0,0 +1,8 @@
+{
+  "chat": {
+    "general": "Sorry, something went wrong. Please try again."
+  },
+  "ai": {
+    "content_filtered": "I couldn't find an answer. Please try rephrasing your question or consult your healthcare provider."
+  }
+} 

frontend/src/common/utils/i18n/resources/en/index.ts

@@ -1,7 +1,8 @@
 import account from './account.json';
 import auth from './auth.json';
 import common from './common.json';
+import errors from './errors.json';
 import home from './home.json';
 import user from './user.json';
 
-export default { account, auth, common, home, user };
+export default { account, auth, common, errors, home, user };

frontend/src/common/utils/i18n/resources/es/errors.json

@@ -0,0 +1,8 @@
+{
+  "chat": {
+    "general": "Lo siento, algo salió mal. Por favor, inténtalo de nuevo."
+  },
+  "ai": {
+    "content_filtered": "No pude encontrar una respuesta. Intenta reformular tu pregunta o consulta a tu proveedor de salud."
+  }
+}

frontend/src/common/utils/i18n/resources/es/index.ts

@@ -1,7 +1,8 @@
 import account from './account.json';
 import auth from './auth.json';
 import common from './common.json';
+import errors from './errors.json';
 import home from './home.json';
 import user from './user.json';
 
-export default { account, auth, common, home, user };
+export default { account, auth, common, errors, home, user };

frontend/src/common/utils/i18n/resources/fr/errors.json

@@ -0,0 +1,8 @@
+{
+  "chat": {
+    "general": "Désolé, une erreur s'est produite. Veuillez réessayer."
+  },
+  "ai": {
+    "content_filtered": "Je n'ai pas pu trouver de réponse. Veuillez reformuler votre question ou consulter votre prestataire de soins de santé."
+  }
+} 

frontend/src/common/utils/i18n/resources/fr/index.ts

@@ -1,7 +1,8 @@
 import account from './account.json';
 import auth from './auth.json';
 import common from './common.json';
+import errors from './errors.json';
 import home from './home.json';
 import user from './user.json';
 
-export default { account, auth, common, home, user };
+export default { account, auth, common, errors, home, user };

frontend/src/pages/Chat/ChatPage.tsx

@@ -15,7 +15,7 @@ import './ChatPage.scss';
  * @returns JSX
  */
 const ChatPage = (): JSX.Element => {
-  const { t } = useTranslation();
+  const { t } = useTranslation(['common', 'errors']);
   const [messages, setMessages] = useState<ChatMessageData[]>([]);
   const location = useLocation();
   const prevPathRef = useRef(location.pathname);
@@ -73,7 +73,12 @@ const ChatPage = (): JSX.Element => {
       setMessages(prevMessages => [...prevMessages, assistantMessage]);
     } catch (error) {
       console.error('Error getting AI response:', error);
-      // You could add error handling here, like showing an error message
+      
+      // Use i18n directly with the full namespace and key
+      const errorMessage = t('chat.general', { ns: 'errors' });
+      
+      const assistantErrorMessage = chatService.createAssistantMessage(errorMessage);
+      setMessages(prevMessages => [...prevMessages, assistantErrorMessage]);
     }
   };