signal checks

Author: qw3rtman

docs/changelog.md

@@ -4,7 +4,12 @@ icon: paste
 # Changelog
 
 ---
-## v0.2.0 [!badge text="LATEST" variant="info"]
+## v0.2.1 [!badge text="LATEST" variant="info"]
+
+Released: [2025-02-22](https://github.com/haizelabs/verdict/releases/tag/v0.2.1)
+- Signal handler check for NVidia NeMo-Guardrails integration
+
+## v0.2.0
 
 Released: [2025-02-19](https://github.com/haizelabs/verdict/releases/tag/v0.2.0)
 - Initial public release.

docs/retype.yml

@@ -4,7 +4,7 @@ url: verdict.haizelabs.com/docs
 
 branding:
   title: Verdict
-  label: 0.2.0
+  label: 0.2.1
   logo: /static/logos/logo.png
   logoDark: /static/logos/logo.png
   logoAlign: left

pyproject.toml

@@ -13,7 +13,7 @@ issues = "https://github.com/haizelabs/verdict/issues"
 
 [project]
 name = "verdict"
-version = "0.2.0"
+version = "0.2.1"
 authors = [{ name = "Nimit Kalra", email = "[email protected]" }]
 license = { text = "MIT License" }
 description = "LLM evaluation pipelines"

uv.lock

@@ -1,11 +1,43 @@
 from __future__ import annotations
 
 import logging
+import os
 import signal
+import sys
+import threading
 from functools import wraps
 from typing import Any, Callable, Optional, Type
 
 
+def is_signal_safe():
+    """
+    Strictly check if the current environment is safe for signal handling.
+
+    Returns:
+        bool: True only if ALL signal safety conditions are met
+    """
+    # Must be in the main thread
+    if threading.current_thread() is not threading.main_thread():
+        return False
+
+    # Must be in the main interpreter
+    if not hasattr(sys, 'argv'):
+        return False
+
+    # Check if we're in a forked process
+    if os.getpid() != os.getppid():
+        return False
+
+    # Additional interpreter state checks
+    if not sys.modules.get('__main__'):
+        return False
+
+    # Check if the interpreter is shutting down
+    if sys is None or threading is None:
+        return False
+
+    return True
+
 def keyboard_interrupt_safe(func: Callable[[Any], Any]) -> Callable[[Any], Any]:
     @wraps(func)
     def wrapped(self: Any, *args, **kwargs) -> Any:
@@ -18,7 +50,11 @@ def wrapped(self: Any, *args, **kwargs) -> Any:
             self.executor.graceful_shutdown()
         finally:
             signal.signal(signal.SIGINT, original_handler)
-    return wrapped
+
+    if is_signal_safe():
+        return wrapped
+    else:
+        return func
 
 class DisableLogger:
     def __init__(self, logger_name: str, all: bool=False) -> None:
@@ -40,3 +76,4 @@ def __exit__(self, exc_type, exc_value, traceback) -> None:
 def lightweight(cls: Type) -> Type:
     cls.lightweight = True
     return cls
+