from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain_ollama import ChatOllama


if __name__ == "__main__":
    llm = ChatOllama(model="maternion/hy-mt2:1.8b")
    template = ChatPromptTemplate.from_template("""
                    翻译任务：
                    - 源语言：{srcLanguage}
                    - 目标语言：{targetLanguage}
                    - 文本领域：{domain}
                    - 翻译风格：{style}
                    待翻译内容：
                    {text}
                    """)

    parser = StrOutputParser()

    chain = template | llm | parser

    result = chain.invoke(input={"srcLanguage": "韩文", "targetLanguage": "简体中文",
                                 "domain": "计算机行业", "style": "正式专业",
                                 "text": "API 통신 Logging 파일"})

    print(result)