[PYTHON/LANGCHAIN] @chain 데코레이터 : 도구 내에서 실행 가능 항목 호출시 실행 가능 항목에 콜백 전파하기

■ @chain 데코레이터를 사용해 도구 내에서 실행 가능 항목 호출시 실행 가능 항목에 콜백을 전파하는 방법을 보여준다.

▶ main.py


import asyncio

from langchain_core.runnables import RunnableLambda
from langchain_core.runnables import chain

async def main():
    def reverseWord(word : str):
        return word[::-1]

    runnableLambda = RunnableLambda(reverseWord)

    @chain
    async def reverseDoubleTool(word : str):
        return await runnableLambda.ainvoke(word) * 2

    async for eventDictionary in reverseDoubleTool.astream_events("1234", version = "v2"):
        print(eventDictionary)

asyncio.run(main())

"""
{'event': 'on_chain_start', 'data': {'input': '1234'}, 'name': 'reverseDoubleTool', 'tags': [], 'run_id': '98e36f75-91ed-4f81-90a9-8a8fe1769a6f', 'metadata': {}, 'parent_ids': []}
{'event': 'on_chain_stream', 'run_id': '98e36f75-91ed-4f81-90a9-8a8fe1769a6f', 'name': 'reverseDoubleTool', 'tags': [], 'metadata': {}, 'data': {'chunk': '43214321'}, 'parent_ids': []}
{'event': 'on_chain_end', 'data': {'output': '43214321'}, 'run_id': '98e36f75-91ed-4f81-90a9-8a8fe1769a6f', 'name': 'reverseDoubleTool', 'tags': [], 'metadata': {}, 'parent_ids': []}
"""

import asyncio

from langchain_core.runnables import RunnableLambda

from langchain_core.runnables import chain

async def main():

def reverseWord(word : str):

return word[::-1]

runnableLambda = RunnableLambda(reverseWord)

@chain

async def reverseDoubleTool(word : str):

return await runnableLambda.ainvoke(word) * 2

async for eventDictionary in reverseDoubleTool.astream_events("1234", version = "v2"):

print(eventDictionary)

asyncio.run(main())

"""

{'event': 'on_chain_start', 'data': {'input': '1234'}, 'name': 'reverseDoubleTool', 'tags': [], 'run_id': '98e36f75-91ed-4f81-90a9-8a8fe1769a6f', 'metadata': {}, 'parent_ids': []}

{'event': 'on_chain_stream', 'run_id': '98e36f75-91ed-4f81-90a9-8a8fe1769a6f', 'name': 'reverseDoubleTool', 'tags': [], 'metadata': {}, 'data': {'chunk': '43214321'}, 'parent_ids': []}

{'event': 'on_chain_end', 'data': {'output': '43214321'}, 'run_id': '98e36f75-91ed-4f81-90a9-8a8fe1769a6f', 'name': 'reverseDoubleTool', 'tags': [], 'metadata': {}, 'parent_ids': []}

"""

▶ requirements.txt


aiohttp==3.9.5
aiosignal==1.3.1
annotated-types==0.7.0
async-timeout==4.0.3
attrs==23.2.0
certifi==2024.6.2
charset-normalizer==3.3.2
frozenlist==1.4.1
greenlet==3.0.3
idna==3.7
jsonpatch==1.33
jsonpointer==3.0.0
langchain==0.2.5
langchain-core==0.2.8
langchain-text-splitters==0.2.1
langsmith==0.1.79
multidict==6.0.5
numpy==1.26.4
orjson==3.10.5
packaging==24.1
pydantic==2.7.4
pydantic_core==2.18.4
PyYAML==6.0.1
requests==2.32.3
SQLAlchemy==2.0.30
tenacity==8.4.1
typing_extensions==4.12.2
urllib3==2.2.2
yarl==1.9.4

aiohttp==3.9.5

aiosignal==1.3.1

annotated-types==0.7.0

async-timeout==4.0.3

attrs==23.2.0

certifi==2024.6.2

charset-normalizer==3.3.2

frozenlist==1.4.1

greenlet==3.0.3

idna==3.7

jsonpatch==1.33

jsonpointer==3.0.0

langchain==0.2.5

langchain-core==0.2.8

langchain-text-splitters==0.2.1

langsmith==0.1.79

multidict==6.0.5

numpy==1.26.4

orjson==3.10.5

packaging==24.1

pydantic==2.7.4

pydantic_core==2.18.4

PyYAML==6.0.1

requests==2.32.3

SQLAlchemy==2.0.30

tenacity==8.4.1

typing_extensions==4.12.2

urllib3==2.2.2

yarl==1.9.4

※ pip install langchain 명령을 실행했다.