Study-with-ChampAI / agents /language_agent.py
SolusOps's picture
feat: agents package
dc124db verified
Raw
History Blame Contribute Delete
705 Bytes
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()