Upload folder using huggingface_hub (part 2)

src/config.py

@@ -596,11 +596,20 @@ class Config:
         self.data_db = os.getenv("DATA_DB", "mongo")
         self.graph_db = os.getenv("GRAPH_DB", "neo4j")
 
+        need_mongo = self.data_db.strip().lower() == "mongo"
+        need_milvus = self.vector_db.strip().lower() == "milvus"
+        need_neo4j = self.graph_db.strip().lower() == "neo4j"
+        need_postgres = (
+            self.data_db.strip().lower() == "postgresql"
+            or self.vector_db.strip().lower() == "postgresql"
+            or self.graph_db.strip().lower() == "networkx"
+        )
+
         self.redis = RedisConfig()
-        self.neo4j = Neo4jConfig()
-        self.postgresql = PostgreSQLConfig()
-        self.milvus = MilvusConfig()
-        self.mongo = MongoConfig()
+        self.neo4j = Neo4jConfig() if need_neo4j else None
+        self.postgresql = PostgreSQLConfig() if need_postgres else None
+        self.milvus = MilvusConfig() if need_milvus else None
+        self.mongo = MongoConfig() if need_mongo else None
         self.celery = CeleryConfig()
         self.spacy = SpacyConfig()
 

src/services/api/middlewares/auth.py

@@ -39,7 +39,7 @@ class BrainPATMiddleware(BaseHTTPMiddleware):
         if is_console_path(request.url.path):
             return await call_next(request)
 
-        if any(request.url.path.startswith(p) for p in self.excluded_prefixes):
+        if any(request.url.path.startswith(p) for p in self.excluded_prefixes) or "/mcp" in request.url.path:
             return await call_next(request)
 
         if request.url.path in self.auth_exempt_paths:

src/services/api/middlewares/brains.py

@@ -59,7 +59,7 @@ class BrainMiddleware(BaseHTTPMiddleware):
         if is_console_path(request.url.path):
             return await call_next(request)
 
-        if any(request.url.path.startswith(p) for p in self.excluded_prefixes):
+        if any(request.url.path.startswith(p) for p in self.excluded_prefixes) or "/mcp" in request.url.path:
             return await call_next(request)
 
         if request.url.path in self.brain_exempt_paths: