Configuring Custom Actions

1
from typing import Optional
2
from nemoguardrails.actions import action
3
4
@action()
5
async def check_custom_policy(context: Optional[dict] = None):
6
    """Check if the input complies with custom policy."""
7
    user_message = context.get("last_user_message", "")
8
9
    # Custom validation logic
10
    forbidden_words = ["spam", "phishing"]
11
    for word in forbidden_words:
12
        if word in user_message.lower():
13
            return False
14
15
    return True
16
17
@action(name="fetch_user_data")
18
async def get_user_info(user_id: str):
19
    """Fetch user data from external service."""
20
    # External API call
21
    return {"user_id": user_id, "status": "active"}

1
from typing import Optional
2
from nemoguardrails.actions import action
3
4
@action(is_system_action=True)
5
async def check_blocked_terms(context: Optional[dict] = None):
6
    """Check if bot response contains blocked terms."""
7
    bot_response = context.get("bot_message", "")
8
9
    blocked_terms = ["confidential", "proprietary", "secret"]
10
11
    for term in blocked_terms:
12
        if term in bot_response.lower():
13
            return True  # Term found, block the response
14
15
    return False  # No blocked terms found