feat: add health endpoint and auto-generate pod_id

- Add /health endpoint to FastAPI server for load balancer health checks
- Auto-generate pod_id when not set: WAVERLESS_POD_ID > DEVICE_ID > {endpoint}-{uuid}

Author: arabot777

src/wavespeed/config.py

@@ -2,6 +2,7 @@
 
 import os
 import sys
+import uuid
 from typing import Optional
 
 from ._config_module import install_config_module
@@ -85,16 +86,37 @@ def _detect_serverless_env() -> Optional[str]:
     """
 
     # Check for native Waverless environment
-    if os.environ.get("WAVERLESS_POD_ID"):
+    if os.environ.get("WAVERLESS_ENDPOINT_ID"):
         return "waverless"
 
     # Check for RunPod environment
-    if os.environ.get("RUNPOD_POD_ID"):
+    if os.environ.get("RUNPOD_ENDPOINT_ID"):
         return "runpod"
 
     return None
 
 
+def _generate_pod_id(endpoint_id: Optional[str], raw_pod_id: Optional[str]) -> str:
+    """Generate or resolve pod_id.
+
+    Priority: raw_pod_id > DEVICE_ID > auto-generate
+
+    Args:
+        endpoint_id: The endpoint identifier.
+        raw_pod_id: The raw pod_id from environment variable.
+
+    Returns:
+        The resolved pod_id.
+    """
+    if raw_pod_id:
+        return raw_pod_id
+    device_id = os.environ.get("DEVICE_ID")
+    if device_id:
+        return device_id
+    prefix = endpoint_id or "worker"
+    return f"{prefix}-{uuid.uuid4().hex}"
+
+
 def _resolve_runpod_url(url_template: Optional[str], pod_id: str) -> Optional[str]:
     """Replace pod ID placeholder in RunPod URL template.
 
@@ -127,8 +149,14 @@ def _resolve_waverless_url(url_template: Optional[str], pod_id: str) -> Optional
 
 def _load_runpod_serverless_config() -> None:
     """Load RunPod environment variables into serverless config."""
+    # Endpoint identification (load first for pod_id generation)
+    serverless.endpoint_id = os.environ.get("RUNPOD_ENDPOINT_ID")
+    serverless.project_id = os.environ.get("RUNPOD_PROJECT_ID")
+
     # Worker identification
-    serverless.pod_id = os.environ.get("RUNPOD_POD_ID") or ""
+    raw_pod_id = os.environ.get("RUNPOD_POD_ID")
+    serverless.pod_id = _generate_pod_id(serverless.endpoint_id, raw_pod_id)
+    serverless.pod_hostname = os.environ.get("RUNPOD_POD_HOSTNAME", serverless.pod_id)
 
     # API endpoint templates
     serverless.webhook_get_job = os.environ.get("RUNPOD_WEBHOOK_GET_JOB")
@@ -163,11 +191,6 @@ def _load_runpod_serverless_config() -> None:
         log_level = os.environ.get("RUNPOD_DEBUG_LEVEL")
     serverless.log_level = log_level or "INFO"
 
-    # Endpoint identification
-    serverless.endpoint_id = os.environ.get("RUNPOD_ENDPOINT_ID")
-    serverless.project_id = os.environ.get("RUNPOD_PROJECT_ID")
-    serverless.pod_hostname = os.environ.get("RUNPOD_POD_HOSTNAME")
-
     # Timing and concurrency
     ping_interval = os.environ.get("RUNPOD_PING_INTERVAL")
     if ping_interval:
@@ -184,8 +207,17 @@ def _load_runpod_serverless_config() -> None:
 
 def _load_waverless_serverless_config() -> None:
     """Load Waverless environment variables into serverless config."""
+    # Endpoint identification (load first for pod_id generation)
+    serverless.endpoint_id = os.environ.get("WAVERLESS_ENDPOINT_ID")
+    # Endpoint identification (endpoint_id already set above)
+    serverless.project_id = os.environ.get("WAVERLESS_PROJECT_ID")
+
     # Worker identification
-    serverless.pod_id = os.environ.get("WAVERLESS_POD_ID") or ""
+    raw_pod_id = os.environ.get("WAVERLESS_POD_ID")
+    serverless.pod_id = _generate_pod_id(serverless.endpoint_id, raw_pod_id)
+    serverless.pod_hostname = os.environ.get(
+        "WAVERLESS_POD_HOSTNAME", serverless.pod_id
+    )
 
     # API endpoint templates
     serverless.webhook_get_job = os.environ.get("WAVERLESS_WEBHOOK_GET_JOB")
@@ -217,11 +249,6 @@ def _load_waverless_serverless_config() -> None:
     # Logging
     serverless.log_level = os.environ.get("WAVERLESS_LOG_LEVEL", "INFO")
 
-    # Endpoint identification
-    serverless.endpoint_id = os.environ.get("WAVERLESS_ENDPOINT_ID")
-    serverless.project_id = os.environ.get("WAVERLESS_PROJECT_ID")
-    serverless.pod_hostname = os.environ.get("WAVERLESS_POD_HOSTNAME")
-
     # Timing and concurrency
     ping_interval = os.environ.get("WAVERLESS_PING_INTERVAL")
     if ping_interval:

src/wavespeed/serverless/modules/fastapi.py

@@ -220,6 +220,15 @@ def __init__(self, config: Dict[str, Any]):
             tags=["Status"],
         )
 
+        # Health check endpoint
+        router.add_api_route(
+            "/health",
+            lambda: {"status": "ok"},
+            methods=["GET"],
+            summary="Health check",
+            tags=["Status"],
+        )
+
         self.app.include_router(router)
 
     def start(