실습 5 — RunnableParallel 로 요약과 키워드 한 번에 뽑기

from langchain_core.language_models.fake_chat_models import FakeListChatModel
from langchain_core.output_parsers import CommaSeparatedListOutputParser, StrOutputParser
from langchain_core.prompts import ChatPromptTemplate

MESSAGE = "다음 달 셋째 주에 강아지 한 마리랑 제주도 2박 3일 가고 싶어요. 숙소는 바다가 보이면 좋겠고, 렌터카도 필요해요."

# 갈래 1: 한 줄 요약. 가짜 모델이 정해 둔 요약을 돌려준다
summary_prompt = ChatPromptTemplate.from_messages([("system", "손님 글을 한 문장으로 요약한다."), ("human", "{message}")])
summary_chain = summary_prompt | FakeListChatModel(responses=["강아지와 함께 다음 달 제주도 2박 3일, 바다 전망 숙소와 렌터카를 원합니다."]) | StrOutputParser()

# 갈래 2: 키워드. 목록 파서의 형식 지시문을 시스템 메시지에 붙인다
list_parser = CommaSeparatedListOutputParser()
keyword_prompt = ChatPromptTemplate.from_messages([
    ("system", "손님 글에서 키워드를 다섯 개 뽑는다. {format}"),
    ("human", "{message}"),
]).partial(format=list_parser.get_format_instructions())
keyword_chain = keyword_prompt | FakeListChatModel(responses=["제주도, 2박 3일, 반려견, 바다 전망, 렌터카"]) | list_parser

print("요약:", summary_chain.invoke({"message": MESSAGE}))
print("키워드:", keyword_chain.invoke({"message": MESSAGE}))

from langchain_core.runnables import RunnableParallel

# 같은 입력 {"message": …} 가 두 갈래에 똑같이 들어가고, 결과는 사전 하나로 모인다
both = RunnableParallel(summary=summary_chain, keywords=keyword_chain)
result = both.invoke({"message": MESSAGE})

print(type(result).__name__, list(result))
print("요약:", result["summary"])
print("키워드:", result["keywords"], "→", len(result["keywords"]), "개")

from langchain_core.runnables import RunnableLambda, RunnablePassthrough


def make_ticket(d):
    """두 갈래의 결과와 원래 글을 상담 접수표 사전으로 정리한다."""
    return {
        "원래 글자 수": len(d["message"]),
        "요약": d["summary"],
        "키워드": d["keywords"],
        # 키워드에 반려동물 낱말이 있으면 반려동물 담당에게 넘긴다
        "담당": "반려동물" if any(k in ["반려견", "강아지", "고양이"] for k in d["keywords"]) else "일반",
    }


# assign: 입력 사전의 message 는 그대로 두고 summary, keywords 를 옆에 붙인다
# 그다음 RunnableLambda 로 접수표를 만든다
ticket_chain = RunnablePassthrough.assign(summary=summary_chain, keywords=keyword_chain) | RunnableLambda(make_ticket)
ticket = ticket_chain.invoke({"message": MESSAGE})

for key, value in ticket.items():
    print(f"{key}: {value}")

import json
from pathlib import Path

# stream: 요약이 다 될 때까지 기다리지 않고 조각이 생기는 대로 받는다
pieces = []
for chunk in summary_chain.stream({"message": MESSAGE}):
    pieces.append(chunk)
print("조각 수:", len(pieces))
print("앞의 다섯 조각:", pieces[:5])
# 조각을 모두 이어 붙이면 invoke 의 결과와 같다
print("이어 붙인 글:", "".join(pieces))

# 접수표를 json 으로 남긴다. ensure_ascii=False: 한글을 그대로 적는다
Path("out_ch03_ticket.json").write_text(json.dumps(ticket, ensure_ascii=False, indent=2), encoding="utf-8")
print("저장: out_ch03_ticket.json")