Add system check

Author: YoanSallami

synalinks/cli/constants.py

@@ -114,6 +114,7 @@ def get_fake_data(filename):
 
 MAIN_TEMPLATE = """import os
 import synalinks
+import psutil
 from dotenv import load_dotenv
 from fastapi import FastAPI
 # Uncomment for streaming apps
@@ -161,10 +162,23 @@ async def create_program():
 # Setup FastAPI
 app = FastAPI()
 
-@app.get("/v1/{{config.package_name}}/health")
+@app.get("/health")
 def health_check():
     return {"status": "healthy"}
 
+
+@app.get("/system_check")
+def system_check():
+    cpu_usage = psutil.cpu_percent(interval=1)
+    memory_info = psutil.virtual_memory()
+    disk_usage = psutil.disk_usage("/")
+    return {
+        "cpu_usage": cpu_usage,
+        "memory_usage": memory_info.percent,
+        "disk_usage": disk_usage.percent,
+    }
+
+
 @app.post("/v1/{{config.package_name}}")
 async def {{config.package_name}}(inputs: Query):
     result = await program(inputs)
@@ -219,6 +233,7 @@ async def {{config.package_name}}(inputs: Query):
 """
 
 REQUIREMENTS_TEMPLATE = """fastapi[standard]
+psutil
 synalinks
 """
 
@@ -277,6 +292,9 @@ async def build(self, inputs):
         super().__init__(
             inputs=inputs,
             outputs=outputs,
+            name=self.name,
+            description=self.description,
+            trainable=self.trainable,
         )
 """
 

synalinks/src/version.py

@@ -3,7 +3,7 @@
 from synalinks.src.api_export import synalinks_export
 
 # Unique source of truth for the version number.
-__version__ = "0.2.014"
+__version__ = "0.2.015"
 
 
 @synalinks_export("synalinks.version")