에이전트 시각화
에이전트 시각화는Graphviz를 사용하여 에이전트와 그 관계를 구조화된 그래프로 생성할 수 있게 합니다. 이는 애플리케이션 내에서 에이전트, 도구, 핸드오프가 어떻게 상호작용하는지 이해하는 데 유용합니다.
설치
선택적viz 의존성 그룹을 설치하세요:
그래프 생성
draw_graph 함수를 사용하여 에이전트 시각화를 생성할 수 있습니다. 이 함수는 다음과 같은 방향성 그래프를 만듭니다:
- 에이전트는 노란색 상자로 표시
- MCP 서버는 회색 상자로 표시
- 도구는 녹색 타원으로 표시
- 핸드오프는 한 에이전트에서 다른 에이전트로의 방향성 간선
사용 예시
importosfromagentsimportAgent,function_toolfromagents.mcp.serverimportMCPServerStdiofromagents.extensions.visualizationimportdraw_graph@function_tooldefget_weather(city:str)->str:returnf"The weather in{city} is sunny."spanish_agent=Agent(name="Spanish agent",instructions="You only speak Spanish.",)english_agent=Agent(name="English agent",instructions="You only speak English",)current_dir=os.path.dirname(os.path.abspath(__file__))samples_dir=os.path.join(current_dir,"sample_files")mcp_server=MCPServerStdio(name="Filesystem Server, via npx",params={"command":"npx","args":["-y","@modelcontextprotocol/server-filesystem",samples_dir],},)triage_agent=Agent(name="Triage agent",instructions="Handoff to the appropriate agent based on the language of the request.",handoffs=[spanish_agent,english_agent],tools=[get_weather],mcp_servers=[mcp_server],)draw_graph(triage_agent)
이는triage 에이전트와 하위 에이전트 및 도구와의 연결 구조를 시각적으로 나타내는 그래프를 생성합니다.
시각화 이해
생성된 그래프에는 다음이 포함됩니다:
- 진입점을 나타내는start 노드(
__start__) - 노란색 채움의사각형으로 표시된 에이전트
- 녹색 채움의타원으로 표시된 도구
- 회색 채움의사각형으로 표시된 MCP 서버
- 상호작용을 나타내는 방향성 간선:
- 에이전트 간 핸드오프는실선 화살표
- 도구 호출은점선 화살표
- MCP 서버 호출은파선 화살표
- 실행 종료 지점을 나타내는end 노드(
__end__)
참고: MCP 서버는 최신 버전의agents 패키지에서 렌더링됩니다(v0.2.8 에서 확인). 시각화에 MCP 상자가 보이지 않는 경우 최신 릴리스로 업그레이드하세요.
그래프 사용자 지정
그래프 표시
기본적으로draw_graph는 그래프를 인라인으로 표시합니다. 그래프를 별도의 창에서 보려면 다음을 작성하세요:
그래프 저장
기본적으로draw_graph는 그래프를 인라인으로 표시합니다. 파일로 저장하려면 파일 이름을 지정하세요:
그러면 작업 디렉터리에agent_graph.png가 생성됩니다.