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
|