First_agent_template / tools /translate_text.py
alexander1198's picture
Update tools/translate_text.py
2fb7227 verified
raw
history blame contribute delete
697 Bytes
from typing import TypedDict
import requests
class InputSchema(TypedDict):
text: str
source_lang: str
target_lang: str
class OutputSchema(TypedDict):
translated_text: str
def translate_text(inputs: InputSchema) -> OutputSchema:
url = "https://libretranslate.de/translate"
payload = {
"q": inputs["text"],
"source": inputs["source_lang"],
"target": inputs["target_lang"],
"format": "text"
}
headers = {"Content-Type": "application/json"}
response = requests.post(url, json=payload, headers=headers)
response.raise_for_status()
translated = response.json()["translatedText"]
return {"translated_text": translated}