모델이 도구를 잘못 고르는 이유는 설명서에 있다

from langchain.tools import tool

# 1 이름: 함수 이름
# 2 설명: docstring 첫 문단
# 3 인자 설명: Args 칸
# 4 인자 종류: 타입 힌트
@tool(parse_docstring=True)
def get_ticket(ticket_id: str) -> str:
    """티켓 번호로 처리 상태와 도착 예정일을 조회한다.
    번호를 알 때만 쓴다. 번호를 모르면 list_my_tickets를 먼저 쓴다.

    Args:
        ticket_id: T-1042 같은 티켓 번호.
    """
    return lookup(ticket_id)

# 5 도구가 실패하면 예외 대신 오류 결과로 바꾼다
@wrap_tool_call
def tool_errors(request, handler):
    try:
        return handler(request)
    except Exception as e:
        return ToolMessage(
            content=str(e), status="error",
            tool_call_id=request.tool_call["id"])

# 6 에이전트에 끼워 넣는다
agent = create_agent(model, tools=[get_ticket],
                     middleware=[tool_errors])
# 7 도구 결과의 상태와 최종 답을 찍는다
result = agent.invoke({"messages": [{"role": "user", "content": "내 티켓 상태 알려줘"}]})
for m in result["messages"]:
    if m.type == "tool":
        print(m.status, m.content)
print(result["messages"][-1].content)

from dataclasses import dataclass
from langchain.tools import tool, ToolRuntime

# 8 로그인한 사용자 정보의 모양
@dataclass
class UserContext:
    user_id: str

# 9 runtime 인자는 모델에게 보이지 않는다
@tool
def list_my_tickets(runtime: ToolRuntime[UserContext]) -> str:
    """로그인한 사용자가 연 티켓 번호 목록을 돌려준다."""
    return ", ".join(find_tickets(runtime.context.user_id))

# 10 만들 때 모양을, 부를 때 값을 넘긴다
agent = create_agent(model, tools=[list_my_tickets],
                     context_schema=UserContext)
result = agent.invoke(messages, context=UserContext(user_id="kim"))
print(result["messages"][-1].content)