创建了新的反注入

This commit is contained in:
明天好像没什么
2025-11-09 12:31:38 +08:00
parent 626dbfe998
commit 6a5648ba07
36 changed files with 1930 additions and 2600 deletions

View File

@@ -13,7 +13,6 @@ from src.common.logger import get_logger
from src.config.config_base import ValidatedConfigBase
from src.config.official_configs import (
AffinityFlowConfig,
AntiPromptInjectionConfig,
BotConfig,
ChatConfig,
ChineseTypoConfig,
@@ -397,9 +396,6 @@ class Config(ValidatedConfigBase):
command: CommandConfig = Field(..., description="命令系统配置")
# 有默认值的字段放在后面
anti_prompt_injection: AntiPromptInjectionConfig = Field(
default_factory=lambda: AntiPromptInjectionConfig(), description="反提示注入配置"
)
video_analysis: VideoAnalysisConfig = Field(
default_factory=lambda: VideoAnalysisConfig(), description="视频分析配置"
)

View File

@@ -646,28 +646,6 @@ class WebSearchConfig(ValidatedConfigBase):
search_strategy: Literal["fallback", "single", "parallel"] = Field(default="single", description="搜索策略")
class AntiPromptInjectionConfig(ValidatedConfigBase):
"""LLM反注入系统配置类"""
enabled: bool = Field(default=True, description="启用")
enabled_LLM: bool = Field(default=True, description="启用LLM")
enabled_rules: bool = Field(default=True, description="启用规则")
process_mode: str = Field(default="lenient", description="处理模式")
whitelist: list[list[str]] = Field(default_factory=list, description="白名单")
llm_detection_enabled: bool = Field(default=True, description="启用LLM检测")
llm_model_name: str = Field(default="anti_injection", description="LLM模型名称")
llm_detection_threshold: float = Field(default=0.7, description="LLM检测阈值")
cache_enabled: bool = Field(default=True, description="启用缓存")
cache_ttl: int = Field(default=3600, description="缓存TTL")
max_message_length: int = Field(default=4096, description="最大消息长度")
stats_enabled: bool = Field(default=True, description="启用统计信息")
auto_ban_enabled: bool = Field(default=True, description="启用自动禁用")
auto_ban_violation_threshold: int = Field(default=3, description="自动禁用违规阈值")
auto_ban_duration_hours: int = Field(default=2, description="自动禁用持续时间(小时)")
shield_prefix: str = Field(default="🛡️ ", description="保护前缀")
shield_suffix: str = Field(default=" 🛡️", description="保护后缀")
class ContextGroup(ValidatedConfigBase):
"""
上下文共享组配置