실습 3 — bind_tools 로 진짜 모델에게 도구 쥐여 주기

import os

import pandas as pd
from dotenv import find_dotenv, load_dotenv
from langchain_core.tools import tool
from langchain_openai import ChatOpenAI

# 과목 폴더의 .env 파일에서 OPENAI_API_KEY 를 읽어 환경 변수에 넣는다
# find_dotenv(usecwd=True): 지금 폴더(data)에서 위로 올라가며 .env 를 찾는다
load_dotenv(find_dotenv(usecwd=True))
# 모델 이름도 .env 에서 바꿀 수 있게 하고, 없으면 gpt-4.1-mini 를 쓴다
MODEL = os.getenv("OPENAI_MODEL", "gpt-4.1-mini")
model = ChatOpenAI(model=MODEL, temperature=0)

products = pd.read_csv("trip_products.csv")


@tool
def find_products(region: str = "", max_price: int = 0) -> str:
    """구름다리여행 상품을 지역과 가격(어른 1명, 만 원 상한)으로 찾는다. 0 이면 가격을 따지지 않는다."""
    df = products
    if region:
        df = df[df["지역"] == region]
    if max_price > 0:
        df = df[df["가격"] <= max_price]
    if df.empty:
        return "조건에 맞는 상품이 없습니다."
    return "\n".join(
        f"{r.상품코드} {r.상품명} | {r.가격}만 원 | 남은 자리 {r.남은자리} | 출발 {r.출발일}" for r in df.itertuples()
    )


@tool
def calc_price(code: str, adults: int, children: int = 0) -> str:
    """상품코드와 인원으로 총 여행 요금(만 원)을 계산한다. 아동 요금은 어른의 80% 이다."""
    rows = products[products["상품코드"] == code]
    if rows.empty:
        return f"없는 상품코드입니다: {code}"
    row = rows.iloc[0]
    total = round(row["가격"] * adults + row["가격"] * 0.8 * children, 1)
    return f"{code} {row['상품명']} · 어른 {adults}명, 아동 {children}명 · {float(total):g}만 원"


TOOLS = {t.name: t for t in [find_products, calc_price]}
# bind_tools: 도구 설명서를 모델에 묶은 새 모델을 만든다
model_with_tools = model.bind_tools(list(TOOLS.values()))
print("모델:", MODEL, "/ 묶은 도구:", list(TOOLS))

from langchain_core.messages import HumanMessage, SystemMessage

SYSTEM = SystemMessage("너는 구름다리여행의 상담 직원이다. 상품, 가격, 남은 자리는 반드시 도구로 확인하고 답한다.")
messages = [SYSTEM, HumanMessage("P01 상품에 어른 2명, 아이 1명이 가면 모두 얼마예요?")]

ai = model_with_tools.invoke(messages)
print("글 답:", repr(ai.content))
for call in ai.tool_calls:
    # id 는 부를 때마다 바뀌므로 이름과 인자만 찍는다
    print("도구 호출:", call["name"], call["args"])

messages.append(ai)                          # 도구 호출 요청도 기록에 넣는다
for call in ai.tool_calls:
    # 요청(call)을 통째로 넘기면 같은 id 를 단 ToolMessage 가 돌아온다
    result = TOOLS[call["name"]].invoke(call)
    print("도구 결과:", result.content)
    messages.append(result)

final = model_with_tools.invoke(messages)     # 둘째 호출: 결과를 읽고 답을 쓴다
print("남은 도구 호출:", final.tool_calls)
print("최종 답:", final.content)
print("메시지:", [m.type for m in messages + [final]])

def ask(question):
    """질문 하나로 모델을 부르고, 도구 요청이 있으면 한 번 실행한 뒤 최종 답을 받는다."""
    msgs = [SYSTEM, HumanMessage(question)]
    ai = model_with_tools.invoke(msgs)
    msgs.append(ai)
    for call in ai.tool_calls:
        print("  도구 호출:", call["name"], call["args"])
        msgs.append(TOOLS[call["name"]].invoke(call))
    answer = model_with_tools.invoke(msgs).content if ai.tool_calls else ai.content
    print(f"  도구 {len(ai.tool_calls)}개 → {answer}")


for q in ["P03 과 P05 중 어른 2명이면 어느 쪽이 더 싸요?", "안녕하세요. 상담 시간이 언제예요?"]:
    print("질문:", q)
    ask(q)