easy-web-summarizer/translator.py

20 lines
721 B
Python

from langchain.chains.llm import LLMChain
from langchain.prompts import PromptTemplate
from langchain_community.chat_models import ChatOllama
def setup_translator_chain():
"""Setup the translation chain with a prompt template and ChatOllama."""
prompt_template = PromptTemplate(
template="""As a professional translator, provide a detailed and comprehensive translation of the provided text into turkish, ensuring that the translation is accurate, coherent, and faithful to the original text.
"{text}"
DETAILED TRANSLATION:""",
input_variables=["text"],
)
llm = ChatOllama(model="llama3")
llm_chain = LLMChain(llm=llm, prompt=prompt_template)
return llm_chain