[PYTHON/LANGCHAIN] BaseOutputParser 클래스 : 커스텀 논리값 출력 파서 만들기

■ BaseOutputParser 클래스를 사용해 커스텀 논리값 출력 파서를 만드는 방법을 보여준다.

▶ main.py


from langchain_core.exceptions     import OutputParserException
from langchain_core.output_parsers import BaseOutputParser

# [bool]에서는 제네릭의 매개변수화에 대해 설명한다.
# 기본적으로 구문 분석의 반환 유형이 무엇인지 나타낸다.
# 이 경우 반환 유형은 True 또는 False이다.
class BooleanOutputParser(BaseOutputParser[bool]):
    """사용자 논리값 파서"""

    true_value  : str = "YES"
    false_value : str = "NO"

    def parse(self, sourceText : str) -> bool:
        targetText = sourceText.strip().upper()

        if targetText not in (self.true_value.upper(), self.false_value.upper()):
            raise OutputParserException(
                f"BooleanOutputParser expected output value to either be "
                f"{self.true_value} or {self.false_value} (case-insensitive). "
                f"Received {targetText}."
            )

        return targetText == self.true_value.upper()

    @property
    def _type(self) -> str:
        return "boolean_output_parser"

booleanOutputParser1 = BooleanOutputParser()

responseBool1 = booleanOutputParser1.invoke("yes")

print(responseBool1)

"""
True
"""

booleanOutputParser2 = BooleanOutputParser(true_value = "ON", false_value = "OFF")

responseBool2 = booleanOutputParser2.invoke("on")

print(responseBool2)

"""
True
"""

booleanOutputParser3 = BooleanOutputParser(true_value = "OPEN", false_value = "CLOSED")

responseBoolList3 = booleanOutputParser3.batch(["open", "closed"])

print(responseBoolList3)

"""
[True, False]
"""

from langchain_core.exceptions import OutputParserException

from langchain_core.output_parsers import BaseOutputParser

# [bool]에서는 제네릭의 매개변수화에 대해 설명한다.

# 기본적으로 구문 분석의 반환 유형이 무엇인지 나타낸다.

# 이 경우 반환 유형은 True 또는 False이다.

class BooleanOutputParser(BaseOutputParser[bool]):

"""사용자 논리값 파서"""

true_value : str = "YES"

false_value : str = "NO"

def parse(self, sourceText : str) -> bool:

targetText = sourceText.strip().upper()

if targetText not in (self.true_value.upper(), self.false_value.upper()):

raise OutputParserException(

f"BooleanOutputParser expected output value to either be "

f"{self.true_value} or {self.false_value} (case-insensitive). "

f"Received {targetText}."

)

return targetText == self.true_value.upper()

@property

def _type(self) -> str:

return "boolean_output_parser"

booleanOutputParser1 = BooleanOutputParser()

responseBool1 = booleanOutputParser1.invoke("yes")

print(responseBool1)

"""

True

"""

booleanOutputParser2 = BooleanOutputParser(true_value = "ON", false_value = "OFF")

responseBool2 = booleanOutputParser2.invoke("on")

print(responseBool2)

"""

True

"""

booleanOutputParser3 = BooleanOutputParser(true_value = "OPEN", false_value = "CLOSED")

responseBoolList3 = booleanOutputParser3.batch(["open", "closed"])

print(responseBoolList3)

"""

[True, False]

"""

▶ 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.6
langchain-core==0.2.10
langchain-text-splitters==0.2.2
langsmith==0.1.82
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.31
tenacity==8.4.2
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.6

langchain-core==0.2.10

langchain-text-splitters==0.2.2

langsmith==0.1.82

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.31

tenacity==8.4.2

typing_extensions==4.12.2

urllib3==2.2.2

yarl==1.9.4

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