Change Gemini model version

app/config.py

@@ -1,4 +1,5 @@
 import os
 
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "")
+GEMINI_MODEL = os.getenv("GEMINI_MODEL", "")
 FIREBASE_SERVICE_ACCOUNT = os.getenv("FIREBASE_SERVICE_ACCOUNT", "")

app/services/mindmap_service.py

@@ -3,23 +3,25 @@ import asyncio
 import json
 import logging
 
-from app.config import GEMINI_API_KEY
+from app.config import GEMINI_API_KEY, GEMINI_MODEL
 
 import google.genai as genai
 from google.api_core.exceptions import GoogleAPIError
 
-_MINDMAP_MODEL = "gemini-1.5-flash"
+_MINDMAP_MODEL = GEMINI_MODEL
 _gemini_client = None
 
-if GEMINI_API_KEY:
+if not GEMINI_API_KEY:
+    logging.warning("[mindmap_service] GEMINI_API_KEY is not set; mindmap generation disabled")
+elif not _MINDMAP_MODEL:
+    logging.warning("[mindmap_service] GEMINI_MODEL is not set; mindmap generation disabled")
+else:
     try:
         _gemini_client = genai.Client(api_key=GEMINI_API_KEY)
-        logging.info(f"[mindmap_service] Initialized google.genai client with model={_MINDMAP_MODEL}")
+        logging.info(f"[mindmap_service] Initialized google.genai client; model={_MINDMAP_MODEL}")
     except Exception as e:
         logging.exception(f"[mindmap_service] Failed to init google.genai client: {e}")
         _gemini_client = None
-else:
-    logging.warning("[mindmap_service] GEMINI_API_KEY is not set, mindmap generation will be disabled")
 
 
 async def generate_mindmap(text: str) -> dict:

app/services/summary_service.py

@@ -1,7 +1,7 @@
 import asyncio
 import logging
 
-from app.config import GEMINI_API_KEY
+from app.config import GEMINI_API_KEY, GEMINI_MODEL
 
 try:
     import google.genai as genai
@@ -11,17 +11,19 @@ except Exception:
 
 from google.api_core.exceptions import GoogleAPIError
 
-_SUMMARY_MODEL = "gemini-1.5-flash"
+_SUMMARY_MODEL = GEMINI_MODEL
 _gemini_client = None
 
 if not genai:
-    logging.warning("[summary_service] google.genai not available, summary will be empty")
+    logging.warning("[summary_service] google.genai not available; summaries disabled")
 elif not GEMINI_API_KEY:
-    logging.warning("[summary_service] GEMINI_API_KEY is not set, summary will be empty")
+    logging.warning("[summary_service] GEMINI_API_KEY is not set; summaries disabled")
+elif not _SUMMARY_MODEL:
+    logging.warning("[summary_service] GEMINI_MODEL is not set; summaries disabled")
 else:
     try:
         _gemini_client = genai.Client(api_key=GEMINI_API_KEY)
-        logging.info(f"[summary_service] Initialized google.genai client with model={_SUMMARY_MODEL}")
+        logging.info(f"[summary_service] Initialized google.genai client; model={_SUMMARY_MODEL}")
     except Exception as e:
         logging.exception(f"[summary_service] Failed to init google.genai client: {e}")
         _gemini_client = None