Update main.py

main.py

@@ -8,6 +8,7 @@ from dotenv import load_dotenv
 #from kb_embed import search_knowledge_base
 from services.kb_creation import collection, ingest_documents, search_knowledge_base
 from contextlib import asynccontextmanager
+import google.generativeai as genai
 
 
 # --- 0. Config ---