brestok commited on
Commit
9eb128a
·
1 Parent(s): 34c9c72

add features

Browse files
Files changed (1) hide show
  1. ocr/api/message/db_requests.py +2 -1
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