From 172f4ba9fdc21979030876fe628b89ab87bbb9d3 Mon Sep 17 00:00:00 2001 From: xunbu Date: Wed, 27 Aug 2025 21:23:25 +0800 Subject: [PATCH] fix bug --- docutranslate/glossary/glossary.py | 5 +++-- docutranslate/translator/ai_translator/base.py | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/docutranslate/glossary/glossary.py b/docutranslate/glossary/glossary.py index 8c9dbc1..c8ce131 100644 --- a/docutranslate/glossary/glossary.py +++ b/docutranslate/glossary/glossary.py @@ -1,5 +1,5 @@ class Glossary: - def __init__(self,glossary_dict:dict[str:str]|None=None): + def __init__(self,glossary_dict:dict[str:str]=None): self.glossary_dict=glossary_dict def update(self,update_dict:dict[str:str]): @@ -9,7 +9,8 @@ class Glossary: def append_system_prompt(self,text:str): prompt="\n以下为参考术语表:\n" - for src,dst in self.glossary_dict: + for src,dst in self.glossary_dict.items(): if src in text: prompt+=f"{src}=>{dst}\n" prompt+="术语表结束\n" + return prompt diff --git a/docutranslate/translator/ai_translator/base.py b/docutranslate/translator/ai_translator/base.py index 5a85c4b..d79ca9c 100644 --- a/docutranslate/translator/ai_translator/base.py +++ b/docutranslate/translator/ai_translator/base.py @@ -1,5 +1,5 @@ from abc import abstractmethod -from dataclasses import dataclass +from dataclasses import dataclass, field from typing import TypeVar from docutranslate.agents.agent import ThinkingMode