실습 1 — @tool 로 상품 찾기·요금 계산 도구 만들기

import pandas as pd

# 가상 상품 12개. 가격은 어른 1명 기준(만 원), 남은자리가 0 이면 매진
products = pd.read_csv("trip_products.csv")

print(products.shape)
print(products.to_string(index=False))

from langchain_core.tools import tool


@tool(parse_docstring=True)
def find_products(region: str = "", max_price: int = 0) -> str:
    """구름다리여행 상품을 지역과 가격으로 찾는다.

    Args:
        region: 지역 이름. 예: 제주, 부산. 비우면 모든 지역
        max_price: 어른 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(parse_docstring=True)
def calc_price(code: str, adults: int, children: int = 0) -> str:
    """상품코드와 인원으로 총 여행 요금(만 원)을 계산한다. 아동 요금은 어른의 80% 이다.

    Args:
        code: 상품코드. 예: P01
        adults: 어른 수
        children: 아동(만 2~12세 미만) 수
    """
    rows = products[products["상품코드"] == code]
    if rows.empty:
        return f"없는 상품코드입니다: {code}"
    row = rows.iloc[0]
    total = round(row["가격"] * adults + row["가격"] * 0.8 * children, 1)
    # :g 는 109.2 는 그대로, 36.0 은 36 으로 짧게 적는 형식이다
    return f"{code} {row['상품명']} · 어른 {adults}명, 아동 {children}명 · {float(total):g}만 원"


for t in [find_products, calc_price]:
    print("이름:", t.name)
    print("설명:", t.description)
    print("인자:", {k: v["type"] for k, v in t.args.items()})

# invoke 에 인자 사전을 넘기면 함수를 실행한 결과 글이 돌아온다
print(find_products.invoke({"region": "제주", "max_price": 40}))
print("---")
print(calc_price.invoke({"code": "P01", "adults": 2, "children": 1}))
print(calc_price.invoke({"code": "P03", "adults": 2}))
print(calc_price.invoke({"code": "P99", "adults": 1}))

import json
from pathlib import Path

from langchain_core.utils.function_calling import convert_to_openai_tool

# bind_tools 가 모델에게 보낼 설명서를 미리 만들어 본다
specs = [convert_to_openai_tool(t) for t in [find_products, calc_price]]
for spec in specs:
    fn = spec["function"]
    print(fn["name"], "| 꼭 넣을 인자:", fn["parameters"].get("required", []))
    for arg, info in fn["parameters"]["properties"].items():
        print(f"  {arg:10} {info['type']:8} {info['description']}")

Path("out_ch05_tool_schema.json").write_text(json.dumps(specs, ensure_ascii=False, indent=2), encoding="utf-8")
print("저장: out_ch05_tool_schema.json")