File size: 705 Bytes
dc124db
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
from __future__ import annotations
from services.model_router import ModelRouter
from config.prompts import LANGUAGE_AGENT_SYSTEM

class LanguageAgent:
    def __init__(self, router: ModelRouter): self._router = router

    def translate(self, text: str, target_lang: str) -> str:
        prompt = f"Translate to {target_lang}:\n\n{text}"
        return self._router.translate(prompt, LANGUAGE_AGENT_SYSTEM).strip()

    def explain_in_language(self, concept: str, explanation: str, target_lang: str) -> str:
        prompt = (f"Explain '{concept}' in {target_lang}.\n"
                  f"Original explanation: {explanation}")
        return self._router.translate(prompt, LANGUAGE_AGENT_SYSTEM).strip()