[Python] 랭체인 @tool 사용하기
Posted by Albert 17Day 6Hour 10Min 58Sec ago [2025-11-11]
랭체인도 GPT처럼 현재 시점 시간, 날씨 및 최신 정보를 가져올수없고 GPT의 Function calling 처럼
@tool 이란 기능을 통하여 개발자가 만든 특정함수를 실행하여 원하는 최신정보를 제공할수있게 한다.
사용예제
from dotenv import load_dotenv
load_dotenv()
from langchain_openai import ChatOpenAI
from langchain_core.messages import HumanMessage
from langchain_core.tools import tool
from datetime import datetime
from langchain_core.messages import SystemMessage
import pytz
llm = ChatOpenAI(model="gpt-4o-mini")
@tool ' @tool 데코레이터를 사용하여 함수를 도구로 등록
def get_current_time(timezone: str, location: str) -> str:
""" 현재 시각을 반환하는 함수
Args:
timezone (str): 타임존 (예: 'Asia/Seoul') 실제 존재하는 타임존이어야 함
location (str): 지역명. 타임존이 모든 지명에 대응되지 않기 때문에 이후 llm 답변 생성에 사용됨
"""
tz = pytz.timezone(timezone)
now = datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
location_and_local_time = f'{timezone} ({location}) 현재시각 {now} ' ' 타임존, 지역명, 현재시각을 문자열로 반환
print(location_and_local_time)
return location_and_local_time
' 도구를 tools 리스트에 추가하고, tool_dict에도 추가
tools = [get_current_time,]
tool_dict = {"get_current_time": get_current_time,}
' 도구를 모델에 바인딩: 모델에 도구를 바인딩하면, 도구를 사용하여 llm 답변을 생성할 수 있음
llm_with_tools = llm.bind_tools(tools)
messages = [
SystemMessage("너는 사용자의 질문에 답변을 하기 위해 tools를 사용할 수 있다."),
HumanMessage("서울은 지금 몇시야?"),
]
' (5) llm_with_tools를 사용하여 사용자의 질문에 대한 llm 답변 생성
response = llm_with_tools.invoke(messages)
messages.append(response)
' (6) 생성된 llm 답변 출력
for tool_call in response.tool_calls:
selected_tool = tool_dict[tool_call["name"]] ' (7) tool_dict를 사용하여 도구 함수를 선택
print(tool_call["args"]) ' (8) 도구 호출 시 전달된 인자 출력
tool_msg = selected_tool.invoke(tool_call) ' (9) 도구 함수를 호출하여 결과를 반환
messages.append(tool_msg)
messages
결과
Connected to pydev debugger (build 241.19416.15)
{'timezone': 'Asia/Seoul', 'location': '서울'}
Asia/Seoul (서울) 현재시각 2025-11-11 16:48:11