■ BaseCallbackHandler 클래스를 사용해 커스텀 콜백 핸들러를 만드는 방법을 보여준다.
※ OPENAI_API_KEY 환경 변수 값은 .env 파일에 정의한다.
▶ 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 |
from dotenv import load_dotenv from langchain_core.callbacks import BaseCallbackHandler from typing import Dict from typing import Any from typing import List from langchain_core.messages import BaseMessage from langchain_core.outputs import LLMResult from langchain_core.prompts import ChatPromptTemplate from langchain_openai import ChatOpenAI load_dotenv() class CustomCallbackHandler(BaseCallbackHandler): def on_chat_model_start(self, serializedDictionary : Dict[str, Any], messageListList : List[List[BaseMessage]], **keywordArgumentDictionary) -> None: print("on_chat_model_start") print() def on_llm_end(self, result : LLMResult, **keywordArgumentDictionary) -> None: print(f"on_llm_end : {result}") print() def on_chain_start(self, serializedDictionary : Dict[str, Any], inputDictionary : Dict[str, Any], **keywordArgumentDictionary) -> None: if serializedDictionary: print(f"on_chain_start : {serializedDictionary.get("name")}") else: print(f"on_chain_start : {serializedDictionary}") print() def on_chain_end(self, outputDictionary : Dict[str, Any], **keywordArgumentDictionary) -> None: print(f"on_chain_end : {outputDictionary}") print() chatPromptTemplate = ChatPromptTemplate.from_template("What is 1 + {number}?") chatOpenAI = ChatOpenAI(model = "gpt-4o") runnableSequence = chatPromptTemplate | chatOpenAI callBackHandlerList = [CustomCallbackHandler()] responseAIMessage = runnableSequence.invoke({"number" : "2"}, config = {"callbacks" : callBackHandlerList}) print("-" * 100) print(responseAIMessage) |
▶ 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 39 40 41 42 43 44 45 46 47 48 |
aiohappyeyeballs==2.4.3 aiohttp==3.10.10 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 colorama==0.4.6 distro==1.9.0 frozenlist==1.5.0 greenlet==3.1.1 h11==0.14.0 httpcore==1.0.6 httpx==0.27.2 idna==3.10 jiter==0.7.0 jsonpatch==1.33 jsonpointer==3.0.0 langchain==0.3.7 langchain-core==0.3.15 langchain-openai==0.2.6 langchain-text-splitters==0.3.2 langsmith==0.1.141 multidict==6.1.0 numpy==1.26.4 openai==1.54.3 orjson==3.10.11 packaging==24.2 propcache==0.2.0 pydantic==2.9.2 pydantic_core==2.23.4 python-dotenv==1.0.1 PyYAML==6.0.2 regex==2024.11.6 requests==2.32.3 requests-toolbelt==1.0.0 six==1.16.0 sniffio==1.3.1 SQLAlchemy==2.0.36 tenacity==9.0.0 tiktoken==0.8.0 tqdm==4.67.0 typing_extensions==4.12.2 urllib3==2.2.3 yarl==1.17.1 |
※ pip install python-dotenv langchain langchain-openai six 명령을 실행했다.