Leon/docutranslate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

术语表不区分大小写

Browse Source
This commit is contained in:
xunbu
2025-12-26 23:07:24 +08:00
parent 583615ebde
commit 8aa0d6cb8c
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
docutranslate/glossary/glossary.py
View File

@@ -8,13 +8,17 @@ from docutranslate.ir.document import Document
class Glossary: class Glossary:
def __init__(self, glossary_dict: dict[str:str] = None): def __init__(self, glossary_dict: dict[str,str] = None):
if glossary_dict:
self.glossary_dict = glossary_dict self.glossary_dict = glossary_dict
else:
self.glossary_dict={}
def update(self, update_dict: dict[str:str]):
def update(self, update_dict: dict[str,str]):
for src, dst in update_dict.items(): for src, dst in update_dict.items():
if src.strip() not in self.glossary_dict: if src.strip().lower() not in self.glossary_dict:
self.glossary_dict[src.strip()] = dst self.glossary_dict[src.strip().lower()] = dst
def append_system_prompt(self, text: str): def append_system_prompt(self, text: str):
flag = False flag = False
@@ -23,9 +27,7 @@ class Glossary:
Here is the reference glossary: Here is the reference glossary:
""" """
for src, dst in self.glossary_dict.items(): for src, dst in self.glossary_dict.items():
text=re.sub(r'\s+', '', text)#去除所有空白字符 if src.lower() in text.lower():
src=re.sub(r'\s+', '', src)#去除所有空白字符
if src in text:
prompt += f"{src}=>{dst}\n" prompt += f"{src}=>{dst}\n"
flag = True flag = True
prompt += "Glossary ends\n" prompt += "Glossary ends\n"