fix

ai_service/models/chat_service.py

@@ -286,21 +286,35 @@ class ChatService:
 
         # Get response from model
         model_id = chat['model_id']
-        response_text = model_service.generate_response(
-            model_id=model_id,
-            prompt=message,
-            context=context,
-            use_rag=use_rag,
-            model_params=model_params
-        )
+        try:
+            print(f"Requesting response from model {model_id} for chat {chat_id}")
+            response_text = model_service.generate_response(
+                model_id=model_id,
+                prompt=message,
+                context=context,
+                use_rag=use_rag,
+                model_params=model_params
+            )
 
-        # Add bot response to chat
-        response_message = self.add_message(
-            chat_id=chat_id,
-            content=response_text,
-            user_id=user_id,
-            is_user_message=False
-        )
+            # Add bot response to chat
+            response_message = self.add_message(
+                chat_id=chat_id,
+                content=response_text,
+                user_id=user_id,
+                is_user_message=False
+            )
+
+        except Exception as e:
+            error_message = f"Error generating response: {str(e)}"
+            print(f"ERROR: {error_message}")
+
+            # Add error message to chat
+            response_message = self.add_message(
+                chat_id=chat_id,
+                content=error_message,
+                user_id=user_id,
+                is_user_message=False
+            )
 
         return response_message