feat(api): handled multi brain deployment

Author: ChrisCoder9000

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "brainapi2"
-version = "1.6.1-dev"
+version = "1.6.2-dev"
 description = "Version 1.x.x of the BrainAPI memory layer."
 authors = [
     {name = "Christian",email = "[email protected]"}

src/services/api/middlewares/brains.py

@@ -22,10 +22,15 @@ class BrainMiddleware(BaseHTTPMiddleware):
     async def dispatch(self, request: Request, call_next):
         brain_id = None
 
-        brain_id = request.query_params.get("brain_id")
+        brain_id = request.headers.get("X-Brain-ID")
         if brain_id:
             brain_id = brain_id.rstrip()
 
+        if brain_id is None:
+            brain_id = request.query_params.get("brain_id")
+            if brain_id:
+                brain_id = brain_id.rstrip()
+
         if brain_id is None and request.method in ("POST", "PUT", "PATCH"):
             body = await request.body()
             if body:

src/services/api/routes/system.py

@@ -26,8 +26,8 @@ async def get_brains_list():
 
 
 @system_router.post(path="/brains")
-async def create_brain():
+async def create_brain(brain_id: str):
     """
     Create a new brain
     """
-    return await create_new_brain_controller()
+    return await create_new_brain_controller(brain_id)