diff --git a/translator.py b/translator.py new file mode 100644 index 0000000..3601718 --- /dev/null +++ b/translator.py @@ -0,0 +1,19 @@ +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