[PYTHON/LANGCHAIN] CompiledStateGraph 클래스 : stream 메소드에서 재귀 제한 수 설정하기

■ CompiledStateGraph 클래스의 stream 메소드에서 재귀 제한 수를 설정하는 방법을 보여준다.

※ OPENAI_API_KEY 환경 변수 값은 .env 파일에 정의한다.

▶ main.py


from dotenv               import load_dotenv
from langchain_openai     import ChatOpenAI
from langchain_core.tools import tool
from langgraph.prebuilt   import create_react_agent
from langgraph.errors     import GraphRecursionError

load_dotenv()

chatOpenAI = ChatOpenAI(model = "gpt-4o")

@tool
def magicFunction(input : int) -> int:
    """Applies a magic function to an input."""
    return input + 2

toolList = [magicFunction]

query = "what is the value of magicFunction(3)?"

RECURSION_LIMIT_COUNT = 2 * 3 + 1

compiledStateGraph = create_react_agent(chatOpenAI, tools = toolList)

try:
    for addableValuesDict in compiledStateGraph.stream(
        {"messages" : [("human", query)]},
        {"recursion_limit" : RECURSION_LIMIT_COUNT},
        stream_mode = "values"
    ):
        print(addableValuesDict["messages"][-1])
except GraphRecursionError:
    print({"input" : query, "output" : "Agent stopped due to max iterations."})

from dotenv import load_dotenv

from langchain_openai import ChatOpenAI

from langchain_core.tools import tool

from langgraph.prebuilt import create_react_agent

from langgraph.errors import GraphRecursionError

load_dotenv()

chatOpenAI = ChatOpenAI(model = "gpt-4o")

@tool

def magicFunction(input : int) -> int:

"""Applies a magic function to an input."""

return input + 2

toolList = [magicFunction]

query = "what is the value of magicFunction(3)?"

RECURSION_LIMIT_COUNT = 2 * 3 + 1

compiledStateGraph = create_react_agent(chatOpenAI, tools = toolList)

try:

for addableValuesDict in compiledStateGraph.stream(

{"messages" : [("human", query)]},

{"recursion_limit" : RECURSION_LIMIT_COUNT},

stream_mode = "values"

print(addableValuesDict["messages"][-1])

except GraphRecursionError:

print({"input" : query, "output" : "Agent stopped due to max iterations."})

▶ requirements.txt


aiohappyeyeballs==2.4.3
aiohttp==3.10.10
aiosignal==1.3.1
annotated-types==0.7.0
anyio==4.6.2
async-timeout==4.0.3
attrs==24.2.0
certifi==2024.8.30
charset-normalizer==3.4.0
distro==1.9.0
exceptiongroup==1.2.2
frozenlist==1.4.1
greenlet==3.1.1
h11==0.14.0
httpcore==1.0.6
httpx==0.27.2
idna==3.10
jiter==0.6.1
jsonpatch==1.33
jsonpointer==3.0.0
langchain==0.3.3
langchain-core==0.3.10
langchain-openai==0.2.2
langchain-text-splitters==0.3.0
langgraph==0.2.35
langgraph-checkpoint==2.0.1
langsmith==0.1.134
msgpack==1.1.0
multidict==6.1.0
numpy==1.26.4
openai==1.51.2
orjson==3.10.7
packaging==24.1
propcache==0.2.0
pydantic==2.9.2
pydantic_core==2.23.4
python-dotenv==1.0.1
PyYAML==6.0.2
regex==2024.9.11
requests==2.32.3
requests-toolbelt==1.0.0
sniffio==1.3.1
SQLAlchemy==2.0.35
tenacity==8.5.0
tiktoken==0.8.0
tqdm==4.66.5
typing_extensions==4.12.2
urllib3==2.2.3
yarl==1.15.2

aiohappyeyeballs==2.4.3

aiohttp==3.10.10

aiosignal==1.3.1

annotated-types==0.7.0

anyio==4.6.2

async-timeout==4.0.3

attrs==24.2.0

certifi==2024.8.30

charset-normalizer==3.4.0

distro==1.9.0

exceptiongroup==1.2.2

frozenlist==1.4.1

greenlet==3.1.1

h11==0.14.0

httpcore==1.0.6

httpx==0.27.2

idna==3.10

jiter==0.6.1

jsonpatch==1.33

jsonpointer==3.0.0

langchain==0.3.3

langchain-core==0.3.10

langchain-openai==0.2.2

langchain-text-splitters==0.3.0

langgraph==0.2.35

langgraph-checkpoint==2.0.1

langsmith==0.1.134

msgpack==1.1.0

multidict==6.1.0

numpy==1.26.4

openai==1.51.2

orjson==3.10.7

packaging==24.1

propcache==0.2.0

pydantic==2.9.2

pydantic_core==2.23.4

python-dotenv==1.0.1

PyYAML==6.0.2

regex==2024.9.11

requests==2.32.3

requests-toolbelt==1.0.0

sniffio==1.3.1

SQLAlchemy==2.0.35

tenacity==8.5.0

tiktoken==0.8.0

tqdm==4.66.5

typing_extensions==4.12.2

urllib3==2.2.3

yarl==1.15.2

※ pip install python-dotenv langchain langchain-openai langgraph 명령을 실행했다.