Update requirement

app/services/nlp_postprocess.py

@@ -7,8 +7,14 @@ from app.utils.hashing import sha256
 from app.config.settings import GEMINI_API_KEY
 
 # New official client
-import google.genai as genai
-from google.api_core.exceptions import GoogleAPIError  # optional but useful
+try:
+    import google.genai as genai
+    from google.api_core.exceptions import GoogleAPIError  # optional but useful
+except Exception:
+    genai = None
+    # fallback exception type so except GoogleAPIError still works
+    class GoogleAPIError(Exception):
+        pass
 
 CACHE_TTL = 60 * 60 * 24 * 3  # 3 days
 
@@ -16,13 +22,15 @@ CACHE_TTL = 60 * 60 * 24 * 3  # 3 days
 _gemini_client = None
 _GEMINI_MODEL = "gemini-1.5-flash"  # hoặc "gemini-1.5-flash-latest"
 
-if GEMINI_API_KEY:
+if GEMINI_API_KEY and genai is not None:
     try:
         _gemini_client = genai.Client(api_key=GEMINI_API_KEY)
         logging.info(f"[nlp_postprocess] Initialized google.genai client with model={_GEMINI_MODEL}")
     except Exception as e:
         logging.exception(f"[nlp_postprocess] Failed to init google.genai client: {e}")
         _gemini_client = None
+elif GEMINI_API_KEY and genai is None:
+    logging.warning("[nlp_postprocess] google.genai package not installed; GEMINI API disabled")
 else:
     logging.warning("[nlp_postprocess] GEMINI_API_KEY is not set, using raw_text as normalization fallback")
 

requirements.txt

@@ -11,5 +11,6 @@ rq
 tenacity
 prometheus-client
 google-generativeai
+google-genai
 numpy
 pytest