codeBOKER commited on
Commit
f0d2c6c
·
verified ·
1 Parent(s): 7af4284

fix latency update

Browse files
Files changed (1) hide show
  1. ai_service.py +4 -3
ai_service.py CHANGED
@@ -244,8 +244,9 @@ async def get_ai_response(user_query: str, telegram_id: int):
244
  final_response = clean_ai_response(response_message.content if response_message.content else "")
245
 
246
  if db_manager:
247
- print("arriving to save message: \n")
248
- db_manager.save_message(telegram_id, user_query, "user")
249
- db_manager.save_message(telegram_id, final_response, "assistant")
 
250
 
251
  return final_response
 
244
  final_response = clean_ai_response(response_message.content if response_message.content else "")
245
 
246
  if db_manager:
247
+ await asyncio.gather(
248
+ db_manager.save_message_async(telegram_id, user_query, "user"),
249
+ db_manager.save_message_async(telegram_id, final_response, "assistant")
250
+ )
251
 
252
  return final_response