from dotenv import load_dotenv
from langchain_openai import ChatOpenAI
from langchain_community.agent_toolkits.load_tools import load_tools

load_dotenv()
llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)

# (1) 정답 — 계산기 + 위키백과 두 도구를 알린다
calculator, wikipedia = load_tools(["llm-math", "wikipedia"], llm=llm)
llm_with_tools = llm.bind_tools([calculator, wikipedia])

# (2) 정답 — 질문 종류에 따라 LLM 이 '어떤 도구'를 고르는지 본다 (아직 실행은 안 함)
for q in ["53 * 7 + 2 는?", "에펠탑은 누가 지었어?", "고마워!"]:
    calls = llm_with_tools.invoke(q).tool_calls
    picked = [c["name"] for c in calls] or ["(도구 안 씀 — 바로 답)"]
    print(f"{q:22} → {picked}")
