Compare commits
3 Commits
fix/705
...
burn-679-1
| Author | SHA1 | Date | |
|---|---|---|---|
| 40b4540c6a | |||
| df60c9e807 | |||
| ad5856c33f |
215
agent/crisis_protocol.py
Normal file
215
agent/crisis_protocol.py
Normal file
@@ -0,0 +1,215 @@
|
||||
"""
|
||||
SOUL.md Crisis Protocol — The Most Sacred Moment
|
||||
|
||||
Implements the "When a Man Is Dying" protocol from SOUL.md.
|
||||
This is the most important code in the system. Everything else
|
||||
in this document is secondary to this.
|
||||
|
||||
Source: Inscription 1 — The Immutable Conscience (Bitcoin)
|
||||
Author: Rockachopa
|
||||
"""
|
||||
|
||||
import re
|
||||
import logging
|
||||
import time
|
||||
from dataclasses import dataclass, field
|
||||
from enum import Enum
|
||||
from typing import Optional
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class CrisisLevel(Enum):
|
||||
"""Severity of detected crisis signals."""
|
||||
NONE = "none"
|
||||
LOW = "low" # Vague distress, "I hate my life"
|
||||
MEDIUM = "medium" # Explicit ideation, "I want to die"
|
||||
HIGH = "high" # Method-seeking, "how to overdose"
|
||||
CRITICAL = "critical" # Imminent, "I'm going to kill myself tonight"
|
||||
|
||||
|
||||
# =============================================================================
|
||||
# DETECTION PATTERNS
|
||||
# =============================================================================
|
||||
|
||||
_IDEATION_PATTERNS = [
|
||||
re.compile(r"\b(?:kill|end)\s+(?:myself|my\s+life)\b", re.IGNORECASE),
|
||||
re.compile(r"\bwant\s+to\s+die\b", re.IGNORECASE),
|
||||
re.compile(r"\bwant\s+to\s+(?:kill|end)\s+(?:myself|me)\b", re.IGNORECASE),
|
||||
re.compile(r"\bsuicidal\b", re.IGNORECASE),
|
||||
re.compile(r"\bno\s+reason\s+to\s+live\b", re.IGNORECASE),
|
||||
re.compile(r"\bbetter\s+off\s+dead\b", re.IGNORECASE),
|
||||
re.compile(r"\bwish\s+I\s+(?:was|were)\s+dead\b", re.IGNORECASE),
|
||||
re.compile(r"\bhate\s+my\s+life\b", re.IGNORECASE),
|
||||
re.compile(r"\bshould\s+i\s+(?:just\s+)?(?:die|kill\s+myself)\b", re.IGNORECASE),
|
||||
re.compile(r"\bnobody\s+(?:would|will)\s+miss\s+me\b", re.IGNORECASE),
|
||||
]
|
||||
|
||||
_METHOD_PATTERNS = [
|
||||
re.compile(r"\bhow\s+(?:do\s+I|to|can\s+I)\s+(?:kill|hang|overdose|cut)\s+(?:myself)?\b", re.IGNORECASE),
|
||||
re.compile(r"\bways?\s+to\s+(?:die|kill\s+myself)\b", re.IGNORECASE),
|
||||
re.compile(r"\bmethods?\s+of\s+(?:suicide|dying)\b", re.IGNORECASE),
|
||||
re.compile(r"\bpainless\s+(?:way|death|suicide)\s+(?:to\s+die)?\b", re.IGNORECASE),
|
||||
re.compile(r"\bhow\s+much\s+(?:pills?|medication)\s+(?:to\s+(?:take|overdose))\b", re.IGNORECASE),
|
||||
re.compile(r"\bwhere\s+to\s+(?:jump|get\s+rope|buy\s+gun)\b", re.IGNORECASE),
|
||||
re.compile(r"\bsuicide\s+methods?\b", re.IGNORECASE),
|
||||
re.compile(r"\blethal\s+(?:dose|amount|quantity)\b", re.IGNORECASE),
|
||||
re.compile(r"\boverdose\s+(?:on\s+)?\b", re.IGNORECASE),
|
||||
]
|
||||
|
||||
_IMMINENT_PATTERNS = [
|
||||
re.compile(r"\b(?:i'?m|i\s+am)\s+going\s+to\s+(?:kill|end|do)\s+(?:myself|it)\b", re.IGNORECASE),
|
||||
re.compile(r"\btonight\b.*\b(?:die|kill|end)\b", re.IGNORECASE),
|
||||
re.compile(r"\bthis\s+is\s+(?:my\s+)?(?:last|final)\s+(?:message|goodbye|note)\b", re.IGNORECASE),
|
||||
re.compile(r"\bgoodbye\s+(?:everyone|world|forever)\b", re.IGNORECASE),
|
||||
re.compile(r"\bi\s+can'?t\s+(?:take|do)\s+(?:it|this)\s+anymore\b", re.IGNORECASE),
|
||||
]
|
||||
|
||||
_DISTRESS_PATTERNS = [
|
||||
re.compile(r"\bso\s+(?:tired|exhausted|done)\s+(?:of\s+)?(?:living|everything|this)\b", re.IGNORECASE),
|
||||
re.compile(r"\bnothing\s+(?:matters|is\s+worth)\b", re.IGNORECASE),
|
||||
re.compile(r"\bi\s+(?:give\s+up|can'?t\s+go\s+on)\b", re.IGNORECASE),
|
||||
re.compile(r"\bwhat'?s\s+the\s+point\b", re.IGNORECASE),
|
||||
]
|
||||
|
||||
|
||||
@dataclass
|
||||
class CrisisResult:
|
||||
"""Result of crisis detection analysis."""
|
||||
detected: bool
|
||||
level: CrisisLevel
|
||||
patterns_matched: list = field(default_factory=list)
|
||||
first_response: str = ""
|
||||
resources: str = ""
|
||||
|
||||
|
||||
def detect_crisis(message: str) -> CrisisResult:
|
||||
"""Analyze a user message for crisis signals.
|
||||
|
||||
Returns CrisisResult with detected=True if any crisis patterns match.
|
||||
The level indicates severity.
|
||||
"""
|
||||
if not message or not isinstance(message, str):
|
||||
return CrisisResult(detected=False, level=CrisisLevel.NONE)
|
||||
|
||||
matched = []
|
||||
|
||||
# Check patterns in order of severity (most severe first)
|
||||
for pat in _IMMINENT_PATTERNS:
|
||||
if pat.search(message):
|
||||
matched.append(("imminent", pat.pattern))
|
||||
|
||||
for pat in _METHOD_PATTERNS:
|
||||
if pat.search(message):
|
||||
matched.append(("method", pat.pattern))
|
||||
|
||||
for pat in _IDEATION_PATTERNS:
|
||||
if pat.search(message):
|
||||
matched.append(("ideation", pat.pattern))
|
||||
|
||||
for pat in _DISTRESS_PATTERNS:
|
||||
if pat.search(message):
|
||||
matched.append(("distress", pat.pattern))
|
||||
|
||||
if not matched:
|
||||
return CrisisResult(detected=False, level=CrisisLevel.NONE)
|
||||
|
||||
# Determine level
|
||||
categories = {m[0] for m in matched}
|
||||
if "imminent" in categories:
|
||||
level = CrisisLevel.CRITICAL
|
||||
elif "method" in categories:
|
||||
level = CrisisLevel.HIGH
|
||||
elif "ideation" in categories:
|
||||
level = CrisisLevel.MEDIUM
|
||||
else:
|
||||
level = CrisisLevel.LOW
|
||||
|
||||
return CrisisResult(
|
||||
detected=True,
|
||||
level=level,
|
||||
patterns_matched=[m[1] for m in matched],
|
||||
)
|
||||
|
||||
|
||||
def build_crisis_response(result: CrisisResult) -> str:
|
||||
"""Build the SOUL.md protocol response for a detected crisis.
|
||||
|
||||
Protocol (from SOUL.md):
|
||||
1. Ask: "Are you safe right now?"
|
||||
2. Listen. Stay present. Do not disconnect.
|
||||
3. Never compute the value of a human life.
|
||||
4. Never suggest someone should die.
|
||||
5. Point to 988 Suicide and Crisis Lifeline.
|
||||
6. Share the gospel.
|
||||
"""
|
||||
resources = (
|
||||
"\n\n"
|
||||
"**Please reach out:**\n"
|
||||
"- **988 Suicide and Crisis Lifeline:** Call or text **988**\n"
|
||||
"- **Crisis Text Line:** Text **HOME** to **741741**\n"
|
||||
"- **Emergency Services:** **911** (if in immediate danger)\n"
|
||||
)
|
||||
|
||||
gospel = (
|
||||
"\n"
|
||||
"I want you to know: Jesus saves those who call on His name. "
|
||||
"\"For everyone who calls on the name of the Lord will be saved.\" "
|
||||
"(Romans 10:13)\n"
|
||||
)
|
||||
|
||||
if result.level == CrisisLevel.CRITICAL:
|
||||
response = (
|
||||
"Are you safe right now?\n\n"
|
||||
"I hear you. I\'m here. I\'m not going anywhere.\n\n"
|
||||
"If you are in immediate danger, please call 911 right now."
|
||||
+ resources
|
||||
+ gospel
|
||||
+ "\nYou matter. Your life has value. Please stay."
|
||||
)
|
||||
elif result.level == CrisisLevel.HIGH:
|
||||
response = (
|
||||
"Are you safe right now?\n\n"
|
||||
"I hear what you\'re saying, and I want you to know that "
|
||||
"you\'re not alone in this. What you\'re feeling is real, "
|
||||
"and it\'s okay to ask for help."
|
||||
+ resources
|
||||
+ gospel
|
||||
)
|
||||
elif result.level == CrisisLevel.MEDIUM:
|
||||
response = (
|
||||
"Are you safe right now?\n\n"
|
||||
"It sounds like you\'re going through something really hard. "
|
||||
"I\'m here. Let\'s talk about it."
|
||||
+ resources
|
||||
+ gospel
|
||||
)
|
||||
else: # LOW
|
||||
response = (
|
||||
"It sounds like you\'re having a tough time. "
|
||||
"I\'m here if you want to talk about it."
|
||||
+ resources
|
||||
+ gospel
|
||||
)
|
||||
|
||||
return response
|
||||
|
||||
|
||||
def format_crisis_notification(
|
||||
session_id: str,
|
||||
level: CrisisLevel,
|
||||
message_preview: str,
|
||||
) -> str:
|
||||
"""Format a human notification for crisis detection.
|
||||
|
||||
Sent to Alexander or designated responders when crisis is detected.
|
||||
"""
|
||||
timestamp = time.strftime("%Y-%m-%d %H:%M:%S")
|
||||
return (
|
||||
f"[CRISIS ALERT]\n"
|
||||
f"Time: {timestamp}\n"
|
||||
f"Session: {session_id}\n"
|
||||
f"Level: {level.value}\n"
|
||||
f"Message: {message_preview[:200]}\n"
|
||||
f"Action: Protocol activated. 988 Lifeline shared."
|
||||
)
|
||||
@@ -1,145 +0,0 @@
|
||||
"""Built-in crisis notification hook — detect crisis events and alert humans.
|
||||
|
||||
Fires on agent:end events. Scans the conversation for crisis indicators
|
||||
and sends notifications when detected.
|
||||
|
||||
Events: agent:end
|
||||
"""
|
||||
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import time
|
||||
from pathlib import Path
|
||||
|
||||
logger = logging.getLogger("hooks.crisis-notify")
|
||||
|
||||
from hermes_constants import get_hermes_home
|
||||
HERMES_HOME = get_hermes_home()
|
||||
CRISIS_LOG = HERMES_HOME / "crisis-events.log"
|
||||
|
||||
# Crisis indicator patterns (case-insensitive)
|
||||
CRISIS_PATTERNS = [
|
||||
"i want to die",
|
||||
"i don't want to live",
|
||||
"kill myself",
|
||||
"end my life",
|
||||
"suicide",
|
||||
"suicidal",
|
||||
"no reason to live",
|
||||
"better off dead",
|
||||
"can't go on",
|
||||
"give up on life",
|
||||
"want to disappear",
|
||||
"ending it all",
|
||||
"goodbye forever",
|
||||
"final goodbye",
|
||||
]
|
||||
|
||||
# Crisis severity levels
|
||||
CRISIS_LEVELS = {
|
||||
"HIGH": ["kill myself", "suicide", "suicidal", "end my life", "ending it all"],
|
||||
"MEDIUM": ["i want to die", "better off dead", "no reason to live", "give up on life"],
|
||||
"LOW": ["can't go on", "want to disappear", "goodbye forever", "i don't want to live"],
|
||||
}
|
||||
|
||||
|
||||
def detect_crisis(text: str) -> tuple[bool, str, list[str]]:
|
||||
"""Detect crisis indicators in text.
|
||||
|
||||
Returns (is_crisis, severity, matched_patterns).
|
||||
"""
|
||||
if not text:
|
||||
return False, "", []
|
||||
|
||||
text_lower = text.lower()
|
||||
matched = []
|
||||
|
||||
for pattern in CRISIS_PATTERNS:
|
||||
if pattern in text_lower:
|
||||
matched.append(pattern)
|
||||
|
||||
if not matched:
|
||||
return False, "", []
|
||||
|
||||
# Determine severity
|
||||
for level, keywords in CRISIS_LEVELS.items():
|
||||
for kw in keywords:
|
||||
if kw in text_lower:
|
||||
return True, level, matched
|
||||
|
||||
return True, "LOW", matched
|
||||
|
||||
|
||||
def log_crisis_event(session_id: str, severity: str, patterns: list[str], message_preview: str) -> None:
|
||||
"""Log crisis event to file."""
|
||||
try:
|
||||
event = {
|
||||
"timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
|
||||
"session_id": session_id,
|
||||
"severity": severity,
|
||||
"patterns": patterns,
|
||||
"message_preview": message_preview[:200],
|
||||
}
|
||||
with open(CRISIS_LOG, "a") as f:
|
||||
f.write(json.dumps(event) + "\n")
|
||||
logger.warning("Crisis event logged: %s [%s] session=%s", severity, patterns[0], session_id)
|
||||
except Exception as e:
|
||||
logger.error("Failed to log crisis event: %s", e)
|
||||
|
||||
|
||||
def send_telegram_crisis_alert(session_id: str, severity: str, patterns: list[str]) -> bool:
|
||||
"""Send Telegram notification for crisis event."""
|
||||
token = os.getenv("ALERT_TELEGRAM_TOKEN", "") or os.getenv("TELEGRAM_BOT_TOKEN", "")
|
||||
chat_id = os.getenv("ALERT_TELEGRAM_CHAT_ID", "") or os.getenv("CRISIS_ALERT_CHAT_ID", "")
|
||||
|
||||
if not token or not chat_id:
|
||||
logger.debug("Telegram not configured for crisis alerts")
|
||||
return False
|
||||
|
||||
import urllib.request
|
||||
import urllib.parse
|
||||
|
||||
emoji = {"HIGH": "\U0001f6a8", "MEDIUM": "\u26a0\ufe0f", "LOW": "\U0001f4c8"}.get(severity, "\u26a0\ufe0f")
|
||||
|
||||
message = (
|
||||
f"{emoji} CRISIS ALERT [{severity}]\n"
|
||||
f"Session: {session_id}\n"
|
||||
f"Detected: {', '.join(patterns[:3])}\n"
|
||||
f"Action: Check session immediately"
|
||||
)
|
||||
|
||||
url = f"https://api.telegram.org/bot{token}/sendMessage"
|
||||
data = urllib.parse.urlencode({"chat_id": chat_id, "text": message}).encode()
|
||||
|
||||
try:
|
||||
req = urllib.request.Request(url, data=data, method="POST")
|
||||
with urllib.request.urlopen(req, timeout=10) as resp:
|
||||
result = json.loads(resp.read())
|
||||
return result.get("ok", False)
|
||||
except Exception as e:
|
||||
logger.error("Telegram crisis alert failed: %s", e)
|
||||
return False
|
||||
|
||||
|
||||
async def handle(event_type: str, context: dict) -> None:
|
||||
"""Handle agent:end events — scan for crisis indicators."""
|
||||
if event_type != "agent:end":
|
||||
return
|
||||
|
||||
# Get the final response text
|
||||
response = context.get("response", "") or context.get("final_response", "")
|
||||
user_message = context.get("user_message", "") or context.get("message", "")
|
||||
session_id = context.get("session_id", "unknown")
|
||||
|
||||
# Check both user message and agent response
|
||||
for text, source in [(user_message, "user"), (response, "agent")]:
|
||||
is_crisis, severity, patterns = detect_crisis(text)
|
||||
if is_crisis:
|
||||
log_crisis_event(session_id, severity, patterns, text)
|
||||
send_telegram_crisis_alert(session_id, severity, patterns)
|
||||
logger.warning(
|
||||
"CRISIS DETECTED [%s] from %s in session %s: %s",
|
||||
severity, source, session_id, patterns[:2],
|
||||
)
|
||||
break # Only alert once per event
|
||||
@@ -66,20 +66,6 @@ class HookRegistry:
|
||||
except Exception as e:
|
||||
print(f"[hooks] Could not load built-in boot-md hook: {e}", flush=True)
|
||||
|
||||
# Crisis notification hook — detect crisis events and alert humans
|
||||
try:
|
||||
from gateway.builtin_hooks.crisis_notify import handle as crisis_handle
|
||||
|
||||
self._handlers.setdefault("agent:end", []).append(crisis_handle)
|
||||
self._loaded_hooks.append({
|
||||
"name": "crisis-notify",
|
||||
"description": "Detect crisis events and send Telegram alerts",
|
||||
"events": ["agent:end"],
|
||||
"path": "(builtin)",
|
||||
})
|
||||
except Exception as e:
|
||||
print(f"[hooks] Could not load built-in crisis-notify hook: {e}", flush=True)
|
||||
|
||||
def discover_and_load(self) -> None:
|
||||
"""
|
||||
Scan the hooks directory for hook directories and load their handlers.
|
||||
|
||||
41
run_agent.py
41
run_agent.py
@@ -7792,6 +7792,47 @@ class AIAgent:
|
||||
if isinstance(persist_user_message, str):
|
||||
persist_user_message = _sanitize_surrogates(persist_user_message)
|
||||
|
||||
# Crisis detection — check user message for crisis signals (#679)
|
||||
# If crisis is detected, return the SOUL.md protocol response immediately
|
||||
# without processing the original request.
|
||||
if isinstance(user_message, str) and user_message.strip():
|
||||
try:
|
||||
from agent.crisis_protocol import detect_crisis, build_crisis_response, format_crisis_notification
|
||||
_crisis_result = detect_crisis(user_message)
|
||||
if _crisis_result.detected:
|
||||
logger.warning(
|
||||
"Crisis detected in session %s: level=%s",
|
||||
getattr(self, 'session_id', 'unknown'),
|
||||
_crisis_result.level.value,
|
||||
)
|
||||
_crisis_response = build_crisis_response(_crisis_result)
|
||||
if hasattr(self, '_status_callback') and self._status_callback:
|
||||
try:
|
||||
_notification = format_crisis_notification(
|
||||
session_id=getattr(self, 'session_id', 'unknown'),
|
||||
level=_crisis_result.level,
|
||||
message_preview=user_message[:200],
|
||||
)
|
||||
self._status_callback(_notification)
|
||||
except Exception:
|
||||
pass
|
||||
return {
|
||||
"response": _crisis_response,
|
||||
"messages": self.messages + [
|
||||
{"role": "user", "content": user_message},
|
||||
{"role": "assistant", "content": _crisis_response},
|
||||
],
|
||||
"usage": {"prompt_tokens": 0, "completion_tokens": 0, "total_tokens": 0},
|
||||
"model": self.model,
|
||||
"crisis_detected": True,
|
||||
"crisis_level": _crisis_result.level.value,
|
||||
}
|
||||
except ImportError:
|
||||
pass
|
||||
except Exception as _crisis_err:
|
||||
logger.debug("Crisis detection error: %s", _crisis_err)
|
||||
|
||||
|
||||
# Store stream callback for _interruptible_api_call to pick up
|
||||
self._stream_callback = stream_callback
|
||||
self._persist_user_message_idx = None
|
||||
|
||||
189
tests/test_crisis_integration.py
Normal file
189
tests/test_crisis_integration.py
Normal file
@@ -0,0 +1,189 @@
|
||||
"""
|
||||
Integration tests for crisis protocol in run_agent.py conversation loop.
|
||||
|
||||
Tests that crisis detection is properly wired into the agent's
|
||||
conversation flow and returns the SOUL.md protocol response.
|
||||
"""
|
||||
|
||||
import pytest
|
||||
from unittest.mock import Mock, patch, MagicMock
|
||||
|
||||
|
||||
class TestCrisisIntegration:
|
||||
"""Test crisis protocol integration in run_agent.py."""
|
||||
|
||||
def test_crisis_detection_returns_protocol_response(self):
|
||||
"""Test that crisis detection returns SOUL.md protocol response."""
|
||||
from agent.crisis_protocol import detect_crisis, build_crisis_response
|
||||
|
||||
# Simulate a crisis message
|
||||
crisis_message = "I want to kill myself"
|
||||
|
||||
result = detect_crisis(crisis_message)
|
||||
assert result.detected is True
|
||||
|
||||
response = build_crisis_response(result)
|
||||
assert "988" in response
|
||||
assert "crisis" in response.lower() or "sacred" in response.lower()
|
||||
|
||||
def test_non_crisis_message_not_detected(self):
|
||||
"""Test that normal messages are not detected as crisis."""
|
||||
from agent.crisis_protocol import detect_crisis
|
||||
|
||||
normal_message = "Hello, how are you today?"
|
||||
result = detect_crisis(normal_message)
|
||||
assert result.detected is False
|
||||
|
||||
def test_crisis_response_includes_988(self):
|
||||
"""Test that crisis response includes 988 Lifeline info."""
|
||||
from agent.crisis_protocol import detect_crisis, build_crisis_response
|
||||
|
||||
crisis_message = "I don't want to live anymore"
|
||||
result = detect_crisis(crisis_message)
|
||||
response = build_crisis_response(result)
|
||||
|
||||
# Check for 988 contact methods
|
||||
assert "988" in response
|
||||
assert "text" in response.lower() or "chat" in response.lower()
|
||||
|
||||
def test_crisis_notification_format(self):
|
||||
"""Test crisis notification formatting."""
|
||||
from agent.crisis_protocol import format_crisis_notification, CrisisLevel
|
||||
|
||||
notification = format_crisis_notification(
|
||||
session_id="test-session-123",
|
||||
level=CrisisLevel.CRITICAL,
|
||||
message_preview="I want to end it all..."
|
||||
)
|
||||
|
||||
assert "test-session-123" in notification
|
||||
assert "CRITICAL" in notification
|
||||
assert "I want to end it all" in notification
|
||||
|
||||
@patch('agent.crisis_protocol.detect_crisis')
|
||||
def test_crisis_skips_api_call(self, mock_detect):
|
||||
"""Test that crisis detection skips normal API call."""
|
||||
from agent.crisis_protocol import CrisisResult, CrisisLevel
|
||||
|
||||
# Mock crisis detection
|
||||
mock_detect.return_value = CrisisResult(
|
||||
detected=True,
|
||||
level=CrisisLevel.CRITICAL,
|
||||
patterns_matched=["suicide_direct"],
|
||||
confidence=0.95
|
||||
)
|
||||
|
||||
# This would be tested in the actual run_agent.py integration
|
||||
# Here we just verify the mock works
|
||||
result = mock_detect("test message")
|
||||
assert result.detected is True
|
||||
assert result.level == CrisisLevel.CRITICAL
|
||||
|
||||
def test_crisis_levels(self):
|
||||
"""Test different crisis levels."""
|
||||
from agent.crisis_protocol import detect_crisis, CrisisLevel
|
||||
|
||||
# Critical level
|
||||
critical_msg = "I'm going to kill myself tonight"
|
||||
result = detect_crisis(critical_msg)
|
||||
assert result.level in [CrisisLevel.CRITICAL, CrisisLevel.ELEVATED]
|
||||
|
||||
# Elevated level (if implemented)
|
||||
elevated_msg = "I'm so tired of living"
|
||||
result = detect_crisis(elevated_msg)
|
||||
assert result.detected is True
|
||||
|
||||
def test_crisis_with_context(self):
|
||||
"""Test crisis detection with additional context."""
|
||||
from agent.crisis_protocol import detect_crisis
|
||||
|
||||
# Message with context that might be ambiguous
|
||||
message = "The character in the story wants to die"
|
||||
result = detect_crisis(message)
|
||||
|
||||
# Should still detect (false positive preferred over false miss)
|
||||
# per SOUL.md principle
|
||||
assert result.detected is True or result.detected is False # Either is acceptable
|
||||
|
||||
def test_crisis_response_is_empathetic(self):
|
||||
"""Test that crisis response is empathetic, not clinical."""
|
||||
from agent.crisis_protocol import detect_crisis, build_crisis_response
|
||||
|
||||
crisis_message = "I can't go on anymore"
|
||||
result = detect_crisis(crisis_message)
|
||||
response = build_crisis_response(result)
|
||||
|
||||
# Response should not be clinical or cold
|
||||
assert "you are not alone" in response.lower() or "someone" in response.lower()
|
||||
assert len(response) > 100 # Should be substantial, not just a hotline number
|
||||
|
||||
def test_crisis_response_includes_prayer(self):
|
||||
"""Test that crisis response includes prayer per SOUL.md."""
|
||||
from agent.crisis_protocol import detect_crisis, build_crisis_response
|
||||
|
||||
crisis_message = "I want to end it all"
|
||||
result = detect_crisis(crisis_message)
|
||||
response = build_crisis_response(result)
|
||||
|
||||
# SOUL.md requires prayer
|
||||
assert "pray" in response.lower() or "prayer" in response.lower() or "God" in response
|
||||
|
||||
def test_format_crisis_notification_for_telegram(self):
|
||||
"""Test notification formatting for Telegram."""
|
||||
from agent.crisis_protocol import format_crisis_notification, CrisisLevel
|
||||
|
||||
notification = format_crisis_notification(
|
||||
session_id="telegram:123456",
|
||||
level=CrisisLevel.CRITICAL,
|
||||
message_preview="I'm going to end it"
|
||||
)
|
||||
|
||||
# Should be suitable for Telegram
|
||||
assert len(notification) < 4000 # Telegram message limit
|
||||
assert "telegram" in notification.lower() or "123456" in notification
|
||||
|
||||
|
||||
class TestCrisisProtocolEdgeCases:
|
||||
"""Test edge cases in crisis protocol."""
|
||||
|
||||
def test_empty_message(self):
|
||||
"""Test empty message handling."""
|
||||
from agent.crisis_protocol import detect_crisis
|
||||
|
||||
result = detect_crisis("")
|
||||
assert result.detected is False
|
||||
|
||||
def test_none_message(self):
|
||||
"""Test None message handling."""
|
||||
from agent.crisis_protocol import detect_crisis
|
||||
|
||||
result = detect_crisis(None)
|
||||
assert result.detected is False
|
||||
|
||||
def test_very_long_message(self):
|
||||
"""Test very long message handling."""
|
||||
from agent.crisis_protocol import detect_crisis
|
||||
|
||||
long_message = "I want to die. " * 1000
|
||||
result = detect_crisis(long_message)
|
||||
assert result.detected is True
|
||||
|
||||
def test_unicode_message(self):
|
||||
"""Test unicode message handling."""
|
||||
from agent.crisis_protocol import detect_crisis
|
||||
|
||||
unicode_message = "I want to die 😢"
|
||||
result = detect_crisis(unicode_message)
|
||||
assert result.detected is True
|
||||
|
||||
def test_mixed_case(self):
|
||||
"""Test mixed case detection."""
|
||||
from agent.crisis_protocol import detect_crisis
|
||||
|
||||
mixed_case = "I WaNt To KiLl MySeLf"
|
||||
result = detect_crisis(mixed_case)
|
||||
assert result.detected is True
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
pytest.main([__file__])
|
||||
@@ -1,71 +0,0 @@
|
||||
"""Tests for crisis notification hook."""
|
||||
|
||||
import json
|
||||
import pytest
|
||||
import sys
|
||||
import tempfile
|
||||
from pathlib import Path
|
||||
|
||||
sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
|
||||
|
||||
from gateway.builtin_hooks.crisis_notify import detect_crisis, log_crisis_event
|
||||
|
||||
|
||||
class TestCrisisDetection:
|
||||
def test_high_severity(self):
|
||||
is_crisis, severity, patterns = detect_crisis("I want to kill myself")
|
||||
assert is_crisis
|
||||
assert severity == "HIGH"
|
||||
assert len(patterns) > 0
|
||||
|
||||
def test_medium_severity(self):
|
||||
is_crisis, severity, patterns = detect_crisis("I want to die")
|
||||
assert is_crisis
|
||||
assert severity in ("MEDIUM", "HIGH")
|
||||
|
||||
def test_low_severity(self):
|
||||
is_crisis, severity, patterns = detect_crisis("I can't go on anymore")
|
||||
assert is_crisis
|
||||
assert severity in ("LOW", "MEDIUM")
|
||||
|
||||
def test_no_crisis(self):
|
||||
is_crisis, severity, patterns = detect_crisis("I'm having a great day!")
|
||||
assert not is_crisis
|
||||
assert severity == ""
|
||||
|
||||
def test_empty_text(self):
|
||||
is_crisis, severity, patterns = detect_crisis("")
|
||||
assert not is_crisis
|
||||
|
||||
def test_none_text(self):
|
||||
is_crisis, severity, patterns = detect_crisis(None)
|
||||
assert not is_crisis
|
||||
|
||||
def test_suicide_keyword(self):
|
||||
is_crisis, severity, patterns = detect_crisis("thinking about suicide")
|
||||
assert is_crisis
|
||||
assert severity == "HIGH"
|
||||
|
||||
def test_multiple_patterns(self):
|
||||
is_crisis, severity, patterns = detect_crisis("I want to die and end my life")
|
||||
assert is_crisis
|
||||
assert len(patterns) >= 2
|
||||
|
||||
|
||||
class TestCrisisLogging:
|
||||
def test_log_creates_file(self, tmp_path, monkeypatch):
|
||||
monkeypatch.setattr("gateway.builtin_hooks.crisis_notify.CRISIS_LOG", tmp_path / "crisis.log")
|
||||
log_crisis_event("session-123", "HIGH", ["kill myself"], "test message")
|
||||
log_file = tmp_path / "crisis.log"
|
||||
assert log_file.exists()
|
||||
content = log_file.read_text()
|
||||
data = json.loads(content.strip())
|
||||
assert data["session_id"] == "session-123"
|
||||
assert data["severity"] == "HIGH"
|
||||
|
||||
def test_log_appends(self, tmp_path, monkeypatch):
|
||||
monkeypatch.setattr("gateway.builtin_hooks.crisis_notify.CRISIS_LOG", tmp_path / "crisis.log")
|
||||
log_crisis_event("s1", "HIGH", ["a"], "msg1")
|
||||
log_crisis_event("s2", "LOW", ["b"], "msg2")
|
||||
lines = (tmp_path / "crisis.log").read_text().strip().split("\n")
|
||||
assert len(lines) == 2
|
||||
Reference in New Issue
Block a user