Update api.py

api.py

@@ -21,13 +21,11 @@ NEO4J_PASSWORD = os.getenv("NEO4J_PASSWORD")
 if not NEO4J_URI or not NEO4J_USER or not NEO4J_PASSWORD:
     logger.critical("CRITICAL ERROR: NEO4J_URI, NEO4J_USER, and NEO4J_PASSWORD environment variables must be set.")
 
-client = None
 # --- Application Lifecycle (Startup/Shutdown) ---
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     """Handles startup and shutdown events."""
     # Initialize Gemini Client
-    global client
     logger.info("Initializing Gemini client...")
     if genai:
         try:
@@ -35,8 +33,9 @@ async def lifespan(app: FastAPI):
             api_key = os.getenv("GEMINI_API_KEY") or getattr(settings, "GEMINI_API_KEY", None)
             if not api_key:
                  raise ValueError("GEMINI_API_KEY not found in environment or settings.")
-            client = genai.Client(api_key=api_key)
-            logger.info("Gemini client configured successfully.")
+            else:
+                genai.configure(api_key=api_key)
+                logger.info("Gemini client configured successfully.")
         except Exception as e:
             logger.error(f"Failed to configure Gemini client: {e}", exc_info=True)
     else:
@@ -46,9 +45,6 @@ async def lifespan(app: FastAPI):
     
     # --- Shutdown ---
     logger.info("API shutting down...")
-    # Close Neo4j connection (handled by atexit in graph_client.py)
-    # neo4j_client.close() # Usually not needed due to atexit registration
-    logger.info("Neo4j client closed (likely via atexit).")
     logger.info("API shutdown complete.")
 
 # Initialize FastAPI application