■ BaseOutputParser 클래스를 사용해 커스텀 출력 파서를 만드는 방법을 보여준다.
▶ 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 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
from langchain_core.output_parsers import BaseOutputParser from langchain_core.exceptions import OutputParserException # [bool]은 제네릭의 매개변수화를 설명한다. # 기본적으로 parse의 반환 유형이 무엇인지 나타낸다. 이 경우 반환 유형은 True 또는 False이다. class CustomOutputParser(BaseOutputParser[bool]): """Custom boolean parser.""" trueString : str = "YES" falseString : str = "NO" def parse(self, text : str) -> bool: textCleaned = text.strip().upper() if textCleaned not in (self.trueString.upper(), self.falseString.upper()): raise OutputParserException( f"BooleanOutputParser expected output value to either be " f"{self.trueString} or {self.falseString} (case-insensitive). " f"Received {textCleaned}." ) return textCleaned == self.trueString.upper() @property def _type(self) -> str: return "custom_output_parser" customOutputParser = CustomOutputParser() responseValue1 = customOutputParser.invoke("YES") print(responseValue1) print("-" * 50) try: customOutputParser.invoke("MEOW") except Exception as exception: print(f"Triggered an exception of type: {type(exception)}") print("-" * 50) customOutputParser2 = CustomOutputParser(trueString = "OKAY") responseValue2 = customOutputParser2.invoke("OKAY") print(responseValue2) print("-" * 50) responseValueList = customOutputParser2.batch(["OKAY", "NO"]) print(responseValueList) print("-" * 50) """ True -------------------------------------------------- Triggered an exception of type: <class 'langchain_core.exceptions.OutputParserException'> -------------------------------------------------- True -------------------------------------------------- [True, False] -------------------------------------------------- """ |
▶ 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.7 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.8 langchain-core==0.3.21 langchain-text-splitters==0.3.2 langsmith==0.1.145 multidict==6.1.0 numpy==1.26.4 orjson==3.10.12 packaging==24.2 propcache==0.2.0 pydantic==2.10.1 pydantic_core==2.27.1 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.18.0 |
※ pip install langchain 명령을 실행했다.