■ BaseCallbackHandler 클래스의 on_custom_event 메소드를 재정의해서 커스텀 이벤트를 처리하는 방법을 보여준다.
▶ main.py
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
import asyncio from langchain_core.callbacks import BaseCallbackHandler from typing import Any from uuid import UUID from typing import Optional from typing import List from typing import Dict from langchain_core.callbacks.manager import dispatch_custom_event from langchain_core.runnables import RunnableLambda from langchain_core.runnables.config import RunnableConfig class CustomCallbackHandler(BaseCallbackHandler): def on_custom_event( self, name : str, data : Any, *, run_id : UUID, tags : Optional[List[str]] = None, metadata : Optional[Dict[str, Any]] = None, **kwargs : Any ) -> None: print(f"Received event {name} with data : {data}, with tags : {tags}, with metadata : {metadata} and run_id : {run_id}") @RunnableLambda def testRunnableLambda(x : int, config : RunnableConfig) -> int: dispatch_custom_event("event1", {"x" : x}) dispatch_custom_event("event2", {"x" : x}) return x customCallbackHandler = CustomCallbackHandler() async def main(): responseValue = testRunnableLambda.invoke(1, {"callbacks" : [customCallbackHandler], "tags" : ["foo", "bar"]}) print(responseValue) asyncio.run(main()) """ Received event event1 with data : {'x': 1}, with tags : ['foo', 'bar'], with metadata : {} and run_id : 745a81f9-b39a-4003-8a09-f4b0592a5be8 Received event event2 with data : {'x': 1}, with tags : ['foo', 'bar'], with metadata : {} and run_id : 745a81f9-b39a-4003-8a09-f4b0592a5be8 1 """ |
▶ requirements.txt
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
aiohappyeyeballs==2.4.3 aiohttp==3.11.6 aiosignal==1.3.1 annotated-types==0.7.0 anyio==4.6.2.post1 attrs==24.2.0 certifi==2024.8.30 charset-normalizer==3.4.0 frozenlist==1.5.0 greenlet==3.1.1 h11==0.14.0 httpcore==1.0.7 httpx==0.27.2 idna==3.10 jsonpatch==1.33 jsonpointer==3.0.0 langchain==0.3.7 langchain-core==0.3.19 langchain-text-splitters==0.3.2 langsmith==0.1.143 multidict==6.1.0 numpy==1.26.4 orjson==3.10.11 packaging==24.2 propcache==0.2.0 pydantic==2.9.2 pydantic_core==2.23.4 PyYAML==6.0.2 requests==2.32.3 requests-toolbelt==1.0.0 sniffio==1.3.1 SQLAlchemy==2.0.36 tenacity==9.0.0 typing_extensions==4.12.2 urllib3==2.2.3 yarl==1.17.2 |
※ pip install langchain 명령을 실행했다.