17 lines
550 B
Python
17 lines
550 B
Python
|
"""投顾 Agent 输出合规校验。"""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from common.common_const import ERR_CODE_LLM_ERROR
|
||
|
|
from utils.exceptions import ApiError
|
||
|
|
|
||
|
|
|
||
|
|
def find_sensitive_words(content: str, words: list[str]) -> list[str]:
|
||
|
|
return [word for word in dict.fromkeys(words) if word and word in content]
|
||
|
|
|
||
|
|
|
||
|
|
def ensure_safe_content(content: str, words: list[str]) -> bool:
|
||
|
|
matched = find_sensitive_words(content, words)
|
||
|
|
if matched:
|
||
|
|
raise ApiError(ERR_CODE_LLM_ERROR, "AI输出包含敏感或违规表述")
|
||
|
|
return True
|