vioott commited on
Commit
410d534
·
1 Parent(s): 4be8d5a

feat(error-handling): add specific error message for quota exhaustion

Browse files
Files changed (1) hide show
  1. routes/chat.py +5 -0
routes/chat.py CHANGED
@@ -3,6 +3,7 @@ import sys
3
  import logging
4
  import traceback
5
  from flask import Blueprint, render_template, request
 
6
  from services.genai_service import configure_genai, list_available_models
7
  from logs import get_user_history, save_log
8
 
@@ -62,6 +63,10 @@ def chat(user_id):
62
 
63
  save_log(user_id, history, response)
64
 
 
 
 
 
65
  except Exception as e:
66
  logger.error(f"Erro CRÍTICO ao usar a IA ({current_model}): {e}")
67
  logger.error(traceback.format_exc())
 
3
  import logging
4
  import traceback
5
  from flask import Blueprint, render_template, request
6
+ from google.api_core.exceptions import ResourceExhausted
7
  from services.genai_service import configure_genai, list_available_models
8
  from logs import get_user_history, save_log
9
 
 
63
 
64
  save_log(user_id, history, response)
65
 
66
+ except ResourceExhausted:
67
+ logger.warning(f"Cota diária excedida para o modelo {current_model}.")
68
+ response = "⚠️ A cota gratuita diária se esgotou. Por favor, retorne em 24h."
69
+
70
  except Exception as e:
71
  logger.error(f"Erro CRÍTICO ao usar a IA ({current_model}): {e}")
72
  logger.error(traceback.format_exc())