feat:单独对名字进行NER

This commit is contained in:
oliviamn 2025-06-25 01:31:12 +08:00
parent ee95f1daa7
commit 5ddef90e8b
1 changed files with 82 additions and 0 deletions

View File

@ -78,4 +78,86 @@ def get_masking_mapping_prompt(text: str) -> str:
{{}}
""")
return prompt.format(text=text)
def get_masking_mapping_prompt_v2(text: str) -> str:
"""
Returns a prompt that generates a mapping of original names/companies to their masked versions.
Args:
text (str): The input text to be analyzed for masking
Returns:
str: The formatted prompt that will generate a mapping dictionary
"""
prompt = textwrap.dedent("""
请根据以下脱敏规则对给定文本进行分析并生成脱敏映射mapJSON格式
脱敏规则说明
1.人名保留姓名变为""同姓者按"某1""某2"依次编号律师姓名审判人员姓名不脱敏
2.公司名替换为大写英文字母A公司B公司等律师事务所不脱敏
3.地址仅保留区级以上地址删除详细位置
4.英文人名保留姓名首字母其余替换为"***"
5.英文公司名替换为所属行业名称的英文大写形式
6.项目名替换为小写英文字母a项目b项目等
7.案号具体案号部分替换为"** *"
8.身份证号替换为6个"X"
9.统一社会信用代码替换为8个"X"
输入文本
{text}
输出要求
请生成一个JSON格式的映射map包含以下结构
{{
"原文1": "脱敏后1",
"原文2": "脱敏后2",
...
}}
如无需要输出的映射请输出空json如下:
{{}}
注意事项
请准确识别文本中的各类敏感信息
同一类别的多个实体请在对应类别下全部列出
如果文本中没有某类敏感信息可以省略该类别
请确保脱敏后的文本符合上述规则要求
""")
return prompt.format(text=text)
def get_ner_name_prompt(text: str) -> str:
"""
Returns a prompt that generates a mapping of original names/companies to their masked versions.
Args:
text (str): The input text to be analyzed for masking
Returns:
str: The formatted prompt that will generate a mapping dictionary
"""
prompt = textwrap.dedent("""
你是一个专业的法律文本实体识别助手请从以下文本中抽取出所有需要脱敏的敏感信息并按照指定的类别进行分类请严格按照JSON格式输出结果
实体类别包括:
- 人名 (不包括律师法官书记员检察官等公职人员)
- 英文人名
待处理文本:
{text}
输出格式:
{{
"entities": [
{{"text": "原始文本内容", "type": "实体类别"}},
...
]
}}
请严格按照JSON格式输出结果
""")
return prompt.format(text=text)