in src/co_op_translator/core/llm/text_translator.py [0:0]
def translate_text(self, text, target_language):
"""
Translate a given text into the target language using the LLM API.
Args:
text (str): The text to be translated.
target_language (str): The target language code.
Returns:
str: The translated text.
"""
prompt = f"Translate the following text into {target_language}:\n\n{text}"
response = self.client.chat.completions.create(
model=self.get_model_name(),
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": prompt},
],
max_tokens=2000,
)
translated_text = remove_code_backticks(response.choices[0].message.content)
return translated_text