From 8aa0d6cb8cfbad4a7c10432a9f208b861d1da2cd Mon Sep 17 00:00:00 2001 From: xunbu Date: Fri, 26 Dec 2025 23:07:24 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9C=AF=E8=AF=AD=E8=A1=A8=E4=B8=8D=E5=8C=BA?= =?UTF-8?q?=E5=88=86=E5=A4=A7=E5=B0=8F=E5=86=99?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docutranslate/glossary/glossary.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/docutranslate/glossary/glossary.py b/docutranslate/glossary/glossary.py index 33d57c7..6799711 100644 --- a/docutranslate/glossary/glossary.py +++ b/docutranslate/glossary/glossary.py @@ -8,13 +8,17 @@ from docutranslate.ir.document import Document class Glossary: - def __init__(self, glossary_dict: dict[str:str] = None): - self.glossary_dict = glossary_dict + def __init__(self, glossary_dict: dict[str,str] = None): + if 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(): - if src.strip() not in self.glossary_dict: - self.glossary_dict[src.strip()] = dst + if src.strip().lower() not in self.glossary_dict: + self.glossary_dict[src.strip().lower()] = dst def append_system_prompt(self, text: str): flag = False @@ -23,9 +27,7 @@ class Glossary: Here is the reference glossary: """ for src, dst in self.glossary_dict.items(): - text=re.sub(r'\s+', '', text)#去除所有空白字符 - src=re.sub(r'\s+', '', src)#去除所有空白字符 - if src in text: + if src.lower() in text.lower(): prompt += f"{src}=>{dst}\n" flag = True prompt += "Glossary ends\n"