fix(brains)

Author: ChrisCoder9000

.gitignore

@@ -5,4 +5,5 @@ __pycache__
 *.excalidraw
 dist
 common.egg-info
-.venv
+.venv
+.env.development

Dockerfile

@@ -84,4 +84,4 @@ HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
     CMD curl -f http://localhost:8000/docs || exit 1
 
 # Default command
-CMD ["/app/.venv/bin/python", "-m", "uvicorn", "src.services.api.app:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["/app/.venv/bin/python", "-m", "uvicorn", "src.services.api.app:app", "--host", "0.0.0.0", "--port", "8000", "--access-log", "--log-level", "info"]

Makefile

@@ -45,7 +45,7 @@ delete-mongo-volumes:
 	docker compose -f src/lib/mongo/docker-compose.yaml down -v --remove-orphans
 
 start-api:
-	poetry run uvicorn src.services.api.app:app --host 0.0.0.0 --port 8000
+	ENV=development poetry run uvicorn src.services.api.app:app --host 0.0.0.0 --port 8000 --access-log --log-level info --reload
 
 stop-api:
 	pkill -f uvicorn

pyproject.toml

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

src/services/api/app.py

@@ -10,7 +10,7 @@
 
 import os
 
-from fastapi import FastAPI
+from fastapi import FastAPI, Response
 from fastapi.middleware.cors import CORSMiddleware
 from uvicorn import run
 
@@ -21,8 +21,7 @@
 from src.services.api.routes.meta import meta_router
 from src.services.api.routes.system import system_router
 
-
-app = FastAPI()
+app = FastAPI(debug=os.getenv("ENV") == "development")
 
 app.add_middleware(BrainPATMiddleware)
 app.add_middleware(BrainMiddleware)
@@ -40,5 +39,11 @@
 app.include_router(meta_router)
 app.include_router(system_router)
 
+
+@app.get("/")
+async def root():
+    return Response(content="ok", status_code=200)
+
+
 if __name__ == "__main__":
     run(app, host="0.0.0.0", port=8000, reload=os.getenv("ENV") == "development")

src/services/api/middlewares/auth.py

@@ -40,18 +40,20 @@ async def dispatch(self, request: Request, call_next):
         if not cached_brainpat and not use_only_system_pat:
             stored_brain = data_adapter.get_brain(name_key=brain_id)
             system_pat = os.getenv("BRAINPAT_TOKEN")
-            if not stored_brain and brainpat != system_pat:
-                return JSONResponse(
-                    status_code=status.HTTP_401_UNAUTHORIZED,
-                    content={"detail": "Invalid or missing BrainPAT header"},
+            if not stored_brain:
+                if brainpat != system_pat:
+                    return JSONResponse(
+                        status_code=status.HTTP_401_UNAUTHORIZED,
+                        content={"detail": "Invalid or missing BrainPAT header"},
+                    )
+                cached_brainpat = system_pat
+            else:
+                cached_brainpat = stored_brain.pat
+                cache_adapter.set(
+                    key=cachepat_key,
+                    value=stored_brain.pat,
+                    brain_id="system",
                 )
-
-            cached_brainpat = stored_brain.pat
-            cache_adapter.set(
-                key=cachepat_key,
-                value=stored_brain.pat,
-                brain_id="system",
-            )
         if not cached_brainpat and use_only_system_pat:
             system_pat = os.getenv("BRAINPAT_TOKEN")
             cached_brainpat = system_pat

src/services/api/middlewares/brains.py

@@ -20,6 +20,9 @@
 
 class BrainMiddleware(BaseHTTPMiddleware):
     async def dispatch(self, request: Request, call_next):
+        if request.method == "POST" and request.url.path == "/system/brains":
+            return await call_next(request)
+
         brain_id = None
 
         brain_id = request.headers.get("X-Brain-ID")

src/utils/logging.py

@@ -13,7 +13,6 @@
 import time
 from typing import Callable
 
-logging.basicConfig(level=logging.INFO, format="%(levelname)s: %(message)s")
 logger = logging.getLogger(__name__)