Merge pull request #823 from SciPhi-AI/Nolan/FastAPI

Introduce Server Stats

r2r/main/api/client.py

@@ -147,6 +147,10 @@ def _ensure_authenticated(self):
     def health(self) -> dict:
         return self._make_request("GET", "health")
 
+    def server_stats(self) -> dict:
+        self._ensure_authenticated()
+        return self._make_request("GET", "server_stats")
+
     def update_prompt(
         self,
         name: str = "default_system",

r2r/main/api/routes/management/base.py

@@ -1,4 +1,7 @@
 # TODO - Cleanup the handling for non-auth configurations
+from datetime import datetime, timezone
+
+import psutil
 from fastapi import Depends
 from pydantic import BaseModel
 
@@ -21,13 +24,37 @@
 class ManagementRouter(BaseRouter):
     def __init__(self, engine: R2REngine):
         super().__init__(engine)
+        self.start_time = datetime.now(timezone.utc)
         self.setup_routes()
 
     def setup_routes(self):
         @self.router.get("/health")
         async def health_check():
             return {"response": "ok"}
 
+        @self.router.get("/server_stats")
+        @self.base_endpoint
+        async def server_stats(
+            auth_user=(
+                Depends(self.engine.providers.auth.auth_wrapper)
+                if self.engine.providers.auth
+                else None
+            ),
+        ):
+            if not auth_user.is_superuser:
+                raise R2RException(
+                    "Only an authorized user can call the `server_stats` endpoint.",
+                    403,
+                )
+            return {
+                "start_time": self.start_time.isoformat(),
+                "uptime_seconds": (
+                    datetime.now(timezone.utc) - self.start_time
+                ).total_seconds(),
+                "cpu_usage": psutil.cpu_percent(),
+                "memory_usage": psutil.virtual_memory().percent,
+            }
+
         @self.router.post("/update_prompt")
         @self.base_endpoint
         async def update_prompt_app(

r2r/pipes/ingestion/chunking_pipe.py

@@ -55,7 +55,6 @@ async def _run_logic(
 
             try:
                 iteration = 0
-                print("item.data = ", item.data)
                 async for chunk in self.chunking_provider.chunk(item.data):
                     yield Fragment(
                         id=generate_id_from_label(f"{item.id}-{iteration}"),