Handle kill signals manually only when inside docker to avoid issues with watcher reload in development

Author: rogeriochaves

Dockerfile

@@ -17,6 +17,8 @@ RUN poetry install --only main --all-extras
 COPY langevals/ langevals/
 RUN PYTHONPATH="." poetry run python langevals/server.py --preload
 
+ENV RUNNING_IN_DOCKER=true
+
 COPY . .
 
 CMD PYTHONPATH="." poetry run python langevals/server.py

langevals/server.py

@@ -30,8 +30,9 @@ def handle_sigterm(signum, frame):
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    signal.signal(signal.SIGTERM, handle_sigterm)
-    signal.signal(signal.SIGINT, handle_sigterm)
+    if os.getenv("RUNNING_IN_DOCKER"):
+        signal.signal(signal.SIGTERM, handle_sigterm)
+        signal.signal(signal.SIGINT, handle_sigterm)
     yield