더북(TheBook)

args_schema에 Pydantic 모델 사용하기

첫 번째로, Pydantic 모델을 사용하면 Tool 입력 구조를 별도의 클래스로 정의할 수 있습니다. 이때 Field()를 사용해 각 입력값에 설명, 기본값, 길이 제한 같은 추가 설정을 함께 지정할 수 있습니다.

RewriteInput 클래스는 Tool이 받을 입력값 구조를 정의하는 Pydantic 모델입니다. 이 모델을 @tool(args_schema=RewriteInput)에 전달하면 랭체인은 RewriteInput을 기준으로 Tool 입력 구조를 생성합니다.