nemoguardrails.library.trend_micro.actions

Module Contents

Classes

Name	Description
`Guard`	Represents a guard entity with a single string attribute.
`GuardResult`	Represents the result of a guard analysis, specifying the action to take and the reason.

Functions

Name	Description
`get_config`	Retrieves the TrendMicroRailConfig from the provided RailsConfig object.
`trend_ai_guard`	Custom action to invoke the Trend Micro AI Guard API.
`trend_ai_guard_mapping`	Convert Trend Micro result to boolean for flow logic.

Data

log

API

class nemoguardrails.library.trend_micro.actions.Guard()

Bases: BaseModel

Represents a guard entity with a single string attribute.

prompt

str

class nemoguardrails.library.trend_micro.actions.GuardResult()

Bases: BaseModel

Represents the result of a guard analysis, specifying the action to take and the reason.

action

Literal['Block', 'Allow']

blocked

bool

reason

str

nemoguardrails.library.trend_micro.actions.GuardResult.set_blocked(
    values
)

nemoguardrails.library.trend_micro.actions.GuardResult.validate_action(
    v
)

nemoguardrails.library.trend_micro.actions.get_config(
    config: nemoguardrails.rails.llm.config.RailsConfig
) -> nemoguardrails.rails.llm.config.TrendMicroRailConfig

Retrieves the TrendMicroRailConfig from the provided RailsConfig object.

Parameters:

config

RailsConfig

The Rails configuration object containing possible

Returns: TrendMicroRailConfig

The Trend Micro configuration, either from the provided

nemoguardrails.library.trend_micro.actions.trend_ai_guard(
    config: nemoguardrails.rails.llm.config.RailsConfig,
    text: str
)

async

Custom action to invoke the Trend Micro AI Guard API.

nemoguardrails.library.trend_micro.actions.trend_ai_guard_mapping(
    result: nemoguardrails.library.trend_micro.actions.GuardResult
) -> bool

Convert Trend Micro result to boolean for flow logic.

nemoguardrails.library.trend_micro.actions.log = logging.getLogger(__name__)