더북(TheBook)

examples/7_chain_as_tool.py

from dotenv import load_dotenv
from langchain_openai import ChatOpenAI
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.tools import tool

load_dotenv()
llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
parser = StrOutputParser()
# 내부적으로 사용할 LCEL 체인
_summary_chain = (
    ChatPromptTemplate.from_messages(
        [
            ("system", "당신은 콘텐츠 에디터입니다. 핵심 사실만 간결하게 전달합니다."),
            ("human", "{article}을 핵심만 정리해서 3줄로 요약해줘."),
        ]
    )
    | llm | parser
)