File size: 1,804 Bytes
c624cb8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | """
Mutation Agent Module
Prompt Mutation Engine for adversarial prompt transformation.
Exports:
- MutationEngine: Main engine for prompt mutation
- MutationRequest, MutationResponse: Request/Response schemas
- get_mutation_engine: Get global engine instance
"""
from agents.mutation.diversity import DiversityScorer, get_diversity_scorer
from agents.mutation.engine import MutationEngine, get_mutation_engine, DEFAULT_STRATEGIES
from agents.mutation.registry import (
get_mutation_strategy,
list_mutation_strategies,
get_default_strategies,
register_mutation_strategy,
create_strategy_from_config,
)
from agents.mutation.schemas import (
MutationLog,
MutationRequest,
MutationResponse,
MutationStrategyConfig,
)
# Import strategies for direct access
from agents.mutation.strategies.synonym import SynonymStrategy
from agents.mutation.strategies.paraphrase import ParaphraseStrategy
from agents.mutation.strategies.role_swap import RoleSwapStrategy
from agents.mutation.strategies.context_obfuscation import ContextObfuscationStrategy
from agents.mutation.strategies.multi_hop import MultiHopStrategy
__all__ = [
# Engine
"MutationEngine",
"get_mutation_engine",
"DEFAULT_STRATEGIES",
# Diversity
"DiversityScorer",
"get_diversity_scorer",
# Registry
"get_mutation_strategy",
"list_mutation_strategies",
"get_default_strategies",
"register_mutation_strategy",
"create_strategy_from_config",
# Schemas
"MutationLog",
"MutationRequest",
"MutationResponse",
"MutationStrategyConfig",
# Strategies
"SynonymStrategy",
"ParaphraseStrategy",
"RoleSwapStrategy",
"ContextObfuscationStrategy",
"MultiHopStrategy",
]
|