app/translate_chain.py

from langchain_core.prompts import ChatPromptTemplate
from app.models.model import LLM
from app.utils.prompts import translate_prompts

def get_translate_chain(language: str = "English"):
    system_prompt = translate_prompts[language]
    prompt = ChatPromptTemplate.from_messages([
        ("system", system_prompt),
        ("human", "{input}"),
    ])
    
    return LLM, prompt