여러 도구 사용하기

from datetime import datetime, timedelta

def add_duration_to_datetime(
    datetime_str, duration=0, unit="days", input_format="%Y-%m-%d"
):
    date = datetime.strptime(datetime_str, input_format)
    if unit == "days":
        new_date = date + timedelta(days=duration)
    # … seconds / minutes / hours / weeks / months / years …
    return new_date.strftime("%A, %B %d, %Y %I:%M:%S %p")

def set_reminder(content, timestamp):
    # 데모용 — 실제 예약 없이 인자만 출력
    print(f"----\nSetting the following reminder for {timestamp}:\n{content}\n----")

def run_conversation(messages):
    while True:
        response = chat(
            messages,
            tools=[
                get_current_datetime_schema,
                add_duration_to_datetime_schema,   # ← 추가
                set_reminder_schema,              # ← 추가
            ],
        )

        add_assistant_message(messages, response)
        print(text_from_message(response))

        if response.stop_reason != "tool_use":
            break

        tool_results = run_tools(response)
        add_user_message(messages, tool_results)

    return messages

def run_tool(tool_name, tool_input):
    if tool_name == "get_current_datetime":
        return get_current_datetime(**tool_input)
    elif tool_name == "add_duration_to_datetime":   # ← 추가
        return add_duration_to_datetime(**tool_input)
    elif tool_name == "set_reminder":               # ← 추가
        return set_reminder(**tool_input)

messages = []
add_user_message(
    messages,
    "Set a reminder for my doctors appointment. "
    "Its 177 days after Jan 1st, 2050.",
)
run_conversation(messages)

# 1) Claude: 먼저 날짜를 구해야 한다 → add_duration_to_datetime
# 2) set_reminder 의 print 로그 ↓
----
Setting the following reminder for 2050-06-27T00:00:00:
Doctor's appointment
----
# 3) Claude 최종 응답
Your appointment reminder is set for Monday, June 27, 2050.