Add feedback support to messages

Introduced a `Feedback` model and integrated it into `MessageModel` for storing user feedback. Added API endpoint for updating message feedback and implemented database logic to handle feedback updates. Updated corresponding imports and removed unused functions in the AI engine.

trauma/api/message/ai/engine.py

@@ -14,9 +14,7 @@ from trauma.api.message.ai.openai_request import (update_entity_data_with_ai,
                                                   choose_closest_treatment_area,
                                                   check_is_valid_request,
                                                   generate_invalid_response,
-                                                  set_entity_score,
-                                                  retrieve_semantic_answer,
-                                                  generate_searched_entity_response)
+                                                  set_entity_score)
 from trauma.api.message.db_requests import (save_assistant_user_message,
                                             filter_entities_by_age_location,
                                             update_entity_data_obj, get_entity_by_index)

trauma/api/message/db_requests.py

@@ -6,7 +6,7 @@ from trauma.api.account.dto import AccountType
 from trauma.api.account.model import AccountModel
 from trauma.api.chat.model import ChatModel
 from trauma.api.data.model import EntityModel, EntityModelExtended
-from trauma.api.message.dto import Author
+from trauma.api.message.dto import Author, Feedback
 from trauma.api.message.model import MessageModel
 from trauma.api.message.schemas import CreateMessageRequest
 from trauma.core.config import settings
@@ -89,3 +89,14 @@ async def filter_entities_by_age_location(entity_data: dict) -> list[int]:
 async def get_entity_by_index(index: int) -> EntityModel:
     entity = await settings.DB_CLIENT.entities.find_one({"index": index})
     return EntityModel.from_mongo(entity)
+
+
+async def update_message_feedback_obj(message_id: str, feedback_data: Feedback) -> MessageModel:
+    message = await settings.DB_CLIENT.messages.find_one({"id": message_id})
+    if not message:
+        raise HTTPException(status_code=404, detail="Message not found")
+    message = MessageModel.from_mongo(message)
+    message.feedback = feedback_data
+    await settings.DB_CLIENT.messages.update_one({"id": message_id},
+                                                 {"$set": {"feedback": feedback_data.model_dump(mode='json')}})
+    return message

trauma/api/message/dto.py

@@ -11,3 +11,7 @@ class Author(Enum):
 class File(BaseModel):
     name: str
     url: str
+
+class Feedback(BaseModel):
+    isPositive: bool
+    comment: str | None = None

trauma/api/message/model.py

@@ -3,7 +3,7 @@ from datetime import datetime
 from pydantic import Field
 
 from trauma.api.data.model import EntityModelExtended
-from trauma.api.message.dto import Author
+from trauma.api.message.dto import Author, Feedback
 from trauma.core.database import MongoBaseModel
 
 
@@ -12,5 +12,6 @@ class MessageModel(MongoBaseModel):
     author: Author
     text: str
     entities: list[EntityModelExtended] | None = None
+    feedback: Feedback | None = None
     datetimeInserted: datetime = Field(default_factory=datetime.now)
     datetimeUpdated: datetime = Field(default_factory=datetime.now)

trauma/api/message/views.py

@@ -5,7 +5,9 @@ from trauma.api.account.model import AccountModel
 from trauma.api.common.dto import Paging
 from trauma.api.message import message_router
 from trauma.api.message.ai.engine import search_entities
-from trauma.api.message.db_requests import get_all_chat_messages_obj
+from trauma.api.message.db_requests import get_all_chat_messages_obj, update_message_feedback_obj
+from trauma.api.message.dto import Feedback
+from trauma.api.message.model import MessageModel
 from trauma.api.message.schemas import (AllMessageWrapper,
                                         AllMessageResponse,
                                         CreateMessageRequest,
@@ -38,3 +40,13 @@ async def create_message(
     message_history = transform_messages_to_openai(messages)
     response = await search_entities(message_data.text, message_history, chat)
     return TraumaResponseWrapper(data=response)
+
+
+@message_router.patch('/{messageId}/feedback')
+async def create_feedback(
+        messageId: str,
+        feedback_data: Feedback,
+        _: AccountModel = Depends(PermissionDependency([AccountType.Admin, AccountType.User]))
+) -> TraumaResponseWrapper[MessageModel]:
+    message = await update_message_feedback_obj(messageId, feedback_data)
+    return TraumaResponseWrapper(data=message)