def agent_translate(text: str) -> str: """A tool that translate text with next logic: 1) If text starts with 'PL:', translate the rest to Polish. 2) If detected language == 'ru', translate to English. 3) Else translate to Russian. and return translated text Args: text: text in russian/english or polish languages. """ if text.startswith("PL:"): original = text[3:].strip() target_lang = "PL" else: detected_lang = detect(text) if detected_lang == "ru": target_lang = "EN" original = text else: target_lang = "RU" original = text try: model_name = "facebook/m2m100_418M" tokenizer = M2M100Tokenizer.from_pretrained(model_name) translation_model = M2M100ForConditionalGeneration.from_pretrained(model_name) inputs = tokenizer(original, return_tensors="pt") forced_bos_token_id = tokenizer.get_lang_id(target_lang.lower()) generated_tokens = translation_model.generate( **inputs, forced_bos_token_id=forced_bos_token_id ) translated_text = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0] return translated_text except Exception as e: return f"Error during translating: { str(e)}"