add features
Browse files
ocr/api/message/db_requests.py
CHANGED
|
@@ -26,7 +26,6 @@ async def get_all_chat_messages_obj(report_id: str) -> list[MessageModel]:
|
|
| 26 |
settings.DB_CLIENT.reports.find_one({"id": report_id})
|
| 27 |
)
|
| 28 |
messages = [MessageModel.from_mongo(message) for message in messages]
|
| 29 |
-
|
| 30 |
if not report:
|
| 31 |
raise HTTPException(status_code=404, detail="Report not found")
|
| 32 |
return messages
|
|
@@ -35,6 +34,8 @@ async def get_all_chat_messages_obj(report_id: str) -> list[MessageModel]:
|
|
| 35 |
async def save_assistant_user_message(user_message: str, assistant_message: str, report_id: str) -> MessageModel:
|
| 36 |
user_message = MessageModel(reportId=report_id, author=Author.User, text=user_message)
|
| 37 |
assistant_message = MessageModel(reportId=report_id, author=Author.Assistant, text=assistant_message)
|
|
|
|
| 38 |
await settings.DB_CLIENT.messages.insert_one(user_message.to_mongo())
|
| 39 |
await settings.DB_CLIENT.messages.insert_one(assistant_message.to_mongo())
|
|
|
|
| 40 |
return user_message
|
|
|
|
| 26 |
settings.DB_CLIENT.reports.find_one({"id": report_id})
|
| 27 |
)
|
| 28 |
messages = [MessageModel.from_mongo(message) for message in messages]
|
|
|
|
| 29 |
if not report:
|
| 30 |
raise HTTPException(status_code=404, detail="Report not found")
|
| 31 |
return messages
|
|
|
|
| 34 |
async def save_assistant_user_message(user_message: str, assistant_message: str, report_id: str) -> MessageModel:
|
| 35 |
user_message = MessageModel(reportId=report_id, author=Author.User, text=user_message)
|
| 36 |
assistant_message = MessageModel(reportId=report_id, author=Author.Assistant, text=assistant_message)
|
| 37 |
+
|
| 38 |
await settings.DB_CLIENT.messages.insert_one(user_message.to_mongo())
|
| 39 |
await settings.DB_CLIENT.messages.insert_one(assistant_message.to_mongo())
|
| 40 |
+
|
| 41 |
return user_message
|