wony617 commited on
Commit
4f7e3cd
·
1 Parent(s): aa4594e

Fix scaffold issue when translating with AWS BedRock

Browse files
Files changed (1) hide show
  1. agent/workflow.py +3 -0
agent/workflow.py CHANGED
@@ -94,6 +94,9 @@ def translate_docs(lang: str, file_path: str, additional_instruction: str = "",
94
  callback_result, translated_content = llm_translate(to_translate_with_prompt)
95
  print("translated_content:\n")
96
  print(translated_content)
 
 
 
97
  # step 4. Add scaffold to translation result
98
  translated_doc = fill_scaffold(content, to_translate, translated_content)
99
  print("translated_doc:\n")
 
94
  callback_result, translated_content = llm_translate(to_translate_with_prompt)
95
  print("translated_content:\n")
96
  print(translated_content)
97
+ if translated_content.startswith("```md\n") and translated_content.endswith("```"):
98
+ print("Satisfied translated_content.startswith ``` md")
99
+ translated_content = translated_content[5:-3].strip()
100
  # step 4. Add scaffold to translation result
101
  translated_doc = fill_scaffold(content, to_translate, translated_content)
102
  print("translated_doc:\n")