[React] agent 에 structured output 설정하기

---

• 개요
  • Tools 정의
  • 구조화된 출력 스키마 정의
• 질문
• Prompt 와 함께 사용하기
• Reference

개요

create_react_agent() 에 response_format 파라미터로 출력 스키마를 지정하면, ReAct 에이전트로부터 구조화된 응답을 받을 수 있다.

• pre-built 된 ReAct agent 는 ReAct 루프 종료 시, LLM을 한; 번 더 호출하여 구조화된 결과를 생성한다.

  Tools 정의

from langchain_core.tools import tool

@tool
def get_weather(location:str) -> str:
    """Use this to get weather information"""
    if any([city in location.lower() for city in ['nyc','new york city']]):
        return "It might be cloudy in nyc"
    elif any([city in location.lower() for city in ['sf', 'san francisco']]):
        return "It's always sunny in sf"
    else:
        return f"I am not sure what the weather is in {location}"

tools = [get_weather]

구조화된 출력 스키마 정의

from pydantic import BaseModel, Field

class WeatherResponse(BaseModel):
	"""Response This Format"""
	conditions: str = Field(description="Weather Conditions")

질문

inputs = {"messages": [("user", "What's the weather in NYC?")]}
response = graph.invoke(inputs)
response['structured_response']

WeatherResponse(conditions='cloudy')

Prompt 와 함께 사용하기

graph = create_react_agent(
	model,
	tools=tools,
	response_format=("Always return capitalized weather conditions", WeatherResponse)
)

inputs = {"messages": [("user", "What's the weather in NYC?")]}
response = graph.invoke(inputs)

WeatherResponse(conditions='CLOUDY')

• Prompt 를 따라 대문자로 Return 해준다.

• 코드 주소

---

Reference

• How to return structured output from the prebuilt ReAct agent