LangChain 그래프 시각화 오류 해결 방법

---

LangChain 그래프 시각화 오류 해결 방법

LangChain에서 그래프를 시각화할 때, 보통 아래와 같이 코드를 실행한다.

from IPython.display import display, Image
display(Image(graph.get_graph().draw_mermaid_png()))

하지만 이 방식은 내부적으로 mermaid.ink API를 호출하기 때문에, 간헐적으로 500 오류 등 외부 API 문제로 인해 다음과 같은 ValueError가 발생할 수 있다.

ValueError: Failed to reach https://mermaid.ink/ API while trying to render your graph. Status code: 500.
...

이 문제를 피하기 위해 draw_mermaid_png(draw_method=MermaidDrawMethod.PYPPETEER) 옵션을 사용해 로컬에서 직접 렌더링을 시도하면, 또 다른 오류(OSError, RuntimeError)가 발생할 수 있다. 이 글에서는 이러한 오류의 원인과, 로컬 렌더링(PYPPETEER) 환경에서 발생하는 문제를 단계별로 해결하는 방법을 설명한다.

---

1. 문제 분석: 복합적인 오류의 원인

LangChain의 draw_mermaid_png() 함수는 기본적으로 외부 API(MermaidDrawMethod.API)를 사용하거나, 로컬에서 pyppeteer를 사용하여 그래프를 렌더링한다. 그러나 두 방식 모두 다음과 같은 문제점을 내포하고 있다.

• MermaidDrawMethod.API 오류: https://mermaid.ink API 서버가 불안정하거나 접속할 수 없을 때 ValueError가 발생한다.
• MermaidDrawMethod.PYPPETEER 오류: 이 방식은 pyppeteer를 사용한다. pyppeteer는 내부적으로 Chromium 브라우저에 의존한다. 여기서 두 가지 주요 오류가 발생한다.
  • Chromium 다운로드 오류 (OSError): pyppeteer가 특정 버전의 Chromium을 다운로드하려 하지만, 해당 URL에 파일이 없어서 발생한다. 이는 pyppeteer가 최신 버전의 Chromium을 자동으로 참조하지 못하기 때문에 발생한다.
  • 비동기 이벤트 루프 오류 (RuntimeError): Jupyter Notebook과 같은 환경은 이미 비동기 이벤트 루프가 실행 중이다. 이때 pyppeteer가 내부적으로 asyncio.run()을 호출하면, 이벤트 루프가 중첩되어 RuntimeError가 발생한다.

이러한 문제들은 단순히 한두 가지 설정을 변경하는 것으로는 해결되지 않으며, 두 가지 오류를 모두 해결하는 복합적인 접근이 필요하다.

---

2. 해결 방법: 두 가지 문제에 대한 동시 해결

1단계: 비동기 이벤트 루프 충돌 해결

먼저 RuntimeError를 해결해야 한다. nest_asyncio 라이브러리를 사용하면 이미 실행 중인 이벤트 루프 내에서 새로운 이벤트 루프를 시작할 수 있게 되어 이 문제를 해결할 수 있다.

1. nest_asyncio 설치:

   pip install nest_asyncio

2. nest_asyncio 적용:

   코드를 실행하기 전에 nest_asyncio.apply()를 호출한다.

   import nest_asyncio
   nest_asyncio.apply()
   
   # ... 나머지 코드

2단계: Pyppeteer Chromium 다운로드 오류 해결

다음으로 OSError를 해결해야 한다. pyppeteer가 Chromium을 다운로드하지 않고, 사용자의 시스템에 이미 설치된 Chrome 또는 Edge 브라우저를 사용하도록 강제하는 방법이다.

pyppeteer는 환경 변수 설정만으로는 내부 다운로드 로직을 건너뛰지 못한다. 따라서 langchain_core 라이브러리의 소스 코드를 직접 수정해야 한다.

1. graph_mermaid.py 파일 찾기:

   langchain_core 라이브러리가 설치된 경로에서 graph_mermaid.py 파일을 찾는다.

   예시 경로: [venv 경로]\Lib\site-packages\langchain_core\runnables\graph_mermaid.py

2. _render_mermaid_using_pyppeteer 함수 수정:

   파일을 열고 _render_mermaid_using_pyppeteer 함수 내부에 있는 pyppeteer.launch() 호출 부분을 찾아 executablePath 인자를 추가한다.

   # 기존 코드
   # browser = await launch()
   
   # 수정 코드
   # Windows 환경 예시
   browser = await launch(executablePath=r'C:\Program Files\Google\Chrome\Application\chrome.exe')
   # macOS 환경 예시
   # browser = await launch(executablePath='/Applications/Google Chrome.app/Contents/MacOS/Google Chrome')

   크롬 실행 파일 경로를 정확하게 지정해야 한다.

---

3. 최종 코드

위 두 가지 해결책을 모두 적용하면 LangChain의 그래프 시각화 오류를 안정적으로 해결할 수 있다.

import nest_asyncio
from IPython.display import display, Image
from langchain_core.runnables.graph_mermaid import MermaidDrawMethod

# 1. 비동기 이벤트 루프 중첩 오류 해결
nest_asyncio.apply()

# 2. (라이브러리 파일 수정 후) 그래프 시각화
display(Image(graph.get_graph().draw_mermaid_png(draw_method=MermaidDrawMethod.PYPPETEER)))

이 방법은 라이브러리 소스 코드를 직접 수정하기 때문에, 라이브러리를 업데이트할 때마다 동일한 수정을 다시 적용해야 하는 번거로움이 있다. 하지만 현재 시점에서 가장 확실하게 문제를 해결하고 그래프를 성공적으로 렌더링할 수 있는 방법이다.

---