scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-10-07 03:00:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

fix(translate_ts): correct translation output handling

Browse source
This commit is contained in:
jinli.yl 2026-02-01 15:32:23 +08:00
parent 2233aa8403
commit b6d3ce69a7
1 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
reme/workflow/gallery/translate_ts.py
View file

@ -19,8 +19,6 @@ class TranslateTs(BaseOp):
async def execute_single_file(self, ts_file: Path):
"""Translate a single ts file."""
ts_code = ts_file.read_text(encoding="utf-8")
python_file = ts_file.with_suffix(".py")
logger.info(f"Translating {ts_file}")
def parse_python(assistant_message: Message):
@ -29,9 +27,9 @@ class TranslateTs(BaseOp):
python_code = python_code.split("```python", 1)[1]
python_code_split = python_code.split("```")
python_code = "```".join(python_code_split[:-1])
return python_code.strip()
return python_code.strip(), assistant_message.content.strip()
output_code = await self.llm.chat(
output = await self.llm.chat(
messages=[
Message(
role=Role.USER,
@ -41,8 +39,9 @@ class TranslateTs(BaseOp):
callback_fn=parse_python,
)
python_file.write_text(output_code, encoding="utf-8")
logger.info(f"Writing {python_file}")
ts_file.with_suffix(".py").write_text(output[0], encoding="utf-8")
ts_file.with_suffix(".py").write_text(output[1], encoding="utf-8")
logger.info(f"Translate {ts_file} complete.")
async def execute(self):
"""Execute the operation."""